Function splitASTConditionByVariables

packages/graph/src/astToSteps.ts:1841–1957

* Split an AST condition into early and late parts based on which variables are referenced. * Early conditions only reference variables in the allowedVariables list. * Late conditions reference other variables.

(
  condition: Condition,
  allowedVariables: string[],
)

 * Late conditions reference other variables.
 */
function splitASTConditionByVariables(
  condition: Condition,
  allowedVariables: string[],
): { early?: Condition; late?: Condition } {
  // Handle logical operators (and, or, xor)
  if (
    condition.type === "AndCondition" ||
    condition.type === "OrCondition" ||
    condition.type === "XorCondition"
  ) {
    const leftSplit = splitASTConditionByVariables(
      (condition as AndCondition | OrCondition | XorCondition).left,
      allowedVariables,
    );
    const rightSplit = splitASTConditionByVariables(
      (condition as AndCondition | OrCondition | XorCondition).right,
      allowedVariables,
    );

    const earlyParts = [leftSplit.early, rightSplit.early].filter(
      (c): c is Condition => c !== undefined,
    );
    const lateParts = [leftSplit.late, rightSplit.late].filter(
      (c): c is Condition => c !== undefined,
    );

    const buildCondition = (parts: Condition[], type: string): Condition => {
      if (type === "AndCondition") {
        return {
          type: "AndCondition",
          left: parts[0]!,
          right: parts[1]!,
        } as AndCondition;
      } else if (type === "OrCondition") {
        return {
          type: "OrCondition",
          left: parts[0]!,
          right: parts[1]!,
        } as OrCondition;
      } else {
        return {
          type: "XorCondition",
          left: parts[0]!,
          right: parts[1]!,
        } as XorCondition;
      }
    };

    const early =
      earlyParts.length === 0
        ? undefined
        : earlyParts.length === 1
          ? earlyParts[0]
          : buildCondition(earlyParts, condition.type);

    const late =
      lateParts.length === 0
        ? undefined