9efd78aaff
- 修正 RustFS 端点地址为 192.168.1.154:9000（对应服务器实际部署） - 移除 config.go 中硬编码的 AccessKey/SecretKey 默认值，改为环境变量传入 - 移除 middleware/auth.go 中硬编码的 API_KEY_VALUE 常量，改为参数注入 - 新增 Config.AuthAPIKey 字段，通过 AUTH_API_KEY 环境变量配置 - 移除 login.html 页面上的密钥格式提示，防止信息泄露 - docker-compose.yml 补全 RustFS 连接所需的环境变量 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
129 lines
4.8 KiB
Go
129 lines
4.8 KiB
Go
package main
|
||
|
||
import (
|
||
_ "file-system/docs" // Import generated docs
|
||
"file-system/internal/api/endpoints"
|
||
"file-system/internal/api/handlers"
|
||
"file-system/internal/api/validators"
|
||
"file-system/internal/common"
|
||
"file-system/internal/domain/repository"
|
||
"file-system/internal/infrastructure/mediator"
|
||
"file-system/internal/infrastructure/s3"
|
||
"file-system/internal/middleware"
|
||
"io"
|
||
"net/http"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
swaggerFiles "github.com/swaggo/files"
|
||
ginSwagger "github.com/swaggo/gin-swagger"
|
||
)
|
||
|
||
// @title RustFS File System API
|
||
// @version 1.1
|
||
// @description RustFS 文件存储系统 API，支持分片上传、文件预览、分页查询等高级功能。
|
||
// @host localhost:8080
|
||
// @BasePath /
|
||
func main() {
|
||
cfg := common.LoadConfig()
|
||
|
||
// Infrastructure
|
||
rustfsClient := s3.NewRustFSClient(cfg)
|
||
s3Repo := s3.NewS3FileRepository(rustfsClient)
|
||
m := mediator.NewMediator()
|
||
|
||
// Handlers
|
||
uploadHandler := handlers.NewUploadFileHandler(s3Repo)
|
||
downloadHandler := handlers.NewDownloadFileHandler(s3Repo)
|
||
createBucketHandler := handlers.NewCreateBucketHandler(s3Repo)
|
||
listBucketsHandler := handlers.NewListBucketsHandler(s3Repo)
|
||
deleteBucketHandler := handlers.NewDeleteBucketHandler(s3Repo)
|
||
// New Handlers
|
||
listFilesHandler := handlers.NewListFilesHandler(s3Repo)
|
||
previewHandler := handlers.NewGetFilePreviewHandler(s3Repo)
|
||
initMultipartHandler := handlers.NewInitMultipartHandler(s3Repo)
|
||
uploadPartHandler := handlers.NewUploadPartHandler(s3Repo)
|
||
completeMultipartHandler := handlers.NewCompleteMultipartHandler(s3Repo)
|
||
deleteFileHandler := handlers.NewDeleteFileHandler(s3Repo)
|
||
loginHandler := handlers.NewLoginHandler(cfg.AuthAPIKey)
|
||
|
||
// Register Handlers
|
||
mediator.Register[handlers.UploadFileCommand, string](m, uploadHandler)
|
||
mediator.Register[handlers.DownloadFileQuery, io.ReadCloser](m, downloadHandler)
|
||
mediator.Register[handlers.CreateBucketCommand, string](m, createBucketHandler)
|
||
mediator.Register[handlers.ListBucketsQuery, []string](m, listBucketsHandler)
|
||
mediator.Register[handlers.DeleteBucketCommand, string](m, deleteBucketHandler)
|
||
// New Registrations
|
||
mediator.Register[handlers.ListFilesQuery, *repository.ListFilesResult](m, listFilesHandler)
|
||
mediator.Register[handlers.GetFilePreviewQuery, string](m, previewHandler)
|
||
mediator.Register[handlers.InitMultipartCommand, string](m, initMultipartHandler)
|
||
mediator.Register[handlers.UploadPartCommand, string](m, uploadPartHandler)
|
||
mediator.Register[handlers.CompleteMultipartCommand, string](m, completeMultipartHandler)
|
||
mediator.Register[handlers.DeleteFileCommand, string](m, deleteFileHandler)
|
||
mediator.Register[handlers.LoginQuery, handlers.LoginResult](m, loginHandler)
|
||
|
||
// Validators
|
||
uploadValidator := validators.NewUploadFileValidator()
|
||
downloadValidator := validators.NewDownloadFileValidator()
|
||
createBucketValidator := validators.NewCreateBucketValidator()
|
||
newFeaturesValidator := validators.NewNewFeaturesValidator()
|
||
|
||
// Endpoints
|
||
fileEndpoint := endpoints.NewFileEndpoint(m, uploadValidator, downloadValidator, newFeaturesValidator)
|
||
bucketEndpoint := endpoints.NewBucketEndpoint(m, createBucketValidator)
|
||
authEndpoint := endpoints.NewAuthEndpoint(m)
|
||
|
||
// Router
|
||
r := gin.Default()
|
||
|
||
// CORS
|
||
r.Use(func(c *gin.Context) {
|
||
c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
|
||
c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
|
||
c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization, accept, origin, Cache-Control, X-Requested-With")
|
||
c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS, GET, PUT, DELETE")
|
||
if c.Request.Method == "OPTIONS" {
|
||
c.AbortWithStatus(204)
|
||
return
|
||
}
|
||
c.Next()
|
||
})
|
||
|
||
// 公开接口（无需授权）
|
||
r.POST("/auth/login", authEndpoint.Login)
|
||
|
||
// API授权中间件组
|
||
api := r.Group("/")
|
||
api.Use(middleware.AuthMiddleware(cfg.AuthAPIKey))
|
||
{
|
||
// File operations
|
||
api.POST("/files/upload", fileEndpoint.UploadFile)
|
||
api.GET("/files/download", fileEndpoint.DownloadFile)
|
||
api.GET("/files/list", fileEndpoint.ListFiles)
|
||
api.GET("/files/preview", fileEndpoint.GetPreviewURL)
|
||
|
||
// Delete file
|
||
api.DELETE("/files/delete", fileEndpoint.DeleteFile)
|
||
|
||
// Multipart Upload
|
||
api.POST("/files/multipart/init", fileEndpoint.InitMultipart)
|
||
api.PUT("/files/multipart/part", fileEndpoint.UploadPart)
|
||
api.POST("/files/multipart/complete", fileEndpoint.CompleteMultipart)
|
||
|
||
// Bucket operations
|
||
api.POST("/buckets", bucketEndpoint.CreateBucket)
|
||
api.GET("/buckets", bucketEndpoint.ListBuckets)
|
||
api.DELETE("/buckets", bucketEndpoint.DeleteBucket)
|
||
}
|
||
|
||
// Swagger
|
||
r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
|
||
|
||
// Web UI
|
||
r.Static("/web", "./web")
|
||
r.GET("/", func(c *gin.Context) {
|
||
c.Redirect(http.StatusMovedPermanently, "/web")
|
||
})
|
||
|
||
r.Run(":" + cfg.ServerPort)
|
||
}
|