From 6a5a06a2061121f551523dc2faca9d9cf46989a2 Mon Sep 17 00:00:00 2001
From: Laytan Laats <laytanlaats@hotmail.com>
Date: Mon, 10 Apr 2023 02:17:36 +0200
Subject: [PATCH] fix: add some boundary checks so the formatter panics less
 often

---
 pkg/visitor/formatter/formatter.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/pkg/visitor/formatter/formatter.go b/pkg/visitor/formatter/formatter.go
index fcb6e81..c9e4d60 100644
--- a/pkg/visitor/formatter/formatter.go
+++ b/pkg/visitor/formatter/formatter.go
@@ -22,6 +22,9 @@ type formatter struct {
 	lastSemiColon *token.Token
 }
 
+// NewFormatter returns a new formatter. This formatter is not great, it can not be customized,
+// but most importantly, loses a lot of doc comments and regular comments, so don't use on code
+// that wants to keep all comments in tact.
 func NewFormatter() *formatter {
 	return &formatter{}
 }
@@ -83,6 +86,10 @@ func (f *formatter) newToken(id token.ID, val []byte) *token.Token {
 }
 
 func (f *formatter) formatList(nodes []ast.Vertex, separator byte) []*token.Token {
+	if len(nodes) == 0 {
+		return nil
+	}
+
 	separatorTkns := make([]*token.Token, len(nodes)-1)
 	for i, v := range nodes {
 		v.Accept(f)
@@ -844,7 +851,6 @@ func (f *formatter) StmtNamespace(n *ast.StmtNamespace) {
 	} else {
 		n.SemiColonTkn = f.newSemicolonTkn()
 	}
-
 }
 
 func (f *formatter) StmtNop(n *ast.StmtNop) {
@@ -1189,7 +1195,9 @@ func (f *formatter) ExprArrayItem(n *ast.ExprArrayItem) {
 		f.addFreeFloating(token.T_WHITESPACE, []byte(" "))
 	}
 
-	n.Val.Accept(f)
+	if n.Val != nil {
+		n.Val.Accept(f)
+	}
 }
 
 func (f *formatter) ExprArrowFunction(n *ast.ExprArrowFunction) {