Initial commit

node_modules/drizzle-orm/d1/session.cjs

@@ -0,0 +1,206 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var session_exports = {};
+__export(session_exports, {
+  D1PreparedQuery: () => D1PreparedQuery,
+  D1Transaction: () => D1Transaction,
+  SQLiteD1Session: () => SQLiteD1Session
+});
+module.exports = __toCommonJS(session_exports);
+var import_entity = require("../entity.cjs");
+var import_logger = require("../logger.cjs");
+var import_sql = require("../sql/sql.cjs");
+var import_sqlite_core = require("../sqlite-core/index.cjs");
+var import_session = require("../sqlite-core/session.cjs");
+var import_utils = require("../utils.cjs");
+class SQLiteD1Session extends import_session.SQLiteSession {
+  constructor(client, dialect, schema, options = {}) {
+    super(dialect);
+    this.client = client;
+    this.schema = schema;
+    this.options = options;
+    this.logger = options.logger ?? new import_logger.NoopLogger();
+  }
+  static [import_entity.entityKind] = "SQLiteD1Session";
+  logger;
+  prepareQuery(query, fields, executeMethod, isResponseInArrayMode, customResultMapper) {
+    const stmt = this.client.prepare(query.sql);
+    return new D1PreparedQuery(
+      stmt,
+      query,
+      this.logger,
+      fields,
+      executeMethod,
+      isResponseInArrayMode,
+      customResultMapper
+    );
+  }
+  async batch(queries) {
+    const preparedQueries = [];
+    const builtQueries = [];
+    for (const query of queries) {
+      const preparedQuery = query._prepare();
+      const builtQuery = preparedQuery.getQuery();
+      preparedQueries.push(preparedQuery);
+      if (builtQuery.params.length > 0) {
+        builtQueries.push(preparedQuery.stmt.bind(...builtQuery.params));
+      } else {
+        const builtQuery2 = preparedQuery.getQuery();
+        builtQueries.push(
+          this.client.prepare(builtQuery2.sql).bind(...builtQuery2.params)
+        );
+      }
+    }
+    const batchResults = await this.client.batch(builtQueries);
+    return batchResults.map((result, i) => preparedQueries[i].mapResult(result, true));
+  }
+  extractRawAllValueFromBatchResult(result) {
+    return result.results;
+  }
+  extractRawGetValueFromBatchResult(result) {
+    return result.results[0];
+  }
+  extractRawValuesValueFromBatchResult(result) {
+    return d1ToRawMapping(result.results);
+  }
+  async transaction(transaction, config) {
+    const tx = new D1Transaction("async", this.dialect, this, this.schema);
+    await this.run(import_sql.sql.raw(`begin${config?.behavior ? " " + config.behavior : ""}`));
+    try {
+      const result = await transaction(tx);
+      await this.run(import_sql.sql`commit`);
+      return result;
+    } catch (err) {
+      await this.run(import_sql.sql`rollback`);
+      throw err;
+    }
+  }
+}
+class D1Transaction extends import_sqlite_core.SQLiteTransaction {
+  static [import_entity.entityKind] = "D1Transaction";
+  async transaction(transaction) {
+    const savepointName = `sp${this.nestedIndex}`;
+    const tx = new D1Transaction("async", this.dialect, this.session, this.schema, this.nestedIndex + 1);
+    await this.session.run(import_sql.sql.raw(`savepoint ${savepointName}`));
+    try {
+      const result = await transaction(tx);
+      await this.session.run(import_sql.sql.raw(`release savepoint ${savepointName}`));
+      return result;
+    } catch (err) {
+      await this.session.run(import_sql.sql.raw(`rollback to savepoint ${savepointName}`));
+      throw err;
+    }
+  }
+}
+function d1ToRawMapping(results) {
+  const rows = [];
+  for (const row of results) {
+    const entry = Object.keys(row).map((k) => row[k]);
+    rows.push(entry);
+  }
+  return rows;
+}
+class D1PreparedQuery extends import_session.SQLitePreparedQuery {
+  constructor(stmt, query, logger, fields, executeMethod, _isResponseInArrayMode, customResultMapper) {
+    super("async", executeMethod, query);
+    this.logger = logger;
+    this._isResponseInArrayMode = _isResponseInArrayMode;
+    this.customResultMapper = customResultMapper;
+    this.fields = fields;
+    this.stmt = stmt;
+  }
+  static [import_entity.entityKind] = "D1PreparedQuery";
+  /** @internal */
+  customResultMapper;
+  /** @internal */
+  fields;
+  /** @internal */
+  stmt;
+  run(placeholderValues) {
+    const params = (0, import_sql.fillPlaceholders)(this.query.params, placeholderValues ?? {});
+    this.logger.logQuery(this.query.sql, params);
+    return this.stmt.bind(...params).run();
+  }
+  async all(placeholderValues) {
+    const { fields, query, logger, stmt, customResultMapper } = this;
+    if (!fields && !customResultMapper) {
+      const params = (0, import_sql.fillPlaceholders)(query.params, placeholderValues ?? {});
+      logger.logQuery(query.sql, params);
+      return stmt.bind(...params).all().then(({ results }) => this.mapAllResult(results));
+    }
+    const rows = await this.values(placeholderValues);
+    return this.mapAllResult(rows);
+  }
+  mapAllResult(rows, isFromBatch) {
+    if (isFromBatch) {
+      rows = d1ToRawMapping(rows.results);
+    }
+    if (!this.fields && !this.customResultMapper) {
+      return rows;
+    }
+    if (this.customResultMapper) {
+      return this.customResultMapper(rows);
+    }
+    return rows.map((row) => (0, import_utils.mapResultRow)(this.fields, row, this.joinsNotNullableMap));
+  }
+  async get(placeholderValues) {
+    const { fields, joinsNotNullableMap, query, logger, stmt, customResultMapper } = this;
+    if (!fields && !customResultMapper) {
+      const params = (0, import_sql.fillPlaceholders)(query.params, placeholderValues ?? {});
+      logger.logQuery(query.sql, params);
+      return stmt.bind(...params).all().then(({ results }) => results[0]);
+    }
+    const rows = await this.values(placeholderValues);
+    if (!rows[0]) {
+      return void 0;
+    }
+    if (customResultMapper) {
+      return customResultMapper(rows);
+    }
+    return (0, import_utils.mapResultRow)(fields, rows[0], joinsNotNullableMap);
+  }
+  mapGetResult(result, isFromBatch) {
+    if (isFromBatch) {
+      result = d1ToRawMapping(result.results)[0];
+    }
+    if (!this.fields && !this.customResultMapper) {
+      return result;
+    }
+    if (this.customResultMapper) {
+      return this.customResultMapper([result]);
+    }
+    return (0, import_utils.mapResultRow)(this.fields, result, this.joinsNotNullableMap);
+  }
+  values(placeholderValues) {
+    const params = (0, import_sql.fillPlaceholders)(this.query.params, placeholderValues ?? {});
+    this.logger.logQuery(this.query.sql, params);
+    return this.stmt.bind(...params).raw();
+  }
+  /** @internal */
+  isResponseInArrayMode() {
+    return this._isResponseInArrayMode;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  D1PreparedQuery,
+  D1Transaction,
+  SQLiteD1Session
+});
+//# sourceMappingURL=session.cjs.map