@treeseed/sdk 0.6.15 → 0.6.17

package/dist/db/node-sqlite.js

@@ -0,0 +1,119 @@
+import { mkdirSync } from "node:fs";
+import { dirname, resolve } from "node:path";
+import { DatabaseSync } from "node:sqlite";
+import { drizzle } from "drizzle-orm/sqlite-proxy";
+import { treeseedSchema } from "./schema.js";
+function isDirectoryLike(path) {
+  return !/\.(sqlite|sqlite3|db)$/iu.test(path);
+}
+function resolveTreeseedSqlitePath(input) {
+  const base = input?.trim() || ".treeseed/generated/environments/local/site-data.sqlite";
+  return isDirectoryLike(base) ? resolve(base, "site-data.sqlite") : resolve(base);
+}
+function toD1Result(result, rows = []) {
+  const changes = Number(result?.changes ?? 0);
+  return {
+    success: true,
+    results: rows,
+    meta: {
+      duration: 0,
+      size_after: 0,
+      rows_read: rows.length,
+      rows_written: changes,
+      last_row_id: Number(result?.lastInsertRowid ?? 0),
+      changed_db: changes > 0,
+      changes
+    }
+  };
+}
+class NodeSqliteD1PreparedStatement {
+  constructor(client, query) {
+    this.client = client;
+    this.query = query;
+  }
+  client;
+  query;
+  bindings = [];
+  bind(...values) {
+    this.bindings = values;
+    return this;
+  }
+  async run() {
+    const statement = this.client.prepare(this.query);
+    return toD1Result(statement.run(...this.bindings));
+  }
+  async all() {
+    const statement = this.client.prepare(this.query);
+    const rows = statement.all(...this.bindings);
+    return toD1Result(void 0, rows);
+  }
+  async first() {
+    const statement = this.client.prepare(this.query);
+    return statement.get(...this.bindings) ?? null;
+  }
+  async raw() {
+    const { results } = await this.all();
+    return results.map((entry) => Object.values(entry));
+  }
+}
+class NodeSqliteD1Database {
+  client;
+  path;
+  constructor(path) {
+    this.path = resolveTreeseedSqlitePath(path);
+    mkdirSync(dirname(this.path), { recursive: true });
+    this.client = new DatabaseSync(this.path);
+    this.client.exec("PRAGMA foreign_keys = ON;");
+    this.client.exec("PRAGMA journal_mode = WAL;");
+  }
+  prepare(query) {
+    return new NodeSqliteD1PreparedStatement(this.client, query);
+  }
+  async exec(query) {
+    this.client.exec(query);
+    return toD1Result(void 0);
+  }
+  async batch(statements) {
+    const results = [];
+    this.client.exec("BEGIN");
+    try {
+      for (const statement of statements) {
+        results.push(await statement.run());
+      }
+      this.client.exec("COMMIT");
+      return results;
+    } catch (error) {
+      this.client.exec("ROLLBACK");
+      throw error;
+    }
+  }
+  close() {
+    this.client.close();
+  }
+}
+function createTreeseedNodeSqliteDrizzle(path) {
+  const database = new NodeSqliteD1Database(path);
+  return {
+    d1: database,
+    db: drizzle(async (sql, params, method) => {
+      const statement = database.client.prepare(sql);
+      if (method === "run") {
+        statement.run(...params);
+        return { rows: [] };
+      }
+      if (method === "get") {
+        const row = statement.get(...params);
+        return { rows: row ? [row] : [] };
+      }
+      if (method === "values") {
+        return { rows: statement.all(...params).map((row) => Object.values(row)) };
+      }
+      return { rows: statement.all(...params) };
+    }, { schema: treeseedSchema })
+  };
+}
+export {
+  NodeSqliteD1Database,
+  createTreeseedNodeSqliteDrizzle,
+  resolveTreeseedSqlitePath
+};