Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 | 30x 30x 30x 30x 30x 30x 30x 15x 15x 30x 30x 30x 30x 1x 1x 1x 30x 1x 29x 28x 1x 1x 1x 1x 1x 18x | import { DefaultLogger, type Logger } from 'drizzle-orm/logger'
import { createTableRelationsHelpers, extractTablesRelationalConfig } from 'drizzle-orm/relations'
import type { Casing } from 'drizzle-orm/utils'
import { YdbDialect } from './dialect.js'
import {
YdbDriver,
type YdbExecutor,
type YdbRemoteCallback,
type YdbTransactionalExecutor,
} from './driver.js'
import type {
YdbSchemaDefinition,
YdbSchemaRelations,
YdbSchemaWithoutTables,
} from '../ydb-core/schema.types.js'
import { YdbSession } from '../ydb-core/session.js'
import { YdbDatabase } from '../ydb-core/db.js'
export interface YdbDrizzleConfig<
TSchemaDefinition extends YdbSchemaDefinition = YdbSchemaWithoutTables,
> {
casing?: Casing | undefined
logger?: boolean | Logger | undefined
/** Exact schema object that powers typed queries like `db.query.users.findMany()`. */
schema?: TSchemaDefinition | undefined
}
export interface YdbDrizzleOptions<
TSchemaDefinition extends YdbSchemaDefinition = YdbSchemaWithoutTables,
> extends YdbDrizzleConfig<TSchemaDefinition> {
connectionString?: string | undefined
client?: YdbExecutor | YdbTransactionalExecutor | undefined
}
export type YdbDrizzleDatabase<
TSchemaDefinition extends YdbSchemaDefinition = YdbSchemaWithoutTables,
> = YdbDatabase<TSchemaDefinition, YdbSchemaRelations<TSchemaDefinition>> & { $client: YdbExecutor }
function isYdbExecutor(value: unknown): value is YdbExecutor {
return (
!!value && typeof value === 'object' && typeof (value as YdbExecutor).execute === 'function'
)
}
function makeDb<TSchemaDefinition extends YdbSchemaDefinition>(
executor: YdbExecutor | YdbTransactionalExecutor,
config: YdbDrizzleConfig<TSchemaDefinition> = {}
): YdbDrizzleDatabase<TSchemaDefinition> {
let dialect = new YdbDialect(config.casing === undefined ? {} : { casing: config.casing })
let logger: Logger | undefined = undefined
Iif (config.logger === true) {
logger = new DefaultLogger()
} else Eif (config.logger !== false) {
logger = config.logger
}
let schema = config.schema
? (() => {
let tablesConfig = extractTablesRelationalConfig(
config.schema,
createTableRelationsHelpers
)
return {
fullSchema: config.schema,
schema: tablesConfig.tables as YdbSchemaRelations<TSchemaDefinition>,
tableNamesMap: tablesConfig.tableNamesMap,
}
})()
: undefined
let session = new YdbSession(executor, dialect, logger === undefined ? {} : { logger })
let db = new YdbDatabase<TSchemaDefinition>(
dialect,
session,
schema
) as YdbDrizzleDatabase<TSchemaDefinition>
db.$client = executor
return db
}
function isYdbOptions<TSchemaDefinition extends YdbSchemaDefinition>(
value: unknown
): value is YdbDrizzleOptions<TSchemaDefinition> {
Iif (!value || typeof value !== 'object') {
return false
}
Iif (isYdbExecutor(value)) {
return false
}
return 'connectionString' in value || 'client' in value || 'schema' in value
}
export function createDrizzle<TSchemaDefinition extends YdbSchemaDefinition>(
input: YdbExecutor | YdbTransactionalExecutor | YdbRemoteCallback,
config: YdbDrizzleConfig<TSchemaDefinition> & { schema: TSchemaDefinition }
): YdbDrizzleDatabase<TSchemaDefinition>
export function createDrizzle(
input: YdbExecutor | YdbTransactionalExecutor | YdbRemoteCallback,
config?: YdbDrizzleConfig<YdbSchemaWithoutTables>
): YdbDrizzleDatabase<YdbSchemaWithoutTables>
export function createDrizzle<TSchemaDefinition extends YdbSchemaDefinition>(
input: YdbDrizzleOptions<TSchemaDefinition> & { schema: TSchemaDefinition }
): YdbDrizzleDatabase<TSchemaDefinition>
export function createDrizzle(
input: YdbDrizzleOptions<YdbSchemaWithoutTables>
): YdbDrizzleDatabase<YdbSchemaWithoutTables>
export function createDrizzle<
TSchemaDefinition extends YdbSchemaDefinition = YdbSchemaWithoutTables,
>(
input:
| YdbExecutor
| YdbTransactionalExecutor
| YdbRemoteCallback
| YdbDrizzleOptions<TSchemaDefinition>,
config?: YdbDrizzleConfig<TSchemaDefinition>
): YdbDrizzleDatabase<TSchemaDefinition> {
if (typeof input === 'function') {
return makeDb(YdbDriver.fromCallback(input), config)
}
if (isYdbExecutor(input)) {
return makeDb(input, config)
}
Eif (isYdbOptions<TSchemaDefinition>(input)) {
Iif (input.client) {
return makeDb(input.client, input)
}
Eif (input.connectionString) {
let client = new YdbDriver(input.connectionString)
return makeDb(client, input)
}
throw new Error('Must include either `client` or `connectionString`.')
}
return makeDb(input, config)
}
export let drizzle = createDrizzle
|