@apibara/indexer 2.0.0-beta.9 → 2.1.0-beta.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (127) hide show
  1. package/dist/index.cjs +271 -39
  2. package/dist/index.d.cts +1 -16
  3. package/dist/index.d.mts +1 -16
  4. package/dist/index.d.ts +1 -16
  5. package/dist/index.mjs +262 -25
  6. package/dist/internal/index.cjs +10 -0
  7. package/dist/internal/index.d.cts +3 -0
  8. package/dist/internal/index.d.mts +3 -0
  9. package/dist/internal/index.d.ts +3 -0
  10. package/dist/internal/index.mjs +8 -0
  11. package/dist/internal/plugins.cjs +38 -0
  12. package/dist/internal/plugins.d.cts +13 -0
  13. package/dist/internal/plugins.d.mts +13 -0
  14. package/dist/internal/plugins.d.ts +13 -0
  15. package/dist/internal/plugins.mjs +34 -0
  16. package/dist/internal/testing.cjs +118 -0
  17. package/dist/internal/testing.d.cts +42 -0
  18. package/dist/internal/testing.d.mts +42 -0
  19. package/dist/internal/testing.d.ts +42 -0
  20. package/dist/internal/testing.mjs +113 -0
  21. package/dist/plugins/index.cjs +39 -3
  22. package/dist/plugins/index.d.cts +16 -2
  23. package/dist/plugins/index.d.mts +16 -2
  24. package/dist/plugins/index.d.ts +16 -2
  25. package/dist/plugins/index.mjs +36 -3
  26. package/dist/shared/indexer.077335f3.cjs +15 -0
  27. package/dist/shared/indexer.2416906c.cjs +29 -0
  28. package/dist/shared/indexer.601ceab0.cjs +7 -0
  29. package/dist/shared/indexer.9b21ddd2.mjs +5 -0
  30. package/dist/shared/indexer.a55ad619.mjs +12 -0
  31. package/dist/shared/indexer.fedcd831.d.cts +100 -0
  32. package/dist/shared/indexer.fedcd831.d.mts +100 -0
  33. package/dist/shared/indexer.fedcd831.d.ts +100 -0
  34. package/dist/shared/indexer.ff25c953.mjs +26 -0
  35. package/dist/testing/index.cjs +52 -50
  36. package/dist/testing/index.d.cts +8 -36
  37. package/dist/testing/index.d.mts +8 -36
  38. package/dist/testing/index.d.ts +8 -36
  39. package/dist/testing/index.mjs +47 -47
  40. package/dist/vcr/index.cjs +84 -17
  41. package/dist/vcr/index.d.cts +16 -7
  42. package/dist/vcr/index.d.mts +16 -7
  43. package/dist/vcr/index.d.ts +16 -7
  44. package/dist/vcr/index.mjs +75 -11
  45. package/package.json +22 -42
  46. package/src/compose.test.ts +76 -0
  47. package/src/compose.ts +71 -0
  48. package/src/context.ts +14 -8
  49. package/src/index.ts +0 -5
  50. package/src/indexer.test.ts +125 -186
  51. package/src/indexer.ts +278 -151
  52. package/src/internal/index.ts +6 -0
  53. package/src/internal/plugins.ts +1 -0
  54. package/src/internal/testing.ts +148 -0
  55. package/src/plugins/config.ts +4 -4
  56. package/src/plugins/context.ts +40 -0
  57. package/src/plugins/index.ts +8 -1
  58. package/src/plugins/logger.ts +30 -0
  59. package/src/plugins/persistence.ts +24 -187
  60. package/src/testing/index.ts +58 -3
  61. package/src/vcr/record.ts +5 -3
  62. package/src/vcr/replay.ts +8 -18
  63. package/dist/plugins/kv.cjs +0 -131
  64. package/dist/plugins/kv.d.cts +0 -32
  65. package/dist/plugins/kv.d.mts +0 -32
  66. package/dist/plugins/kv.d.ts +0 -32
  67. package/dist/plugins/kv.mjs +0 -124
  68. package/dist/plugins/persistence.cjs +0 -182
  69. package/dist/plugins/persistence.d.cts +0 -50
  70. package/dist/plugins/persistence.d.mts +0 -50
  71. package/dist/plugins/persistence.d.ts +0 -50
  72. package/dist/plugins/persistence.mjs +0 -179
  73. package/dist/shared/indexer.2c23c9cd.mjs +0 -35
  74. package/dist/shared/indexer.318d3617.cjs +0 -47
  75. package/dist/shared/indexer.36530330.mjs +0 -249
  76. package/dist/shared/indexer.500fd281.d.cts +0 -23
  77. package/dist/shared/indexer.541d43eb.cjs +0 -266
  78. package/dist/shared/indexer.93d6b2eb.mjs +0 -17
  79. package/dist/shared/indexer.a8b7ab1f.cjs +0 -25
  80. package/dist/shared/indexer.b9c8f0d8.d.cts +0 -19
  81. package/dist/shared/indexer.b9c8f0d8.d.mts +0 -19
  82. package/dist/shared/indexer.b9c8f0d8.d.ts +0 -19
  83. package/dist/shared/indexer.c7ed6b83.d.cts +0 -82
  84. package/dist/shared/indexer.e1856641.d.mts +0 -23
  85. package/dist/shared/indexer.e4f2430f.d.ts +0 -23
  86. package/dist/shared/indexer.e8bd138d.d.mts +0 -82
  87. package/dist/shared/indexer.f761abcd.d.ts +0 -82
  88. package/dist/sinks/csv.cjs +0 -85
  89. package/dist/sinks/csv.d.cts +0 -66
  90. package/dist/sinks/csv.d.mts +0 -66
  91. package/dist/sinks/csv.d.ts +0 -66
  92. package/dist/sinks/csv.mjs +0 -78
  93. package/dist/sinks/drizzle/index.cjs +0 -212
  94. package/dist/sinks/drizzle/index.d.cts +0 -153
  95. package/dist/sinks/drizzle/index.d.mts +0 -153
  96. package/dist/sinks/drizzle/index.d.ts +0 -153
  97. package/dist/sinks/drizzle/index.mjs +0 -198
  98. package/dist/sinks/sqlite.cjs +0 -90
  99. package/dist/sinks/sqlite.d.cts +0 -71
  100. package/dist/sinks/sqlite.d.mts +0 -71
  101. package/dist/sinks/sqlite.d.ts +0 -71
  102. package/dist/sinks/sqlite.mjs +0 -87
  103. package/src/hooks/index.ts +0 -2
  104. package/src/hooks/useKVStore.ts +0 -12
  105. package/src/hooks/useSink.ts +0 -13
  106. package/src/plugins/kv.test.ts +0 -120
  107. package/src/plugins/kv.ts +0 -132
  108. package/src/plugins/persistence.test.ts +0 -151
  109. package/src/sink.ts +0 -36
  110. package/src/sinks/csv.test.ts +0 -65
  111. package/src/sinks/csv.ts +0 -159
  112. package/src/sinks/drizzle/Int8Range.ts +0 -52
  113. package/src/sinks/drizzle/delete.ts +0 -42
  114. package/src/sinks/drizzle/drizzle.test.ts +0 -239
  115. package/src/sinks/drizzle/drizzle.ts +0 -115
  116. package/src/sinks/drizzle/index.ts +0 -6
  117. package/src/sinks/drizzle/insert.ts +0 -42
  118. package/src/sinks/drizzle/select.ts +0 -44
  119. package/src/sinks/drizzle/transaction.ts +0 -49
  120. package/src/sinks/drizzle/update.ts +0 -47
  121. package/src/sinks/drizzle/utils.ts +0 -99
  122. package/src/sinks/sqlite.test.ts +0 -99
  123. package/src/sinks/sqlite.ts +0 -170
  124. package/src/testing/helper.ts +0 -13
  125. package/src/testing/indexer.ts +0 -35
  126. package/src/testing/setup.ts +0 -59
  127. package/src/testing/vcr.ts +0 -54
@@ -1,153 +0,0 @@
1
- import { Cursor } from '@apibara/protocol';
2
- import { TablesRelationalConfig, ExtractTablesWithRelations, SQL, NotNull, BuildExtraConfigColumns, BuildColumns, Placeholder } from 'drizzle-orm';
3
- import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
4
- import { PgTable, PgQueryResultHKT, PgTransaction, PgUpdateBase, PgCustomColumnBuilder, PgColumnBuilderBase, PgTableExtraConfig, PgTableWithColumns, PgUpdateSetSource, PgSelectBuilder, SelectedFields, TableConfig, PgDatabase, AnyPgTable } from 'drizzle-orm/pg-core';
5
- import { S as Sink, a as SinkCursorParams } from '../../shared/indexer.b9c8f0d8.cjs';
6
- import { Range } from 'postgres-range';
7
-
8
- declare class DrizzleSinkDelete<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
9
- private db;
10
- private table;
11
- private endCursor?;
12
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
13
- where(where: SQL): Omit<PgUpdateBase<TTable, TQueryResult, undefined, false, "where">, "where">;
14
- }
15
-
16
- type Comparable = string | number;
17
- type RangeBound<T extends Comparable> = T | {
18
- value: T;
19
- inclusive: boolean;
20
- };
21
- declare class Int8Range {
22
- readonly range: Range<number>;
23
- constructor(range: Range<number>);
24
- get start(): RangeBound<number> | null;
25
- get end(): RangeBound<number> | null;
26
- }
27
- declare const int8range: {
28
- (): drizzle_orm_pg_core.PgCustomColumnBuilder<{
29
- name: "";
30
- dataType: "custom";
31
- columnType: "PgCustomColumn";
32
- data: Int8Range;
33
- driverParam: unknown;
34
- enumValues: undefined;
35
- generated: undefined;
36
- }>;
37
- <TConfig extends Record<string, any>>(fieldConfig?: TConfig | undefined): drizzle_orm_pg_core.PgCustomColumnBuilder<{
38
- name: "";
39
- dataType: "custom";
40
- columnType: "PgCustomColumn";
41
- data: Int8Range;
42
- driverParam: unknown;
43
- enumValues: undefined;
44
- generated: undefined;
45
- }>;
46
- <TName extends string>(dbName: TName, fieldConfig?: unknown): drizzle_orm_pg_core.PgCustomColumnBuilder<{
47
- name: TName;
48
- dataType: "custom";
49
- columnType: "PgCustomColumn";
50
- data: Int8Range;
51
- driverParam: unknown;
52
- enumValues: undefined;
53
- generated: undefined;
54
- }>;
55
- };
56
-
57
- type CursorColumnBuilder = NotNull<PgCustomColumnBuilder<{
58
- name: "_cursor";
59
- dataType: "custom";
60
- columnType: "PgCustomColumn";
61
- data: Int8Range;
62
- driverParam: undefined;
63
- enumValues: undefined;
64
- generated: undefined;
65
- }>>;
66
- type PgTableWithCursorFn<TSchema extends string | undefined = undefined> = <TTableName extends string, TColumnsMap extends Record<string, PgColumnBuilderBase>>(name: TTableName, columns: TColumnsMap, extraConfig?: (self: BuildExtraConfigColumns<TTableName, TColumnsMap & {
67
- _cursor: CursorColumnBuilder;
68
- }, "pg">) => PgTableExtraConfig) => PgTableWithColumns<{
69
- name: TTableName;
70
- schema: TSchema;
71
- columns: BuildColumns<TTableName, TColumnsMap & {
72
- _cursor: CursorColumnBuilder;
73
- }, "pg">;
74
- dialect: "pg";
75
- }>;
76
- type PgInsertValue<TTable extends PgTable> = Omit<{
77
- [Key in keyof TTable["$inferInsert"]]: TTable["$inferInsert"][Key] | SQL | Placeholder;
78
- } & {}, "_cursor">;
79
- declare const pgTable: PgTableWithCursorFn;
80
- declare const getDrizzleCursor: (cursor_range: [bigint | undefined, bigint | undefined] | bigint | undefined) => Int8Range;
81
-
82
- declare class DrizzleSinkInsert<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
83
- private db;
84
- private table;
85
- private endCursor?;
86
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
87
- values(values: PgInsertValue<TTable> | PgInsertValue<TTable>[]): drizzle_orm_pg_core.PgInsertBase<TTable, TQueryResult, undefined, false, never>;
88
- }
89
-
90
- declare class DrizzleSinkUpdate<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
91
- private db;
92
- private table;
93
- private endCursor?;
94
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
95
- set(values: PgUpdateSetSource<TTable>): PgUpdateBase<TTable, TQueryResult>;
96
- }
97
-
98
- declare class DrizzleSinkTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
99
- private db;
100
- private endCursor?;
101
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, endCursor?: Cursor | undefined);
102
- insert<TTable extends PgTable>(table: TTable): DrizzleSinkInsert<TTable, TQueryResult, TFullSchema, TSchema>;
103
- update<TTable extends PgTable>(table: TTable): DrizzleSinkUpdate<TTable, TQueryResult, TFullSchema, TSchema>;
104
- delete<TTable extends PgTable>(table: TTable): DrizzleSinkDelete<TTable, TQueryResult, TFullSchema, TSchema>;
105
- select(): PgSelectBuilder<undefined>;
106
- select<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
107
- }
108
-
109
- type DrizzleSinkTables<TTableConfig extends Record<string, TableConfig>> = {
110
- [K in keyof TTableConfig]: PgTableWithColumns<TTableConfig[K]>;
111
- };
112
- type DrizzleSinkOptions<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> = {
113
- /**
114
- * Database instance of drizzle-orm
115
- */
116
- database: PgDatabase<TQueryResult, TFullSchema, TSchema>;
117
- tables: AnyPgTable[];
118
- };
119
- /**
120
- * A sink that writes data to a PostgreSQL database using Drizzle ORM.
121
- *
122
- * @example
123
- *
124
- * ```ts
125
- * const sink = drizzle({
126
- * database: db,
127
- * });
128
- *
129
- * ...
130
- * async transform({context, endCursor}){
131
- * const { transaction } = useSink(context);
132
- * const db = transaction(endCursor);
133
- *
134
- * db.insert(users).values([
135
- * { id: 1, name: "John" },
136
- * { id: 2, name: "Jane" },
137
- * ]);
138
- * }
139
- *
140
- * ```
141
- */
142
- declare class DrizzleSink<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> extends Sink {
143
- private _db;
144
- private _tables;
145
- constructor(options: DrizzleSinkOptions<TQueryResult, TFullSchema, TSchema>);
146
- transaction({ cursor, endCursor, finality }: SinkCursorParams, cb: (params: {
147
- db: DrizzleSinkTransaction<TQueryResult, TFullSchema, TSchema>;
148
- }) => Promise<void>): Promise<void>;
149
- invalidate(cursor?: Cursor): Promise<void>;
150
- }
151
- declare const drizzle: <TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>>(args: DrizzleSinkOptions<TQueryResult, TFullSchema, TSchema>) => DrizzleSink<TQueryResult, TFullSchema, TSchema>;
152
-
153
- export { type CursorColumnBuilder, DrizzleSink, DrizzleSinkDelete, type DrizzleSinkOptions, type DrizzleSinkTables, DrizzleSinkTransaction, DrizzleSinkUpdate, Int8Range, type PgInsertValue, type PgTableWithCursorFn, drizzle, getDrizzleCursor, int8range, pgTable };
@@ -1,153 +0,0 @@
1
- import { Cursor } from '@apibara/protocol';
2
- import { TablesRelationalConfig, ExtractTablesWithRelations, SQL, NotNull, BuildExtraConfigColumns, BuildColumns, Placeholder } from 'drizzle-orm';
3
- import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
4
- import { PgTable, PgQueryResultHKT, PgTransaction, PgUpdateBase, PgCustomColumnBuilder, PgColumnBuilderBase, PgTableExtraConfig, PgTableWithColumns, PgUpdateSetSource, PgSelectBuilder, SelectedFields, TableConfig, PgDatabase, AnyPgTable } from 'drizzle-orm/pg-core';
5
- import { S as Sink, a as SinkCursorParams } from '../../shared/indexer.b9c8f0d8.mjs';
6
- import { Range } from 'postgres-range';
7
-
8
- declare class DrizzleSinkDelete<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
9
- private db;
10
- private table;
11
- private endCursor?;
12
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
13
- where(where: SQL): Omit<PgUpdateBase<TTable, TQueryResult, undefined, false, "where">, "where">;
14
- }
15
-
16
- type Comparable = string | number;
17
- type RangeBound<T extends Comparable> = T | {
18
- value: T;
19
- inclusive: boolean;
20
- };
21
- declare class Int8Range {
22
- readonly range: Range<number>;
23
- constructor(range: Range<number>);
24
- get start(): RangeBound<number> | null;
25
- get end(): RangeBound<number> | null;
26
- }
27
- declare const int8range: {
28
- (): drizzle_orm_pg_core.PgCustomColumnBuilder<{
29
- name: "";
30
- dataType: "custom";
31
- columnType: "PgCustomColumn";
32
- data: Int8Range;
33
- driverParam: unknown;
34
- enumValues: undefined;
35
- generated: undefined;
36
- }>;
37
- <TConfig extends Record<string, any>>(fieldConfig?: TConfig | undefined): drizzle_orm_pg_core.PgCustomColumnBuilder<{
38
- name: "";
39
- dataType: "custom";
40
- columnType: "PgCustomColumn";
41
- data: Int8Range;
42
- driverParam: unknown;
43
- enumValues: undefined;
44
- generated: undefined;
45
- }>;
46
- <TName extends string>(dbName: TName, fieldConfig?: unknown): drizzle_orm_pg_core.PgCustomColumnBuilder<{
47
- name: TName;
48
- dataType: "custom";
49
- columnType: "PgCustomColumn";
50
- data: Int8Range;
51
- driverParam: unknown;
52
- enumValues: undefined;
53
- generated: undefined;
54
- }>;
55
- };
56
-
57
- type CursorColumnBuilder = NotNull<PgCustomColumnBuilder<{
58
- name: "_cursor";
59
- dataType: "custom";
60
- columnType: "PgCustomColumn";
61
- data: Int8Range;
62
- driverParam: undefined;
63
- enumValues: undefined;
64
- generated: undefined;
65
- }>>;
66
- type PgTableWithCursorFn<TSchema extends string | undefined = undefined> = <TTableName extends string, TColumnsMap extends Record<string, PgColumnBuilderBase>>(name: TTableName, columns: TColumnsMap, extraConfig?: (self: BuildExtraConfigColumns<TTableName, TColumnsMap & {
67
- _cursor: CursorColumnBuilder;
68
- }, "pg">) => PgTableExtraConfig) => PgTableWithColumns<{
69
- name: TTableName;
70
- schema: TSchema;
71
- columns: BuildColumns<TTableName, TColumnsMap & {
72
- _cursor: CursorColumnBuilder;
73
- }, "pg">;
74
- dialect: "pg";
75
- }>;
76
- type PgInsertValue<TTable extends PgTable> = Omit<{
77
- [Key in keyof TTable["$inferInsert"]]: TTable["$inferInsert"][Key] | SQL | Placeholder;
78
- } & {}, "_cursor">;
79
- declare const pgTable: PgTableWithCursorFn;
80
- declare const getDrizzleCursor: (cursor_range: [bigint | undefined, bigint | undefined] | bigint | undefined) => Int8Range;
81
-
82
- declare class DrizzleSinkInsert<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
83
- private db;
84
- private table;
85
- private endCursor?;
86
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
87
- values(values: PgInsertValue<TTable> | PgInsertValue<TTable>[]): drizzle_orm_pg_core.PgInsertBase<TTable, TQueryResult, undefined, false, never>;
88
- }
89
-
90
- declare class DrizzleSinkUpdate<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
91
- private db;
92
- private table;
93
- private endCursor?;
94
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
95
- set(values: PgUpdateSetSource<TTable>): PgUpdateBase<TTable, TQueryResult>;
96
- }
97
-
98
- declare class DrizzleSinkTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
99
- private db;
100
- private endCursor?;
101
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, endCursor?: Cursor | undefined);
102
- insert<TTable extends PgTable>(table: TTable): DrizzleSinkInsert<TTable, TQueryResult, TFullSchema, TSchema>;
103
- update<TTable extends PgTable>(table: TTable): DrizzleSinkUpdate<TTable, TQueryResult, TFullSchema, TSchema>;
104
- delete<TTable extends PgTable>(table: TTable): DrizzleSinkDelete<TTable, TQueryResult, TFullSchema, TSchema>;
105
- select(): PgSelectBuilder<undefined>;
106
- select<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
107
- }
108
-
109
- type DrizzleSinkTables<TTableConfig extends Record<string, TableConfig>> = {
110
- [K in keyof TTableConfig]: PgTableWithColumns<TTableConfig[K]>;
111
- };
112
- type DrizzleSinkOptions<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> = {
113
- /**
114
- * Database instance of drizzle-orm
115
- */
116
- database: PgDatabase<TQueryResult, TFullSchema, TSchema>;
117
- tables: AnyPgTable[];
118
- };
119
- /**
120
- * A sink that writes data to a PostgreSQL database using Drizzle ORM.
121
- *
122
- * @example
123
- *
124
- * ```ts
125
- * const sink = drizzle({
126
- * database: db,
127
- * });
128
- *
129
- * ...
130
- * async transform({context, endCursor}){
131
- * const { transaction } = useSink(context);
132
- * const db = transaction(endCursor);
133
- *
134
- * db.insert(users).values([
135
- * { id: 1, name: "John" },
136
- * { id: 2, name: "Jane" },
137
- * ]);
138
- * }
139
- *
140
- * ```
141
- */
142
- declare class DrizzleSink<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> extends Sink {
143
- private _db;
144
- private _tables;
145
- constructor(options: DrizzleSinkOptions<TQueryResult, TFullSchema, TSchema>);
146
- transaction({ cursor, endCursor, finality }: SinkCursorParams, cb: (params: {
147
- db: DrizzleSinkTransaction<TQueryResult, TFullSchema, TSchema>;
148
- }) => Promise<void>): Promise<void>;
149
- invalidate(cursor?: Cursor): Promise<void>;
150
- }
151
- declare const drizzle: <TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>>(args: DrizzleSinkOptions<TQueryResult, TFullSchema, TSchema>) => DrizzleSink<TQueryResult, TFullSchema, TSchema>;
152
-
153
- export { type CursorColumnBuilder, DrizzleSink, DrizzleSinkDelete, type DrizzleSinkOptions, type DrizzleSinkTables, DrizzleSinkTransaction, DrizzleSinkUpdate, Int8Range, type PgInsertValue, type PgTableWithCursorFn, drizzle, getDrizzleCursor, int8range, pgTable };
@@ -1,153 +0,0 @@
1
- import { Cursor } from '@apibara/protocol';
2
- import { TablesRelationalConfig, ExtractTablesWithRelations, SQL, NotNull, BuildExtraConfigColumns, BuildColumns, Placeholder } from 'drizzle-orm';
3
- import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
4
- import { PgTable, PgQueryResultHKT, PgTransaction, PgUpdateBase, PgCustomColumnBuilder, PgColumnBuilderBase, PgTableExtraConfig, PgTableWithColumns, PgUpdateSetSource, PgSelectBuilder, SelectedFields, TableConfig, PgDatabase, AnyPgTable } from 'drizzle-orm/pg-core';
5
- import { S as Sink, a as SinkCursorParams } from '../../shared/indexer.b9c8f0d8.js';
6
- import { Range } from 'postgres-range';
7
-
8
- declare class DrizzleSinkDelete<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
9
- private db;
10
- private table;
11
- private endCursor?;
12
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
13
- where(where: SQL): Omit<PgUpdateBase<TTable, TQueryResult, undefined, false, "where">, "where">;
14
- }
15
-
16
- type Comparable = string | number;
17
- type RangeBound<T extends Comparable> = T | {
18
- value: T;
19
- inclusive: boolean;
20
- };
21
- declare class Int8Range {
22
- readonly range: Range<number>;
23
- constructor(range: Range<number>);
24
- get start(): RangeBound<number> | null;
25
- get end(): RangeBound<number> | null;
26
- }
27
- declare const int8range: {
28
- (): drizzle_orm_pg_core.PgCustomColumnBuilder<{
29
- name: "";
30
- dataType: "custom";
31
- columnType: "PgCustomColumn";
32
- data: Int8Range;
33
- driverParam: unknown;
34
- enumValues: undefined;
35
- generated: undefined;
36
- }>;
37
- <TConfig extends Record<string, any>>(fieldConfig?: TConfig | undefined): drizzle_orm_pg_core.PgCustomColumnBuilder<{
38
- name: "";
39
- dataType: "custom";
40
- columnType: "PgCustomColumn";
41
- data: Int8Range;
42
- driverParam: unknown;
43
- enumValues: undefined;
44
- generated: undefined;
45
- }>;
46
- <TName extends string>(dbName: TName, fieldConfig?: unknown): drizzle_orm_pg_core.PgCustomColumnBuilder<{
47
- name: TName;
48
- dataType: "custom";
49
- columnType: "PgCustomColumn";
50
- data: Int8Range;
51
- driverParam: unknown;
52
- enumValues: undefined;
53
- generated: undefined;
54
- }>;
55
- };
56
-
57
- type CursorColumnBuilder = NotNull<PgCustomColumnBuilder<{
58
- name: "_cursor";
59
- dataType: "custom";
60
- columnType: "PgCustomColumn";
61
- data: Int8Range;
62
- driverParam: undefined;
63
- enumValues: undefined;
64
- generated: undefined;
65
- }>>;
66
- type PgTableWithCursorFn<TSchema extends string | undefined = undefined> = <TTableName extends string, TColumnsMap extends Record<string, PgColumnBuilderBase>>(name: TTableName, columns: TColumnsMap, extraConfig?: (self: BuildExtraConfigColumns<TTableName, TColumnsMap & {
67
- _cursor: CursorColumnBuilder;
68
- }, "pg">) => PgTableExtraConfig) => PgTableWithColumns<{
69
- name: TTableName;
70
- schema: TSchema;
71
- columns: BuildColumns<TTableName, TColumnsMap & {
72
- _cursor: CursorColumnBuilder;
73
- }, "pg">;
74
- dialect: "pg";
75
- }>;
76
- type PgInsertValue<TTable extends PgTable> = Omit<{
77
- [Key in keyof TTable["$inferInsert"]]: TTable["$inferInsert"][Key] | SQL | Placeholder;
78
- } & {}, "_cursor">;
79
- declare const pgTable: PgTableWithCursorFn;
80
- declare const getDrizzleCursor: (cursor_range: [bigint | undefined, bigint | undefined] | bigint | undefined) => Int8Range;
81
-
82
- declare class DrizzleSinkInsert<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
83
- private db;
84
- private table;
85
- private endCursor?;
86
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
87
- values(values: PgInsertValue<TTable> | PgInsertValue<TTable>[]): drizzle_orm_pg_core.PgInsertBase<TTable, TQueryResult, undefined, false, never>;
88
- }
89
-
90
- declare class DrizzleSinkUpdate<TTable extends PgTable, TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
91
- private db;
92
- private table;
93
- private endCursor?;
94
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, table: TTable, endCursor?: Cursor | undefined);
95
- set(values: PgUpdateSetSource<TTable>): PgUpdateBase<TTable, TQueryResult>;
96
- }
97
-
98
- declare class DrizzleSinkTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> {
99
- private db;
100
- private endCursor?;
101
- constructor(db: PgTransaction<TQueryResult, TFullSchema, TSchema>, endCursor?: Cursor | undefined);
102
- insert<TTable extends PgTable>(table: TTable): DrizzleSinkInsert<TTable, TQueryResult, TFullSchema, TSchema>;
103
- update<TTable extends PgTable>(table: TTable): DrizzleSinkUpdate<TTable, TQueryResult, TFullSchema, TSchema>;
104
- delete<TTable extends PgTable>(table: TTable): DrizzleSinkDelete<TTable, TQueryResult, TFullSchema, TSchema>;
105
- select(): PgSelectBuilder<undefined>;
106
- select<TSelection extends SelectedFields>(fields: TSelection): PgSelectBuilder<TSelection>;
107
- }
108
-
109
- type DrizzleSinkTables<TTableConfig extends Record<string, TableConfig>> = {
110
- [K in keyof TTableConfig]: PgTableWithColumns<TTableConfig[K]>;
111
- };
112
- type DrizzleSinkOptions<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> = {
113
- /**
114
- * Database instance of drizzle-orm
115
- */
116
- database: PgDatabase<TQueryResult, TFullSchema, TSchema>;
117
- tables: AnyPgTable[];
118
- };
119
- /**
120
- * A sink that writes data to a PostgreSQL database using Drizzle ORM.
121
- *
122
- * @example
123
- *
124
- * ```ts
125
- * const sink = drizzle({
126
- * database: db,
127
- * });
128
- *
129
- * ...
130
- * async transform({context, endCursor}){
131
- * const { transaction } = useSink(context);
132
- * const db = transaction(endCursor);
133
- *
134
- * db.insert(users).values([
135
- * { id: 1, name: "John" },
136
- * { id: 2, name: "Jane" },
137
- * ]);
138
- * }
139
- *
140
- * ```
141
- */
142
- declare class DrizzleSink<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>> extends Sink {
143
- private _db;
144
- private _tables;
145
- constructor(options: DrizzleSinkOptions<TQueryResult, TFullSchema, TSchema>);
146
- transaction({ cursor, endCursor, finality }: SinkCursorParams, cb: (params: {
147
- db: DrizzleSinkTransaction<TQueryResult, TFullSchema, TSchema>;
148
- }) => Promise<void>): Promise<void>;
149
- invalidate(cursor?: Cursor): Promise<void>;
150
- }
151
- declare const drizzle: <TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TSchema extends TablesRelationalConfig = ExtractTablesWithRelations<TFullSchema>>(args: DrizzleSinkOptions<TQueryResult, TFullSchema, TSchema>) => DrizzleSink<TQueryResult, TFullSchema, TSchema>;
152
-
153
- export { type CursorColumnBuilder, DrizzleSink, DrizzleSinkDelete, type DrizzleSinkOptions, type DrizzleSinkTables, DrizzleSinkTransaction, DrizzleSinkUpdate, Int8Range, type PgInsertValue, type PgTableWithCursorFn, drizzle, getDrizzleCursor, int8range, pgTable };
@@ -1,198 +0,0 @@
1
- import { sql, gt } from 'drizzle-orm';
2
- import { S as Sink } from '../../shared/indexer.93d6b2eb.mjs';
3
- import { customType, pgTable as pgTable$1 } from 'drizzle-orm/pg-core';
4
- import range, { parse, serialize } from 'postgres-range';
5
- import 'consola';
6
-
7
- class DrizzleSinkDelete {
8
- constructor(db, table, endCursor) {
9
- this.db = db;
10
- this.table = table;
11
- this.endCursor = endCursor;
12
- }
13
- //@ts-ignore
14
- where(where) {
15
- return this.db.update(this.table).set({
16
- // @ts-ignore
17
- _cursor: sql`int8range(lower(_cursor), ${Number(this.endCursor?.orderKey)}, '[)')`
18
- }).where(where);
19
- }
20
- }
21
-
22
- class Int8Range {
23
- constructor(range) {
24
- this.range = range;
25
- }
26
- get start() {
27
- return this.range.lower != null ? {
28
- value: this.range.lower,
29
- inclusive: this.range.isLowerBoundClosed()
30
- } : null;
31
- }
32
- get end() {
33
- return this.range.upper != null ? {
34
- value: this.range.upper,
35
- inclusive: this.range.isUpperBoundClosed()
36
- } : null;
37
- }
38
- }
39
- const int8range = customType({
40
- dataType: () => "int8range",
41
- fromDriver: (value) => {
42
- if (typeof value !== "string") {
43
- throw new Error("Expected string");
44
- }
45
- const parsed = parse(value, (val) => Number.parseInt(val, 10));
46
- return new Int8Range(parsed);
47
- },
48
- toDriver: (value) => serialize(value.range)
49
- });
50
-
51
- const pgTable = (name, columns, extraConfig) => {
52
- return pgTable$1(
53
- name,
54
- {
55
- ...columns,
56
- _cursor: int8range("_cursor").notNull()
57
- },
58
- extraConfig
59
- );
60
- };
61
- const getDrizzleCursor = (cursor_range) => {
62
- const isArray = Array.isArray(cursor_range);
63
- const [lower, upper] = isArray ? cursor_range : [cursor_range, void 0];
64
- let isNoUpperBound = false;
65
- if (!lower) {
66
- throw new Error("Lower bound cursor is required");
67
- }
68
- if (!upper) {
69
- isNoUpperBound = true;
70
- }
71
- return new Int8Range(
72
- new range.Range(
73
- Number(lower),
74
- Number(upper),
75
- range.RANGE_LB_INC | (isNoUpperBound ? range.RANGE_UB_INF : 0)
76
- )
77
- );
78
- };
79
-
80
- class DrizzleSinkInsert {
81
- constructor(db, table, endCursor) {
82
- this.db = db;
83
- this.table = table;
84
- this.endCursor = endCursor;
85
- }
86
- values(values) {
87
- const originalInsert = this.db.insert(this.table);
88
- const cursoredValues = (Array.isArray(values) ? values : [values]).map(
89
- (v) => {
90
- return {
91
- ...v,
92
- _cursor: getDrizzleCursor(this.endCursor?.orderKey)
93
- };
94
- }
95
- );
96
- return originalInsert.values(
97
- cursoredValues
98
- );
99
- }
100
- }
101
-
102
- class DrizzleSinkSelect {
103
- constructor(db, fields, endCursor) {
104
- this.db = db;
105
- this.fields = fields;
106
- this.endCursor = endCursor;
107
- }
108
- from(source) {
109
- if (this.fields) {
110
- const originalFrom = this.db.select(this.fields).from(source);
111
- return {
112
- ...originalFrom,
113
- where: (where) => {
114
- const combinedWhere = sql`${where ? sql`${where} AND ` : sql``}upper_inf(_cursor)`;
115
- return originalFrom.where(combinedWhere);
116
- }
117
- };
118
- }
119
- return this.db.select().from(source).where(sql`upper_inf(_cursor)`);
120
- }
121
- }
122
-
123
- class DrizzleSinkUpdate {
124
- constructor(db, table, endCursor) {
125
- this.db = db;
126
- this.table = table;
127
- this.endCursor = endCursor;
128
- }
129
- set(values) {
130
- const originalUpdate = this.db.update(this.table);
131
- const originalSet = originalUpdate.set(values);
132
- return {
133
- ...originalSet,
134
- where: async (where) => {
135
- await this.db.update(this.table).set({
136
- _cursor: sql`int8range(lower(_cursor), ${Number(this.endCursor?.orderKey)}, '[)')`
137
- }).where(sql`${where ? sql`${where} AND ` : sql``}upper_inf(_cursor)`).execute();
138
- return originalSet.where(where);
139
- }
140
- };
141
- }
142
- }
143
-
144
- class DrizzleSinkTransaction {
145
- constructor(db, endCursor) {
146
- this.db = db;
147
- this.endCursor = endCursor;
148
- }
149
- insert(table) {
150
- return new DrizzleSinkInsert(this.db, table, this.endCursor);
151
- }
152
- update(table) {
153
- return new DrizzleSinkUpdate(this.db, table, this.endCursor);
154
- }
155
- delete(table) {
156
- return new DrizzleSinkDelete(this.db, table, this.endCursor);
157
- }
158
- select(fields) {
159
- return new DrizzleSinkSelect(this.db, fields, this.endCursor);
160
- }
161
- }
162
-
163
- var __defProp = Object.defineProperty;
164
- var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
165
- var __publicField = (obj, key, value) => {
166
- __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
167
- return value;
168
- };
169
- class DrizzleSink extends Sink {
170
- constructor(options) {
171
- super();
172
- __publicField(this, "_db");
173
- __publicField(this, "_tables");
174
- const { database, tables } = options;
175
- this._db = database;
176
- this._tables = tables;
177
- }
178
- async transaction({ cursor, endCursor, finality }, cb) {
179
- await this._db.transaction(async (db) => {
180
- await cb({ db: new DrizzleSinkTransaction(db, endCursor) });
181
- });
182
- }
183
- async invalidate(cursor) {
184
- await this._db.transaction(async (db) => {
185
- for (const table of this._tables) {
186
- await db.delete(table).where(gt(sql`lower(_cursor)`, sql`${Number(cursor?.orderKey)}`)).returning();
187
- await db.update(table).set({
188
- _cursor: sql`int8range(lower(_cursor), NULL, '[)')`
189
- }).where(gt(sql`upper(_cursor)`, sql`${Number(cursor?.orderKey)}`));
190
- }
191
- });
192
- }
193
- }
194
- const drizzle = (args) => {
195
- return new DrizzleSink(args);
196
- };
197
-
198
- export { DrizzleSink, DrizzleSinkDelete, DrizzleSinkTransaction, DrizzleSinkUpdate, Int8Range, drizzle, getDrizzleCursor, int8range, pgTable };