feat: add pre and post command

internal/models/agent.go

@@ -50,11 +50,13 @@ func (AgentToken) TableName() string {
 type AgentTask struct {
 	ID        string              `json:"id"`
 	Name      string              `json:"name"`
-	Command   string              `json:"command"`
-	Schedule  string              `json:"schedule"`
-	Timeout   int                 `json:"timeout"`
-	WorkDir   string              `json:"work_dir"`
-	Envs      string              `json:"envs"`
+	Command     string              `json:"command"`
+	PreCommand  string              `json:"pre_command"`
+	PostCommand string              `json:"post_command"`
+	Schedule    string              `json:"schedule"`
+	Timeout     int                 `json:"timeout"`
+	WorkDir     string              `json:"work_dir"`
+	Envs        string              `json:"envs"`
 	Languages   []map[string]string `json:"languages"`
 	RandomRange int                 `json:"random_range"`
 	Secrets     []string            `json:"secrets"`
@@ -73,6 +75,14 @@ func (t AgentTask) GetCommand() string {
 	return t.Command
 }
 
+func (t AgentTask) GetPreCommand() string {
+	return t.PreCommand
+}
+
+func (t AgentTask) GetPostCommand() string {
+	return t.PostCommand
+}
+
 func (t AgentTask) GetSchedule() string {
 	return t.Schedule
 }

internal/models/task.go

@@ -74,6 +74,8 @@ type Task struct {
 	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:"size:255;default:''"`            // 标签，逗号分隔
 	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.TriggerTypeBaihuStartup
@@ -123,6 +125,14 @@ 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
 }

internal/models/vo/task_vo.go

@@ -11,6 +11,8 @@ type TaskVO struct {
 	ID          string              `json:"id"`
 	Name        string              `json:"name"`
 	Command     string              `json:"command"`
+	PreCommand  string              `json:"pre_command"`
+	PostCommand string              `json:"post_command"`
 	Tags        string              `json:"tags"`
 	Type        string              `json:"type"`
 	TriggerType string              `json:"trigger_type"`
@@ -44,6 +46,8 @@ func ToTaskVO(task *models.Task) *TaskVO {
 		ID:          task.ID,
 		Name:        task.Name,
 		Command:     string(task.Command),
+		PreCommand:  string(task.PreCommand),
+		PostCommand: string(task.PostCommand),
 		Tags:        task.Tags,
 		Type:        task.Type,
 		TriggerType: task.TriggerType,