/*
 * 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 { HvigorNode, HvigorPlugin, hvigor } from '@ohos/hvigor';
import PBExecuteController from './src/tasks/PBExecuteController';
import { Config } from './src/constants/TypesConfig';

const PLUGIN_ID = "PBPlugin";

export function turboTransProtobufPlugin(options: Config): HvigorPlugin {
  return {
    pluginId: PLUGIN_ID,
    async apply(node: HvigorNode) {
      hvigor.nodesEvaluated(async () => {
        node.subNodes(subNode => {
          const subNodeName = subNode.getNodeName();
          if (options.ignoreModuleNames?.includes(subNodeName)) {
            return;
          }
          // 注册模块级任务
          PBExecuteController.execute(subNode, options);
        });
      });
    }
  };
}