adding file upload support

Author: Mzack9999

simplehttpserver.go

@@ -4,30 +4,38 @@ import (
 	"bytes"
 	"flag"
 	"fmt"
+	"io/ioutil"
 	"log"
 	"net/http"
 	"net/http/httputil"
+	"path"
 )
 
 type options struct {
 	ListenAddress string
 	Folder        string
 	Verbose       bool
+	Upload        bool
 }
 
 var opts options
 
 func main() {
 	flag.StringVar(&opts.ListenAddress, "listen", "0.0.0.0:8000", "Address:Port")
 	flag.StringVar(&opts.Folder, "path", ".", "Folder")
+	flag.BoolVar(&opts.Upload, "upload", false, "Enable upload via PUT")
 	flag.BoolVar(&opts.Verbose, "v", false, "Verbose")
+
 	flag.Parse()
 
 	if flag.NArg() > 0 && opts.Folder == "." {
 		opts.Folder = flag.Args()[0]
 	}
 
 	log.Printf("Serving %s on http://%s/...", opts.Folder, opts.ListenAddress)
+	if opts.Upload {
+		log.Println("Upload enabled")
+	}
 	fmt.Println(http.ListenAndServe(opts.ListenAddress, loglayer(http.FileServer(http.Dir(opts.Folder)))))
 }
 
@@ -37,6 +45,18 @@ func loglayer(handler http.Handler) http.Handler {
 		lrw := newLoggingResponseWriter(w)
 		handler.ServeHTTP(lrw, r)
 
+		// Handles file write if enabled
+		if opts.Upload && r.Method == http.MethodPut {
+			data, err := ioutil.ReadAll(r.Body)
+			if err != nil {
+				log.Println(err)
+			}
+			err = handleUpload(path.Base(r.URL.Path), data)
+			if err != nil {
+				log.Println(err)
+			}
+		}
+
 		if opts.Verbose {
 			headers := new(bytes.Buffer)
 			lrw.Header().Write(headers) //nolint
@@ -70,3 +90,7 @@ func (lrw *loggingResponseWriter) WriteHeader(code int) {
 	lrw.statusCode = code
 	lrw.ResponseWriter.WriteHeader(code)
 }
+
+func handleUpload(file string, data []byte) error {
+	return ioutil.WriteFile(file, data, 0655)
+}