@atomic-ehr/fhirpath 0.0.1-canary.69eb286.20250724163205

package/src/registry/operations/collection.ts

@@ -0,0 +1,422 @@
+import type { Function, Operator } from '../types';
+import { defaultFunctionAnalyze, defaultOperatorAnalyze } from '../default-analyzers';
+import { defaultFunctionCompile, defaultOperatorCompile } from '../default-compilers';
+import { TokenType } from '../../lexer/token';
+
+export const unionFunction: Function = {
+  name: 'union',
+  kind: 'function',
+
+  syntax: {
+    notation: 'union(other)'
+  },
+
+  signature: {
+    input: {
+      types: { kind: 'any' },
+      cardinality: 'any'
+    },
+    parameters: [
+      {
+        name: 'other',
+        kind: 'expression',
+        types: { kind: 'any' },
+        cardinality: 'any',
+        optional: false
+      }
+    ],
+    output: {
+      type: 'preserve-input',
+      cardinality: 'collection'
+    },
+    propagatesEmpty: false,
+    deterministic: true
+  },
+
+  analyze: defaultFunctionAnalyze,
+
+  evaluate: (interpreter, context, input, otherExpr) => {
+    const otherResult = interpreter.evaluate(otherExpr, input, context);
+    const other = otherResult.value;
+
+    const seen = new Set();
+    const result: any[] = [];
+
+    for (const item of input) {
+      const key = JSON.stringify(item);
+      if (!seen.has(key)) {
+        seen.add(key);
+        result.push(item);
+      }
+    }
+
+    for (const item of other) {
+      const key = JSON.stringify(item);
+      if (!seen.has(key)) {
+        seen.add(key);
+        result.push(item);
+      }
+    }
+
+    return { value: result, context: otherResult.context };
+  },
+
+  compile: (compiler, input, args) => {
+    const otherExpr = args[0];
+    if (!otherExpr) {
+      throw new Error('union() requires an argument');
+    }
+
+    return {
+      fn: (ctx) => {
+        const inputVal = input.fn(ctx);
+        const otherVal = otherExpr.fn(ctx);
+
+        const seen = new Set();
+        const result: any[] = [];
+
+        for (const item of inputVal) {
+          const key = JSON.stringify(item);
+          if (!seen.has(key)) {
+            seen.add(key);
+            result.push(item);
+          }
+        }
+
+        for (const item of otherVal) {
+          const key = JSON.stringify(item);
+          if (!seen.has(key)) {
+            seen.add(key);
+            result.push(item);
+          }
+        }
+
+        return result;
+      },
+      type: input.type,
+      isSingleton: false,
+      source: `${input.source || ''}.union(${otherExpr.source || ''})`
+    };
+  }
+};
+
+export const combineFunction: Function = {
+  name: 'combine',
+  kind: 'function',
+
+  syntax: {
+    notation: 'combine(other)'
+  },
+
+  signature: {
+    input: {
+      types: { kind: 'any' },
+      cardinality: 'any'
+    },
+    parameters: [
+      {
+        name: 'other',
+        kind: 'expression',
+        types: { kind: 'any' },
+        cardinality: 'any',
+        optional: false
+      }
+    ],
+    output: {
+      type: 'preserve-input',
+      cardinality: 'collection'
+    },
+    propagatesEmpty: false,
+    deterministic: true
+  },
+
+  analyze: defaultFunctionAnalyze,
+
+  evaluate: (interpreter, context, input, otherExpr) => {
+    const otherResult = interpreter.evaluate(otherExpr, input, context);
+    const other = otherResult.value;
+
+    return { value: [...input, ...other], context: otherResult.context };
+  },
+
+  compile: (compiler, input, args) => {
+    const otherExpr = args[0];
+    if (!otherExpr) {
+      throw new Error('combine() requires an argument');
+    }
+
+    return {
+      fn: (ctx) => {
+        const inputVal = input.fn(ctx);
+        const otherVal = otherExpr.fn(ctx);
+        return [...inputVal, ...otherVal];
+      },
+      type: input.type,
+      isSingleton: false,
+      source: `${input.source || ''}.combine(${otherExpr.source || ''})`
+    };
+  }
+};
+
+export const intersectFunction: Function = {
+  name: 'intersect',
+  kind: 'function',
+
+  syntax: {
+    notation: 'intersect(other)'
+  },
+
+  signature: {
+    input: {
+      types: { kind: 'any' },
+      cardinality: 'any'
+    },
+    parameters: [
+      {
+        name: 'other',
+        kind: 'expression',
+        types: { kind: 'any' },
+        cardinality: 'any',
+        optional: false
+      }
+    ],
+    output: {
+      type: 'preserve-input',
+      cardinality: 'collection'
+    },
+    propagatesEmpty: true,
+    deterministic: true
+  },
+
+  analyze: defaultFunctionAnalyze,
+
+  evaluate: (interpreter, context, input, otherExpr) => {
+    const otherResult = interpreter.evaluate(otherExpr, input, context);
+    const other = otherResult.value;
+
+    const otherSet = new Set(other.map((item: any) => JSON.stringify(item)));
+    const result: any[] = [];
+    const seen = new Set();
+
+    for (const item of input) {
+      const key = JSON.stringify(item);
+      if (otherSet.has(key) && !seen.has(key)) {
+        seen.add(key);
+        result.push(item);
+      }
+    }
+
+    return { value: result, context: otherResult.context };
+  },
+
+  compile: (compiler, input, args) => {
+    const otherExpr = args[0];
+    if (!otherExpr) {
+      throw new Error('intersect() requires an argument');
+    }
+
+    return {
+      fn: (ctx) => {
+        const inputVal = input.fn(ctx);
+        const otherVal = otherExpr.fn(ctx);
+
+        const otherSet = new Set(otherVal.map((item: any) => JSON.stringify(item)));
+        const result: any[] = [];
+        const seen = new Set();
+
+        for (const item of inputVal) {
+          const key = JSON.stringify(item);
+          if (otherSet.has(key) && !seen.has(key)) {
+            seen.add(key);
+            result.push(item);
+          }
+        }
+
+        return result;
+      },
+      type: input.type,
+      isSingleton: false,
+      source: `${input.source || ''}.intersect(${otherExpr.source || ''})`
+    };
+  }
+};
+
+export const excludeFunction: Function = {
+  name: 'exclude',
+  kind: 'function',
+
+  syntax: {
+    notation: 'exclude(other)'
+  },
+
+  signature: {
+    input: {
+      types: { kind: 'any' },
+      cardinality: 'any'
+    },
+    parameters: [
+      {
+        name: 'other',
+        kind: 'expression',
+        types: { kind: 'any' },
+        cardinality: 'any',
+        optional: false
+      }
+    ],
+    output: {
+      type: 'preserve-input',
+      cardinality: 'collection'
+    },
+    propagatesEmpty: true,
+    deterministic: true
+  },
+
+  analyze: defaultFunctionAnalyze,
+
+  evaluate: (interpreter, context, input, otherExpr) => {
+    const otherResult = interpreter.evaluate(otherExpr, input, context);
+    const other = otherResult.value;
+
+    const excludeSet = new Set(other.map((item: any) => JSON.stringify(item)));
+    const result: any[] = [];
+    const seen = new Set();
+
+    for (const item of input) {
+      const key = JSON.stringify(item);
+      if (!excludeSet.has(key) && !seen.has(key)) {
+        seen.add(key);
+        result.push(item);
+      }
+    }
+
+    return { value: result, context: otherResult.context };
+  },
+
+  compile: (compiler, input, args) => {
+    const otherExpr = args[0];
+    if (!otherExpr) {
+      throw new Error('exclude() requires an argument');
+    }
+
+    return {
+      fn: (ctx) => {
+        const inputVal = input.fn(ctx);
+        const otherVal = otherExpr.fn(ctx);
+
+        const excludeSet = new Set(otherVal.map((item: any) => JSON.stringify(item)));
+        const result: any[] = [];
+        const seen = new Set();
+
+        for (const item of inputVal) {
+          const key = JSON.stringify(item);
+          if (!excludeSet.has(key) && !seen.has(key)) {
+            seen.add(key);
+            result.push(item);
+          }
+        }
+
+        return result;
+      },
+      type: input.type,
+      isSingleton: false,
+      source: `${input.source || ''}.exclude(${otherExpr.source || ''})`
+    };
+  }
+};
+
+// Union operator (|) - combines collections removing duplicates
+export const unionOperator: Operator = {
+  name: '|',
+  kind: 'operator',
+
+  syntax: {
+    form: 'infix',
+    token: TokenType.PIPE,
+    precedence: 13,  // Lower precedence than most operators
+    associativity: 'left',
+    notation: 'a | b'
+  },
+
+  signature: {
+    parameters: [
+      { name: 'left' },
+      { name: 'right' }
+    ],
+    output: {
+      type: 'preserve-left',
+      cardinality: 'collection'
+    },
+    propagatesEmpty: false
+  },
+
+  analyze: defaultOperatorAnalyze,
+
+  evaluate: (interpreter, context, input, left, right) => {
+    // Union removes duplicates
+    const seen = new Set();
+    const result: any[] = [];
+
+    for (const item of left) {
+      const key = JSON.stringify(item);
+      if (!seen.has(key)) {
+        seen.add(key);
+        result.push(item);
+      }
+    }
+
+    for (const item of right) {
+      const key = JSON.stringify(item);
+      if (!seen.has(key)) {
+        seen.add(key);
+        result.push(item);
+      }
+    }
+
+    return { value: result, context };
+  },
+  //TODO: Fix later
+  //@ts-expect-error it's okay currently
+  compile: (compiler, input, left, right) => {
+    return {
+      fn: (ctx) => {
+        // Evaluate both sides with SEPARATE context copies
+        // This prevents variable definitions from leaking between branches
+        const leftCtx = { ...ctx, env: { ...ctx.env } };
+        const rightCtx = { ...ctx, env: { ...ctx.env } };
+
+        const leftVal = left.fn(leftCtx);
+        const rightVal = right.fn(rightCtx);
+
+        // Union removes duplicates
+        const seen = new Set();
+        const result: any[] = [];
+
+        for (const item of leftVal) {
+          const key = JSON.stringify(item);
+          if (!seen.has(key)) {
+            seen.add(key);
+            result.push(item);
+          }
+        }
+
+        for (const item of rightVal) {
+          const key = JSON.stringify(item);
+          if (!seen.has(key)) {
+            seen.add(key);
+            result.push(item);
+          }
+        }
+
+        return result;
+      },
+      type: left.type,
+      isSingleton: false,
+      source: `${left.source || ''} | ${right.source || ''}`
+    };
+  }
+};
+
+// Export all collection operations
+export const collectionOperators = [
+  unionOperator
+];