Support USER/SYSTEM mode on SOSL

CHANGELOG.md

@@ -1,6 +1,10 @@
 # apex-parser - Changelog
 
-## 4.0.0 - 2023-03-28
+## 4.1.0 - 2024-05-12
+
+- Allow WITH USER_MODE or SYSTEM_MODE on SOSL queries
+
+## 4.0.0 - 2024-03-28
 
 - Correct trigger body parsing to allow member declarations
 - Add support for TYPEOF in SOQL subqueries

antlr/ApexParser.g4

@@ -826,17 +826,23 @@ soslLiteralAlt
 soslClauses
     : (IN searchGroup)?
       (RETURNING fieldSpecList)?
-      (WITH DIVISION ASSIGN StringLiteral)?
-      (WITH DATA CATEGORY filteringExpression)?
-      (WITH SNIPPET (LPAREN TARGET_LENGTH ASSIGN IntegerLiteral RPAREN)? )?
-      (WITH NETWORK IN LPAREN networkList RPAREN)?
-      (WITH NETWORK ASSIGN StringLiteral)?
-      (WITH PRICEBOOKID ASSIGN StringLiteral)?
-      (WITH METADATA ASSIGN StringLiteral)?
+      soslWithClause*
       limitClause?
       (UPDATE updateList)?
     ;
 
+soslWithClause
+    : WITH DIVISION ASSIGN StringLiteral
+    | WITH DATA CATEGORY filteringExpression
+    | WITH SNIPPET (LPAREN TARGET_LENGTH ASSIGN IntegerLiteral RPAREN)?
+    | WITH NETWORK IN LPAREN networkList RPAREN
+    | WITH NETWORK ASSIGN StringLiteral
+    | WITH PRICEBOOKID ASSIGN StringLiteral
+    | WITH METADATA ASSIGN StringLiteral
+    | WITH USER_MODE
+    | WITH SYSTEM_MODE
+    ;
+
 searchGroup
     : (ALL|EMAIL|NAME|PHONE|SIDEBAR) FIELDS
     ;

jvm/src/test/java/io/github/apexdevtools/apexparser/SOSLParserTest.java

@@ -60,4 +60,20 @@ void testQuotesFailOnAltFormat() {
         assertNotNull(context);
         assertEquals(1, parserAndCounter.getValue().getNumErrors());
     }
+
+    @Test
+    void testWithUserModeQuery() {
+        Map.Entry<ApexParser, SyntaxErrorCounter> parserAndCounter = createParser("[Find 'something' RETURNING Account WITH USER_MODE WITH METADATA='Labels']");
+        ApexParser.SoslLiteralContext context = parserAndCounter.getKey().soslLiteral();
+        assertNotNull(context);
+        assertEquals(0, parserAndCounter.getValue().getNumErrors());
+    }
+
+    @Test
+    void testWithSystemModeQuery() {
+        Map.Entry<ApexParser, SyntaxErrorCounter> parserAndCounter = createParser("[Find 'something' RETURNING Account WITH METADATA='Labels' WITH SYSTEM_MODE]");
+        ApexParser.SoslLiteralContext context = parserAndCounter.getKey().soslLiteral();
+        assertNotNull(context);
+        assertEquals(0, parserAndCounter.getValue().getNumErrors());
+    }
 }

npm/src/__tests__/SOSLParserTest.ts

@@ -61,3 +61,21 @@ test('testQuotesFailOnAltFormat', () => {
     expect(context).toBeInstanceOf(SoslLiteralAltContext)
     expect(errorCounter.getNumErrors()).toEqual(1)
 })
+
+test('testWithUserModeQuery', () => {
+    const [parser, errorCounter] = createParser("[Find 'something' RETURNING Account WITH USER_MODE WITH METADATA='Labels']")
+
+    const context = parser.soslLiteralAlt()
+
+    expect(context).toBeInstanceOf(SoslLiteralAltContext)
+    expect(errorCounter.getNumErrors()).toEqual(1)
+})
+
+test('testWithSystemModeQuery', () => {
+    const [parser, errorCounter] = createParser("[Find 'something' RETURNING Account WITH METADATA='Labels' WITH SYSTEM_MODE]")
+
+    const context = parser.soslLiteralAlt()
+
+    expect(context).toBeInstanceOf(SoslLiteralAltContext)
+    expect(errorCounter.getNumErrors()).toEqual(1)
+})