feat: Initial commit

internal/controllers/dashboard_controller.go

@@ -0,0 +1,51 @@
+package controllers
+
+import (
+	"baihu/internal/database"
+	"baihu/internal/models"
+	"baihu/internal/services"
+	"baihu/internal/utils"
+
+	"github.com/gin-gonic/gin"
+)
+
+type DashboardController struct {
+	cronService     *services.CronService
+	executorService *services.ExecutorService
+}
+
+func NewDashboardController(cronService *services.CronService, executorService *services.ExecutorService) *DashboardController {
+	return &DashboardController{
+		cronService:     cronService,
+		executorService: executorService,
+	}
+}
+
+type StatsResponse struct {
+	Tasks     int64 `json:"tasks"`
+	Scripts   int64 `json:"scripts"`
+	Envs      int64 `json:"envs"`
+	Logs      int64 `json:"logs"`
+	Scheduled int   `json:"scheduled"`
+	Running   int   `json:"running"`
+}
+
+func (dc *DashboardController) GetStats(c *gin.Context) {
+	var taskCount, scriptCount, envCount, logCount int64
+
+	database.DB.Model(&models.Task{}).Count(&taskCount)
+	database.DB.Model(&models.Script{}).Count(&scriptCount)
+	database.DB.Model(&models.EnvironmentVariable{}).Count(&envCount)
+	database.DB.Model(&models.TaskLog{}).Count(&logCount)
+
+	stats := StatsResponse{
+		Tasks:     taskCount,
+		Scripts:   scriptCount,
+		Envs:      envCount,
+		Logs:      logCount,
+		Scheduled: dc.cronService.GetScheduledCount(),
+		Running:   dc.executorService.GetRunningCount(),
+	}
+
+	utils.Success(c, stats)
+}