فهرست منبع

feat: 静态文件也需要经过权限认证 (#52)

jeessy2 2 سال پیش
والد
کامیت
9b5c429136
1فایلهای تغییر یافته به همراه10 افزوده شده و 2 حذف شده
  1. 10 2
      main.go

+ 10 - 2
main.go

@@ -73,10 +73,18 @@ func main() {
 	}
 }
 
+func staticFsFunc(writer http.ResponseWriter, request *http.Request) {
+	http.FileServer(http.FS(staticEmbededFiles)).ServeHTTP(writer, request)
+}
+
+func faviconFsFunc(writer http.ResponseWriter, request *http.Request) {
+	http.FileServer(http.FS(faviconEmbededFile)).ServeHTTP(writer, request)
+}
+
 func run(firstDelay time.Duration) {
 	// 启动静态文件服务
-	http.Handle("/static/", http.FileServer(http.FS(staticEmbededFiles)))
-	http.Handle("/favicon.ico", http.FileServer(http.FS(faviconEmbededFile)))
+	http.HandleFunc("/static/", web.BasicAuth(staticFsFunc))
+	http.HandleFunc("/favicon.ico", web.BasicAuth(faviconFsFunc))
 
 	http.HandleFunc("/", web.BasicAuth(web.WritingConfig))
 	http.HandleFunc("/save", web.BasicAuth(web.Save))