From c4cd7d24d256e3f633fad02d5ca153ce881de2c5 Mon Sep 17 00:00:00 2001
From: Silvan Jegen <s.jegen@gmail.com>
Date: Tue, 19 Jan 2016 17:18:57 +0100
Subject: Read from stdin instead of a file

---
 htmlfilter.go | 20 +++++++-------------
 1 file changed, 7 insertions(+), 13 deletions(-)

(limited to 'htmlfilter.go')

diff --git a/htmlfilter.go b/htmlfilter.go
index 13b7fa4..0b00e51 100644
--- a/htmlfilter.go
+++ b/htmlfilter.go
@@ -11,21 +11,17 @@ import (
 )
 
 func main() {
-	var unxmlreader io.Reader
-	var n int
+	var (
+		unxmlreader io.Reader
+		n           int
+		err         error
+	)
 
-	f, err := os.Open(os.Args[1])
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error when opening file: %s\n", err)
-	}
-
-	unxmlreader = unxml.NewReaderKeepTags(f, []string{"a"})
-	//unxmlreader = unxml.NewReaderKeepTags(f, make([]string, 0))
+	unxmlreader = unxml.NewReaderKeepTags(os.Stdin, make([]string, 0))
 	lineno := 0
 
-	bbuffer := make([]byte, 100)
+	bbuffer := make([]byte, 4096)
 	for {
-
 		n, err = unxmlreader.Read(bbuffer)
 		if err != nil {
 			if err == io.EOF || n == 0 {
@@ -40,6 +36,4 @@ func main() {
 		fmt.Printf("%s", bbuffer[:n])
 	}
 	fmt.Printf("%s", bbuffer[:n])
-
-	f.Close()
 }
-- 
cgit v1.2.1-18-gbd029