UNPKG

generator-begcode

Version:

Spring Boot + Angular/React/Vue in one handy generator

114 lines (113 loc) 3.68 kB
import { snakeCase } from 'lodash-es'; import validations from './validations.js'; import JDLEnum from '../models/jdl-enum.js'; import databaseTypes from './database-types.js'; const { Validations: { REQUIRED, UNIQUE, MAX, MAXBYTES, MAXLENGTH, MIN, MINBYTES, MINLENGTH, PATTERN }, } = validations; const { MONGODB, MARIADB, COUCHBASE, NEO4J, CASSANDRA, MSSQL, MYSQL, NO, ORACLE, POSTGRESQL, SQL } = databaseTypes; export const CommonDBTypes = { STRING: 'String', INTEGER: 'Integer', LONG: 'Long', BIG_DECIMAL: 'BigDecimal', FLOAT: 'Float', DOUBLE: 'Double', UUID: 'UUID', ENUM: 'Enum', BOOLEAN: 'Boolean', LOCAL_DATE: 'LocalDate', ZONED_DATE_TIME: 'ZonedDateTime', BLOB: 'Blob', ANY_BLOB: 'AnyBlob', IMAGE_BLOB: 'ImageBlob', TEXT_BLOB: 'TextBlob', INSTANT: 'Instant', DURATION: 'Duration', BYTES: 'byte[]', BYTE_BUFFER: 'ByteBuffer', }; export const RelationalOnlyDBTypes = { BYTES: 'byte[]', BYTE_BUFFER: 'ByteBuffer', }; export const BlobTypes = { IMAGE: 'image', ANY: 'any', TEXT: 'text', }; const CommonDBValidations = { String: new Set([REQUIRED, UNIQUE, MINLENGTH, MAXLENGTH, PATTERN]), Integer: new Set([REQUIRED, UNIQUE, MIN, MAX]), Long: new Set([REQUIRED, UNIQUE, MIN, MAX]), BigDecimal: new Set([REQUIRED, UNIQUE, MIN, MAX]), Float: new Set([REQUIRED, UNIQUE, MIN, MAX]), Double: new Set([REQUIRED, UNIQUE, MIN, MAX]), Enum: new Set([REQUIRED, UNIQUE]), Boolean: new Set([REQUIRED, UNIQUE]), LocalDate: new Set([REQUIRED, UNIQUE]), ZonedDateTime: new Set([REQUIRED, UNIQUE]), Blob: new Set([REQUIRED, UNIQUE, MINBYTES, MAXBYTES]), AnyBlob: new Set([REQUIRED, UNIQUE, MINBYTES, MAXBYTES]), ImageBlob: new Set([REQUIRED, UNIQUE, MINBYTES, MAXBYTES]), TextBlob: new Set([REQUIRED, UNIQUE]), UUID: new Set([REQUIRED, UNIQUE]), Instant: new Set([REQUIRED, UNIQUE]), Duration: new Set([REQUIRED, UNIQUE]), }; export default { CommonDBTypes, RelationalOnlyDBTypes, isCommonDBType, hasValidation, getIsType, isBlobType, BlobTypes, }; export function isCommonDBType(type) { if (!type) { throw new Error('The passed type must not be nil.'); } return snakeCase(type).toUpperCase() in CommonDBTypes || type instanceof JDLEnum; } export function isBlobType(type) { if (!type) { return false; } return (CommonDBTypes.BLOB === type || CommonDBTypes.ANY_BLOB === type || CommonDBTypes.IMAGE_BLOB === type || CommonDBTypes.TEXT_BLOB === type); } export function hasValidation(type, validation, isAnEnum) { if (!type || !validation) { throw new Error('The passed type and value must not be nil.'); } if (isAnEnum) { type = 'Enum'; } return isCommonDBType(type) && CommonDBValidations[type].has(validation); } export function getIsType(databaseType, callback) { if (!databaseType) { throw new Error('The passed type must not be nil.'); } let isType; switch (databaseType) { case SQL: case MYSQL: case MARIADB: case POSTGRESQL: case ORACLE: case MSSQL: case MONGODB: case COUCHBASE: case CASSANDRA: case NEO4J: isType = isCommonDBType; break; case NO: isType = () => true; break; default: callback && callback(); throw new Error("The passed database type must either be 'sql', 'mysql', 'mariadb', 'postgresql'," + " 'oracle', 'mssql', 'mongodb', 'couchbase', 'neo4j' or 'cassandra'"); } return isType; }