filebrowser/routes/browse/post.go

package browse

import (
	"bytes"
	"encoding/json"
	"io"
	"mime/multipart"
	"net/http"
	"os"
	"path/filepath"
	"strings"

	"github.com/hacdias/caddy-hugo/tools/commands"
	s "github.com/hacdias/caddy-hugo/tools/server"
)

// 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) (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)
	}

	// 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 s.RespondJSON(w, &response{"Filename not specified.", ""}, http.StatusBadRequest, nil)
	}

	if _, ok := info["archetype"]; !ok {
		return s.RespondJSON(w, &response{"Archtype not specified.", ""}, http.StatusBadRequest, 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 = conf.Path + r.URL.Path + filename

	if strings.HasPrefix(filename, conf.Path+"content/") &&
		(strings.HasSuffix(filename, ".md") || strings.HasSuffix(filename, ".markdown")) {

		filename = strings.Replace(filename, conf.Path+"content/", "", 1)
		args := []string{"new", filename}
		archetype := info["archetype"].(string)

		if archetype != "" {
			args = append(args, "--kind", archetype)
		}

		if err := commands.Run(conf.Hugo, args, conf.Path); err != nil {
			return s.RespondJSON(w, &response{"Something went wrong.", ""}, http.StatusInternalServerError, 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 s.RespondJSON(w, &response{"Something went wrong.", ""}, http.StatusInternalServerError, err)
		}

	}

	return s.RespondJSON(w, &response{"File created!", url}, http.StatusOK, nil)
}

func upload(w http.ResponseWriter, r *http.Request) (int, error) {
	// Parse the multipart form in the request
	err := r.ParseMultipartForm(100000)
	if err != nil {
		return s.RespondJSON(w, &response{"Something went wrong.", ""}, http.StatusInternalServerError, 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 s.RespondJSON(w, &response{"Something went wrong.", ""}, http.StatusInternalServerError, err)
			}

			// Create the file
			var outfile *os.File
			if outfile, err = os.Create(conf.Path + r.URL.Path + hdr.Filename); nil != err {
				return s.RespondJSON(w, &response{"Something went wrong.", ""}, http.StatusInternalServerError, err)
			}

			// Copy the file content
			if _, err = io.Copy(outfile, infile); nil != err {
				return s.RespondJSON(w, &response{"Something went wrong.", ""}, http.StatusInternalServerError, err)
			}

			defer outfile.Close()
		}
	}

	return s.RespondJSON(w, nil, http.StatusOK, nil)
}