@leonardovida-md/drizzle-neo-duckdb 1.0.0

package/dist/columns.d.ts

@@ -0,0 +1,129 @@
+import { type SQL } from 'drizzle-orm';
+import type { SQLWrapper } from 'drizzle-orm/sql/sql';
+type IntColType = 'SMALLINT' | 'INTEGER' | 'BIGINT' | 'HUGEINT' | 'USMALLINT' | 'UINTEGER' | 'UBIGINT' | 'UHUGEINT' | 'INT' | 'INT16' | 'INT32' | 'INT64' | 'INT128' | 'LONG' | 'VARINT';
+type FloatColType = 'FLOAT' | 'DOUBLE';
+type StringColType = 'STRING' | 'VARCHAR' | 'TEXT';
+type BoolColType = 'BOOLEAN' | 'BOOL';
+type BlobColType = 'BLOB' | 'BYTEA' | 'VARBINARY';
+type DateColType = 'DATE' | 'TIME' | 'TIMETZ' | 'TIMESTAMP' | 'DATETIME' | 'TIMESTAMPTZ' | 'TIMESTAMP_MS' | 'TIMESTAMP_S';
+type AnyColType = IntColType | FloatColType | StringColType | BoolColType | DateColType | BlobColType;
+type ListColType = `${AnyColType}[]`;
+type ArrayColType = `${AnyColType}[${number}]`;
+type StructColType = `STRUCT (${string})`;
+type Primitive = AnyColType | ListColType | ArrayColType | StructColType;
+export declare const duckDbList: <TData = unknown>(name: string, elementType: AnyColType) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: TData[];
+    driverParam: string | unknown[] | SQL<unknown>;
+    enumValues: undefined;
+}>;
+export declare const duckDbArray: <TData = unknown>(name: string, elementType: AnyColType, fixedLength?: number) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: TData[];
+    driverParam: string | unknown[] | SQL<unknown>;
+    enumValues: undefined;
+}>;
+export declare const duckDbMap: <TData extends Record<string, any>>(name: string, valueType: AnyColType | ListColType | ArrayColType) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: TData;
+    driverParam: TData;
+    enumValues: undefined;
+}>;
+export declare const duckDbStruct: <TData extends Record<string, any>>(name: string, schema: Record<string, Primitive>) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: TData;
+    driverParam: TData;
+    enumValues: undefined;
+}>;
+export declare const duckDbJson: <TData = unknown>(name: string) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: TData;
+    driverParam: string | SQL<unknown>;
+    enumValues: undefined;
+}>;
+export declare const duckDbBlob: {
+    (): import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+        name: "";
+        dataType: "custom";
+        columnType: "PgCustomColumn";
+        data: Buffer;
+        driverParam: unknown;
+        enumValues: undefined;
+    }>;
+    <TConfig extends Record<string, any>>(fieldConfig?: TConfig | undefined): import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+        name: "";
+        dataType: "custom";
+        columnType: "PgCustomColumn";
+        data: Buffer;
+        driverParam: unknown;
+        enumValues: undefined;
+    }>;
+    <TName extends string>(dbName: TName, fieldConfig?: unknown): import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+        name: TName;
+        dataType: "custom";
+        columnType: "PgCustomColumn";
+        data: Buffer;
+        driverParam: unknown;
+        enumValues: undefined;
+    }>;
+};
+export declare const duckDbInet: (name: string) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: string;
+    driverParam: string;
+    enumValues: undefined;
+}>;
+export declare const duckDbInterval: (name: string) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: string;
+    driverParam: string;
+    enumValues: undefined;
+}>;
+type TimestampMode = 'date' | 'string';
+interface TimestampOptions {
+    withTimezone?: boolean;
+    mode?: TimestampMode;
+    precision?: number;
+}
+export declare const duckDbTimestamp: (name: string, options?: TimestampOptions) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: string | Date;
+    driverParam: string | Date | SQL<unknown>;
+    enumValues: undefined;
+}>;
+export declare const duckDbDate: (name: string) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: string | Date;
+    driverParam: string | Date;
+    enumValues: undefined;
+}>;
+export declare const duckDbTime: (name: string) => import("drizzle-orm/pg-core").PgCustomColumnBuilder<{
+    name: string;
+    dataType: "custom";
+    columnType: "PgCustomColumn";
+    data: string;
+    driverParam: string | bigint;
+    enumValues: undefined;
+}>;
+export declare function duckDbArrayContains(column: SQLWrapper, values: unknown[] | SQLWrapper): SQL;
+export declare function duckDbArrayContained(column: SQLWrapper, values: unknown[] | SQLWrapper): SQL;
+export declare function duckDbArrayOverlaps(column: SQLWrapper, values: unknown[] | SQLWrapper): SQL;
+export {};

package/dist/dialect.d.ts

@@ -0,0 +1,11 @@
+import { entityKind } from 'drizzle-orm/entity';
+import type { MigrationConfig, MigrationMeta } from 'drizzle-orm/migrator';
+import { PgDialect, PgSession } from 'drizzle-orm/pg-core';
+import { type DriverValueEncoder, type QueryTypingsValue } from 'drizzle-orm';
+export declare class DuckDBDialect extends PgDialect {
+    static readonly [entityKind]: string;
+    private hasPgJsonColumn;
+    assertNoPgJsonColumns(): void;
+    migrate(migrations: MigrationMeta[], session: PgSession, config: MigrationConfig | string): Promise<void>;
+    prepareTyping(encoder: DriverValueEncoder<unknown, unknown>): QueryTypingsValue;
+}

package/dist/driver.d.ts

@@ -0,0 +1,37 @@
+import { entityKind } from 'drizzle-orm/entity';
+import type { Logger } from 'drizzle-orm/logger';
+import { PgDatabase } from 'drizzle-orm/pg-core/db';
+import type { SelectedFields } from 'drizzle-orm/pg-core/query-builders';
+import { type ExtractTablesWithRelations, type RelationalSchemaConfig, type TablesRelationalConfig } from 'drizzle-orm/relations';
+import { type DrizzleConfig } from 'drizzle-orm/utils';
+import type { DuckDBClientLike, DuckDBQueryResultHKT, DuckDBTransaction } from './session.ts';
+import { DuckDBSession } from './session.ts';
+import { DuckDBDialect } from './dialect.ts';
+import { DuckDBSelectBuilder } from './select-builder.ts';
+export interface PgDriverOptions {
+    logger?: Logger;
+    rewriteArrays?: boolean;
+    rejectStringArrayLiterals?: boolean;
+}
+export declare class DuckDBDriver {
+    private client;
+    private dialect;
+    private options;
+    static readonly [entityKind]: string;
+    constructor(client: DuckDBClientLike, dialect: DuckDBDialect, options?: PgDriverOptions);
+    createSession(schema: RelationalSchemaConfig<TablesRelationalConfig> | undefined): DuckDBSession<Record<string, unknown>, TablesRelationalConfig>;
+}
+export interface DuckDBDrizzleConfig<TSchema extends Record<string, unknown> = Record<string, never>> extends DrizzleConfig<TSchema> {
+    rewriteArrays?: boolean;
+    rejectStringArrayLiterals?: boolean;
+}
+export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(client: DuckDBClientLike, config?: DuckDBDrizzleConfig<TSchema>): DuckDBDatabase<TSchema, ExtractTablesWithRelations<TSchema>>;
+export declare class DuckDBDatabase<TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> extends PgDatabase<DuckDBQueryResultHKT, TFullSchema, TSchema> {
+    readonly dialect: DuckDBDialect;
+    readonly session: DuckDBSession<TFullSchema, TSchema>;
+    static readonly [entityKind]: string;
+    constructor(dialect: DuckDBDialect, session: DuckDBSession<TFullSchema, TSchema>, schema: RelationalSchemaConfig<TSchema> | undefined);
+    select(): DuckDBSelectBuilder<undefined>;
+    select<TSelection extends SelectedFields>(fields: TSelection): DuckDBSelectBuilder<TSelection>;
+    transaction<T>(transaction: (tx: DuckDBTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}