builder-lambda/custom-component/custom-component-call.test.ts

/*
 * Copyright (c) 2025 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import * as path from 'path';
import { PluginTestContext, PluginTester } from '../../../../utils/plugin-tester';
import { BuildConfig, mockBuildConfig } from '../../../../utils/artkts-config';
import { getRootPath, MOCK_ENTRY_DIR_PATH } from '../../../../utils/path-config';
import { parseDumpSrc } from '../../../../utils/parse-string';
import { recheck, uiNoRecheck } from '../../../../utils/plugins';
import { uiTransform } from '../../../../../ui-plugins';
import { Plugins } from '../../../../../common/plugin-context';

const BUILDER_LAMBDA_DIR_PATH: string = 'builder-lambda';
const CUSTOM_COMPONENT_DIR_PATH: string = 'custom-component';

const buildConfig: BuildConfig = mockBuildConfig();
buildConfig.compileFiles = [
    path.resolve(getRootPath(), MOCK_ENTRY_DIR_PATH, BUILDER_LAMBDA_DIR_PATH, CUSTOM_COMPONENT_DIR_PATH, 'custom-component-call.ets'),
];

const pluginTester = new PluginTester('test custom component call transformation', buildConfig);

const parsedTransform: Plugins = {
    name: 'custom-component-call',
    parsed: uiTransform().parsed
};

const expectedParsedScript: string = `
import { __memo_id_type as __memo_id_type } from "@ohos.arkui.stateManagement";
import { __memo_context_type as __memo_context_type } from "@ohos.arkui.stateManagement";
import { memo as memo } from "@ohos.arkui.stateManagement";
import { UIColumnAttribute as UIColumnAttribute } from "@ohos.arkui.component";
import { UITextAttribute as UITextAttribute } from "@ohos.arkui.component";
import { CustomComponent as CustomComponent } from "@ohos.arkui.component";
import { Text as Text, Column as Column, Component as Component, Builder as Builder, BuilderParam as BuilderParam } from "@ohos.arkui.component";

@Component() final class CustomContainer extends CustomComponent<CustomContainer, __Options_CustomContainer> {
  @Builder() public closerBuilder() {}
  @BuilderParam() public closer: (()=> void) = (this).closerBuilder;
  public build() {}
  public constructor() {}
}

@Component() final class CustomContainerUser extends CustomComponent<CustomContainerUser, __Options_CustomContainerUser> {
  public build() {
    Column(undefined){
      CustomContainer(undefined){
        Column(undefined){
          Text("hello", undefined);
        };
      };
      CustomContainer(({} as __Options_CustomContainer)){
        Column(undefined){};
      };
      CustomContainer(undefined){};
      CustomContainer();
    };
  }
  public constructor() {}
}

interface __Options_CustomContainer {
  closer?: @memo() (()=> void);
}

interface __Options_CustomContainerUser {
}
`;

function testParedTransformer(this: PluginTestContext): void {
    expect(parseDumpSrc(this.scriptSnapshot ?? '')).toBe(parseDumpSrc(expectedParsedScript));
}

const expectedBuilderLambdaScript: string = `
import { __memo_id_type as __memo_id_type } from "@ohos.arkui.stateManagement";
import { __memo_context_type as __memo_context_type } from "@ohos.arkui.stateManagement";
import { memo as memo } from "@ohos.arkui.stateManagement";
import { UIColumnAttribute as UIColumnAttribute } from "@ohos.arkui.component";
import { UITextAttribute as UITextAttribute } from "@ohos.arkui.component";
import { CustomComponent as CustomComponent } from "@ohos.arkui.component";
import { Text as Text, Column as Column, Component as Component, Builder as Builder, BuilderParam as BuilderParam } from "@ohos.arkui.component";

function main() {}

@Component({freezeWhenInactive:false}) final class CustomContainer extends CustomComponent<CustomContainer, __Options_CustomContainer> {
  public __initializeStruct(initializers: __Options_CustomContainer | undefined, @memo() content: (()=> void) | undefined): void {
    (this).__backing_closer = ((((({let gensym___38813563 = initializers;
    (((gensym___38813563) == (null)) ? undefined : gensym___38813563.closer)})) ?? (content))) ?? ((this).closerBuilder))
  }
  public __updateStruct(initializers: __Options_CustomContainer | undefined): void {}
  private __backing_closer?: @memo() (()=> void);
  public get closer(): @memo() (()=> void) {
    return (this).__backing_closer!;
  }
  public set closer(@memo() value: (()=> void)) {
    (this).__backing_closer = value;
  }
  @memo() public closerBuilder() {}
  @memo() public _build(@memo() style: ((instance: CustomContainer)=> CustomContainer) | undefined, @memo() content: (()=> void) | undefined, initializers: __Options_CustomContainer | undefined): void {}
  public constructor() {}
}

@Component({freezeWhenInactive:false}) final class CustomContainerUser extends CustomComponent<CustomContainerUser, __Options_CustomContainerUser> {
  public __initializeStruct(initializers: __Options_CustomContainerUser | undefined, @memo() content: (()=> void) | undefined): void {}
  public __updateStruct(initializers: __Options_CustomContainerUser | undefined): void {}
  @memo() public _build(@memo() style: ((instance: CustomContainerUser)=> CustomContainerUser) | undefined, @memo() content: (()=> void) | undefined, initializers: __Options_CustomContainerUser | undefined): void {
    Column(undefined, undefined, (() => {
      CustomContainer._instantiateImpl(undefined, (() => {
        return new CustomContainer();
      }), undefined, (() => {
        Column(undefined, undefined, (() => {
          Text(undefined, "hello", undefined, undefined);
        }));
      }), undefined);
      CustomContainer._instantiateImpl(undefined, (() => {
        return new CustomContainer();
      }), ({} as __Options_CustomContainer), (() => {
        Column(undefined, undefined, (() => {}));
      }), undefined);
      CustomContainer._instantiateImpl(undefined, (() => {
        return new CustomContainer();
      }), undefined, (() => {}), undefined);
      CustomContainer._instantiateImpl(undefined, (() => {
        return new CustomContainer();
      }), undefined, undefined, undefined);
    }));
  }
  public constructor() {}
}

interface __Options_CustomContainer {
  set closer(closer: @memo() (()=> void) | undefined)
  get closer(): @memo() (()=> void) | undefined
}

interface __Options_CustomContainerUser {
}
`;

function testCustomComponentTransformer(this: PluginTestContext): void {
    expect(parseDumpSrc(this.scriptSnapshot ?? '')).toBe(parseDumpSrc(expectedBuilderLambdaScript));
}

pluginTester.run(
    'test custom component call transformation',
    [parsedTransform, recheck, uiNoRecheck],
    {
        parsed: [testParedTransformer],
        'checked:builder-lambda-no-recheck': [testCustomComponentTransformer],
    },
    {
        stopAfter: 'checked',
    }
);