this repo has no description
internal/buildattr: factor out from cue/load
This factors out the build attribute evaluation logic so that it
can be used from other places.
This does not change any behavior.
For #3180.
Signed-off-by: Roger Peppe <rogpeppe@gmail.com>
Change-Id: Id2a72d69695cb09118ead5c42fdb4c22feb66c76
Reviewed-on: https://review.gerrithub.io/c/cue-lang/cue/+/1197159
Unity-Result: CUE porcuepine <cue.porcuepine@gmail.com>
Reviewed-by: Daniel Martí <mvdan@mvdan.cc>
TryBot-Result: CUEcueckoo <cueckoo@cuelang.org>
+143
-103
+1
-1
cue/load/loader_common.go
+1
-1
cue/load/loader_common.go
+2
-2
cue/load/search.go
+2
-2
cue/load/search.go
···
406
406
_err = err
407
407
return false
408
408
}
409
-
if err := shouldBuildFile(f.Syntax, tg); err != nil {
409
+
if err := shouldBuildFile(f.Syntax, tg.tagIsSet); err != nil {
410
410
// Later build logic should pick up and report the same error.
411
411
return true
412
412
}
···
522
522
if err != nil {
523
523
return false
524
524
}
525
-
if err := shouldBuildFile(f.Syntax, tg); err != nil {
525
+
if err := shouldBuildFile(f.Syntax, tg.tagIsSet); err != nil {
526
526
// Later build logic should pick up and report the same error.
527
527
return true
528
528
}
+102
internal/buildattr/buildattr.go
+102
internal/buildattr/buildattr.go
···
1
+
// Package buildattr implements support for interpreting the @if
2
+
// build attributes in CUE files.
3
+
package buildattr
4
+
5
+
import (
6
+
"cuelang.org/go/cue/ast"
7
+
"cuelang.org/go/cue/errors"
8
+
"cuelang.org/go/cue/parser"
9
+
"cuelang.org/go/cue/token"
10
+
)
11
+
12
+
// ShouldBuildFile reports whether a File should be included based on its
13
+
// attributes. It uses tagIsSet to determine whether a given attribute
14
+
// key should be treated as set.
15
+
//
16
+
// It also returns the build attribute if one was found.
17
+
func ShouldBuildFile(f *ast.File, tagIsSet func(key string) bool) (bool, *ast.Attribute, errors.Error) {
18
+
a, err := getBuildAttr(f)
19
+
if err != nil {
20
+
return false, nil, err
21
+
}
22
+
if a == nil {
23
+
return true, nil, nil
24
+
}
25
+
26
+
_, body := a.Split()
27
+
28
+
expr, parseErr := parser.ParseExpr("", body)
29
+
if parseErr != nil {
30
+
return false, a, errors.Promote(parseErr, "")
31
+
}
32
+
33
+
include, err := shouldInclude(expr, tagIsSet)
34
+
if err != nil {
35
+
return false, a, err
36
+
}
37
+
return include, a, nil
38
+
}
39
+
40
+
func getBuildAttr(f *ast.File) (*ast.Attribute, errors.Error) {
41
+
var a *ast.Attribute
42
+
for _, d := range f.Decls {
43
+
switch x := d.(type) {
44
+
case *ast.Attribute:
45
+
key, _ := x.Split()
46
+
if key != "if" {
47
+
continue
48
+
}
49
+
if a != nil {
50
+
err := errors.Newf(d.Pos(), "multiple @if attributes")
51
+
err = errors.Append(err,
52
+
errors.Newf(a.Pos(), "previous declaration here"))
53
+
return nil, err
54
+
}
55
+
a = x
56
+
57
+
case *ast.Package:
58
+
break
59
+
}
60
+
}
61
+
return a, nil
62
+
}
63
+
64
+
func shouldInclude(expr ast.Expr, tagIsSet func(key string) bool) (bool, errors.Error) {
65
+
switch x := expr.(type) {
66
+
case *ast.Ident:
67
+
return tagIsSet(x.Name), nil
68
+
69
+
case *ast.BinaryExpr:
70
+
switch x.Op {
71
+
case token.LAND, token.LOR:
72
+
a, err := shouldInclude(x.X, tagIsSet)
73
+
if err != nil {
74
+
return false, err
75
+
}
76
+
b, err := shouldInclude(x.Y, tagIsSet)
77
+
if err != nil {
78
+
return false, err
79
+
}
80
+
if x.Op == token.LAND {
81
+
return a && b, nil
82
+
}
83
+
return a || b, nil
84
+
85
+
default:
86
+
return false, errors.Newf(token.NoPos, "invalid operator %v", x.Op)
87
+
}
88
+
89
+
case *ast.UnaryExpr:
90
+
if x.Op != token.NOT {
91
+
return false, errors.Newf(token.NoPos, "invalid operator %v", x.Op)
92
+
}
93
+
v, err := shouldInclude(x.X, tagIsSet)
94
+
if err != nil {
95
+
return false, err
96
+
}
97
+
return !v, nil
98
+
99
+
default:
100
+
return false, errors.Newf(token.NoPos, "invalid type %T in build attribute", expr)
101
+
}
102
+
}