diff --git a/src/Engine.ts b/src/Engine.ts index e477e6f..af444af 100644 --- a/src/Engine.ts +++ b/src/Engine.ts @@ -146,6 +146,10 @@ function addImportsAndGenerationOptions( element.Imports.filter((elem, index, self) => { return index === self.indexOf(elem); }); + if (generationOptions.skipSchema) { + element.Schema = undefined; + element.Database = undefined; + } }); return dbModel; } diff --git a/src/IGenerationOptions.ts b/src/IGenerationOptions.ts index 9a825f4..94f8750 100644 --- a/src/IGenerationOptions.ts +++ b/src/IGenerationOptions.ts @@ -21,4 +21,6 @@ export default class IGenerationOptions { public customNamingStrategyPath: string = ""; public relationIds: boolean = false; + + public skipSchema: boolean = false; } diff --git a/src/entity.mst b/src/entity.mst index 97de093..c474516 100644 --- a/src/entity.mst +++ b/src/entity.mst @@ -3,7 +3,7 @@ import {BaseEntity,Column,Entity,Index,JoinColumn,JoinTable,ManyToMany,ManyToOne {{/each}} -@Entity("{{sqlEntityName}}"{{#Schema}},{schema:"{{.}}"{{#if ../Database}}, database:"{{../Database}}"{{/if}} } {{/Schema}}) +@Entity("{{sqlEntityName}}"{{#Schema}} ,{schema:"{{.}}"{{#if ../Database}}, database:"{{../Database}}"{{/if}} } {{/Schema}}) {{#Indexes}}{{^isPrimaryKey}}@Index("{{name}}",[{{#columns}}"{{toPropertyName name}}",{{/columns}}]{{#isUnique}},{unique:true}{{/isUnique}}) {{/isPrimaryKey}}{{/Indexes}}export class {{toEntityName tsEntityName}}{{#IsActiveRecord}} extends BaseEntity{{/IsActiveRecord}} { {{#Columns}} diff --git a/src/index.ts b/src/index.ts index ba1bf2c..85a8758 100644 --- a/src/index.ts +++ b/src/index.ts @@ -153,6 +153,11 @@ function GetUtilParametersByArgs() { default: false, describe: "Generate RelationId fields" }) + .option("skipSchema", { + boolean: true, + default: false, + describe: "Omits schema identifier in generated entities" + }) .option("generateConstructor", { boolean: true, default: false, @@ -194,8 +199,8 @@ function GetUtilParametersByArgs() { generationOptions.noConfigs = argv.noConfig; generationOptions.propertyVisibility = argv.pv; generationOptions.relationIds = argv.relationIds; + generationOptions.skipSchema = argv.skipSchema; generationOptions.resultsPath = argv.o ? argv.o.toString() : null; - return { driver, connectionOptions, generationOptions }; } @@ -363,6 +368,10 @@ async function GetUtilParametersByInquirer() { name: "Generate RelationId fields", value: "relationId" }, + { + name: "Omits schema identifier in generated entities", + value: "skipSchema" + }, { name: "Generate constructor allowing partial initialization", @@ -396,6 +405,7 @@ async function GetUtilParametersByInquirer() { "activeRecord" ); generationOptions.relationIds = customizations.includes("relationId"); + generationOptions.skipSchema = customizations.includes("skipSchema"); generationOptions.generateConstructor = customizations.includes( "constructor" ); diff --git a/src/models/EntityInfo.ts b/src/models/EntityInfo.ts index cdcd06c..b1cf4a7 100644 --- a/src/models/EntityInfo.ts +++ b/src/models/EntityInfo.ts @@ -14,13 +14,13 @@ export default class EntityInfo { public Indexes: IndexInfo[]; - public Schema: string; + public Schema?: string; public GenerateConstructor: boolean; public IsActiveRecord: boolean; - public Database: string; + public Database?: string; public relationImports() { const imports: string[] = []; diff --git a/test/utils/GeneralTestUtils.ts b/test/utils/GeneralTestUtils.ts index b49767b..c3981d9 100644 --- a/test/utils/GeneralTestUtils.ts +++ b/test/utils/GeneralTestUtils.ts @@ -23,6 +23,7 @@ export function getGenerationOptions(resultsPath: string): IGenerationOptions { generateConstructor: false, customNamingStrategyPath: "", relationIds: false, + skipSchema: false, activeRecord: false }; }