openAgent

1/* oxlint-disable */
2import * as Effect from "effect/Effect"
3import * as Layer from "effect/Layer"
4import { SqlClient } from "effect/unstable/sql/SqlClient"
5import { EffectCache } from "drizzle-orm/cache/core/cache-effect"
6import { EffectLogger } from "drizzle-orm/effect-core"
7import { entityKind } from "drizzle-orm/entity"
8import type { AnyRelations, EmptyRelations } from "drizzle-orm/relations"
9import { SQLiteAsyncDialect } from "drizzle-orm/sqlite-core/dialect"
10import { SQLiteEffectDatabase } from "../sqlite-core/effect/db"
11import type { DrizzleConfig } from "drizzle-orm/utils"
12import { jitCompatCheck } from "../internal/drizzle-utils"
13import { type EffectSQLiteQueryEffectHKT, type EffectSQLiteRunResult, EffectSQLiteSession } from "./session"
14 
15export class EffectSQLiteDatabase<TRelations extends AnyRelations = EmptyRelations> extends SQLiteEffectDatabase<
16  EffectSQLiteQueryEffectHKT,
17  EffectSQLiteRunResult,
18  TRelations
19> {
20  static override readonly [entityKind]: string = "EffectSQLiteDatabase"
21}
22 
23export type EffectDrizzleSQLiteConfig<TRelations extends AnyRelations = EmptyRelations> = Omit<
24  DrizzleConfig<Record<string, never>, TRelations>,
25  "cache" | "logger" | "schema"
26>
27 
28export const DefaultServices = Layer.merge(EffectCache.Default, EffectLogger.Default)
29 
30/**
31 * Creates an EffectSQLiteDatabase instance.
32 *
33 * Requires a generic Effect `SqlClient`, `EffectLogger`, and `EffectCache` services to be provided.
34 * Drizzle only depends on the generic `SqlClient`; install and provide a compatible SQLite provider such as
35 * `@effect/sql-sqlite-node`, `@effect/sql-sqlite-bun`, or another package that exposes `SqlClient`.
36 *
37 * @example
38 * ```ts
39 * import { SqliteClient } from '@effect/sql-sqlite-node';
40 * import * as SQLiteDrizzle from 'drizzle-orm/effect-sqlite';
41 * import * as Effect from 'effect/Effect';
42 *
43 * const db = yield* SQLiteDrizzle.make({ relations }).pipe(
44 *   Effect.provide(SQLiteDrizzle.DefaultServices),
45 *   Effect.provide(SqliteClient.layer({ filename: 'sqlite.db' })),
46 * );
47 * ```
48 */
49export const make = Effect.fn("SQLiteDrizzle.make")(function* <TRelations extends AnyRelations = EmptyRelations>(
50  config: EffectDrizzleSQLiteConfig<TRelations> = {},
51) {
52  const client = yield* SqlClient
53  const cache = yield* EffectCache
54  const logger = yield* EffectLogger
55 
56  const dialect = new SQLiteAsyncDialect()
57  const relations = config.relations ?? ({} as TRelations)
58  const session = new EffectSQLiteSession(client, dialect, relations, {
59    logger,
60    cache,
61    useJitMappers: jitCompatCheck(config.jit),
62  })
63  const db = new EffectSQLiteDatabase(dialect, session, relations) as EffectSQLiteDatabase<TRelations> & {
64    $client: SqlClient
65  }
66  db.$client = client
67  db.$cache.invalidate = cache.onMutate
68 
69  return db
70})
71 
72/**
73 * Convenience function that creates an EffectSQLiteDatabase with `DefaultServices` already provided.
74 */
75export const makeWithDefaults = <TRelations extends AnyRelations = EmptyRelations>(
76  config: EffectDrizzleSQLiteConfig<TRelations> = {},
77) => make(config).pipe(Effect.provide(DefaultServices))
78