func Image(ctx *tango.Context) {
token := ctx.Params().Get(":path")
// split token and file ext
var filePath string
if i := strings.IndexRune(token, '.'); i == -1 {
return
} else {
filePath = token[i+1:]
token = token[:i]
}
// decode token to file path
var image models.Image
if err := image.DecodeToken(token); err != nil {
log.Info(err)
return
}
// file real path
filePath = attachment.GenImagePath(&image) + filePath
// if x-send on then set header and http status
// fall back use proxy serve file
if setting.ImageXSend {
//ext := filepath.Ext(filePath)
// TODO:
//ctx.Header().ContentType(ext)
ctx.Header().Set(setting.ImageXSendHeader, "/"+filePath)
ctx.WriteHeader(http.StatusOK)
} else {
// direct serve file use go
ctx.ServeFile(filePath)
}
}
// get auth token via http header, cookie or form value
func (ar *AuthRouter) GetAuthToken(ctx *tango.Context) string {
var token string
if token = ctx.Header().Get("X-Token"); token != "" {
return token
}
if token = ctx.Cookie("x-token"); token != "" {
return token
}
return ctx.Form("x-token")
}
// Handle implements tango handler,
// copy from tango static.go
func (s *Static) Handle(ctx *tango.Context) {
if ctx.Req().Method != "GET" && ctx.Req().Method != "HEAD" {
ctx.Next()
return
}
opt := prepareStaticOptions(s.Options)
var rPath = ctx.Req().URL.Path
// if defined prefix, then only check prefix
if opt.Prefix != "" {
if !strings.HasPrefix(ctx.Req().URL.Path, opt.Prefix) {
ctx.Next()
return
} else {
if len(opt.Prefix) == len(ctx.Req().URL.Path) {
rPath = ""
} else {
rPath = ctx.Req().URL.Path[len(opt.Prefix):]
}
}
}
fPath, _ := filepath.Abs(filepath.Join(opt.RootPath, rPath))
finfo, err := os.Stat(fPath)
if err != nil {
if !os.IsNotExist(err) {
ctx.Result = tango.InternalServerError(err.Error())
ctx.HandleError()
return
}
} else if !finfo.IsDir() {
if len(opt.FilterExts) > 0 {
var matched bool
for _, ext := range opt.FilterExts {
if filepath.Ext(fPath) == ext {
matched = true
break
}
}
if !matched {
ctx.Next()
return
}
}
err := ctx.ServeFile(fPath)
if err != nil {
ctx.Result = tango.InternalServerError(err.Error())
ctx.HandleError()
}
return
} else {
// try serving index.html or index.htm
if len(opt.IndexFiles) > 0 {
for _, index := range opt.IndexFiles {
nPath := filepath.Join(fPath, index)
finfo, err = os.Stat(nPath)
if err != nil {
if !os.IsNotExist(err) {
ctx.Result = tango.InternalServerError(err.Error())
ctx.HandleError()
return
}
} else if !finfo.IsDir() {
err = ctx.ServeFile(nPath)
if err != nil {
ctx.Result = tango.InternalServerError(err.Error())
ctx.HandleError()
}
return
}
}
}
// list dir files
if opt.ListDir {
ctx.Header().Set("Content-Type", "text/html; charset=UTF-8")
ctx.Write([]byte(`<ul style="list-style-type:none;line-height:32px;">`))
rootPath, _ := filepath.Abs(opt.RootPath)
rPath, _ := filepath.Rel(rootPath, fPath)
if fPath != rootPath {
ctx.Write([]byte(`<li> <a href="/` + path.Join(opt.Prefix, filepath.Dir(rPath)) + `">..</a></li>`))
}
err = filepath.Walk(fPath, func(p string, fi os.FileInfo, err error) error {
rPath, _ := filepath.Rel(fPath, p)
if rPath == "." || len(strings.Split(rPath, string(filepath.Separator))) > 1 {
return nil
}
rPath, _ = filepath.Rel(rootPath, p)
ps, _ := os.Stat(p)
if ps.IsDir() {
ctx.Write([]byte(`<li>┖ <a href="/` + path.Join(opt.Prefix, rPath) + `">` + filepath.Base(p) + `</a></li>`))
} else {
if len(opt.FilterExts) > 0 {
var matched bool
for _, ext := range opt.FilterExts {
if filepath.Ext(p) == ext {
matched = true
break
}
}
if !matched {
return nil
}
}
ctx.Write([]byte(`<li> <a href="/` + path.Join(opt.Prefix, rPath) + `">` + filepath.Base(p) + `</a></li>`))
}
return nil
})
ctx.Write([]byte("</ul>"))
return
}
}
}