加固后台管理安全

2026-06-11 07:23:00 +08:00
parent 5255b21141
commit 88b1df64e7
41 changed files with 1276 additions and 293 deletions
@@ -5,12 +5,24 @@ import (
 )

 type Handler struct {
-	service *Service
-	storage *filemodule.Storage
+	service                  *Service
+	storage                  *filemodule.Storage
+	externalUploadSecret     string
+	externalUploadAllowedIPs []string
 }

 const maxExternalUploadBodyBytes = 256 * 1024

-func NewHandler(service *Service, storage *filemodule.Storage) *Handler {
-	return &Handler{service: service, storage: storage}
+type HandlerOptions struct {
+	ExternalUploadSecret     string
+	ExternalUploadAllowedIPs []string
+}
+
+func NewHandler(service *Service, storage *filemodule.Storage, opts ...HandlerOptions) *Handler {
+	handler := &Handler{service: service, storage: storage}
+	if len(opts) > 0 {
+		handler.externalUploadSecret = opts[0].ExternalUploadSecret
+		handler.externalUploadAllowedIPs = opts[0].ExternalUploadAllowedIPs
+	}
+	return handler
 }