feat(): add basic authentication

.gitignore

@@ -1 +1,2 @@
 files/
+.key

docker-compose.yml

@@ -7,4 +7,5 @@ services:
       - "58080:8080"
     volumes:
       - ./files:/app/files
+      - ./.key:/app/.key
     restart: unless-stopped

main.go

@@ -42,6 +42,11 @@ func uploadHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	if !checkAuth(w, r) {
+		http.Error(w, "You're not authorized.", http.StatusBadRequest)
+		return
+	}
+
 	r.Body = http.MaxBytesReader(w, r.Body, maxFileSize)
 
 	file, _, err := r.FormFile("file")
@@ -77,3 +82,8 @@ func uploadHandler(w http.ResponseWriter, r *http.Request) {
 		fmt.Fprintf(w, "http://%s/%d\n", url, time)
 	}
 }
+
+func checkAuth(w http.ResponseWriter, r *http.Request) bool {
+	authKey, _ := os.ReadFile(".key")
+	return r.Header.Get("X-Auth")+"\n" == string(authKey)
+}