generator-pyhipster/jdl/models/jdl-enum.js

Python (Flask) + Angular/React/Vue in one handy generator

/**
 * Copyright 2013-2022 the original author or authors from the JHipster project.
 *
 * This file is part of the JHipster project, see https://www.jhipster.tech/
 * for more information.
 *
 * 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
 *
 *      https://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.
 */

const merge = require('../utils/object-utils').merge;
const JDLEnumValue = require('./jdl-enum-value');

class JDLEnum {
  constructor(args) {
    const merged = merge(defaults(), args);
    if (!merged.name) {
      throw new Error("The enum's name must be passed to create an enum.");
    }
    this.comment = merged.comment;
    this.name = merged.name;
    this.values = new Map(
      merged.values.map(entry => {
        return [entry.key, new JDLEnumValue(entry.key, entry.value, entry.comment)];
      })
    );
  }

  getValuesAsString() {
    return stringifyValues(this.values).join(',');
  }

  getValueJavadocs() {
    const javadocs = {};
    this.values.forEach(jdlEnumValue => {
      if (jdlEnumValue.comment) {
        javadocs[jdlEnumValue.name] = jdlEnumValue.comment;
      }
    });
    return javadocs;
  }

  toString() {
    let comment = '';
    if (this.comment) {
      comment += `/**\n * ${this.comment}\n */\n`;
    }
    const values = stringifyValues(this.values);
    return `${comment}enum ${this.name} {\n  ${values.join(',\n  ')}\n}`;
  }
}

module.exports = JDLEnum;

function defaults() {
  return {
    values: [],
  };
}

function stringifyValues(jdlEnumValues) {
  const values = [];
  jdlEnumValues.forEach(jdlEnumValue => {
    values.push(jdlEnumValue.toString());
  });
  return values;
}