-
Notifications
You must be signed in to change notification settings - Fork 64
/
static.go
107 lines (92 loc) · 2.07 KB
/
static.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package plugin
import (
"fmt"
"os"
"path/filepath"
"github.com/labstack/echo"
)
type (
Static struct {
Base `json:",squash"`
// Root directory from where the static content is served.
// Required.
Root string `json:"root"`
// Index file for serving a directory.
// Optional. Default value "index.html".
Index string `json:"index"`
// Enable HTML5 mode by forwarding all not-found requests to root so that
// SPA (single-page application) can handle the routing.
// Optional. Default value false.
HTML5 bool `json:"html5"`
// Enable directory browsing.
// Optional. Default value false.
Browse bool `json:"browse"`
}
)
func (s *Static) Init() (err error) {
// Defaults
if s.Root == "" {
// TODO:
}
if s.Index == "" {
s.Index = "index.html"
}
return
}
func (*Static) Priority() int {
return 1
}
func (s *Static) Process(next echo.HandlerFunc) echo.HandlerFunc {
return func(c echo.Context) error {
p := c.Param("*")
if p == "" {
p = c.Request().URL.Path // For hosts with no paths defined
}
name := filepath.Join(s.Root, p)
fi, err := os.Stat(name)
if err != nil {
if os.IsNotExist(err) {
if s.HTML5 {
return c.File(filepath.Join(s.Root, s.Index))
}
return echo.ErrNotFound
}
return err
}
if fi.IsDir() {
if s.Browse {
return s.listDir(name, c.Response())
}
return c.File(filepath.Join(name, s.Index))
}
return c.File(name)
}
}
func (s *Static) listDir(name string, res *echo.Response) error {
dir, err := os.Open(name)
if err != nil {
return err
}
dirs, err := dir.Readdir(-1)
if err != nil {
return err
}
// Create a directory index
res.Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
if _, err = fmt.Fprintf(res, "<pre>\n"); err != nil {
return err
}
for _, d := range dirs {
name := d.Name()
color := "#212121"
if d.IsDir() {
color = "#e91e63"
name += "/"
}
if _, err = fmt.Fprintf(res, "<a href=\"%s\" style=\"color: %s;\">%s</a>\n", name, color, name); err != nil {
return err
}
}
_, err = fmt.Fprintf(res, "</pre>\n")
return err
}