gen-jhipster
Version:
VHipster - Spring Boot + Angular/React/Vue in one handy generator
152 lines (151 loc) • 6.26 kB
JavaScript
/**
* Copyright 2013-2026 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.
*/
import assert from 'node:assert';
import { clientFrameworkTypes, fieldTypes } from "../../../lib/jhipster/index.js";
import { normalizePathEnd } from "../../../lib/utils/path.js";
import { getEntryIfTypeOrTypeAttribute } from "./types-utils.js";
const { STRING: TYPE_STRING, UUID: TYPE_UUID } = fieldTypes.CommonDBTypes;
const { ANGULAR, VUE } = clientFrameworkTypes;
/**
* @private
* Filter the relevant relationships fields on the model
* @param {Array|Object} relationships - array of relationships
* @returns {Array|Object} filtered relationships
*/
export const filterRelevantRelationships = (relationships) => relationships.filter(rel => rel.persistableRelationship || rel.relationshipEagerLoad);
/**
* @private
* Generate Entity Client Imports
*
* @param {Array|Object} relationships - array of relationships
* @param {string} dto - dto
* @param {string} clientFramework the client framework, 'angular', 'vue' or 'react'.
* @returns typeImports: Map
*/
export const generateEntityClientImports = (relationships, _dto, clientFramework) => {
const typeImports = new Map();
const relevantRelationships = filterRelevantRelationships(relationships);
relevantRelationships.forEach(relationship => {
const importType = `I${relationship.otherEntity.entityAngularName}`;
let importPath;
if (relationship.otherEntity?.builtInUser) {
importPath = clientFramework === ANGULAR ? 'app/entities/user/user.model' : 'app/shared/model/user.model';
}
else {
importPath =
clientFramework === ANGULAR
? `app/entities/${normalizePathEnd(relationship.otherEntity.clientRootFolder)}${relationship.otherEntity.entityFileName}.model`
: `app/shared/model/${normalizePathEnd(relationship.otherEntity.clientRootFolder)}${relationship.otherEntity.entityFileName}.model`;
}
typeImports.set(importType, importPath);
});
return typeImports;
};
/**
* @private
* Generate Entity Client Enum Imports
*/
export const generateEntityClientEnumImports = (fields, clientFramework) => {
const typeImports = new Map();
const uniqueEnums = {};
for (const field of fields) {
const { enumFileName, fieldType } = field;
if (field.fieldIsEnum && (!uniqueEnums[fieldType] || (uniqueEnums[fieldType] && field.fieldValues?.length !== 0))) {
const importType = `${fieldType}`;
const basePath = clientFramework === VUE ? '@' : 'app';
const modelPath = clientFramework === ANGULAR ? 'entities' : 'shared/model';
const importPath = `${basePath}/${modelPath}/enumerations/${enumFileName}.model`;
uniqueEnums[fieldType] = field.fieldType;
typeImports.set(importType, importPath);
}
}
return typeImports;
};
/**
* @private
* Generate a primary key, according to the type
*/
export const generateTestEntityId = (primaryKey, index = 0, wrapped = true) => {
if (index === 'random' && typeof primaryKey === 'object') {
return primaryKey.fields[0].generateFakeData('ts');
}
assert(index === 0 || index === 1, 'index must be 0 or 1');
const primaryKeyType = getEntryIfTypeOrTypeAttribute(primaryKey);
let value;
if (primaryKeyType === TYPE_STRING) {
value = index === 0 ? 'ABC' : 'CBA';
}
else if (primaryKeyType === TYPE_UUID) {
value = index === 0 ? '9fec3727-3421-4967-b213-ba36557ca194' : '1361f429-3817-4123-8ee3-fdf8943310b2';
}
else {
value = index === 0 ? 123 : 456;
}
if (wrapped && [TYPE_UUID, TYPE_STRING].includes(primaryKeyType)) {
return `'${value}'`;
}
return value;
};
/**
* Generate a test entity, according to the type
*/
export const generateTsTestEntityForFields = (fields) => {
const entries = fields.flatMap(field => {
const { fieldWithContentType, contentTypeFieldName, fieldTypeTimed, fieldTypeLocalDate } = field;
const fakeData = field.generateFakeData('ts');
if (fieldWithContentType) {
return [
[field.propertyName, fakeData],
[contentTypeFieldName, "'unknown'"],
];
}
if (fieldTypeTimed || fieldTypeLocalDate) {
return [[field.propertyName, `dayjs(${fakeData})`]];
}
return [[field.propertyName, fakeData]];
});
return Object.fromEntries(entries);
};
export const stringifyTsEntity = (data, options = {}) => {
const entries = Object.entries(data);
const { sep = entries.length > 1 ? '\n ' : '' } = options;
return `{${sep}${entries.map(([key, value]) => `${key}: ${value}`).join(`,${sep}`)}${sep.trim()}}`;
};
/**
* @private
* @deprecated
* Generate a test entity, according to the type
*/
export const generateTestEntity = (fields, index = 'random') => {
const entries = fields.flatMap(field => {
if (index === 'random') {
const { fieldWithContentType, contentTypeFieldName } = field;
const fakeData = field.generateFakeData('json-serializable');
if (fieldWithContentType) {
return [
[field.propertyName, fakeData],
[contentTypeFieldName, 'unknown'],
];
}
return [[field.propertyName, fakeData]];
}
return [[field.propertyName, generateTestEntityId(field.fieldType, index, false)]];
});
return Object.fromEntries(entries);
};