From 86863fa36908c5b24499e16c810f4a250c97bffe Mon Sep 17 00:00:00 2001
From: Li2CO3 <liyun.339@bytedance.com>
Date: Tue, 4 Jun 2024 14:18:48 +0800
Subject: [PATCH] fix: fix root scope nil pointer (#27)

---
 validator/generator.go | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/validator/generator.go b/validator/generator.go
index 2afcbfe..ef99bf8 100644
--- a/validator/generator.go
+++ b/validator/generator.go
@@ -178,12 +178,15 @@ func (g *generator) renderHeader(ast *tp.Thrift) (string, error) {
 	for _, impt := range g.enumImport {
 		if _, exist := enumUnique[impt]; !exist {
 			var importAlias string
-			for _, inc := range g.utils.RootScope().Includes() {
-				if inc.ImportPath == impt {
-					if inc.PackageName == filepath.Base(inc.ImportPath) {
-						importAlias = inc.PackageName + " "
+			rootScope := g.utils.RootScope()
+			if rootScope != nil {
+				for _, inc := range rootScope.Includes() {
+					if inc.ImportPath == impt {
+						if inc.PackageName == filepath.Base(inc.ImportPath) {
+							importAlias = inc.PackageName + " "
+						}
+						break
 					}
-					break
 				}
 			}
 			importStr = importStr + importAlias + "\"" + impt + "\"\n"