@cat-factory/node-server 0.6.0
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.
- package/LICENSE +21 -0
- package/dist/config.d.ts +3 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +297 -0
- package/dist/config.js.map +1 -0
- package/dist/container.d.ts +88 -0
- package/dist/container.d.ts.map +1 -0
- package/dist/container.js +937 -0
- package/dist/container.js.map +1 -0
- package/dist/db/client.d.ts +13 -0
- package/dist/db/client.d.ts.map +1 -0
- package/dist/db/client.js +21 -0
- package/dist/db/client.js.map +1 -0
- package/dist/db/migrate.d.ts +12 -0
- package/dist/db/migrate.d.ts.map +1 -0
- package/dist/db/migrate.js +40 -0
- package/dist/db/migrate.js.map +1 -0
- package/dist/db/schema.d.ts +7858 -0
- package/dist/db/schema.d.ts.map +1 -0
- package/dist/db/schema.js +928 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/environments.d.ts +11 -0
- package/dist/environments.d.ts.map +1 -0
- package/dist/environments.js +31 -0
- package/dist/environments.js.map +1 -0
- package/dist/execution/bootstrapRunner.d.ts +27 -0
- package/dist/execution/bootstrapRunner.d.ts.map +1 -0
- package/dist/execution/bootstrapRunner.js +79 -0
- package/dist/execution/bootstrapRunner.js.map +1 -0
- package/dist/execution/config.d.ts +37 -0
- package/dist/execution/config.d.ts.map +1 -0
- package/dist/execution/config.js +86 -0
- package/dist/execution/config.js.map +1 -0
- package/dist/execution/drive.d.ts +6 -0
- package/dist/execution/drive.d.ts.map +1 -0
- package/dist/execution/drive.js +13 -0
- package/dist/execution/drive.js.map +1 -0
- package/dist/execution/pgBossRunner.d.ts +82 -0
- package/dist/execution/pgBossRunner.d.ts.map +1 -0
- package/dist/execution/pgBossRunner.js +163 -0
- package/dist/execution/pgBossRunner.js.map +1 -0
- package/dist/gateways.d.ts +4 -0
- package/dist/gateways.d.ts.map +1 -0
- package/dist/gateways.js +91 -0
- package/dist/gateways.js.map +1 -0
- package/dist/index.d.ts +13 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +22 -0
- package/dist/index.js.map +1 -0
- package/dist/main.d.ts +2 -0
- package/dist/main.d.ts.map +1 -0
- package/dist/main.js +9 -0
- package/dist/main.js.map +1 -0
- package/dist/modelProvider.d.ts +6 -0
- package/dist/modelProvider.d.ts.map +1 -0
- package/dist/modelProvider.js +72 -0
- package/dist/modelProvider.js.map +1 -0
- package/dist/realtime.d.ts +62 -0
- package/dist/realtime.d.ts.map +1 -0
- package/dist/realtime.js +171 -0
- package/dist/realtime.js.map +1 -0
- package/dist/recurring.d.ts +11 -0
- package/dist/recurring.d.ts.map +1 -0
- package/dist/recurring.js +33 -0
- package/dist/recurring.js.map +1 -0
- package/dist/repositories/bootstrap.d.ts +25 -0
- package/dist/repositories/bootstrap.d.ts.map +1 -0
- package/dist/repositories/bootstrap.js +280 -0
- package/dist/repositories/bootstrap.js.map +1 -0
- package/dist/repositories/containerExecution.d.ts +33 -0
- package/dist/repositories/containerExecution.d.ts.map +1 -0
- package/dist/repositories/containerExecution.js +199 -0
- package/dist/repositories/containerExecution.js.map +1 -0
- package/dist/repositories/documents.d.ts +31 -0
- package/dist/repositories/documents.d.ts.map +1 -0
- package/dist/repositories/documents.js +176 -0
- package/dist/repositories/documents.js.map +1 -0
- package/dist/repositories/drizzle.d.ts +105 -0
- package/dist/repositories/drizzle.d.ts.map +1 -0
- package/dist/repositories/drizzle.js +1872 -0
- package/dist/repositories/drizzle.js.map +1 -0
- package/dist/repositories/environments.d.ts +23 -0
- package/dist/repositories/environments.d.ts.map +1 -0
- package/dist/repositories/environments.js +162 -0
- package/dist/repositories/environments.js.map +1 -0
- package/dist/repositories/fragments.d.ts +23 -0
- package/dist/repositories/fragments.d.ts.map +1 -0
- package/dist/repositories/fragments.js +190 -0
- package/dist/repositories/fragments.js.map +1 -0
- package/dist/repositories/github.d.ts +53 -0
- package/dist/repositories/github.d.ts.map +1 -0
- package/dist/repositories/github.js +441 -0
- package/dist/repositories/github.js.map +1 -0
- package/dist/repositories/localModelEndpoint.d.ts +12 -0
- package/dist/repositories/localModelEndpoint.d.ts.map +1 -0
- package/dist/repositories/localModelEndpoint.js +75 -0
- package/dist/repositories/localModelEndpoint.js.map +1 -0
- package/dist/repositories/notifications.d.ts +11 -0
- package/dist/repositories/notifications.d.ts.map +1 -0
- package/dist/repositories/notifications.js +88 -0
- package/dist/repositories/notifications.js.map +1 -0
- package/dist/repositories/personalSubscription.d.ts +22 -0
- package/dist/repositories/personalSubscription.d.ts.map +1 -0
- package/dist/repositories/personalSubscription.js +159 -0
- package/dist/repositories/personalSubscription.js.map +1 -0
- package/dist/repositories/providerApiKey.d.ts +18 -0
- package/dist/repositories/providerApiKey.d.ts.map +1 -0
- package/dist/repositories/providerApiKey.js +111 -0
- package/dist/repositories/providerApiKey.js.map +1 -0
- package/dist/repositories/providerSubscription.d.ts +16 -0
- package/dist/repositories/providerSubscription.d.ts.map +1 -0
- package/dist/repositories/providerSubscription.js +88 -0
- package/dist/repositories/providerSubscription.js.map +1 -0
- package/dist/repositories/slack.d.ts +23 -0
- package/dist/repositories/slack.d.ts.map +1 -0
- package/dist/repositories/slack.js +150 -0
- package/dist/repositories/slack.js.map +1 -0
- package/dist/repositories/tasks.d.ts +24 -0
- package/dist/repositories/tasks.d.ts.map +1 -0
- package/dist/repositories/tasks.js +194 -0
- package/dist/repositories/tasks.js.map +1 -0
- package/dist/retention.d.ts +38 -0
- package/dist/retention.d.ts.map +1 -0
- package/dist/retention.js +53 -0
- package/dist/retention.js.map +1 -0
- package/dist/runtime.d.ts +10 -0
- package/dist/runtime.d.ts.map +1 -0
- package/dist/runtime.js +13 -0
- package/dist/runtime.js.map +1 -0
- package/dist/server.d.ts +41 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +138 -0
- package/dist/server.js.map +1 -0
- package/dist/tasks/JiraProvider.d.ts +27 -0
- package/dist/tasks/JiraProvider.d.ts.map +1 -0
- package/dist/tasks/JiraProvider.js +79 -0
- package/dist/tasks/JiraProvider.js.map +1 -0
- package/drizzle/20260622175812_flashy_maginty/migration.sql +689 -0
- package/drizzle/20260622175812_flashy_maginty/snapshot.json +8318 -0
- package/drizzle/20260623172634_loud_wallop/migration.sql +11 -0
- package/drizzle/20260623172634_loud_wallop/snapshot.json +8439 -0
- package/drizzle/20260623174706_acoustic_zemo/migration.sql +16 -0
- package/drizzle/20260623174706_acoustic_zemo/snapshot.json +8506 -0
- package/drizzle/20260623184400_silent_cardiac/migration.sql +24 -0
- package/drizzle/20260623184400_silent_cardiac/snapshot.json +8639 -0
- package/drizzle/20260623205323_quick_arclight/migration.sql +1 -0
- package/drizzle/20260623205323_quick_arclight/snapshot.json +8963 -0
- package/drizzle/20260623221910_black_zombie/migration.sql +22 -0
- package/drizzle/20260623221910_black_zombie/snapshot.json +9189 -0
- package/drizzle/20260624131343_far_lily_hollister/migration.sql +3 -0
- package/drizzle/20260624131343_far_lily_hollister/snapshot.json +9228 -0
- package/drizzle/20260624135452_tiny_norman_osborn/migration.sql +11 -0
- package/drizzle/20260624135452_tiny_norman_osborn/snapshot.json +9126 -0
- package/drizzle/20260624140138_wandering_avengers/migration.sql +1 -0
- package/drizzle/20260624140138_wandering_avengers/snapshot.json +9045 -0
- package/package.json +62 -0
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
import { and, desc, eq } from 'drizzle-orm';
|
|
2
|
+
import { notifications } from '../db/schema.js';
|
|
3
|
+
function rowToNotification(row) {
|
|
4
|
+
let payload = null;
|
|
5
|
+
if (row.payload) {
|
|
6
|
+
try {
|
|
7
|
+
payload = JSON.parse(row.payload);
|
|
8
|
+
}
|
|
9
|
+
catch {
|
|
10
|
+
payload = null;
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
return {
|
|
14
|
+
id: row.id,
|
|
15
|
+
type: row.type,
|
|
16
|
+
status: row.status,
|
|
17
|
+
blockId: row.block_id,
|
|
18
|
+
executionId: row.execution_id,
|
|
19
|
+
title: row.title,
|
|
20
|
+
body: row.body,
|
|
21
|
+
payload,
|
|
22
|
+
createdAt: row.created_at,
|
|
23
|
+
resolvedAt: row.resolved_at,
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
export class DrizzleNotificationRepository {
|
|
27
|
+
db;
|
|
28
|
+
constructor(db) {
|
|
29
|
+
this.db = db;
|
|
30
|
+
}
|
|
31
|
+
async get(workspaceId, id) {
|
|
32
|
+
const rows = await this.db
|
|
33
|
+
.select()
|
|
34
|
+
.from(notifications)
|
|
35
|
+
.where(and(eq(notifications.workspace_id, workspaceId), eq(notifications.id, id)))
|
|
36
|
+
.limit(1);
|
|
37
|
+
return rows[0] ? rowToNotification(rows[0]) : null;
|
|
38
|
+
}
|
|
39
|
+
async listOpen(workspaceId) {
|
|
40
|
+
const rows = await this.db
|
|
41
|
+
.select()
|
|
42
|
+
.from(notifications)
|
|
43
|
+
.where(and(eq(notifications.workspace_id, workspaceId), eq(notifications.status, 'open')))
|
|
44
|
+
.orderBy(desc(notifications.created_at));
|
|
45
|
+
return rows.map(rowToNotification);
|
|
46
|
+
}
|
|
47
|
+
async findOpenByBlock(workspaceId, blockId, type) {
|
|
48
|
+
const rows = await this.db
|
|
49
|
+
.select()
|
|
50
|
+
.from(notifications)
|
|
51
|
+
.where(and(eq(notifications.workspace_id, workspaceId), eq(notifications.block_id, blockId), eq(notifications.type, type), eq(notifications.status, 'open')))
|
|
52
|
+
.orderBy(desc(notifications.created_at))
|
|
53
|
+
.limit(1);
|
|
54
|
+
return rows[0] ? rowToNotification(rows[0]) : null;
|
|
55
|
+
}
|
|
56
|
+
async upsert(workspaceId, notification) {
|
|
57
|
+
const values = {
|
|
58
|
+
workspace_id: workspaceId,
|
|
59
|
+
id: notification.id,
|
|
60
|
+
type: notification.type,
|
|
61
|
+
status: notification.status,
|
|
62
|
+
block_id: notification.blockId,
|
|
63
|
+
execution_id: notification.executionId,
|
|
64
|
+
title: notification.title,
|
|
65
|
+
body: notification.body,
|
|
66
|
+
payload: notification.payload ? JSON.stringify(notification.payload) : null,
|
|
67
|
+
created_at: notification.createdAt,
|
|
68
|
+
resolved_at: notification.resolvedAt,
|
|
69
|
+
};
|
|
70
|
+
await this.db
|
|
71
|
+
.insert(notifications)
|
|
72
|
+
.values(values)
|
|
73
|
+
.onConflictDoUpdate({
|
|
74
|
+
target: [notifications.workspace_id, notifications.id],
|
|
75
|
+
set: {
|
|
76
|
+
type: values.type,
|
|
77
|
+
status: values.status,
|
|
78
|
+
block_id: values.block_id,
|
|
79
|
+
execution_id: values.execution_id,
|
|
80
|
+
title: values.title,
|
|
81
|
+
body: values.body,
|
|
82
|
+
payload: values.payload,
|
|
83
|
+
resolved_at: values.resolved_at,
|
|
84
|
+
},
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
//# sourceMappingURL=notifications.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"notifications.js","sourceRoot":"","sources":["../../src/repositories/notifications.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAE3C,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAA;AAU/C,SAAS,iBAAiB,CAAC,GAAoB;IAC7C,IAAI,OAAO,GAA+B,IAAI,CAAA;IAC9C,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAwB,CAAA;QAC1D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,IAAI,CAAA;QAChB,CAAC;IACH,CAAC;IACD,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,IAAI,EAAE,GAAG,CAAC,IAAwB;QAClC,MAAM,EAAE,GAAG,CAAC,MAAgC;QAC5C,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,OAAO;QACP,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,UAAU,EAAE,GAAG,CAAC,WAAW;KAC5B,CAAA;AACH,CAAC;AAED,MAAM,OAAO,6BAA6B;IACX,EAAE;IAA/B,YAA6B,EAAa;kBAAb,EAAE;IAAc,CAAC;IAE9C,KAAK,CAAC,GAAG,CAAC,WAAmB,EAAE,EAAU;QACvC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;aACjF,KAAK,CAAC,CAAC,CAAC,CAAA;QACX,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IACpD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,WAAmB;QAChC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;aACzF,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAA;QAC1C,OAAO,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAA;IACpC,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,WAAmB,EACnB,OAAe,EACf,IAAsB;QAEtB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,aAAa,CAAC;aACnB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,WAAW,CAAC,EAC3C,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,EACnC,EAAE,CAAC,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,EAC5B,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CACjC,CACF;aACA,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;aACvC,KAAK,CAAC,CAAC,CAAC,CAAA;QACX,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IACpD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,WAAmB,EAAE,YAA0B;QAC1D,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,WAAW;YACzB,EAAE,EAAE,YAAY,CAAC,EAAE;YACnB,IAAI,EAAE,YAAY,CAAC,IAAI;YACvB,MAAM,EAAE,YAAY,CAAC,MAAM;YAC3B,QAAQ,EAAE,YAAY,CAAC,OAAO;YAC9B,YAAY,EAAE,YAAY,CAAC,WAAW;YACtC,KAAK,EAAE,YAAY,CAAC,KAAK;YACzB,IAAI,EAAE,YAAY,CAAC,IAAI;YACvB,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI;YAC3E,UAAU,EAAE,YAAY,CAAC,SAAS;YAClC,WAAW,EAAE,YAAY,CAAC,UAAU;SACrC,CAAA;QACD,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,aAAa,CAAC;aACrB,MAAM,CAAC,MAAM,CAAC;aACd,kBAAkB,CAAC;YAClB,MAAM,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,aAAa,CAAC,EAAE,CAAC;YACtD,GAAG,EAAE;gBACH,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC;SACF,CAAC,CAAA;IACN,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { PersonalSubscriptionRecord, PersonalSubscriptionRepository, SubscriptionActivationRecord, SubscriptionActivationRepository, SubscriptionVendor } from '@cat-factory/kernel';
|
|
2
|
+
import type { DrizzleDb } from '../db/client.js';
|
|
3
|
+
export declare class DrizzlePersonalSubscriptionRepository implements PersonalSubscriptionRepository {
|
|
4
|
+
private readonly db;
|
|
5
|
+
constructor(db: DrizzleDb);
|
|
6
|
+
getByUserVendor(userId: string, vendor: SubscriptionVendor): Promise<PersonalSubscriptionRecord | null>;
|
|
7
|
+
listByUser(userId: string): Promise<PersonalSubscriptionRecord[]>;
|
|
8
|
+
upsert(record: PersonalSubscriptionRecord): Promise<void>;
|
|
9
|
+
markUsed(userId: string, vendor: SubscriptionVendor, at: number): Promise<void>;
|
|
10
|
+
softDelete(userId: string, vendor: SubscriptionVendor, at: number): Promise<void>;
|
|
11
|
+
listExpiring(now: number, before: number): Promise<PersonalSubscriptionRecord[]>;
|
|
12
|
+
}
|
|
13
|
+
export declare class DrizzleSubscriptionActivationRepository implements SubscriptionActivationRepository {
|
|
14
|
+
private readonly db;
|
|
15
|
+
constructor(db: DrizzleDb);
|
|
16
|
+
get(executionId: string, userId: string, vendor: SubscriptionVendor, now: number): Promise<SubscriptionActivationRecord | null>;
|
|
17
|
+
upsert(record: SubscriptionActivationRecord): Promise<void>;
|
|
18
|
+
refresh(executionId: string, userId: string, vendor: SubscriptionVendor, expiresAt: number): Promise<void>;
|
|
19
|
+
deleteByExecution(executionId: string): Promise<void>;
|
|
20
|
+
deleteExpired(now: number): Promise<number>;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=personalSubscription.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"personalSubscription.d.ts","sourceRoot":"","sources":["../../src/repositories/personalSubscription.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,0BAA0B,EAC1B,8BAA8B,EAC9B,4BAA4B,EAC5B,gCAAgC,EAChC,kBAAkB,EACnB,MAAM,qBAAqB,CAAA;AAE5B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAwBhD,qBAAa,qCAAsC,YAAW,8BAA8B;IAC9E,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,eAAe,CACnB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,kBAAkB,GACzB,OAAO,CAAC,0BAA0B,GAAG,IAAI,CAAC,CAa5C;IAEK,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,0BAA0B,EAAE,CAAC,CAStE;IAEK,MAAM,CAAC,MAAM,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC,CAqC9D;IAEK,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAWpF;IAEK,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAWtF;IAEK,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,0BAA0B,EAAE,CAAC,CAarF;CACF;AAgBD,qBAAa,uCAAwC,YAAW,gCAAgC;IAClF,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,GAAG,CACP,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,kBAAkB,EAC1B,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,4BAA4B,GAAG,IAAI,CAAC,CAc9C;IAEK,MAAM,CAAC,MAAM,EAAE,4BAA4B,GAAG,OAAO,CAAC,IAAI,CAAC,CAyBhE;IAEK,OAAO,CACX,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,kBAAkB,EAC1B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,IAAI,CAAC,CAWf;IAEK,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAI1D;IAEK,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAMhD;CACF"}
|
|
@@ -0,0 +1,159 @@
|
|
|
1
|
+
import { and, asc, eq, gt, gte, isNull, lte, ne } from 'drizzle-orm';
|
|
2
|
+
import { personalSubscriptions, subscriptionActivations } from '../db/schema.js';
|
|
3
|
+
function toRecord(row) {
|
|
4
|
+
return {
|
|
5
|
+
id: row.id,
|
|
6
|
+
userId: row.user_id,
|
|
7
|
+
vendor: row.vendor,
|
|
8
|
+
label: row.label,
|
|
9
|
+
tokenCipher: row.token_cipher,
|
|
10
|
+
expiresAt: row.expires_at,
|
|
11
|
+
createdAt: row.created_at,
|
|
12
|
+
updatedAt: row.updated_at,
|
|
13
|
+
lastUsedAt: row.last_used_at,
|
|
14
|
+
deletedAt: row.deleted_at,
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
export class DrizzlePersonalSubscriptionRepository {
|
|
18
|
+
db;
|
|
19
|
+
constructor(db) {
|
|
20
|
+
this.db = db;
|
|
21
|
+
}
|
|
22
|
+
async getByUserVendor(userId, vendor) {
|
|
23
|
+
const rows = await this.db
|
|
24
|
+
.select()
|
|
25
|
+
.from(personalSubscriptions)
|
|
26
|
+
.where(and(eq(personalSubscriptions.user_id, userId), eq(personalSubscriptions.vendor, vendor), isNull(personalSubscriptions.deleted_at)))
|
|
27
|
+
.limit(1);
|
|
28
|
+
return rows[0] ? toRecord(rows[0]) : null;
|
|
29
|
+
}
|
|
30
|
+
async listByUser(userId) {
|
|
31
|
+
const rows = await this.db
|
|
32
|
+
.select()
|
|
33
|
+
.from(personalSubscriptions)
|
|
34
|
+
.where(and(eq(personalSubscriptions.user_id, userId), isNull(personalSubscriptions.deleted_at)))
|
|
35
|
+
.orderBy(asc(personalSubscriptions.created_at));
|
|
36
|
+
return rows.map(toRecord);
|
|
37
|
+
}
|
|
38
|
+
async upsert(record) {
|
|
39
|
+
// One live row per user+vendor: tombstone any other live row, then upsert by id.
|
|
40
|
+
await this.db
|
|
41
|
+
.update(personalSubscriptions)
|
|
42
|
+
.set({ deleted_at: record.updatedAt })
|
|
43
|
+
.where(and(eq(personalSubscriptions.user_id, record.userId), eq(personalSubscriptions.vendor, record.vendor), isNull(personalSubscriptions.deleted_at), ne(personalSubscriptions.id, record.id)));
|
|
44
|
+
await this.db
|
|
45
|
+
.insert(personalSubscriptions)
|
|
46
|
+
.values({
|
|
47
|
+
id: record.id,
|
|
48
|
+
user_id: record.userId,
|
|
49
|
+
vendor: record.vendor,
|
|
50
|
+
label: record.label,
|
|
51
|
+
token_cipher: record.tokenCipher,
|
|
52
|
+
expires_at: record.expiresAt,
|
|
53
|
+
created_at: record.createdAt,
|
|
54
|
+
updated_at: record.updatedAt,
|
|
55
|
+
last_used_at: record.lastUsedAt,
|
|
56
|
+
deleted_at: null,
|
|
57
|
+
})
|
|
58
|
+
.onConflictDoUpdate({
|
|
59
|
+
target: personalSubscriptions.id,
|
|
60
|
+
set: {
|
|
61
|
+
label: record.label,
|
|
62
|
+
token_cipher: record.tokenCipher,
|
|
63
|
+
expires_at: record.expiresAt,
|
|
64
|
+
updated_at: record.updatedAt,
|
|
65
|
+
deleted_at: null,
|
|
66
|
+
},
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
async markUsed(userId, vendor, at) {
|
|
70
|
+
await this.db
|
|
71
|
+
.update(personalSubscriptions)
|
|
72
|
+
.set({ last_used_at: at })
|
|
73
|
+
.where(and(eq(personalSubscriptions.user_id, userId), eq(personalSubscriptions.vendor, vendor), isNull(personalSubscriptions.deleted_at)));
|
|
74
|
+
}
|
|
75
|
+
async softDelete(userId, vendor, at) {
|
|
76
|
+
await this.db
|
|
77
|
+
.update(personalSubscriptions)
|
|
78
|
+
.set({ deleted_at: at })
|
|
79
|
+
.where(and(eq(personalSubscriptions.user_id, userId), eq(personalSubscriptions.vendor, vendor), isNull(personalSubscriptions.deleted_at)));
|
|
80
|
+
}
|
|
81
|
+
async listExpiring(now, before) {
|
|
82
|
+
const rows = await this.db
|
|
83
|
+
.select()
|
|
84
|
+
.from(personalSubscriptions)
|
|
85
|
+
.where(and(isNull(personalSubscriptions.deleted_at), gte(personalSubscriptions.expires_at, now), lte(personalSubscriptions.expires_at, before)))
|
|
86
|
+
.orderBy(asc(personalSubscriptions.expires_at));
|
|
87
|
+
return rows.map(toRecord);
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
function toActivation(row) {
|
|
91
|
+
return {
|
|
92
|
+
id: row.id,
|
|
93
|
+
executionId: row.execution_id,
|
|
94
|
+
userId: row.user_id,
|
|
95
|
+
vendor: row.vendor,
|
|
96
|
+
tokenCipher: row.token_cipher,
|
|
97
|
+
createdAt: row.created_at,
|
|
98
|
+
expiresAt: row.expires_at,
|
|
99
|
+
};
|
|
100
|
+
}
|
|
101
|
+
export class DrizzleSubscriptionActivationRepository {
|
|
102
|
+
db;
|
|
103
|
+
constructor(db) {
|
|
104
|
+
this.db = db;
|
|
105
|
+
}
|
|
106
|
+
async get(executionId, userId, vendor, now) {
|
|
107
|
+
const rows = await this.db
|
|
108
|
+
.select()
|
|
109
|
+
.from(subscriptionActivations)
|
|
110
|
+
.where(and(eq(subscriptionActivations.execution_id, executionId), eq(subscriptionActivations.user_id, userId), eq(subscriptionActivations.vendor, vendor), gt(subscriptionActivations.expires_at, now)))
|
|
111
|
+
.limit(1);
|
|
112
|
+
return rows[0] ? toActivation(rows[0]) : null;
|
|
113
|
+
}
|
|
114
|
+
async upsert(record) {
|
|
115
|
+
await this.db
|
|
116
|
+
.insert(subscriptionActivations)
|
|
117
|
+
.values({
|
|
118
|
+
id: record.id,
|
|
119
|
+
execution_id: record.executionId,
|
|
120
|
+
user_id: record.userId,
|
|
121
|
+
vendor: record.vendor,
|
|
122
|
+
token_cipher: record.tokenCipher,
|
|
123
|
+
created_at: record.createdAt,
|
|
124
|
+
expires_at: record.expiresAt,
|
|
125
|
+
})
|
|
126
|
+
.onConflictDoUpdate({
|
|
127
|
+
// Matches the (execution_id, user_id, vendor) unique index.
|
|
128
|
+
target: [
|
|
129
|
+
subscriptionActivations.execution_id,
|
|
130
|
+
subscriptionActivations.user_id,
|
|
131
|
+
subscriptionActivations.vendor,
|
|
132
|
+
],
|
|
133
|
+
set: {
|
|
134
|
+
token_cipher: record.tokenCipher,
|
|
135
|
+
created_at: record.createdAt,
|
|
136
|
+
expires_at: record.expiresAt,
|
|
137
|
+
},
|
|
138
|
+
});
|
|
139
|
+
}
|
|
140
|
+
async refresh(executionId, userId, vendor, expiresAt) {
|
|
141
|
+
await this.db
|
|
142
|
+
.update(subscriptionActivations)
|
|
143
|
+
.set({ expires_at: expiresAt })
|
|
144
|
+
.where(and(eq(subscriptionActivations.execution_id, executionId), eq(subscriptionActivations.user_id, userId), eq(subscriptionActivations.vendor, vendor)));
|
|
145
|
+
}
|
|
146
|
+
async deleteByExecution(executionId) {
|
|
147
|
+
await this.db
|
|
148
|
+
.delete(subscriptionActivations)
|
|
149
|
+
.where(eq(subscriptionActivations.execution_id, executionId));
|
|
150
|
+
}
|
|
151
|
+
async deleteExpired(now) {
|
|
152
|
+
const deleted = await this.db
|
|
153
|
+
.delete(subscriptionActivations)
|
|
154
|
+
.where(lte(subscriptionActivations.expires_at, now))
|
|
155
|
+
.returning({ id: subscriptionActivations.id });
|
|
156
|
+
return deleted.length;
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
//# sourceMappingURL=personalSubscription.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"personalSubscription.js","sourceRoot":"","sources":["../../src/repositories/personalSubscription.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAEpE,OAAO,EAAE,qBAAqB,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAA;AAQhF,SAAS,QAAQ,CAAC,GAAW;IAC3B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,MAAM,EAAE,GAAG,CAAC,OAAO;QACnB,MAAM,EAAE,GAAG,CAAC,MAA4B;QACxC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,UAAU,EAAE,GAAG,CAAC,YAAY;QAC5B,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAA;AACH,CAAC;AAED,MAAM,OAAO,qCAAqC;IACnB,EAAE;IAA/B,YAA6B,EAAa;kBAAb,EAAE;IAAc,CAAC;IAE9C,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,MAA0B;QAE1B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,qBAAqB,CAAC;aAC3B,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,EACzC,EAAE,CAAC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,EACxC,MAAM,CAAC,qBAAqB,CAAC,UAAU,CAAC,CACzC,CACF;aACA,KAAK,CAAC,CAAC,CAAC,CAAA;QACX,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,qBAAqB,CAAC;aAC3B,KAAK,CACJ,GAAG,CAAC,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,MAAM,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CACzF;aACA,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAA;QACjD,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAkC;QAC7C,iFAAiF;QACjF,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,qBAAqB,CAAC;aAC7B,GAAG,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC;aACrC,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,EAChD,EAAE,CAAC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,EAC/C,MAAM,CAAC,qBAAqB,CAAC,UAAU,CAAC,EACxC,EAAE,CAAC,qBAAqB,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,CACxC,CACF,CAAA;QACH,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,qBAAqB,CAAC;aAC7B,MAAM,CAAC;YACN,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,OAAO,EAAE,MAAM,CAAC,MAAM;YACtB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,YAAY,EAAE,MAAM,CAAC,UAAU;YAC/B,UAAU,EAAE,IAAI;SACjB,CAAC;aACD,kBAAkB,CAAC;YAClB,MAAM,EAAE,qBAAqB,CAAC,EAAE;YAChC,GAAG,EAAE;gBACH,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,YAAY,EAAE,MAAM,CAAC,WAAW;gBAChC,UAAU,EAAE,MAAM,CAAC,SAAS;gBAC5B,UAAU,EAAE,MAAM,CAAC,SAAS;gBAC5B,UAAU,EAAE,IAAI;aACjB;SACF,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,MAA0B,EAAE,EAAU;QACnE,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,qBAAqB,CAAC;aAC7B,GAAG,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;aACzB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,EACzC,EAAE,CAAC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,EACxC,MAAM,CAAC,qBAAqB,CAAC,UAAU,CAAC,CACzC,CACF,CAAA;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,MAA0B,EAAE,EAAU;QACrE,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,qBAAqB,CAAC;aAC7B,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;aACvB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,CAAC,EACzC,EAAE,CAAC,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,EACxC,MAAM,CAAC,qBAAqB,CAAC,UAAU,CAAC,CACzC,CACF,CAAA;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAW,EAAE,MAAc;QAC5C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,qBAAqB,CAAC;aAC3B,KAAK,CACJ,GAAG,CACD,MAAM,CAAC,qBAAqB,CAAC,UAAU,CAAC,EACxC,GAAG,CAAC,qBAAqB,CAAC,UAAU,EAAE,GAAG,CAAC,EAC1C,GAAG,CAAC,qBAAqB,CAAC,UAAU,EAAE,MAAM,CAAC,CAC9C,CACF;aACA,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC,CAAA;QACjD,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAC3B,CAAC;CACF;AAID,SAAS,YAAY,CAAC,GAAW;IAC/B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,MAAM,EAAE,GAAG,CAAC,OAAO;QACnB,MAAM,EAAE,GAAG,CAAC,MAA4B;QACxC,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAA;AACH,CAAC;AAED,MAAM,OAAO,uCAAuC;IACrB,EAAE;IAA/B,YAA6B,EAAa;kBAAb,EAAE;IAAc,CAAC;IAE9C,KAAK,CAAC,GAAG,CACP,WAAmB,EACnB,MAAc,EACd,MAA0B,EAC1B,GAAW;QAEX,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,uBAAuB,CAAC;aAC7B,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,uBAAuB,CAAC,YAAY,EAAE,WAAW,CAAC,EACrD,EAAE,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,CAAC,EAC3C,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,EAC1C,EAAE,CAAC,uBAAuB,CAAC,UAAU,EAAE,GAAG,CAAC,CAC5C,CACF;aACA,KAAK,CAAC,CAAC,CAAC,CAAA;QACX,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IAC/C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAoC;QAC/C,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,uBAAuB,CAAC;aAC/B,MAAM,CAAC;YACN,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,OAAO,EAAE,MAAM,CAAC,MAAM;YACtB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,UAAU,EAAE,MAAM,CAAC,SAAS;SAC7B,CAAC;aACD,kBAAkB,CAAC;YAClB,4DAA4D;YAC5D,MAAM,EAAE;gBACN,uBAAuB,CAAC,YAAY;gBACpC,uBAAuB,CAAC,OAAO;gBAC/B,uBAAuB,CAAC,MAAM;aAC/B;YACD,GAAG,EAAE;gBACH,YAAY,EAAE,MAAM,CAAC,WAAW;gBAChC,UAAU,EAAE,MAAM,CAAC,SAAS;gBAC5B,UAAU,EAAE,MAAM,CAAC,SAAS;aAC7B;SACF,CAAC,CAAA;IACN,CAAC;IAED,KAAK,CAAC,OAAO,CACX,WAAmB,EACnB,MAAc,EACd,MAA0B,EAC1B,SAAiB;QAEjB,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,uBAAuB,CAAC;aAC/B,GAAG,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;aAC9B,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,uBAAuB,CAAC,YAAY,EAAE,WAAW,CAAC,EACrD,EAAE,CAAC,uBAAuB,CAAC,OAAO,EAAE,MAAM,CAAC,EAC3C,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,MAAM,CAAC,CAC3C,CACF,CAAA;IACL,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,WAAmB;QACzC,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,uBAAuB,CAAC;aAC/B,KAAK,CAAC,EAAE,CAAC,uBAAuB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAA;IACjE,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAW;QAC7B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,EAAE;aAC1B,MAAM,CAAC,uBAAuB,CAAC;aAC/B,KAAK,CAAC,GAAG,CAAC,uBAAuB,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;aACnD,SAAS,CAAC,EAAE,EAAE,EAAE,uBAAuB,CAAC,EAAE,EAAE,CAAC,CAAA;QAChD,OAAO,OAAO,CAAC,MAAM,CAAA;IACvB,CAAC;CACF"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { ApiKeyProvider, ApiKeyScope, ApiKeyScopeRef, ProviderApiKeyRecord, ProviderApiKeyRepository } from '@cat-factory/kernel';
|
|
2
|
+
import type { DrizzleDb } from '../db/client.js';
|
|
3
|
+
export declare class DrizzleProviderApiKeyRepository implements ProviderApiKeyRepository {
|
|
4
|
+
private readonly db;
|
|
5
|
+
constructor(db: DrizzleDb);
|
|
6
|
+
listByScope(scope: ApiKeyScope, scopeId: string, provider?: ApiKeyProvider): Promise<ProviderApiKeyRecord[]>;
|
|
7
|
+
listForPool(scopes: ApiKeyScopeRef[], provider: ApiKeyProvider): Promise<ProviderApiKeyRecord[]>;
|
|
8
|
+
listConfiguredProviders(scopes: ApiKeyScopeRef[]): Promise<ApiKeyProvider[]>;
|
|
9
|
+
getById(scope: ApiKeyScope, scopeId: string, id: string): Promise<ProviderApiKeyRecord | null>;
|
|
10
|
+
add(record: ProviderApiKeyRecord): Promise<void>;
|
|
11
|
+
markLeased(id: string, at: number): Promise<void>;
|
|
12
|
+
recordUsage(id: string, usage: {
|
|
13
|
+
inputTokens: number;
|
|
14
|
+
outputTokens: number;
|
|
15
|
+
}, at: number, windowMs: number): Promise<void>;
|
|
16
|
+
softDelete(scope: ApiKeyScope, scopeId: string, id: string, at: number): Promise<void>;
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=providerApiKey.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerApiKey.d.ts","sourceRoot":"","sources":["../../src/repositories/providerApiKey.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,WAAW,EACX,cAAc,EACd,oBAAoB,EACpB,wBAAwB,EACzB,MAAM,qBAAqB,CAAA;AAE5B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAmChD,qBAAa,+BAAgC,YAAW,wBAAwB;IAClE,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,WAAW,CACf,KAAK,EAAE,WAAW,EAClB,OAAO,EAAE,MAAM,EACf,QAAQ,CAAC,EAAE,cAAc,GACxB,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAcjC;IAEK,WAAW,CACf,MAAM,EAAE,cAAc,EAAE,EACxB,QAAQ,EAAE,cAAc,GACvB,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAcjC;IAEK,uBAAuB,CAAC,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAOjF;IAEK,OAAO,CACX,KAAK,EAAE,WAAW,EAClB,OAAO,EAAE,MAAM,EACf,EAAE,EAAE,MAAM,GACT,OAAO,CAAC,oBAAoB,GAAG,IAAI,CAAC,CAetC;IAEK,GAAG,CAAC,MAAM,EAAE,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC,CAgBrD;IAEK,UAAU,CAAC,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAKtD;IAEK,WAAW,CACf,EAAE,EAAE,MAAM,EACV,KAAK,EAAE;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,EACpD,EAAE,EAAE,MAAM,EACV,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CAef;IAEK,UAAU,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAY3F;CACF"}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
import { and, asc, eq, isNull, or, sql } from 'drizzle-orm';
|
|
2
|
+
import { providerApiKeys } from '../db/schema.js';
|
|
3
|
+
function rowToRecord(row) {
|
|
4
|
+
return {
|
|
5
|
+
id: row.id,
|
|
6
|
+
scope: row.scope,
|
|
7
|
+
scopeId: row.scope_id,
|
|
8
|
+
provider: row.provider,
|
|
9
|
+
label: row.label,
|
|
10
|
+
keyCipher: row.key_cipher,
|
|
11
|
+
createdAt: row.created_at,
|
|
12
|
+
lastUsedAt: row.last_used_at,
|
|
13
|
+
windowStartedAt: row.window_started_at,
|
|
14
|
+
inputTokens: row.input_tokens,
|
|
15
|
+
outputTokens: row.output_tokens,
|
|
16
|
+
requestCount: row.request_count,
|
|
17
|
+
deletedAt: row.deleted_at,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
/** OR-predicate matching any of the given (scope, scopeId) segments. */
|
|
21
|
+
function scopeMatch(scopes) {
|
|
22
|
+
return or(...scopes.map((s) => and(eq(providerApiKeys.scope, s.scope), eq(providerApiKeys.scope_id, s.scopeId))));
|
|
23
|
+
}
|
|
24
|
+
export class DrizzleProviderApiKeyRepository {
|
|
25
|
+
db;
|
|
26
|
+
constructor(db) {
|
|
27
|
+
this.db = db;
|
|
28
|
+
}
|
|
29
|
+
async listByScope(scope, scopeId, provider) {
|
|
30
|
+
const rows = await this.db
|
|
31
|
+
.select()
|
|
32
|
+
.from(providerApiKeys)
|
|
33
|
+
.where(and(eq(providerApiKeys.scope, scope), eq(providerApiKeys.scope_id, scopeId), provider ? eq(providerApiKeys.provider, provider) : undefined, isNull(providerApiKeys.deleted_at)))
|
|
34
|
+
.orderBy(asc(providerApiKeys.created_at));
|
|
35
|
+
return rows.map(rowToRecord);
|
|
36
|
+
}
|
|
37
|
+
async listForPool(scopes, provider) {
|
|
38
|
+
if (scopes.length === 0)
|
|
39
|
+
return [];
|
|
40
|
+
const rows = await this.db
|
|
41
|
+
.select()
|
|
42
|
+
.from(providerApiKeys)
|
|
43
|
+
.where(and(scopeMatch(scopes), eq(providerApiKeys.provider, provider), isNull(providerApiKeys.deleted_at)))
|
|
44
|
+
.orderBy(asc(providerApiKeys.created_at));
|
|
45
|
+
return rows.map(rowToRecord);
|
|
46
|
+
}
|
|
47
|
+
async listConfiguredProviders(scopes) {
|
|
48
|
+
if (scopes.length === 0)
|
|
49
|
+
return [];
|
|
50
|
+
const rows = await this.db
|
|
51
|
+
.selectDistinct({ provider: providerApiKeys.provider })
|
|
52
|
+
.from(providerApiKeys)
|
|
53
|
+
.where(and(scopeMatch(scopes), isNull(providerApiKeys.deleted_at)));
|
|
54
|
+
return rows.map((r) => r.provider);
|
|
55
|
+
}
|
|
56
|
+
async getById(scope, scopeId, id) {
|
|
57
|
+
const rows = await this.db
|
|
58
|
+
.select()
|
|
59
|
+
.from(providerApiKeys)
|
|
60
|
+
.where(and(eq(providerApiKeys.id, id), eq(providerApiKeys.scope, scope), eq(providerApiKeys.scope_id, scopeId), isNull(providerApiKeys.deleted_at)))
|
|
61
|
+
.limit(1);
|
|
62
|
+
const row = rows[0];
|
|
63
|
+
return row ? rowToRecord(row) : null;
|
|
64
|
+
}
|
|
65
|
+
async add(record) {
|
|
66
|
+
await this.db.insert(providerApiKeys).values({
|
|
67
|
+
id: record.id,
|
|
68
|
+
scope: record.scope,
|
|
69
|
+
scope_id: record.scopeId,
|
|
70
|
+
provider: record.provider,
|
|
71
|
+
label: record.label,
|
|
72
|
+
key_cipher: record.keyCipher,
|
|
73
|
+
created_at: record.createdAt,
|
|
74
|
+
last_used_at: record.lastUsedAt,
|
|
75
|
+
window_started_at: record.windowStartedAt,
|
|
76
|
+
input_tokens: record.inputTokens,
|
|
77
|
+
output_tokens: record.outputTokens,
|
|
78
|
+
request_count: record.requestCount,
|
|
79
|
+
deleted_at: record.deletedAt,
|
|
80
|
+
});
|
|
81
|
+
}
|
|
82
|
+
async markLeased(id, at) {
|
|
83
|
+
await this.db
|
|
84
|
+
.update(providerApiKeys)
|
|
85
|
+
.set({ last_used_at: at })
|
|
86
|
+
.where(eq(providerApiKeys.id, id));
|
|
87
|
+
}
|
|
88
|
+
async recordUsage(id, usage, at, windowMs) {
|
|
89
|
+
// A single atomic statement (no read-modify-write) — mirrors the D1 repo and
|
|
90
|
+
// the subscription pool. Keyed by row id alone (the leased key may belong to
|
|
91
|
+
// any scope segment merged at lease time).
|
|
92
|
+
const cols = providerApiKeys;
|
|
93
|
+
const active = sql `(${cols.window_started_at} IS NOT NULL AND ${at} - ${cols.window_started_at} < ${windowMs})`;
|
|
94
|
+
await this.db
|
|
95
|
+
.update(cols)
|
|
96
|
+
.set({
|
|
97
|
+
window_started_at: sql `CASE WHEN ${active} THEN ${cols.window_started_at} ELSE ${at} END`,
|
|
98
|
+
input_tokens: sql `CASE WHEN ${active} THEN ${cols.input_tokens} ELSE 0 END + ${usage.inputTokens}`,
|
|
99
|
+
output_tokens: sql `CASE WHEN ${active} THEN ${cols.output_tokens} ELSE 0 END + ${usage.outputTokens}`,
|
|
100
|
+
request_count: sql `CASE WHEN ${active} THEN ${cols.request_count} ELSE 0 END + 1`,
|
|
101
|
+
})
|
|
102
|
+
.where(eq(cols.id, id));
|
|
103
|
+
}
|
|
104
|
+
async softDelete(scope, scopeId, id, at) {
|
|
105
|
+
await this.db
|
|
106
|
+
.update(providerApiKeys)
|
|
107
|
+
.set({ deleted_at: at })
|
|
108
|
+
.where(and(eq(providerApiKeys.id, id), eq(providerApiKeys.scope, scope), eq(providerApiKeys.scope_id, scopeId), isNull(providerApiKeys.deleted_at)));
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
//# sourceMappingURL=providerApiKey.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerApiKey.js","sourceRoot":"","sources":["../../src/repositories/providerApiKey.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AAE3D,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA;AAOjD,SAAS,WAAW,CAAC,GAAQ;IAC3B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,KAAK,EAAE,GAAG,CAAC,KAAoB;QAC/B,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,QAAQ,EAAE,GAAG,CAAC,QAA0B;QACxC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,UAAU,EAAE,GAAG,CAAC,YAAY;QAC5B,eAAe,EAAE,GAAG,CAAC,iBAAiB;QACtC,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAA;AACH,CAAC;AAED,wEAAwE;AACxE,SAAS,UAAU,CAAC,MAAwB;IAC1C,OAAO,EAAE,CACP,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAClB,GAAG,CAAC,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CACjF,CACF,CAAA;AACH,CAAC;AAED,MAAM,OAAO,+BAA+B;IACb,EAAE;IAA/B,YAA6B,EAAa;kBAAb,EAAE;IAAc,CAAC;IAE9C,KAAK,CAAC,WAAW,CACf,KAAkB,EAClB,OAAe,EACf,QAAyB;QAEzB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,eAAe,CAAC;aACrB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,EAChC,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,EACrC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,EAC7D,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CACnC,CACF;aACA,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAA;QAC3C,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,WAAW,CACf,MAAwB,EACxB,QAAwB;QAExB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAA;QAClC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,eAAe,CAAC;aACrB,KAAK,CACJ,GAAG,CACD,UAAU,CAAC,MAAM,CAAC,EAClB,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,EACtC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CACnC,CACF;aACA,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAA;QAC3C,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,uBAAuB,CAAC,MAAwB;QACpD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAA;QAClC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,cAAc,CAAC,EAAE,QAAQ,EAAE,eAAe,CAAC,QAAQ,EAAE,CAAC;aACtD,IAAI,CAAC,eAAe,CAAC;aACrB,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,CAAA;QACrE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAA0B,CAAC,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,OAAO,CACX,KAAkB,EAClB,OAAe,EACf,EAAU;QAEV,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,eAAe,CAAC;aACrB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,EAC1B,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,EAChC,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,EACrC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CACnC,CACF;aACA,KAAK,CAAC,CAAC,CAAC,CAAA;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;QACnB,OAAO,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAA4B;QACpC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC;YAC3C,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,OAAO;YACxB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,YAAY,EAAE,MAAM,CAAC,UAAU;YAC/B,iBAAiB,EAAE,MAAM,CAAC,eAAe;YACzC,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,aAAa,EAAE,MAAM,CAAC,YAAY;YAClC,aAAa,EAAE,MAAM,CAAC,YAAY;YAClC,UAAU,EAAE,MAAM,CAAC,SAAS;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU,EAAE,EAAU;QACrC,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,eAAe,CAAC;aACvB,GAAG,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;aACzB,KAAK,CAAC,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,WAAW,CACf,EAAU,EACV,KAAoD,EACpD,EAAU,EACV,QAAgB;QAEhB,6EAA6E;QAC7E,6EAA6E;QAC7E,2CAA2C;QAC3C,MAAM,IAAI,GAAG,eAAe,CAAA;QAC5B,MAAM,MAAM,GAAG,GAAG,CAAA,IAAI,IAAI,CAAC,iBAAiB,oBAAoB,EAAE,MAAM,IAAI,CAAC,iBAAiB,MAAM,QAAQ,GAAG,CAAA;QAC/G,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,IAAI,CAAC;aACZ,GAAG,CAAC;YACH,iBAAiB,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,iBAAiB,SAAS,EAAE,MAAM;YACzF,YAAY,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,YAAY,iBAAiB,KAAK,CAAC,WAAW,EAAE;YAClG,aAAa,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,aAAa,iBAAiB,KAAK,CAAC,YAAY,EAAE;YACrG,aAAa,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,aAAa,iBAAiB;SAClF,CAAC;aACD,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAkB,EAAE,OAAe,EAAE,EAAU,EAAE,EAAU;QAC1E,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,eAAe,CAAC;aACvB,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;aACvB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,EAAE,CAAC,EAC1B,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,EAChC,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,EACrC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CACnC,CACF,CAAA;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { ProviderSubscriptionTokenRecord, ProviderSubscriptionTokenRepository, SubscriptionVendor } from '@cat-factory/kernel';
|
|
2
|
+
import type { DrizzleDb } from '../db/client.js';
|
|
3
|
+
export declare class DrizzleProviderSubscriptionTokenRepository implements ProviderSubscriptionTokenRepository {
|
|
4
|
+
private readonly db;
|
|
5
|
+
constructor(db: DrizzleDb);
|
|
6
|
+
listByVendor(workspaceId: string, vendor: SubscriptionVendor): Promise<ProviderSubscriptionTokenRecord[]>;
|
|
7
|
+
getById(workspaceId: string, id: string): Promise<ProviderSubscriptionTokenRecord | null>;
|
|
8
|
+
add(record: ProviderSubscriptionTokenRecord): Promise<void>;
|
|
9
|
+
markLeased(workspaceId: string, id: string, at: number): Promise<void>;
|
|
10
|
+
recordUsage(workspaceId: string, id: string, usage: {
|
|
11
|
+
inputTokens: number;
|
|
12
|
+
outputTokens: number;
|
|
13
|
+
}, at: number, windowMs: number): Promise<void>;
|
|
14
|
+
softDelete(workspaceId: string, id: string, at: number): Promise<void>;
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=providerSubscription.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerSubscription.d.ts","sourceRoot":"","sources":["../../src/repositories/providerSubscription.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,+BAA+B,EAC/B,mCAAmC,EACnC,kBAAkB,EACnB,MAAM,qBAAqB,CAAA;AAE5B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AA0BhD,qBAAa,0CAA2C,YAAW,mCAAmC;IACxF,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,YAAY,CAChB,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,kBAAkB,GACzB,OAAO,CAAC,+BAA+B,EAAE,CAAC,CAa5C;IAEK,OAAO,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,+BAA+B,GAAG,IAAI,CAAC,CAc9F;IAEK,GAAG,CAAC,MAAM,EAAE,+BAA+B,GAAG,OAAO,CAAC,IAAI,CAAC,CAehE;IAEK,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAU3E;IAEK,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,EAAE,EAAE,MAAM,EACV,KAAK,EAAE;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,EACpD,EAAE,EAAE,MAAM,EACV,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CAiBf;IAEK,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAW3E;CACF"}
|
|
@@ -0,0 +1,88 @@
|
|
|
1
|
+
import { and, asc, eq, isNull, sql } from 'drizzle-orm';
|
|
2
|
+
import { providerSubscriptionTokens } from '../db/schema.js';
|
|
3
|
+
function rowToRecord(row) {
|
|
4
|
+
return {
|
|
5
|
+
id: row.id,
|
|
6
|
+
workspaceId: row.workspace_id,
|
|
7
|
+
vendor: row.vendor,
|
|
8
|
+
label: row.label,
|
|
9
|
+
tokenCipher: row.token_cipher,
|
|
10
|
+
createdAt: row.created_at,
|
|
11
|
+
lastUsedAt: row.last_used_at,
|
|
12
|
+
windowStartedAt: row.window_started_at,
|
|
13
|
+
inputTokens: row.input_tokens,
|
|
14
|
+
outputTokens: row.output_tokens,
|
|
15
|
+
requestCount: row.request_count,
|
|
16
|
+
deletedAt: row.deleted_at,
|
|
17
|
+
};
|
|
18
|
+
}
|
|
19
|
+
export class DrizzleProviderSubscriptionTokenRepository {
|
|
20
|
+
db;
|
|
21
|
+
constructor(db) {
|
|
22
|
+
this.db = db;
|
|
23
|
+
}
|
|
24
|
+
async listByVendor(workspaceId, vendor) {
|
|
25
|
+
const rows = await this.db
|
|
26
|
+
.select()
|
|
27
|
+
.from(providerSubscriptionTokens)
|
|
28
|
+
.where(and(eq(providerSubscriptionTokens.workspace_id, workspaceId), eq(providerSubscriptionTokens.vendor, vendor), isNull(providerSubscriptionTokens.deleted_at)))
|
|
29
|
+
.orderBy(asc(providerSubscriptionTokens.created_at));
|
|
30
|
+
return rows.map(rowToRecord);
|
|
31
|
+
}
|
|
32
|
+
async getById(workspaceId, id) {
|
|
33
|
+
const rows = await this.db
|
|
34
|
+
.select()
|
|
35
|
+
.from(providerSubscriptionTokens)
|
|
36
|
+
.where(and(eq(providerSubscriptionTokens.id, id), eq(providerSubscriptionTokens.workspace_id, workspaceId), isNull(providerSubscriptionTokens.deleted_at)))
|
|
37
|
+
.limit(1);
|
|
38
|
+
const row = rows[0];
|
|
39
|
+
return row ? rowToRecord(row) : null;
|
|
40
|
+
}
|
|
41
|
+
async add(record) {
|
|
42
|
+
await this.db.insert(providerSubscriptionTokens).values({
|
|
43
|
+
id: record.id,
|
|
44
|
+
workspace_id: record.workspaceId,
|
|
45
|
+
vendor: record.vendor,
|
|
46
|
+
label: record.label,
|
|
47
|
+
token_cipher: record.tokenCipher,
|
|
48
|
+
created_at: record.createdAt,
|
|
49
|
+
last_used_at: record.lastUsedAt,
|
|
50
|
+
window_started_at: record.windowStartedAt,
|
|
51
|
+
input_tokens: record.inputTokens,
|
|
52
|
+
output_tokens: record.outputTokens,
|
|
53
|
+
request_count: record.requestCount,
|
|
54
|
+
deleted_at: record.deletedAt,
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
async markLeased(workspaceId, id, at) {
|
|
58
|
+
await this.db
|
|
59
|
+
.update(providerSubscriptionTokens)
|
|
60
|
+
.set({ last_used_at: at })
|
|
61
|
+
.where(and(eq(providerSubscriptionTokens.id, id), eq(providerSubscriptionTokens.workspace_id, workspaceId)));
|
|
62
|
+
}
|
|
63
|
+
async recordUsage(workspaceId, id, usage, at, windowMs) {
|
|
64
|
+
// A single atomic statement (no read-modify-write) so two jobs finishing on the
|
|
65
|
+
// same token can't lose each other's counters — mirrors the D1 repository. The
|
|
66
|
+
// window-active test is evaluated against the row's pre-update values in every
|
|
67
|
+
// branch: a stale window resets to `at` and counts from this run; an active one
|
|
68
|
+
// accumulates.
|
|
69
|
+
const cols = providerSubscriptionTokens;
|
|
70
|
+
const active = sql `(${cols.window_started_at} IS NOT NULL AND ${at} - ${cols.window_started_at} < ${windowMs})`;
|
|
71
|
+
await this.db
|
|
72
|
+
.update(cols)
|
|
73
|
+
.set({
|
|
74
|
+
window_started_at: sql `CASE WHEN ${active} THEN ${cols.window_started_at} ELSE ${at} END`,
|
|
75
|
+
input_tokens: sql `CASE WHEN ${active} THEN ${cols.input_tokens} ELSE 0 END + ${usage.inputTokens}`,
|
|
76
|
+
output_tokens: sql `CASE WHEN ${active} THEN ${cols.output_tokens} ELSE 0 END + ${usage.outputTokens}`,
|
|
77
|
+
request_count: sql `CASE WHEN ${active} THEN ${cols.request_count} ELSE 0 END + 1`,
|
|
78
|
+
})
|
|
79
|
+
.where(and(eq(cols.id, id), eq(cols.workspace_id, workspaceId)));
|
|
80
|
+
}
|
|
81
|
+
async softDelete(workspaceId, id, at) {
|
|
82
|
+
await this.db
|
|
83
|
+
.update(providerSubscriptionTokens)
|
|
84
|
+
.set({ deleted_at: at })
|
|
85
|
+
.where(and(eq(providerSubscriptionTokens.id, id), eq(providerSubscriptionTokens.workspace_id, workspaceId), isNull(providerSubscriptionTokens.deleted_at)));
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
//# sourceMappingURL=providerSubscription.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providerSubscription.js","sourceRoot":"","sources":["../../src/repositories/providerSubscription.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AAEvD,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAA;AAQ5D,SAAS,WAAW,CAAC,GAAQ;IAC3B,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,EAAE;QACV,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,MAAM,EAAE,GAAG,CAAC,MAA4B;QACxC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,UAAU,EAAE,GAAG,CAAC,YAAY;QAC5B,eAAe,EAAE,GAAG,CAAC,iBAAiB;QACtC,WAAW,EAAE,GAAG,CAAC,YAAY;QAC7B,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,YAAY,EAAE,GAAG,CAAC,aAAa;QAC/B,SAAS,EAAE,GAAG,CAAC,UAAU;KAC1B,CAAA;AACH,CAAC;AAED,MAAM,OAAO,0CAA0C;IACxB,EAAE;IAA/B,YAA6B,EAAa;kBAAb,EAAE;IAAc,CAAC;IAE9C,KAAK,CAAC,YAAY,CAChB,WAAmB,EACnB,MAA0B;QAE1B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,0BAA0B,CAAC;aAChC,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,0BAA0B,CAAC,YAAY,EAAE,WAAW,CAAC,EACxD,EAAE,CAAC,0BAA0B,CAAC,MAAM,EAAE,MAAM,CAAC,EAC7C,MAAM,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAC9C,CACF;aACA,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAAC,CAAA;QACtD,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,EAAU;QAC3C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE;aACvB,MAAM,EAAE;aACR,IAAI,CAAC,0BAA0B,CAAC;aAChC,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,0BAA0B,CAAC,EAAE,EAAE,EAAE,CAAC,EACrC,EAAE,CAAC,0BAA0B,CAAC,YAAY,EAAE,WAAW,CAAC,EACxD,MAAM,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAC9C,CACF;aACA,KAAK,CAAC,CAAC,CAAC,CAAA;QACX,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;QACnB,OAAO,GAAG,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IACtC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAuC;QAC/C,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,0BAA0B,CAAC,CAAC,MAAM,CAAC;YACtD,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,YAAY,EAAE,MAAM,CAAC,UAAU;YAC/B,iBAAiB,EAAE,MAAM,CAAC,eAAe;YACzC,YAAY,EAAE,MAAM,CAAC,WAAW;YAChC,aAAa,EAAE,MAAM,CAAC,YAAY;YAClC,aAAa,EAAE,MAAM,CAAC,YAAY;YAClC,UAAU,EAAE,MAAM,CAAC,SAAS;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,WAAmB,EAAE,EAAU,EAAE,EAAU;QAC1D,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,0BAA0B,CAAC;aAClC,GAAG,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;aACzB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,0BAA0B,CAAC,EAAE,EAAE,EAAE,CAAC,EACrC,EAAE,CAAC,0BAA0B,CAAC,YAAY,EAAE,WAAW,CAAC,CACzD,CACF,CAAA;IACL,CAAC;IAED,KAAK,CAAC,WAAW,CACf,WAAmB,EACnB,EAAU,EACV,KAAoD,EACpD,EAAU,EACV,QAAgB;QAEhB,gFAAgF;QAChF,+EAA+E;QAC/E,+EAA+E;QAC/E,gFAAgF;QAChF,eAAe;QACf,MAAM,IAAI,GAAG,0BAA0B,CAAA;QACvC,MAAM,MAAM,GAAG,GAAG,CAAA,IAAI,IAAI,CAAC,iBAAiB,oBAAoB,EAAE,MAAM,IAAI,CAAC,iBAAiB,MAAM,QAAQ,GAAG,CAAA;QAC/G,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,IAAI,CAAC;aACZ,GAAG,CAAC;YACH,iBAAiB,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,iBAAiB,SAAS,EAAE,MAAM;YACzF,YAAY,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,YAAY,iBAAiB,KAAK,CAAC,WAAW,EAAE;YAClG,aAAa,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,aAAa,iBAAiB,KAAK,CAAC,YAAY,EAAE;YACrG,aAAa,EAAE,GAAG,CAAA,aAAa,MAAM,SAAS,IAAI,CAAC,aAAa,iBAAiB;SAClF,CAAC;aACD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC,CAAA;IACpE,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,WAAmB,EAAE,EAAU,EAAE,EAAU;QAC1D,MAAM,IAAI,CAAC,EAAE;aACV,MAAM,CAAC,0BAA0B,CAAC;aAClC,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;aACvB,KAAK,CACJ,GAAG,CACD,EAAE,CAAC,0BAA0B,CAAC,EAAE,EAAE,EAAE,CAAC,EACrC,EAAE,CAAC,0BAA0B,CAAC,YAAY,EAAE,WAAW,CAAC,EACxD,MAAM,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAC9C,CACF,CAAA;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { SlackConnectionRecord, SlackConnectionRepository, SlackMemberMappingEntry, SlackMemberMappingRepository, SlackSettingsRecord, SlackSettingsRepository } from '@cat-factory/kernel';
|
|
2
|
+
import type { DrizzleDb } from '../db/client.js';
|
|
3
|
+
export declare class DrizzleSlackConnectionRepository implements SlackConnectionRepository {
|
|
4
|
+
private readonly db;
|
|
5
|
+
constructor(db: DrizzleDb);
|
|
6
|
+
getByAccount(accountId: string): Promise<SlackConnectionRecord | null>;
|
|
7
|
+
getByTeam(teamId: string): Promise<SlackConnectionRecord | null>;
|
|
8
|
+
upsert(record: SlackConnectionRecord): Promise<void>;
|
|
9
|
+
softDelete(accountId: string, at: number): Promise<void>;
|
|
10
|
+
}
|
|
11
|
+
export declare class DrizzleSlackSettingsRepository implements SlackSettingsRepository {
|
|
12
|
+
private readonly db;
|
|
13
|
+
constructor(db: DrizzleDb);
|
|
14
|
+
getByWorkspace(workspaceId: string): Promise<SlackSettingsRecord | null>;
|
|
15
|
+
upsert(record: SlackSettingsRecord): Promise<void>;
|
|
16
|
+
}
|
|
17
|
+
export declare class DrizzleSlackMemberMappingRepository implements SlackMemberMappingRepository {
|
|
18
|
+
private readonly db;
|
|
19
|
+
constructor(db: DrizzleDb);
|
|
20
|
+
getByAccount(accountId: string): Promise<SlackMemberMappingEntry[]>;
|
|
21
|
+
upsert(accountId: string, entries: SlackMemberMappingEntry[], at: number): Promise<void>;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=slack.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"slack.d.ts","sourceRoot":"","sources":["../../src/repositories/slack.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,EACvB,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACxB,MAAM,qBAAqB,CAAA;AAE5B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAwBhD,qBAAa,gCAAiC,YAAW,yBAAyB;IACpE,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAO3E;IAEK,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAOrE;IAEK,MAAM,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CA4BzD;IAEK,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAK7D;CACF;AAID,qBAAa,8BAA+B,YAAW,uBAAuB;IAChE,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,cAAc,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC,CAc7E;IAEK,MAAM,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC,CAkBvD;CACF;AAED,qBAAa,mCAAoC,YAAW,4BAA4B;IAC1E,OAAO,CAAC,QAAQ,CAAC,EAAE;IAA/B,YAA6B,EAAE,EAAE,SAAS,EAAI;IAExC,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,uBAAuB,EAAE,CAAC,CAaxE;IAEK,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,uBAAuB,EAAE,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAa7F;CACF"}
|