|
| 1 | +// SPDX-License-Identifier: Apache-2.0 |
| 2 | + |
| 3 | +package linter |
| 4 | + |
| 5 | +import ( |
| 6 | + "strings" |
| 7 | + |
| 8 | + "go.starlark.net/starlark" |
| 9 | + "go.starlark.net/starlarkstruct" |
| 10 | + |
| 11 | + "github.com/mendixlabs/mxcli/mdl/exprcheck" |
| 12 | +) |
| 13 | + |
| 14 | +// stripXPathBrackets removes the outer [ and ] from a Mendix XPath constraint string. |
| 15 | +// Returns the inner expression ready for parsing. |
| 16 | +// Only strips when the opening [ at position 0 is matched by the final ] (i.e. they |
| 17 | +// form a single outer pair). Chained predicates like [a = 1][b = 2] are returned as-is. |
| 18 | +func stripXPathBrackets(s string) string { |
| 19 | + s = strings.TrimSpace(s) |
| 20 | + if !strings.HasPrefix(s, "[") || !strings.HasSuffix(s, "]") { |
| 21 | + return s |
| 22 | + } |
| 23 | + // Walk forward to find where the first '[' closes. |
| 24 | + depth := 0 |
| 25 | + for i, ch := range s { |
| 26 | + switch ch { |
| 27 | + case '[': |
| 28 | + depth++ |
| 29 | + case ']': |
| 30 | + depth-- |
| 31 | + if depth == 0 { |
| 32 | + if i == len(s)-1 { |
| 33 | + return s[1 : len(s)-1] |
| 34 | + } |
| 35 | + // First '[' closes before the end — chained predicates; don't strip. |
| 36 | + return s |
| 37 | + } |
| 38 | + } |
| 39 | + } |
| 40 | + return s |
| 41 | +} |
| 42 | + |
| 43 | +// robustExprToStarlark converts a RobustExpr AST node to a Starlark struct tree. |
| 44 | +// Each node is a struct with a "kind" field and type-specific fields. |
| 45 | +// Returns a struct with kind="null" for a nil node. |
| 46 | +func robustExprToStarlark(expr exprcheck.RobustExpr) starlark.Value { |
| 47 | + if expr == nil { |
| 48 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 49 | + "kind": starlark.String("null"), |
| 50 | + }) |
| 51 | + } |
| 52 | + |
| 53 | + switch n := expr.(type) { |
| 54 | + case *exprcheck.BinExpr: |
| 55 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 56 | + "kind": starlark.String("bin"), |
| 57 | + "op": starlark.String(n.Op), |
| 58 | + "left": robustExprToStarlark(n.L), |
| 59 | + "right": robustExprToStarlark(n.R), |
| 60 | + }) |
| 61 | + case *exprcheck.UnaryExpr: |
| 62 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 63 | + "kind": starlark.String("unary"), |
| 64 | + "op": starlark.String(n.Op), |
| 65 | + "operand": robustExprToStarlark(n.Operand), |
| 66 | + }) |
| 67 | + case *exprcheck.CallExpr: |
| 68 | + args := make([]starlark.Value, len(n.Args)) |
| 69 | + for i, a := range n.Args { |
| 70 | + args[i] = robustExprToStarlark(a) |
| 71 | + } |
| 72 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 73 | + "kind": starlark.String("call"), |
| 74 | + "name": starlark.String(n.Name), |
| 75 | + "args": starlark.NewList(args), |
| 76 | + }) |
| 77 | + case *exprcheck.StringLit: |
| 78 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 79 | + "kind": starlark.String("string"), |
| 80 | + "value": starlark.String(n.Value), |
| 81 | + }) |
| 82 | + case *exprcheck.NumberLit: |
| 83 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 84 | + "kind": starlark.String("number"), |
| 85 | + "value": starlark.String(n.Value), |
| 86 | + }) |
| 87 | + case *exprcheck.BoolLit: |
| 88 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 89 | + "kind": starlark.String("bool"), |
| 90 | + "value": starlark.Bool(n.Value), |
| 91 | + }) |
| 92 | + case *exprcheck.EmptyExpr: |
| 93 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 94 | + "kind": starlark.String("empty"), |
| 95 | + }) |
| 96 | + case *exprcheck.VariableExpr: |
| 97 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 98 | + "kind": starlark.String("variable"), |
| 99 | + "name": starlark.String(n.Name), |
| 100 | + }) |
| 101 | + case *exprcheck.AttributePathExpr: |
| 102 | + pathParts := make([]starlark.Value, len(n.Path)) |
| 103 | + for i, p := range n.Path { |
| 104 | + pathParts[i] = starlark.String(p) |
| 105 | + } |
| 106 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 107 | + "kind": starlark.String("attr_path"), |
| 108 | + "variable": starlark.String(n.Variable), |
| 109 | + "path": starlark.NewList(pathParts), |
| 110 | + }) |
| 111 | + case *exprcheck.QNameExpr: |
| 112 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 113 | + "kind": starlark.String("qname"), |
| 114 | + "module": starlark.String(n.Module), |
| 115 | + "name": starlark.String(n.Name), |
| 116 | + "sub": starlark.String(n.Sub), |
| 117 | + }) |
| 118 | + case *exprcheck.ParenExpr: |
| 119 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 120 | + "kind": starlark.String("paren"), |
| 121 | + "inner": robustExprToStarlark(n.Inner), |
| 122 | + }) |
| 123 | + case *exprcheck.IfThenElseExpr: |
| 124 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 125 | + "kind": starlark.String("if"), |
| 126 | + "cond": robustExprToStarlark(n.Cond), |
| 127 | + "then": robustExprToStarlark(n.Then), |
| 128 | + "else_": robustExprToStarlark(n.Else), |
| 129 | + }) |
| 130 | + case *exprcheck.ConstantRef: |
| 131 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 132 | + "kind": starlark.String("constant"), |
| 133 | + "qname": starlark.String(n.QName), |
| 134 | + }) |
| 135 | + case *exprcheck.TokenExpr: |
| 136 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 137 | + "kind": starlark.String("token"), |
| 138 | + "token": starlark.String(n.Token), |
| 139 | + "arg": starlark.String(n.Arg), |
| 140 | + }) |
| 141 | + case *exprcheck.RecoveredExpr: |
| 142 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 143 | + "kind": starlark.String("recovered"), |
| 144 | + "source": starlark.String(n.SourceFragment), |
| 145 | + "reason": starlark.String(n.Reason), |
| 146 | + }) |
| 147 | + default: |
| 148 | + return starlarkstruct.FromStringDict(starlark.String("expr"), starlark.StringDict{ |
| 149 | + "kind": starlark.String("unknown"), |
| 150 | + }) |
| 151 | + } |
| 152 | +} |
| 153 | + |
| 154 | +// xpathExpressionEntryToStarlark converts an XPathExpressionEntry to a Starlark struct. |
| 155 | +func xpathExpressionEntryToStarlark(e XPathExpressionEntry) starlark.Value { |
| 156 | + return starlarkstruct.FromStringDict(starlark.String("xpath_expression"), starlark.StringDict{ |
| 157 | + "id": starlark.String(e.ID), |
| 158 | + "document_type": starlark.String(e.DocumentType), |
| 159 | + "document_id": starlark.String(e.DocumentID), |
| 160 | + "document_qualified_name": starlark.String(e.DocumentQualifiedName), |
| 161 | + "component_type": starlark.String(e.ComponentType), |
| 162 | + "component_id": starlark.String(e.ComponentID), |
| 163 | + "component_name": starlark.String(e.ComponentName), |
| 164 | + "xpath_expression": starlark.String(e.XPathExpression), |
| 165 | + "target_entity": starlark.String(e.TargetEntity), |
| 166 | + "referenced_entities": starlark.String(e.ReferencedEntities), |
| 167 | + "is_parameterized": starlark.Bool(e.IsParameterized), |
| 168 | + "usage_type": starlark.String(e.UsageType), |
| 169 | + "module_name": starlark.String(e.ModuleName), |
| 170 | + }) |
| 171 | +} |
0 commit comments