-
-
Notifications
You must be signed in to change notification settings - Fork 182
/
controller.go
50 lines (40 loc) · 1.13 KB
/
controller.go
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
package controller
import (
// Embed templates
_ "embed"
"fmt"
"github.com/livebud/bud/internal/gotemplate"
"github.com/livebud/bud/package/di"
"github.com/livebud/bud/package/genfs"
"github.com/livebud/bud/package/gomod"
"github.com/livebud/bud/package/parser"
)
//go:embed controller.gotext
var template string
var generator = gotemplate.MustParse("framework/controller/controller.gotext", template)
// Generate the controller template from state
func Generate(state *State) ([]byte, error) {
return generator.Generate(state)
}
// New controller generator
func New(injector *di.Injector, module *gomod.Module, parser *parser.Parser) *Generator {
return &Generator{injector, module, parser}
}
// Generator for controllers
type Generator struct {
injector *di.Injector
module *gomod.Module
parser *parser.Parser
}
func (g *Generator) GenerateFile(fsys genfs.FS, file *genfs.File) error {
state, err := Load(fsys, g.injector, g.module, g.parser)
if err != nil {
return fmt.Errorf("framework/controller: unable to load. %w", err)
}
code, err := Generate(state)
if err != nil {
return err
}
file.Data = code
return nil
}