137 lines
2.9 KiB
Go
137 lines
2.9 KiB
Go
package fstore
|
|
|
|
import (
|
|
"embed"
|
|
"io"
|
|
"git.crumpington.com/public/jldb/fstore/pages"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
//go:embed static/*
|
|
var staticFS embed.FS
|
|
|
|
type browser struct {
|
|
store *Store
|
|
}
|
|
|
|
func (s *Store) ServeBrowser(listenAddr string) error {
|
|
b := &browser{s}
|
|
http.HandleFunc("/", b.handle)
|
|
http.Handle("/static/", http.FileServer(http.FS(staticFS)))
|
|
return http.ListenAndServe(listenAddr, nil)
|
|
}
|
|
|
|
func (b *browser) handle(w http.ResponseWriter, r *http.Request) {
|
|
switch r.Method {
|
|
case http.MethodGet:
|
|
b.handleGet(w, r)
|
|
case http.MethodPost:
|
|
b.handlePOST(w, r)
|
|
default:
|
|
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
|
|
}
|
|
}
|
|
|
|
func (b *browser) handleGet(w http.ResponseWriter, r *http.Request) {
|
|
path := cleanPath(r.URL.Path)
|
|
if err := validatePath(path); err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
info, err := b.store.Stat(path)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
pages.Page{Path: path}.Render(w)
|
|
return
|
|
}
|
|
http.Error(w, err.Error(), http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
b.store.Serve(w, r, path)
|
|
return
|
|
}
|
|
|
|
dirs, files, err := b.store.List(path)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
pages.Page{
|
|
Path: path,
|
|
Dirs: dirs,
|
|
Files: files,
|
|
}.Render(w)
|
|
}
|
|
|
|
// Handle actions:
|
|
// - upload (multipart),
|
|
// - delete
|
|
func (b *browser) handlePOST(w http.ResponseWriter, r *http.Request) {
|
|
if err := r.ParseMultipartForm(1024 * 1024); err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
switch r.Form.Get("Action") {
|
|
case "Upload":
|
|
b.handlePOSTUpload(w, r)
|
|
case "Delete":
|
|
b.handlePOSTDelete(w, r)
|
|
default:
|
|
http.Error(w, "unknown action", http.StatusBadRequest)
|
|
}
|
|
}
|
|
|
|
func (b *browser) handlePOSTUpload(w http.ResponseWriter, r *http.Request) {
|
|
file, handler, err := r.FormFile("File")
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
relativePath := handler.Filename
|
|
if p := r.Form.Get("Path"); p != "" {
|
|
relativePath = p
|
|
}
|
|
fullPath := filepath.Join(r.URL.Path, relativePath)
|
|
|
|
tmpPath := b.store.GetTempFilePath()
|
|
defer os.RemoveAll(tmpPath)
|
|
|
|
tmpFile, err := os.Create(tmpPath)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
defer tmpFile.Close()
|
|
|
|
if _, err := io.Copy(tmpFile, file); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
if err := b.store.Store(tmpPath, fullPath); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, filepath.Dir(fullPath), http.StatusSeeOther)
|
|
}
|
|
|
|
func (b *browser) handlePOSTDelete(w http.ResponseWriter, r *http.Request) {
|
|
path := r.Form.Get("Path")
|
|
if err := b.store.Remove(path); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, filepath.Dir(path), http.StatusSeeOther)
|
|
}
|