PoC: Add syntax for escaped field selectors.

cel/cel_test.go

@@ -2130,6 +2130,52 @@ func TestParserRecursionLimit(t *testing.T) {
 	}
 }
 
+func TestQuotedFields(t *testing.T) {
+	testCases := []struct {
+		expr        string
+		errorSubstr string
+		out         ref.Val
+	}{
+		{
+			expr: "{'$key': 64}.`$key`",
+			out:  types.Int(64),
+		},
+		{
+			expr:        "{'$key': 64}.`$key2`",
+			errorSubstr: "no such key: $key2",
+		},
+		{
+			expr: "has({'$key': 64}.`$key`)",
+			out:  types.True,
+		},
+		{
+			expr: "has({'$key': 64}.`$key2`)",
+			out:  types.False,
+		},
+	}
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.expr, func(t *testing.T) {
+			env := testEnv(t, ParserRecursionLimit(10))
+			out, err := interpret(t, env,
+				tc.expr, map[string]any{})
+
+			if tc.errorSubstr != "" {
+				if err == nil || !strings.Contains(err.Error(), tc.errorSubstr) {
+					t.Fatalf("prg.Eval() wanted error containing '%s' got %v", tc.errorSubstr, err)
+				}
+			}
+
+			if tc.out != nil {
+				if tc.out != out {
+					t.Errorf("prg.Eval() wanted %v got %v", tc.out, out)
+				}
+			}
+		})
+
+	}
+}
+
 func TestDynamicDispatch(t *testing.T) {
 	env := testEnv(t,
 		HomogeneousAggregateLiterals(),

ext/protos_test.go

@@ -42,12 +42,21 @@ func TestProtos(t *testing.T) {
 		{expr: `proto.getExt(ExampleType{}, google.expr.proto2.test.nested_example) == ExampleType{}`},
 		{expr: `proto.getExt(ExampleType{}, google.expr.proto2.test.ExtendedExampleType.extended_examples) == []`},
 		{expr: `proto.getExt(ExampleType{}, google.expr.proto2.test.ExtendedExampleType.enum_ext) == GlobalEnum.GOO`},
+		{expr: "ExampleType{`in`: 64}.`in` == 64"},
+		// TODO(): can't assign extension fields, only read. Unclear if needed.
+		// ExampleType{`google.expr.proto2.test.int32_ext`: 42}.`google.expr.proto2.test.int32_ext` == 42
 		{expr: `proto.getExt(msg, google.expr.proto2.test.int32_ext) == 42`},
+		{expr: "msg.`google.expr.proto2.test.int32_ext` == 42"},
 		{expr: `proto.getExt(msg, google.expr.proto2.test.int32_wrapper_ext) == 21`},
+		{expr: "msg.`google.expr.proto2.test.int32_wrapper_ext` == 21"},
 		{expr: `proto.hasExt(msg, google.expr.proto2.test.nested_example)`},
+		{expr: "has(msg.`google.expr.proto2.test.nested_example`)"},
 		{expr: `proto.getExt(msg, google.expr.proto2.test.nested_example) == ExampleType{name: 'nested'}`},
+		{expr: "msg.`google.expr.proto2.test.nested_example` == ExampleType{name: 'nested'}"},
 		{expr: `proto.getExt(msg, google.expr.proto2.test.ExtendedExampleType.extended_examples) == ['example1', 'example2']`},
+		{expr: "msg.`google.expr.proto2.test.ExtendedExampleType.extended_examples` == ['example1', 'example2']"},
 		{expr: `proto.getExt(msg, google.expr.proto2.test.ExtendedExampleType.enum_ext) == GlobalEnum.GAZ`},
+		{expr: "msg.`google.expr.proto2.test.ExtendedExampleType.enum_ext` == GlobalEnum.GAZ"},
 	}
 
 	env := testProtosEnv(t)
@@ -174,6 +183,15 @@ func TestProtosParseErrors(t *testing.T) {
 		| proto.getExt(ExtendedExampleType{}, has(google.expr.proto2.test.int32_ext))
 		| .......................................^`,
 		},
+		{
+			expr: `ExampleType{}.in`,
+			err: `ERROR: <input>:1:15: Syntax error: no viable alternative at input '.in'
+			| ExampleType{}.in
+			| ..............^
+		   ERROR: <input>:1:17: Syntax error: mismatched input '<EOF>' expecting {'[', '{', '(', '.', '-', '!', 'true', 'false', 'null', NUM_FLOAT, NUM_INT, NUM_UINT, STRING, BYTES, IDENTIFIER}
+			| ExampleType{}.in
+			| ................^`,
+		},
 	}
 	env := testProtosEnv(t)
 	for i, tst := range protosTests {

parser/gen/CEL.g4

@@ -52,13 +52,14 @@ unary
 
 member
     : primary                                                       # PrimaryExpr
-    | member op='.' (opt='?')? id=IDENTIFIER                        # Select
+    | member op='.' (opt='?')? id=escapeIdent                       # Select
     | member op='.' id=IDENTIFIER open='(' args=exprList? ')'       # MemberCall
     | member op='[' (opt='?')? index=expr ']'                       # Index
     ;
 
 primary
-    : leadingDot='.'? id=IDENTIFIER (op='(' args=exprList? ')')?    # IdentOrGlobalCall
+    : leadingDot='.'? id=IDENTIFIER                                # Ident
+    | leadingDot='.'? id=IDENTIFIER (op='(' args=exprList? ')')     # GlobalCall
     | '(' e=expr ')'                                                # Nested
     | op='[' elems=listInit? ','? ']'                               # CreateList
     | op='{' entries=mapInitializerList? ','? '}'                   # CreateStruct
@@ -80,13 +81,18 @@ fieldInitializerList
     ;
 
 optField
-    : (opt='?')? IDENTIFIER
+    : (opt='?')? escapeIdent
     ;
 
 mapInitializerList
     : keys+=optExpr cols+=':' values+=expr (',' keys+=optExpr cols+=':' values+=expr)*
     ;
 
+escapeIdent
+    : id=IDENTIFIER      # SimpleIdentifier
+    | id=ESC_IDENTIFIER  # EscapedIdentifier
+;
+
 optExpr
     : (opt='?')? e=expr
     ;
@@ -198,3 +204,4 @@ STRING
 BYTES : ('b' | 'B') STRING;
 
 IDENTIFIER : (LETTER | '_') ( LETTER | DIGIT | '_')*;
+ESC_IDENTIFIER : '`' (BACKSLASH '`' | BACKSLASH BACKSLASH | ~('\r'|'\n'|'\\'|'`'))* '`';

parser/gen/CEL.tokens

@@ -34,6 +34,7 @@ NUM_UINT=33
 STRING=34
 BYTES=35
 IDENTIFIER=36
+ESC_IDENTIFIER=37
 '=='=1
 '!='=2
 'in'=3

parser/gen/CELLexer.tokens

@@ -34,6 +34,7 @@ NUM_UINT=33
 STRING=34
 BYTES=35
 IDENTIFIER=36
+ESC_IDENTIFIER=37
 '=='=1
 '!='=2
 'in'=3