Initial commit: TaskPool React panel

- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
2026-07-26 08:43:52 +08:00
commit e6956aa001
397 changed files with 73621 additions and 0 deletions
@@ -0,0 +1,196 @@
+package models
+
+import (
+	"database/sql/driver"
+	"encoding/json"
+	"fmt"
+	"github.com/engigu/taskpool/internal/constant"
+)
+
+// TaskLanguages 自定义语言配置列表类型，处理 JSON 序列化
+type TaskLanguages []map[string]string
+
+func (t TaskLanguages) Value() (driver.Value, error) {
+	if t == nil {
+		return "[]", nil
+	}
+	b, err := json.Marshal(t)
+	return string(b), err
+}
+
+func (t *TaskLanguages) Scan(v interface{}) error {
+	if v == nil {
+		*t = nil
+		return nil
+	}
+	var data []byte
+	switch s := v.(type) {
+	case string:
+		data = []byte(s)
+	case []byte:
+		data = s
+	default:
+		return fmt.Errorf("invalid type for TaskLanguages: %T", v)
+	}
+	return json.Unmarshal(data, t)
+}
+
+// CleanConfig 清理配置结构
+type CleanConfig struct {
+	Type string `json:"type"` // "day" 或 "count"
+	Keep int    `json:"keep"` // 保留天数或条数
+}
+
+// RepoConfig 仓库同步配置
+type RepoConfig struct {
+	SourceType     string `json:"source_type"`     // url 或 git
+	SourceURL      string `json:"source_url"`      // 源地址
+	TargetPath     string `json:"target_path"`     // 目标路径
+	Branch         string `json:"branch"`          // Git 分支
+	SparsePath     string `json:"sparse_path"`     // 稀疏检出路径（仅拉取指定目录或文件）
+	SingleFile     bool   `json:"single_file"`     // 单文件模式（直接下载文件而非 sparse-checkout）
+	Proxy          string `json:"proxy"`           // 代理类型: none, ghproxy, mirror, custom
+	ProxyURL       string `json:"proxy_url"`       // 自定义代理地址
+	AuthToken      string `json:"auth_token"`      // 认证 Token
+	WhitelistPaths string `json:"whitelist_paths"` // 同步时保留的路径及脚本筛选白名单关键词，逗号或竖线分割
+	Blacklist      string `json:"blacklist"`       // 脚本筛选黑名单关键词，竖线分割
+	Dependence     string `json:"dependence"`      // 脚本依赖文件关键词，竖线分割
+	Extensions     string `json:"extensions"`      // 脚本文件后缀关键词，竖线分割
+	AutoAddCron    bool   `json:"auto_add_cron"`   // 自动解析脚本注释添加定时任务
+	CommentToTask  string `json:"commenttotask"`   // 兼容 QL 格式任务脚本注释解析
+	RepoSource     string `json:"repo_source"`     // 仓库来源，如果是选择了这个 ql 导入的仓库，= ql
+	RepoDirName    string `json:"repo_dir_name"`   // 自定义仓库目录名
+}
+
+// TaskConfig  任务配置  RepoConfig+TaskConfig=task.config
+type TaskConfig struct {
+	Concurrency int  `json:"$task_concurrency"` // 0: disable concurrency, 1: enable concurrency
+	AllEnvs     bool `json:"$task_all_envs"`    // 开启则注入全部环境变量
+}
+
+// Task 代表一个计划任务
+type Task struct {
+	ID             string        `json:"id" gorm:"primaryKey;size:20"`
+	Name           string        `json:"name" gorm:"size:255;not null"`
+	Remark         string        `json:"remark" gorm:"size:255;default:''"`
+	PinType        string        `json:"pin_type" gorm:"size:20;default:none;index"` // 置顶类型: constant.PinTypeNone, constant.PinTypeTop
+	Command        BigText       `json:"command"`                                    // 普通任务的命令
+	PreCommand     BigText       `json:"pre_command"`                                // 执行前的命令
+	PostCommand    BigText       `json:"post_command"`                               // 执行后的命令
+	Tags           string        `json:"tags" gorm:"-"`                              // 标签，逗号分隔
+	Type           string        `json:"type" gorm:"size:20;default:'task'"`         // 任务类型: constant.TaskTypeNormal, constant.TaskTypeRepo
+	TriggerType    string        `json:"trigger_type" gorm:"size:25;default:'cron'"` // 触发类型: constant.TriggerTypeCron, constant.TriggerTypeTaskPoolStartup
+	Config         BigText       `json:"config"`                                     // 配置 JSON（仓库同步配置等）
+	Schedule       string        `json:"schedule" gorm:"size:100"`                   // cron 表达式
+	Timeout        int           `json:"timeout" gorm:"default:30"`                  // 超时时间（分钟），默认30分钟
+	WorkDir        string        `json:"work_dir" gorm:"size:255;default:''"`        // 工作目录，为空则使用 scripts 目录
+	CleanConfig    string        `json:"clean_config" gorm:"size:255;default:''"`    // 清理配置 JSON
+	Envs           BigText       `json:"envs" gorm:"-"`                              // 环境变量ID列表，逗号分隔
+	Languages      TaskLanguages `json:"languages" gorm:"type:text"`                 // 针对本地任务的语言配置列表
+	AgentID        *string       `json:"agent_id" gorm:"size:20;index"`              // Agent ID，为空表示本地执行
+	RetryCount     int           `json:"retry_count" gorm:"default:0"`               // 失败重试次数
+	RetryInterval  int           `json:"retry_interval" gorm:"default:0"`            // 失败重试间隔(秒)
+	RandomRange    int           `json:"random_range" gorm:"default:0"`              // 随机延迟范围(秒)
+	Enabled        *bool         `json:"enabled" gorm:"default:true"`
+	RunningGo      BigText       `json:"running_go"` // 正在运行的 go routine id 数组 (JSON)
+	RuntimeEnvs    []string      `json:"-" gorm:"-"` // 运行时环境变量（非持久化）
+	RuntimeSecrets []string      `json:"-" gorm:"-"` // 运行时安全机密（非持久化）
+	LastRun        *LocalTime    `json:"last_run"`
+	NextRun        *LocalTime    `json:"next_run"`
+	SourceID       string        `json:"source_id" gorm:"size:255;index"`   // 脚本资源唯一标识（路径 sanitized）
+	RepoTaskID     string        `json:"repo_task_id" gorm:"size:20;index"` // 所属的仓库任务 ID
+	CreatedAt      LocalTime     `json:"created_at"`
+	UpdatedAt      LocalTime     `json:"updated_at"`
+}
+
+func (t *Task) IsRunning() bool {
+	if string(t.RunningGo) == "" || string(t.RunningGo) == "[]" {
+		return false
+	}
+	return true
+}
+
+func (Task) TableName() string {
+	return constant.TablePrefix + "tasks"
+}
+
+func (t *Task) GetID() string {
+	return t.ID
+}
+
+func (t *Task) GetName() string {
+	return t.Name
+}
+
+func (t *Task) GetCommand() string {
+	return string(t.Command)
+}
+
+func (t *Task) GetPreCommand() string {
+	return string(t.PreCommand)
+}
+
+func (t *Task) GetPostCommand() string {
+	return string(t.PostCommand)
+}
+
+func (t *Task) GetTimeout() int {
+	return t.Timeout
+}
+
+func (t *Task) GetWorkDir() string {
+	return t.WorkDir
+}
+
+func (t *Task) GetEnvs() string {
+	return string(t.Envs)
+}
+
+func (t *Task) GetLanguages() []map[string]string {
+	return []map[string]string(t.Languages)
+}
+
+func (t *Task) GetEnvVars() []string {
+	return t.RuntimeEnvs
+}
+
+func (t *Task) GetSecrets() []string {
+	return t.RuntimeSecrets
+}
+
+func (t *Task) GetUseMise() bool {
+	return t.AgentID == nil || *t.AgentID == ""
+}
+
+func (t *Task) UseMise() bool {
+	return t.GetUseMise()
+}
+
+// CronTask 计划任务接口
+func (t *Task) GetSchedule() string {
+	return t.Schedule
+}
+
+func (t *Task) GetRandomRange() int {
+	return t.RandomRange
+}
+
+// TaskLog 代表任务执行的日志记录
+type TaskLog struct {
+	ID        string     `json:"id" gorm:"primaryKey;size:20"`
+	TaskID    string     `json:"task_id" gorm:"size:20;index"`
+	AgentID   *string    `json:"agent_id" gorm:"size:20;index"` // Agent ID，为空表示本地执行
+	Command   BigText    `json:"command"`
+	Output    BigText    `json:"-"`                           // gzip+base64 压缩后的日志
+	Error     BigText    `json:"error"`                       // 额外的系统错误信息
+	Status    string     `json:"status" gorm:"size:20;index"` // success, failed
+	Duration  int64      `json:"duration"`                    // 执行耗时（毫秒）
+	ExitCode  int        `json:"exit_code"`
+	StartTime *LocalTime `json:"start_time"`
+	EndTime   *LocalTime `json:"end_time"`
+	CreatedAt LocalTime  `json:"created_at"`
+}
+
+func (TaskLog) TableName() string {
+	return constant.TablePrefix + "task_logs"
+}