1/*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16import { ESMODULE } from './common/ark_define';
17import { ModuleBuildMode } from './module/module_build_mode';
18import { ModuleColdreloadMode } from './module/module_coldreload_mode';
19import { ModuleHotfixMode } from './module/module_hotfix_mode';
20import { ModuleHotreloadMode } from './module/module_hotreload_mode';
21import { ModulePreviewMode } from './module/module_preview_mode';
22import { ModuleSourceFile } from './module/module_source_file';
23import {
24  getHookEventFactory,
25  createAndStartEvent,
26  stopEvent
27} from '../../ark_utils';
28import type { ModuleMode } from './module/module_mode';
29import { SourceMapGenerator } from './generate_sourcemap';
30
31let moduleMode: ModuleMode = null;
32
33export async function generateModuleAbc(error) {
34  const hookEventFactory = getHookEventFactory(this.share, 'genAbc', 'buildEnd');
35  if (error) {
36    // When error thrown in previous plugins, rollup will catch and call buildEnd plugin.
37    // Stop generate abc if error exists
38    return;
39  }
40  if (this.share.projectConfig.compileMode === ESMODULE) {
41    await ModuleSourceFile.processModuleSourceFiles(this, hookEventFactory);
42    if (this.share.projectConfig.compileHar) {
43      SourceMapGenerator.getInstance().buildModuleSourceMapInfo(hookEventFactory);
44      // compileHar: compile har of project, which convert .ets to .d.ts and js
45      if (!this.share.projectConfig.byteCodeHar) {
46        return;
47      }
48    }
49    generateAbc(this, hookEventFactory);
50  }
51}
52
53function generateAbc(rollupObject: Object, parentEvent: Object): void {
54  const eventGenerateAbc = createAndStartEvent(parentEvent, 'generate abc');
55  if (rollupObject.share.projectConfig.watchMode !== 'true') {
56    if (rollupObject.share.arkProjectConfig.coldReload) {
57      const moduleColdreloadMode: ModuleColdreloadMode = new ModuleColdreloadMode(rollupObject);
58      moduleColdreloadMode.generateAbc(rollupObject, eventGenerateAbc);
59      moduleMode = moduleColdreloadMode;
60    } else if (rollupObject.share.arkProjectConfig.hotReload) {
61      // If file changes are monitored by the IDE, rollup is not started in watch mode, 
62      // so rollupObject.share.projectConfig.watchMode is not true. Hotreload in this mode 
63      // supports scenarios where entry depends on HAR and HSP.
64      const moduleHotreloadMode: ModuleHotreloadMode = new ModuleHotreloadMode(rollupObject);
65      moduleHotreloadMode.generateAbc(rollupObject, eventGenerateAbc);
66      moduleMode = moduleHotreloadMode;
67    } else {
68      const moduleBuildMode: ModuleBuildMode = new ModuleBuildMode(rollupObject);
69      moduleBuildMode.generateAbc(rollupObject, eventGenerateAbc);
70      moduleMode = moduleBuildMode;
71    }
72  } else if (rollupObject.share.arkProjectConfig.hotReload) {
73    // If file changes are monitored by rollup, rollup must be started in watch mode, 
74    // so rollupObject.share.projectConfig.watchMode needs to be true. Hotreload in this mode
75    // does not support scenarios where entry depends on HSP.
76    const moduleHotreloadMode: ModuleHotreloadMode = new ModuleHotreloadMode(rollupObject);
77    moduleHotreloadMode.generateAbc(rollupObject, eventGenerateAbc);
78    moduleMode = moduleHotreloadMode;
79  } else if (rollupObject.share.arkProjectConfig.hotFix) {
80    const moduleHotfixMode: ModuleHotfixMode = new ModuleHotfixMode(rollupObject);
81    moduleHotfixMode.generateAbc(rollupObject, eventGenerateAbc);
82    moduleMode = moduleHotfixMode;
83  } else {
84    const modulePreviewMode: ModulePreviewMode = new ModulePreviewMode(rollupObject);
85    modulePreviewMode.generateAbc(rollupObject, eventGenerateAbc);
86    moduleMode = modulePreviewMode;
87  }
88  stopEvent(eventGenerateAbc);
89}
90
91export function cleanModuleMode(): void {
92  if (moduleMode) {
93    moduleMode.triggerAsync = null;
94    moduleMode.triggerEndSignal = null;
95    moduleMode = null;
96  }
97}
98