grindlemire/go-tui code browser

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

// Package formatter provides code formatting for .gsx files.
package formatter

import (
	"github.com/grindlemire/go-tui/internal/tuigen"
)

// Formatter formats .gsx source code.
type Formatter struct {
	// IndentString is the string used for indentation (default: tab).
	IndentString string
	// FixImports enables auto-import resolution (default: true).
	FixImports bool
}

// New creates a new Formatter with default settings.
func New() *Formatter {
	return &Formatter{
		IndentString: "\t",
		FixImports:   true,
	}
}

// Format parses and reformats the given .gsx source code.
// Returns the formatted code and any error encountered during parsing.
func (f *Formatter) Format(filename, source string) (string, error) {
	// Parse the source
	lexer := tuigen.NewLexer(filename, source)
	parser := tuigen.NewParser(lexer)

	file, err := parser.ParseFile()
	if err != nil {
		return source, err
	}

	// Fix imports using goimports
	if f.FixImports {
		err = fixImports(file, filename)
		if err != nil {
			return source, err
		}
	}

	// Generate formatted output
	printer := newPrinter(f.IndentString)
	return printer.PrintFile(file), nil
}

// FormatResult contains the result of formatting a file.
type FormatResult struct {
	// Content is the formatted content.
	Content string
	// Changed indicates if the content was different from the original.
	Changed bool
}

// FormatWithResult formats the source and indicates if it changed.
func (f *Formatter) FormatWithResult(filename, source string) (FormatResult, error) {
	formatted, err := f.Format(filename, source)
	if err != nil {
		return FormatResult{}, err
	}

	return FormatResult{
		Content: formatted,
		Changed: formatted != source,
	}, nil
}