From 85c0cde9c7bb57b9399b3907a595c41b3c8e0e83 Mon Sep 17 00:00:00 2001
From: Christian Muehlhaeuser <muesli@gmail.com>
Date: Wed, 8 Jan 2020 21:50:34 +0100
Subject: [PATCH] Pass writer to executeArg to make code easier to test

---
 main.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 189d779..0d53e02 100644
--- a/main.go
+++ b/main.go
@@ -102,18 +102,18 @@ func readerFromArg(s string) (*Source, error) {
 
 func execute(cmd *cobra.Command, args []string) error {
 	if len(args) == 0 {
-		return executeArg(cmd, "")
+		return executeArg(cmd, "", os.Stdout)
 	}
 
 	for _, arg := range args {
-		if err := executeArg(cmd, arg); err != nil {
+		if err := executeArg(cmd, arg, os.Stdout); err != nil {
 			return err
 		}
 	}
 	return nil
 }
 
-func executeArg(cmd *cobra.Command, arg string) error {
+func executeArg(cmd *cobra.Command, arg string, w io.Writer) error {
 	// create an io.Reader from the markdown source in cli-args
 	src, err := readerFromArg(arg)
 	if err != nil {
@@ -180,7 +180,7 @@ func executeArg(cmd *cobra.Command, arg string) error {
 		return c.Run()
 	}
 
-	fmt.Print(content)
+	fmt.Fprint(w, content)
 	return nil
 }