refactor helper function

zerowidth · Apr 3, 2023 · 19d553c · 19d553c
1 parent 3d1af35
commit 19d553c
Showing 1 changed file with 12 additions and 18 deletions.
diff --git a/main.go b/main.go
@@ -55,12 +55,7 @@ Examples:
 		if err != nil {
 			fmt.Fprintln(os.Stderr, err)
 		}
-		noNewline, _ := cmd.Flags().GetBool("no-newline")
-		if noNewline {
-			fmt.Print(out)
-		} else {
-			fmt.Println(out)
-		}
+		printOptionalNewline(cmd, out)
 	},
 }
 
@@ -88,12 +83,7 @@ Example:
 		if err != nil {
 			fmt.Fprintln(os.Stderr, err)
 		}
-		noNewline, _ := cmd.Flags().GetBool("no-newline")
-		if noNewline {
-			fmt.Print(out)
-		} else {
-			fmt.Println(out)
-		}
+		printOptionalNewline(cmd, out)
 	},
 }
 
@@ -121,12 +111,7 @@ Example:
 		if err != nil {
 			fmt.Fprintln(os.Stderr, err)
 		}
-		noNewline, _ := cmd.Flags().GetBool("no-newline")
-		if noNewline {
-			fmt.Print(out)
-		} else {
-			fmt.Println(out)
-		}
+		printOptionalNewline(cmd, out)
 	},
 }
 
@@ -213,3 +198,12 @@ func title(input string, sanitize bool) (string, error) {
 
 	return title, nil
 }
+
+func printOptionalNewline(cmd *cobra.Command, output string) {
+	noNewline, _ := cmd.Flags().GetBool("no-newline")
+	if noNewline {
+		fmt.Print(output)
+	} else {
+		fmt.Println(output)
+	}
+}