@polpo-ai/drizzle 0.1.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/__tests__/stores.test.d.ts +2 -0
- package/dist/__tests__/stores.test.d.ts.map +1 -0
- package/dist/__tests__/stores.test.js +922 -0
- package/dist/__tests__/stores.test.js.map +1 -0
- package/dist/index.d.ts +4811 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +129 -0
- package/dist/index.js.map +1 -0
- package/dist/migrate.d.ts +8 -0
- package/dist/migrate.d.ts.map +1 -0
- package/dist/migrate.js +214 -0
- package/dist/migrate.js.map +1 -0
- package/dist/schema/approvals.d.ts +413 -0
- package/dist/schema/approvals.d.ts.map +1 -0
- package/dist/schema/approvals.js +37 -0
- package/dist/schema/approvals.js.map +1 -0
- package/dist/schema/index.d.ts +17 -0
- package/dist/schema/index.d.ts.map +1 -0
- package/dist/schema/index.js +19 -0
- package/dist/schema/index.js.map +1 -0
- package/dist/schema/logs.d.ts +281 -0
- package/dist/schema/logs.d.ts.map +1 -0
- package/dist/schema/logs.js +33 -0
- package/dist/schema/logs.js.map +1 -0
- package/dist/schema/memory.d.ts +87 -0
- package/dist/schema/memory.d.ts.map +1 -0
- package/dist/schema/memory.js +13 -0
- package/dist/schema/memory.js.map +1 -0
- package/dist/schema/notifications.d.ts +521 -0
- package/dist/schema/notifications.d.ts.map +1 -0
- package/dist/schema/notifications.js +47 -0
- package/dist/schema/notifications.js.map +1 -0
- package/dist/schema/peers.d.ts +743 -0
- package/dist/schema/peers.d.ts.map +1 -0
- package/dist/schema/peers.js +71 -0
- package/dist/schema/peers.js.map +1 -0
- package/dist/schema/runs.d.ts +483 -0
- package/dist/schema/runs.d.ts.map +1 -0
- package/dist/schema/runs.js +41 -0
- package/dist/schema/runs.js.map +1 -0
- package/dist/schema/sessions.d.ts +389 -0
- package/dist/schema/sessions.d.ts.map +1 -0
- package/dist/schema/sessions.js +37 -0
- package/dist/schema/sessions.js.map +1 -0
- package/dist/schema/tasks.d.ts +1843 -0
- package/dist/schema/tasks.d.ts.map +1 -0
- package/dist/schema/tasks.js +135 -0
- package/dist/schema/tasks.js.map +1 -0
- package/dist/stores/approval-store.d.ts +19 -0
- package/dist/stores/approval-store.d.ts.map +1 -0
- package/dist/stores/approval-store.js +77 -0
- package/dist/stores/approval-store.js.map +1 -0
- package/dist/stores/checkpoint-store.d.ts +14 -0
- package/dist/stores/checkpoint-store.d.ts.map +1 -0
- package/dist/stores/checkpoint-store.js +44 -0
- package/dist/stores/checkpoint-store.js.map +1 -0
- package/dist/stores/config-store.d.ts +15 -0
- package/dist/stores/config-store.d.ts.map +1 -0
- package/dist/stores/config-store.js +31 -0
- package/dist/stores/config-store.js.map +1 -0
- package/dist/stores/delay-store.d.ts +14 -0
- package/dist/stores/delay-store.d.ts.map +1 -0
- package/dist/stores/delay-store.js +42 -0
- package/dist/stores/delay-store.js.map +1 -0
- package/dist/stores/index.d.ts +13 -0
- package/dist/stores/index.d.ts.map +1 -0
- package/dist/stores/index.js +12 -0
- package/dist/stores/index.js.map +1 -0
- package/dist/stores/log-store.d.ts +20 -0
- package/dist/stores/log-store.d.ts.map +1 -0
- package/dist/stores/log-store.js +87 -0
- package/dist/stores/log-store.js.map +1 -0
- package/dist/stores/memory-store.d.ts +14 -0
- package/dist/stores/memory-store.d.ts.map +1 -0
- package/dist/stores/memory-store.js +39 -0
- package/dist/stores/memory-store.js.map +1 -0
- package/dist/stores/notification-store.d.ts +20 -0
- package/dist/stores/notification-store.d.ts.map +1 -0
- package/dist/stores/notification-store.js +111 -0
- package/dist/stores/notification-store.js.map +1 -0
- package/dist/stores/peer-store.d.ts +40 -0
- package/dist/stores/peer-store.d.ts.map +1 -0
- package/dist/stores/peer-store.js +203 -0
- package/dist/stores/peer-store.js.map +1 -0
- package/dist/stores/run-store.d.ts +23 -0
- package/dist/stores/run-store.d.ts.map +1 -0
- package/dist/stores/run-store.js +120 -0
- package/dist/stores/run-store.js.map +1 -0
- package/dist/stores/session-store.d.ts +26 -0
- package/dist/stores/session-store.d.ts.map +1 -0
- package/dist/stores/session-store.js +166 -0
- package/dist/stores/session-store.js.map +1 -0
- package/dist/stores/task-store.d.ts +42 -0
- package/dist/stores/task-store.d.ts.map +1 -0
- package/dist/stores/task-store.js +387 -0
- package/dist/stores/task-store.js.map +1 -0
- package/dist/utils.d.ts +11 -0
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +24 -0
- package/dist/utils.js.map +1 -0
- package/package.json +74 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tasks.d.ts","sourceRoot":"","sources":["../../src/schema/tasks.ts"],"names":[],"mappings":"AAKA,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAmCtB,CAAC;AAEH,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgBzB,CAAC;AAEH,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGzB,CAAC;AAEH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAO1B,CAAC;AAIH,eAAO,MAAM,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAmClB,CAAC;AAEH,eAAO,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAgBrB,CAAC;AAEH,eAAO,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGrB,CAAC;AAEH,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAOtB,CAAC"}
|
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
import { sqliteTable, text, integer, index } from "drizzle-orm/sqlite-core";
|
|
2
|
+
import { pgTable, text as pgText, integer as pgInteger, jsonb, varchar, index as pgIndex } from "drizzle-orm/pg-core";
|
|
3
|
+
// ── SQLite schema ──────────────────────────────────────────────────────
|
|
4
|
+
export const tasksSqlite = sqliteTable("tasks", {
|
|
5
|
+
id: text("id").primaryKey(),
|
|
6
|
+
title: text("title").notNull(),
|
|
7
|
+
description: text("description").notNull(),
|
|
8
|
+
assignTo: text("assign_to").notNull(),
|
|
9
|
+
group: text("group"),
|
|
10
|
+
missionId: text("mission_id"),
|
|
11
|
+
dependsOn: text("depends_on").notNull().default("[]"),
|
|
12
|
+
status: text("status").notNull().default("pending"),
|
|
13
|
+
retries: integer("retries").notNull().default(0),
|
|
14
|
+
maxRetries: integer("max_retries").notNull().default(2),
|
|
15
|
+
maxDuration: integer("max_duration"),
|
|
16
|
+
retryPolicy: text("retry_policy"),
|
|
17
|
+
expectations: text("expectations").notNull().default("[]"),
|
|
18
|
+
metrics: text("metrics").notNull().default("[]"),
|
|
19
|
+
result: text("result"),
|
|
20
|
+
phase: text("phase"),
|
|
21
|
+
fixAttempts: integer("fix_attempts").notNull().default(0),
|
|
22
|
+
resolutionAttempts: integer("resolution_attempts").notNull().default(0),
|
|
23
|
+
originalDescription: text("original_description"),
|
|
24
|
+
sessionId: text("session_id"),
|
|
25
|
+
notifications: text("notifications"),
|
|
26
|
+
outcomes: text("outcomes"),
|
|
27
|
+
expectedOutcomes: text("expected_outcomes"),
|
|
28
|
+
deadline: text("deadline"),
|
|
29
|
+
priority: text("priority"),
|
|
30
|
+
sideEffects: integer("side_effects"),
|
|
31
|
+
revisionCount: integer("revision_count"),
|
|
32
|
+
createdAt: text("created_at").notNull(),
|
|
33
|
+
updatedAt: text("updated_at").notNull(),
|
|
34
|
+
}, (table) => [
|
|
35
|
+
index("idx_tasks_status").on(table.status),
|
|
36
|
+
index("idx_tasks_group").on(table.group),
|
|
37
|
+
index("idx_tasks_assign_to").on(table.assignTo),
|
|
38
|
+
index("idx_tasks_mission_id").on(table.missionId),
|
|
39
|
+
]);
|
|
40
|
+
export const missionsSqlite = sqliteTable("missions", {
|
|
41
|
+
id: text("id").primaryKey(),
|
|
42
|
+
name: text("name").notNull().unique(),
|
|
43
|
+
data: text("data").notNull(),
|
|
44
|
+
prompt: text("prompt"),
|
|
45
|
+
status: text("status").notNull().default("draft"),
|
|
46
|
+
schedule: text("schedule"),
|
|
47
|
+
endDate: text("end_date"),
|
|
48
|
+
qualityThreshold: text("quality_threshold"),
|
|
49
|
+
deadline: text("deadline"),
|
|
50
|
+
notifications: text("notifications"),
|
|
51
|
+
executionCount: integer("execution_count").notNull().default(0),
|
|
52
|
+
createdAt: text("created_at").notNull(),
|
|
53
|
+
updatedAt: text("updated_at").notNull(),
|
|
54
|
+
}, (table) => [
|
|
55
|
+
index("idx_missions_status").on(table.status),
|
|
56
|
+
]);
|
|
57
|
+
export const metadataSqlite = sqliteTable("metadata", {
|
|
58
|
+
key: text("key").primaryKey(),
|
|
59
|
+
value: text("value").notNull(),
|
|
60
|
+
});
|
|
61
|
+
export const processesSqlite = sqliteTable("processes", {
|
|
62
|
+
agentName: text("agent_name").notNull(),
|
|
63
|
+
pid: integer("pid").notNull(),
|
|
64
|
+
taskId: text("task_id").notNull(),
|
|
65
|
+
startedAt: text("started_at").notNull(),
|
|
66
|
+
alive: integer("alive").notNull().default(1),
|
|
67
|
+
activity: text("activity").notNull().default("{}"),
|
|
68
|
+
});
|
|
69
|
+
// ── PostgreSQL schema ──────────────────────────────────────────────────
|
|
70
|
+
export const tasksPg = pgTable("tasks", {
|
|
71
|
+
id: pgText("id").primaryKey(),
|
|
72
|
+
title: pgText("title").notNull(),
|
|
73
|
+
description: pgText("description").notNull(),
|
|
74
|
+
assignTo: pgText("assign_to").notNull(),
|
|
75
|
+
group: pgText("group"),
|
|
76
|
+
missionId: pgText("mission_id"),
|
|
77
|
+
dependsOn: jsonb("depends_on").notNull().default([]),
|
|
78
|
+
status: varchar("status", { length: 32 }).notNull().default("pending"),
|
|
79
|
+
retries: pgInteger("retries").notNull().default(0),
|
|
80
|
+
maxRetries: pgInteger("max_retries").notNull().default(2),
|
|
81
|
+
maxDuration: pgInteger("max_duration"),
|
|
82
|
+
retryPolicy: jsonb("retry_policy"),
|
|
83
|
+
expectations: jsonb("expectations").notNull().default([]),
|
|
84
|
+
metrics: jsonb("metrics").notNull().default([]),
|
|
85
|
+
result: jsonb("result"),
|
|
86
|
+
phase: varchar("phase", { length: 32 }),
|
|
87
|
+
fixAttempts: pgInteger("fix_attempts").notNull().default(0),
|
|
88
|
+
resolutionAttempts: pgInteger("resolution_attempts").notNull().default(0),
|
|
89
|
+
originalDescription: pgText("original_description"),
|
|
90
|
+
sessionId: pgText("session_id"),
|
|
91
|
+
notifications: jsonb("notifications"),
|
|
92
|
+
outcomes: jsonb("outcomes"),
|
|
93
|
+
expectedOutcomes: jsonb("expected_outcomes"),
|
|
94
|
+
deadline: pgText("deadline"),
|
|
95
|
+
priority: pgText("priority"),
|
|
96
|
+
sideEffects: pgInteger("side_effects"),
|
|
97
|
+
revisionCount: pgInteger("revision_count"),
|
|
98
|
+
createdAt: pgText("created_at").notNull(),
|
|
99
|
+
updatedAt: pgText("updated_at").notNull(),
|
|
100
|
+
}, (table) => [
|
|
101
|
+
pgIndex("idx_pg_tasks_status").on(table.status),
|
|
102
|
+
pgIndex("idx_pg_tasks_group").on(table.group),
|
|
103
|
+
pgIndex("idx_pg_tasks_assign_to").on(table.assignTo),
|
|
104
|
+
pgIndex("idx_pg_tasks_mission_id").on(table.missionId),
|
|
105
|
+
]);
|
|
106
|
+
export const missionsPg = pgTable("missions", {
|
|
107
|
+
id: pgText("id").primaryKey(),
|
|
108
|
+
name: pgText("name").notNull().unique(),
|
|
109
|
+
data: pgText("data").notNull(),
|
|
110
|
+
prompt: pgText("prompt"),
|
|
111
|
+
status: varchar("status", { length: 32 }).notNull().default("draft"),
|
|
112
|
+
schedule: pgText("schedule"),
|
|
113
|
+
endDate: pgText("end_date"),
|
|
114
|
+
qualityThreshold: pgText("quality_threshold"),
|
|
115
|
+
deadline: pgText("deadline"),
|
|
116
|
+
notifications: jsonb("notifications"),
|
|
117
|
+
executionCount: pgInteger("execution_count").notNull().default(0),
|
|
118
|
+
createdAt: pgText("created_at").notNull(),
|
|
119
|
+
updatedAt: pgText("updated_at").notNull(),
|
|
120
|
+
}, (table) => [
|
|
121
|
+
pgIndex("idx_pg_missions_status").on(table.status),
|
|
122
|
+
]);
|
|
123
|
+
export const metadataPg = pgTable("metadata", {
|
|
124
|
+
key: pgText("key").primaryKey(),
|
|
125
|
+
value: pgText("value").notNull(),
|
|
126
|
+
});
|
|
127
|
+
export const processesPg = pgTable("processes", {
|
|
128
|
+
agentName: pgText("agent_name").notNull(),
|
|
129
|
+
pid: pgInteger("pid").notNull(),
|
|
130
|
+
taskId: pgText("task_id").notNull(),
|
|
131
|
+
startedAt: pgText("started_at").notNull(),
|
|
132
|
+
alive: pgInteger("alive").notNull().default(1),
|
|
133
|
+
activity: jsonb("activity").notNull().default({}),
|
|
134
|
+
});
|
|
135
|
+
//# sourceMappingURL=tasks.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tasks.js","sourceRoot":"","sources":["../../src/schema/tasks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAC5E,OAAO,EAAE,OAAO,EAAE,IAAI,IAAI,MAAM,EAAE,OAAO,IAAI,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEtH,0EAA0E;AAE1E,MAAM,CAAC,MAAM,WAAW,GAAG,WAAW,CAAC,OAAO,EAAE;IAC9C,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE;IAC9B,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE;IAC1C,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE;IACrC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC;IACpB,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;IAC7B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;IACrD,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC;IACnD,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAChD,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACvD,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC;IACpC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC;IACjC,YAAY,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;IAC1D,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;IAChD,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC;IACtB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,kBAAkB,EAAE,OAAO,CAAC,qBAAqB,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACvE,mBAAmB,EAAE,IAAI,CAAC,sBAAsB,CAAC;IACjD,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC;IAC7B,aAAa,EAAE,IAAI,CAAC,eAAe,CAAC;IACpC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC;IAC1B,gBAAgB,EAAE,IAAI,CAAC,mBAAmB,CAAC;IAC3C,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC;IAC1B,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC;IAC1B,WAAW,EAAE,OAAO,CAAC,cAAc,CAAC;IACpC,aAAa,EAAE,OAAO,CAAC,gBAAgB,CAAC;IACxC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CACxC,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;IACZ,KAAK,CAAC,kBAAkB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;IAC1C,KAAK,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC;IACxC,KAAK,CAAC,qBAAqB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC;IAC/C,KAAK,CAAC,sBAAsB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;CAClD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,EAAE;IACpD,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC3B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC,MAAM,EAAE;IACrC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE;IAC5B,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC;IACtB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;IACjD,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC;IAC1B,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC;IACzB,gBAAgB,EAAE,IAAI,CAAC,mBAAmB,CAAC;IAC3C,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC;IAC1B,aAAa,EAAE,IAAI,CAAC,eAAe,CAAC;IACpC,cAAc,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/D,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CACxC,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;IACZ,KAAK,CAAC,qBAAqB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;CAC9C,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,cAAc,GAAG,WAAW,CAAC,UAAU,EAAE;IACpD,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE;IAC7B,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE;CAC/B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC,WAAW,EAAE;IACtD,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,GAAG,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE;IAC7B,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IACjC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACvC,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC5C,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;CACnD,CAAC,CAAC;AAEH,0EAA0E;AAE1E,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE;IACtC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC7B,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE;IAChC,WAAW,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE;IAC5C,QAAQ,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE;IACvC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC;IAC/B,SAAS,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IACpD,MAAM,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC;IACtE,OAAO,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAClD,UAAU,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,WAAW,EAAE,SAAS,CAAC,cAAc,CAAC;IACtC,WAAW,EAAE,KAAK,CAAC,cAAc,CAAC;IAClC,YAAY,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IACzD,OAAO,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/C,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC;IACvB,KAAK,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACvC,WAAW,EAAE,SAAS,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3D,kBAAkB,EAAE,SAAS,CAAC,qBAAqB,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACzE,mBAAmB,EAAE,MAAM,CAAC,sBAAsB,CAAC;IACnD,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC;IAC/B,aAAa,EAAE,KAAK,CAAC,eAAe,CAAC;IACrC,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC;IAC3B,gBAAgB,EAAE,KAAK,CAAC,mBAAmB,CAAC;IAC5C,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC;IAC5B,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC;IAC5B,WAAW,EAAE,SAAS,CAAC,cAAc,CAAC;IACtC,aAAa,EAAE,SAAS,CAAC,gBAAgB,CAAC;IAC1C,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACzC,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;IACZ,OAAO,CAAC,qBAAqB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;IAC/C,OAAO,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC;IAC7C,OAAO,CAAC,wBAAwB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC;IACpD,OAAO,CAAC,yBAAyB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;CACvD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,EAAE;IAC5C,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE;IAC7B,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC,MAAM,EAAE;IACvC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE;IAC9B,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC;IACxB,MAAM,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;IACpE,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC;IAC5B,OAAO,EAAE,MAAM,CAAC,UAAU,CAAC;IAC3B,gBAAgB,EAAE,MAAM,CAAC,mBAAmB,CAAC;IAC7C,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC;IAC5B,aAAa,EAAE,KAAK,CAAC,eAAe,CAAC;IACrC,cAAc,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IACjE,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACzC,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CAC1C,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;IACZ,OAAO,CAAC,wBAAwB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;CACnD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,EAAE;IAC5C,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE;IAC/B,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE;CACjC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE;IAC9C,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACzC,GAAG,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE;IAC/B,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IACnC,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;IACzC,KAAK,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9C,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;CAClD,CAAC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { ApprovalStore } from "@polpo-ai/core/approval-store";
|
|
2
|
+
import type { ApprovalRequest, ApprovalStatus } from "@polpo-ai/core/types";
|
|
3
|
+
import { type Dialect } from "../utils.js";
|
|
4
|
+
type AnyTable = any;
|
|
5
|
+
export declare class DrizzleApprovalStore implements ApprovalStore {
|
|
6
|
+
private db;
|
|
7
|
+
private approvals;
|
|
8
|
+
private dialect;
|
|
9
|
+
constructor(db: any, approvals: AnyTable, dialect: Dialect);
|
|
10
|
+
private rowToRequest;
|
|
11
|
+
upsert(request: ApprovalRequest): Promise<void>;
|
|
12
|
+
get(id: string): Promise<ApprovalRequest | undefined>;
|
|
13
|
+
list(status?: ApprovalStatus): Promise<ApprovalRequest[]>;
|
|
14
|
+
listByTask(taskId: string): Promise<ApprovalRequest[]>;
|
|
15
|
+
delete(id: string): Promise<boolean>;
|
|
16
|
+
close(): Promise<void>;
|
|
17
|
+
}
|
|
18
|
+
export {};
|
|
19
|
+
//# sourceMappingURL=approval-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"approval-store.d.ts","sourceRoot":"","sources":["../../src/stores/approval-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AACnE,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC5E,OAAO,EAAE,KAAK,OAAO,EAAkC,MAAM,aAAa,CAAC;AAE3E,KAAK,QAAQ,GAAG,GAAG,CAAC;AAEpB,qBAAa,oBAAqB,YAAW,aAAa;IAEtD,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,OAAO;gBAFP,EAAE,EAAE,GAAG,EACP,SAAS,EAAE,QAAQ,EACnB,OAAO,EAAE,OAAO;IAG1B,OAAO,CAAC,YAAY;IAgBd,MAAM,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IA2B/C,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;IAMrD,IAAI,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAOzD,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAMtD,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAMpC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAC7B"}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
import { eq } from "drizzle-orm";
|
|
2
|
+
import { serializeJson, deserializeJson } from "../utils.js";
|
|
3
|
+
export class DrizzleApprovalStore {
|
|
4
|
+
db;
|
|
5
|
+
approvals;
|
|
6
|
+
dialect;
|
|
7
|
+
constructor(db, approvals, dialect) {
|
|
8
|
+
this.db = db;
|
|
9
|
+
this.approvals = approvals;
|
|
10
|
+
this.dialect = dialect;
|
|
11
|
+
}
|
|
12
|
+
rowToRequest(row) {
|
|
13
|
+
return {
|
|
14
|
+
id: row.id,
|
|
15
|
+
gateId: row.gateId,
|
|
16
|
+
gateName: row.gateName,
|
|
17
|
+
taskId: row.taskId ?? undefined,
|
|
18
|
+
missionId: row.missionId ?? undefined,
|
|
19
|
+
status: row.status,
|
|
20
|
+
payload: deserializeJson(row.payload, undefined, this.dialect),
|
|
21
|
+
requestedAt: row.requestedAt,
|
|
22
|
+
resolvedAt: row.resolvedAt ?? undefined,
|
|
23
|
+
resolvedBy: row.resolvedBy ?? undefined,
|
|
24
|
+
note: row.note ?? undefined,
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
async upsert(request) {
|
|
28
|
+
const values = {
|
|
29
|
+
id: request.id,
|
|
30
|
+
gateId: request.gateId,
|
|
31
|
+
gateName: request.gateName,
|
|
32
|
+
taskId: request.taskId ?? null,
|
|
33
|
+
missionId: request.missionId ?? null,
|
|
34
|
+
status: request.status,
|
|
35
|
+
payload: serializeJson(request.payload, this.dialect),
|
|
36
|
+
requestedAt: request.requestedAt,
|
|
37
|
+
resolvedAt: request.resolvedAt ?? null,
|
|
38
|
+
resolvedBy: request.resolvedBy ?? null,
|
|
39
|
+
note: request.note ?? null,
|
|
40
|
+
};
|
|
41
|
+
await this.db.insert(this.approvals).values(values)
|
|
42
|
+
.onConflictDoUpdate({
|
|
43
|
+
target: this.approvals.id,
|
|
44
|
+
set: {
|
|
45
|
+
status: values.status,
|
|
46
|
+
payload: values.payload,
|
|
47
|
+
resolvedAt: values.resolvedAt,
|
|
48
|
+
resolvedBy: values.resolvedBy,
|
|
49
|
+
note: values.note,
|
|
50
|
+
},
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
async get(id) {
|
|
54
|
+
const rows = await this.db.select().from(this.approvals)
|
|
55
|
+
.where(eq(this.approvals.id, id));
|
|
56
|
+
return rows.length > 0 ? this.rowToRequest(rows[0]) : undefined;
|
|
57
|
+
}
|
|
58
|
+
async list(status) {
|
|
59
|
+
let q = this.db.select().from(this.approvals);
|
|
60
|
+
if (status)
|
|
61
|
+
q = q.where(eq(this.approvals.status, status));
|
|
62
|
+
const rows = await q;
|
|
63
|
+
return rows.map((r) => this.rowToRequest(r));
|
|
64
|
+
}
|
|
65
|
+
async listByTask(taskId) {
|
|
66
|
+
const rows = await this.db.select().from(this.approvals)
|
|
67
|
+
.where(eq(this.approvals.taskId, taskId));
|
|
68
|
+
return rows.map((r) => this.rowToRequest(r));
|
|
69
|
+
}
|
|
70
|
+
async delete(id) {
|
|
71
|
+
const result = await this.db.delete(this.approvals)
|
|
72
|
+
.where(eq(this.approvals.id, id));
|
|
73
|
+
return (result?.rowCount ?? result?.changes ?? 0) > 0;
|
|
74
|
+
}
|
|
75
|
+
async close() { }
|
|
76
|
+
}
|
|
77
|
+
//# sourceMappingURL=approval-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"approval-store.js","sourceRoot":"","sources":["../../src/stores/approval-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAGjC,OAAO,EAAgB,aAAa,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAI3E,MAAM,OAAO,oBAAoB;IAErB;IACA;IACA;IAHV,YACU,EAAO,EACP,SAAmB,EACnB,OAAgB;QAFhB,OAAE,GAAF,EAAE,CAAK;QACP,cAAS,GAAT,SAAS,CAAU;QACnB,YAAO,GAAP,OAAO,CAAS;IACvB,CAAC;IAEI,YAAY,CAAC,GAAQ;QAC3B,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,SAAS;YAC/B,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,SAAS;YACrC,MAAM,EAAE,GAAG,CAAC,MAAwB;YACpC,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC;YAC9D,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,SAAS;YACvC,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,SAAS;YACvC,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,SAAS;SAC5B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAwB;QACnC,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI;YAC9B,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;YACpC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,aAAa,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC;YACrD,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,IAAI;YACtC,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,IAAI;YACtC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;SAC3B,CAAC;QACF,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;aAChD,kBAAkB,CAAC;YAClB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE;YACzB,GAAG,EAAE;gBACH,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB;SACF,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,EAAU;QAClB,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;aAC5D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACpC,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAuB;QAChC,IAAI,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,MAAM;YAAE,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;QAC3D,MAAM,IAAI,GAAU,MAAM,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc;QAC7B,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;aAC5D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;QAC5C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC;aAChD,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACpC,OAAO,CAAC,MAAM,EAAE,QAAQ,IAAI,MAAM,EAAE,OAAO,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,KAAK,KAAmB,CAAC;CAChC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { CheckpointStore, CheckpointState } from "@polpo-ai/core/checkpoint-store";
|
|
2
|
+
import { type Dialect } from "../utils.js";
|
|
3
|
+
type AnyTable = any;
|
|
4
|
+
export declare class DrizzleCheckpointStore implements CheckpointStore {
|
|
5
|
+
private db;
|
|
6
|
+
private metadata;
|
|
7
|
+
private dialect;
|
|
8
|
+
constructor(db: any, metadata: AnyTable, dialect: Dialect);
|
|
9
|
+
load(): Promise<CheckpointState>;
|
|
10
|
+
save(state: CheckpointState): Promise<void>;
|
|
11
|
+
removeGroup(state: CheckpointState, group: string): Promise<CheckpointState>;
|
|
12
|
+
}
|
|
13
|
+
export {};
|
|
14
|
+
//# sourceMappingURL=checkpoint-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"checkpoint-store.d.ts","sourceRoot":"","sources":["../../src/stores/checkpoint-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AACxF,OAAO,EAAE,KAAK,OAAO,EAAmB,MAAM,aAAa,CAAC;AAE5D,KAAK,QAAQ,GAAG,GAAG,CAAC;AAMpB,qBAAa,sBAAuB,YAAW,eAAe;IAE1D,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,OAAO;gBAFP,EAAE,EAAE,GAAG,EACP,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,OAAO;IAGpB,IAAI,IAAI,OAAO,CAAC,eAAe,CAAC;IAOhC,IAAI,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ3C,WAAW,CAAC,KAAK,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;CAcnF"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { eq } from "drizzle-orm";
|
|
2
|
+
import { deserializeJson } from "../utils.js";
|
|
3
|
+
const CHECKPOINT_KEY = "checkpoints";
|
|
4
|
+
const EMPTY_STATE = { definitions: {}, active: {}, resumed: [] };
|
|
5
|
+
export class DrizzleCheckpointStore {
|
|
6
|
+
db;
|
|
7
|
+
metadata;
|
|
8
|
+
dialect;
|
|
9
|
+
constructor(db, metadata, dialect) {
|
|
10
|
+
this.db = db;
|
|
11
|
+
this.metadata = metadata;
|
|
12
|
+
this.dialect = dialect;
|
|
13
|
+
}
|
|
14
|
+
async load() {
|
|
15
|
+
const rows = await this.db.select().from(this.metadata)
|
|
16
|
+
.where(eq(this.metadata.key, CHECKPOINT_KEY));
|
|
17
|
+
if (rows.length === 0)
|
|
18
|
+
return { ...EMPTY_STATE };
|
|
19
|
+
return deserializeJson(rows[0].value, { ...EMPTY_STATE }, this.dialect);
|
|
20
|
+
}
|
|
21
|
+
async save(state) {
|
|
22
|
+
const value = this.dialect === "sqlite"
|
|
23
|
+
? JSON.stringify(state)
|
|
24
|
+
: JSON.stringify(state); // metadata.value is always TEXT, not JSONB
|
|
25
|
+
await this.db.insert(this.metadata).values({ key: CHECKPOINT_KEY, value })
|
|
26
|
+
.onConflictDoUpdate({ target: this.metadata.key, set: { value } });
|
|
27
|
+
}
|
|
28
|
+
async removeGroup(state, group) {
|
|
29
|
+
const { definitions, active, resumed } = state;
|
|
30
|
+
const next = {
|
|
31
|
+
definitions: { ...definitions },
|
|
32
|
+
active: { ...active },
|
|
33
|
+
resumed: resumed.filter((r) => !r.startsWith(`${group}:`)),
|
|
34
|
+
};
|
|
35
|
+
delete next.definitions[group];
|
|
36
|
+
for (const key of Object.keys(next.active)) {
|
|
37
|
+
if (key.startsWith(`${group}:`))
|
|
38
|
+
delete next.active[key];
|
|
39
|
+
}
|
|
40
|
+
await this.save(next);
|
|
41
|
+
return next;
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=checkpoint-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"checkpoint-store.js","sourceRoot":"","sources":["../../src/stores/checkpoint-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAEjC,OAAO,EAAgB,eAAe,EAAE,MAAM,aAAa,CAAC;AAI5D,MAAM,cAAc,GAAG,aAAa,CAAC;AAErC,MAAM,WAAW,GAAoB,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AAElF,MAAM,OAAO,sBAAsB;IAEvB;IACA;IACA;IAHV,YACU,EAAO,EACP,QAAkB,EAClB,OAAgB;QAFhB,OAAE,GAAF,EAAE,CAAK;QACP,aAAQ,GAAR,QAAQ,CAAU;QAClB,YAAO,GAAP,OAAO,CAAS;IACvB,CAAC;IAEJ,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;aAC3D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC,CAAC;QAChD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,GAAG,WAAW,EAAE,CAAC;QACjD,OAAO,eAAe,CAAkB,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE,GAAG,WAAW,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3F,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAsB;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,KAAK,QAAQ;YACrC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;YACvB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,2CAA2C;QACtE,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;aACvE,kBAAkB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAsB,EAAE,KAAa;QACrD,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;QAC/C,MAAM,IAAI,GAAoB;YAC5B,WAAW,EAAE,EAAE,GAAG,WAAW,EAAE;YAC/B,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE;YACrB,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;SAC3D,CAAC;QACF,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC/B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,KAAK,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { ConfigStore } from "@polpo-ai/core/config-store";
|
|
2
|
+
import type { PolpoConfig } from "@polpo-ai/core/types";
|
|
3
|
+
import { type Dialect } from "../utils.js";
|
|
4
|
+
type AnyTable = any;
|
|
5
|
+
export declare class DrizzleConfigStore implements ConfigStore {
|
|
6
|
+
private db;
|
|
7
|
+
private metadata;
|
|
8
|
+
private dialect;
|
|
9
|
+
constructor(db: any, metadata: AnyTable, dialect: Dialect);
|
|
10
|
+
exists(): Promise<boolean>;
|
|
11
|
+
get(): Promise<PolpoConfig | undefined>;
|
|
12
|
+
save(config: PolpoConfig): Promise<void>;
|
|
13
|
+
}
|
|
14
|
+
export {};
|
|
15
|
+
//# sourceMappingURL=config-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config-store.d.ts","sourceRoot":"","sources":["../../src/stores/config-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,KAAK,OAAO,EAAmB,MAAM,aAAa,CAAC;AAE5D,KAAK,QAAQ,GAAG,GAAG,CAAC;AAIpB,qBAAa,kBAAmB,YAAW,WAAW;IAElD,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,OAAO;gBAFP,EAAE,EAAE,GAAG,EACP,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,OAAO;IAGpB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC;IAM1B,GAAG,IAAI,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC;IAOvC,IAAI,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;CAK/C"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import { eq } from "drizzle-orm";
|
|
2
|
+
import { deserializeJson } from "../utils.js";
|
|
3
|
+
const CONFIG_KEY = "config";
|
|
4
|
+
export class DrizzleConfigStore {
|
|
5
|
+
db;
|
|
6
|
+
metadata;
|
|
7
|
+
dialect;
|
|
8
|
+
constructor(db, metadata, dialect) {
|
|
9
|
+
this.db = db;
|
|
10
|
+
this.metadata = metadata;
|
|
11
|
+
this.dialect = dialect;
|
|
12
|
+
}
|
|
13
|
+
async exists() {
|
|
14
|
+
const rows = await this.db.select().from(this.metadata)
|
|
15
|
+
.where(eq(this.metadata.key, CONFIG_KEY));
|
|
16
|
+
return rows.length > 0;
|
|
17
|
+
}
|
|
18
|
+
async get() {
|
|
19
|
+
const rows = await this.db.select().from(this.metadata)
|
|
20
|
+
.where(eq(this.metadata.key, CONFIG_KEY));
|
|
21
|
+
if (rows.length === 0)
|
|
22
|
+
return undefined;
|
|
23
|
+
return deserializeJson(rows[0].value, undefined, this.dialect);
|
|
24
|
+
}
|
|
25
|
+
async save(config) {
|
|
26
|
+
const value = JSON.stringify(config);
|
|
27
|
+
await this.db.insert(this.metadata).values({ key: CONFIG_KEY, value })
|
|
28
|
+
.onConflictDoUpdate({ target: this.metadata.key, set: { value } });
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=config-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config-store.js","sourceRoot":"","sources":["../../src/stores/config-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAGjC,OAAO,EAAgB,eAAe,EAAE,MAAM,aAAa,CAAC;AAI5D,MAAM,UAAU,GAAG,QAAQ,CAAC;AAE5B,MAAM,OAAO,kBAAkB;IAEnB;IACA;IACA;IAHV,YACU,EAAO,EACP,QAAkB,EAClB,OAAgB;QAFhB,OAAE,GAAF,EAAE,CAAK;QACP,aAAQ,GAAR,QAAQ,CAAU;QAClB,YAAO,GAAP,OAAO,CAAS;IACvB,CAAC;IAEJ,KAAK,CAAC,MAAM;QACV,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;aAC3D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,CAAC;QAC5C,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,GAAG;QACP,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;aAC3D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QACxC,OAAO,eAAe,CAA0B,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1F,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAmB;QAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;aACnE,kBAAkB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IACvE,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { DelayStore, DelayState } from "@polpo-ai/core/delay-store";
|
|
2
|
+
import { type Dialect } from "../utils.js";
|
|
3
|
+
type AnyTable = any;
|
|
4
|
+
export declare class DrizzleDelayStore implements DelayStore {
|
|
5
|
+
private db;
|
|
6
|
+
private metadata;
|
|
7
|
+
private dialect;
|
|
8
|
+
constructor(db: any, metadata: AnyTable, dialect: Dialect);
|
|
9
|
+
load(): Promise<DelayState>;
|
|
10
|
+
save(state: DelayState): Promise<void>;
|
|
11
|
+
removeGroup(state: DelayState, group: string): Promise<DelayState>;
|
|
12
|
+
}
|
|
13
|
+
export {};
|
|
14
|
+
//# sourceMappingURL=delay-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delay-store.d.ts","sourceRoot":"","sources":["../../src/stores/delay-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAE,KAAK,OAAO,EAAmB,MAAM,aAAa,CAAC;AAE5D,KAAK,QAAQ,GAAG,GAAG,CAAC;AAMpB,qBAAa,iBAAkB,YAAW,UAAU;IAEhD,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,OAAO;gBAFP,EAAE,EAAE,GAAG,EACP,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,OAAO;IAGpB,IAAI,IAAI,OAAO,CAAC,UAAU,CAAC;IAO3B,IAAI,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAMtC,WAAW,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;CAczE"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { eq } from "drizzle-orm";
|
|
2
|
+
import { deserializeJson } from "../utils.js";
|
|
3
|
+
const DELAY_KEY = "delays";
|
|
4
|
+
const EMPTY_STATE = { definitions: {}, active: {}, expired: [] };
|
|
5
|
+
export class DrizzleDelayStore {
|
|
6
|
+
db;
|
|
7
|
+
metadata;
|
|
8
|
+
dialect;
|
|
9
|
+
constructor(db, metadata, dialect) {
|
|
10
|
+
this.db = db;
|
|
11
|
+
this.metadata = metadata;
|
|
12
|
+
this.dialect = dialect;
|
|
13
|
+
}
|
|
14
|
+
async load() {
|
|
15
|
+
const rows = await this.db.select().from(this.metadata)
|
|
16
|
+
.where(eq(this.metadata.key, DELAY_KEY));
|
|
17
|
+
if (rows.length === 0)
|
|
18
|
+
return { ...EMPTY_STATE };
|
|
19
|
+
return deserializeJson(rows[0].value, { ...EMPTY_STATE }, this.dialect);
|
|
20
|
+
}
|
|
21
|
+
async save(state) {
|
|
22
|
+
const value = JSON.stringify(state);
|
|
23
|
+
await this.db.insert(this.metadata).values({ key: DELAY_KEY, value })
|
|
24
|
+
.onConflictDoUpdate({ target: this.metadata.key, set: { value } });
|
|
25
|
+
}
|
|
26
|
+
async removeGroup(state, group) {
|
|
27
|
+
const { definitions, active, expired } = state;
|
|
28
|
+
const next = {
|
|
29
|
+
definitions: { ...definitions },
|
|
30
|
+
active: { ...active },
|
|
31
|
+
expired: expired.filter((e) => !e.startsWith(`${group}:`)),
|
|
32
|
+
};
|
|
33
|
+
delete next.definitions[group];
|
|
34
|
+
for (const key of Object.keys(next.active)) {
|
|
35
|
+
if (key.startsWith(`${group}:`))
|
|
36
|
+
delete next.active[key];
|
|
37
|
+
}
|
|
38
|
+
await this.save(next);
|
|
39
|
+
return next;
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
//# sourceMappingURL=delay-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delay-store.js","sourceRoot":"","sources":["../../src/stores/delay-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAC;AAEjC,OAAO,EAAgB,eAAe,EAAE,MAAM,aAAa,CAAC;AAI5D,MAAM,SAAS,GAAG,QAAQ,CAAC;AAE3B,MAAM,WAAW,GAAe,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AAE7E,MAAM,OAAO,iBAAiB;IAElB;IACA;IACA;IAHV,YACU,EAAO,EACP,QAAkB,EAClB,OAAgB;QAFhB,OAAE,GAAF,EAAE,CAAK;QACP,aAAQ,GAAR,QAAQ,CAAU;QAClB,YAAO,GAAP,OAAO,CAAS;IACvB,CAAC;IAEJ,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC;aAC3D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;QAC3C,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,GAAG,WAAW,EAAE,CAAC;QACjD,OAAO,eAAe,CAAa,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE,GAAG,WAAW,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAiB;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;aAClE,kBAAkB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAiB,EAAE,KAAa;QAChD,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;QAC/C,MAAM,IAAI,GAAe;YACvB,WAAW,EAAE,EAAE,GAAG,WAAW,EAAE;YAC/B,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE;YACrB,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC;SAC3D,CAAC;QACF,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC/B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,KAAK,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
export { DrizzleTaskStore } from "./task-store.js";
|
|
2
|
+
export type { TaskStoreSchema } from "./task-store.js";
|
|
3
|
+
export { DrizzleRunStore } from "./run-store.js";
|
|
4
|
+
export { DrizzleSessionStore } from "./session-store.js";
|
|
5
|
+
export { DrizzleNotificationStore } from "./notification-store.js";
|
|
6
|
+
export { DrizzleLogStore } from "./log-store.js";
|
|
7
|
+
export { DrizzleApprovalStore } from "./approval-store.js";
|
|
8
|
+
export { DrizzleMemoryStore } from "./memory-store.js";
|
|
9
|
+
export { DrizzlePeerStore } from "./peer-store.js";
|
|
10
|
+
export { DrizzleCheckpointStore } from "./checkpoint-store.js";
|
|
11
|
+
export { DrizzleDelayStore } from "./delay-store.js";
|
|
12
|
+
export { DrizzleConfigStore } from "./config-store.js";
|
|
13
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/stores/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,YAAY,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
export { DrizzleTaskStore } from "./task-store.js";
|
|
2
|
+
export { DrizzleRunStore } from "./run-store.js";
|
|
3
|
+
export { DrizzleSessionStore } from "./session-store.js";
|
|
4
|
+
export { DrizzleNotificationStore } from "./notification-store.js";
|
|
5
|
+
export { DrizzleLogStore } from "./log-store.js";
|
|
6
|
+
export { DrizzleApprovalStore } from "./approval-store.js";
|
|
7
|
+
export { DrizzleMemoryStore } from "./memory-store.js";
|
|
8
|
+
export { DrizzlePeerStore } from "./peer-store.js";
|
|
9
|
+
export { DrizzleCheckpointStore } from "./checkpoint-store.js";
|
|
10
|
+
export { DrizzleDelayStore } from "./delay-store.js";
|
|
11
|
+
export { DrizzleConfigStore } from "./config-store.js";
|
|
12
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/stores/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAEnD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { LogStore, LogEntry, SessionInfo } from "@polpo-ai/core/log-store";
|
|
2
|
+
import { type Dialect } from "../utils.js";
|
|
3
|
+
type AnyTable = any;
|
|
4
|
+
export declare class DrizzleLogStore implements LogStore {
|
|
5
|
+
private db;
|
|
6
|
+
private logSessions;
|
|
7
|
+
private logEntries;
|
|
8
|
+
private dialect;
|
|
9
|
+
private currentSessionId;
|
|
10
|
+
constructor(db: any, logSessions: AnyTable, logEntries: AnyTable, dialect: Dialect);
|
|
11
|
+
startSession(): Promise<string>;
|
|
12
|
+
getSessionId(): Promise<string | undefined>;
|
|
13
|
+
append(entry: LogEntry): Promise<void>;
|
|
14
|
+
getSessionEntries(sessionId?: string): Promise<LogEntry[]>;
|
|
15
|
+
listSessions(): Promise<SessionInfo[]>;
|
|
16
|
+
prune(keepSessions: number): Promise<number>;
|
|
17
|
+
close(): Promise<void>;
|
|
18
|
+
}
|
|
19
|
+
export {};
|
|
20
|
+
//# sourceMappingURL=log-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"log-store.d.ts","sourceRoot":"","sources":["../../src/stores/log-store.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAChF,OAAO,EAAE,KAAK,OAAO,EAAkC,MAAM,aAAa,CAAC;AAE3E,KAAK,QAAQ,GAAG,GAAG,CAAC;AAEpB,qBAAa,eAAgB,YAAW,QAAQ;IAI5C,OAAO,CAAC,EAAE;IACV,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,OAAO;IANjB,OAAO,CAAC,gBAAgB,CAAqB;gBAGnC,EAAE,EAAE,GAAG,EACP,WAAW,EAAE,QAAQ,EACrB,UAAU,EAAE,QAAQ,EACpB,OAAO,EAAE,OAAO;IAGpB,YAAY,IAAI,OAAO,CAAC,MAAM,CAAC;IAQ/B,YAAY,IAAI,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAI3C,MAAM,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IAatC,iBAAiB,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAe1D,YAAY,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAmBtC,KAAK,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAiB5C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG7B"}
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import { eq, desc, asc, sql } from "drizzle-orm";
|
|
2
|
+
import { nanoid } from "nanoid";
|
|
3
|
+
import { serializeJson, deserializeJson } from "../utils.js";
|
|
4
|
+
export class DrizzleLogStore {
|
|
5
|
+
db;
|
|
6
|
+
logSessions;
|
|
7
|
+
logEntries;
|
|
8
|
+
dialect;
|
|
9
|
+
currentSessionId;
|
|
10
|
+
constructor(db, logSessions, logEntries, dialect) {
|
|
11
|
+
this.db = db;
|
|
12
|
+
this.logSessions = logSessions;
|
|
13
|
+
this.logEntries = logEntries;
|
|
14
|
+
this.dialect = dialect;
|
|
15
|
+
}
|
|
16
|
+
async startSession() {
|
|
17
|
+
const id = nanoid(10);
|
|
18
|
+
const now = new Date().toISOString();
|
|
19
|
+
await this.db.insert(this.logSessions).values({ id, startedAt: now });
|
|
20
|
+
this.currentSessionId = id;
|
|
21
|
+
return id;
|
|
22
|
+
}
|
|
23
|
+
async getSessionId() {
|
|
24
|
+
return this.currentSessionId;
|
|
25
|
+
}
|
|
26
|
+
async append(entry) {
|
|
27
|
+
if (!this.currentSessionId) {
|
|
28
|
+
await this.startSession();
|
|
29
|
+
}
|
|
30
|
+
await this.db.insert(this.logEntries).values({
|
|
31
|
+
id: nanoid(),
|
|
32
|
+
sessionId: this.currentSessionId,
|
|
33
|
+
ts: entry.ts,
|
|
34
|
+
event: entry.event,
|
|
35
|
+
data: serializeJson(entry.data, this.dialect),
|
|
36
|
+
});
|
|
37
|
+
}
|
|
38
|
+
async getSessionEntries(sessionId) {
|
|
39
|
+
const sid = sessionId ?? this.currentSessionId;
|
|
40
|
+
if (!sid)
|
|
41
|
+
return [];
|
|
42
|
+
const rows = await this.db.select().from(this.logEntries)
|
|
43
|
+
.where(eq(this.logEntries.sessionId, sid))
|
|
44
|
+
.orderBy(asc(this.logEntries.ts));
|
|
45
|
+
return rows.map((r) => ({
|
|
46
|
+
ts: r.ts,
|
|
47
|
+
event: r.event,
|
|
48
|
+
data: deserializeJson(r.data, null, this.dialect),
|
|
49
|
+
}));
|
|
50
|
+
}
|
|
51
|
+
async listSessions() {
|
|
52
|
+
const rows = await this.db
|
|
53
|
+
.select({
|
|
54
|
+
sessionId: this.logSessions.id,
|
|
55
|
+
startedAt: this.logSessions.startedAt,
|
|
56
|
+
entries: sql `count(${this.logEntries.id})`,
|
|
57
|
+
})
|
|
58
|
+
.from(this.logSessions)
|
|
59
|
+
.leftJoin(this.logEntries, eq(this.logSessions.id, this.logEntries.sessionId))
|
|
60
|
+
.groupBy(this.logSessions.id)
|
|
61
|
+
.orderBy(desc(this.logSessions.startedAt));
|
|
62
|
+
return rows.map((r) => ({
|
|
63
|
+
sessionId: r.sessionId,
|
|
64
|
+
startedAt: r.startedAt,
|
|
65
|
+
entries: Number(r.entries),
|
|
66
|
+
}));
|
|
67
|
+
}
|
|
68
|
+
async prune(keepSessions) {
|
|
69
|
+
const all = await this.db.select({ id: this.logSessions.id })
|
|
70
|
+
.from(this.logSessions)
|
|
71
|
+
.orderBy(desc(this.logSessions.startedAt));
|
|
72
|
+
if (all.length <= keepSessions)
|
|
73
|
+
return 0;
|
|
74
|
+
const toDelete = all.slice(keepSessions).map((r) => r.id);
|
|
75
|
+
let deleted = 0;
|
|
76
|
+
for (const id of toDelete) {
|
|
77
|
+
// Entries cascade-deleted via FK
|
|
78
|
+
await this.db.delete(this.logSessions).where(eq(this.logSessions.id, id));
|
|
79
|
+
deleted++;
|
|
80
|
+
}
|
|
81
|
+
return deleted;
|
|
82
|
+
}
|
|
83
|
+
async close() {
|
|
84
|
+
this.currentSessionId = undefined;
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
//# sourceMappingURL=log-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"log-store.js","sourceRoot":"","sources":["../../src/stores/log-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAgB,aAAa,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAI3E,MAAM,OAAO,eAAe;IAIhB;IACA;IACA;IACA;IANF,gBAAgB,CAAqB;IAE7C,YACU,EAAO,EACP,WAAqB,EACrB,UAAoB,EACpB,OAAgB;QAHhB,OAAE,GAAF,EAAE,CAAK;QACP,gBAAW,GAAX,WAAW,CAAU;QACrB,eAAU,GAAV,UAAU,CAAU;QACpB,YAAO,GAAP,OAAO,CAAS;IACvB,CAAC;IAEJ,KAAK,CAAC,YAAY;QAChB,MAAM,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC;QACtB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QACtE,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAe;QAC1B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC5B,CAAC;QACD,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC;YAC3C,EAAE,EAAE,MAAM,EAAE;YACZ,SAAS,EAAE,IAAI,CAAC,gBAAiB;YACjC,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,IAAI,EAAE,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC;SAC9C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,SAAkB;QACxC,MAAM,GAAG,GAAG,SAAS,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC/C,IAAI,CAAC,GAAG;YAAE,OAAO,EAAE,CAAC;QAEpB,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;aAC7D,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;aACzC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC;QAEpC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC;SAClD,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,MAAM,IAAI,GAAU,MAAM,IAAI,CAAC,EAAE;aAC9B,MAAM,CAAC;YACN,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,EAAE;YAC9B,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS;YACrC,OAAO,EAAE,GAAG,CAAQ,SAAS,IAAI,CAAC,UAAU,CAAC,EAAE,GAAG;SACnD,CAAC;aACD,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;aACtB,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;aAC7E,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;aAC5B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC;QAE7C,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;SAC3B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,YAAoB;QAC9B,MAAM,GAAG,GAAU,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC;aACjE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;aACtB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC;QAE7C,IAAI,GAAG,CAAC,MAAM,IAAI,YAAY;YAAE,OAAO,CAAC,CAAC;QAEzC,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1D,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,iCAAiC;YACjC,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1E,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC;IACpC,CAAC;CACF"}
|