refactor: move write file to cmd

cmd/arch.go

 package cmd
 
 import (
+	"bufio"
 	"github.com/phodal/coca/cmd/cmd_util"
 	"github.com/phodal/coca/config"
 	"github.com/phodal/coca/core/adapter"
 	"github.com/phodal/coca/core/domain/arch"
 	"github.com/spf13/cobra"
+	"os"
 	"strings"
 )
 
@@ -39,7 +41,12 @@ var archCmd = &cobra.Command{
 			return false
 		}
 
-		dotContent.ToDot("coca_reporter/arch.dot", ".", nodeFilter)
+		graph := dotContent.ToDot(".", nodeFilter)
+		f, _ := os.Create("coca_reporter/arch.dot")
+		w := bufio.NewWriter(f)
+		w.WriteString("di" + graph.String())
+		w.Flush()
+
 		cmd_util.ConvertToSvg("arch")
 	},
 }

core/domain/arch/tequila/incl_viz.go

 package tequila
 
 import (
-	"bufio"
 	"fmt"
 	"github.com/awalterschulze/gographviz"
 	"io/ioutil"
@@ -232,7 +231,7 @@ func ParseInclude(codeDir string) *FullGraph {
 	return fullGraph
 }
 
-func (fullGraph *FullGraph) ToDot(fileName string, split string, filter func(string) bool) {
+func (fullGraph *FullGraph) ToDot(split string, filter func(string) bool) *gographviz.Graph {
 	graph := gographviz.NewGraph()
 	graph.SetName("G")
 
@@ -294,10 +293,7 @@ func (fullGraph *FullGraph) ToDot(fileName string, split string, filter func(str
 		}
 	}
 
-	f, _ := os.Create(fileName)
-	w := bufio.NewWriter(f)
-	w.WriteString("di" + graph.String())
-	w.Flush()
+	return graph
 }
 
 var Foo = func() string {