From aab771498bcd2e7d8cc5fd2911a4f167f7418a35 Mon Sep 17 00:00:00 2001
From: engigu <sayheya@qq.com>
Date: Fri, 13 Feb 2026 22:13:04 +0800
Subject: [PATCH] fix: demo running stop

---
 internal/executor/executor.go  | 26 ++++++++++++++++++++++++++
 internal/executor/scheduler.go | 29 +++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+)

diff --git a/internal/executor/executor.go b/internal/executor/executor.go
index 56732d8..88e1932 100644
--- a/internal/executor/executor.go
+++ b/internal/executor/executor.go
@@ -76,6 +76,32 @@ func Execute(ctx context.Context, req Request, stdout, stderr io.Writer) (*Resul
 func ExecuteWithHooks(ctx context.Context, req Request, stdout, stderr io.Writer, hooks Hooks) (*Result, error) {
 	start := time.Now()
 
+	// 演示模式拦截
+	if constant.DemoMode {
+		logger.Warnf("[Executor] 演示模式下已拦截命令执行: %s", req.Command)
+		if stdout != nil {
+			stdout.Write([]byte("\r\n\033[1;33m[演示模式] 命令执行已跳过\033[0m\r\n"))
+		}
+
+		// 仍然触发 PreExecute 以便流程完整
+		var logID uint
+		if hooks != nil {
+			logID, _ = hooks.PreExecute(ctx, req)
+		}
+
+		result := &Result{
+			Status:    constant.TaskStatusFailed,
+			Output:    "[演示模式] 该任务在演示模式下被禁用执行",
+			StartTime: start,
+			EndTime:   time.Now(),
+		}
+
+		if hooks != nil {
+			hooks.PostExecute(ctx, logID, result)
+		}
+		return result, nil
+	}
+
 	// 2. 执行命令
 	timeout := req.Timeout
 	if timeout <= 0 {
diff --git a/internal/executor/scheduler.go b/internal/executor/scheduler.go
index bbc5e25..6d2214a 100644
--- a/internal/executor/scheduler.go
+++ b/internal/executor/scheduler.go
@@ -314,6 +314,35 @@ func (s *Scheduler) executeTask(req *ExecutionRequest) (*ExecutionResult, error)
 
 	s.logger.Infof("[Scheduler] 执行任务 %s (名称: %s, 类型: %s)", req.TaskID, req.Name, req.Type)
 
+	// 演示模式拦截
+	if constant.DemoMode {
+		s.logger.Infof("[Scheduler] 演示模式下已跳过任务 %s (%s) 的执行", req.TaskID, req.Name)
+		// 仍然触发 OnTaskExecuting 以便创建初始日志记录（业务层面的 Handler 会处理）
+		var stdout io.Writer
+		if s.handler != nil {
+			stdout, _, _ = s.handler.OnTaskExecuting(req)
+		}
+
+		result := &ExecutionResult{
+			TaskID:    req.TaskID,
+			LogID:     req.LogID,
+			Success:   false,
+			Status:    constant.TaskStatusFailed,
+			Error:     "[演示模式] 该任务在演示模式下被禁用执行",
+			StartTime: start,
+			EndTime:   time.Now(),
+		}
+
+		if stdout != nil {
+			stdout.Write([]byte("\r\n\033[1;33m[演示模式] 定时任务/手动任务执行已跳过\033[0m\r\n"))
+		}
+
+		if s.handler != nil {
+			s.handler.OnTaskCompleted(req, result)
+		}
+		return result, nil
+	}
+
 	// 如果指定使用 mise，则预先构建好带 mise 的命令，这样 OnTaskExecuting 记录的就是完整命令
 	if req.UseMise {
 		req.Command = utils.BuildMiseCommand(req.Command, req.Languages)