From f42ff6b393b010a697d14a85b80058532245ba8b Mon Sep 17 00:00:00 2001
From: Eric Chiang <eric.chiang.m@gmail.com>
Date: Sun, 1 Feb 2015 13:52:02 -0500
Subject: [PATCH] number displayer added back in and unescape renamed to plain

---
 display.go | 7 +++++++
 parse.go   | 6 ++++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/display.go b/display.go
index 9a1d381..5e8cda6 100644
--- a/display.go
+++ b/display.go
@@ -288,3 +288,10 @@ func (j JSONDisplayer) Display(nodes []*html.Node) {
 	}
 	fmt.Printf("%s", data)
 }
+
+// Print the number of features returned
+type NumDisplayer struct{}
+
+func (d NumDisplayer) Display(nodes []*html.Node) {
+	fmt.Println(len(nodes))
+}
diff --git a/parse.go b/parse.go
index 8432292..b70a115 100644
--- a/parse.go
+++ b/parse.go
@@ -54,7 +54,7 @@ Flags
     -i --indent        number of spaces to use for indent or character
     -n --number        print number of elements selected
     -l --limit         restrict number of levels printed
-    -u --unescape      don't escape html
+    -p --plain         don't escape html
     --charset          specify the charset for pup to use
     --version          display version
 `
@@ -85,7 +85,7 @@ func ProcessFlags(cmds []string) (nonFlagCmds []string, err error) {
 		switch cmd {
 		case "-c", "--color":
 			pupPrintColor = true
-		case "-u", "--unescape":
+		case "-p", "--plain":
 			pupEscapeHTML = false
 		case "-f", "--file":
 			filename := cmds[i+1]
@@ -117,6 +117,8 @@ func ProcessFlags(cmds []string) (nonFlagCmds []string, err error) {
 		case "--version":
 			fmt.Println(VERSION)
 			os.Exit(0)
+		case "-n", "--number":
+			pupDisplayer = NumDisplayer{}
 		default:
 			if cmd[0] == '-' {
 				return []string{}, fmt.Errorf("Unrecognized flag '%s'", cmd)