You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package web
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func RegisterRoute(router *gin.Engine) error {
|
|
files, err := WalkDirs("", false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, f := range files {
|
|
router.GET(f, func(name string) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
data, err := ReadFile(name)
|
|
if err != nil {
|
|
c.AbortWithError(500, err)
|
|
return
|
|
}
|
|
switch filepath.Ext(name) {
|
|
case ".html":
|
|
c.Header("Content-Type", "text/html; charset=utf-8")
|
|
case ".css":
|
|
c.Header("Content-Type", "text/css; charset=utf-8")
|
|
case ".js":
|
|
c.Header("Content-Type", "text/javascript")
|
|
case ".svg":
|
|
c.Header("Content-Type", "image/svg+xml")
|
|
}
|
|
if path, err := exec.LookPath(os.Args[0]); err == nil {
|
|
if file, err := os.Stat(path); err == nil {
|
|
c.Header("Last-Modified", file.ModTime().UTC().Format("Mon, 02 Jan 2006 15:04:05 GMT"))
|
|
}
|
|
}
|
|
if _, err := c.Writer.Write(data); err != nil {
|
|
c.AbortWithError(500, err)
|
|
return
|
|
}
|
|
}
|
|
}(f))
|
|
}
|
|
router.GET("/", func(c *gin.Context) {
|
|
data, err := ReadFile("index.html")
|
|
if err != nil {
|
|
c.AbortWithError(500, err)
|
|
return
|
|
}
|
|
c.Header("Content-Type", "text/html; charset=utf-8")
|
|
if _, err := c.Writer.Write(data); err != nil {
|
|
c.AbortWithError(500, err)
|
|
return
|
|
}
|
|
})
|
|
return nil
|
|
}
|