go-svc-configure/main.go

package main

import (
	"context"
	"embed"
	"io/fs"
	"log"
	"net/http"
	"strings"

	"git.x2erp.com/qdy/go-base/authbase"
	"git.x2erp.com/qdy/go-base/config"
	"git.x2erp.com/qdy/go-base/container"
	"git.x2erp.com/qdy/go-base/ctx"
	"git.x2erp.com/qdy/go-base/graceful"
	"git.x2erp.com/qdy/go-base/logger"
	"git.x2erp.com/qdy/go-base/model/request/configreq"
	"git.x2erp.com/qdy/go-base/model/response"
	"git.x2erp.com/qdy/go-base/webx"
	"git.x2erp.com/qdy/go-base/webx/router"
	"git.x2erp.com/qdy/go-db/factory/database"

	"git.x2erp.com/qdy/go-db/sqldef"

	"git.x2erp.com/qdy/go-svc-configure/internal/service/admin"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/project"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/projectagent"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/register"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/role"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/skill"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/startup"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/svcstartup"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/tenant"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/token"
	"git.x2erp.com/qdy/go-svc-configure/internal/service/user"
	"git.x2erp.com/qdy/go-svc-configure/internal/tables"
	_ "git.x2erp.com/qdy/go-svc-configure/internal/tables" // 导入表定义包，触发 init() 函数
)

//go:embed web/dist
var frontendFS embed.FS

var (
	appName    = "svc-configure"
	appVersion = "1"
)

func main() {

	// 0. 初始化日志
	logBootFactory := logger.InitBootLog()

	// 1. 获取配置文件
	cfg := config.GetConfig()
	cfg.SetAppName(appName)
	cfg.SetAppVersion(appVersion)

	// 2. 创建关闭容器
	ctr := container.NewContainer(cfg)

	// 注册日志，实现自动关闭
	container.Reg(ctr, logBootFactory)

	// 3. 创建数据库工厂
	dbFactory := container.Create(ctr, database.CreateDBFactory)
	dbFactory.TestConnection()
	// 创建表
	creteTabel(dbFactory)

	//启用运行日志
	container.Create(ctr, logger.InitRuntimeLogger)

	//建立httpservice
	//得到webservice服务工厂
	webxFactory := webx.GetWebServiceFactory()

	//建立hhtpService服务
	webServcie, _ := webxFactory.CreateService(cfg.GetServiceConfig())

	//建立路由-api
	routerService := router.NewWebService(webServcie.GetRouter())
	//注册路由--api
	registerRoutes(routerService, dbFactory)

	// 注册前端静态文件服务
	frontendHandler := ServeFrontend()
	webServcie.GetRouter().Handle("/", frontendHandler)

	//启动服务
	webServcie.Run()

	//等待关闭
	graceful.WaitForShutdown(cfg.GetServiceConfig().ServiceName, ctr, webServcie.GetServer())

}

// 注册所有路由
func registerRoutes(ws *router.RouterService, dbFactory *database.DBFactory) {
	// 建立访问配置中心Token
	ws.POST("/api/create/config/token",
		func(tokenRequest *configreq.ConfigTokenRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[string], error) {
			return token.CreateConfigureToken(tokenRequest, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Desc("用户创建自身token").Register()

	// svc
	ws.POST("/api/delete/config/startup/{servicename}",
		func(servicename string, req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return svcstartup.DeleteSVCtartupConfig(servicename, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Register()

	ws.POST("/api/create/config/startup/{servicename}",
		func(servicename string, req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return svcstartup.CreateStartupSVCConfig(servicename, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Register()

	// startup
	ws.POST("/api/delete/config/startup",
		func(req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return startup.DeleteStartupConfig(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Register()

	ws.POST("/api/create/config/startup",
		func(req *configreq.ConfigRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return startup.CreateStartupConfig(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Register()

	//init
	ws.POST("/api/init/config/meta",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[map[string]interface{}], error) {
			return register.RegisterConfigMeta(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Register()

	// project skill
	ws.POST("/api/create/config/project/skill",
		func(req *configreq.ProjectSkillRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return skill.CreateProjectSkill(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建项目Skill").Register()

	ws.POST("/api/update/config/project/skill/{skill_id}",
		func(skillID string, req *configreq.ProjectSkillRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return skill.UpdateProjectSkill(skillID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新项目Skill").Register()

	ws.POST("/api/delete/config/project/skill/{skill_id}",
		func(skillID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return skill.DeleteProjectSkill(skillID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除项目Skill").Register()

	ws.POST("/api/query/config/project/skill/{skill_id}",
		func(skillID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.ProjectSkillDB], error) {
			return skill.GetProjectSkill(skillID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目Skill").Register()

	ws.POST("/api/query/config/project/skills",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.ProjectSkillDB], error) {
			return skill.ListProjectSkills(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目Skill列表").Register()

	// project
	ws.POST("/api/create/config/project",
		func(req *configreq.ProjectRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return project.CreateProject(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建项目").Register()

	ws.POST("/api/update/config/project/{project_id}",
		func(projectID string, req *configreq.ProjectRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return project.UpdateProject(projectID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新项目").Register()

	ws.POST("/api/delete/config/project/{project_id}",
		func(projectID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return project.DeleteProject(projectID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除项目").Register()

	ws.POST("/api/query/config/project/{project_id}",
		func(projectID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.ProjectDB], error) {
			return project.GetProject(projectID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目").Register()

	ws.POST("/api/query/config/projects",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.ProjectDB], error) {
			return project.ListProjects(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目列表").Register()

	// project agent
	ws.POST("/api/create/config/project/agent",
		func(req *configreq.ProjectAgentRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return projectagent.CreateProjectAgent(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建项目Agent").Register()

	ws.POST("/api/update/config/project/agent/{agent_id}",
		func(agentID string, req *configreq.ProjectAgentRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return projectagent.UpdateProjectAgent(agentID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新项目Agent").Register()

	ws.POST("/api/delete/config/project/agent/{agent_id}",
		func(agentID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return projectagent.DeleteProjectAgent(agentID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除项目Agent").Register()

	ws.POST("/api/query/config/project/agent/{agent_id}",
		func(agentID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.ProjectAgentDB], error) {
			return projectagent.GetProjectAgent(agentID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目Agent").Register()

	ws.POST("/api/query/config/project/agents",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.ProjectAgentDB], error) {
			return projectagent.ListProjectAgents(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询项目Agent列表").Register()

	// tenant
	ws.POST("/api/create/config/tenant",
		func(req *configreq.TenantRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return tenant.CreateTenant(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建租户").Register()

	ws.POST("/api/update/config/tenant/{tenant_id}",
		func(tenantID string, req *configreq.TenantRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return tenant.UpdateTenant(tenantID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新租户").Register()

	ws.POST("/api/delete/config/tenant/{tenant_id}",
		func(tenantID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return tenant.DeleteTenant(tenantID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除租户").Register()

	ws.POST("/api/query/config/tenant/{tenant_id}",
		func(tenantID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.TenantDB], error) {
			return tenant.GetTenant(tenantID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询租户").Register()

	ws.POST("/api/query/config/tenants",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.TenantDB], error) {
			return tenant.ListTenants(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询租户列表").Register()

	// role
	ws.POST("/api/create/config/role",
		func(req *configreq.RoleRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return role.CreateRole(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建角色").Register()

	ws.POST("/api/update/config/role/{role_id}",
		func(roleID string, req *configreq.RoleRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return role.UpdateRole(roleID, req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("更新角色").Register()

	ws.POST("/api/delete/config/role/{role_id}",
		func(roleID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return role.DeleteRole(roleID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("删除角色").Register()

	ws.POST("/api/query/config/role/{role_id}",
		func(roleID string, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[tables.RoleDB], error) {
			return role.GetRole(roleID, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询角色").Register()

	ws.POST("/api/query/config/roles",
		func(ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[[]tables.RoleDB], error) {
			return role.ListRoles(ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("查询角色列表").Register()

	// 用户注册（无需认证）
	ws.POST("/api/register/user",
		func(req *configreq.UserRegisterRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return user.RegisterUser(req, ctx, dbFactory, reqCtx), nil
		},
	).Desc("用户注册（需邀请码）").Register()

	// 用户登录（无需认证）
	ws.POST("/api/login/user",
		func(req *configreq.UserLoginRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[string], error) {
			return user.GetUserToken(req, ctx, dbFactory, reqCtx), nil
		},
	).Desc("用户登录（返回token）").Register()

	// 管理员登录（无需认证）
	ws.POST("/api/login/admin",
		func(req *configreq.UserLoginRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[bool], error) {
			return admin.LoginAdmin(req, ctx, reqCtx), nil
		},
	).Desc("管理员登录（验证配置文件凭据）").Register()

	// 创建租户管理员用户（BasicAuth认证）
	ws.POST("/api/create/tenant/admin",
		func(req *configreq.CreateTenantAdminRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[int64], error) {
			return user.CreateTenantAdmin(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.BasicAuth).Desc("创建租户管理员用户").Register()

	// 创建邀请码（TokenAuth认证）
	ws.POST("/api/create/invitation/code",
		func(req *configreq.CreateInvitationCodeRequest, ctx context.Context, reqCtx *ctx.RequestContext) (*response.QueryResult[string], error) {
			return user.CreateInvitationCode(req, ctx, dbFactory, reqCtx), nil
		},
	).Use(authbase.TokenAuth).Desc("创建邀请码").Register()

}

func creteTabel(factory *database.DBFactory) {

	// 获取数据库连接和类型
	db := factory.GetDB()
	dbType := factory.GetDBType()

	// 创建表同步器
	syncer, err := sqldef.NewTableSyncer(db, dbType)
	if err != nil {
		log.Printf("创建 - 建立器失败: %v", err)
	}

	// 创建表
	if err := syncer.CreateTables(); err != nil {
		log.Printf("建表失败: %v", err)
	}

	log.Println("数据库表建立完成！")
}

// ServeFrontend 返回处理前端静态文件的服务处理器
func ServeFrontend() http.Handler {
	// 从嵌入的文件系统中获取 dist 子目录
	fsys, err := fs.Sub(frontendFS, "web/dist")
	if err != nil {
		// 如果失败，回退到空文件系统
		return http.FileServer(http.FS(frontendFS))
	}

	// 创建自定义文件服务器，处理 SPA 路由
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// 只处理 GET 和 HEAD 请求
		if r.Method != "GET" && r.Method != "HEAD" {
			// 对于非 GET/HEAD 请求，返回 404 让 API 路由处理
			http.NotFound(w, r)
			return
		}

		// API 路径不处理
		if strings.HasPrefix(r.URL.Path, "/api/") {
			http.NotFound(w, r)
			return
		}

		// 尝试从文件系统读取请求的文件
		filePath := strings.TrimPrefix(r.URL.Path, "/")
		if filePath == "" {
			filePath = "index.html"
		}

		// 检查文件是否存在
		_, err := fs.Stat(fsys, filePath)
		if err != nil {
			// 文件不存在，返回 index.html 用于 SPA 路由
			indexData, err := frontendFS.ReadFile("web/dist/index.html")
			if err != nil {
				http.Error(w, "Not Found", http.StatusNotFound)
				return
			}
			w.Header().Set("Content-Type", "text/html; charset=utf-8")
			w.Write(indexData)
			return
		}

		// 文件存在，正常提供
		http.FileServer(http.FS(fsys)).ServeHTTP(w, r)
	})
}