drizzle-orm 0.43.1-d1468f0 → 0.43.1-e3d1c4f

package/extensions/s3-file/singlestore/column.js

@@ -0,0 +1,97 @@
+import { entityKind } from "../../../entity.js";
+import { extensionColumnConfig, requiredExtension } from "../../../extension-core/index.js";
+import { SingleStoreColumn, SingleStoreColumnBuilder } from "../../../singlestore-core/index.js";
+import { getColumnNameAndConfig } from "../../../utils.js";
+import {
+  objectIdToText
+} from "../common.js";
+import { DrizzleSingleStoreS3Extension } from "./extension.js";
+class SingleStoreS3FileBuilder extends SingleStoreColumnBuilder {
+  static [entityKind] = "SingleStoreS3FileBuilder";
+  [extensionColumnConfig];
+  constructor(name, config) {
+    super(name, "custom", "SingleStoreS3File");
+    this[extensionColumnConfig] = {
+      fileMode: config.mode,
+      fetchMode: "file"
+    };
+  }
+  /**
+   * Allows to set default to an existing file
+   *
+   * @example
+   * ```
+   * fileColumn: s3File().default(sql`${bucket}:${key}`)
+   * ```
+   *
+   * @returns
+   */
+  default(value) {
+    return super.default(value);
+  }
+  generatedAlwaysAs(as) {
+    return super.generatedAlwaysAs(as);
+  }
+  /** @internal */
+  build(table) {
+    return new SingleStoreS3File(
+      table,
+      this.config,
+      this[extensionColumnConfig]
+    );
+  }
+}
+class SingleStoreS3File extends SingleStoreColumn {
+  static [entityKind] = "SingleStoreS3File";
+  [requiredExtension] = DrizzleSingleStoreS3Extension;
+  [extensionColumnConfig];
+  constructor(table, config, extensionConfig) {
+    super(table, config);
+    this[extensionColumnConfig] = extensionConfig ?? {
+      fileMode: "buffer",
+      fetchMode: "file"
+    };
+  }
+  mapToDriverValue(value) {
+    if (typeof value === "string")
+      return value;
+    return objectIdToText(value);
+  }
+  data() {
+    return new SingleStoreS3File(
+      this.table,
+      this.config,
+      { fileMode: this[extensionColumnConfig].fileMode, fetchMode: "data" }
+    );
+  }
+  // meta(): Omit<SingleStoreS3File<Omit<T, 'data'> & { data: DrizzleS3ObjectMeta }>, 'data' | 'meta' | 'presigned'> {
+  // 	return new SingleStoreS3File(
+  // 		this.table,
+  // 		this.config as ColumnRuntimeConfig<
+  // 			DrizzleS3ObjectMeta,
+  // 			Omit<T, 'data'> & { data: DrizzleS3ObjectMeta }
+  // 		>,
+  // 		{ fileMode: this[extensionColumnConfig].fileMode, fetchMode: 'meta' },
+  // 	);
+  // }
+  presigned(options) {
+    return new SingleStoreS3File(
+      this.table,
+      this.config,
+      { fileMode: this[extensionColumnConfig].fileMode, fetchMode: "presigned", options }
+    );
+  }
+  getSQLType() {
+    return "text";
+  }
+}
+function s3File(a, b) {
+  const { name, config } = getColumnNameAndConfig(a, b);
+  return new SingleStoreS3FileBuilder(name, config);
+}
+export {
+  SingleStoreS3File,
+  SingleStoreS3FileBuilder,
+  s3File
+};
+//# sourceMappingURL=column.js.map

package/extensions/s3-file/singlestore/column.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/extensions/s3-file/singlestore/column.ts"],"sourcesContent":["import type {\n\tColumnBuilderBaseConfig,\n\tColumnBuilderRuntimeConfig,\n\tHasDefault,\n\tHasGenerated,\n\tMakeColumnConfig,\n} from '~/column-builder.ts';\nimport type { ColumnBaseConfig, ColumnRuntimeConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { extensionColumnConfig, requiredExtension } from '~/extension-core/index.ts';\nimport { SingleStoreColumn, SingleStoreColumnBuilder } from '~/singlestore-core/index.ts';\nimport type { AnySingleStoreTable, SingleStoreTable } from '~/singlestore-core/table.ts';\nimport type { SQL } from '~/sql/index.ts';\nimport { type Assume, getColumnNameAndConfig } from '~/utils.ts';\nimport {\n\ttype DrizzleS3FetchMode,\n\ttype DrizzleS3FileMode,\n\ttype DrizzleS3FileModeToData,\n\ttype DrizzleS3ObjectFile,\n\ttype DrizzleS3ObjectIdentification,\n\tobjectIdToText,\n\ttype RequestPresigningArguments,\n} from '../common.ts';\nimport { DrizzleSingleStoreS3Extension } from './extension.ts';\n\nexport type SingleStoreS3FileBuilderInitial<TName extends string, TMode extends DrizzleS3FileMode> =\n\tSingleStoreS3FileBuilder<{\n\t\tname: TName;\n\t\tdataType: 'custom';\n\t\tcolumnType: 'SingleStoreS3File';\n\t\tdata: DrizzleS3ObjectFile<DrizzleS3FileModeToData[TMode]>;\n\t\tenumValues: undefined;\n\t\tdriverParam: string;\n\t}>;\n\nexport class SingleStoreS3FileBuilder<\n\tT extends ColumnBuilderBaseConfig<'custom', 'SingleStoreS3File'>,\n\tTRuntimeConfig extends object = {},\n\tTTypeConfig extends object = {},\n> extends SingleStoreColumnBuilder<T, TRuntimeConfig, TTypeConfig> {\n\tstatic override readonly [entityKind]: string = 'SingleStoreS3FileBuilder';\n\treadonly [extensionColumnConfig]: SingleStoreS3FileExtensionColumnConfig;\n\n\tconstructor(\n\t\tname: T['name'],\n\t\tconfig: SingleStoreS3FileConfig,\n\t) {\n\t\tsuper(name, 'custom', 'SingleStoreS3File');\n\t\tthis[extensionColumnConfig] = {\n\t\t\tfileMode: config.mode,\n\t\t\tfetchMode: 'file',\n\t\t};\n\t}\n\n\t/**\n\t * Allows to set default to an existing file\n\t *\n\t * @example\n\t * ```\n\t * fileColumn: s3File().default(sql`${bucket}:${key}`)\n\t * ```\n\t *\n\t * @returns\n\t */\n\toverride default(value: SQL<unknown>): HasDefault<this> {\n\t\treturn super.default(value);\n\t}\n\n\toverride generatedAlwaysAs(as: SQL<unknown> | (() => SQL)): HasGenerated<this, { type: 'always' }> {\n\t\treturn super.generatedAlwaysAs(as);\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnySingleStoreTable<{ name: TTableName }>,\n\t): SingleStoreS3File<MakeColumnConfig<T, TTableName>> {\n\t\treturn new SingleStoreS3File<MakeColumnConfig<T, TTableName>>(\n\t\t\ttable,\n\t\t\tthis.config as ColumnBuilderRuntimeConfig<any, any>,\n\t\t\tthis[extensionColumnConfig],\n\t\t);\n\t}\n}\n\nexport type SingleStoreS3FileExtensionColumnConfig = {\n\tfetchMode: DrizzleS3FetchMode;\n\tfileMode: DrizzleS3FileMode;\n\toptions?: RequestPresigningArguments;\n};\n\nexport class SingleStoreS3File<\n\tT extends ColumnBaseConfig<'custom', 'SingleStoreS3File'> = ColumnBaseConfig<'custom', 'SingleStoreS3File'>,\n> extends SingleStoreColumn<T, { enumValues: T['enumValues'] }> {\n\tstatic override readonly [entityKind]: string = 'SingleStoreS3File';\n\toverride [requiredExtension] = DrizzleSingleStoreS3Extension;\n\t[extensionColumnConfig]: SingleStoreS3FileExtensionColumnConfig;\n\n\tconstructor(\n\t\ttable: SingleStoreTable,\n\t\tconfig: ColumnBuilderRuntimeConfig<T['data'], T>,\n\t\textensionConfig?: SingleStoreS3FileExtensionColumnConfig,\n\t) {\n\t\tsuper(table, config);\n\n\t\tthis[extensionColumnConfig] = extensionConfig ?? {\n\t\t\tfileMode: 'buffer',\n\t\t\tfetchMode: 'file',\n\t\t};\n\t}\n\n\toverride mapToDriverValue(value: unknown): string {\n\t\tif (typeof value === 'string') return value;\n\n\t\treturn objectIdToText(value as DrizzleS3ObjectIdentification);\n\t}\n\n\tdata(): SingleStoreColumn<Omit<T, 'data'> & { data: Assume<T['data'], DrizzleS3ObjectFile>['data'] }> {\n\t\treturn new SingleStoreS3File(\n\t\t\tthis.table,\n\t\t\tthis.config as ColumnRuntimeConfig<\n\t\t\t\tDrizzleS3ObjectFile['data'],\n\t\t\t\tOmit<T, 'data'> & { data: Assume<T['data'], DrizzleS3ObjectFile>['data'] }\n\t\t\t>,\n\t\t\t{ fileMode: this[extensionColumnConfig].fileMode, fetchMode: 'data' },\n\t\t);\n\t}\n\n\t// meta(): Omit<SingleStoreS3File<Omit<T, 'data'> & { data: DrizzleS3ObjectMeta }>, 'data' | 'meta' | 'presigned'> {\n\t// \treturn new SingleStoreS3File(\n\t// \t\tthis.table,\n\t// \t\tthis.config as ColumnRuntimeConfig<\n\t// \t\t\tDrizzleS3ObjectMeta,\n\t// \t\t\tOmit<T, 'data'> & { data: DrizzleS3ObjectMeta }\n\t// \t\t>,\n\t// \t\t{ fileMode: this[extensionColumnConfig].fileMode, fetchMode: 'meta' },\n\t// \t);\n\t// }\n\n\tpresigned(options?: RequestPresigningArguments): SingleStoreColumn<Omit<T, 'data'> & { data: string }> {\n\t\treturn new SingleStoreS3File(\n\t\t\tthis.table,\n\t\t\tthis.config as ColumnRuntimeConfig<\n\t\t\t\tstring,\n\t\t\t\tOmit<T, 'data'> & { data: string }\n\t\t\t>,\n\t\t\t{ fileMode: this[extensionColumnConfig].fileMode, fetchMode: 'presigned', options },\n\t\t);\n\t}\n\n\tgetSQLType(): string {\n\t\treturn 'text';\n\t}\n}\n\nexport interface SingleStoreS3FileConfig<\n\tTMode extends DrizzleS3FileMode = DrizzleS3FileMode,\n> {\n\tmode: TMode;\n}\n\nexport function s3File<TName extends string, TMode extends DrizzleS3FileMode>(\n\tconfig: SingleStoreS3FileConfig<TMode>,\n): SingleStoreS3FileBuilderInitial<TName, TMode>;\nexport function s3File<TName extends string, TMode extends DrizzleS3FileMode>(\n\tname: TName,\n\tconfig: SingleStoreS3FileConfig<TMode>,\n): SingleStoreS3FileBuilderInitial<TName, TMode>;\nexport function s3File(a: string | SingleStoreS3FileConfig, b?: SingleStoreS3FileConfig): any {\n\tconst { name, config } = getColumnNameAndConfig<SingleStoreS3FileConfig>(a, b);\n\treturn new SingleStoreS3FileBuilder(name, config);\n}\n"],"mappings":"AAQA,SAAS,kBAAkB;AAC3B,SAAS,uBAAuB,yBAAyB;AACzD,SAAS,mBAAmB,gCAAgC;AAG5D,SAAsB,8BAA8B;AACpD;AAAA,EAMC;AAAA,OAEM;AACP,SAAS,qCAAqC;AAYvC,MAAM,iCAIH,yBAAyD;AAAA,EAClE,QAA0B,UAAU,IAAY;AAAA,EAChD,CAAU,qBAAqB;AAAA,EAE/B,YACC,MACA,QACC;AACD,UAAM,MAAM,UAAU,mBAAmB;AACzC,SAAK,qBAAqB,IAAI;AAAA,MAC7B,UAAU,OAAO;AAAA,MACjB,WAAW;AAAA,IACZ;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYS,QAAQ,OAAuC;AACvD,WAAO,MAAM,QAAQ,KAAK;AAAA,EAC3B;AAAA,EAES,kBAAkB,IAAwE;AAClG,WAAO,MAAM,kBAAkB,EAAE;AAAA,EAClC;AAAA;AAAA,EAGS,MACR,OACqD;AACrD,WAAO,IAAI;AAAA,MACV;AAAA,MACA,KAAK;AAAA,MACL,KAAK,qBAAqB;AAAA,IAC3B;AAAA,EACD;AACD;AAQO,MAAM,0BAEH,kBAAsD;AAAA,EAC/D,QAA0B,UAAU,IAAY;AAAA,EAChD,CAAU,iBAAiB,IAAI;AAAA,EAC/B,CAAC,qBAAqB;AAAA,EAEtB,YACC,OACA,QACA,iBACC;AACD,UAAM,OAAO,MAAM;AAEnB,SAAK,qBAAqB,IAAI,mBAAmB;AAAA,MAChD,UAAU;AAAA,MACV,WAAW;AAAA,IACZ;AAAA,EACD;AAAA,EAES,iBAAiB,OAAwB;AACjD,QAAI,OAAO,UAAU;AAAU,aAAO;AAEtC,WAAO,eAAe,KAAsC;AAAA,EAC7D;AAAA,EAEA,OAAsG;AACrG,WAAO,IAAI;AAAA,MACV,KAAK;AAAA,MACL,KAAK;AAAA,MAIL,EAAE,UAAU,KAAK,qBAAqB,EAAE,UAAU,WAAW,OAAO;AAAA,IACrE;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,UAAU,SAA6F;AACtG,WAAO,IAAI;AAAA,MACV,KAAK;AAAA,MACL,KAAK;AAAA,MAIL,EAAE,UAAU,KAAK,qBAAqB,EAAE,UAAU,WAAW,aAAa,QAAQ;AAAA,IACnF;AAAA,EACD;AAAA,EAEA,aAAqB;AACpB,WAAO;AAAA,EACR;AACD;AAeO,SAAS,OAAO,GAAqC,GAAkC;AAC7F,QAAM,EAAE,MAAM,OAAO,IAAI,uBAAgD,GAAG,CAAC;AAC7E,SAAO,IAAI,yBAAyB,MAAM,MAAM;AACjD;","names":[]}

package/extensions/s3-file/singlestore/extension.cjs

@@ -0,0 +1,247 @@
+"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 extension_exports = {};
+__export(extension_exports, {
+  DrizzleSingleStoreS3Extension: () => DrizzleSingleStoreS3Extension,
+  s3FileExt: () => s3FileExt
+});
+module.exports = __toCommonJS(extension_exports);
+var import_client_s3 = require("@aws-sdk/client-s3");
+var import_entity = require("../../../entity.cjs");
+var import_errors = require("../../../errors.cjs");
+var import_extension_core = require("../../../extension-core/index.cjs");
+var import_singlestore = require("../../../extension-core/singlestore/index.cjs");
+var import_relations = require("../../../relations.cjs");
+var import_singlestore_core = require("../../../singlestore-core/index.cjs");
+var import_sql = require("../../../sql/index.cjs");
+var import_utils = require("../../../utils.cjs");
+var import_common = require("../common.cjs");
+class DrizzleSingleStoreS3Extension extends import_singlestore.DrizzleSingleStoreExtension {
+  static [import_entity.entityKind] = "DrizzleSingleStoreS3Extension";
+  static [import_extension_core.extensionName] = "Drizzle SingleStore S3 Extension";
+  s3;
+  constructor(client) {
+    super();
+    this.s3 = typeof client === "object" && (client instanceof import_client_s3.S3Client || client.constructor.name !== "Object") ? client : new import_client_s3.S3Client(client);
+  }
+  buildOutputMappingInstructions(selection) {
+    const paths = [];
+    for (const f of selection) {
+      if (!(0, import_entity.is)(f.field, import_singlestore_core.SingleStoreColumn) || !f.field[import_extension_core.requiredExtension] || !(0, import_entity.is)(this, f.field[import_extension_core.requiredExtension]))
+        continue;
+      const column = f.field;
+      const { fetchMode, fileMode, options } = column[import_extension_core.extensionColumnConfig];
+      paths.push({
+        path: f.path.map((p) => `[${JSON.stringify(p)}]`),
+        fetchMode,
+        fileMode,
+        options
+      });
+    }
+    return paths;
+  }
+  buildRqbOutputMappingInstructions(schema, table, selection, parent, prefix = []) {
+    const p = parent ?? {
+      path: [],
+      subinstructions: []
+    };
+    for (const s of selection) {
+      if ((0, import_entity.is)(s.field, import_singlestore_core.SingleStoreColumn)) {
+        if (!s.field[import_extension_core.requiredExtension] || !(0, import_entity.is)(this, s.field[import_extension_core.requiredExtension]))
+          continue;
+        const extCol = s.field;
+        (p.subinstructions ??= []).push({
+          path: [...prefix, `[${JSON.stringify(s.tsKey)}]`],
+          fileMode: extCol[import_extension_core.extensionColumnConfig].fileMode,
+          fetchMode: extCol[import_extension_core.extensionColumnConfig].fetchMode,
+          options: extCol[import_extension_core.extensionColumnConfig].options
+        });
+        continue;
+      }
+      if (s.selection) {
+        const relation = schema[s.relationTableTsKey];
+        if ((0, import_entity.is)(table.relations[s.tsKey], import_relations.One)) {
+          this.buildRqbOutputMappingInstructions(schema, relation, s.selection, p, [
+            ...prefix,
+            `[${JSON.stringify(s.tsKey)}]`
+          ]);
+          continue;
+        }
+        const nestedSubs = this.buildRqbOutputMappingInstructions(schema, relation, s.selection);
+        if (!nestedSubs.length)
+          continue;
+        (p.subinstructions ??= []).push({
+          path: [...prefix, `[${JSON.stringify(s.tsKey)}]`],
+          subinstructions: nestedSubs
+        });
+      }
+    }
+    return p.subinstructions;
+  }
+  buildParamMappingInstructions(params) {
+    const indexes = [];
+    const placeholders = [];
+    let i = 0;
+    for (const item of params) {
+      if (!(0, import_entity.is)(item, import_sql.ExtensionParam) || !(0, import_entity.is)(this, item.extension)) {
+        ++i;
+        continue;
+      }
+      if ((0, import_entity.is)(item.value, import_sql.Placeholder)) {
+        placeholders.push({
+          key: item.value.name,
+          mode: item.encoder[import_extension_core.extensionColumnConfig].fileMode
+        });
+        ++i;
+        continue;
+      }
+      indexes.push({
+        index: i++,
+        mode: item.encoder[import_extension_core.extensionColumnConfig].fileMode
+      });
+    }
+    return {
+      placeholders,
+      indexes
+    };
+  }
+  async hookBefore(context) {
+    if (!context?.params?.length)
+      return;
+    let indexes = [];
+    let placeholders = [];
+    const errors = [];
+    switch (context.query) {
+      case "select":
+      case "delete": {
+        break;
+      }
+      case "update":
+      case "insert": {
+        const instr = this.buildParamMappingInstructions(context.params);
+        indexes = instr.indexes;
+        placeholders = instr.placeholders;
+        break;
+      }
+    }
+    await (0, import_common.mapParams)(this.s3, context.params, indexes, errors);
+    if (errors.length) {
+      throw new import_errors.DrizzleError({
+        message: `Error${errors.length > 1 ? "s" : ""} occured during S3 file upload.
+Log "(e as DrizzleError).cause" for details.`,
+        cause: errors
+      });
+    }
+    if (!context.metadata?.encoder) {
+      context.metadata = context.metadata ?? {};
+      context.metadata.encoder = (0, import_common.buildEncoder)(placeholders);
+    }
+    await context.metadata.encoder(this.s3, context.placeholders, errors, import_common.uploadFile, import_common.objectIdToText);
+    if (errors.length) {
+      throw new import_errors.DrizzleError({
+        message: `Error${errors.length > 1 ? "s" : ""} occured during S3 file upload.
+Log "(e as DrizzleError).cause" for details.`,
+        cause: errors
+      });
+    }
+    return;
+  }
+  async hookAfter(context) {
+    const errors = [];
+    const data = context.data;
+    switch (context.query) {
+      case "select": {
+        if (!data.length)
+          return;
+        if (!context.metadata?.decoder) {
+          const fields = context.fieldsOrdered ?? context.config.fieldsFlat ?? (0, import_utils.orderSelectedFields)(context.config.fields);
+          const instructions = this.buildOutputMappingInstructions(fields);
+          const decoder = (0, import_common.buildDecoder)(instructions);
+          context.metadata = context.metadata ?? {};
+          context.metadata.decoder = decoder;
+        }
+        break;
+      }
+      case "_query": {
+        if (context.mode === "first" && !data || context.mode === "many" && !data.length)
+          return;
+        if (!context.metadata?.decoder) {
+          const instructions = this.buildRqbOutputMappingInstructions(
+            context.tablesConfig,
+            context.tableConfig,
+            context.config.selection
+          );
+          const decoder = (0, import_common.buildDecoder)(instructions, context.mode);
+          context.metadata = context.metadata ?? {};
+          context.metadata.decoder = decoder;
+        }
+        break;
+      }
+      case "update":
+      case "delete":
+      case "insert": {
+        if (!data.length)
+          return;
+        if (!context.metadata?.decoder) {
+          if (!context.config.returning) {
+            const decoder2 = (0, import_common.buildDecoder)([]);
+            context.metadata = context.metadata ?? {};
+            context.metadata.decoder = decoder2;
+            break;
+          }
+          const fields = context.config.returning;
+          const instructions = this.buildOutputMappingInstructions(fields);
+          const decoder = (0, import_common.buildDecoder)(instructions);
+          context.metadata = context.metadata ?? {};
+          context.metadata.decoder = decoder;
+        }
+        break;
+      }
+    }
+    await context.metadata.decoder(
+      this.s3,
+      data,
+      errors,
+      import_common.downloadFile,
+      // downloadFileMeta,
+      import_common.presignDownload,
+      import_common.textToObjectId
+    );
+    if (errors.length) {
+      throw new import_errors.DrizzleError({
+        message: `Error${errors.length > 1 ? "s" : ""} occured during S3 file download.
+Log "(e as DrizzleError).cause" for details.`,
+        cause: errors
+      });
+    }
+    return;
+  }
+  async hook(context) {
+    return context.stage === "before" ? await this.hookBefore(context) : await this.hookAfter(context);
+  }
+}
+function s3FileExt(client) {
+  return new DrizzleSingleStoreS3Extension(client);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  DrizzleSingleStoreS3Extension,
+  s3FileExt
+});
+//# sourceMappingURL=extension.cjs.map

package/extensions/s3-file/singlestore/extension.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/extensions/s3-file/singlestore/extension.ts"],"sourcesContent":["import { S3Client, type S3ClientConfig } from '@aws-sdk/client-s3';\nimport { entityKind, is } from '~/entity.ts';\nimport { DrizzleError } from '~/errors.ts';\nimport { extensionColumnConfig, extensionName, requiredExtension } from '~/extension-core/index.ts';\nimport { DrizzleSingleStoreExtension, type DrizzleSingleStoreHookContext } from '~/extension-core/singlestore/index.ts';\nimport type { SelectedFieldsOrdered } from '~/operations.ts';\nimport {\n\ttype BuildRelationalQueryResult,\n\tOne,\n\ttype TableRelationalConfig,\n\ttype TablesRelationalConfig,\n} from '~/relations.ts';\nimport { SingleStoreColumn } from '~/singlestore-core/index.ts';\nimport { ExtensionParam, Placeholder } from '~/sql/index.ts';\nimport { orderSelectedFields } from '~/utils.ts';\nimport {\n\tbuildDecoder,\n\tbuildEncoder,\n\tdownloadFile,\n\t// downloadFileMeta,\n\tmapParams,\n\tobjectIdToText,\n\tpresignDownload,\n\ttextToObjectId,\n\tuploadFile,\n} from '../common.ts';\nimport type {\n\tDrizzleS3FileInputMappingInstruction,\n\tDrizzleS3FileOutputMappingInstruction,\n\tDrizzleS3FileOutputMappingInstructionIterableNode,\n\tDrizzleS3FilePlaceholderMappingInstruction,\n\tS3ExtMeta,\n} from '../common.ts';\nimport type { SingleStoreS3File } from './column.ts';\n\nexport class DrizzleSingleStoreS3Extension extends DrizzleSingleStoreExtension<S3ExtMeta> {\n\tstatic override readonly [entityKind]: string = 'DrizzleSingleStoreS3Extension';\n\n\tstatic override readonly [extensionName] = 'Drizzle SingleStore S3 Extension';\n\n\tprivate s3: S3Client;\n\n\tconstructor(client: S3Client | S3ClientConfig) {\n\t\tsuper();\n\n\t\t// eslint-disable-next-line no-instanceof/no-instanceof\n\t\tthis.s3 = typeof client === 'object' && (client instanceof S3Client || client.constructor.name !== 'Object')\n\t\t\t? client as S3Client\n\t\t\t: new S3Client(client);\n\t}\n\n\tprivate buildOutputMappingInstructions(selection: SelectedFieldsOrdered<SingleStoreColumn>) {\n\t\tconst paths: DrizzleS3FileOutputMappingInstruction[] = [];\n\n\t\tfor (const f of selection) {\n\t\t\tif (\n\t\t\t\t!is(f.field, SingleStoreColumn) || !(<SingleStoreColumn> f.field)[requiredExtension]\n\t\t\t\t|| !is(this, f.field[requiredExtension]!)\n\t\t\t) continue;\n\n\t\t\tconst column = f.field as SingleStoreS3File;\n\n\t\t\tconst { fetchMode, fileMode, options } = column[extensionColumnConfig];\n\n\t\t\tpaths.push({\n\t\t\t\tpath: f.path.map((p) => `[${JSON.stringify(p)}]`),\n\t\t\t\tfetchMode: fetchMode,\n\t\t\t\tfileMode: fileMode,\n\t\t\t\toptions: options,\n\t\t\t});\n\t\t}\n\n\t\treturn paths;\n\t}\n\n\tprivate buildRqbOutputMappingInstructions(\n\t\tschema: TablesRelationalConfig,\n\t\ttable: TableRelationalConfig,\n\t\tselection: BuildRelationalQueryResult['selection'],\n\t\tparent?: DrizzleS3FileOutputMappingInstructionIterableNode,\n\t\tprefix: string[] = [],\n\t) {\n\t\tconst p: DrizzleS3FileOutputMappingInstructionIterableNode = parent ?? {\n\t\t\tpath: [],\n\t\t\tsubinstructions: [],\n\t\t};\n\n\t\tfor (const s of selection) {\n\t\t\tif (is(s.field, SingleStoreColumn)) {\n\t\t\t\tif (!s.field[requiredExtension] || !is(this, s.field[requiredExtension])) continue;\n\n\t\t\t\tconst extCol = s.field as SingleStoreS3File;\n\n\t\t\t\t(p.subinstructions ??= []).push({\n\t\t\t\t\tpath: [...prefix, `[${JSON.stringify(s.tsKey)}]`],\n\t\t\t\t\tfileMode: extCol[extensionColumnConfig].fileMode,\n\t\t\t\t\tfetchMode: extCol[extensionColumnConfig].fetchMode,\n\t\t\t\t\toptions: extCol[extensionColumnConfig].options,\n\t\t\t\t});\n\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tif (s.selection) {\n\t\t\t\tconst relation = schema[s.relationTableTsKey!]!;\n\n\t\t\t\tif (is(table.relations[s.tsKey]!, One)) {\n\t\t\t\t\tthis.buildRqbOutputMappingInstructions(schema, relation, s.selection, p, [\n\t\t\t\t\t\t...prefix,\n\t\t\t\t\t\t`[${JSON.stringify(s.tsKey)}]`,\n\t\t\t\t\t]);\n\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\n\t\t\t\tconst nestedSubs = this.buildRqbOutputMappingInstructions(schema, relation, s.selection);\n\t\t\t\tif (!nestedSubs.length) continue;\n\n\t\t\t\t(p.subinstructions ??= []).push({\n\t\t\t\t\tpath: [...prefix, `[${JSON.stringify(s.tsKey)}]`],\n\t\t\t\t\tsubinstructions: nestedSubs,\n\t\t\t\t});\n\t\t\t}\n\t\t}\n\n\t\treturn p.subinstructions;\n\t}\n\n\tprivate buildParamMappingInstructions(params: unknown[]) {\n\t\tconst indexes: DrizzleS3FileInputMappingInstruction[] = [];\n\t\tconst placeholders: DrizzleS3FilePlaceholderMappingInstruction[] = [];\n\t\tlet i = 0;\n\t\tfor (const item of params) {\n\t\t\tif (!is(item, ExtensionParam) || !is(this, item.extension)) {\n\t\t\t\t++i;\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tif (is(item.value, Placeholder)) {\n\t\t\t\tplaceholders.push({\n\t\t\t\t\tkey: item.value.name,\n\t\t\t\t\tmode: (item.encoder as SingleStoreS3File)[extensionColumnConfig].fileMode,\n\t\t\t\t});\n\t\t\t\t++i;\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tindexes.push({\n\t\t\t\tindex: i++,\n\t\t\t\tmode: (item.encoder as SingleStoreS3File)[extensionColumnConfig].fileMode,\n\t\t\t});\n\t\t}\n\n\t\treturn {\n\t\t\tplaceholders,\n\t\t\tindexes,\n\t\t};\n\t}\n\n\tprivate async hookBefore(context: DrizzleSingleStoreHookContext<S3ExtMeta> & { stage: 'before' }) {\n\t\tif (!context?.params?.length) return;\n\n\t\tlet indexes: DrizzleS3FileInputMappingInstruction[] = [];\n\t\tlet placeholders: DrizzleS3FilePlaceholderMappingInstruction[] = [];\n\t\tconst errors: unknown[] = [];\n\n\t\tswitch (context.query) {\n\t\t\tcase 'select':\n\t\t\tcase 'delete': {\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase 'update':\n\t\t\tcase 'insert': {\n\t\t\t\tconst instr = this.buildParamMappingInstructions(context.params as Record<string, unknown>[]);\n\n\t\t\t\tindexes = instr.indexes;\n\t\t\t\tplaceholders = instr.placeholders;\n\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\tawait mapParams(this.s3, context.params, indexes, errors);\n\n\t\tif (errors.length) {\n\t\t\tthrow new DrizzleError({\n\t\t\t\tmessage: `Error${\n\t\t\t\t\terrors.length > 1 ? 's' : ''\n\t\t\t\t} occured during S3 file upload.\\nLog \"(e as DrizzleError).cause\" for details.`,\n\t\t\t\tcause: errors,\n\t\t\t});\n\t\t}\n\n\t\tif (!context.metadata?.encoder) {\n\t\t\tcontext.metadata = context.metadata ?? {};\n\n\t\t\tcontext.metadata.encoder = buildEncoder(placeholders);\n\t\t}\n\n\t\tawait context.metadata.encoder(this.s3, context.placeholders, errors, uploadFile, objectIdToText);\n\n\t\tif (errors.length) {\n\t\t\tthrow new DrizzleError({\n\t\t\t\tmessage: `Error${\n\t\t\t\t\terrors.length > 1 ? 's' : ''\n\t\t\t\t} occured during S3 file upload.\\nLog \"(e as DrizzleError).cause\" for details.`,\n\t\t\t\tcause: errors,\n\t\t\t});\n\t\t}\n\n\t\treturn;\n\t}\n\n\tprivate async hookAfter(context: DrizzleSingleStoreHookContext<S3ExtMeta> & { stage: 'after' }) {\n\t\tconst errors: unknown[] = [];\n\t\tconst data = context.data as Record<string, unknown>[];\n\n\t\tswitch (context.query) {\n\t\t\tcase 'select': {\n\t\t\t\tif (!data.length) return;\n\n\t\t\t\tif (!context.metadata?.decoder) {\n\t\t\t\t\tconst fields = context.fieldsOrdered ?? context.config.fieldsFlat\n\t\t\t\t\t\t?? orderSelectedFields<SingleStoreColumn>(context.config.fields);\n\n\t\t\t\t\tconst instructions = this.buildOutputMappingInstructions(fields);\n\n\t\t\t\t\tconst decoder = buildDecoder(instructions);\n\n\t\t\t\t\tcontext.metadata = context.metadata ?? {};\n\t\t\t\t\tcontext.metadata.decoder = decoder;\n\t\t\t\t}\n\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase '_query': {\n\t\t\t\tif ((context.mode === 'first' && !data) || (context.mode === 'many' && !data.length)) return;\n\n\t\t\t\tif (!context.metadata?.decoder) {\n\t\t\t\t\tconst instructions = this.buildRqbOutputMappingInstructions(\n\t\t\t\t\t\tcontext.tablesConfig,\n\t\t\t\t\t\tcontext.tableConfig,\n\t\t\t\t\t\tcontext.config.selection,\n\t\t\t\t\t);\n\n\t\t\t\t\tconst decoder = buildDecoder(instructions, context.mode);\n\n\t\t\t\t\tcontext.metadata = context.metadata ?? {};\n\t\t\t\t\tcontext.metadata.decoder = decoder;\n\t\t\t\t}\n\n\t\t\t\tbreak;\n\t\t\t}\n\n\t\t\tcase 'update':\n\t\t\tcase 'delete':\n\t\t\tcase 'insert': {\n\t\t\t\tif (!data.length) return;\n\n\t\t\t\tif (!context.metadata?.decoder) {\n\t\t\t\t\tif (!(context.config.returning)) {\n\t\t\t\t\t\tconst decoder = buildDecoder([]);\n\n\t\t\t\t\t\tcontext.metadata = context.metadata ?? {};\n\t\t\t\t\t\tcontext.metadata.decoder = decoder;\n\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\n\t\t\t\t\tconst fields = context.config.returning;\n\t\t\t\t\tconst instructions = this.buildOutputMappingInstructions(fields);\n\n\t\t\t\t\tconst decoder = buildDecoder(instructions);\n\n\t\t\t\t\tcontext.metadata = context.metadata ?? {};\n\t\t\t\t\tcontext.metadata.decoder = decoder;\n\t\t\t\t}\n\t\t\t\tbreak;\n\t\t\t}\n\t\t}\n\n\t\tawait context.metadata.decoder(\n\t\t\tthis.s3,\n\t\t\tdata,\n\t\t\terrors,\n\t\t\tdownloadFile,\n\t\t\t// downloadFileMeta,\n\t\t\tpresignDownload,\n\t\t\ttextToObjectId,\n\t\t);\n\n\t\tif (errors.length) {\n\t\t\tthrow new DrizzleError({\n\t\t\t\tmessage: `Error${\n\t\t\t\t\terrors.length > 1 ? 's' : ''\n\t\t\t\t} occured during S3 file download.\\nLog \"(e as DrizzleError).cause\" for details.`,\n\t\t\t\tcause: errors,\n\t\t\t});\n\t\t}\n\n\t\treturn;\n\t}\n\n\toverride async hook(context: DrizzleSingleStoreHookContext<S3ExtMeta>) {\n\t\treturn context.stage === 'before' ? await this.hookBefore(context) : await this.hookAfter(context);\n\t}\n}\n\nexport function s3FileExt(client: S3Client | S3ClientConfig): DrizzleSingleStoreS3Extension {\n\treturn new DrizzleSingleStoreS3Extension(client);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,uBAA8C;AAC9C,oBAA+B;AAC/B,oBAA6B;AAC7B,4BAAwE;AACxE,yBAAgF;AAEhF,uBAKO;AACP,8BAAkC;AAClC,iBAA4C;AAC5C,mBAAoC;AACpC,oBAUO;AAUA,MAAM,sCAAsC,+CAAuC;AAAA,EACzF,QAA0B,wBAAU,IAAY;AAAA,EAEhD,QAA0B,mCAAa,IAAI;AAAA,EAEnC;AAAA,EAER,YAAY,QAAmC;AAC9C,UAAM;AAGN,SAAK,KAAK,OAAO,WAAW,aAAa,kBAAkB,6BAAY,OAAO,YAAY,SAAS,YAChG,SACA,IAAI,0BAAS,MAAM;AAAA,EACvB;AAAA,EAEQ,+BAA+B,WAAqD;AAC3F,UAAM,QAAiD,CAAC;AAExD,eAAW,KAAK,WAAW;AAC1B,UACC,KAAC,kBAAG,EAAE,OAAO,yCAAiB,KAAK,CAAsB,EAAE,MAAO,uCAAiB,KAChF,KAAC,kBAAG,MAAM,EAAE,MAAM,uCAAiB,CAAE;AACvC;AAEF,YAAM,SAAS,EAAE;AAEjB,YAAM,EAAE,WAAW,UAAU,QAAQ,IAAI,OAAO,2CAAqB;AAErE,YAAM,KAAK;AAAA,QACV,MAAM,EAAE,KAAK,IAAI,CAAC,MAAM,IAAI,KAAK,UAAU,CAAC,CAAC,GAAG;AAAA,QAChD;AAAA,QACA;AAAA,QACA;AAAA,MACD,CAAC;AAAA,IACF;AAEA,WAAO;AAAA,EACR;AAAA,EAEQ,kCACP,QACA,OACA,WACA,QACA,SAAmB,CAAC,GACnB;AACD,UAAM,IAAuD,UAAU;AAAA,MACtE,MAAM,CAAC;AAAA,MACP,iBAAiB,CAAC;AAAA,IACnB;AAEA,eAAW,KAAK,WAAW;AAC1B,cAAI,kBAAG,EAAE,OAAO,yCAAiB,GAAG;AACnC,YAAI,CAAC,EAAE,MAAM,uCAAiB,KAAK,KAAC,kBAAG,MAAM,EAAE,MAAM,uCAAiB,CAAC;AAAG;AAE1E,cAAM,SAAS,EAAE;AAEjB,SAAC,EAAE,oBAAoB,CAAC,GAAG,KAAK;AAAA,UAC/B,MAAM,CAAC,GAAG,QAAQ,IAAI,KAAK,UAAU,EAAE,KAAK,CAAC,GAAG;AAAA,UAChD,UAAU,OAAO,2CAAqB,EAAE;AAAA,UACxC,WAAW,OAAO,2CAAqB,EAAE;AAAA,UACzC,SAAS,OAAO,2CAAqB,EAAE;AAAA,QACxC,CAAC;AAED;AAAA,MACD;AAEA,UAAI,EAAE,WAAW;AAChB,cAAM,WAAW,OAAO,EAAE,kBAAmB;AAE7C,gBAAI,kBAAG,MAAM,UAAU,EAAE,KAAK,GAAI,oBAAG,GAAG;AACvC,eAAK,kCAAkC,QAAQ,UAAU,EAAE,WAAW,GAAG;AAAA,YACxE,GAAG;AAAA,YACH,IAAI,KAAK,UAAU,EAAE,KAAK,CAAC;AAAA,UAC5B,CAAC;AAED;AAAA,QACD;AAEA,cAAM,aAAa,KAAK,kCAAkC,QAAQ,UAAU,EAAE,SAAS;AACvF,YAAI,CAAC,WAAW;AAAQ;AAExB,SAAC,EAAE,oBAAoB,CAAC,GAAG,KAAK;AAAA,UAC/B,MAAM,CAAC,GAAG,QAAQ,IAAI,KAAK,UAAU,EAAE,KAAK,CAAC,GAAG;AAAA,UAChD,iBAAiB;AAAA,QAClB,CAAC;AAAA,MACF;AAAA,IACD;AAEA,WAAO,EAAE;AAAA,EACV;AAAA,EAEQ,8BAA8B,QAAmB;AACxD,UAAM,UAAkD,CAAC;AACzD,UAAM,eAA6D,CAAC;AACpE,QAAI,IAAI;AACR,eAAW,QAAQ,QAAQ;AAC1B,UAAI,KAAC,kBAAG,MAAM,yBAAc,KAAK,KAAC,kBAAG,MAAM,KAAK,SAAS,GAAG;AAC3D,UAAE;AACF;AAAA,MACD;AAEA,cAAI,kBAAG,KAAK,OAAO,sBAAW,GAAG;AAChC,qBAAa,KAAK;AAAA,UACjB,KAAK,KAAK,MAAM;AAAA,UAChB,MAAO,KAAK,QAA8B,2CAAqB,EAAE;AAAA,QAClE,CAAC;AACD,UAAE;AACF;AAAA,MACD;AAEA,cAAQ,KAAK;AAAA,QACZ,OAAO;AAAA,QACP,MAAO,KAAK,QAA8B,2CAAqB,EAAE;AAAA,MAClE,CAAC;AAAA,IACF;AAEA,WAAO;AAAA,MACN;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAAA,EAEA,MAAc,WAAW,SAAyE;AACjG,QAAI,CAAC,SAAS,QAAQ;AAAQ;AAE9B,QAAI,UAAkD,CAAC;AACvD,QAAI,eAA6D,CAAC;AAClE,UAAM,SAAoB,CAAC;AAE3B,YAAQ,QAAQ,OAAO;AAAA,MACtB,KAAK;AAAA,MACL,KAAK,UAAU;AACd;AAAA,MACD;AAAA,MAEA,KAAK;AAAA,MACL,KAAK,UAAU;AACd,cAAM,QAAQ,KAAK,8BAA8B,QAAQ,MAAmC;AAE5F,kBAAU,MAAM;AAChB,uBAAe,MAAM;AAErB;AAAA,MACD;AAAA,IACD;AAEA,cAAM,yBAAU,KAAK,IAAI,QAAQ,QAAQ,SAAS,MAAM;AAExD,QAAI,OAAO,QAAQ;AAClB,YAAM,IAAI,2BAAa;AAAA,QACtB,SAAS,QACR,OAAO,SAAS,IAAI,MAAM,EAC3B;AAAA;AAAA,QACA,OAAO;AAAA,MACR,CAAC;AAAA,IACF;AAEA,QAAI,CAAC,QAAQ,UAAU,SAAS;AAC/B,cAAQ,WAAW,QAAQ,YAAY,CAAC;AAExC,cAAQ,SAAS,cAAU,4BAAa,YAAY;AAAA,IACrD;AAEA,UAAM,QAAQ,SAAS,QAAQ,KAAK,IAAI,QAAQ,cAAc,QAAQ,0BAAY,4BAAc;AAEhG,QAAI,OAAO,QAAQ;AAClB,YAAM,IAAI,2BAAa;AAAA,QACtB,SAAS,QACR,OAAO,SAAS,IAAI,MAAM,EAC3B;AAAA;AAAA,QACA,OAAO;AAAA,MACR,CAAC;AAAA,IACF;AAEA;AAAA,EACD;AAAA,EAEA,MAAc,UAAU,SAAwE;AAC/F,UAAM,SAAoB,CAAC;AAC3B,UAAM,OAAO,QAAQ;AAErB,YAAQ,QAAQ,OAAO;AAAA,MACtB,KAAK,UAAU;AACd,YAAI,CAAC,KAAK;AAAQ;AAElB,YAAI,CAAC,QAAQ,UAAU,SAAS;AAC/B,gBAAM,SAAS,QAAQ,iBAAiB,QAAQ,OAAO,kBACnD,kCAAuC,QAAQ,OAAO,MAAM;AAEhE,gBAAM,eAAe,KAAK,+BAA+B,MAAM;AAE/D,gBAAM,cAAU,4BAAa,YAAY;AAEzC,kBAAQ,WAAW,QAAQ,YAAY,CAAC;AACxC,kBAAQ,SAAS,UAAU;AAAA,QAC5B;AAEA;AAAA,MACD;AAAA,MAEA,KAAK,UAAU;AACd,YAAK,QAAQ,SAAS,WAAW,CAAC,QAAU,QAAQ,SAAS,UAAU,CAAC,KAAK;AAAS;AAEtF,YAAI,CAAC,QAAQ,UAAU,SAAS;AAC/B,gBAAM,eAAe,KAAK;AAAA,YACzB,QAAQ;AAAA,YACR,QAAQ;AAAA,YACR,QAAQ,OAAO;AAAA,UAChB;AAEA,gBAAM,cAAU,4BAAa,cAAc,QAAQ,IAAI;AAEvD,kBAAQ,WAAW,QAAQ,YAAY,CAAC;AACxC,kBAAQ,SAAS,UAAU;AAAA,QAC5B;AAEA;AAAA,MACD;AAAA,MAEA,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK,UAAU;AACd,YAAI,CAAC,KAAK;AAAQ;AAElB,YAAI,CAAC,QAAQ,UAAU,SAAS;AAC/B,cAAI,CAAE,QAAQ,OAAO,WAAY;AAChC,kBAAMA,eAAU,4BAAa,CAAC,CAAC;AAE/B,oBAAQ,WAAW,QAAQ,YAAY,CAAC;AACxC,oBAAQ,SAAS,UAAUA;AAE3B;AAAA,UACD;AAEA,gBAAM,SAAS,QAAQ,OAAO;AAC9B,gBAAM,eAAe,KAAK,+BAA+B,MAAM;AAE/D,gBAAM,cAAU,4BAAa,YAAY;AAEzC,kBAAQ,WAAW,QAAQ,YAAY,CAAC;AACxC,kBAAQ,SAAS,UAAU;AAAA,QAC5B;AACA;AAAA,MACD;AAAA,IACD;AAEA,UAAM,QAAQ,SAAS;AAAA,MACtB,KAAK;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA;AAAA,MAEA;AAAA,MACA;AAAA,IACD;AAEA,QAAI,OAAO,QAAQ;AAClB,YAAM,IAAI,2BAAa;AAAA,QACtB,SAAS,QACR,OAAO,SAAS,IAAI,MAAM,EAC3B;AAAA;AAAA,QACA,OAAO;AAAA,MACR,CAAC;AAAA,IACF;AAEA;AAAA,EACD;AAAA,EAEA,MAAe,KAAK,SAAmD;AACtE,WAAO,QAAQ,UAAU,WAAW,MAAM,KAAK,WAAW,OAAO,IAAI,MAAM,KAAK,UAAU,OAAO;AAAA,EAClG;AACD;AAEO,SAAS,UAAU,QAAkE;AAC3F,SAAO,IAAI,8BAA8B,MAAM;AAChD;","names":["decoder"]}

package/extensions/s3-file/singlestore/extension.d.cts

@@ -0,0 +1,18 @@
+import { S3Client, type S3ClientConfig } from '@aws-sdk/client-s3';
+import { entityKind } from "../../../entity.cjs";
+import { extensionName } from "../../../extension-core/index.cjs";
+import { DrizzleSingleStoreExtension, type DrizzleSingleStoreHookContext } from "../../../extension-core/singlestore/index.cjs";
+import type { S3ExtMeta } from "../common.cjs";
+export declare class DrizzleSingleStoreS3Extension extends DrizzleSingleStoreExtension<S3ExtMeta> {
+    static readonly [entityKind]: string;
+    static readonly [extensionName] = "Drizzle SingleStore S3 Extension";
+    private s3;
+    constructor(client: S3Client | S3ClientConfig);
+    private buildOutputMappingInstructions;
+    private buildRqbOutputMappingInstructions;
+    private buildParamMappingInstructions;
+    private hookBefore;
+    private hookAfter;
+    hook(context: DrizzleSingleStoreHookContext<S3ExtMeta>): Promise<void>;
+}
+export declare function s3FileExt(client: S3Client | S3ClientConfig): DrizzleSingleStoreS3Extension;

package/extensions/s3-file/singlestore/extension.d.ts

@@ -0,0 +1,18 @@
+import { S3Client, type S3ClientConfig } from '@aws-sdk/client-s3';
+import { entityKind } from "../../../entity.js";
+import { extensionName } from "../../../extension-core/index.js";
+import { DrizzleSingleStoreExtension, type DrizzleSingleStoreHookContext } from "../../../extension-core/singlestore/index.js";
+import type { S3ExtMeta } from "../common.js";
+export declare class DrizzleSingleStoreS3Extension extends DrizzleSingleStoreExtension<S3ExtMeta> {
+    static readonly [entityKind]: string;
+    static readonly [extensionName] = "Drizzle SingleStore S3 Extension";
+    private s3;
+    constructor(client: S3Client | S3ClientConfig);
+    private buildOutputMappingInstructions;
+    private buildRqbOutputMappingInstructions;
+    private buildParamMappingInstructions;
+    private hookBefore;
+    private hookAfter;
+    hook(context: DrizzleSingleStoreHookContext<S3ExtMeta>): Promise<void>;
+}
+export declare function s3FileExt(client: S3Client | S3ClientConfig): DrizzleSingleStoreS3Extension;

package/extensions/s3-file/singlestore/extension.js

@@ -0,0 +1,233 @@
+import { S3Client } from "@aws-sdk/client-s3";
+import { entityKind, is } from "../../../entity.js";
+import { DrizzleError } from "../../../errors.js";
+import { extensionColumnConfig, extensionName, requiredExtension } from "../../../extension-core/index.js";
+import { DrizzleSingleStoreExtension } from "../../../extension-core/singlestore/index.js";
+import {
+  One
+} from "../../../relations.js";
+import { SingleStoreColumn } from "../../../singlestore-core/index.js";
+import { ExtensionParam, Placeholder } from "../../../sql/index.js";
+import { orderSelectedFields } from "../../../utils.js";
+import {
+  buildDecoder,
+  buildEncoder,
+  downloadFile,
+  mapParams,
+  objectIdToText,
+  presignDownload,
+  textToObjectId,
+  uploadFile
+} from "../common.js";
+class DrizzleSingleStoreS3Extension extends DrizzleSingleStoreExtension {
+  static [entityKind] = "DrizzleSingleStoreS3Extension";
+  static [extensionName] = "Drizzle SingleStore S3 Extension";
+  s3;
+  constructor(client) {
+    super();
+    this.s3 = typeof client === "object" && (client instanceof S3Client || client.constructor.name !== "Object") ? client : new S3Client(client);
+  }
+  buildOutputMappingInstructions(selection) {
+    const paths = [];
+    for (const f of selection) {
+      if (!is(f.field, SingleStoreColumn) || !f.field[requiredExtension] || !is(this, f.field[requiredExtension]))
+        continue;
+      const column = f.field;
+      const { fetchMode, fileMode, options } = column[extensionColumnConfig];
+      paths.push({
+        path: f.path.map((p) => `[${JSON.stringify(p)}]`),
+        fetchMode,
+        fileMode,
+        options
+      });
+    }
+    return paths;
+  }
+  buildRqbOutputMappingInstructions(schema, table, selection, parent, prefix = []) {
+    const p = parent ?? {
+      path: [],
+      subinstructions: []
+    };
+    for (const s of selection) {
+      if (is(s.field, SingleStoreColumn)) {
+        if (!s.field[requiredExtension] || !is(this, s.field[requiredExtension]))
+          continue;
+        const extCol = s.field;
+        (p.subinstructions ??= []).push({
+          path: [...prefix, `[${JSON.stringify(s.tsKey)}]`],
+          fileMode: extCol[extensionColumnConfig].fileMode,
+          fetchMode: extCol[extensionColumnConfig].fetchMode,
+          options: extCol[extensionColumnConfig].options
+        });
+        continue;
+      }
+      if (s.selection) {
+        const relation = schema[s.relationTableTsKey];
+        if (is(table.relations[s.tsKey], One)) {
+          this.buildRqbOutputMappingInstructions(schema, relation, s.selection, p, [
+            ...prefix,
+            `[${JSON.stringify(s.tsKey)}]`
+          ]);
+          continue;
+        }
+        const nestedSubs = this.buildRqbOutputMappingInstructions(schema, relation, s.selection);
+        if (!nestedSubs.length)
+          continue;
+        (p.subinstructions ??= []).push({
+          path: [...prefix, `[${JSON.stringify(s.tsKey)}]`],
+          subinstructions: nestedSubs
+        });
+      }
+    }
+    return p.subinstructions;
+  }
+  buildParamMappingInstructions(params) {
+    const indexes = [];
+    const placeholders = [];
+    let i = 0;
+    for (const item of params) {
+      if (!is(item, ExtensionParam) || !is(this, item.extension)) {
+        ++i;
+        continue;
+      }
+      if (is(item.value, Placeholder)) {
+        placeholders.push({
+          key: item.value.name,
+          mode: item.encoder[extensionColumnConfig].fileMode
+        });
+        ++i;
+        continue;
+      }
+      indexes.push({
+        index: i++,
+        mode: item.encoder[extensionColumnConfig].fileMode
+      });
+    }
+    return {
+      placeholders,
+      indexes
+    };
+  }
+  async hookBefore(context) {
+    if (!context?.params?.length)
+      return;
+    let indexes = [];
+    let placeholders = [];
+    const errors = [];
+    switch (context.query) {
+      case "select":
+      case "delete": {
+        break;
+      }
+      case "update":
+      case "insert": {
+        const instr = this.buildParamMappingInstructions(context.params);
+        indexes = instr.indexes;
+        placeholders = instr.placeholders;
+        break;
+      }
+    }
+    await mapParams(this.s3, context.params, indexes, errors);
+    if (errors.length) {
+      throw new DrizzleError({
+        message: `Error${errors.length > 1 ? "s" : ""} occured during S3 file upload.
+Log "(e as DrizzleError).cause" for details.`,
+        cause: errors
+      });
+    }
+    if (!context.metadata?.encoder) {
+      context.metadata = context.metadata ?? {};
+      context.metadata.encoder = buildEncoder(placeholders);
+    }
+    await context.metadata.encoder(this.s3, context.placeholders, errors, uploadFile, objectIdToText);
+    if (errors.length) {
+      throw new DrizzleError({
+        message: `Error${errors.length > 1 ? "s" : ""} occured during S3 file upload.
+Log "(e as DrizzleError).cause" for details.`,
+        cause: errors
+      });
+    }
+    return;
+  }
+  async hookAfter(context) {
+    const errors = [];
+    const data = context.data;
+    switch (context.query) {
+      case "select": {
+        if (!data.length)
+          return;
+        if (!context.metadata?.decoder) {
+          const fields = context.fieldsOrdered ?? context.config.fieldsFlat ?? orderSelectedFields(context.config.fields);
+          const instructions = this.buildOutputMappingInstructions(fields);
+          const decoder = buildDecoder(instructions);
+          context.metadata = context.metadata ?? {};
+          context.metadata.decoder = decoder;
+        }
+        break;
+      }
+      case "_query": {
+        if (context.mode === "first" && !data || context.mode === "many" && !data.length)
+          return;
+        if (!context.metadata?.decoder) {
+          const instructions = this.buildRqbOutputMappingInstructions(
+            context.tablesConfig,
+            context.tableConfig,
+            context.config.selection
+          );
+          const decoder = buildDecoder(instructions, context.mode);
+          context.metadata = context.metadata ?? {};
+          context.metadata.decoder = decoder;
+        }
+        break;
+      }
+      case "update":
+      case "delete":
+      case "insert": {
+        if (!data.length)
+          return;
+        if (!context.metadata?.decoder) {
+          if (!context.config.returning) {
+            const decoder2 = buildDecoder([]);
+            context.metadata = context.metadata ?? {};
+            context.metadata.decoder = decoder2;
+            break;
+          }
+          const fields = context.config.returning;
+          const instructions = this.buildOutputMappingInstructions(fields);
+          const decoder = buildDecoder(instructions);
+          context.metadata = context.metadata ?? {};
+          context.metadata.decoder = decoder;
+        }
+        break;
+      }
+    }
+    await context.metadata.decoder(
+      this.s3,
+      data,
+      errors,
+      downloadFile,
+      // downloadFileMeta,
+      presignDownload,
+      textToObjectId
+    );
+    if (errors.length) {
+      throw new DrizzleError({
+        message: `Error${errors.length > 1 ? "s" : ""} occured during S3 file download.
+Log "(e as DrizzleError).cause" for details.`,
+        cause: errors
+      });
+    }
+    return;
+  }
+  async hook(context) {
+    return context.stage === "before" ? await this.hookBefore(context) : await this.hookAfter(context);
+  }
+}
+function s3FileExt(client) {
+  return new DrizzleSingleStoreS3Extension(client);
+}
+export {
+  DrizzleSingleStoreS3Extension,
+  s3FileExt
+};
+//# sourceMappingURL=extension.js.map