Write to output file given via flag

Author: samuell

README.md

@@ -41,11 +41,11 @@ Usage
 -----
 
 Call the rdf2smw binary, specifying a file with triples in n-triples or turtle
-format, with the `--infile` flag. Output is written to stdout, so you have to
-redirect it to a file of a chosen name:
+format, with the `--in` flag, and an output file in XML format with the
+`--out` flag, like so:
 
 ```bash
-./rdf2smw --infile triples.nt > semantic_mediawiki_pages.xml
+./rdf2smw --in triples.nt --out semantic_mediawiki_pages.xml
 ```
 
 The resulting XML file, can then be imported into MediaWiki / Semantic

main.go

@@ -23,10 +23,20 @@ const (
 func main() {
 	flowbase.InitLogInfo()
 
-	inFileName := flag.String("infile", "", "The input file name")
+	inFileName := flag.String("in", "", "The input file name")
+	outFileName := flag.String("out", "", "The output file name")
 	flag.Parse()
+
+	doExit := false
 	if *inFileName == "" {
-		fmt.Println("No filename specified to --infile")
+		fmt.Println("No filename specified to --in")
+		doExit = true
+	} else if *outFileName == "" {
+		fmt.Println("No filename specified to --out")
+		doExit = true
+	}
+
+	if doExit {
 		os.Exit(1)
 	}
 
@@ -72,8 +82,10 @@ func main() {
 	xmlCreator := NewMWXMLCreator(useTemplates)
 	pipeRunner.AddProcess(xmlCreator)
 
-	printer := NewStringPrinter()
-	pipeRunner.AddProcess(printer)
+	//printer := NewStringPrinter()
+	//pipeRunner.AddProcess(printer)
+	strFileWriter := NewStringFileWriter(*outFileName)
+	pipeRunner.AddProcess(strFileWriter)
 
 	// ------------------------------------------
 	// Connect network
@@ -760,6 +772,33 @@ func (p *StringPrinter) Run() {
 	}
 }
 
+// --------------------------------------------------------------------------------
+// String File Writer
+// --------------------------------------------------------------------------------
+
+type StringFileWriter struct {
+	In       chan string
+	fileName string
+}
+
+func NewStringFileWriter(fileName string) *StringFileWriter {
+	return &StringFileWriter{
+		In:       make(chan string, BUFSIZE),
+		fileName: fileName,
+	}
+}
+
+func (p *StringFileWriter) Run() {
+	fh, err := os.Create(p.fileName)
+	if err != nil {
+		panic("Could not create output file: " + err.Error())
+	}
+	defer fh.Close()
+	for s := range p.In {
+		fh.WriteString(s)
+	}
+}
+
 // --------------------------------------------------------------------------------
 // IP: RDFTriple
 // --------------------------------------------------------------------------------