xref: /ide/tools/previewer/automock/mock-generate/src/main.ts

/*
 * Copyright (c) 2023 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 fs from 'fs';
import path from 'path';
import { createSourceFile, ScriptTarget } from 'typescript';
import {
  collectAllFileName,
  getAllClassDeclaration,
  dtsFileList,
  getOhosInterfacesDir,
  specialFiles,
  generateKitMap,
  generateDependJsonFile,
  generateMyComponent
} from './common/commonUtils';
import { getSourceFileAssembly } from './declaration-node/sourceFileElementsAssemply';
import { generateEntry } from './generate/generateEntry';
import { generateIndex } from './generate/generateIndex';
import { generateSourceFileElements } from './generate/generateMockJsFile';
import { generateSystemIndex } from './generate/generateSystemIndex';

/**
 * get all api .d.ts file path
 * @param dir
 * @returns
 */
function getAllDtsFile(dir: string): void {
  const arr = fs.readdirSync(dir);
  if (!dir.toString().includes('node_modules') && !dir.toString().includes(path.join('@internal', 'component'))) {
    arr.forEach(value => {
      const fullPath = path.join(dir, value);
      const stats = fs.statSync(fullPath);
      if (stats.isDirectory()) {
        getAllDtsFile(fullPath);
      } else {
        dtsFileList.push(fullPath);
      }
    });
  }
}

/**
 * get all component .d.ts file path
 * @param dir
 * @returns
 */
function getAllComponentsFilePath(dir: string): Array<string> {
  const componentPath = path.join(dir, '@internal', 'component', 'ets');
  if (!fs.existsSync(componentPath)) {
    return;
  }
  const componentPathArr = fs.readdirSync(componentPath);
  componentPathArr.forEach(value => {
    const fullPath = path.join(componentPath, value);
    if (fs.existsSync(fullPath) && !fs.statSync(fullPath).isDirectory()) {
      const componentName = `@internal/component/ets/${value}`;
      if (!specialFiles.includes(componentName)) {
        specialFiles.push(componentName);
      }
    }
  });
}

/**
 * mkdir
 * @param dirname
 * @returns
 */
function mkdirsSync(dirname): boolean {
  if (fs.existsSync(dirname)) {
    return true;
  } else {
    if (mkdirsSync(path.dirname(dirname))) {
      fs.mkdirSync(dirname);
      return true;
    }
  }
  return false;
}

/**
 * hgandle all ets file mock logic
 * @param outMockJsFileDir automated mock file output path
 * @returns
 */
function etsFileToMock(outMockJsFileDir: string): void {
  let index = 0;
  while (index < dtsFileList.length) {
    const value = dtsFileList[index];
    index++;

    if (!value.endsWith('.d.ts') && !value.endsWith('.d.ets')) {
      continue;
    }

    const code = fs.readFileSync(value);
    const sourceFile = createSourceFile(value, code.toString(), ScriptTarget.Latest);
    let fileName: string;
    if (value.endsWith('.d.ts')) {
      fileName = path.basename(value, '.d.ts');
    } else if (value.endsWith('.d.ets')) {
      fileName = path.basename(value, '.d.ets');
    } else {
      continue;
    }
    let outputFileName = '';
    if (fileName.includes('@')) {
      outputFileName = fileName.split('@')[1].replace(/\./g, '_');
    } else {
      outputFileName = fileName;
    }

    let tmpOutputMockJsFileDir = outMockJsFileDir;
    if (!outputFileName.startsWith('system_')) {
      tmpOutputMockJsFileDir = path.join(outMockJsFileDir, 'napi');
    }

    if (value.startsWith(getOhosInterfacesDir()) && !apiInputPath.startsWith(getOhosInterfacesDir())) {
      tmpOutputMockJsFileDir = path.join(tmpOutputMockJsFileDir, '@ohos');
    }

    let dirName = '';
    dirName = path.join(tmpOutputMockJsFileDir, path.dirname(value).split(`${path.sep}api`)[1]);
    if (!fs.existsSync(dirName)) {
      mkdirsSync(dirName);
    }
    const sourceFileEntity = getSourceFileAssembly(sourceFile, fileName);
    const filePath = path.join(dirName, outputFileName + '.js');
    fs.writeFileSync(filePath, '');
    fs.appendFileSync(
      path.join(filePath),
      generateSourceFileElements('', sourceFileEntity, sourceFile, outputFileName)
    );
  }
}

/**
 * Project Entry Function
 * @param apiInputPath interface_sdk-js\api absolute file path
 * @returns
 */
function main(apiInputPath): void {
  const dtsDir = apiInputPath;
  const outMockJsFileDir = path.join(__dirname, '../../runtime/main/extend/systemplugin');
  generateKitMap(apiInputPath);
  getAllDtsFile(dtsDir);
  getAllComponentsFilePath(dtsDir);
  dtsFileList.forEach(value => {
    collectAllFileName(value);
    if (value.endsWith('.d.ts') || value.endsWith('.d.ets')) {
      const code = fs.readFileSync(value);
      const sourceFile = createSourceFile(value, code.toString(), ScriptTarget.Latest);
      getAllClassDeclaration(sourceFile);
    }
  });

  etsFileToMock(outMockJsFileDir);

  if (!fs.existsSync(path.join(outMockJsFileDir, 'napi'))) {
    mkdirsSync(path.join(outMockJsFileDir, 'napi'));
  }
  generateDependJsonFile();
  generateMyComponent(outMockJsFileDir);
  fs.writeFileSync(path.join(outMockJsFileDir, 'napi', 'index.js'), generateIndex());
  fs.writeFileSync(path.join(outMockJsFileDir, 'index.js'), generateSystemIndex());
  fs.writeFileSync(path.join(outMockJsFileDir, 'entry.js'), generateEntry());
}

const paramIndex = 2;
const apiInputPath = process.argv[paramIndex];
main(apiInputPath);