-
Notifications
You must be signed in to change notification settings - Fork 233
/
userfacing.go
93 lines (80 loc) · 1.85 KB
/
userfacing.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package compiler
import (
"bytes"
"os"
"path/filepath"
"encr.dev/parser"
"encr.dev/parser/est"
"encr.dev/pkg/errinsrc/srcerrors"
)
// GenUserFacing generates user-facing application code.
func GenUserFacing(appRoot string, res *parser.Result) error {
b := &builder{
cfg: &Config{
Parse: res,
},
appRoot: appRoot,
res: res,
}
return b.GenUserFacing()
}
func (b *builder) GenUserFacing() (err error) {
defer func() {
if e := recover(); e != nil {
if b, ok := e.(bailout); ok {
err = b.err
} else {
err = srcerrors.UnhandledPanic(e)
}
}
}()
for _, fn := range []func() error{
b.parseApp,
b.genUserFacing,
} {
if err := fn(); err != nil {
return err
}
}
return nil
}
func (b *builder) genUserFacing() error {
for _, svc := range b.res.App.Services {
if err := b.generateUserFacingGoCode(svc); err != nil {
return err
}
if err := b.generateUserFacingCueCode(svc); err != nil {
return err
}
}
return nil
}
func (b *builder) generateUserFacingGoCode(svc *est.Service) (err error) {
dst := filepath.Join(b.appRoot, svc.Root.RelPath, "encore.gen.go")
f := b.codegen.UserFacing(svc, false)
if f == nil {
// No need for any generated code. Try to remove the existing file
// if it's there as it's no longer needed.
_ = os.Remove(dst)
return nil
}
var buf bytes.Buffer
if err := f.Render(&buf); err != nil {
return err
}
return os.WriteFile(dst, buf.Bytes(), 0644)
}
func (b *builder) generateUserFacingCueCode(svc *est.Service) (err error) {
dst := filepath.Join(b.appRoot, svc.Root.RelPath, "encore.gen.cue")
f, err := b.cuegen.UserFacing(svc)
if err != nil {
return err
}
if f == nil || len(f) == 0 {
// No need for any generated code. Try to remove the existing file
// if it's there as it's no longer needed.
_ = os.Remove(dst)
return nil
}
return os.WriteFile(dst, f, 0644)
}