Merge pull request #24 from kefniark/chore/improve-preparser

Improve Preparser
kefniark · Sep 2, 2024 · b6970a5 · b6970a5
2 parents 150ed5e + a95249f
commit b6970a5
Show file tree

Hide file tree

Showing 109 changed files with 235,846 additions and 58 deletions.
diff --git a/cmd/mangosql/actions/codegen/codegen.go b/cmd/mangosql/actions/codegen/codegen.go
@@ -64,6 +64,7 @@ func generate(opts generateOptions) error {
 	// parse schema
 	schema, err := internal.ParseSchema(sql)
 	if err != nil {
+		fmt.Printf("schema parsing error: %+v\n", err)
 		return err
 	}
 

diff --git a/cmd/mangosql/actions/diagram/diagram.go b/cmd/mangosql/actions/diagram/diagram.go
@@ -103,6 +103,7 @@ func diagram(opts diagramOptions) error {
 	// parse schema
 	schema, err := internal.ParseSchema(sql)
 	if err != nil {
+		fmt.Printf("schema parsing error: %+v\n", err)
 		return err
 	}
 
@@ -275,6 +276,11 @@ func renderColumn(t *core.SQLTable, columns map[string]*core.SQLColumn, writer *
 			if err != nil {
 				continue
 			}
+		} else {
+			_, err := writer.WriteString(fmt.Sprintf("	%s: %s\n", c.Name, ""))
+			if err != nil {
+				continue
+			}
 		}
 	}
 }

diff --git a/internal/parser.go b/internal/parser.go
@@ -105,6 +105,10 @@ func ParseSchema(sql string) (*core.SQLSchema, error) {
 	for _, table := range schema.Tables {
 		for _, ref := range table.References {
 			refTable := schema.Tables[ref.Table]
+			if refTable == nil {
+				fmt.Println("Cannot find foreignKey", ref)
+				continue
+			}
 
 			refTable.Referenced = append(refTable.Referenced, &core.SQLTableReference{
 				Name:         ref.Name,