handlers.go

package main

import (
	"io"
	"net/http"
	"os"
	"path/filepath"
)

func registerHandlers() {
	if goFileServer {
		http.Handle("/", http.FileServer(http.Dir(exports[0])))
	} else {
		registerShitHandlers()
	}
}

func registerShitHandlers() {
	http.HandleFunc("GET /", get)
	if upload {
		http.HandleFunc("POST /", post)
	}
}

func get(w http.ResponseWriter, r *http.Request) {
	if r.URL.Path == "/" {
		if err := serveRoot(w, r); err != nil {
			httpErr(w, err)
		}

		return
	}

	if err := serveExports(w, r); err != nil {
		httpErr(w, err)
	}
}

func post(w http.ResponseWriter, r *http.Request) {
	if err := r.ParseMultipartForm(maxUploadMemory); err != nil {
		httpErr(w, err)
	}

	path, err := getRealPath(r.URL.Path)
	if err != nil {
		httpErr(w, err)
	}

	if path == "" {
		http.NotFound(w, r)
	}

	fileinfo, err := os.Stat(path)
	if err != nil {
		httpErr(w, err)
	}

	if !fileinfo.IsDir() {
		http.NotFound(w, r)
		return
	}

	files := r.MultipartForm.File["files"]
	for _, header := range files {
		file, err := header.Open()
		if err != nil {
			httpErr(w, err)
			return
		}
		defer file.Close()

		dst, err := os.Create(filepath.Join(path, header.Filename))
		if err != nil {
			http.Error(w, "Error saving file", http.StatusInternalServerError)
			return
		}
		defer dst.Close()

		_, err = io.Copy(dst, file)
		if err != nil {
			httpErr(w, err)
			return
		}
	}

	http.Redirect(w, r, r.URL.Path, http.StatusSeeOther)
}