More content types

main.go

@@ -99,6 +99,7 @@ func endpoints(foo http.HandlerFunc) http.HandlerFunc {
 				fmt.Fprintln(w, err.Error())
 			}
 		} else {
+			setContentTypeIfMedia(w, r)
 			foo(w, r)
 		}
 	}
@@ -189,7 +190,7 @@ func del(w http.ResponseWriter, r *http.Request) error {
 	if err != nil {
 		return err
 	}
-	err = os.Remove(p)
+	err = os.RemoveAll(p)
 	if err != nil {
 		return err
 	}
@@ -201,3 +202,28 @@ func toRealPath(p string) string {
 	d := path.Join(fs.Get("d").GetString())
 	return path.Join(d, p)
 }
+
+func setContentTypeIfMedia(w http.ResponseWriter, r *http.Request) {
+	switch strings.ToLower(path.Ext(r.URL.Path)) {
+	case ".mp4":
+		w.Header().Set("Content-Type", "video/mp4")
+	case ".webm":
+		w.Header().Set("Content-Type", "video/webm")
+	case ".mkv":
+		w.Header().Set("Content-Type", "video/x-matroska")
+	case ".mp3":
+		w.Header().Set("Content-Type", "audio/mpeg3")
+	case ".epub", ".mobi":
+		w.Header().Set("Content-Disposition", "attachment")
+	case ".jpg", ".jpeg":
+		w.Header().Set("Content-Type", "image/jpeg")
+	case ".gif":
+		w.Header().Set("Content-Type", "image/gif")
+	case ".png":
+		w.Header().Set("Content-Type", "image/png")
+	case ".ico":
+		w.Header().Set("Content-Type", "image/x-icon")
+	case ".svg":
+		w.Header().Set("Content-Type", "image/svg+xml")
+	}
+}