update file structure [ci skip]

routes/browse/post.go

@@ -0,0 +1,141 @@
+package browse
+
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"mime/multipart"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/hacdias/caddy-hugo/config"
+	"github.com/hacdias/caddy-hugo/tools/commands"
+	"github.com/hacdias/caddy-hugo/tools/utils"
+)
+
+// POST handles the POST method on browse page. It's used to create new files,
+// folders and upload content.
+func POST(w http.ResponseWriter, r *http.Request, c *config.Config) (int, error) {
+	// Remove prefix slash
+	r.URL.Path = strings.TrimPrefix(r.URL.Path, "/")
+
+	// If it's the upload of a file
+	if r.Header.Get("X-Upload") == "true" {
+		return upload(w, r, c)
+	}
+
+	// Get the JSON information sent using a buffer
+	buffer := new(bytes.Buffer)
+	buffer.ReadFrom(r.Body)
+
+	// Creates the raw file "map" using the JSON
+	var info map[string]interface{}
+	json.Unmarshal(buffer.Bytes(), &info)
+
+	// Check if filename and archetype are specified in
+	// the request
+	if _, ok := info["filename"]; !ok {
+		return utils.RespondJSON(w, map[string]string{
+			"message": "Filename not specified.",
+		}, 500, nil)
+	}
+
+	if _, ok := info["archetype"]; !ok {
+		return utils.RespondJSON(w, map[string]string{
+			"message": "Archtype not specified.",
+		}, 500, nil)
+	}
+
+	// Sanitize the file name path
+	filename := info["filename"].(string)
+	filename = strings.TrimPrefix(filename, "/")
+	filename = strings.TrimSuffix(filename, "/")
+	url := "/admin/edit/" + r.URL.Path + filename
+	filename = c.Path + r.URL.Path + filename
+
+	if strings.HasPrefix(filename, c.Path+"content/") &&
+		(strings.HasSuffix(filename, ".md") || strings.HasSuffix(filename, ".markdown")) {
+
+		filename = strings.Replace(filename, c.Path+"content/", "", 1)
+		args := []string{"new", filename}
+		archetype := info["archetype"].(string)
+
+		if archetype != "" {
+			args = append(args, "--kind", archetype)
+		}
+
+		if err := commands.Run(c.Hugo, args, c.Path); err != nil {
+			return utils.RespondJSON(w, map[string]string{
+				"message": "Something went wrong.",
+			}, 500, err)
+		}
+	} else {
+		var err error
+
+		if filepath.Ext(filename) == "" {
+			err = os.MkdirAll(filename, 0755)
+			url = strings.Replace(url, "edit", "browse", 1)
+		} else {
+			var wf *os.File
+			wf, err = os.Create(filename)
+			defer wf.Close()
+		}
+
+		if err != nil {
+			return utils.RespondJSON(w, map[string]string{
+				"message": "Something went wrong.",
+			}, 500, err)
+		}
+
+	}
+
+	return utils.RespondJSON(w, map[string]string{
+		"location": url,
+		"message":  "File created.",
+	}, 200, nil)
+}
+
+func upload(w http.ResponseWriter, r *http.Request, c *config.Config) (int, error) {
+	// Parse the multipart form in the request
+	err := r.ParseMultipartForm(100000)
+	if err != nil {
+		return utils.RespondJSON(w, map[string]string{
+			"message": "Something went wrong.",
+		}, 500, err)
+	}
+
+	// For each file header in the multipart form
+	for _, fheaders := range r.MultipartForm.File {
+		// Handle each file
+		for _, hdr := range fheaders {
+			// Open the first file
+			var infile multipart.File
+			if infile, err = hdr.Open(); nil != err {
+				return utils.RespondJSON(w, map[string]string{
+					"message": "Something went wrong.",
+				}, 500, err)
+			}
+
+			// Create the file
+			var outfile *os.File
+			if outfile, err = os.Create(c.Path + r.URL.Path + hdr.Filename); nil != err {
+				return utils.RespondJSON(w, map[string]string{
+					"message": "Something went wrong.",
+				}, 500, err)
+			}
+
+			// Copy the file content
+			if _, err = io.Copy(outfile, infile); nil != err {
+				return utils.RespondJSON(w, map[string]string{
+					"message": "Something went wrong.",
+				}, 500, err)
+			}
+
+			defer outfile.Close()
+		}
+	}
+
+	return utils.RespondJSON(w, nil, 200, nil)
+}