@entity-access/entity-access 1.0.26 → 1.0.28

package/src/migrations/sql-server/SqlServerAutomaticMigrations.ts

@@ -1,4 +1,5 @@
 import { IColumn } from "../../decorators/IColumn.js";
+import { IIndex } from "../../decorators/IIndex.js";
 import { BaseDriver } from "../../drivers/base/BaseDriver.js";
 import { SqlServerLiteral } from "../../drivers/sql-server/SqlServerLiteral.js";
 import EntityType from "../../entity-query/EntityType.js";
@@ -21,27 +22,21 @@ export default class SqlServerAutomaticMigrations extends SqlServerMigrations {
 
         await this.createColumns(driver, type, nonKeyColumns);
 
-        await this.createIndexes(driver, type, nonKeyColumns.filter((x) => x.fkRelation && !x.fkRelation?.dotNotCreateIndex));
+        await this.createIndexes(context, type, nonKeyColumns.filter((x) => x.fkRelation && !x.fkRelation?.dotNotCreateIndex));
 
     }
 
-    async createIndexes(driver: BaseDriver, type: EntityType, fkColumns: IColumn[]) {
-
-        const name = type.schema
-        ? SqlServerLiteral.quotedLiteral(type.schema) + "." + SqlServerLiteral.quotedLiteral(type.name)
-        : SqlServerLiteral.quotedLiteral(type.name);
-
+    async createIndexes(context: EntityContext, type: EntityType, fkColumns: IColumn[]) {
         for (const iterator of fkColumns) {
-            const indexName =  SqlServerLiteral.quotedLiteral(`IX_${type.name}_${iterator.columnName}`);
-            const columnName = SqlServerLiteral.quotedLiteral(iterator.columnName);
-            let query = `IF NOT EXISTS(SELECT * FROM sys.indexes WHERE name = '${indexName}' AND object_id = OBJECT_ID('${name}'))
-                BEGIN   
-                    CREATE INDEX ${indexName} ON ${name} ( ${columnName})`;
-            if (iterator.nullable !== true) {
-                query += ` WHERE (${columnName} is not null)`;
-            }
-            query += `END`;
-            await driver.executeQuery(query);
+            const filter = iterator.nullable
+                ? `${ SqlServerLiteral.quotedLiteral(iterator.columnName)} IS NOT NULL`
+                : "";
+            const indexDef: IIndex = {
+                name: `IX_${type.name}_${iterator.columnName}`,
+                columns: [{ name: iterator.columnName, descending: iterator.indexOrder !== "ascending"}],
+                filter
+            };
+            await this.migrateIndex(context, indexDef, type);
         }
     }
 
@@ -101,5 +96,26 @@ export default class SqlServerAutomaticMigrations extends SqlServerMigrations {
 
     }
 
+    async migrateIndex(context: EntityContext, index: IIndex, type: EntityType) {
+        const driver = context.driver;
+        const name = type.schema
+            ? SqlServerLiteral.quotedLiteral(type.schema) + "." + SqlServerLiteral.quotedLiteral(type.name)
+            : SqlServerLiteral.quotedLiteral(type.name);
+        const indexName =  SqlServerLiteral.quotedLiteral(index.name);
+        const columns = [];
+        for (const column of index.columns) {
+            const columnName = SqlServerLiteral.quotedLiteral(column.name);
+            columns.push(`${columnName} ${column.descending ? "DESC" : "ASC"}`);
+        }
+        let query = `IF NOT EXISTS(SELECT * FROM sys.indexes WHERE name = '${indexName}' AND object_id = OBJECT_ID('${name}'))
+        BEGIN   
+            CREATE ${index.unique ? "UNIQUE" : ""} INDEX ${indexName} ON ${name} ( ${columns.join(", ")})`;
+        if (index.filter) {
+            query += ` WHERE (${index.filter})`;
+        }
+        query += `\nEND`;
+        await driver.executeQuery(query);
+    }
+
 
 }

package/src/model/changes/ChangeEntry.ts

@@ -49,12 +49,36 @@ export default class ChangeEntry implements IChanges {
             return;
         }
 
-        const { type: { columns }, entity, original } = this;
+        const { type: { columns, keys }, entity } = this;
+        let { original } = this;
 
         if (original === void 0) {
-            this.status = "inserted";
-            this.detectDependencies();
-            return;
+
+            // check if all keys are set or not...
+
+            let keysSet = true;
+            let autoGenerate = false;
+            for (const iterator of keys) {
+                autoGenerate ||= iterator.autoGenerate;
+                if(entity[iterator.name] === void 0) {
+                    keysSet = false;
+                }
+            }
+
+            if (keysSet) {
+                if (!autoGenerate) {
+                    this.status = "inserted";
+                    this.detectDependencies();
+                    return;
+                }
+            } else if (autoGenerate) {
+                this.status = "inserted";
+                this.detectDependencies();
+                return;
+            }
+
+            this.original = { ... entity };
+            original = this.original;
         }
 
         this.detectDependencies();
@@ -133,6 +157,11 @@ export default class ChangeEntry implements IChanges {
             // if related has key defined.. set it...
             const rKey = iterator.relatedEntity.keys[0];
 
+            // lets set the prototype...
+            const prototype = iterator.relatedTypeClass.prototype;
+            if (Object.getPrototypeOf(related) !== prototype) {
+                Object.setPrototypeOf(related, prototype);
+            }
             const relatedChanges = this.changeSet.getEntry(related);
 
             const keyValue = related[rKey.name];

package/src/model/changes/ChangeSet.ts

@@ -1,3 +1,4 @@
+import EntityAccessError from "../../common/EntityAccessError.js";
 import SchemaRegistry from "../../decorators/SchemaRegistry.js";
 import EntityContext from "../EntityContext.js";
 import IdentityService, { identityMapSymbol } from "../identity/IdentityService.js";
@@ -47,7 +48,11 @@ export default class ChangeSet {
         if (entry) {
             return entry;
         }
-        const type = SchemaRegistry.model(Object.getPrototypeOf(entity).constructor);
+        const c = Object.getPrototypeOf(entity).constructor;
+        if (c === Object) {
+            throw new EntityAccessError("Entity type not set");
+        }
+        const type = SchemaRegistry.model(c);
         const jsonKey = IdentityService.getIdentity(entity);
         if (jsonKey) {
             const existing = this.identityMap.get(jsonKey);

package/src/tests/model/ShoppingContext.ts

@@ -2,6 +2,7 @@ import EntityContext from "../../model/EntityContext.js";
 import Column from "../../decorators/Column.js";
 import { RelateTo } from "../../decorators/Relate.js";
 import Table from "../../decorators/Table.js";
+import Index from "../../decorators/Index.js";
 
 export const statusPublished = "published";
 
@@ -24,6 +25,11 @@ export class ShoppingContext extends EntityContext {
 }
 
 @Table("Users")
+@Index({
+    name: "IX_Unique_Users",
+    columns: [(x) => x.userName],
+    unique: true
+})
 export class User {
 
     @Column({ key: true , autoGenerate: true, dataType: "BigInt" })
@@ -32,6 +38,9 @@ export class User {
     @Column({})
     public dateCreated: Date;
 
+    @Column({ dataType: "Char", length: 200 })
+    public userName: string;
+
     public ownedProducts: Product[];
 
     public orders: Order[];

package/src/tests/model/createContext.ts

@@ -17,6 +17,7 @@ export async function createContext(driver: BaseDriver) {
 
     await seed(context);
 
+    driver.connectionString.database = context.driver.connectionString.database;
     return context;
 
 }
@@ -29,9 +30,15 @@ async function seed(context: ShoppingContext) {
     const now = new Date();
 
     // add admin user...
-    context.users.add({ dateCreated: new Date()});
+    context.users.add({
+        userName: "admin",
+        dateCreated: new Date()
+    });
     // add seller
-    const seller = context.users.add({ dateCreated: new Date()});
+    const seller = context.users.add({
+        userName: "self",
+        dateCreated: new Date()
+    });
 
     addHeadPhones(context, now, seller);
 
@@ -45,6 +52,7 @@ async function seed(context: ShoppingContext) {
     const productPrice = product.prices[0];
 
     const user = context.users.add({
+        userName: "customer1",
         dateCreated: new Date(),
         orders: [
             context.orders.add({

package/src/tests/security/tests/place-order.ts

@@ -94,11 +94,11 @@ async function addNewOrder(this: TestConfig, customer: User, userID?) {
 
 async function createUser(config: TestConfig) {
     const context = await createContext(config.driver);
-    config.driver.connectionString.database = context.driver.connectionString.database;
     const user = context.users.add({
+        userName: "customer",
         dateCreated: new Date()
     });
     await context.saveChanges();
-    return user;
+    return { ... user };
 }