xref: /ide/tools/previewer/automock/mock-generate/src/generate/generateEnumDeclaration.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 { SyntaxKind } from 'typescript';
import type { EnumEntity } from '../declaration-node/enumDeclaration';

const colorSpaceSpecialData = {
  LINEAR_BT709: 24,
  DISPLAY_SRGB: 4,
  DISPLAY_P3_SRGB: 3,
  DISPLAY_P3_HLG: 11,
  DISPLAY_P3_PQ: 12
};

/**
 * generate enum
 * @param rootName
 * @param enumDeclaration
 * @returns
 */
export function generateEnumDeclaration(rootName: string, enumDeclaration: EnumEntity): string {
  let enumBody = '';
  if (enumDeclaration.exportModifiers.length !== 0) {
    enumBody += `export const ${enumDeclaration.enumName} = {\n`;
  } else {
    enumBody += `const ${enumDeclaration.enumName} = {\n`;
  }

  let defaultValue = 0;
  enumDeclaration.enumMembers.forEach(member => {
    if (member.enumKind === SyntaxKind.TypeReference) {
      enumBody += `${member.enumValueName}: new ${member.enumValue},\n`;
    } else if (member.enumKind === SyntaxKind.NumericLiteral) {
      defaultValue = parseInt(member.enumValue.replace(/"/g, '').replace(/'/g, ''));
      enumBody += `${member.enumValueName}: ${defaultValue},\n`;
      defaultValue++;
    } else if (member.enumKind === SyntaxKind.StringLiteral) {
      enumBody += `${member.enumValueName}: ${member.enumValue},\n`;
    } else {
      if (member.enumValue === '' || member.enumValue === null || member.enumValue === undefined) {
        enumBody += `${member.enumValueName}: ${defaultValue},\n`;
        defaultValue++;
      } else {
        if (Object.keys(colorSpaceSpecialData).includes(member.enumValueName)) {
          enumBody += `${member.enumValueName}: ${colorSpaceSpecialData[member.enumValueName]},\n`;
        } else {
          enumBody += `${member.enumValueName}: ${member.enumValue},\n`;
        }
      }
    }
  });
  enumBody += '}\n';
  return enumBody;
}