@cogitator-ai/channels 0.2.4 → 0.3.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/README.md +141 -39
- package/dist/assistant-config.d.ts +1 -0
- package/dist/assistant-config.d.ts.map +1 -1
- package/dist/assistant-config.js +1 -0
- package/dist/assistant-config.js.map +1 -1
- package/dist/channels/discord.d.ts +4 -0
- package/dist/channels/discord.d.ts.map +1 -1
- package/dist/channels/discord.js +123 -59
- package/dist/channels/discord.js.map +1 -1
- package/dist/channels/slack.d.ts +9 -0
- package/dist/channels/slack.d.ts.map +1 -1
- package/dist/channels/slack.js +153 -34
- package/dist/channels/slack.js.map +1 -1
- package/dist/channels/telegram.d.ts +13 -0
- package/dist/channels/telegram.d.ts.map +1 -1
- package/dist/channels/telegram.js +206 -111
- package/dist/channels/telegram.js.map +1 -1
- package/dist/channels/terminal.d.ts +10 -0
- package/dist/channels/terminal.d.ts.map +1 -1
- package/dist/channels/terminal.js +46 -10
- package/dist/channels/terminal.js.map +1 -1
- package/dist/channels/webchat.d.ts +7 -2
- package/dist/channels/webchat.d.ts.map +1 -1
- package/dist/channels/webchat.js +112 -50
- package/dist/channels/webchat.js.map +1 -1
- package/dist/channels/whatsapp.d.ts +19 -3
- package/dist/channels/whatsapp.d.ts.map +1 -1
- package/dist/channels/whatsapp.js +204 -66
- package/dist/channels/whatsapp.js.map +1 -1
- package/dist/cron.d.ts +2 -0
- package/dist/cron.d.ts.map +1 -0
- package/dist/cron.js +9 -0
- package/dist/cron.js.map +1 -0
- package/dist/envelope.d.ts.map +1 -1
- package/dist/envelope.js +7 -6
- package/dist/envelope.js.map +1 -1
- package/dist/formatters/code-protect.d.ts +2 -0
- package/dist/formatters/code-protect.d.ts.map +1 -0
- package/dist/formatters/code-protect.js +12 -0
- package/dist/formatters/code-protect.js.map +1 -0
- package/dist/formatters/markdown.d.ts +1 -1
- package/dist/formatters/markdown.d.ts.map +1 -1
- package/dist/formatters/markdown.js +79 -21
- package/dist/formatters/markdown.js.map +1 -1
- package/dist/formatters/whatsapp-markdown.d.ts.map +1 -1
- package/dist/formatters/whatsapp-markdown.js +7 -21
- package/dist/formatters/whatsapp-markdown.js.map +1 -1
- package/dist/gateway.d.ts +32 -3
- package/dist/gateway.d.ts.map +1 -1
- package/dist/gateway.js +275 -174
- package/dist/gateway.js.map +1 -1
- package/dist/heartbeat.d.ts +9 -2
- package/dist/heartbeat.d.ts.map +1 -1
- package/dist/heartbeat.js +121 -84
- package/dist/heartbeat.js.map +1 -1
- package/dist/inbound-debounce.js +2 -2
- package/dist/inbound-debounce.js.map +1 -1
- package/dist/index.d.ts +6 -3
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -1
- package/dist/index.js.map +1 -1
- package/dist/media/audio-format.d.ts +3 -0
- package/dist/media/audio-format.d.ts.map +1 -0
- package/dist/media/audio-format.js +20 -0
- package/dist/media/audio-format.js.map +1 -0
- package/dist/media/deepgram-stt.d.ts +1 -1
- package/dist/media/deepgram-stt.d.ts.map +1 -1
- package/dist/media/deepgram-stt.js +4 -3
- package/dist/media/deepgram-stt.js.map +1 -1
- package/dist/media/media-processor.d.ts +12 -1
- package/dist/media/media-processor.d.ts.map +1 -1
- package/dist/media/media-processor.js +75 -18
- package/dist/media/media-processor.js.map +1 -1
- package/dist/media/whisper-api.d.ts.map +1 -1
- package/dist/media/whisper-api.js +20 -32
- package/dist/media/whisper-api.js.map +1 -1
- package/dist/media/whisper-local.d.ts +1 -0
- package/dist/media/whisper-local.d.ts.map +1 -1
- package/dist/media/whisper-local.js +100 -45
- package/dist/media/whisper-local.js.map +1 -1
- package/dist/message-queue.d.ts +1 -0
- package/dist/message-queue.d.ts.map +1 -1
- package/dist/message-queue.js +17 -10
- package/dist/message-queue.js.map +1 -1
- package/dist/middleware/auto-extract.d.ts +2 -0
- package/dist/middleware/auto-extract.d.ts.map +1 -1
- package/dist/middleware/auto-extract.js +94 -30
- package/dist/middleware/auto-extract.js.map +1 -1
- package/dist/middleware/dm-policy.d.ts +2 -1
- package/dist/middleware/dm-policy.d.ts.map +1 -1
- package/dist/middleware/dm-policy.js +26 -27
- package/dist/middleware/dm-policy.js.map +1 -1
- package/dist/middleware/owner-commands.d.ts +1 -1
- package/dist/middleware/owner-commands.d.ts.map +1 -1
- package/dist/middleware/owner-commands.js +8 -7
- package/dist/middleware/owner-commands.js.map +1 -1
- package/dist/middleware/pairing-codes.d.ts +12 -0
- package/dist/middleware/pairing-codes.d.ts.map +1 -0
- package/dist/middleware/pairing-codes.js +26 -0
- package/dist/middleware/pairing-codes.js.map +1 -0
- package/dist/middleware/pairing.d.ts +3 -3
- package/dist/middleware/pairing.d.ts.map +1 -1
- package/dist/middleware/pairing.js +18 -25
- package/dist/middleware/pairing.js.map +1 -1
- package/dist/middleware/rate-limit.d.ts +3 -1
- package/dist/middleware/rate-limit.d.ts.map +1 -1
- package/dist/middleware/rate-limit.js +21 -6
- package/dist/middleware/rate-limit.js.map +1 -1
- package/dist/runtime-builder.d.ts +11 -0
- package/dist/runtime-builder.d.ts.map +1 -1
- package/dist/runtime-builder.js +212 -62
- package/dist/runtime-builder.js.map +1 -1
- package/dist/simple-timer-store.d.ts +6 -0
- package/dist/simple-timer-store.d.ts.map +1 -1
- package/dist/simple-timer-store.js +8 -1
- package/dist/simple-timer-store.js.map +1 -1
- package/dist/stream-buffer.d.ts +14 -7
- package/dist/stream-buffer.d.ts.map +1 -1
- package/dist/stream-buffer.js +129 -84
- package/dist/stream-buffer.js.map +1 -1
- package/dist/tools/path-guard.d.ts +4 -0
- package/dist/tools/path-guard.d.ts.map +1 -0
- package/dist/tools/path-guard.js +53 -0
- package/dist/tools/path-guard.js.map +1 -0
- package/dist/tools/self-config.d.ts +12 -2
- package/dist/tools/self-config.d.ts.map +1 -1
- package/dist/tools/self-config.js +53 -16
- package/dist/tools/self-config.js.map +1 -1
- package/package.json +19 -12
|
@@ -24,12 +24,14 @@ export interface AutoExtractConfig {
|
|
|
24
24
|
set(key: string, value: string): Promise<void>;
|
|
25
25
|
};
|
|
26
26
|
coreFactPatterns?: Record<string, RegExp>;
|
|
27
|
+
onError?: (error: Error) => void;
|
|
27
28
|
}
|
|
28
29
|
export declare class AutoExtractMiddleware implements GatewayMiddleware {
|
|
29
30
|
private config;
|
|
30
31
|
readonly name = "auto-extract";
|
|
31
32
|
constructor(config: AutoExtractConfig);
|
|
32
33
|
handle(msg: ChannelMessage, _ctx: MiddlewareContext, next: () => Promise<void>): Promise<void>;
|
|
34
|
+
private resolveNodeId;
|
|
33
35
|
private extractAndStore;
|
|
34
36
|
}
|
|
35
37
|
export declare function autoExtract(config: AutoExtractConfig): GatewayMiddleware;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"auto-extract.d.ts","sourceRoot":"","sources":["../../src/middleware/auto-extract.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,cAAc,EACd,iBAAiB,EACjB,YAAY,EAGb,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC,CAAC;IACH,SAAS,EAAE,KAAK,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;CACrE;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,eAAe,CAAC;IAC3B,YAAY,EAAE,YAAY,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE;QAAE,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;KAAE,CAAC;IAC/D,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"auto-extract.d.ts","sourceRoot":"","sources":["../../src/middleware/auto-extract.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,cAAc,EACd,iBAAiB,EACjB,YAAY,EAGb,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC,CAAC;IACH,SAAS,EAAE,KAAK,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;CACrE;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,eAAe,CAAC;IAC3B,YAAY,EAAE,YAAY,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE;QAAE,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;KAAE,CAAC;IAC/D,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAoCD,qBAAa,qBAAsB,YAAW,iBAAiB;IAGjD,OAAO,CAAC,MAAM;IAF1B,QAAQ,CAAC,IAAI,kBAAkB;gBAEX,MAAM,EAAE,iBAAiB;IAEvC,MAAM,CACV,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE,iBAAiB,EACvB,IAAI,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GACxB,OAAO,CAAC,IAAI,CAAC;YAYF,aAAa;YASb,eAAe;CAiF9B;AAED,wBAAgB,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,iBAAiB,CAExE"}
|
|
@@ -1,3 +1,33 @@
|
|
|
1
|
+
const ENTITY_TYPES = new Set([
|
|
2
|
+
'person',
|
|
3
|
+
'organization',
|
|
4
|
+
'location',
|
|
5
|
+
'concept',
|
|
6
|
+
'event',
|
|
7
|
+
'object',
|
|
8
|
+
'custom',
|
|
9
|
+
]);
|
|
10
|
+
const RELATION_TYPES = new Set([
|
|
11
|
+
'knows',
|
|
12
|
+
'works_at',
|
|
13
|
+
'located_in',
|
|
14
|
+
'part_of',
|
|
15
|
+
'related_to',
|
|
16
|
+
'created_by',
|
|
17
|
+
'belongs_to',
|
|
18
|
+
'associated_with',
|
|
19
|
+
'causes',
|
|
20
|
+
'precedes',
|
|
21
|
+
'custom',
|
|
22
|
+
]);
|
|
23
|
+
function toEntityType(type) {
|
|
24
|
+
const normalized = type.toLowerCase();
|
|
25
|
+
return ENTITY_TYPES.has(normalized) ? normalized : 'custom';
|
|
26
|
+
}
|
|
27
|
+
function toRelationType(type) {
|
|
28
|
+
const normalized = type.toLowerCase();
|
|
29
|
+
return RELATION_TYPES.has(normalized) ? normalized : 'custom';
|
|
30
|
+
}
|
|
1
31
|
export class AutoExtractMiddleware {
|
|
2
32
|
config;
|
|
3
33
|
name = 'auto-extract';
|
|
@@ -6,59 +36,93 @@ export class AutoExtractMiddleware {
|
|
|
6
36
|
}
|
|
7
37
|
async handle(msg, _ctx, next) {
|
|
8
38
|
await next();
|
|
9
|
-
if (msg.text) {
|
|
39
|
+
if (msg.text?.trim()) {
|
|
10
40
|
void this.extractAndStore(msg.text).catch((err) => {
|
|
11
|
-
|
|
41
|
+
const error = err instanceof Error ? err : new Error(String(err));
|
|
42
|
+
if (this.config.onError)
|
|
43
|
+
this.config.onError(error);
|
|
44
|
+
else
|
|
45
|
+
console.error('[auto-extract] Error:', error.message);
|
|
12
46
|
});
|
|
13
47
|
}
|
|
14
48
|
}
|
|
49
|
+
async resolveNodeId(name, known) {
|
|
50
|
+
const cached = known.get(name.toLowerCase());
|
|
51
|
+
if (cached)
|
|
52
|
+
return cached;
|
|
53
|
+
const found = await this.config.graphAdapter.getNodeByName(this.config.agentId, name);
|
|
54
|
+
if (!found.success || !found.data)
|
|
55
|
+
return null;
|
|
56
|
+
known.set(name.toLowerCase(), found.data.id);
|
|
57
|
+
return found.data.id;
|
|
58
|
+
}
|
|
15
59
|
async extractAndStore(text) {
|
|
16
60
|
const { extractor, graphAdapter, agentId } = this.config;
|
|
17
61
|
const result = await extractor.extract(text);
|
|
62
|
+
const known = new Map();
|
|
18
63
|
for (const entity of result.entities) {
|
|
19
64
|
const existing = await graphAdapter.getNodeByName(agentId, entity.name);
|
|
20
65
|
if (existing.success && existing.data) {
|
|
66
|
+
known.set(entity.name.toLowerCase(), existing.data.id);
|
|
21
67
|
if (entity.confidence > existing.data.confidence) {
|
|
22
68
|
await graphAdapter.updateNode(existing.data.id, {
|
|
23
69
|
confidence: entity.confidence,
|
|
24
|
-
description: entity.description,
|
|
70
|
+
...(entity.description ? { description: entity.description } : {}),
|
|
25
71
|
});
|
|
26
72
|
}
|
|
73
|
+
continue;
|
|
27
74
|
}
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
}
|
|
75
|
+
const rawType = entity.type;
|
|
76
|
+
const type = toEntityType(rawType);
|
|
77
|
+
const created = await graphAdapter.addNode({
|
|
78
|
+
agentId,
|
|
79
|
+
name: entity.name,
|
|
80
|
+
type,
|
|
81
|
+
description: entity.description,
|
|
82
|
+
confidence: entity.confidence,
|
|
83
|
+
source: 'extracted',
|
|
84
|
+
aliases: [],
|
|
85
|
+
properties: type === 'custom' && rawType ? { originalType: rawType } : {},
|
|
86
|
+
metadata: {},
|
|
87
|
+
});
|
|
88
|
+
if (created.success && created.data) {
|
|
89
|
+
known.set(entity.name.toLowerCase(), created.data.id);
|
|
40
90
|
}
|
|
41
91
|
}
|
|
42
92
|
for (const relation of result.relations) {
|
|
43
|
-
const
|
|
44
|
-
const
|
|
45
|
-
if (
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
93
|
+
const sourceNodeId = await this.resolveNodeId(relation.from, known);
|
|
94
|
+
const targetNodeId = await this.resolveNodeId(relation.to, known);
|
|
95
|
+
if (!sourceNodeId || !targetNodeId)
|
|
96
|
+
continue;
|
|
97
|
+
const type = toRelationType(relation.type);
|
|
98
|
+
const label = type === 'custom' ? relation.type : undefined;
|
|
99
|
+
const between = await graphAdapter.getEdgesBetween(sourceNodeId, targetNodeId);
|
|
100
|
+
const duplicate = between.success
|
|
101
|
+
? between.data.find((edge) => edge.type === type && edge.label === label)
|
|
102
|
+
: undefined;
|
|
103
|
+
if (duplicate) {
|
|
104
|
+
if (relation.confidence > duplicate.confidence) {
|
|
105
|
+
await graphAdapter.updateEdge(duplicate.id, { confidence: relation.confidence });
|
|
106
|
+
}
|
|
107
|
+
continue;
|
|
58
108
|
}
|
|
109
|
+
await graphAdapter.addEdge({
|
|
110
|
+
agentId,
|
|
111
|
+
sourceNodeId,
|
|
112
|
+
targetNodeId,
|
|
113
|
+
type,
|
|
114
|
+
...(label ? { label } : {}),
|
|
115
|
+
confidence: relation.confidence,
|
|
116
|
+
source: 'extracted',
|
|
117
|
+
weight: 1.0,
|
|
118
|
+
bidirectional: false,
|
|
119
|
+
properties: {},
|
|
120
|
+
metadata: {},
|
|
121
|
+
});
|
|
59
122
|
}
|
|
60
123
|
if (this.config.coreFacts && this.config.coreFactPatterns) {
|
|
61
124
|
for (const [key, pattern] of Object.entries(this.config.coreFactPatterns)) {
|
|
125
|
+
pattern.lastIndex = 0;
|
|
62
126
|
const match = pattern.exec(text);
|
|
63
127
|
if (match?.[1]) {
|
|
64
128
|
await this.config.coreFacts.set(key, match[1].trim());
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"auto-extract.js","sourceRoot":"","sources":["../../src/middleware/auto-extract.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"auto-extract.js","sourceRoot":"","sources":["../../src/middleware/auto-extract.ts"],"names":[],"mappings":"AAqCA,MAAM,YAAY,GAAwB,IAAI,GAAG,CAAa;IAC5D,QAAQ;IACR,cAAc;IACd,UAAU;IACV,SAAS;IACT,OAAO;IACP,QAAQ;IACR,QAAQ;CACT,CAAC,CAAC;AAEH,MAAM,cAAc,GAAwB,IAAI,GAAG,CAAe;IAChE,OAAO;IACP,UAAU;IACV,YAAY;IACZ,SAAS;IACT,YAAY;IACZ,YAAY;IACZ,YAAY;IACZ,iBAAiB;IACjB,QAAQ;IACR,UAAU;IACV,QAAQ;CACT,CAAC,CAAC;AAEH,SAAS,YAAY,CAAC,IAAY;IAChC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACtC,OAAO,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAE,UAAyB,CAAC,CAAC,CAAC,QAAQ,CAAC;AAC9E,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACtC,OAAO,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAE,UAA2B,CAAC,CAAC,CAAC,QAAQ,CAAC;AAClF,CAAC;AAED,MAAM,OAAO,qBAAqB;IAGZ;IAFX,IAAI,GAAG,cAAc,CAAC;IAE/B,YAAoB,MAAyB;QAAzB,WAAM,GAAN,MAAM,CAAmB;IAAG,CAAC;IAEjD,KAAK,CAAC,MAAM,CACV,GAAmB,EACnB,IAAuB,EACvB,IAAyB;QAEzB,MAAM,IAAI,EAAE,CAAC;QAEb,IAAI,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC;YACrB,KAAK,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;gBACzD,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClE,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO;oBAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;;oBAC/C,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAC7D,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,IAAY,EAAE,KAA0B;QAClE,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAC7C,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QAC1B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACtF,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAC/C,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7C,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,IAAY;QACxC,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QACzD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;QAExC,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACxE,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACtC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACvD,IAAI,MAAM,CAAC,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;oBACjD,MAAM,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE;wBAC9C,UAAU,EAAE,MAAM,CAAC,UAAU;wBAC7B,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBACnE,CAAC,CAAC;gBACL,CAAC;gBACD,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;YAC5B,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;YACnC,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC;gBACzC,OAAO;gBACP,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI;gBACJ,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,MAAM,EAAE,WAAW;gBACnB,OAAO,EAAE,EAAE;gBACX,UAAU,EAAE,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE;gBACzE,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;gBACpC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACxC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACpE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YAClE,IAAI,CAAC,YAAY,IAAI,CAAC,YAAY;gBAAE,SAAS;YAE7C,MAAM,IAAI,GAAG,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,KAAK,GAAG,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAE5D,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,eAAe,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;YAC/E,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO;gBAC/B,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC;gBACzE,CAAC,CAAC,SAAS,CAAC;YAEd,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,QAAQ,CAAC,UAAU,GAAG,SAAS,CAAC,UAAU,EAAE,CAAC;oBAC/C,MAAM,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;gBACnF,CAAC;gBACD,SAAS;YACX,CAAC;YAED,MAAM,YAAY,CAAC,OAAO,CAAC;gBACzB,OAAO;gBACP,YAAY;gBACZ,YAAY;gBACZ,IAAI;gBACJ,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,MAAM,EAAE,WAAW;gBACnB,MAAM,EAAE,GAAG;gBACX,aAAa,EAAE,KAAK;gBACpB,UAAU,EAAE,EAAE;gBACd,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC1D,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC1E,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC;gBACtB,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjC,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACf,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBACxD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,WAAW,CAAC,MAAyB;IACnD,OAAO,IAAI,qBAAqB,CAAC,MAAM,CAAC,CAAC;AAC3C,CAAC"}
|
|
@@ -10,6 +10,7 @@ export interface DmPolicyConfig {
|
|
|
10
10
|
groupAllowlist?: string[];
|
|
11
11
|
ownerIds?: Record<string, string>;
|
|
12
12
|
onPairingRequest?: (userId: string, code: string) => void;
|
|
13
|
+
onStoreError?: (error: Error) => void;
|
|
13
14
|
}
|
|
14
15
|
export declare class DmPolicyMiddleware implements GatewayMiddleware {
|
|
15
16
|
readonly name = "dm-policy";
|
|
@@ -23,6 +24,7 @@ export declare class DmPolicyMiddleware implements GatewayMiddleware {
|
|
|
23
24
|
private readonly groupAllowlist;
|
|
24
25
|
private readonly ownerKeys;
|
|
25
26
|
private readonly onPairingRequest?;
|
|
27
|
+
private readonly onStoreError?;
|
|
26
28
|
constructor(config: DmPolicyConfig);
|
|
27
29
|
handle(msg: ChannelMessage, ctx: MiddlewareContext, next: () => Promise<void>): Promise<void>;
|
|
28
30
|
isApproved(channelType: string, userId: string): boolean;
|
|
@@ -31,7 +33,6 @@ export declare class DmPolicyMiddleware implements GatewayMiddleware {
|
|
|
31
33
|
private checkGroupAccess;
|
|
32
34
|
private handlePairingRequest;
|
|
33
35
|
private handleApproval;
|
|
34
|
-
private findPendingByUser;
|
|
35
36
|
private loadStore;
|
|
36
37
|
private saveStore;
|
|
37
38
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dm-policy.d.ts","sourceRoot":"","sources":["../../src/middleware/dm-policy.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"dm-policy.d.ts","sourceRoot":"","sources":["../../src/middleware/dm-policy.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAUhG,MAAM,MAAM,YAAY,GAAG,MAAM,GAAG,WAAW,GAAG,SAAS,GAAG,UAAU,CAAC;AAEzE,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,YAAY,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,GAAG,WAAW,GAAG,UAAU,CAAC;IAChD,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAClC,gBAAgB,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;IAC1D,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACvC;AAWD,qBAAa,kBAAmB,YAAW,iBAAiB;IAC1D,QAAQ,CAAC,IAAI,eAAe;IAC5B,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAe;IACpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqC;IAC7D,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAoC;IAChE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAc;IAC7C,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAqB;IAC/C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAyC;IAC3E,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAyB;gBAE3C,MAAM,EAAE,cAAc;IA0B5B,MAAM,CACV,GAAG,EAAE,cAAc,EACnB,GAAG,EAAE,iBAAiB,EACtB,IAAI,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GACxB,OAAO,CAAC,IAAI,CAAC;IA8ChB,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO;IAKxD,gBAAgB,IAAI,SAAS,MAAM,EAAE;IAIrC,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,gBAAgB;YAWV,oBAAoB;YAiCpB,cAAc;IAkB5B,OAAO,CAAC,SAAS;IAajB,OAAO,CAAC,SAAS;CAmBlB;AAED,wBAAgB,QAAQ,CAAC,MAAM,EAAE,cAAc,GAAG,iBAAiB,CAElE"}
|
|
@@ -1,8 +1,7 @@
|
|
|
1
|
-
import { readFileSync, writeFileSync, mkdirSync, existsSync } from 'node:fs';
|
|
2
|
-
import { dirname } from 'node:path';
|
|
3
|
-
import { join } from 'node:path';
|
|
1
|
+
import { readFileSync, writeFileSync, mkdirSync, existsSync, renameSync } from 'node:fs';
|
|
2
|
+
import { dirname, join } from 'node:path';
|
|
4
3
|
import { homedir } from 'node:os';
|
|
5
|
-
import {
|
|
4
|
+
import { findPendingByUser, generatePairingCode, isPairCommand, parsePairCommand, prunePending, } from './pairing-codes.js';
|
|
6
5
|
const DEFAULT_STORE_PATH = join(homedir(), '.cogitator', 'dm-allowlist.json');
|
|
7
6
|
const DEFAULT_PAIRING_CODE_LENGTH = 6;
|
|
8
7
|
const DEFAULT_PAIRING_EXPIRES_MS = 300_000;
|
|
@@ -18,6 +17,7 @@ export class DmPolicyMiddleware {
|
|
|
18
17
|
groupAllowlist;
|
|
19
18
|
ownerKeys = new Set();
|
|
20
19
|
onPairingRequest;
|
|
20
|
+
onStoreError;
|
|
21
21
|
constructor(config) {
|
|
22
22
|
this.mode = config.mode;
|
|
23
23
|
this.storePath = config.storePath ?? DEFAULT_STORE_PATH;
|
|
@@ -26,6 +26,7 @@ export class DmPolicyMiddleware {
|
|
|
26
26
|
this.groupPolicy = config.groupPolicy ?? 'open';
|
|
27
27
|
this.groupAllowlist = new Set(config.groupAllowlist ?? []);
|
|
28
28
|
this.onPairingRequest = config.onPairingRequest;
|
|
29
|
+
this.onStoreError = config.onStoreError;
|
|
29
30
|
if (config.ownerIds) {
|
|
30
31
|
for (const [channelType, ownerId] of Object.entries(config.ownerIds)) {
|
|
31
32
|
this.ownerKeys.add(`${channelType}:${ownerId}`);
|
|
@@ -50,7 +51,7 @@ export class DmPolicyMiddleware {
|
|
|
50
51
|
return;
|
|
51
52
|
}
|
|
52
53
|
if (isOwner) {
|
|
53
|
-
if (this.mode === 'pairing' && msg.text
|
|
54
|
+
if (this.mode === 'pairing' && isPairCommand(msg.text)) {
|
|
54
55
|
await this.handleApproval(msg, ctx);
|
|
55
56
|
return;
|
|
56
57
|
}
|
|
@@ -101,47 +102,37 @@ export class DmPolicyMiddleware {
|
|
|
101
102
|
}
|
|
102
103
|
}
|
|
103
104
|
async handlePairingRequest(msg, ctx, userKey) {
|
|
104
|
-
const
|
|
105
|
-
|
|
105
|
+
const now = Date.now();
|
|
106
|
+
prunePending(this.pending, now);
|
|
107
|
+
const existing = findPendingByUser(this.pending, userKey);
|
|
108
|
+
if (existing) {
|
|
106
109
|
await ctx.channel.sendText(msg.channelId, `Waiting for approval. Your code: \`${existing.code}\``);
|
|
107
110
|
return;
|
|
108
111
|
}
|
|
109
|
-
const
|
|
110
|
-
const code = genCode(this.codeLength);
|
|
112
|
+
const code = generatePairingCode(this.codeLength);
|
|
111
113
|
this.pending.set(code, {
|
|
112
114
|
code,
|
|
113
115
|
userId: msg.userId,
|
|
114
116
|
channelType: msg.channelType,
|
|
115
|
-
expiresAt:
|
|
117
|
+
expiresAt: now + this.expiresMs,
|
|
116
118
|
});
|
|
117
119
|
this.onPairingRequest?.(msg.userId, code);
|
|
118
120
|
await ctx.channel.sendText(msg.channelId, `Hi! Ask the bot owner to approve you with: \`/pair ${code}\``);
|
|
119
121
|
}
|
|
120
122
|
async handleApproval(msg, ctx) {
|
|
121
|
-
|
|
122
|
-
const
|
|
123
|
-
|
|
123
|
+
prunePending(this.pending, Date.now());
|
|
124
|
+
const code = parsePairCommand(msg.text);
|
|
125
|
+
const pairing = code ? this.pending.get(code) : undefined;
|
|
126
|
+
if (!code || !pairing) {
|
|
124
127
|
await ctx.channel.sendText(msg.channelId, 'Invalid or expired pairing code.');
|
|
125
128
|
return;
|
|
126
129
|
}
|
|
127
|
-
if (pairing.expiresAt < Date.now()) {
|
|
128
|
-
this.pending.delete(code);
|
|
129
|
-
await ctx.channel.sendText(msg.channelId, 'Pairing code expired.');
|
|
130
|
-
return;
|
|
131
|
-
}
|
|
132
130
|
const userKey = `${pairing.channelType}:${pairing.userId}`;
|
|
133
131
|
this.approved.add(userKey);
|
|
134
132
|
this.pending.delete(code);
|
|
135
133
|
this.saveStore();
|
|
136
134
|
await ctx.channel.sendText(msg.channelId, `User approved (${userKey}).`);
|
|
137
135
|
}
|
|
138
|
-
findPendingByUser(userKey) {
|
|
139
|
-
for (const p of this.pending.values()) {
|
|
140
|
-
if (`${p.channelType}:${p.userId}` === userKey)
|
|
141
|
-
return p;
|
|
142
|
-
}
|
|
143
|
-
return undefined;
|
|
144
|
-
}
|
|
145
136
|
loadStore() {
|
|
146
137
|
try {
|
|
147
138
|
if (!existsSync(this.storePath))
|
|
@@ -166,9 +157,17 @@ export class DmPolicyMiddleware {
|
|
|
166
157
|
if (!existsSync(dir)) {
|
|
167
158
|
mkdirSync(dir, { recursive: true });
|
|
168
159
|
}
|
|
169
|
-
|
|
160
|
+
const tmpPath = `${this.storePath}.${process.pid}.tmp`;
|
|
161
|
+
writeFileSync(tmpPath, JSON.stringify(store, null, 2));
|
|
162
|
+
renameSync(tmpPath, this.storePath);
|
|
163
|
+
}
|
|
164
|
+
catch (err) {
|
|
165
|
+
const error = err instanceof Error ? err : new Error(String(err));
|
|
166
|
+
if (this.onStoreError)
|
|
167
|
+
this.onStoreError(error);
|
|
168
|
+
else
|
|
169
|
+
console.error('[dm-policy] Failed to persist allowlist:', error.message);
|
|
170
170
|
}
|
|
171
|
-
catch { }
|
|
172
171
|
}
|
|
173
172
|
}
|
|
174
173
|
export function dmPolicy(config) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"dm-policy.js","sourceRoot":"","sources":["../../src/middleware/dm-policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;
|
|
1
|
+
{"version":3,"file":"dm-policy.js","sourceRoot":"","sources":["../../src/middleware/dm-policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACzF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAElC,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,aAAa,EACb,gBAAgB,EAChB,YAAY,GAEb,MAAM,iBAAiB,CAAC;AAsBzB,MAAM,kBAAkB,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;AAC9E,MAAM,2BAA2B,GAAG,CAAC,CAAC;AACtC,MAAM,0BAA0B,GAAG,OAAO,CAAC;AAE3C,MAAM,OAAO,kBAAkB;IACpB,IAAI,GAAG,WAAW,CAAC;IACX,IAAI,CAAe;IACnB,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IAC7B,OAAO,GAAG,IAAI,GAAG,EAA0B,CAAC;IAC5C,SAAS,CAAS;IAClB,UAAU,CAAS;IACnB,SAAS,CAAS;IAClB,WAAW,CAAoC;IAC/C,cAAc,CAAc;IAC5B,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,gBAAgB,CAA0C;IAC1D,YAAY,CAA0B;IAEvD,YAAY,MAAsB;QAChC,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;QACxD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,iBAAiB,IAAI,2BAA2B,CAAC;QAC1E,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,gBAAgB,IAAI,0BAA0B,CAAC;QACvE,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC;QAChD,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;QAC3D,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QAExC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,KAAK,MAAM,CAAC,WAAW,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC,CAAC;gBAChD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,WAAW,IAAI,OAAO,EAAE,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,SAAS,EAAE,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAmB,EACnB,GAAsB,EACtB,IAAyB;QAEzB,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;QAC9B,MAAM,OAAO,GAAG,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QACnD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAE5C,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,OAAO,CAAC;gBAAE,OAAO;YACjD,MAAM,IAAI,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvD,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBACpC,OAAO;YACT,CAAC;YACD,MAAM,IAAI,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,MAAM;gBACT,MAAM,IAAI,EAAE,CAAC;gBACb,OAAO;YAET,KAAK,UAAU;gBACb,OAAO;YAET,KAAK,WAAW;gBACd,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;oBACxC,MAAM,IAAI,EAAE,CAAC;gBACf,CAAC;qBAAM,CAAC;oBACN,MAAM,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;gBAC/D,CAAC;gBACD,OAAO;YAET,KAAK,SAAS;gBACZ,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;oBACxC,MAAM,IAAI,EAAE,CAAC;oBACb,OAAO;gBACT,CAAC;gBACD,MAAM,IAAI,CAAC,oBAAoB,CAAC,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;gBACnD,OAAO;QACX,CAAC;IACH,CAAC;IAED,UAAU,CAAC,WAAmB,EAAE,MAAc;QAC5C,MAAM,OAAO,GAAG,GAAG,WAAW,IAAI,MAAM,EAAE,CAAC;QAC3C,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACzC,CAAC;IAED,gBAAgB;QACd,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5B,CAAC;IAEO,SAAS,CAAC,OAAe,EAAE,MAAc;QAC/C,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACjE,CAAC;IAEO,gBAAgB,CAAC,GAAmB,EAAE,OAAgB;QAC5D,QAAQ,IAAI,CAAC,WAAW,EAAE,CAAC;YACzB,KAAK,MAAM;gBACT,OAAO,IAAI,CAAC;YACd,KAAK,UAAU;gBACb,OAAO,KAAK,CAAC;YACf,KAAK,WAAW;gBACd,OAAO,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,OAAQ,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,GAAmB,EACnB,GAAsB,EACtB,OAAe;QAEf,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAEhC,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC1D,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,GAAG,CAAC,OAAO,CAAC,QAAQ,CACxB,GAAG,CAAC,SAAS,EACb,sCAAsC,QAAQ,CAAC,IAAI,IAAI,CACxD,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,mBAAmB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE;YACrB,IAAI;YACJ,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,SAAS,EAAE,GAAG,GAAG,IAAI,CAAC,SAAS;SAChC,CAAC,CAAC;QAEH,IAAI,CAAC,gBAAgB,EAAE,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE1C,MAAM,GAAG,CAAC,OAAO,CAAC,QAAQ,CACxB,GAAG,CAAC,SAAS,EACb,sDAAsD,IAAI,IAAI,CAC/D,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,GAAmB,EAAE,GAAsB;QACtE,YAAY,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE1D,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACtB,MAAM,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,kCAAkC,CAAC,CAAC;YAC9E,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAC3D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1B,IAAI,CAAC,SAAS,EAAE,CAAC;QAEjB,MAAM,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,kBAAkB,OAAO,IAAI,CAAC,CAAC;IAC3E,CAAC;IAEO,SAAS;QACf,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC;gBAAE,OAAO;YACxC,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YAClD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAe,CAAC;YAC3C,IAAI,IAAI,CAAC,OAAO,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACZ,CAAC;IAEO,SAAS;QACf,MAAM,KAAK,GAAe;YACxB,OAAO,EAAE,CAAC;YACV,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;SACpE,CAAC;QACF,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACtC,CAAC;YACD,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,MAAM,CAAC;YACvD,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACvD,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClE,IAAI,IAAI,CAAC,YAAY;gBAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;;gBAC3C,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;CACF;AAED,MAAM,UAAU,QAAQ,CAAC,MAAsB;IAC7C,OAAO,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;AACxC,CAAC"}
|
|
@@ -7,7 +7,7 @@ export interface OwnerCommandsConfig {
|
|
|
7
7
|
onStatus?: () => string;
|
|
8
8
|
onSessions?: () => string;
|
|
9
9
|
onUsers?: () => string;
|
|
10
|
-
onCompact?: (target: string) => Promise<string>;
|
|
10
|
+
onCompact?: (target: string, threadId: string) => Promise<string>;
|
|
11
11
|
onModel?: (model: string, forUser?: string) => string;
|
|
12
12
|
onRestart?: () => Promise<void>;
|
|
13
13
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"owner-commands.d.ts","sourceRoot":"","sources":["../../src/middleware/owner-commands.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAEhG,MAAM,MAAM,YAAY,GAAG,OAAO,GAAG,YAAY,GAAG,QAAQ,CAAC;AAE7D,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"owner-commands.d.ts","sourceRoot":"","sources":["../../src/middleware/owner-commands.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAEhG,MAAM,MAAM,YAAY,GAAG,OAAO,GAAG,YAAY,GAAG,QAAQ,CAAC;AAE7D,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAClE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC;IACtD,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACjC;AA4FD,qBAAa,uBAAwB,YAAW,iBAAiB;IAMnD,OAAO,CAAC,QAAQ,CAAC,MAAM;IALnC,QAAQ,CAAC,IAAI,oBAAoB;IACjC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAc;IACxC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAc;IAC7C,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAc;gBAEhB,MAAM,EAAE,mBAAmB;IAkBlD,MAAM,CACV,GAAG,EAAE,cAAc,EACnB,GAAG,EAAE,iBAAiB,EACtB,IAAI,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GACxB,OAAO,CAAC,IAAI,CAAC;IA+BhB,OAAO,CAAC,YAAY;CAQrB;AAED,wBAAgB,aAAa,CAAC,MAAM,EAAE,mBAAmB,GAAG,iBAAiB,CAE5E"}
|
|
@@ -19,9 +19,9 @@ const COMMANDS = {
|
|
|
19
19
|
},
|
|
20
20
|
'/compact': {
|
|
21
21
|
level: 'owner',
|
|
22
|
-
handler: async (args, _msg,
|
|
22
|
+
handler: async (args, _msg, ctx, config) => {
|
|
23
23
|
if (config.onCompact) {
|
|
24
|
-
return config.onCompact(args.trim() || 'current');
|
|
24
|
+
return config.onCompact(args.trim() || 'current', ctx.threadId);
|
|
25
25
|
}
|
|
26
26
|
return 'No compact handler configured';
|
|
27
27
|
},
|
|
@@ -92,14 +92,15 @@ export class OwnerCommandsMiddleware {
|
|
|
92
92
|
}
|
|
93
93
|
}
|
|
94
94
|
async handle(msg, ctx, next) {
|
|
95
|
-
|
|
95
|
+
const text = msg.text.trim();
|
|
96
|
+
if (!text.startsWith('/')) {
|
|
96
97
|
await next();
|
|
97
98
|
return;
|
|
98
99
|
}
|
|
99
|
-
const
|
|
100
|
-
const command =
|
|
101
|
-
const args =
|
|
102
|
-
const def = COMMANDS[command];
|
|
100
|
+
const match = /^(\/[^\s@]+)(?:@\S+)?(?:\s+([\s\S]*))?$/.exec(text);
|
|
101
|
+
const command = match ? match[1].toLowerCase() : text;
|
|
102
|
+
const args = match?.[2] ?? '';
|
|
103
|
+
const def = Object.hasOwn(COMMANDS, command) ? COMMANDS[command] : undefined;
|
|
103
104
|
if (!def) {
|
|
104
105
|
await next();
|
|
105
106
|
return;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"owner-commands.js","sourceRoot":"","sources":["../../src/middleware/owner-commands.ts"],"names":[],"mappings":"AA4BA,MAAM,QAAQ,GAA+B;IAC3C,SAAS,EAAE;QACT,KAAK,EAAE,YAAY;QACnB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,OAAO,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,8BAA8B,CAAC;QAC/D,CAAC;KACF;IAED,WAAW,EAAE;QACX,KAAK,EAAE,YAAY;QACnB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,OAAO,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,gCAAgC,CAAC;QACnE,CAAC;KACF;IAED,QAAQ,EAAE;QACR,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,OAAO,MAAM,CAAC,OAAO,EAAE,EAAE,IAAI,6BAA6B,CAAC;QAC7D,CAAC;KACF;IAED,UAAU,EAAE;QACV,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,
|
|
1
|
+
{"version":3,"file":"owner-commands.js","sourceRoot":"","sources":["../../src/middleware/owner-commands.ts"],"names":[],"mappings":"AA4BA,MAAM,QAAQ,GAA+B;IAC3C,SAAS,EAAE;QACT,KAAK,EAAE,YAAY;QACnB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,OAAO,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,8BAA8B,CAAC;QAC/D,CAAC;KACF;IAED,WAAW,EAAE;QACX,KAAK,EAAE,YAAY;QACnB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,OAAO,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,gCAAgC,CAAC;QACnE,CAAC;KACF;IAED,QAAQ,EAAE;QACR,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,OAAO,MAAM,CAAC,OAAO,EAAE,EAAE,IAAI,6BAA6B,CAAC;QAC7D,CAAC;KACF;IAED,UAAU,EAAE;QACV,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE;YACzC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,OAAO,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,SAAS,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC;YAClE,CAAC;YACD,OAAO,+BAA+B,CAAC;QACzC,CAAC;KACF;IAED,QAAQ,EAAE;QACR,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACvC,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACvB,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK;gBAAE,OAAO,8BAA8B,CAAC;YAClD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,iBAAiB,KAAK,EAAE,CAAC;QACtE,CAAC;KACF;IAED,UAAU,EAAE;QACV,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;YAC3C,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;gBACrB,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBACzB,OAAO,eAAe,CAAC;YACzB,CAAC;YACD,OAAO,+BAA+B,CAAC;QACzC,CAAC;KACF;IAED,OAAO,EAAE;QACP,KAAK,EAAE,YAAY;QACnB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,OAAO;gBACL,WAAW;gBACX,wCAAwC;gBACxC,sCAAsC;gBACtC,6CAA6C;gBAC7C,8CAA8C;gBAC9C,8BAA8B;gBAC9B,mCAAmC;gBACnC,mCAAmC;gBACnC,mCAAmC;aACpC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC;KACF;CACF,CAAC;AAEF,MAAM,eAAe,GAAiC;IACpD,MAAM,EAAE,CAAC;IACT,UAAU,EAAE,CAAC;IACb,KAAK,EAAE,CAAC;CACT,CAAC;AAEF,MAAM,OAAO,uBAAuB;IAML;IALpB,IAAI,GAAG,gBAAgB,CAAC;IAChB,SAAS,CAAc;IACvB,cAAc,CAAc;IAC5B,cAAc,CAAc;IAE7C,YAA6B,MAA2B;QAA3B,WAAM,GAAN,MAAM,CAAqB;QACtD,IAAI,CAAC,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAE/F,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;QACxC,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAC7B,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;gBAC1C,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;QACxC,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YAC1B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBACxC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAmB,EACnB,GAAsB,EACtB,IAAyB;QAEzB,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,yCAAyC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACtD,MAAM,IAAI,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE9B,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7E,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QACzC,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC;QAE9E,IAAI,eAAe,CAAC,SAAS,CAAC,GAAG,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAChE,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,GAAmB;QACtC,MAAM,OAAO,GAAG,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QACnD,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,CAAC;QAChD,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5E,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAED,MAAM,UAAU,aAAa,CAAC,MAA2B;IACvD,OAAO,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC;AAC7C,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
export interface PendingPairing {
|
|
2
|
+
code: string;
|
|
3
|
+
userId: string;
|
|
4
|
+
channelType: string;
|
|
5
|
+
expiresAt: number;
|
|
6
|
+
}
|
|
7
|
+
export declare function generatePairingCode(length: number): string;
|
|
8
|
+
export declare function parsePairCommand(text: string): string | null;
|
|
9
|
+
export declare function isPairCommand(text: string): boolean;
|
|
10
|
+
export declare function prunePending(pending: Map<string, PendingPairing>, now: number): void;
|
|
11
|
+
export declare function findPendingByUser(pending: Map<string, PendingPairing>, userKey: string): PendingPairing | undefined;
|
|
12
|
+
//# sourceMappingURL=pairing-codes.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pairing-codes.d.ts","sourceRoot":"","sources":["../../src/middleware/pairing-codes.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAE1D;AAED,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAG5D;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAEnD;AAED,wBAAgB,YAAY,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,EAAE,GAAG,EAAE,MAAM,GAAG,IAAI,CAIpF;AAED,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,EACpC,OAAO,EAAE,MAAM,GACd,cAAc,GAAG,SAAS,CAK5B"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import { customAlphabet } from 'nanoid';
|
|
2
|
+
const PAIRING_ALPHABET = 'ABCDEFGHJKLMNPQRSTUVWXYZ23456789';
|
|
3
|
+
export function generatePairingCode(length) {
|
|
4
|
+
return customAlphabet(PAIRING_ALPHABET, length)();
|
|
5
|
+
}
|
|
6
|
+
export function parsePairCommand(text) {
|
|
7
|
+
const match = /^\/pair(?:@\S+)?\s+(\S+)\s*$/i.exec(text.trim());
|
|
8
|
+
return match ? match[1].toUpperCase() : null;
|
|
9
|
+
}
|
|
10
|
+
export function isPairCommand(text) {
|
|
11
|
+
return /^\/pair(?:@\S+)?(?:\s|$)/i.test(text.trim());
|
|
12
|
+
}
|
|
13
|
+
export function prunePending(pending, now) {
|
|
14
|
+
for (const [code, entry] of pending) {
|
|
15
|
+
if (entry.expiresAt <= now)
|
|
16
|
+
pending.delete(code);
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
export function findPendingByUser(pending, userKey) {
|
|
20
|
+
for (const p of pending.values()) {
|
|
21
|
+
if (`${p.channelType}:${p.userId}` === userKey)
|
|
22
|
+
return p;
|
|
23
|
+
}
|
|
24
|
+
return undefined;
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=pairing-codes.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pairing-codes.js","sourceRoot":"","sources":["../../src/middleware/pairing-codes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,QAAQ,CAAC;AAExC,MAAM,gBAAgB,GAAG,kCAAkC,CAAC;AAS5D,MAAM,UAAU,mBAAmB,CAAC,MAAc;IAChD,OAAO,cAAc,CAAC,gBAAgB,EAAE,MAAM,CAAC,EAAE,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC3C,MAAM,KAAK,GAAG,+BAA+B,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAChE,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAY;IACxC,OAAO,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,OAAoC,EAAE,GAAW;IAC5E,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,OAAO,EAAE,CAAC;QACpC,IAAI,KAAK,CAAC,SAAS,IAAI,GAAG;YAAE,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,OAAoC,EACpC,OAAe;IAEf,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;QACjC,IAAI,GAAG,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,MAAM,EAAE,KAAK,OAAO;YAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}
|
|
@@ -6,14 +6,14 @@ export interface PairingConfig {
|
|
|
6
6
|
}
|
|
7
7
|
export declare class PairingMiddleware implements GatewayMiddleware {
|
|
8
8
|
readonly name = "pairing";
|
|
9
|
-
private
|
|
10
|
-
private
|
|
9
|
+
private readonly ownerKeys;
|
|
10
|
+
private readonly approved;
|
|
11
|
+
private readonly pending;
|
|
11
12
|
private readonly codeLength;
|
|
12
13
|
private readonly expiresIn;
|
|
13
14
|
constructor(config: PairingConfig);
|
|
14
15
|
handle(msg: ChannelMessage, ctx: MiddlewareContext, next: () => Promise<void>): Promise<void>;
|
|
15
16
|
private handleApproval;
|
|
16
|
-
private findPendingByUser;
|
|
17
17
|
isApproved(channelType: string, userId: string): boolean;
|
|
18
18
|
}
|
|
19
19
|
export declare function pairing(config: PairingConfig): GatewayMiddleware;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"pairing.d.ts","sourceRoot":"","sources":["../../src/middleware/pairing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"pairing.d.ts","sourceRoot":"","sources":["../../src/middleware/pairing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAUhG,MAAM,WAAW,aAAa;IAC5B,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,iBAAkB,YAAW,iBAAiB;IACzD,QAAQ,CAAC,IAAI,aAAa;IAC1B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAqB;IAC/C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqC;IAC7D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEvB,MAAM,EAAE,aAAa;IAW3B,MAAM,CACV,GAAG,EAAE,cAAc,EACnB,GAAG,EAAE,iBAAiB,EACtB,IAAI,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GACxB,OAAO,CAAC,IAAI,CAAC;YAsCF,cAAc;IAgB5B,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO;CAGzD;AAED,wBAAgB,OAAO,CAAC,MAAM,EAAE,aAAa,GAAG,iBAAiB,CAEhE"}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { findPendingByUser, generatePairingCode, isPairCommand, parsePairCommand, prunePending, } from './pairing-codes.js';
|
|
2
2
|
export class PairingMiddleware {
|
|
3
3
|
name = 'pairing';
|
|
4
|
+
ownerKeys = new Set();
|
|
4
5
|
approved = new Set();
|
|
5
6
|
pending = new Map();
|
|
6
7
|
codeLength;
|
|
@@ -9,57 +10,49 @@ export class PairingMiddleware {
|
|
|
9
10
|
this.codeLength = config.codeLength ?? 6;
|
|
10
11
|
this.expiresIn = (config.expiresIn ?? 300) * 1000;
|
|
11
12
|
for (const [channelType, ownerId] of Object.entries(config.ownerIds)) {
|
|
12
|
-
|
|
13
|
+
const key = `${channelType}:${ownerId}`;
|
|
14
|
+
this.ownerKeys.add(key);
|
|
15
|
+
this.approved.add(key);
|
|
13
16
|
}
|
|
14
17
|
}
|
|
15
18
|
async handle(msg, ctx, next) {
|
|
16
19
|
const userKey = `${msg.channelType}:${msg.userId}`;
|
|
20
|
+
const now = Date.now();
|
|
21
|
+
prunePending(this.pending, now);
|
|
22
|
+
if (this.ownerKeys.has(userKey) && isPairCommand(msg.text)) {
|
|
23
|
+
await this.handleApproval(msg, ctx);
|
|
24
|
+
return;
|
|
25
|
+
}
|
|
17
26
|
if (this.approved.has(userKey)) {
|
|
18
|
-
if (msg.text.startsWith('/pair ')) {
|
|
19
|
-
await this.handleApproval(msg, ctx);
|
|
20
|
-
return;
|
|
21
|
-
}
|
|
22
27
|
await next();
|
|
23
28
|
return;
|
|
24
29
|
}
|
|
25
|
-
const existing = this.
|
|
26
|
-
if (existing
|
|
30
|
+
const existing = findPendingByUser(this.pending, userKey);
|
|
31
|
+
if (existing) {
|
|
27
32
|
await ctx.channel.sendText(msg.channelId, `Waiting for approval. Your code: \`${existing.code}\``);
|
|
28
33
|
return;
|
|
29
34
|
}
|
|
30
|
-
const code =
|
|
35
|
+
const code = generatePairingCode(this.codeLength);
|
|
31
36
|
this.pending.set(code, {
|
|
32
37
|
code,
|
|
33
38
|
userId: msg.userId,
|
|
34
39
|
channelType: msg.channelType,
|
|
35
|
-
expiresAt:
|
|
40
|
+
expiresAt: now + this.expiresIn,
|
|
36
41
|
});
|
|
37
42
|
await ctx.channel.sendText(msg.channelId, `Hi! Ask the bot owner to approve you with: \`/pair ${code}\``);
|
|
38
43
|
}
|
|
39
44
|
async handleApproval(msg, ctx) {
|
|
40
|
-
const code = msg.text
|
|
41
|
-
const pairing = this.pending.get(code);
|
|
42
|
-
if (!pairing) {
|
|
45
|
+
const code = parsePairCommand(msg.text);
|
|
46
|
+
const pairing = code ? this.pending.get(code) : undefined;
|
|
47
|
+
if (!code || !pairing) {
|
|
43
48
|
await ctx.channel.sendText(msg.channelId, 'Invalid or expired pairing code.');
|
|
44
49
|
return;
|
|
45
50
|
}
|
|
46
|
-
if (pairing.expiresAt < Date.now()) {
|
|
47
|
-
this.pending.delete(code);
|
|
48
|
-
await ctx.channel.sendText(msg.channelId, 'Pairing code expired.');
|
|
49
|
-
return;
|
|
50
|
-
}
|
|
51
51
|
const userKey = `${pairing.channelType}:${pairing.userId}`;
|
|
52
52
|
this.approved.add(userKey);
|
|
53
53
|
this.pending.delete(code);
|
|
54
54
|
await ctx.channel.sendText(msg.channelId, `User approved (${userKey}).`);
|
|
55
55
|
}
|
|
56
|
-
findPendingByUser(userKey) {
|
|
57
|
-
for (const p of this.pending.values()) {
|
|
58
|
-
if (`${p.channelType}:${p.userId}` === userKey)
|
|
59
|
-
return p;
|
|
60
|
-
}
|
|
61
|
-
return undefined;
|
|
62
|
-
}
|
|
63
56
|
isApproved(channelType, userId) {
|
|
64
57
|
return this.approved.has(`${channelType}:${userId}`);
|
|
65
58
|
}
|