From 4f501c5d65054a6ed7ad9770795e2aa8b5b3f668 Mon Sep 17 00:00:00 2001
From: Diff <diff@tilde.town>
Date: Fri, 13 Nov 2020 04:16:43 +0000
Subject: [PATCH] Fixed handling of blank inputs.

Now properly removes ~/.venture when input is blank, and returns succesfully if
the file came already pre-non-existing.
---
 main.go | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 1ca64c5..5cf1293 100644
--- a/main.go
+++ b/main.go
@@ -9,6 +9,7 @@ import (
 	"io"
 	"io/ioutil"
 	"bufio"
+	"strings"
 )
 
 func main() {
@@ -18,8 +19,7 @@ func main() {
 		err := SetVenture(flag.Args()[1:])
 		if err != nil {
 			fmt.Fprintf(os.Stderr, "Could not set: %v", err)
-		}
-		
+		}		
 	default:
 		fmt.Printf("Usage:\n\t%v set\n", os.Args[0])
 		flag.PrintDefaults()
@@ -45,9 +45,12 @@ func SetVenture(args []string) error {
 	}
 
 	// Remove file on blank input
-	if len(input) < 1 {
+	if strings.TrimSpace(input) == "" {
 		err = os.Remove(outputPath)
-		if err != nil {
+		// File already pre-non-existing is considered success.
+		if os.IsNotExist(err) {
+			return nil
+		} else if err != nil {
 			return err
 		}
 		return nil