filecloud.go 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. package main
  2. import (
  3. "embed"
  4. "flag"
  5. "log"
  6. "net/http"
  7. "github.com/gin-gonic/gin"
  8. swaggerFiles "github.com/swaggo/files"
  9. ginSwagger "github.com/swaggo/gin-swagger"
  10. "filecloud/internal/database"
  11. "filecloud/internal/handler"
  12. // _ "filecloud/docs"
  13. )
  14. var (
  15. configPath = flag.String("file", "./config.toml", "config file")
  16. )
  17. //go:embed public/*
  18. var publicFS embed.FS
  19. // @title FileCloud API
  20. // @version 1.0
  21. // @description FileCloud API documentation
  22. // @host localhost:8080
  23. // @BasePath /api
  24. func main() {
  25. flag.Parse()
  26. logger := service.InitLogger("log", "filecloud")
  27. service.LoadConfig(*configPath)
  28. // Initialize database
  29. if err := database.InitDB(); err != nil {
  30. log.Fatal(err)
  31. }
  32. r := gin.Default()
  33. // Serve static files from embedded frontend
  34. r.StaticFS("/", http.FS(publicFS))
  35. // API routes
  36. api := r.Group("/api")
  37. {
  38. api.POST("/files", handler.UploadFile)
  39. api.GET("/files", handler.ListFiles)
  40. }
  41. // Swagger documentation
  42. r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
  43. service.Launch()
  44. logger.Info("stop")
  45. service.Stop()
  46. log.Fatal(r.Run(":8080"))
  47. }