From 0abe99274bef2ef182cd786290e12f3d7915be0d Mon Sep 17 00:00:00 2001
From: z7zmey <z7zmey@gmail.com>
Date: Fri, 6 Apr 2018 00:59:22 +0300
Subject: [PATCH] pretty printer: print file

---
 printer/printer.go      | 21 +++++++++++++++++
 printer/printer_test.go | 52 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 73 insertions(+)

diff --git a/printer/printer.go b/printer/printer.go
index fcd4690..231c855 100644
--- a/printer/printer.go
+++ b/printer/printer.go
@@ -30,6 +30,27 @@ func NewPrinter(w io.Writer, indentStr string) *Printer {
 	}
 }
 
+func (p *Printer) PrintFile(n *stmt.StmtList) {
+	if len(n.Stmts) > 0 {
+		firstStmt := n.Stmts[0]
+		n.Stmts = n.Stmts[1:]
+
+		switch fs := firstStmt.(type) {
+		case *stmt.InlineHtml:
+			io.WriteString(p.w, fs.Value)
+			io.WriteString(p.w, "<?php\n")
+		default:
+			io.WriteString(p.w, "<?php\n")
+			p.printIndent()
+			p.Print(fs)
+			io.WriteString(p.w, "\n")
+		}
+	}
+	p.indentDepth--
+	p.printNodes(n.Stmts)
+	io.WriteString(p.w, "\n")
+}
+
 func (p *Printer) Print(n node.Node) {
 	p.printNode(n)
 }
diff --git a/printer/printer_test.go b/printer/printer_test.go
index 1a0d37b..6dbb6a2 100644
--- a/printer/printer_test.go
+++ b/printer/printer_test.go
@@ -15,6 +15,58 @@ import (
 	"github.com/z7zmey/php-parser/printer"
 )
 
+func TestPrintFile(t *testing.T) {
+	o := bytes.NewBufferString("")
+
+	p := printer.NewPrinter(o, "    ")
+	p.PrintFile(&stmt.StmtList{
+		Stmts: []node.Node{
+			&stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}},
+			&stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "b"}}},
+		},
+	})
+
+	expected := `<?php
+$a;
+$b;
+`
+	actual := o.String()
+
+	if expected != actual {
+		t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
+	}
+}
+
+func TestPrintFileInlineHtml(t *testing.T) {
+	o := bytes.NewBufferString("")
+
+	p := printer.NewPrinter(o, "    ")
+	p.PrintFile(&stmt.StmtList{
+		Stmts: []node.Node{
+			&stmt.InlineHtml{Value: "<div>HTML</div>"},
+			&stmt.Expression{
+				Expr: &scalar.Heredoc{
+					Label: "\"LBL\"",
+					Parts: []node.Node{
+						&scalar.EncapsedStringPart{Value: "hello world\n"},
+					},
+				},
+			},
+		},
+	})
+
+	expected := `<div>HTML</div><?php
+<<<"LBL"
+hello world
+LBL;
+`
+	actual := o.String()
+
+	if expected != actual {
+		t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
+	}
+}
+
 // node
 
 func TestPrintIdentifier(t *testing.T) {