From 27a2aac537d215f9d8bd055fa78d7b1391a50486 Mon Sep 17 00:00:00 2001
From: Boming Zhang <bomingzh@sjtu.edu.cn>
Date: Fri, 19 Jul 2024 07:12:18 -0400
Subject: [PATCH] feat: return 1 on force quit

---
 cmd/joj3/main.go      | 13 +++++++++++--
 internal/stage/run.go |  7 ++++---
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/cmd/joj3/main.go b/cmd/joj3/main.go
index b8e0e05..52da3fd 100644
--- a/cmd/joj3/main.go
+++ b/cmd/joj3/main.go
@@ -71,17 +71,26 @@ func outputResult(conf Conf, results []stage.StageResult) error {
 }
 
 func main() {
+	retCode := 0
+	defer func() {
+		os.Exit(retCode)
+	}()
 	conf, err := commitMsgToConf()
 	if err != nil {
 		slog.Error("no conf found", "error", err)
-		os.Exit(1)
+		retCode = 1
+		return
 	}
 	setupSlog(conf)
 	executors.InitWithConf(conf.SandboxExecServer, conf.SandboxToken)
 	stages := generateStages(conf)
 	defer stage.Cleanup()
-	results := stage.Run(stages)
+	results, stageErr := stage.Run(stages)
 	if err := outputResult(conf, results); err != nil {
 		slog.Error("output result", "error", err)
 	}
+	if stageErr != nil {
+		retCode = 1
+		return
+	}
 }
diff --git a/internal/stage/run.go b/internal/stage/run.go
index 670f666..46a0218 100644
--- a/internal/stage/run.go
+++ b/internal/stage/run.go
@@ -1,11 +1,11 @@
 package stage
 
 import (
+	"errors"
 	"log/slog"
 )
 
-func Run(stages []Stage) []StageResult {
-	stageResults := []StageResult{}
+func Run(stages []Stage) (stageResults []StageResult, stageErr error) {
 	for _, stage := range stages {
 		slog.Debug("stage start", "name", stage.Name)
 		slog.Debug("executor run start", "cmds", stage.ExecutorCmds)
@@ -38,10 +38,11 @@ func Run(stages []Stage) []StageResult {
 			ForceQuit: forceQuit,
 		})
 		if forceQuit {
+			stageErr = errors.New("parser force quit")
 			break
 		}
 	}
-	return stageResults
+	return
 }
 
 func Cleanup() {