fix: make exit calls only from main package

ivanilves · Oct 15, 2022 · 3ca0067 · 3ca0067
1 parent 3572e99
commit 3ca0067
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/cmd/ttg/main.go b/cmd/ttg/main.go
@@ -70,9 +70,11 @@ func main() {
 	}
 
 	if outFile != "" {
-		if err := file.WriteAndExit(outFile, entries[selected]); err != nil {
+		if err := file.Write(outFile, entries[selected]); err != nil {
 			log.Fatalf("failed to write output file: %s", err.Error())
 		}
+
+		os.Exit(0)
 	}
 
 	shell.Spawn(entries[selected])

diff --git a/pkg/file/file.go b/pkg/file/file.go
@@ -4,12 +4,8 @@ import (
 	"os"
 )
 
-func WriteAndExit(outFile string, path string) error {
+func Write(outFile string, path string) error {
 	err := os.WriteFile(outFile, []byte(path), 0644)
 
-	if err == nil {
-		os.Exit(0)
-	}
-
 	return err
 }