lourenco/mgmt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lang: Add common type to global variables

Browse Source 
We use the list of strings so often, we might as well give it a global
variable.
This commit is contained in:
James Shubin
2024-04-16 14:31:03 -04:00
parent 6782d65577
commit dc45c90ccd
7 changed files with 19 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
lang/ast/structs.go
View File

@@ -589,13 +589,13 @@ func (obj *StmtRes) Unify() ([]interfaces.Invariant, error) {
invarListStr := &interfaces.EqualsInvariant{
Expr: obj.Name,
Type: types.NewType("[]str"),
Type: types.TypeListStr,
}
// Optimization: If we know it's a []str, no need for exclusives!
if expr, ok := obj.Name.(*ExprList); ok {
typ, err := expr.Type()
if err == nil && typ.Cmp(types.NewType("[]str")) == nil {
if err == nil && typ.Cmp(types.TypeListStr) == nil {
invariants = append(invariants, invarListStr)
return invariants, nil
}
@@ -707,7 +707,7 @@ func (obj *StmtRes) Output(table map[interfaces.Func]types.Value) (*interfaces.O
name := nameValue.Str() // must not panic
names = append(names, name)
case types.NewType("[]str").Cmp(nameValue.Type()) == nil:
case types.TypeListStr.Cmp(nameValue.Type()) == nil:
for _, x := range nameValue.List() { // must not panic
name := x.Str() // must not panic
names = append(names, name)
@@ -880,7 +880,7 @@ func (obj *StmtRes) edges(table map[interfaces.Func]types.Value, resName string)
name := nameValue.Str() // must not panic
names = append(names, name)
case types.NewType("[]str").Cmp(nameValue.Type()) == nil:
case types.TypeListStr.Cmp(nameValue.Type()) == nil:
for _, x := range nameValue.List() { // must not panic
name := x.Str() // must not panic
names = append(names, name)
@@ -1953,7 +1953,7 @@ func (obj *StmtResMeta) Unify(kind string) ([]interfaces.Invariant, error) {
invar = static(types.TypeBool)
case "sema":
invar = static(types.NewType("[]str"))
invar = static(types.TypeListStr)
case "rewatch":
invar = static(types.TypeBool)
@@ -2235,14 +2235,14 @@ func (obj *StmtEdge) Unify() ([]interfaces.Invariant, error) {
// display something nicer
if v.Type().Kind == types.KindStr {
p1 = engine.Repr(k1, v.Str())
} else if v.Type().Cmp(types.NewType("[]str")) == nil {
} else if v.Type().Cmp(types.TypeListStr) == nil {
p1 = engine.Repr(k1, v.String())
}
}
if v, err := obj.EdgeHalfList[1].Name.Value(); err == nil {
if v.Type().Kind == types.KindStr {
p2 = engine.Repr(k2, v.Str())
} else if v.Type().Cmp(types.NewType("[]str")) == nil {
} else if v.Type().Cmp(types.TypeListStr) == nil {
p2 = engine.Repr(k2, v.String())
}
}
@@ -2319,7 +2319,7 @@ func (obj *StmtEdge) Output(table map[interfaces.Func]types.Value) (*interfaces.
name := nameValue1.Str() // must not panic
names1 = append(names1, name)
case types.NewType("[]str").Cmp(nameValue1.Type()) == nil:
case types.TypeListStr.Cmp(nameValue1.Type()) == nil:
for _, x := range nameValue1.List() { // must not panic
name := x.Str() // must not panic
names1 = append(names1, name)
@@ -2344,7 +2344,7 @@ func (obj *StmtEdge) Output(table map[interfaces.Func]types.Value) (*interfaces.
name := nameValue2.Str() // must not panic
names2 = append(names2, name)
case types.NewType("[]str").Cmp(nameValue2.Type()) == nil:
case types.TypeListStr.Cmp(nameValue2.Type()) == nil:
for _, x := range nameValue2.List() { // must not panic
name := x.Str() // must not panic
names2 = append(names2, name)
@@ -2501,13 +2501,13 @@ func (obj *StmtEdgeHalf) Unify() ([]interfaces.Invariant, error) {
invarListStr := &interfaces.EqualsInvariant{
Expr: obj.Name,
Type: types.NewType("[]str"),
Type: types.TypeListStr,
}
// Optimization: If we know it's a []str, no need for exclusives!
if expr, ok := obj.Name.(*ExprList); ok {
typ, err := expr.Type()
if err == nil && typ.Cmp(types.NewType("[]str")) == nil {
if err == nil && typ.Cmp(types.TypeListStr) == nil {
invariants = append(invariants, invarListStr)
return invariants, nil
}

2
lang/core/os/args_func.go
View File

@@ -54,6 +54,6 @@ func Args([]types.Value) (types.Value, error) {
}
return &types.ListValue{
V: values,
T: types.NewType("[]str"),
T: types.TypeListStr,
}, nil
}

2
lang/core/strings/split_func.go
View File

@@ -50,7 +50,7 @@ func Split(input []types.Value) (types.Value, error) {
segments := strings.Split(str, sep)
listVal := types.NewList(types.NewType("[]str"))
listVal := types.NewList(types.TypeListStr)
for _, segment := range segments {
listVal.Add(&types.StrValue{

6
lang/core/world/schedule_func.go
View File

@@ -149,7 +149,7 @@ func (obj *ScheduleFunc) Unify(expr interfaces.Expr) ([]interfaces.Invariant, er
// return type of []string
invar = &interfaces.EqualsInvariant{
Expr: dummyOut,
Type: types.NewType("[]str"),
Type: types.TypeListStr,
}
invariants = append(invariants, invar)
@@ -345,7 +345,7 @@ func (obj *ScheduleFunc) Polymorphisms(partialType *types.Type, partialValues []
var typ *types.Type
if tOut := partialType.Out; tOut != nil {
if err := tOut.Cmp(types.NewType("[]str")); err != nil {
if err := tOut.Cmp(types.TypeListStr); err != nil {
return nil, errwrap.Wrapf(err, "return type must be a list of strings")
}
}
@@ -428,7 +428,7 @@ func (obj *ScheduleFunc) Build(typ *types.Type) (*types.Type, error) {
return nil, fmt.Errorf("invalid input type")
}
if err := typ.Out.Cmp(types.NewType("[]str")); err != nil {
if err := typ.Out.Cmp(types.TypeListStr); err != nil {
return nil, errwrap.Wrapf(err, "return type must be a list of strings")
}

2
lang/funcs/funcgen/config.go
View File

@@ -77,7 +77,7 @@ func (obj *arg) ToMcl() (string, error) {
case "float64":
return fmt.Sprintf("%s%s", prefix, types.TypeFloat.String()), nil
case "[]string":
return fmt.Sprintf("%s%s", prefix, types.NewType("[]str").String()), nil
return fmt.Sprintf("%s%s", prefix, types.TypeListStr.String()), nil
default:
return "", fmt.Errorf("cannot convert %v to mcl", obj.Type)
}

1
lang/types/type.go
View File

@@ -50,6 +50,7 @@ var (
TypeStr = NewType("str")
TypeInt = NewType("int")
TypeFloat = NewType("float")
TypeListStr = NewType("[]str")
TypeVariant = NewType("variant")
)

2
lang/unification/solvers/unification_test.go
View File

@@ -125,7 +125,7 @@ func TestUnification1(t *testing.T) {
v1: types.TypeStr,
v2: types.TypeStr,
v3: types.TypeStr,
expr: types.NewType("[]str"),
expr: types.TypeListStr,
},
})
}