From 3e26ed725fef106f8be2bd0d18b0b568cf527448 Mon Sep 17 00:00:00 2001
From: ericchiang <eric.chiang.m@gmail.com>
Date: Sun, 14 Sep 2014 19:00:31 -0400
Subject: [PATCH] class selector fixed

---
 main.go              |  2 +-
 selector/selector.go | 20 +++++++++++++++-----
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 7a04e41..e777eed 100644
--- a/main.go
+++ b/main.go
@@ -11,7 +11,7 @@ import (
 	"strings"
 )
 
-const VERSION string = "0.1.0"
+const VERSION string = "0.1.1"
 
 var (
 	// Flags
diff --git a/selector/selector.go b/selector/selector.go
index 19359f5..4fda1c4 100644
--- a/selector/selector.go
+++ b/selector/selector.go
@@ -42,7 +42,7 @@ func parseAttrField(command string) (attrKey string, matcher *regexp.Regexp,
 		attrKeyLen := len(attrKey)
 		switch attrKey[attrKeyLen-1] {
 		case '~':
-			matcherString = fmt.Sprintf(`[^\s]%s[$\s]`, attrVal)
+			matcherString = fmt.Sprintf(`\b%s\b`, attrVal)
 		case '$':
 			matcherString = fmt.Sprintf("%s$", attrVal)
 		case '^':
@@ -67,18 +67,27 @@ func (s *Selector) setFieldValue(f selectorField, v string) error {
 	if v == "" {
 		return nil
 	}
-	r, err := regexp.Compile(fmt.Sprintf("^%s$", v))
-	if err != nil {
-		return err
-	}
 	switch f {
 	case ClassField:
+		r, err := regexp.Compile(fmt.Sprintf(`\b%s\b`, v))
+		if err != nil {
+			return err
+		}
 		s.Attrs["class"] = r
 	case IDField:
+		r, err := regexp.Compile(fmt.Sprintf("^%s$", v))
+		if err != nil {
+			return err
+		}
 		s.Attrs["id"] = r
 	case NameField:
+		r, err := regexp.Compile(fmt.Sprintf("^%s$", v))
+		if err != nil {
+			return err
+		}
 		s.Name = r
 	case AttrField:
+		// Attribute fields are a little more complicated
 		keystring, matcher, err := parseAttrField(v)
 		if err != nil {
 			return err
@@ -94,6 +103,7 @@ func NewSelector(s string) (*Selector, error) {
 	selector := &Selector{nil, attrs}
 	nextField := NameField
 	start := 0
+	// Parse the selector character by character
 	for i, c := range s {
 		switch c {
 		case '.':