feat(): add basic authentication

jabuxas · Aug 19, 2024 · b8ad1dd · b8ad1dd
1 parent 48528f3
commit b8ad1dd
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
 files/
+.key
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -7,4 +7,5 @@ services:
       - "58080:8080"
     volumes:
       - ./files:/app/files
+      - ./.key:/app/.key
     restart: unless-stopped
diff --git a/main.go b/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)
+}