Support multiple root-level nodes

1 files changed, 19 insertions, 10 deletions

diff --git a/parser/parser.go b/parser/parser.go
index d8f7eb6..09a6f49 100644
--- a/parser/parser.go
+++ b/parser/parser.go
@@ -31,18 +31,27 @@ type AstNode struct {
 
 func ParseFile(file *os.File) (AstNode, error) {
 	r := reader{r: bufio.NewReader(file)}
-	return r.parseDocument()
-}
-
-func (reader *reader) parseDocument() (AstNode, error) {
 	document := AstNode{Type: Tagless}
-	if node, err := reader.parseNode(); err != nil {
-		return AstNode{}, err
-	} else {
-		document.Children = append(document.Children, node)
-	}
 
-	return document, nil
+	for {
+		if _, err := r.readNonSpaceRune(); err == io.EOF {
+			return document, nil
+		} else if err != nil {
+			return AstNode{}, err
+		} else if err := r.unreadRune(); err != nil {
+			return AstNode{}, err
+		}
+
+		if err := r.skipSpaces(); err != nil {
+			return AstNode{}, err
+		}
+
+		if node, err := r.parseNode(); err != nil {
+			return AstNode{}, err
+		} else {
+			document.Children = append(document.Children, node)
+		}
+	}
 }
 
 func (reader *reader) parseNode() (AstNode, error) {