adding key (#33)

h.go

@@ -1,12 +1,17 @@
 package vn
 
-func H(tagName string, attrs *Attrs, children interface{}) *Vnode {
+func H(tagName string, attrs *Attrs, children interface{}, params ...string) *Vnode {
 	sanitizedAttrs := Sanitize(attrs)
+	var key *string
+
+	if len(params) == 1 {
+		key = &params[0]
+	}
 
 	switch (children).(type) {
 	case string:
-		return NewVNode(tagName, sanitizedAttrs, nil, &TextNode{(children).(string), nil}, nil)
+		return NewVNode(tagName, sanitizedAttrs, nil, &TextNode{(children).(string), nil}, nil, key)
 	default:
-		return NewVNode(tagName, sanitizedAttrs, children.(Children), nil, nil)
+		return NewVNode(tagName, sanitizedAttrs, children.(Children), nil, nil, key)
 	}
 }

node.go

@@ -6,6 +6,7 @@ type Node interface {
 	GetElement() *vnd.DomElement
 	GetTagName() string
 	GetAttrs() *Attrs
+	GetKey() *string
 	HasElement() bool
 	SetElement(vnd.DomElement)
 	HashCode() string

vnode.go

@@ -15,13 +15,22 @@ type Vnode struct {
 	Children Children
 	Text     *TextNode
 	Element  *vnd.DomElement
+	key      *string
 }
 
-func NewVNode(tagName string, attrs *Attrs, children Children, text *TextNode, element *vnd.DomElement) *Vnode {
-	return &Vnode{tagName, attrs, children, text, element}
+func NewVNode(tagName string, attrs *Attrs, children Children, text *TextNode, element *vnd.DomElement, key *string) *Vnode {
+	return &Vnode{tagName, attrs, children, text, element, key}
 }
 
 func (vnode *Vnode) IsSame(other Node) bool {
+	currKey := vnode.GetKey()
+	otherKey := other.GetKey()
+
+	if currKey != nil && otherKey != nil {
+		fmt.Println("They are the same => ", *currKey)
+		return *currKey == *otherKey
+	}
+
 	if vnh.IsNil(vnode.Text) {
 		if vnh.IsNil(other.GetText()) {
 			return vnode.HashCode() == other.HashCode()
@@ -52,6 +61,10 @@ func (vnode *Vnode) ChildAt(index int) Node {
 	return nil
 }
 
+func (vnode *Vnode) GetKey() *string {
+	return vnode.key
+}
+
 func (vnode *Vnode) GetText() *TextNode {
 	return vnode.Text
 }

vnode_test.go

@@ -44,13 +44,21 @@ func TestVnode_IsSame_WithChildren(t *testing.T) {
 		H("li", nil, "Second item"),
 	})
 
-	c := H("ul", &Attrs{Props: &Props{"class": "navbar"}}, Children{
+	assert.Equal(t, true, a.IsSame(b))
+}
+
+func TestVnode_IsSame_WithKey(t *testing.T) {
+	a := H("ul", &Attrs{Props: &Props{"class": "navbar"}}, Children{
 		H("li", nil, "First item"),
-		H("li", nil, "Something else"),
-	})
+		H("li", nil, "Second item"),
+	}, "1")
 
-	assert.Equal(t, true, a.IsSame(b))
-	assert.Equal(t, false, a.IsSame(c))
+	b := H("ul", &Attrs{Props: &Props{"class": "navbar"}}, Children{
+		H("li", nil, "First item"),
+		H("li", nil, "Second item"),
+	}, "2")
+
+	assert.Equal(t, false, a.IsSame(b))
 }
 
 func TestVnode_ChildrenCount(t *testing.T) {