linx-server/fileserve.go

68 lines
1.3 KiB
Go
Raw Normal View History

2015-09-24 23:58:50 +00:00
package main
import (
"net/http"
"os"
"path"
2015-09-29 23:28:10 +00:00
"strings"
2015-09-24 23:58:50 +00:00
"github.com/zenazn/goji/web"
)
func fileServeHandler(c web.C, w http.ResponseWriter, r *http.Request) {
fileName := c.URLParams["name"]
filePath := path.Join(Config.filesDir, fileName)
2015-09-24 23:58:50 +00:00
2015-09-29 03:46:43 +00:00
if !fileExistsAndNotExpired(fileName) {
notFoundHandler(c, w, r)
2015-09-24 23:58:50 +00:00
return
}
2015-09-29 23:28:10 +00:00
if !Config.allowHotlink {
referer := r.Header.Get("Referer")
if referer != "" && !strings.HasPrefix(referer, Config.siteURL) {
w.WriteHeader(403)
return
}
}
http.ServeFile(w, r, filePath)
}
2015-09-28 02:17:12 +00:00
2015-09-30 19:54:30 +00:00
func staticHandler(c web.C, w http.ResponseWriter, r *http.Request) {
path := r.URL.Path
if path[len(path)-1:] == "/" {
notFoundHandler(c, w, r)
return
} else {
filePath := strings.TrimPrefix(path, "/static/")
file, err := staticBox.Open(filePath)
if err != nil {
2015-10-04 16:47:20 +00:00
notFoundHandler(c, w, r)
2015-09-30 19:54:30 +00:00
return
}
w.Header().Set("Etag", timeStartedStr)
w.Header().Set("Cache-Control", "max-age=86400")
2015-09-30 19:54:30 +00:00
http.ServeContent(w, r, filePath, timeStarted, file)
return
}
}
func fileExistsAndNotExpired(filename string) bool {
filePath := path.Join(Config.filesDir, filename)
_, err := os.Stat(filePath)
if err != nil {
return false
2015-09-28 02:17:12 +00:00
}
2015-09-24 23:58:50 +00:00
if isFileExpired(filename) {
os.Remove(path.Join(Config.filesDir, filename))
os.Remove(path.Join(Config.metaDir, filename))
return false
}
return true
2015-09-24 23:58:50 +00:00
}