Created a comment function.

element.go

@@ -53,7 +53,7 @@ func (e *Element) parse() error {
 		return errors.New(fmt.Sprintf("The element has no tokens. (line no: %d)", e.LineNo))
 	}
 	switch {
-	case e.Type == TypeContent || e.Type == TypeBlock || e.Type == TypeExpression || e.Type == TypeLiteral:
+	case e.Type == TypeContent || e.Type == TypeBlock || e.Type == TypeExpression || e.Type == TypeLiteral || e.comment():
 	default:
 		for i, token := range e.Tokens {
 			switch {
@@ -190,6 +190,7 @@ func (e *Element) AppendChild(child *Element) {
 // Html writes the element's html to the buffer.
 func (e *Element) Html(bf *bytes.Buffer) error {
 	switch {
+	case e.comment():
 	case e.Type == TypeContent || e.Type == TypeExpression:
 		e.writeText(bf)
 		for _, child := range e.Children {
@@ -386,6 +387,11 @@ func (e *Element) writeLiteralValue(bf *bytes.Buffer) {
 	bf.WriteString(e.literalValue())
 }
 
+// comment returns if the string is a comment or not.
+func (e *Element) comment() bool {
+	return strings.HasPrefix(e.Text, "//")
+}
+
 // NewElement generates a new element and returns it.
 func NewElement(text string, lineNo int, indent int, parent *Element, tpl *Template, block *Block) (*Element, error) {
 	text = strings.TrimSpace(text)

element_test.go

@@ -683,6 +683,20 @@ func TestElementWriteLiteralValue(t *testing.T) {
 	}
 }
 
+func TestElementComment(t *testing.T) {
+	// When the element is a comment.
+	e := &Element{Text: "//aaa"}
+	if e.comment() != true {
+		t.Errorf("Return value should be true.")
+	}
+
+	// When the element is not a comment.
+	e = &Element{Text: "aaa"}
+	if e.comment() != false {
+		t.Errorf("Return value should be true.")
+	}
+}
+
 func TestNewElement(t *testing.T) {
 	// When an error occurs while parsing.
 	_, err := NewElement("div#id1#id2", 1, 0, nil, nil, nil)

generator.go

@@ -143,7 +143,7 @@ indentLoop:
 	return i
 }
 
-// empty returns if the string is empty.
+// empty returns if the string is empty or not.
 func empty(s string) bool {
 	return strings.TrimSpace(s) == ""
 }