172 lines
7.3 KiB
Go
172 lines
7.3 KiB
Go
package initialize
|
||
|
||
import (
|
||
"net/http"
|
||
"os"
|
||
|
||
"git.echol.cn/loser/st/server/docs"
|
||
"git.echol.cn/loser/st/server/global"
|
||
"git.echol.cn/loser/st/server/middleware"
|
||
"git.echol.cn/loser/st/server/router"
|
||
"github.com/gin-gonic/gin"
|
||
swaggerFiles "github.com/swaggo/files"
|
||
ginSwagger "github.com/swaggo/gin-swagger"
|
||
)
|
||
|
||
type justFilesFilesystem struct {
|
||
fs http.FileSystem
|
||
}
|
||
|
||
func (fs justFilesFilesystem) Open(name string) (http.File, error) {
|
||
f, err := fs.fs.Open(name)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
stat, err := f.Stat()
|
||
if stat.IsDir() {
|
||
return nil, os.ErrPermission
|
||
}
|
||
|
||
return f, nil
|
||
}
|
||
|
||
// 初始化总路由
|
||
|
||
func Routers() *gin.Engine {
|
||
Router := gin.New()
|
||
|
||
// 设置文件上传大小限制（10MB）
|
||
Router.MaxMultipartMemory = 10 << 20 // 10 MB
|
||
|
||
// 使用自定义的 Recovery 中间件，记录 panic 并入库
|
||
Router.Use(middleware.GinRecovery(true))
|
||
if gin.Mode() == gin.DebugMode {
|
||
Router.Use(gin.Logger())
|
||
}
|
||
|
||
// 跨域配置（前台应用需要）
|
||
// 必须在静态文件路由之前注册，否则静态文件跨域会失败
|
||
Router.Use(middleware.Cors())
|
||
global.GVA_LOG.Info("use middleware cors")
|
||
|
||
if !global.GVA_CONFIG.MCP.Separate {
|
||
|
||
sseServer := McpRun()
|
||
|
||
// 注册mcp服务
|
||
Router.GET(global.GVA_CONFIG.MCP.SSEPath, func(c *gin.Context) {
|
||
sseServer.SSEHandler().ServeHTTP(c.Writer, c.Request)
|
||
})
|
||
|
||
Router.POST(global.GVA_CONFIG.MCP.MessagePath, func(c *gin.Context) {
|
||
sseServer.MessageHandler().ServeHTTP(c.Writer, c.Request)
|
||
})
|
||
}
|
||
|
||
systemRouter := router.RouterGroupApp.System
|
||
exampleRouter := router.RouterGroupApp.Example
|
||
appRouter := router.RouterGroupApp.App // 前台应用路由
|
||
|
||
// SillyTavern 核心脚本静态文件服务
|
||
// 扩展通过 ES module 相对路径 import 引用这些核心模块（如 ../../../../../script.js → /script.js）
|
||
// 所有核心文件存储在 data/st-core-scripts/ 下，完全独立于 web-app/ 目录
|
||
// 扩展文件存储在 data/st-core-scripts/scripts/extensions/third-party/{name}/ 下
|
||
stCorePath := "data/st-core-scripts"
|
||
if _, err := os.Stat(stCorePath); err == nil {
|
||
Router.Static("/scripts", stCorePath+"/scripts")
|
||
Router.Static("/css", stCorePath+"/css")
|
||
Router.Static("/img", stCorePath+"/img")
|
||
Router.Static("/webfonts", stCorePath+"/webfonts")
|
||
Router.Static("/lib", stCorePath+"/lib") // SillyTavern 扩展依赖的第三方库
|
||
Router.Static("/locales", stCorePath+"/locales") // 国际化文件
|
||
Router.StaticFile("/script.js", stCorePath+"/script.js") // SillyTavern 主入口
|
||
Router.StaticFile("/lib.js", stCorePath+"/lib.js") // Webpack 编译后的 lib.js
|
||
global.GVA_LOG.Info("SillyTavern 核心脚本服务已启动: " + stCorePath)
|
||
} else {
|
||
global.GVA_LOG.Warn("SillyTavern 核心脚本目录不存在: " + stCorePath + "，扩展功能将不可用")
|
||
}
|
||
|
||
// 管理后台前端静态文件（web）
|
||
// 如果想要不使用nginx代理前端网页，可以修改 web/.env.production 下的
|
||
// VUE_APP_BASE_API = /
|
||
// VUE_APP_BASE_PATH = http://localhost
|
||
// 然后执行打包命令 npm run build。在打开下面3行注释
|
||
// Router.StaticFile("/favicon.ico", "./dist/favicon.ico")
|
||
// Router.Static("/assets", "./dist/assets") // dist里面的静态资源
|
||
// Router.StaticFile("/", "./dist/index.html") // 前端网页入口页面
|
||
|
||
Router.StaticFS(global.GVA_CONFIG.Local.StorePath, justFilesFilesystem{http.Dir(global.GVA_CONFIG.Local.StorePath)}) // Router.Use(middleware.LoadTls()) // 如果需要使用https 请打开此中间件 然后前往 core/server.go 将启动模式 更变为 Router.RunTLS("端口","你的cre/pem文件","你的key文件")
|
||
|
||
docs.SwaggerInfo.BasePath = global.GVA_CONFIG.System.RouterPrefix
|
||
Router.GET(global.GVA_CONFIG.System.RouterPrefix+"/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
|
||
global.GVA_LOG.Info("register swagger handler")
|
||
// 方便统一添加路由组前缀 多服务器上线使用
|
||
|
||
PublicGroup := Router.Group(global.GVA_CONFIG.System.RouterPrefix)
|
||
PrivateGroup := Router.Group(global.GVA_CONFIG.System.RouterPrefix)
|
||
|
||
PrivateGroup.Use(middleware.JWTAuth()).Use(middleware.CasbinHandler())
|
||
|
||
{
|
||
// 健康监测
|
||
PublicGroup.GET("/health", func(c *gin.Context) {
|
||
c.JSON(http.StatusOK, "ok")
|
||
})
|
||
}
|
||
{
|
||
systemRouter.InitBaseRouter(PublicGroup) // 注册基础功能路由 不做鉴权
|
||
systemRouter.InitInitRouter(PublicGroup) // 自动初始化相关
|
||
}
|
||
|
||
{
|
||
systemRouter.InitApiRouter(PrivateGroup, PublicGroup) // 注册功能api路由
|
||
systemRouter.InitJwtRouter(PrivateGroup) // jwt相关路由
|
||
systemRouter.InitUserRouter(PrivateGroup) // 注册用户路由
|
||
systemRouter.InitMenuRouter(PrivateGroup) // 注册menu路由
|
||
systemRouter.InitSystemRouter(PrivateGroup) // system相关路由
|
||
systemRouter.InitSysVersionRouter(PrivateGroup) // 发版相关路由
|
||
systemRouter.InitCasbinRouter(PrivateGroup) // 权限相关路由
|
||
systemRouter.InitAutoCodeRouter(PrivateGroup, PublicGroup) // 创建自动化代码
|
||
systemRouter.InitAuthorityRouter(PrivateGroup) // 注册角色路由
|
||
systemRouter.InitSysDictionaryRouter(PrivateGroup) // 字典管理
|
||
systemRouter.InitAutoCodeHistoryRouter(PrivateGroup) // 自动化代码历史
|
||
systemRouter.InitSysOperationRecordRouter(PrivateGroup) // 操作记录
|
||
systemRouter.InitSysDictionaryDetailRouter(PrivateGroup) // 字典详情管理
|
||
systemRouter.InitAuthorityBtnRouterRouter(PrivateGroup) // 按钮权限管理
|
||
systemRouter.InitSysExportTemplateRouter(PrivateGroup, PublicGroup) // 导出模板
|
||
systemRouter.InitSysParamsRouter(PrivateGroup, PublicGroup) // 参数管理
|
||
systemRouter.InitSysErrorRouter(PrivateGroup, PublicGroup) // 错误日志
|
||
systemRouter.InitLoginLogRouter(PrivateGroup) // 登录日志
|
||
systemRouter.InitApiTokenRouter(PrivateGroup) // apiToken签发
|
||
systemRouter.InitSkillsRouter(PrivateGroup) // Skills 定义器
|
||
exampleRouter.InitCustomerRouter(PrivateGroup) // 客户路由
|
||
exampleRouter.InitFileUploadAndDownloadRouter(PrivateGroup) // 文件上传下载功能路由
|
||
exampleRouter.InitAttachmentCategoryRouterRouter(PrivateGroup) // 文件上传下载分类
|
||
|
||
}
|
||
|
||
// 前台应用路由（新增）
|
||
{
|
||
appGroup := PublicGroup.Group("app") // 统一使用 /app 前缀
|
||
appRouter.InitAuthRouter(appGroup) // 认证路由：/app/auth/* 和 /app/user/*
|
||
appRouter.InitCharacterRouter(appGroup) // 角色卡路由：/app/character/*
|
||
appRouter.InitWorldInfoRouter(appGroup) // 世界书路由：/app/worldbook/*
|
||
appRouter.InitExtensionRouter(appGroup) // 扩展路由：/app/extension/*
|
||
appRouter.InitRegexScriptRouter(appGroup) // 正则脚本路由：/app/regex/*
|
||
appRouter.InitProviderRouter(appGroup) // AI提供商路由：/app/provider/*
|
||
appRouter.InitChatRouter(appGroup) // 对话路由：/app/chat/*
|
||
}
|
||
|
||
//插件路由安装
|
||
InstallPlugin(PrivateGroup, PublicGroup, Router)
|
||
|
||
// 注册业务路由
|
||
initBizRouter(PrivateGroup, PublicGroup)
|
||
|
||
global.GVA_ROUTERS = Router.Routes()
|
||
|
||
global.GVA_LOG.Info("router register success")
|
||
return Router
|
||
}
|