mirror of
https://github.com/go-shiori/shiori.git
synced 2024-11-15 21:55:31 +08:00
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package webserver
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
fp "path/filepath"
|
|
"strings"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
)
|
|
|
|
// serveFile is handler for general file request
|
|
func (h *handler) serveFile(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
err := serveFile(w, r.URL.Path, true)
|
|
checkError(err)
|
|
}
|
|
|
|
// serveJsFile is handler for GET /js/*filepath
|
|
func (h *handler) serveJsFile(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
filePath := r.URL.Path
|
|
fileName := path.Base(filePath)
|
|
fileDir := path.Dir(filePath)
|
|
|
|
if developmentMode && fp.Ext(fileName) == ".js" && strings.HasSuffix(fileName, ".min.js") {
|
|
fileName = strings.TrimSuffix(fileName, ".min.js") + ".js"
|
|
filePath = path.Join(fileDir, fileName)
|
|
if assetExists(filePath) {
|
|
redirectPage(w, r, filePath)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
err := serveFile(w, r.URL.Path, true)
|
|
checkError(err)
|
|
}
|
|
|
|
// serveIndexPage is handler for GET /
|
|
func (h *handler) serveIndexPage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
// Make sure session still valid
|
|
err := h.validateSession(r)
|
|
if err != nil {
|
|
redirectPage(w, r, "/login")
|
|
return
|
|
}
|
|
|
|
err = serveFile(w, "index.html", false)
|
|
checkError(err)
|
|
}
|
|
|
|
// serveLoginPage is handler for GET /login
|
|
func (h *handler) serveLoginPage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
// Make sure session is not valid
|
|
err := h.validateSession(r)
|
|
if err == nil {
|
|
redirectPage(w, r, "/")
|
|
return
|
|
}
|
|
|
|
err = serveFile(w, "login.html", false)
|
|
checkError(err)
|
|
}
|
|
|
|
// serveThumbnailImage is handler for GET /thumb/:id
|
|
func (h *handler) serveThumbnailImage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
// Get bookmark ID from URL
|
|
id := ps.ByName("id")
|
|
|
|
// Open image
|
|
imgPath := fp.Join(h.DataDir, "thumb", id)
|
|
img, err := os.Open(imgPath)
|
|
checkError(err)
|
|
defer img.Close()
|
|
|
|
// Get image type from its 512 first bytes
|
|
buffer := make([]byte, 512)
|
|
_, err = img.Read(buffer)
|
|
checkError(err)
|
|
|
|
mimeType := http.DetectContentType(buffer)
|
|
w.Header().Set("Content-Type", mimeType)
|
|
|
|
// Serve image
|
|
img.Seek(0, 0)
|
|
_, err = io.Copy(w, img)
|
|
checkError(err)
|
|
}
|