Format formats an AST into properly indented SQL
(astObj *ast.AST)
| 58 | |
| 59 | // Format formats an AST into properly indented SQL |
| 60 | func (f *SQLFormatter) Format(astObj *ast.AST) (string, error) { |
| 61 | f.builder.Reset() |
| 62 | f.newlineLevel = 0 |
| 63 | |
| 64 | for i, stmt := range astObj.Statements { |
| 65 | if i > 0 { |
| 66 | if f.compact { |
| 67 | f.builder.WriteString("; ") |
| 68 | } else { |
| 69 | f.builder.WriteString(";\n\n") |
| 70 | } |
| 71 | } |
| 72 | |
| 73 | if err := f.formatStatement(stmt); err != nil { |
| 74 | return "", fmt.Errorf("failed to format statement: %w", err) |
| 75 | } |
| 76 | } |
| 77 | |
| 78 | return f.builder.String(), nil |
| 79 | } |
| 80 | |
| 81 | // formatStatement formats individual SQL statements |
| 82 | func (f *SQLFormatter) formatStatement(stmt ast.Statement) error { |