generator-pyhipster
Version:
Python (Flask) + Angular/React/Vue in one handy generator
77 lines (68 loc) • 2.09 kB
JavaScript
/**
* 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;
}