@shadow-dev/orm 1.0.2 → 1.0.3

package/dist/core/Repository.d.ts

@@ -2,13 +2,14 @@ import { Model } from "./Model";
 export declare class Repository<T extends object> {
     readonly model: Model<T>;
     constructor(model: Model<T>);
-    create(data: T): Promise<void>;
+    create(data: T): Promise<T>;
     find(where?: Partial<T>): Promise<T[]>;
     findOne(where: Partial<T>): Promise<T | null>;
-    update(where: Partial<T>, data: Partial<T>): Promise<void>;
+    update(where: Partial<T>, data: Partial<T>): Promise<T | null>;
     delete(where: Partial<T>): Promise<void>;
     count(where?: Partial<T>): Promise<number>;
     exists(where: Partial<T>): Promise<boolean>;
     private normalizeValue;
     private buildWhereClause;
+    private getPrimaryKeyField;
 }

package/dist/core/Repository.js

@@ -6,12 +6,38 @@ const Database_1 = require("./Database");
 class Repository {
     constructor(model) {
         this.model = model;
+        // Validate PK at construction
+        const pk = this.getPrimaryKeyField();
+        if (!pk) {
+            throw new Error(`Model "${model.name}" has no primary key defined (pk: true)`);
+        }
     }
     async create(data) {
         const keys = Object.keys(data);
-        const sql = `INSERT INTO \`${this.model.name}\` (${keys.join(",")}) VALUES (${keys.map(() => "?").join(",")})`;
+        if (keys.length === 0)
+            throw new Error("create(): empty data");
+        const sql = `INSERT INTO \`${this.model.name}\` (${keys.map(k => `\`${k}\``).join(",")})
+                     VALUES (${keys.map(() => "?").join(",")})`;
         const values = keys.map((key) => this.normalizeValue(data[key]));
-        await (0, Database_1.getPool)().execute(sql, values);
+        const [res] = await (0, Database_1.getPool)().execute(sql, values);
+        const pk = this.getPrimaryKeyField();
+        // If PK exists in data, refetch using it
+        if (pk && data[pk] != null) {
+            const row = await this.findOne({ [pk]: data[pk] });
+            if (row)
+                return row;
+        }
+        // If PK is auto-increment and insertId is present
+        if (pk && res.insertId && res.insertId !== 0) {
+            const row = await this.findOne({ [pk]: res.insertId });
+            if (row)
+                return row;
+        }
+        // Fallback — return data + insertId if available
+        if (res.insertId && res.insertId !== 0) {
+            return { ...data, id: res.insertId };
+        }
+        return data;
     }
     async find(where = {}) {
         const { sql, params } = this.buildWhereClause(where);
@@ -27,12 +53,18 @@ class Repository {
         return results.length > 0 ? results[0] : null;
     }
     async update(where, data) {
+        if (!where || Object.keys(where).length === 0) {
+            throw new Error("update(): missing WHERE");
+        }
         const setKeys = Object.keys(data);
-        const setClause = setKeys.map(k => `${k} = ?`).join(", ");
+        if (setKeys.length === 0)
+            return this.findOne(where);
+        const setClause = setKeys.map(k => `\`${k}\` = ?`).join(", ");
         const setValues = setKeys.map(k => this.normalizeValue(data[k]));
         const { sql: whereClause, params: whereValues } = this.buildWhereClause(where);
         const query = `UPDATE \`${this.model.name}\` SET ${setClause} ${whereClause}`;
         await (0, Database_1.getPool)().execute(query, [...setValues, ...whereValues.map(this.normalizeValue)]);
+        return this.findOne(where);
     }
     async delete(where) {
         const { sql, params } = this.buildWhereClause(where);
@@ -69,5 +101,13 @@ class Repository {
             params: values,
         };
     }
+    getPrimaryKeyField() {
+        for (const key of Object.keys(this.model.schema)) {
+            if (this.model.schema[key]?.pk) {
+                return key;
+            }
+        }
+        return undefined;
+    }
 }
 exports.Repository = Repository;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shadow-dev/orm",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "description": "Lightweight dynamic MySQL ORM designed for ShadowCore and modular apps.",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",