@type32/tauri-sqlite-orm 0.1.8 → 0.1.10

package/dist/index.d.mts

@@ -188,6 +188,12 @@ declare const notExists: (subquery: SQL | {
 declare const asc: (column: Column) => string;
 declare const desc: (column: Column) => string;
 
+type InferInsert<T> = T extends {
+    $inferInsert: infer I;
+} ? I : never;
+type InferSelect<T> = T extends {
+    $inferSelect: infer S;
+} ? S : never;
 declare class SelectQueryBuilder<T> {
     private _table;
     private _selectedColumns;
@@ -222,11 +228,17 @@ declare class TauriORM {
     constructor(dbUri: string);
     private getDb;
     configureQuery(tables: Record<string, Table<any>>, relations: Record<string, Record<string, RelationConfig>>): void;
-    select<T>(fields?: Record<string, Column<any>>): SelectQueryBuilder<T>;
-    selectDistinct<T>(fields?: Record<string, Column<any>>): SelectQueryBuilder<T>;
-    insert(table: Table<any>): {
-        _table: any;
-        _rows: Record<string, any>[];
+    select<TFields extends Record<string, Column<any>>>(fields: TFields): SelectQueryBuilder<{
+        [K in keyof TFields]: TFields[K]["_dataType"];
+    }>;
+    select<T = any>(fields?: undefined): SelectQueryBuilder<T>;
+    selectDistinct<TFields extends Record<string, Column<any>>>(fields: TFields): SelectQueryBuilder<{
+        [K in keyof TFields]: TFields[K]["_dataType"];
+    }>;
+    selectDistinct<T = any>(fields?: undefined): SelectQueryBuilder<T>;
+    insert<TTable extends Table<any>>(table: TTable): {
+        _table: TTable;
+        _rows: Array<InferInsert<TTable>>;
         _selectSql: {
             clause: string;
             bindings: any[];
@@ -243,7 +255,7 @@ declare class TauriORM {
             setWhere?: SQL;
         };
         _returning: null | "__RETURNING_ID__" | Record<string, Column<any>>;
-        values(rowOrRows: Record<string, any> | Record<string, any>[]): /*elided*/ any;
+        values(rowOrRows: InferInsert<TTable> | Array<InferInsert<TTable>>): /*elided*/ any;
         select(qb: {
             toSQL?: () => {
                 clause: string;
@@ -266,15 +278,15 @@ declare class TauriORM {
         _buildConflictClause(): string;
         _executeWithReturning(db: any, query: string, bindings: any[]): Promise<any>;
     };
-    update(table: Table<any>): {
-        _table: any;
-        _data: Record<string, any> | null;
+    update<TTable extends Table<any>>(table: TTable): {
+        _table: TTable;
+        _data: Partial<InferInsert<TTable>> | null;
         _where: Record<string, any> | SQL | null;
         _orderBy: Array<string | SQL>;
         _limit: number | null;
         _from: Table<any> | null;
         _returning: null | Record<string, Column<any>>;
-        set(data: Record<string, any>): /*elided*/ any;
+        set(data: Partial<InferInsert<TTable>>): /*elided*/ any;
         where(cond: Record<string, any> | SQL): /*elided*/ any;
         orderBy(...clauses: (string | Column<any> | SQL)[]): /*elided*/ any;
         limit(n: number): /*elided*/ any;
@@ -282,13 +294,13 @@ declare class TauriORM {
         returning(fields?: Record<string, Column<any>>): any;
         execute(): Promise<unknown>;
     };
-    delete(table: Table<any>): {
-        _table: any;
-        _where: Record<string, any> | SQL | null;
+    delete<TTable extends Table<any>>(table: TTable): {
+        _table: TTable;
+        _where: Partial<InferInsert<TTable>> | SQL | null;
         _orderBy: Array<string | SQL>;
         _limit: number | null;
         _returning: null | Record<string, Column<any>>;
-        where(cond: Record<string, any> | SQL): /*elided*/ any;
+        where(cond: Partial<InferInsert<TTable>> | SQL): /*elided*/ any;
         orderBy(...clauses: (string | Column<any> | SQL)[]): /*elided*/ any;
         limit(n: number): /*elided*/ any;
         returning(fields?: Record<string, Column<any>>): any;
@@ -299,6 +311,8 @@ declare class TauriORM {
     private generateCreateTableSql;
     createTableIfNotExists(table: Table<any>): Promise<void>;
     createTablesIfNotExist(tables: Table<any>[]): Promise<void>;
+    private generateCreateIndexSqls;
+    private createIndexesForTable;
     private ensureMigrationsTable;
     private hasMigration;
     private recordMigration;
@@ -306,7 +320,14 @@ declare class TauriORM {
         name?: string;
         track?: boolean;
     }): Promise<void>;
-    configure(tables: Record<string, Table<any>>, relDefs?: Record<string, Record<string, RelationConfig>>): this;
+    configure<TTables extends Record<string, Table<any>>, TRelDefs extends Record<string, Record<string, RelationConfig>> = Record<string, Record<string, RelationConfig>>>(tables: TTables, relDefs?: TRelDefs): this & {
+        query: {
+            [K in keyof TTables]: {
+                findMany: (opts?: any) => Promise<Array<InferSelect<TTables[K]>>>;
+                findFirst: (opts?: any) => Promise<InferSelect<TTables[K]> | null>;
+            };
+        };
+    };
     migrateConfigured(options?: {
         name?: string;
         track?: boolean;
@@ -349,6 +370,8 @@ declare class TauriORM {
     private setSchemaMeta;
     private normalizeColumn;
     private computeModelSignature;
+    getSchemaSignature(): string;
+    printSchemaDiff(): Promise<void>;
     isSchemaDirty(): Promise<{
         dirty: boolean;
         current: string;
@@ -365,6 +388,7 @@ declare class TauriORM {
         dropExtraColumns?: boolean;
         preserveData?: boolean;
     }): Promise<void>;
+    private forcePushForTables;
 }
 type OneConfig = {
     fields?: Column[];

package/dist/index.d.ts

@@ -188,6 +188,12 @@ declare const notExists: (subquery: SQL | {
 declare const asc: (column: Column) => string;
 declare const desc: (column: Column) => string;
 
+type InferInsert<T> = T extends {
+    $inferInsert: infer I;
+} ? I : never;
+type InferSelect<T> = T extends {
+    $inferSelect: infer S;
+} ? S : never;
 declare class SelectQueryBuilder<T> {
     private _table;
     private _selectedColumns;
@@ -222,11 +228,17 @@ declare class TauriORM {
     constructor(dbUri: string);
     private getDb;
     configureQuery(tables: Record<string, Table<any>>, relations: Record<string, Record<string, RelationConfig>>): void;
-    select<T>(fields?: Record<string, Column<any>>): SelectQueryBuilder<T>;
-    selectDistinct<T>(fields?: Record<string, Column<any>>): SelectQueryBuilder<T>;
-    insert(table: Table<any>): {
-        _table: any;
-        _rows: Record<string, any>[];
+    select<TFields extends Record<string, Column<any>>>(fields: TFields): SelectQueryBuilder<{
+        [K in keyof TFields]: TFields[K]["_dataType"];
+    }>;
+    select<T = any>(fields?: undefined): SelectQueryBuilder<T>;
+    selectDistinct<TFields extends Record<string, Column<any>>>(fields: TFields): SelectQueryBuilder<{
+        [K in keyof TFields]: TFields[K]["_dataType"];
+    }>;
+    selectDistinct<T = any>(fields?: undefined): SelectQueryBuilder<T>;
+    insert<TTable extends Table<any>>(table: TTable): {
+        _table: TTable;
+        _rows: Array<InferInsert<TTable>>;
         _selectSql: {
             clause: string;
             bindings: any[];
@@ -243,7 +255,7 @@ declare class TauriORM {
             setWhere?: SQL;
         };
         _returning: null | "__RETURNING_ID__" | Record<string, Column<any>>;
-        values(rowOrRows: Record<string, any> | Record<string, any>[]): /*elided*/ any;
+        values(rowOrRows: InferInsert<TTable> | Array<InferInsert<TTable>>): /*elided*/ any;
         select(qb: {
             toSQL?: () => {
                 clause: string;
@@ -266,15 +278,15 @@ declare class TauriORM {
         _buildConflictClause(): string;
         _executeWithReturning(db: any, query: string, bindings: any[]): Promise<any>;
     };
-    update(table: Table<any>): {
-        _table: any;
-        _data: Record<string, any> | null;
+    update<TTable extends Table<any>>(table: TTable): {
+        _table: TTable;
+        _data: Partial<InferInsert<TTable>> | null;
         _where: Record<string, any> | SQL | null;
         _orderBy: Array<string | SQL>;
         _limit: number | null;
         _from: Table<any> | null;
         _returning: null | Record<string, Column<any>>;
-        set(data: Record<string, any>): /*elided*/ any;
+        set(data: Partial<InferInsert<TTable>>): /*elided*/ any;
         where(cond: Record<string, any> | SQL): /*elided*/ any;
         orderBy(...clauses: (string | Column<any> | SQL)[]): /*elided*/ any;
         limit(n: number): /*elided*/ any;
@@ -282,13 +294,13 @@ declare class TauriORM {
         returning(fields?: Record<string, Column<any>>): any;
         execute(): Promise<unknown>;
     };
-    delete(table: Table<any>): {
-        _table: any;
-        _where: Record<string, any> | SQL | null;
+    delete<TTable extends Table<any>>(table: TTable): {
+        _table: TTable;
+        _where: Partial<InferInsert<TTable>> | SQL | null;
         _orderBy: Array<string | SQL>;
         _limit: number | null;
         _returning: null | Record<string, Column<any>>;
-        where(cond: Record<string, any> | SQL): /*elided*/ any;
+        where(cond: Partial<InferInsert<TTable>> | SQL): /*elided*/ any;
         orderBy(...clauses: (string | Column<any> | SQL)[]): /*elided*/ any;
         limit(n: number): /*elided*/ any;
         returning(fields?: Record<string, Column<any>>): any;
@@ -299,6 +311,8 @@ declare class TauriORM {
     private generateCreateTableSql;
     createTableIfNotExists(table: Table<any>): Promise<void>;
     createTablesIfNotExist(tables: Table<any>[]): Promise<void>;
+    private generateCreateIndexSqls;
+    private createIndexesForTable;
     private ensureMigrationsTable;
     private hasMigration;
     private recordMigration;
@@ -306,7 +320,14 @@ declare class TauriORM {
         name?: string;
         track?: boolean;
     }): Promise<void>;
-    configure(tables: Record<string, Table<any>>, relDefs?: Record<string, Record<string, RelationConfig>>): this;
+    configure<TTables extends Record<string, Table<any>>, TRelDefs extends Record<string, Record<string, RelationConfig>> = Record<string, Record<string, RelationConfig>>>(tables: TTables, relDefs?: TRelDefs): this & {
+        query: {
+            [K in keyof TTables]: {
+                findMany: (opts?: any) => Promise<Array<InferSelect<TTables[K]>>>;
+                findFirst: (opts?: any) => Promise<InferSelect<TTables[K]> | null>;
+            };
+        };
+    };
     migrateConfigured(options?: {
         name?: string;
         track?: boolean;
@@ -349,6 +370,8 @@ declare class TauriORM {
     private setSchemaMeta;
     private normalizeColumn;
     private computeModelSignature;
+    getSchemaSignature(): string;
+    printSchemaDiff(): Promise<void>;
     isSchemaDirty(): Promise<{
         dirty: boolean;
         current: string;
@@ -365,6 +388,7 @@ declare class TauriORM {
         dropExtraColumns?: boolean;
         preserveData?: boolean;
     }): Promise<void>;
+    private forcePushForTables;
 }
 type OneConfig = {
     fields?: Column[];

package/dist/index.js

@@ -213,12 +213,19 @@ function unique(name) {
   return {
     on: (...cols) => ({
       name,
-      columns: cols.map((c) => c.name)
+      columns: cols.map((c) => c.name),
+      // runtime marker for DDL rendering
+      kind: "unique"
     })
   };
 }
 function primaryKey(opts) {
-  return { name: opts.name, columns: opts.columns.map((c) => c.name) };
+  return {
+    name: opts.name,
+    columns: opts.columns.map((c) => c.name),
+    // runtime marker for DDL rendering
+    kind: "primaryKey"
+  };
 }
 function check(name, expr) {
   return { name, expr };
@@ -647,7 +654,10 @@ var TauriORM = class {
   _relations = null;
   _dbPromise;
   constructor(dbUri) {
-    this._dbPromise = import_plugin_sql.default.load(dbUri);
+    this._dbPromise = import_plugin_sql.default.load(dbUri).then(async (db) => {
+      await db.execute("PRAGMA foreign_keys = ON");
+      return db;
+    });
   }
   async getDb() {
     return this._dbPromise;
@@ -721,19 +731,32 @@ var TauriORM = class {
           return ret;
         }
         for (const data of this._rows) {
-          const finalData = { ...data };
+          let coerceValue2 = function(col, value) {
+            if (col && col.mode === "boolean") {
+              return value ? 1 : 0;
+            }
+            if (value instanceof Date) {
+              if (col && col.mode === "timestamp_ms") return value.getTime();
+              if (col && col.mode === "timestamp")
+                return Math.floor(value.getTime() / 1e3);
+            }
+            return value;
+          };
+          var coerceValue = coerceValue2;
+          const finalData = Object.assign({}, data);
           const schema = this._table._schema;
           for (const [key, col] of Object.entries(schema)) {
             if (finalData[key] === void 0) {
               if (col.defaultFn) {
-                finalData[key] = col.defaultFn();
+                finalData[key] = coerceValue2(col, col.defaultFn());
               } else if (col.onUpdateFn) {
-                finalData[key] = col.onUpdateFn();
+                finalData[key] = coerceValue2(col, col.onUpdateFn());
               }
             }
           }
-          const keys = Object.keys(finalData);
-          const values = Object.values(finalData);
+          const entries = Object.entries(finalData);
+          const keys = entries.map(([k]) => schema[k]?.name ?? k);
+          const values = entries.map(([k, v]) => coerceValue2(schema[k], v));
           const placeholders = values.map(() => "?").join(", ");
           let query = `INSERT INTO ${this._table._tableName} (${keys.join(", ")}) VALUES (${placeholders})`;
           const bindings = [...values];
@@ -831,7 +854,13 @@ var TauriORM = class {
         const dataToSet = { ...this._data };
         for (const [key, col] of Object.entries(schema)) {
           if (!(key in dataToSet) && col.onUpdateFn) {
-            dataToSet[key] = col.onUpdateFn();
+            const v = col.onUpdateFn();
+            if (col.mode === "boolean") dataToSet[key] = v ? 1 : 0;
+            else if (v instanceof Date) {
+              if (col.mode === "timestamp_ms") dataToSet[key] = v.getTime();
+              else if (col.mode === "timestamp") dataToSet[key] = Math.floor(v.getTime() / 1e3);
+              else dataToSet[key] = v;
+            } else dataToSet[key] = v;
           }
         }
         const setParts = [];
@@ -840,11 +869,20 @@ var TauriORM = class {
           if (v === void 0) continue;
           if (v && typeof v === "object" && typeof v.toSQL === "function") {
             const s = v.toSQL();
-            setParts.push(`${k} = ${s.clause}`);
+            const colName = schema[k]?.name ?? k;
+            setParts.push(`${colName} = ${s.clause}`);
             bindings.push(...s.bindings);
           } else {
-            setParts.push(`${k} = ?`);
-            bindings.push(v);
+            const colName = schema[k]?.name ?? k;
+            let val = v;
+            const col = schema[k];
+            if (col && col.mode === "boolean") val = v ? 1 : 0;
+            else if (v instanceof Date) {
+              if (col && col.mode === "timestamp_ms") val = v.getTime();
+              else if (col && col.mode === "timestamp") val = Math.floor(v.getTime() / 1e3);
+            }
+            setParts.push(`${colName} = ?`);
+            bindings.push(val);
           }
         }
         let query = `UPDATE ${this._table._tableName} SET ${setParts.join(", ")}`;
@@ -857,7 +895,7 @@ var TauriORM = class {
           } else {
             const entries = Object.entries(this._where);
             if (entries.length > 0) {
-              query += ` WHERE ${entries.map(([k]) => `${k} = ?`).join(" AND ")}`;
+              query += ` WHERE ${entries.map(([k]) => `${schema[k]?.name ?? k} = ?`).join(" AND ")}`;
               bindings.push(...entries.map(([, v]) => v));
             }
           }
@@ -933,7 +971,8 @@ var TauriORM = class {
           } else {
             const entries = Object.entries(this._where);
             if (entries.length > 0) {
-              query += ` WHERE ${entries.map(([k]) => `${k} = ?`).join(" AND ")}`;
+              const schema = this._table._schema;
+              query += ` WHERE ${entries.map(([k]) => `${schema[k]?.name ?? k} = ?`).join(" AND ")}`;
               bindings.push(...entries.map(([, v]) => v));
             }
           }
@@ -1016,19 +1055,86 @@ var TauriORM = class {
       }
       return def;
     });
-    return `CREATE TABLE IF NOT EXISTS ${tableName} (${columnDefs.join(
-      ", "
-    )});`;
+    const tableConstraints = [];
+    const constraints = table._constraints;
+    if (constraints && constraints.length) {
+      for (const spec of constraints) {
+        if (spec.expr) {
+          const name = spec.name;
+          const expr = spec.expr.raw ?? spec.expr?.raw ?? String(spec.expr);
+          tableConstraints.push(
+            name ? `CONSTRAINT ${name} CHECK (${expr})` : `CHECK (${expr})`
+          );
+          continue;
+        }
+        if (spec.foreignColumns) {
+          const name = spec.name;
+          const cols = spec.columns.join(", ");
+          const fTable = spec.foreignTable;
+          const fCols = spec.foreignColumns.join(", ");
+          let clause = `${name ? `CONSTRAINT ${name} ` : ""}FOREIGN KEY (${cols}) REFERENCES ${fTable} (${fCols})`;
+          if (spec.onDelete)
+            clause += ` ON DELETE ${String(
+              spec.onDelete
+            ).toUpperCase()}`;
+          if (spec.onUpdate)
+            clause += ` ON UPDATE ${String(
+              spec.onUpdate
+            ).toUpperCase()}`;
+          tableConstraints.push(clause);
+          continue;
+        }
+        if (spec.columns) {
+          const cols = spec.columns.join(", ");
+          const name = spec.name;
+          const isPk = spec.kind === "primaryKey" || name && name.toLowerCase().includes("pk");
+          if (isPk) {
+            tableConstraints.push(
+              name ? `CONSTRAINT ${name} PRIMARY KEY (${cols})` : `PRIMARY KEY (${cols})`
+            );
+          } else {
+            tableConstraints.push(
+              name ? `CONSTRAINT ${name} UNIQUE (${cols})` : `UNIQUE (${cols})`
+            );
+          }
+          continue;
+        }
+      }
+    }
+    const parts = [...columnDefs, ...tableConstraints];
+    return `CREATE TABLE IF NOT EXISTS ${tableName} (${parts.join(", ")});`;
   }
   async createTableIfNotExists(table) {
     const sql2 = this.generateCreateTableSql(table);
     await this.run(sql2);
+    await this.createIndexesForTable(table);
   }
   async createTablesIfNotExist(tables) {
     for (const t of tables) {
       await this.createTableIfNotExists(t);
     }
   }
+  generateCreateIndexSqls(table) {
+    const tableName = table._tableName;
+    const indexes = table._indexes || [];
+    const stmts = [];
+    for (const idx of indexes) {
+      const unique2 = idx.unique ? "UNIQUE " : "";
+      if (!idx.name) continue;
+      const colList = Array.isArray(idx.columns) ? idx.columns : [];
+      if (colList.length === 0) continue;
+      const cols = `(${colList.join(", ")})`;
+      const where = idx.where?.raw ? ` WHERE ${idx.where.raw}` : "";
+      stmts.push(
+        `CREATE ${unique2}INDEX IF NOT EXISTS ${idx.name} ON ${tableName} ${cols}${where};`
+      );
+    }
+    return stmts;
+  }
+  async createIndexesForTable(table) {
+    const stmts = this.generateCreateIndexSqls(table);
+    for (const s of stmts) await this.run(s);
+  }
   async ensureMigrationsTable() {
     await this.run(
       `CREATE TABLE IF NOT EXISTS _migrations (name TEXT PRIMARY KEY, applied_at INTEGER NOT NULL)`
@@ -1053,20 +1159,23 @@ var TauriORM = class {
     const track = options?.track ?? true;
     if (track) {
       await this.ensureMigrationsTable();
+    }
+    await this.forcePushForTables(tables, { preserveData: true });
+    if (track) {
       const name = options?.name ?? `init:${tables.map((t) => t._tableName).join(",")}`;
       const already = await this.hasMigration(name);
-      if (already) return;
-      await this.createTablesIfNotExist(tables);
-      await this.recordMigration(name);
-      return;
+      if (!already) await this.recordMigration(name);
     }
-    await this.createTablesIfNotExist(tables);
   }
   // Configure schema and relations, and generate db.query automatically
   configure(tables, relDefs) {
     this._tables = tables;
     this._relations = relDefs ?? {};
-    this.query = makeQueryAPI(tables, this._relations, this.getDb.bind(this));
+    this.query = makeQueryAPI(
+      tables,
+      this._relations ?? {},
+      this.getDb.bind(this)
+    );
     return this;
   }
   // Convenience: migrate from configured tables
@@ -1074,6 +1183,7 @@ var TauriORM = class {
     if (!this._tables)
       throw new Error("No tables configured. Call db.configure({...}) first.");
     await this.migrate(Object.values(this._tables), options);
+    await this.setSchemaMeta("schema_signature", this.computeModelSignature());
   }
   // --- Schema diff and CLI-like helpers ---
   async diffSchema() {
@@ -1200,6 +1310,13 @@ var TauriORM = class {
     entries.sort((a, b) => a.table.localeCompare(b.table));
     return JSON.stringify(entries);
   }
+  getSchemaSignature() {
+    return this.computeModelSignature();
+  }
+  async printSchemaDiff() {
+    const diff = await this.diffSchema();
+    console.log("Schema diff:", JSON.stringify(diff, null, 2));
+  }
   async isSchemaDirty() {
     const sig = this.computeModelSignature();
     const stored = await this.getSchemaMeta("schema_signature");
@@ -1209,7 +1326,9 @@ var TauriORM = class {
     const status = await this.isSchemaDirty();
     if (!this._tables) throw new Error("No tables configured.");
     if (status.dirty) {
-      await this.migrate(Object.values(this._tables), options);
+      await this.forcePushForTables(Object.values(this._tables), {
+        preserveData: true
+      });
       await this.setSchemaMeta(
         "schema_signature",
         this.computeModelSignature()
@@ -1250,13 +1369,17 @@ var TauriORM = class {
   // Force push model to DB: add missing tables/columns, rebuild tables if incompatible
   async forcePush(options) {
     if (!this._tables) throw new Error("No tables configured.");
+    await this.forcePushForTables(Object.values(this._tables), options);
+  }
+  async forcePushForTables(tables, options) {
     const dbi = await this.getDb();
     const preserve = options?.preserveData !== false;
-    for (const tbl of Object.values(this._tables)) {
+    for (const tbl of tables) {
       const tableName = tbl._tableName;
       const exists2 = await this.tableExists(tableName);
       if (!exists2) {
         await this.run(this.buildCreateTableSQL(tbl));
+        await this.createIndexesForTable(tbl);
         continue;
       }
       const existingCols = await dbi.select(
@@ -1301,6 +1424,7 @@ var TauriORM = class {
         }
         await this.run(`DROP TABLE ${tableName}`);
         await this.run(`ALTER TABLE ${tmp} RENAME TO ${tableName}`);
+        await this.createIndexesForTable(tbl);
       } else {
         for (const m of missing) {
           let clause = `${m.name} ${m.type}`;
@@ -1311,6 +1435,7 @@ var TauriORM = class {
           }
           await this.run(`ALTER TABLE ${tableName} ADD COLUMN ${clause}`);
         }
+        await this.createIndexesForTable(tbl);
       }
     }
     await this.setSchemaMeta("schema_signature", this.computeModelSignature());

package/dist/index.mjs

@@ -135,12 +135,19 @@ function unique(name) {
   return {
     on: (...cols) => ({
       name,
-      columns: cols.map((c) => c.name)
+      columns: cols.map((c) => c.name),
+      // runtime marker for DDL rendering
+      kind: "unique"
     })
   };
 }
 function primaryKey(opts) {
-  return { name: opts.name, columns: opts.columns.map((c) => c.name) };
+  return {
+    name: opts.name,
+    columns: opts.columns.map((c) => c.name),
+    // runtime marker for DDL rendering
+    kind: "primaryKey"
+  };
 }
 function check(name, expr) {
   return { name, expr };
@@ -569,7 +576,10 @@ var TauriORM = class {
   _relations = null;
   _dbPromise;
   constructor(dbUri) {
-    this._dbPromise = Database.load(dbUri);
+    this._dbPromise = Database.load(dbUri).then(async (db) => {
+      await db.execute("PRAGMA foreign_keys = ON");
+      return db;
+    });
   }
   async getDb() {
     return this._dbPromise;
@@ -643,19 +653,32 @@ var TauriORM = class {
           return ret;
         }
         for (const data of this._rows) {
-          const finalData = { ...data };
+          let coerceValue2 = function(col, value) {
+            if (col && col.mode === "boolean") {
+              return value ? 1 : 0;
+            }
+            if (value instanceof Date) {
+              if (col && col.mode === "timestamp_ms") return value.getTime();
+              if (col && col.mode === "timestamp")
+                return Math.floor(value.getTime() / 1e3);
+            }
+            return value;
+          };
+          var coerceValue = coerceValue2;
+          const finalData = Object.assign({}, data);
           const schema = this._table._schema;
           for (const [key, col] of Object.entries(schema)) {
             if (finalData[key] === void 0) {
               if (col.defaultFn) {
-                finalData[key] = col.defaultFn();
+                finalData[key] = coerceValue2(col, col.defaultFn());
               } else if (col.onUpdateFn) {
-                finalData[key] = col.onUpdateFn();
+                finalData[key] = coerceValue2(col, col.onUpdateFn());
               }
             }
           }
-          const keys = Object.keys(finalData);
-          const values = Object.values(finalData);
+          const entries = Object.entries(finalData);
+          const keys = entries.map(([k]) => schema[k]?.name ?? k);
+          const values = entries.map(([k, v]) => coerceValue2(schema[k], v));
           const placeholders = values.map(() => "?").join(", ");
           let query = `INSERT INTO ${this._table._tableName} (${keys.join(", ")}) VALUES (${placeholders})`;
           const bindings = [...values];
@@ -753,7 +776,13 @@ var TauriORM = class {
         const dataToSet = { ...this._data };
         for (const [key, col] of Object.entries(schema)) {
           if (!(key in dataToSet) && col.onUpdateFn) {
-            dataToSet[key] = col.onUpdateFn();
+            const v = col.onUpdateFn();
+            if (col.mode === "boolean") dataToSet[key] = v ? 1 : 0;
+            else if (v instanceof Date) {
+              if (col.mode === "timestamp_ms") dataToSet[key] = v.getTime();
+              else if (col.mode === "timestamp") dataToSet[key] = Math.floor(v.getTime() / 1e3);
+              else dataToSet[key] = v;
+            } else dataToSet[key] = v;
           }
         }
         const setParts = [];
@@ -762,11 +791,20 @@ var TauriORM = class {
           if (v === void 0) continue;
           if (v && typeof v === "object" && typeof v.toSQL === "function") {
             const s = v.toSQL();
-            setParts.push(`${k} = ${s.clause}`);
+            const colName = schema[k]?.name ?? k;
+            setParts.push(`${colName} = ${s.clause}`);
             bindings.push(...s.bindings);
           } else {
-            setParts.push(`${k} = ?`);
-            bindings.push(v);
+            const colName = schema[k]?.name ?? k;
+            let val = v;
+            const col = schema[k];
+            if (col && col.mode === "boolean") val = v ? 1 : 0;
+            else if (v instanceof Date) {
+              if (col && col.mode === "timestamp_ms") val = v.getTime();
+              else if (col && col.mode === "timestamp") val = Math.floor(v.getTime() / 1e3);
+            }
+            setParts.push(`${colName} = ?`);
+            bindings.push(val);
           }
         }
         let query = `UPDATE ${this._table._tableName} SET ${setParts.join(", ")}`;
@@ -779,7 +817,7 @@ var TauriORM = class {
           } else {
             const entries = Object.entries(this._where);
             if (entries.length > 0) {
-              query += ` WHERE ${entries.map(([k]) => `${k} = ?`).join(" AND ")}`;
+              query += ` WHERE ${entries.map(([k]) => `${schema[k]?.name ?? k} = ?`).join(" AND ")}`;
               bindings.push(...entries.map(([, v]) => v));
             }
           }
@@ -855,7 +893,8 @@ var TauriORM = class {
           } else {
             const entries = Object.entries(this._where);
             if (entries.length > 0) {
-              query += ` WHERE ${entries.map(([k]) => `${k} = ?`).join(" AND ")}`;
+              const schema = this._table._schema;
+              query += ` WHERE ${entries.map(([k]) => `${schema[k]?.name ?? k} = ?`).join(" AND ")}`;
               bindings.push(...entries.map(([, v]) => v));
             }
           }
@@ -938,19 +977,86 @@ var TauriORM = class {
       }
       return def;
     });
-    return `CREATE TABLE IF NOT EXISTS ${tableName} (${columnDefs.join(
-      ", "
-    )});`;
+    const tableConstraints = [];
+    const constraints = table._constraints;
+    if (constraints && constraints.length) {
+      for (const spec of constraints) {
+        if (spec.expr) {
+          const name = spec.name;
+          const expr = spec.expr.raw ?? spec.expr?.raw ?? String(spec.expr);
+          tableConstraints.push(
+            name ? `CONSTRAINT ${name} CHECK (${expr})` : `CHECK (${expr})`
+          );
+          continue;
+        }
+        if (spec.foreignColumns) {
+          const name = spec.name;
+          const cols = spec.columns.join(", ");
+          const fTable = spec.foreignTable;
+          const fCols = spec.foreignColumns.join(", ");
+          let clause = `${name ? `CONSTRAINT ${name} ` : ""}FOREIGN KEY (${cols}) REFERENCES ${fTable} (${fCols})`;
+          if (spec.onDelete)
+            clause += ` ON DELETE ${String(
+              spec.onDelete
+            ).toUpperCase()}`;
+          if (spec.onUpdate)
+            clause += ` ON UPDATE ${String(
+              spec.onUpdate
+            ).toUpperCase()}`;
+          tableConstraints.push(clause);
+          continue;
+        }
+        if (spec.columns) {
+          const cols = spec.columns.join(", ");
+          const name = spec.name;
+          const isPk = spec.kind === "primaryKey" || name && name.toLowerCase().includes("pk");
+          if (isPk) {
+            tableConstraints.push(
+              name ? `CONSTRAINT ${name} PRIMARY KEY (${cols})` : `PRIMARY KEY (${cols})`
+            );
+          } else {
+            tableConstraints.push(
+              name ? `CONSTRAINT ${name} UNIQUE (${cols})` : `UNIQUE (${cols})`
+            );
+          }
+          continue;
+        }
+      }
+    }
+    const parts = [...columnDefs, ...tableConstraints];
+    return `CREATE TABLE IF NOT EXISTS ${tableName} (${parts.join(", ")});`;
   }
   async createTableIfNotExists(table) {
     const sql2 = this.generateCreateTableSql(table);
     await this.run(sql2);
+    await this.createIndexesForTable(table);
   }
   async createTablesIfNotExist(tables) {
     for (const t of tables) {
       await this.createTableIfNotExists(t);
     }
   }
+  generateCreateIndexSqls(table) {
+    const tableName = table._tableName;
+    const indexes = table._indexes || [];
+    const stmts = [];
+    for (const idx of indexes) {
+      const unique2 = idx.unique ? "UNIQUE " : "";
+      if (!idx.name) continue;
+      const colList = Array.isArray(idx.columns) ? idx.columns : [];
+      if (colList.length === 0) continue;
+      const cols = `(${colList.join(", ")})`;
+      const where = idx.where?.raw ? ` WHERE ${idx.where.raw}` : "";
+      stmts.push(
+        `CREATE ${unique2}INDEX IF NOT EXISTS ${idx.name} ON ${tableName} ${cols}${where};`
+      );
+    }
+    return stmts;
+  }
+  async createIndexesForTable(table) {
+    const stmts = this.generateCreateIndexSqls(table);
+    for (const s of stmts) await this.run(s);
+  }
   async ensureMigrationsTable() {
     await this.run(
       `CREATE TABLE IF NOT EXISTS _migrations (name TEXT PRIMARY KEY, applied_at INTEGER NOT NULL)`
@@ -975,20 +1081,23 @@ var TauriORM = class {
     const track = options?.track ?? true;
     if (track) {
       await this.ensureMigrationsTable();
+    }
+    await this.forcePushForTables(tables, { preserveData: true });
+    if (track) {
       const name = options?.name ?? `init:${tables.map((t) => t._tableName).join(",")}`;
       const already = await this.hasMigration(name);
-      if (already) return;
-      await this.createTablesIfNotExist(tables);
-      await this.recordMigration(name);
-      return;
+      if (!already) await this.recordMigration(name);
     }
-    await this.createTablesIfNotExist(tables);
   }
   // Configure schema and relations, and generate db.query automatically
   configure(tables, relDefs) {
     this._tables = tables;
     this._relations = relDefs ?? {};
-    this.query = makeQueryAPI(tables, this._relations, this.getDb.bind(this));
+    this.query = makeQueryAPI(
+      tables,
+      this._relations ?? {},
+      this.getDb.bind(this)
+    );
     return this;
   }
   // Convenience: migrate from configured tables
@@ -996,6 +1105,7 @@ var TauriORM = class {
     if (!this._tables)
       throw new Error("No tables configured. Call db.configure({...}) first.");
     await this.migrate(Object.values(this._tables), options);
+    await this.setSchemaMeta("schema_signature", this.computeModelSignature());
   }
   // --- Schema diff and CLI-like helpers ---
   async diffSchema() {
@@ -1122,6 +1232,13 @@ var TauriORM = class {
     entries.sort((a, b) => a.table.localeCompare(b.table));
     return JSON.stringify(entries);
   }
+  getSchemaSignature() {
+    return this.computeModelSignature();
+  }
+  async printSchemaDiff() {
+    const diff = await this.diffSchema();
+    console.log("Schema diff:", JSON.stringify(diff, null, 2));
+  }
   async isSchemaDirty() {
     const sig = this.computeModelSignature();
     const stored = await this.getSchemaMeta("schema_signature");
@@ -1131,7 +1248,9 @@ var TauriORM = class {
     const status = await this.isSchemaDirty();
     if (!this._tables) throw new Error("No tables configured.");
     if (status.dirty) {
-      await this.migrate(Object.values(this._tables), options);
+      await this.forcePushForTables(Object.values(this._tables), {
+        preserveData: true
+      });
       await this.setSchemaMeta(
         "schema_signature",
         this.computeModelSignature()
@@ -1172,13 +1291,17 @@ var TauriORM = class {
   // Force push model to DB: add missing tables/columns, rebuild tables if incompatible
   async forcePush(options) {
     if (!this._tables) throw new Error("No tables configured.");
+    await this.forcePushForTables(Object.values(this._tables), options);
+  }
+  async forcePushForTables(tables, options) {
     const dbi = await this.getDb();
     const preserve = options?.preserveData !== false;
-    for (const tbl of Object.values(this._tables)) {
+    for (const tbl of tables) {
       const tableName = tbl._tableName;
       const exists2 = await this.tableExists(tableName);
       if (!exists2) {
         await this.run(this.buildCreateTableSQL(tbl));
+        await this.createIndexesForTable(tbl);
         continue;
       }
       const existingCols = await dbi.select(
@@ -1223,6 +1346,7 @@ var TauriORM = class {
         }
         await this.run(`DROP TABLE ${tableName}`);
         await this.run(`ALTER TABLE ${tmp} RENAME TO ${tableName}`);
+        await this.createIndexesForTable(tbl);
       } else {
         for (const m of missing) {
           let clause = `${m.name} ${m.type}`;
@@ -1233,6 +1357,7 @@ var TauriORM = class {
           }
           await this.run(`ALTER TABLE ${tableName} ADD COLUMN ${clause}`);
         }
+        await this.createIndexesForTable(tbl);
       }
     }
     await this.setSchemaMeta("schema_signature", this.computeModelSignature());

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@type32/tauri-sqlite-orm",
-  "version": "0.1.8",
+  "version": "0.1.10",
   "description": "A Drizzle-like ORM for Tauri v2's SQL JS API plugin.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",