add a unit test

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)
 		}
 	}
@@ -145,7 +146,7 @@ func withDel(foo http.HandlerFunc) http.HandlerFunc {
 				if len(match) > 0 {
 					match = bytes.Split(match, []byte(`href="`))[1]
 					match = match[:len(match)-1]
-					b[i] = []byte(fmt.Sprintf(`<a href="%s/%s"><input type="button" value="&#10060;" style="padding: .40em 1em .10em 1em; margin-right: .5em"/></a> %s`, match, ENDPOINT_DELETE, b[i]))
+					b[i] = []byte(fmt.Sprintf(`<a href="%s/%s"><input type="button" value="&#10060;" style="padding: .40em 1em .10em 1em; margin-right: .5em" onclick='return confirm("Delete "+%q+"?");'></input></a> %s`, match, ENDPOINT_DELETE, match, b[i]))
 				}
 			}
 			buff.Write(b[i])
@@ -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,52 @@ 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) {
+	ext := strings.ToLower(path.Ext(r.URL.Path))
+	if i := strings.LastIndex(ext, "."); i != -1 {
+		ext = ext[i:]
+	}
+	switch ext {
+	case ".mp4":
+		w.Header().Set("Content-Type", "video/mp4")
+	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")
+	case ".css":
+		w.Header().Set("Content-Type", "text/css")
+	case ".js":
+		w.Header().Set("Content-Type", "text/javascript")
+	case ".json":
+		w.Header().Set("Content-Type", "application/json")
+	case ".html", ".htm":
+		w.Header().Set("Content-Type", "text/html")
+	case ".pdf":
+		w.Header().Set("Content-Type", "application/pdf")
+	case ".webm":
+		w.Header().Set("Content-Type", "video/webm")
+	case ".weba":
+		w.Header().Set("Content-Type", "audio/webm")
+	case ".webp":
+		w.Header().Set("Content-Type", "image/webp")
+	case ".zip":
+		w.Header().Set("Content-Type", "application/zip")
+	case ".7z":
+		w.Header().Set("Content-Type", "application/x-7z-compressed")
+	case ".tar":
+		w.Header().Set("Content-Type", "application/x-tar")
+	}
+}

main_test.go

@@ -0,0 +1,34 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"testing"
+)
+
+func TestSetContentType(t *testing.T) {
+	t.Parallel()
+
+	cases := map[string]struct {
+		path string
+		want string
+	}{
+		"css with multi .": {
+			path: "/static/css/main.2145ce41.chunk.css",
+			want: "text/css",
+		},
+	}
+
+	for name, d := range cases {
+		c := d
+		t.Run(name, func(t *testing.T) {
+			r := &http.Request{URL: &url.URL{Path: c.path}}
+			w := httptest.NewRecorder()
+			setContentTypeIfMedia(w, r)
+			if ct := w.Header().Get("Content-Type"); ct != c.want {
+				t.Errorf("wrong content type: want %q, got %q", c.want, ct)
+			}
+		})
+	}
+}

public/_index.html

@@ -1 +0,0 @@
-hi