89 lines
3.0 KiB
Go
89 lines
3.0 KiB
Go
package main
|
||
|
||
import (
|
||
"log"
|
||
"net/http"
|
||
"tonav-go/database"
|
||
"tonav-go/handlers"
|
||
config "tonav-go/utils"
|
||
|
||
"github.com/gin-contrib/sessions"
|
||
"github.com/gin-contrib/sessions/cookie"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
func main() {
|
||
cfg := config.LoadConfig()
|
||
database.InitDB(cfg.DBPath)
|
||
database.Seed()
|
||
handlers.StartHealthCheck()
|
||
handlers.StartAutoBackup()
|
||
handlers.SetupMCP()
|
||
|
||
r := gin.Default()
|
||
store := cookie.NewStore([]byte(cfg.SecretKey))
|
||
r.Use(sessions.Sessions("tonav_session", store))
|
||
r.Static("/static", "./static")
|
||
|
||
// 显式逐个加载，彻底杜绝路径猜测问题
|
||
r.LoadHTMLFiles(
|
||
"templates/index.html",
|
||
"templates/admin/login.html",
|
||
"templates/admin/dashboard.html",
|
||
"templates/admin/categories.html",
|
||
"templates/admin/services.html",
|
||
"templates/admin/change_password.html",
|
||
)
|
||
|
||
r.GET("/", handlers.IndexHandler)
|
||
|
||
r.GET("/admin/login", func(c *gin.Context) {
|
||
c.HTML(http.StatusOK, "login.html", nil)
|
||
})
|
||
r.POST("/admin/login", handlers.LoginHandler)
|
||
r.GET("/admin/logout", handlers.LogoutHandler)
|
||
|
||
admin := r.Group("/admin")
|
||
admin.Use(handlers.AuthRequired())
|
||
{
|
||
admin.GET("", func(c *gin.Context) {
|
||
c.Redirect(http.StatusFound, "/admin/dashboard")
|
||
})
|
||
admin.GET("/dashboard", handlers.DashboardHandler)
|
||
admin.GET("/services", handlers.ServicesPageHandler)
|
||
admin.GET("/categories", handlers.CategoriesPageHandler)
|
||
admin.GET("/change-password", handlers.ChangePasswordHandler)
|
||
admin.POST("/change-password", handlers.ChangePasswordHandler)
|
||
|
||
admin.GET("/api/services", handlers.GetServices)
|
||
admin.POST("/api/services", handlers.SaveService)
|
||
admin.PUT("/api/services/:id", handlers.SaveService)
|
||
admin.DELETE("/api/services/:id", handlers.DeleteService)
|
||
admin.GET("/api/categories", handlers.GetCategories)
|
||
admin.POST("/api/categories", handlers.SaveCategory)
|
||
admin.PUT("/api/categories/:id", handlers.SaveCategory)
|
||
admin.DELETE("/api/categories/:id", handlers.DeleteCategory)
|
||
admin.GET("/api/settings", handlers.GetSettings)
|
||
admin.POST("/api/settings", handlers.SaveSettings)
|
||
admin.POST("/api/backup/webdav", handlers.RunCloudBackup)
|
||
admin.GET("/api/backup/list", handlers.ListCloudBackups)
|
||
admin.DELETE("/api/backup/delete", handlers.DeleteCloudBackup)
|
||
admin.POST("/api/backup/restore", handlers.RestoreCloudBackup)
|
||
|
||
// MCP 管理（后台 UI）
|
||
admin.GET("/api/mcp/status", handlers.MCPStatus)
|
||
admin.GET("/api/mcp/token", handlers.MCPGetToken)
|
||
admin.POST("/api/mcp/token", handlers.MCPRegenerateToken)
|
||
admin.POST("/api/mcp/enabled", handlers.MCPSetEnabled)
|
||
}
|
||
|
||
// MCP (Model Context Protocol) 端点
|
||
// POST: 请求-响应模式；GET: SSE 流式（服务端主动推送 notifications/tools/list_changed）
|
||
// 鉴权中间件动态读数据库：后台开关/Token 改动即时生效，无需重启
|
||
r.POST("/mcp", handlers.MCPAuthRequired(), handlers.MCPHandler)
|
||
r.GET("/mcp", handlers.MCPAuthRequired(), handlers.MCPSSEHandler)
|
||
|
||
log.Printf("ToNav-go 启动在端口 %s...", cfg.Port)
|
||
r.Run(":" + cfg.Port)
|
||
}
|