Add SIGINT handling for Git Lfs, refactor

bringauto · Nov 7, 2024 · 58930c0 · 58930c0
1 parent fb50fe3
commit 58930c0
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/bap-builder/PackageMode.go b/bap-builder/PackageMode.go
@@ -241,20 +241,19 @@ func BuildPackage(cmdLine *BuildPackageCmdLineArgs, contextPath string) error {
 	if err != nil {
 		return err
 	}
-	buildAll := cmdLine.All
-	if *buildAll {
+
+	handleRemover := bringauto_process.SignalHandlerAddHandler(repo.RestoreAllChanges)
+	if *cmdLine.All {
 		err = buildAllPackages(cmdLine, contextPath, platformString, repo)
-		if err != nil {
-			repo.RestoreAllChanges()
-			return err
-		}
+	} else {
+		err = buildSinglePackage(cmdLine, contextPath, platformString, repo)
 	}
-	err = buildSinglePackage(cmdLine, contextPath, platformString, repo)
 	if err != nil {
-		repo.RestoreAllChanges()
+		handleRemover()
 		return err
 	}
 	repo.CommitAllChanges()
+	handleRemover()
 	return nil
 }