@easy-cms/drizzle 0.13.0 → 0.14.0

package/dist/index.d.ts

@@ -70,6 +70,11 @@ interface Dialect {
     firstElementValue(array: AnyColumn | SQL, lists: readonly (readonly string[])[], path: readonly string[], type: JsonValueType): SQL;
     /** SQL that copies the whole database to a file while it is in use (SQLite: `VACUUM INTO`). */
     backupSQL?(file: string): string;
+    /**
+     * SQL that moves a serial column's sequence past its largest value, after rows were inserted
+     * with their ids (Postgres). SQLite needs none: AUTOINCREMENT follows inserted ids.
+     */
+    resetSequenceSQL?(table: string, column: string): string;
     /** Placeholder for the n-th parameter (1-based) in raw SQL. */
     param(n: number): string;
     /** CREATE TABLE IF NOT EXISTS for the migrations table. */

package/dist/index.js

@@ -1,7 +1,7 @@
 // src/database.ts
 import { resolve } from "path";
 import { QueryError as QueryError2 } from "@easy-cms/core";
-import { count, eq as eq3, inArray as inArray3 } from "drizzle-orm";
+import { asc as asc3, count, eq as eq3, getTableColumns, getTableName, inArray as inArray3 } from "drizzle-orm";
 
 // src/documents.ts
 import { asc, eq, inArray } from "drizzle-orm";
@@ -1196,6 +1196,47 @@ var DrizzleDatabase = class {
       await this.connection.runner.query(backupSQL(file));
     });
   }
+  get transfer() {
+    const { tables, hash } = this.schema;
+    const table = (name) => {
+      const found = tables[name];
+      if (!found) throw new QueryError2(`Unknown table "${name}"`);
+      return found;
+    };
+    const key = (t) => {
+      const columns = getTableColumns(t);
+      return columns.id ?? columns.slug;
+    };
+    return {
+      schemaHash: hash,
+      tables: () => Object.keys(tables),
+      count: async (name) => {
+        const [row] = await this.db.select({ n: count() }).from(table(name));
+        return Number(row?.n ?? 0);
+      },
+      read: (name, offset, limit) => {
+        const t = table(name);
+        return this.db.select().from(t).orderBy(asc3(key(t))).limit(limit).offset(offset);
+      },
+      write: async (name, rows) => {
+        const t = table(name);
+        const size = Math.max(1, Math.floor(1e4 / Object.keys(getTableColumns(t)).length));
+        for (let i = 0; i < rows.length; i += size) {
+          const batch = rows.slice(i, i + size);
+          await this.write(() => this.db.insert(t).values(batch));
+        }
+      },
+      finish: async () => {
+        const { resetSequenceSQL } = this.dialect;
+        if (!resetSequenceSQL) return;
+        for (const t of Object.values(tables)) {
+          const id = getTableColumns(t).id;
+          if (id?.dataType === "number")
+            await this.connection.runner.query(resetSequenceSQL(getTableName(t), "id"));
+        }
+      }
+    };
+  }
   async destroy() {
     await this.connection.close();
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@easy-cms/drizzle",
-  "version": "0.13.0",
+  "version": "0.14.0",
   "description": "Shared Drizzle ORM layer for Easy CMS database adapters",
   "license": "MIT",
   "type": "module",
@@ -23,10 +23,10 @@
     "drizzle-orm": "0.45.3"
   },
   "peerDependencies": {
-    "@easy-cms/core": "^0.13.0"
+    "@easy-cms/core": "^0.14.0"
   },
   "devDependencies": {
-    "@easy-cms/core": "^0.13.0"
+    "@easy-cms/core": "^0.14.0"
   },
   "publishConfig": {
     "access": "public"