added redirect anchors
This commit is contained in:
parent
5e5b73884b
commit
b13ff70a2d
4 changed files with 44 additions and 18 deletions
|
|
@ -7,14 +7,9 @@ import (
|
|||
"encoding/xml"
|
||||
"net/http"
|
||||
"sort"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type dirContents struct {
|
||||
AllPaths []string `xml:"a"`
|
||||
Dirs []string `xml:"-"`
|
||||
Files []string `xml:"-"`
|
||||
}
|
||||
|
||||
type statusInterceptor struct {
|
||||
w http.ResponseWriter
|
||||
status int
|
||||
|
|
@ -23,13 +18,16 @@ type statusInterceptor struct {
|
|||
func (s *statusInterceptor) Header() http.Header {
|
||||
return s.w.Header()
|
||||
}
|
||||
|
||||
func (s *statusInterceptor) Write(p []byte) (int, error) {
|
||||
return s.w.Write(p)
|
||||
}
|
||||
|
||||
func (s *statusInterceptor) WriteHeader(statusCode int) {
|
||||
s.w.WriteHeader(statusCode)
|
||||
s.status = statusCode
|
||||
}
|
||||
|
||||
func (s *statusInterceptor) Status() int {
|
||||
if s.status == 0 {
|
||||
return http.StatusOK
|
||||
|
|
@ -37,6 +35,23 @@ func (s *statusInterceptor) Status() int {
|
|||
return s.status
|
||||
}
|
||||
|
||||
type dirContents struct {
|
||||
AllPaths []string `xml:"a"`
|
||||
Dirs []string `xml:"-"`
|
||||
Files files `xml:"-"`
|
||||
}
|
||||
|
||||
type file struct {
|
||||
Name string
|
||||
Anchor string
|
||||
}
|
||||
|
||||
type files []file
|
||||
|
||||
func (f files) Len() int { return len(f) }
|
||||
func (f files) Less(i, j int) bool { return f[i].Name < f[j].Name }
|
||||
func (f files) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
|
||||
|
||||
type responseInterceptor struct {
|
||||
buf bytes.Buffer
|
||||
w http.ResponseWriter
|
||||
|
|
@ -46,18 +61,22 @@ type responseInterceptor struct {
|
|||
func (r *responseInterceptor) Header() http.Header {
|
||||
return r.w.Header()
|
||||
}
|
||||
|
||||
func (r *responseInterceptor) Write(p []byte) (int, error) {
|
||||
return r.buf.Write(p)
|
||||
}
|
||||
|
||||
func (r *responseInterceptor) WriteHeader(statusCode int) {
|
||||
r.status = statusCode
|
||||
}
|
||||
|
||||
func (r *responseInterceptor) Status() int {
|
||||
if r.status == 0 {
|
||||
return http.StatusOK
|
||||
}
|
||||
return r.status
|
||||
}
|
||||
|
||||
func (r *responseInterceptor) GetPaths() (dir dirContents, err error) {
|
||||
err = xml.Unmarshal(r.buf.Bytes(), &dir)
|
||||
if err != nil {
|
||||
|
|
@ -67,10 +86,18 @@ func (r *responseInterceptor) GetPaths() (dir dirContents, err error) {
|
|||
if p[len(p)-1] == '/' {
|
||||
dir.Dirs = append(dir.Dirs, p)
|
||||
} else {
|
||||
dir.Files = append(dir.Files, p)
|
||||
dir.Files = append(dir.Files, file{Name: p, Anchor: anchor(p)})
|
||||
}
|
||||
}
|
||||
sort.Strings(dir.Dirs)
|
||||
sort.Strings(dir.Files)
|
||||
sort.Sort(dir.Files)
|
||||
return
|
||||
}
|
||||
|
||||
func anchor(s string) string {
|
||||
if i := strings.LastIndex(s, "/"); i > 0 && i+1 < len(s) {
|
||||
s = s[i+1:]
|
||||
}
|
||||
s = strings.ReplaceAll(s, " ", "_")
|
||||
return strings.ToLower(s)
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue