Static file upload via optional direct link or multipart form

view/files.go

@@ -1,10 +1,17 @@
 package view
 
 import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"io/ioutil"
 	"local/dndex/config"
 	"local/dndex/storage"
 	"local/simpleserve/simpleserve"
 	"net/http"
+	"net/url"
+	"os"
 	"path"
 	"strings"
 )
@@ -19,6 +26,8 @@ func files(_ storage.Graph, w http.ResponseWriter, r *http.Request) error {
 	switch r.Method {
 	case http.MethodGet:
 		return filesGet(w, r)
+	case http.MethodPost:
+		return filesPost(w, r)
 	default:
 		http.NotFound(w, r)
 		return nil
@@ -26,6 +35,80 @@ func files(_ storage.Graph, w http.ResponseWriter, r *http.Request) error {
 }
 
 func filesGet(w http.ResponseWriter, r *http.Request) error {
-	http.ServeFile(w, r, path.Join(config.New().FileRoot, r.URL.Path))
+	http.ServeFile(w, r, toLocalPath(r.URL.Path))
 	return nil
 }
+
+func filesPost(w http.ResponseWriter, r *http.Request) error {
+	p := toLocalPath(r.URL.Path)
+	if err := os.MkdirAll(path.Dir(p), os.ModePerm); err != nil {
+		return err
+	}
+	switch r.URL.Query().Get("direct") {
+	case "true":
+		return filesPostFromDirectLink(w, r)
+	default:
+		return filesPostFromUpload(w, r)
+	}
+}
+
+func filesPostFromDirectLink(w http.ResponseWriter, r *http.Request) error {
+	b, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		return err
+	}
+	url, err := url.Parse(string(b))
+	if err != nil {
+		return err
+	}
+	resp, err := http.Get(url.String())
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("bad status from direct: %v", resp.StatusCode)
+	}
+
+	path := toLocalPath(r.URL.Path)
+	f, err := os.Create(path)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	_, err = io.Copy(f, resp.Body)
+	return err
+}
+
+func filesPostFromUpload(w http.ResponseWriter, r *http.Request) error {
+	p := toLocalPath(r.URL.Path)
+	if err := os.MkdirAll(path.Dir(p), os.ModePerm); err != nil {
+		return err
+	}
+	if fi, err := os.Stat(p); err != nil && !os.IsNotExist(err) {
+		return err
+	} else if err == nil && fi.IsDir() {
+		return errors.New("path is a directory")
+	}
+	f, err := os.Create(p)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	megabyte := 100 << 20
+	r.ParseMultipartForm(int64(megabyte))
+	file, _, err := r.FormFile("file")
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+	if _, err := io.Copy(f, file); err != nil {
+		return err
+	}
+	return json.NewEncoder(w).Encode(map[string]interface{}{"status": "ok"})
+}
+
+func toLocalPath(p string) string {
+	return path.Join(config.New().FileRoot, p)
+}