From fb42cc209373d36a59fde32bc64d0b3208639246 Mon Sep 17 00:00:00 2001
From: Dushusir <1414556676@qq.com>
Date: Fri, 13 Dec 2024 18:26:42 +0800
Subject: [PATCH] feat(formula): support lambda in function register

---
 .../src/engine/ast-node/function-node.ts             |  4 ++++
 .../src/engine/value-object/lambda-value-object.ts   | 12 ++++++++++++
 .../engine-formula/src/functions/base-function.ts    |  2 +-
 packages/facade/src/apis/__tests__/facade.spec.ts    | 10 ++++++++++
 4 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/packages/engine-formula/src/engine/ast-node/function-node.ts b/packages/engine-formula/src/engine/ast-node/function-node.ts
index e2f2eafdb44..91ee6496733 100644
--- a/packages/engine-formula/src/engine/ast-node/function-node.ts
+++ b/packages/engine-formula/src/engine/ast-node/function-node.ts
@@ -220,6 +220,10 @@ export class FunctionNode extends BaseAstNode {
                         return (variant as ArrayValueObject).toValue();
                     }
 
+                    if (variant.isLambda()) {
+                        return variant;
+                    }
+
                     return variant.getValue();
                 })
             );
diff --git a/packages/engine-formula/src/engine/value-object/lambda-value-object.ts b/packages/engine-formula/src/engine/value-object/lambda-value-object.ts
index 86bd578eefb..31b3cda2df9 100644
--- a/packages/engine-formula/src/engine/value-object/lambda-value-object.ts
+++ b/packages/engine-formula/src/engine/value-object/lambda-value-object.ts
@@ -20,10 +20,12 @@ import type { BaseAstNode } from '../ast-node/base-ast-node';
 import type { LambdaParameterNode } from '../ast-node/lambda-parameter-node';
 import type { Interpreter } from '../interpreter/interpreter';
 import type { BaseReferenceObject, FunctionVariantType } from '../reference-object/base-reference-object';
+import type { PrimitiveValueType } from './primitive-object';
 import { ErrorType } from '../../basics/error-type';
 import { DEFAULT_TOKEN_TYPE_LAMBDA_RUNTIME_PARAMETER } from '../../basics/token-type';
 import { AsyncObject } from '../reference-object/base-reference-object';
 import { generateExecuteAstNodeData } from '../utils/ast-node-tool';
+import { ValueObjectFactory } from './array-value-object';
 import { BaseValueObject, ErrorValueObject } from './base-value-object';
 
 function getRootLexerHasValueNode(node: Nullable<BaseAstNode>): Nullable<BaseAstNode> {
@@ -110,6 +112,16 @@ export class LambdaValueObjectObject extends BaseValueObject {
         return value;
     }
 
+    /**
+     * Execute custom lambda function, handle basic types
+     * @param variants
+     */
+    executeCustom(...variants: PrimitiveValueType[]) {
+        // Create base value object from primitive value, then execute
+        const baseValueObjects = variants.map((variant) => ValueObjectFactory.create(variant));
+        return this.execute(...baseValueObjects);
+    }
+
     private _setLambdaNodeValue(node: Nullable<BaseAstNode>) {
         if (!node) {
             return;
diff --git a/packages/engine-formula/src/functions/base-function.ts b/packages/engine-formula/src/functions/base-function.ts
index f7f20bb5b6e..edf01b7ba2c 100644
--- a/packages/engine-formula/src/functions/base-function.ts
+++ b/packages/engine-formula/src/functions/base-function.ts
@@ -191,7 +191,7 @@ export class BaseFunction {
     }
 
     calculateCustom(
-        ...arg: Array<PrimitiveValueType | PrimitiveValueType[][]>
+        ...arg: Array<PrimitiveValueType | PrimitiveValueType[][] | BaseValueObject>
     ): PrimitiveValueType | PrimitiveValueType[][] {
         return null;
     }
diff --git a/packages/facade/src/apis/__tests__/facade.spec.ts b/packages/facade/src/apis/__tests__/facade.spec.ts
index d84b0f4fee7..c5fdb90b6ce 100644
--- a/packages/facade/src/apis/__tests__/facade.spec.ts
+++ b/packages/facade/src/apis/__tests__/facade.spec.ts
@@ -183,6 +183,16 @@ describe('Test FUniver', () => {
                 [function (...variants) {
                     let sum = 0;
 
+                    const last = variants[variants.length - 1];
+                    // @ts-ignore
+                    if (last.isLambda()) {
+                        variants.pop();
+
+                        const variantsList = variants.map((variant) => Array.isArray(variant) ? variant[0][0] : variant);
+                        // @ts-ignore
+                        sum += last.executeCustom(...variantsList).getValue();
+                    }
+
                     for (const variant of variants) {
                         sum += Number(variant) || 0;
                     }