engigu
49 lines
1.5 KiB
Go
49 lines
1.5 KiB
Go
package router
|
|
|
|
import (
|
|
"baihu/internal/constant"
|
|
"baihu/internal/controllers"
|
|
"baihu/internal/services"
|
|
)
|
|
|
|
var cronService *services.CronService
|
|
|
|
func RegisterControllers() *Controllers {
|
|
// Initialize services
|
|
taskService := services.NewTaskService()
|
|
userService := services.NewUserService()
|
|
envService := services.NewEnvService()
|
|
scriptService := services.NewScriptService()
|
|
executorService := services.NewExecutorService(taskService)
|
|
settingsService := services.NewSettingsService()
|
|
|
|
// 执行系统初始化
|
|
initService := services.NewInitService(settingsService, userService)
|
|
initService.Initialize()
|
|
|
|
// Initialize cron service
|
|
cronService = services.NewCronService(taskService, executorService)
|
|
cronService.Start()
|
|
|
|
// Initialize and return controllers
|
|
return &Controllers{
|
|
Task: controllers.NewTaskController(taskService, cronService),
|
|
Auth: controllers.NewAuthController(userService),
|
|
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),
|
|
}
|
|
}
|
|
|
|
// StopCron stops the cron service gracefully
|
|
func StopCron() {
|
|
if cronService != nil {
|
|
cronService.Stop()
|
|
}
|
|
}
|