Ability to provide a custom record context besides the standard record Id.

Author: cesarParra

expression-src/main/api/CustomRecordContext.cls

@@ -0,0 +1,10 @@
+// TODO:Make global
+public with sharing class CustomRecordContext {
+    public final String key;
+    public final Id recordId;
+
+    public CustomRecordContext(String key, Id recordId) {
+        this.key = key;
+        this.recordId = recordId;
+    }
+}

expression-src/main/api/CustomRecordContext.cls-meta.xml

@@ -0,0 +1,5 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<ApexClass xmlns="http://soap.sforce.com/2006/04/metadata">
+    <apiVersion>60.0</apiVersion>
+    <status>Active</status>
+</ApexClass>

expression-src/main/api/Evaluator.cls

@@ -3,7 +3,6 @@
  *              It provides a set of static methods that allow you to evaluate expressions
  *              in a variety of contexts.
  */
-// TODO: Add all overloads to the bulk endpoints
 global with sharing abstract class Evaluator {
     /**
      * @description Evaluates an expression and returns the result.
@@ -296,7 +295,7 @@ global with sharing abstract class Evaluator {
      * @description Evaluates a formula and returns the result using a record Id as the context.
      *              When using this endpoint field references will automatically be resolved
      *              and queried.
-     * @param formula The formula to evaluate.
+     * @param expression The formula to evaluate.
      * @param recordId The Id of the record to use as the context for the formula.
      * @return The result of the formula.
      * @example
@@ -306,8 +305,8 @@ global with sharing abstract class Evaluator {
      * String recordName = (String)expression.Evaluator.run('Name', record.Id);
      * ```
      */
-    global static Object run(String formula, Id recordId) {
-        return run(formula, recordId, new Configuration());
+    global static Object run(String expression, Id recordId) {
+        return run(expression, recordId, new Configuration());
     }
 
     /**
@@ -343,6 +342,10 @@ global with sharing abstract class Evaluator {
         return results;
     }
 
+    public static Object run(String expression, CustomRecordContext context) {
+        return EvaluatorResolver.forCustomContext(context).evaluate(expression, new Configuration()).resultValue;
+    }
+
     /**
      * @description Evaluates a formula and returns the result using a set of record Ids as the context.
      *              When using this endpoint field references will automatically be resolved

expression-src/main/src/interpreter/ContextResolver.cls

@@ -1,21 +1,28 @@
 public with sharing class ContextResolver implements Visitor {
     private Boolean shouldExecuteQuery = false;
     private Query queryContext;
+    private String contextReferenceName = '@context';
 
     private final List<Expr.FunctionDeclaration> customFunctionDeclarations;
     private final Query topLevelQuery;
     private final List<Query> subQueries = new List<Query>();
 
-    public ContextResolver(Id recordId, List<Expr.FunctionDeclaration> customFunctionDeclarations) {
+    public ContextResolver(Id recordId, List<Expr.FunctionDeclaration> customFunctionDeclarations, String contextReferenceName) {
         this.queryContext = new Query(recordId);
         this.topLevelQuery = this.queryContext;
         this.customFunctionDeclarations = customFunctionDeclarations;
+        if (String.isNotBlank(contextReferenceName)) {
+            this.contextReferenceName = contextReferenceName;
+        }
     }
 
-    public ContextResolver(Set<Id> recordIds, List<Expr.FunctionDeclaration> customFunctionDeclarations) {
+    public ContextResolver(Set<Id> recordIds, List<Expr.FunctionDeclaration> customFunctionDeclarations, String contextReferenceName) {
         this.queryContext = new Query(recordIds);
         this.topLevelQuery = this.queryContext;
         this.customFunctionDeclarations = customFunctionDeclarations;
+        if (String.isNotBlank(contextReferenceName)) {
+            this.contextReferenceName = contextReferenceName;
+        }
     }
 
     private Object resolve(Expr expr) {
@@ -222,9 +229,9 @@ public with sharing class ContextResolver implements Visitor {
         return relationshipToQuery + getExpr.field.lexeme;
     }
 
-    private static Boolean isReferencingContextField(Expr.GetExpr expr) {
+    private Boolean isReferencingContextField(Expr.GetExpr expr) {
         if (expr.objectExpr instanceof Expr.Variable) {
-            return ((Expr.Variable)expr.objectExpr).isContext();
+            return ((Expr.Variable)expr.objectExpr).isContext(this.contextReferenceName);
         } else if (expr.objectExpr instanceof Expr.GetExpr) {
             // Recursively check if the parent object is referencing a context field
             // since we can have nested get expressions and what is important is the top level
@@ -235,7 +242,7 @@ public with sharing class ContextResolver implements Visitor {
     }
 
     public Object visit(Expr.Variable variable) {
-        if (variable.isContext()) {
+        if (variable.isContext(this.contextReferenceName)) {
             // If context is being accessed, then we always want to run the query.
             this.shouldExecuteQuery = true;
 
@@ -483,7 +490,6 @@ public with sharing class ContextResolver implements Visitor {
         public Query(Set<Id> recordIds) {
             this.recordId = new MultipleIds(recordIds);
             this.objectType = this.recordId.getSobjectType();
-            System.debug('Querying for ' + this.objectType);
             this.queryBuilder = new Q(this.objectType);
             this.queryBuilder.selectField('Id');
         }

expression-src/main/src/interpreter/Environment.cls

@@ -73,6 +73,17 @@ public with sharing class Environment {
         return (Expr.FunctionDeclaration)GLOBAL_CONTEXT_VARIABLES.get(functionName.toLowerCase());
     }
 
+    /**
+     * @description Adds a global context variable that can be accessed from any formula.
+     * @param prefix The prefix to be added to the name of the global context variable.
+     * @param name The name of the global context variable.
+     * @param value The value to be stored in the global context variable.
+     */
+    public static void addGlobalContextVariable(String prefix, String name, Object value) {
+        String prefixedName = prefix + name;
+        GLOBAL_CONTEXT_VARIABLES.put(prefixedName.toLowerCase(), value);
+    }
+
     /**
      * @description Adds a global context variable that can be accessed from any formula.
      * Prefixes the name with '@'.

expression-src/main/src/interpreter/Expr.cls

@@ -165,8 +165,8 @@ public abstract class Expr {
             return v.visit(this);
         }
 
-        public Boolean isContext() {
-            return this.name.lexeme.toLowerCase() == '@context';
+        public Boolean isContext(String contextReferenceName) {
+            return this.name.lexeme.toLowerCase() == contextReferenceName.toLowerCase();
         }
     }