From c1de9b192253d53e7fb2079b15721d4409b0af92 Mon Sep 17 00:00:00 2001
From: mattn <mattn.jp@gmail.com>
Date: Wed, 17 Sep 2014 02:11:16 +0900
Subject: [PATCH] Use charset.NewReader()

---
 main.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 25d5ec4..5e51a7f 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"code.google.com/p/go.net/html"
+	"code.google.com/p/go.net/html/charset"
 	"fmt"
 	"github.com/ericchiang/pup/funcs"
 	"github.com/ericchiang/pup/selector"
@@ -115,7 +116,12 @@ func ProcessFlags(cmds []string) []string {
 // pup
 func main() {
 	cmds := ProcessFlags(os.Args[1:])
-	root, err := html.Parse(inputStream)
+	cr, err := charset.NewReader(inputStream, "")
+	if err != nil {
+		fmt.Fprintf(os.Stderr, err.Error())
+		os.Exit(2)
+	}
+	root, err := html.Parse(cr)
 	if err != nil {
 		fmt.Fprintf(os.Stderr, err.Error())
 		os.Exit(2)