engigu
58 lines
2.1 KiB
Go
58 lines
2.1 KiB
Go
package router
|
||
|
||
import (
|
||
"github.com/engigu/baihu-panel/internal/constant"
|
||
"github.com/engigu/baihu-panel/internal/controllers"
|
||
"github.com/engigu/baihu-panel/internal/services"
|
||
"github.com/engigu/baihu-panel/internal/services/tasks"
|
||
)
|
||
|
||
var cronService *tasks.CronService
|
||
|
||
func RegisterControllers() *Controllers {
|
||
// Initialize services
|
||
settingsService := services.NewSettingsService()
|
||
loginLogService := services.NewLoginLogService()
|
||
|
||
// 执行系统初始化（返回 userService）
|
||
initService := services.NewInitService(settingsService)
|
||
userService := initService.Initialize()
|
||
|
||
taskService := tasks.NewTaskService()
|
||
envService := services.NewEnvService()
|
||
scriptService := services.NewScriptService()
|
||
sendStatsService := services.NewSendStatsService()
|
||
agentWSManager := services.GetAgentWSManager()
|
||
|
||
// 创建任务执行服务（需要依赖注入）
|
||
taskExecutionService := tasks.NewTaskExecutionService(agentWSManager, sendStatsService)
|
||
executorService := tasks.NewExecutorService(taskService, taskExecutionService, settingsService, envService)
|
||
|
||
// Initialize cron service
|
||
cronService = tasks.NewCronService(taskService, executorService)
|
||
cronService.Start()
|
||
|
||
// Initialize and return controllers
|
||
return &Controllers{
|
||
Task: controllers.NewTaskController(taskService, cronService),
|
||
Auth: controllers.NewAuthController(userService, settingsService, loginLogService),
|
||
Env: controllers.NewEnvController(envService),
|
||
Script: controllers.NewScriptController(scriptService),
|
||
Executor: controllers.NewExecutorController(executorService),
|
||
File: controllers.NewFileController(constant.ScriptsWorkDir),
|
||
Dashboard: controllers.NewDashboardController(cronService, executorService),
|
||
Log: controllers.NewLogController(),
|
||
Terminal: controllers.NewTerminalController(),
|
||
Settings: controllers.NewSettingsController(userService, loginLogService, executorService),
|
||
Dependency: controllers.NewDependencyController(),
|
||
Agent: controllers.NewAgentController(),
|
||
}
|
||
}
|
||
|
||
// StopCron stops the cron service gracefully
|
||
func StopCron() {
|
||
if cronService != nil {
|
||
cronService.Stop()
|
||
}
|
||
}
|