sliproad/router/filerouter.go

package router

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"sort"
	"strings"

	"github.com/gmemstr/nas/files"
	"github.com/gorilla/mux"
)

func handleProvider() handler {
	return func(context *requestContext, w http.ResponseWriter, r *http.Request) *httpError {
		vars := mux.Vars(r)
		providerCodename := vars["provider"]
		providerCodename = strings.Replace(providerCodename, "/", "", -1)
		provider := *files.Providers[providerCodename]

		// Directory listing or serve file.
		if r.Method == "GET" {
			filename, err := url.QueryUnescape(vars["file"])
			if err != nil {
				return &httpError{
					Message:    fmt.Sprintf("error determining filetype for %s\n", filename),
					StatusCode: http.StatusInternalServerError,
				}
			}
			ok, isDir, _ := provider.ObjectInfo(filename)
			if !ok {
				return &httpError{
					Message:    fmt.Sprintf("error locating file %s\n", filename),
					StatusCode: http.StatusNotFound,
				}
			}

			if isDir {
				fileList := provider.GetDirectory(filename)

				data, err := json.Marshal(fileList)
				if err != nil {
					return &httpError{
						Message:    fmt.Sprintf("error fetching filelisting for %s\n", vars["file"]),
						StatusCode: http.StatusNotFound,
					}
				}
				w.Write(data)
				return nil
			} else {
				stream, contenttype, err := provider.SendFile(filename)

				if err != nil {
					return &httpError{
						Message:    fmt.Sprintf("error finding file %s\n", vars["file"]),
						StatusCode: http.StatusNotFound,
					}
				}
				w.Header().Set("Content-Type", contenttype)
				_, err = io.Copy(w, stream)
				if err != nil {
					return &httpError{
						Message:    fmt.Sprintf("unable to write %s\n", vars["file"]),
						StatusCode: http.StatusBadGateway,
					}
				}
			}

			return nil
		}

		// File upload or directory creation.
		if r.Method == "POST" {
			xType := r.Header.Get("X-NAS-Type")
			// We only really care about this header of creating a directory.
			if xType == "directory" {
				dirname := vars["file"]
				success := provider.CreateDirectory(dirname)
				if !success {
					return &httpError{
						Message:    fmt.Sprintf("error creating directory %s\n", dirname),
						StatusCode: http.StatusInternalServerError,
					}
				}
				_, _ = w.Write([]byte("created directory"))
				return nil
			}

			err := r.ParseMultipartForm(32 << 20)
			if err != nil {
				return &httpError{
					Message:    fmt.Sprintf("error parsing form for %s\n", vars["file"]),
					StatusCode: http.StatusInternalServerError,
				}
			}
			file, handler, err := r.FormFile("file")
			defer file.Close()

			success := provider.SaveFile(file, handler.Filename, vars["file"])
			if !success {
				return &httpError{
					Message:    fmt.Sprintf("error saving file %s\n", vars["file"]),
					StatusCode: http.StatusInternalServerError,
				}
			}
			_, _  = w.Write([]byte("saved file"))
		}

		// Delete file.
		if r.Method == "DELETE" {
			path := vars["file"]
			success := provider.Delete(path)
			if !success {
				return &httpError{
					Message:    fmt.Sprintf("error deleting %s\n", path),
					StatusCode: http.StatusInternalServerError,
				}
			}
			_, _ = w.Write([]byte("deleted"))
		}

		return nil
	}
}

func listProviders() handler {
	return func(context *requestContext, w http.ResponseWriter, r *http.Request) *httpError {
		var providers []string
		for v := range files.ProviderConfig {
			providers = append(providers, v)
		}
		sort.Strings(providers)
		data, err := json.Marshal(providers)
		if err != nil {
			return &httpError{
				Message:    fmt.Sprintf("error provider listing"),
				StatusCode: http.StatusInternalServerError,
			}
		}
		_, _ = w.Write(data)
		return nil
	}
}