feat(formula): date function with array

packages/engine-formula/src/functions/date/date/__tests__/index.spec.ts

@@ -19,17 +19,99 @@ import { describe, expect, it } from 'vitest';
 import { FUNCTION_NAMES_DATE } from '../../function-names';
 import { DateFunction } from '..';
 import { NumberValueObject } from '../../../../engine/value-object/primitive-object';
+import { ArrayValueObject, transformToValue, transformToValueObject } from '../../../../engine/value-object/array-value-object';
 
 describe('Test date function', () => {
  const textFunction = new DateFunction(FUNCTION_NAMES_DATE.DATE);
 
  describe('Date', () => {
- it('Value is normal', () => {
+ it('All value is normal', () => {
  const year = new NumberValueObject(2024);
  const month = new NumberValueObject(1);
  const day = new NumberValueObject(1);
  const result = textFunction.calculate(year, month, day);
- expect(result.getValue()).toBe(45292);
+ expect(transformToValue(result.getArrayValue())).toStrictEqual([[45292]]);
+ });
+
+ it('Year is single cell, month is one column, day is one row', () => {
+ const year = new NumberValueObject(2024);
+ const month = new ArrayValueObject({
+ calculateValueList: transformToValueObject([[1], [2]]),
+ rowCount: 2,
+ columnCount: 1,
+ unitId: '',
+ sheetId: '',
+ row: 0,
+ column: 0,
+ });
+ const day = new ArrayValueObject({
+ calculateValueList: transformToValueObject([[3, 4, 5]]),
+ rowCount: 1,
+ columnCount: 3,
+ unitId: '',
+ sheetId: '',
+ row: 0,
+ column: 0,
+ });
+ const result = textFunction.calculate(year, month, day);
+ expect(transformToValue(result.getArrayValue())).toStrictEqual([[45294, 45295, 45296], [45325, 45326, 45327]]);
+ });
+
+ it('Year is array with multiple format values', () => {
+ const year = new ArrayValueObject({
+ calculateValueList: transformToValueObject([
+ [1, ' ', 1.23, true, false, null, 18],
+ [0, '100', '2.34', 'test', -3, 1900, 108],
+ ]),
+ rowCount: 2,
+ columnCount: 7,
+ unitId: '',
+ sheetId: '',
+ row: 0,
+ column: 0,
+ });
+ const month = new NumberValueObject(1);
+ const day = new NumberValueObject(1);
+ const result = textFunction.calculate(year, month, day);
+ expect(transformToValue(result.getArrayValue())).toStrictEqual([[367, '#VALUE!', 367, 367, 1, 1, 6576], [1, 36526, 732, '#VALUE!', '#NUM!', 1, 39448]]);
+ });
+
+ it('Month is array with multiple format values', () => {
+ const year = new NumberValueObject(2024);
+ const month = new ArrayValueObject({
+ calculateValueList: transformToValueObject([
+ [1, ' ', 1.23, true, false, null],
+ [0, '100', '2.34', 'test', -3, 14],
+ ]),
+ rowCount: 2,
+ columnCount: 6,
+ unitId: '',
+ sheetId: '',
+ row: 0,
+ column: 0,
+ });
+ const day = new NumberValueObject(1);
+ const result = textFunction.calculate(year, month, day);
+ expect(transformToValue(result.getArrayValue())).toStrictEqual([[45292, '#VALUE!', 45292, 45292, 45261, 45261], [45261, 48305, 45323, '#VALUE!', 45170, 45689]]);
+ });
+
+ it('Day is array with multiple format values', () => {
+ const year = new NumberValueObject(2024);
+ const month = new NumberValueObject(1);
+ const day = new ArrayValueObject({
+ calculateValueList: transformToValueObject([
+ [1, ' ', 1.23, true, false, null],
+ [0, '100', '2.34', 'test', -3, 32],
+ ]),
+ rowCount: 2,
+ columnCount: 6,
+ unitId: '',
+ sheetId: '',
+ row: 0,
+ column: 0,
+ });
+ const result = textFunction.calculate(year, month, day);
+ expect(transformToValue(result.getArrayValue())).toStrictEqual([[45292, '#VALUE!', 45292, 45292, 45291, 45291], [45291, 45391, 45293, '#VALUE!', 45288, 45323]]);
  });
  });
 });

packages/engine-formula/src/functions/date/date/index.ts

@@ -14,9 +14,13 @@
  * limitations under the License.
  */
 
-import type { BaseValueObject } from '../../..';
-import { ErrorType, ErrorValueObject, NumberValueObject } from '../../..';
 import { DEFFAULT_DATE_FORMAT, excelDateSerial } from '../../../basics/date';
+import { ErrorType } from '../../../basics/error-type';
+import { expandArrayValueObject } from '../../../engine/utils/array-object';
+import type { ArrayValueObject } from '../../../engine/value-object/array-value-object';
+import type { BaseValueObject } from '../../../engine/value-object/base-value-object';
+import { ErrorValueObject } from '../../../engine/value-object/base-value-object';
+import { NumberValueObject } from '../../../engine/value-object/primitive-object';
 import { BaseFunction } from '../../base-function';
 
 export class DateFunction extends BaseFunction {
@@ -33,20 +37,63 @@ export class DateFunction extends BaseFunction {
  return day;
  }
 
- // TODO@Dushusir: array
- const yearValue = +year.getValue();
- const monthValue = +month.getValue();
- const dayValue = +day.getValue();
+ // get max row length
+ const maxRowLength = Math.max(
+ year.isArray() ? (year as ArrayValueObject).getRowCount() : 1,
+ month.isArray() ? (month as ArrayValueObject).getRowCount() : 1,
+ day.isArray() ? (day as ArrayValueObject).getRowCount() : 1
+ );
 
- const date = new Date(yearValue, monthValue - 1, dayValue);
+ // get max column length
+ const maxColumnLength = Math.max(
+ year.isArray() ? (year as ArrayValueObject).getColumnCount() : 1,
+ month.isArray() ? (month as ArrayValueObject).getColumnCount() : 1,
+ day.isArray() ? (day as ArrayValueObject).getColumnCount() : 1
+ );
 
- if (date.getMonth() !== monthValue - 1 || date.getDate() !== dayValue) {
- return new ErrorValueObject(ErrorType.VALUE);
- }
+ const yearArray = expandArrayValueObject(maxRowLength, maxColumnLength, year);
+ const monthArray = expandArrayValueObject(maxRowLength, maxColumnLength, month);
+ const dayArray = expandArrayValueObject(maxRowLength, maxColumnLength, day);
+
+ return yearArray.map((yearValueObject, rowIndex, columnIndex) => {
+ const monthValueObject = monthArray.get(rowIndex, columnIndex);
+ const dayValueObject = dayArray.get(rowIndex, columnIndex);
+
+ if (yearValueObject.isError()) {
+ return yearValueObject;
+ }
+
+ if (monthValueObject.isError()) {
+ return monthValueObject;
+ }
+
+ if (dayValueObject.isError()) {
+ return dayValueObject;
+ }
+
+ if (yearValueObject.isString() || monthValueObject.isString() || dayValueObject.isString()) {
+ return new ErrorValueObject(ErrorType.VALUE);
+ }
+
+ let yearValue = +yearValueObject.getValue();
+ const monthValue = Math.floor(+monthValueObject.getValue());
+ const dayValue = +dayValueObject.getValue();
+
+ if (yearValue < 0 || yearValue > 9999) {
+ return new ErrorValueObject(ErrorType.NUM);
+ }
+
+ if (yearValue >= 0 && yearValue < 1899) {
+ yearValue += 1900;
+ }
+
+ const date = new Date(yearValue, monthValue - 1, dayValue);
+
+ const currentSerial = excelDateSerial(date);
+ const valueObject = new NumberValueObject(currentSerial);
+ valueObject.setPattern(DEFFAULT_DATE_FORMAT);
 
- const currentSerial = excelDateSerial(date);
- const valueObject = new NumberValueObject(currentSerial);
- valueObject.setPattern(DEFFAULT_DATE_FORMAT);
- return valueObject;
+ return valueObject;
+ });
  }
 }