@sjawhar/whatsapp-mcp 1.0.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/README.md +122 -0
- package/dist/__tests__/connection.test.d.ts +2 -0
- package/dist/__tests__/connection.test.d.ts.map +1 -0
- package/dist/__tests__/connection.test.js +105 -0
- package/dist/__tests__/connection.test.js.map +1 -0
- package/dist/__tests__/disconnect.test.d.ts +2 -0
- package/dist/__tests__/disconnect.test.d.ts.map +1 -0
- package/dist/__tests__/disconnect.test.js +166 -0
- package/dist/__tests__/disconnect.test.js.map +1 -0
- package/dist/__tests__/download-media.test.d.ts +2 -0
- package/dist/__tests__/download-media.test.d.ts.map +1 -0
- package/dist/__tests__/download-media.test.js +110 -0
- package/dist/__tests__/download-media.test.js.map +1 -0
- package/dist/__tests__/failures/connection-failures.test.d.ts +2 -0
- package/dist/__tests__/failures/connection-failures.test.d.ts.map +1 -0
- package/dist/__tests__/failures/connection-failures.test.js +146 -0
- package/dist/__tests__/failures/connection-failures.test.js.map +1 -0
- package/dist/__tests__/failures/edge-cases.test.d.ts +2 -0
- package/dist/__tests__/failures/edge-cases.test.d.ts.map +1 -0
- package/dist/__tests__/failures/edge-cases.test.js +121 -0
- package/dist/__tests__/failures/edge-cases.test.js.map +1 -0
- package/dist/__tests__/failures/resource-failures.test.d.ts +2 -0
- package/dist/__tests__/failures/resource-failures.test.d.ts.map +1 -0
- package/dist/__tests__/failures/resource-failures.test.js +136 -0
- package/dist/__tests__/failures/resource-failures.test.js.map +1 -0
- package/dist/__tests__/failures/security-failures.test.d.ts +2 -0
- package/dist/__tests__/failures/security-failures.test.d.ts.map +1 -0
- package/dist/__tests__/failures/security-failures.test.js +0 -0
- package/dist/__tests__/failures/security-failures.test.js.map +1 -0
- package/dist/__tests__/helpers/fake-baileys.d.ts +52 -0
- package/dist/__tests__/helpers/fake-baileys.d.ts.map +1 -0
- package/dist/__tests__/helpers/fake-baileys.js +60 -0
- package/dist/__tests__/helpers/fake-baileys.js.map +1 -0
- package/dist/__tests__/helpers/mcp-test-client.d.ts +9 -0
- package/dist/__tests__/helpers/mcp-test-client.d.ts.map +1 -0
- package/dist/__tests__/helpers/mcp-test-client.js +40 -0
- package/dist/__tests__/helpers/mcp-test-client.js.map +1 -0
- package/dist/__tests__/helpers/test-db.d.ts +4 -0
- package/dist/__tests__/helpers/test-db.d.ts.map +1 -0
- package/dist/__tests__/helpers/test-db.js +32 -0
- package/dist/__tests__/helpers/test-db.js.map +1 -0
- package/dist/__tests__/integration/chat-navigation.test.d.ts +2 -0
- package/dist/__tests__/integration/chat-navigation.test.d.ts.map +1 -0
- package/dist/__tests__/integration/chat-navigation.test.js +171 -0
- package/dist/__tests__/integration/chat-navigation.test.js.map +1 -0
- package/dist/__tests__/integration/contacts-flow.test.d.ts +2 -0
- package/dist/__tests__/integration/contacts-flow.test.d.ts.map +1 -0
- package/dist/__tests__/integration/contacts-flow.test.js +144 -0
- package/dist/__tests__/integration/contacts-flow.test.js.map +1 -0
- package/dist/__tests__/integration/media-flow.test.d.ts +2 -0
- package/dist/__tests__/integration/media-flow.test.d.ts.map +1 -0
- package/dist/__tests__/integration/media-flow.test.js +225 -0
- package/dist/__tests__/integration/media-flow.test.js.map +1 -0
- package/dist/__tests__/integration/search-flow.test.d.ts +2 -0
- package/dist/__tests__/integration/search-flow.test.d.ts.map +1 -0
- package/dist/__tests__/integration/search-flow.test.js +44 -0
- package/dist/__tests__/integration/search-flow.test.js.map +1 -0
- package/dist/__tests__/integration/send-message.test.d.ts +2 -0
- package/dist/__tests__/integration/send-message.test.d.ts.map +1 -0
- package/dist/__tests__/integration/send-message.test.js +160 -0
- package/dist/__tests__/integration/send-message.test.js.map +1 -0
- package/dist/__tests__/lock-file.test.d.ts +2 -0
- package/dist/__tests__/lock-file.test.d.ts.map +1 -0
- package/dist/__tests__/lock-file.test.js +63 -0
- package/dist/__tests__/lock-file.test.js.map +1 -0
- package/dist/__tests__/medium-fixes.test.d.ts +2 -0
- package/dist/__tests__/medium-fixes.test.d.ts.map +1 -0
- package/dist/__tests__/medium-fixes.test.js +141 -0
- package/dist/__tests__/medium-fixes.test.js.map +1 -0
- package/dist/__tests__/rate-limit.test.d.ts +2 -0
- package/dist/__tests__/rate-limit.test.d.ts.map +1 -0
- package/dist/__tests__/rate-limit.test.js +193 -0
- package/dist/__tests__/rate-limit.test.js.map +1 -0
- package/dist/__tests__/send-file.test.d.ts +2 -0
- package/dist/__tests__/send-file.test.d.ts.map +1 -0
- package/dist/__tests__/send-file.test.js +237 -0
- package/dist/__tests__/send-file.test.js.map +1 -0
- package/dist/__tests__/smoke.test.d.ts +2 -0
- package/dist/__tests__/smoke.test.d.ts.map +1 -0
- package/dist/__tests__/smoke.test.js +28 -0
- package/dist/__tests__/smoke.test.js.map +1 -0
- package/dist/__tests__/transcribe.test.d.ts +2 -0
- package/dist/__tests__/transcribe.test.d.ts.map +1 -0
- package/dist/__tests__/transcribe.test.js +71 -0
- package/dist/__tests__/transcribe.test.js.map +1 -0
- package/dist/__tests__/zombie.test.d.ts +2 -0
- package/dist/__tests__/zombie.test.d.ts.map +1 -0
- package/dist/__tests__/zombie.test.js +145 -0
- package/dist/__tests__/zombie.test.js.map +1 -0
- package/dist/db.d.ts +53 -0
- package/dist/db.d.ts.map +1 -0
- package/dist/db.js +509 -0
- package/dist/db.js.map +1 -0
- package/dist/import-contacts.d.ts +37 -0
- package/dist/import-contacts.d.ts.map +1 -0
- package/dist/import-contacts.js +242 -0
- package/dist/import-contacts.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +62 -0
- package/dist/index.js.map +1 -0
- package/dist/lock.d.ts +16 -0
- package/dist/lock.d.ts.map +1 -0
- package/dist/lock.js +65 -0
- package/dist/lock.js.map +1 -0
- package/dist/tools.d.ts +6 -0
- package/dist/tools.d.ts.map +1 -0
- package/dist/tools.js +339 -0
- package/dist/tools.js.map +1 -0
- package/dist/transcribe.d.ts +8 -0
- package/dist/transcribe.d.ts.map +1 -0
- package/dist/transcribe.js +63 -0
- package/dist/transcribe.js.map +1 -0
- package/dist/utils.d.ts +51 -0
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +156 -0
- package/dist/utils.js.map +1 -0
- package/dist/whatsapp.d.ts +50 -0
- package/dist/whatsapp.d.ts.map +1 -0
- package/dist/whatsapp.js +896 -0
- package/dist/whatsapp.js.map +1 -0
- package/package.json +52 -0
- package/patches/@whiskeysockets+baileys+6.7.21.patch +46 -0
- package/patches/libsignal+2.0.1.patch +84 -0
|
@@ -0,0 +1,225 @@
|
|
|
1
|
+
import fs from "node:fs";
|
|
2
|
+
import os from "node:os";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import { afterAll, beforeAll, describe, expect, it, vi } from "vitest";
|
|
5
|
+
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
6
|
+
import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
|
|
7
|
+
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
8
|
+
import { createFakeBaileysSocket } from "../helpers/fake-baileys.js";
|
|
9
|
+
import { closeTestDb, seedTestDb, setupTestDb } from "../helpers/test-db.js";
|
|
10
|
+
const mockState = vi.hoisted(() => ({
|
|
11
|
+
sockets: [],
|
|
12
|
+
}));
|
|
13
|
+
vi.mock("@whiskeysockets/baileys", () => ({
|
|
14
|
+
default: vi.fn(() => {
|
|
15
|
+
const socket = createFakeBaileysSocket();
|
|
16
|
+
socket.sendMessage = async (jid, content) => {
|
|
17
|
+
const streamLike = [content.image, content.video, content.audio, content.document].find((value) => value && typeof value.on === "function");
|
|
18
|
+
if (streamLike) {
|
|
19
|
+
await new Promise((resolve) => {
|
|
20
|
+
streamLike.on("open", () => {
|
|
21
|
+
streamLike.destroy();
|
|
22
|
+
resolve();
|
|
23
|
+
});
|
|
24
|
+
streamLike.on("error", () => resolve());
|
|
25
|
+
});
|
|
26
|
+
}
|
|
27
|
+
socket.sentMessages.push({ jid, content });
|
|
28
|
+
return { key: { id: `fake-msg-${socket.sentMessages.length}` } };
|
|
29
|
+
};
|
|
30
|
+
mockState.sockets.push(socket);
|
|
31
|
+
return socket;
|
|
32
|
+
}),
|
|
33
|
+
DisconnectReason: {
|
|
34
|
+
loggedOut: 401,
|
|
35
|
+
connectionReplaced: 440,
|
|
36
|
+
},
|
|
37
|
+
fetchLatestBaileysVersion: vi.fn(async () => ({ version: [2, 3000, 0] })),
|
|
38
|
+
downloadMediaMessage: vi.fn(async () => Buffer.from("fake-media-bytes")),
|
|
39
|
+
getContentType: vi.fn((message) => Object.keys(message)[0]),
|
|
40
|
+
initAuthCreds: vi.fn(() => ({})),
|
|
41
|
+
BufferJSON: {
|
|
42
|
+
replacer: (_key, value) => value,
|
|
43
|
+
reviver: (_key, value) => value,
|
|
44
|
+
},
|
|
45
|
+
proto: {
|
|
46
|
+
Message: {
|
|
47
|
+
AppStateSyncKeyData: {
|
|
48
|
+
fromObject: (value) => value,
|
|
49
|
+
},
|
|
50
|
+
},
|
|
51
|
+
},
|
|
52
|
+
}));
|
|
53
|
+
function parseToolJson(result) {
|
|
54
|
+
const text = result.content?.find((entry) => entry.type === "text")?.text || "";
|
|
55
|
+
try {
|
|
56
|
+
return JSON.parse(text);
|
|
57
|
+
}
|
|
58
|
+
catch {
|
|
59
|
+
return text;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
describe("integration: media flow", () => {
|
|
63
|
+
const originalAllowedDir = process.env.ALLOWED_SEND_DIR;
|
|
64
|
+
const originalDownloadsDir = process.env.DOWNLOADS_DIR;
|
|
65
|
+
const originalWhisperApiKey = process.env.WHISPER_API_KEY;
|
|
66
|
+
const originalWhisperApiUrl = process.env.WHISPER_API_URL;
|
|
67
|
+
const originalWhisperModel = process.env.WHISPER_MODEL;
|
|
68
|
+
let uploadDir;
|
|
69
|
+
let downloadsDir;
|
|
70
|
+
let server;
|
|
71
|
+
let client;
|
|
72
|
+
let closeWhatsApp;
|
|
73
|
+
beforeAll(async () => {
|
|
74
|
+
uploadDir = fs.mkdtempSync(path.join(os.tmpdir(), "integration-upload-"));
|
|
75
|
+
downloadsDir = fs.mkdtempSync(path.join(os.tmpdir(), "integration-download-"));
|
|
76
|
+
process.env.ALLOWED_SEND_DIR = uploadDir;
|
|
77
|
+
process.env.DOWNLOADS_DIR = downloadsDir;
|
|
78
|
+
process.env.MIN_SEND_INTERVAL_MS = "0";
|
|
79
|
+
process.env.SEND_JITTER_MS = "0";
|
|
80
|
+
process.env.WHISPER_API_KEY = "test-whisper-key";
|
|
81
|
+
process.env.WHISPER_API_URL = "https://api.example.com/v1/audio/transcriptions";
|
|
82
|
+
process.env.WHISPER_MODEL = "test-model";
|
|
83
|
+
await setupTestDb();
|
|
84
|
+
seedTestDb();
|
|
85
|
+
const { getDb } = await import("../../db.js");
|
|
86
|
+
const db = getDb();
|
|
87
|
+
db.prepare("INSERT INTO messages (id, chat_jid, from_me, sender_jid, sender_name, type, text, timestamp, has_media, message_blob, transcription) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)").run("img-msg-1", "15550001111@s.whatsapp.net", 0, "15550001111@s.whatsapp.net", "Alice Test", "image", null, 1700000100, 1, JSON.stringify({ message: { imageMessage: { mimetype: "image/png" } } }), null);
|
|
88
|
+
db.prepare("INSERT INTO messages (id, chat_jid, from_me, sender_jid, sender_name, type, text, timestamp, has_media, message_blob, transcription) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)").run("voice-msg-1", "15550001111@s.whatsapp.net", 0, "15550001111@s.whatsapp.net", "Alice Test", "voice_note", null, 1700000110, 1, JSON.stringify({ message: { audioMessage: { mimetype: "audio/ogg; codecs=opus" } } }), null);
|
|
89
|
+
vi.stubGlobal("fetch", vi.fn(async () => ({
|
|
90
|
+
ok: true,
|
|
91
|
+
status: 200,
|
|
92
|
+
text: async () => "",
|
|
93
|
+
json: async () => ({
|
|
94
|
+
text: "Transcribed hello world",
|
|
95
|
+
language: "en",
|
|
96
|
+
duration: 1.23,
|
|
97
|
+
}),
|
|
98
|
+
})));
|
|
99
|
+
const whatsapp = await import("../../whatsapp.js");
|
|
100
|
+
closeWhatsApp = whatsapp.closeWhatsApp;
|
|
101
|
+
await whatsapp.initWhatsApp();
|
|
102
|
+
const socket = mockState.sockets.at(-1);
|
|
103
|
+
if (!socket)
|
|
104
|
+
throw new Error("Expected fake Baileys socket");
|
|
105
|
+
socket.emitConnectionOpen();
|
|
106
|
+
server = new McpServer({ name: "whatsapp-test", version: "1.0.0" });
|
|
107
|
+
const { registerTools } = await import("../../tools.js");
|
|
108
|
+
registerTools(server);
|
|
109
|
+
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
|
|
110
|
+
client = new Client({ name: "whatsapp-test-client", version: "1.0.0" });
|
|
111
|
+
await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
|
|
112
|
+
});
|
|
113
|
+
afterAll(async () => {
|
|
114
|
+
await Promise.all([client?.close(), server?.close()]);
|
|
115
|
+
await closeWhatsApp?.();
|
|
116
|
+
closeTestDb();
|
|
117
|
+
vi.unstubAllGlobals();
|
|
118
|
+
fs.rmSync(uploadDir, { recursive: true, force: true });
|
|
119
|
+
fs.rmSync(downloadsDir, { recursive: true, force: true });
|
|
120
|
+
if (originalAllowedDir === undefined)
|
|
121
|
+
delete process.env.ALLOWED_SEND_DIR;
|
|
122
|
+
else
|
|
123
|
+
process.env.ALLOWED_SEND_DIR = originalAllowedDir;
|
|
124
|
+
if (originalDownloadsDir === undefined)
|
|
125
|
+
delete process.env.DOWNLOADS_DIR;
|
|
126
|
+
else
|
|
127
|
+
process.env.DOWNLOADS_DIR = originalDownloadsDir;
|
|
128
|
+
if (originalWhisperApiKey === undefined)
|
|
129
|
+
delete process.env.WHISPER_API_KEY;
|
|
130
|
+
else
|
|
131
|
+
process.env.WHISPER_API_KEY = originalWhisperApiKey;
|
|
132
|
+
if (originalWhisperApiUrl === undefined)
|
|
133
|
+
delete process.env.WHISPER_API_URL;
|
|
134
|
+
else
|
|
135
|
+
process.env.WHISPER_API_URL = originalWhisperApiUrl;
|
|
136
|
+
if (originalWhisperModel === undefined)
|
|
137
|
+
delete process.env.WHISPER_MODEL;
|
|
138
|
+
else
|
|
139
|
+
process.env.WHISPER_MODEL = originalWhisperModel;
|
|
140
|
+
});
|
|
141
|
+
it("handles send_file preview + send flow", async () => {
|
|
142
|
+
const filePath = path.join(uploadDir, "story.txt");
|
|
143
|
+
fs.writeFileSync(filePath, "integration media test");
|
|
144
|
+
const preview = (await client.callTool({
|
|
145
|
+
name: "send_file",
|
|
146
|
+
arguments: {
|
|
147
|
+
jid: "15550001111@s.whatsapp.net",
|
|
148
|
+
filePath,
|
|
149
|
+
caption: "preview cap",
|
|
150
|
+
confirmed: false,
|
|
151
|
+
},
|
|
152
|
+
}));
|
|
153
|
+
expect(preview.isError).toBeFalsy();
|
|
154
|
+
expect(parseToolJson(preview)).toMatchObject({
|
|
155
|
+
preview: true,
|
|
156
|
+
fileName: "story.txt",
|
|
157
|
+
recipient: expect.objectContaining({ jid: "15550001111@s.whatsapp.net" }),
|
|
158
|
+
});
|
|
159
|
+
const sent = (await client.callTool({
|
|
160
|
+
name: "send_file",
|
|
161
|
+
arguments: {
|
|
162
|
+
jid: "15550001111@s.whatsapp.net",
|
|
163
|
+
filePath,
|
|
164
|
+
caption: "send cap",
|
|
165
|
+
confirmed: true,
|
|
166
|
+
},
|
|
167
|
+
}));
|
|
168
|
+
expect(sent.isError).toBeFalsy();
|
|
169
|
+
expect(parseToolJson(sent)).toMatchObject({
|
|
170
|
+
success: true,
|
|
171
|
+
to: "15550001111@s.whatsapp.net",
|
|
172
|
+
fileType: "document",
|
|
173
|
+
});
|
|
174
|
+
});
|
|
175
|
+
it("downloads media and transcribes voice note", async () => {
|
|
176
|
+
const download = (await client.callTool({
|
|
177
|
+
name: "download_media",
|
|
178
|
+
arguments: {
|
|
179
|
+
jid: "15550001111@s.whatsapp.net",
|
|
180
|
+
messageId: "img-msg-1",
|
|
181
|
+
},
|
|
182
|
+
}));
|
|
183
|
+
const downloadPayload = parseToolJson(download);
|
|
184
|
+
expect(download.isError).toBeFalsy();
|
|
185
|
+
expect(downloadPayload).toMatchObject({
|
|
186
|
+
success: true,
|
|
187
|
+
fileName: "img-msg-1.png",
|
|
188
|
+
mimeType: "image/png",
|
|
189
|
+
type: "imageMessage",
|
|
190
|
+
size: "fake-media-bytes".length,
|
|
191
|
+
});
|
|
192
|
+
expect(fs.existsSync(downloadPayload.filePath)).toBe(true);
|
|
193
|
+
const transcribed = (await client.callTool({
|
|
194
|
+
name: "transcribe_voice_note",
|
|
195
|
+
arguments: {
|
|
196
|
+
jid: "15550001111@s.whatsapp.net",
|
|
197
|
+
messageId: "voice-msg-1",
|
|
198
|
+
},
|
|
199
|
+
}));
|
|
200
|
+
const transcribedPayload = parseToolJson(transcribed);
|
|
201
|
+
expect(transcribed.isError).toBeFalsy();
|
|
202
|
+
expect(transcribedPayload).toMatchObject({
|
|
203
|
+
success: true,
|
|
204
|
+
messageId: "voice-msg-1",
|
|
205
|
+
chatJid: "15550001111@s.whatsapp.net",
|
|
206
|
+
transcription: "Transcribed hello world",
|
|
207
|
+
language: "en",
|
|
208
|
+
cached: false,
|
|
209
|
+
});
|
|
210
|
+
const cached = (await client.callTool({
|
|
211
|
+
name: "transcribe_voice_note",
|
|
212
|
+
arguments: {
|
|
213
|
+
jid: "15550001111@s.whatsapp.net",
|
|
214
|
+
messageId: "voice-msg-1",
|
|
215
|
+
},
|
|
216
|
+
}));
|
|
217
|
+
expect(cached.isError).toBeFalsy();
|
|
218
|
+
expect(parseToolJson(cached)).toMatchObject({
|
|
219
|
+
success: true,
|
|
220
|
+
messageId: "voice-msg-1",
|
|
221
|
+
cached: true,
|
|
222
|
+
});
|
|
223
|
+
});
|
|
224
|
+
});
|
|
225
|
+
//# sourceMappingURL=media-flow.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"media-flow.test.js","sourceRoot":"","sources":["../../../src/__tests__/integration/media-flow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACvE,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAA0B,MAAM,4BAA4B,CAAC;AAC7F,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAE7E,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,EAAyB;CACnC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;QAClB,MAAM,MAAM,GAAG,uBAAuB,EAAE,CAAC;QACzC,MAAM,CAAC,WAAW,GAAG,KAAK,EAAE,GAAW,EAAE,OAAgC,EAAE,EAAE;YAC3E,MAAM,UAAU,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CACrF,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,IAAI,OAAQ,KAAa,CAAC,EAAE,KAAK,UAAU,CACrD,CAAC;YAET,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;oBAClC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;wBACzB,UAAU,CAAC,OAAO,EAAE,CAAC;wBACrB,OAAO,EAAE,CAAC;oBACZ,CAAC,CAAC,CAAC;oBACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC1C,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;YAC3C,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,YAAY,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;QACnE,CAAC,CAAC;QACF,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IACF,gBAAgB,EAAE;QAChB,SAAS,EAAE,GAAG;QACd,kBAAkB,EAAE,GAAG;KACxB;IACD,yBAAyB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IACzE,oBAAoB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACxE,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,OAAY,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAChE,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IAChC,UAAU,EAAE;QACV,QAAQ,EAAE,CAAC,IAAY,EAAE,KAAc,EAAE,EAAE,CAAC,KAAK;QACjD,OAAO,EAAE,CAAC,IAAY,EAAE,KAAc,EAAE,EAAE,CAAC,KAAK;KACjD;IACD,KAAK,EAAE;QACL,OAAO,EAAE;YACP,mBAAmB,EAAE;gBACnB,UAAU,EAAE,CAAC,KAAc,EAAE,EAAE,CAAC,KAAK;aACtC;SACF;KACF;CACF,CAAC,CAAC,CAAC;AAOJ,SAAS,aAAa,CAAC,MAAqB;IAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IAChF,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IACxD,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IACvD,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC1D,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC1D,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;IAEvD,IAAI,SAAiB,CAAC;IACtB,IAAI,YAAoB,CAAC;IACzB,IAAI,MAAiB,CAAC;IACtB,IAAI,MAAc,CAAC;IACnB,IAAI,aAAkC,CAAC;IAEvC,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,SAAS,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,qBAAqB,CAAC,CAAC,CAAC;QAC1E,YAAY,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAC;QAE/E,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,SAAS,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,YAAY,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,GAAG,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,GAAG,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,kBAAkB,CAAC;QACjD,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,iDAAiD,CAAC;QAChF,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,YAAY,CAAC;QAEzC,MAAM,WAAW,EAAE,CAAC;QACpB,UAAU,EAAE,CAAC;QAEb,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;QACnB,EAAE,CAAC,OAAO,CACR,+KAA+K,CAChL,CAAC,GAAG,CACH,WAAW,EACX,4BAA4B,EAC5B,CAAC,EACD,4BAA4B,EAC5B,YAAY,EACZ,OAAO,EACP,IAAI,EACJ,UAAU,EACV,CAAC,EACD,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,EACxE,IAAI,CACL,CAAC;QACF,EAAE,CAAC,OAAO,CACR,+KAA+K,CAChL,CAAC,GAAG,CACH,aAAa,EACb,4BAA4B,EAC5B,CAAC,EACD,4BAA4B,EAC5B,YAAY,EACZ,YAAY,EACZ,IAAI,EACJ,UAAU,EACV,CAAC,EACD,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,EAAE,QAAQ,EAAE,wBAAwB,EAAE,EAAE,EAAE,CAAC,EACrF,IAAI,CACL,CAAC;QAEF,EAAE,CAAC,UAAU,CACX,OAAO,EACP,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACjB,EAAE,EAAE,IAAI;YACR,MAAM,EAAE,GAAG;YACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;YACpB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBACjB,IAAI,EAAE,yBAAyB;gBAC/B,QAAQ,EAAE,IAAI;gBACd,QAAQ,EAAE,IAAI;aACf,CAAC;SACH,CAAC,CAAC,CACJ,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACnD,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC;QACvC,MAAM,QAAQ,CAAC,YAAY,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC7D,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAE5B,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACpE,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACzD,aAAa,CAAC,MAAM,CAAC,CAAC;QAEtB,MAAM,CAAC,eAAe,EAAE,eAAe,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChF,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACxE,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,EAAE,EAAE,CAAC;QACxB,WAAW,EAAE,CAAC;QACd,EAAE,CAAC,gBAAgB,EAAE,CAAC;QAEtB,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,EAAE,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAE1D,IAAI,kBAAkB,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;;YACrE,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,kBAAkB,CAAC;QACvD,IAAI,oBAAoB,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;;YACpE,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,oBAAoB,CAAC;QACtD,IAAI,qBAAqB,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;;YACvE,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,qBAAqB,CAAC;QACzD,IAAI,qBAAqB,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;;YACvE,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,qBAAqB,CAAC;QACzD,IAAI,oBAAoB,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;;YACpE,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,oBAAoB,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACnD,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,wBAAwB,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACrC,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,QAAQ;gBACR,OAAO,EAAE,aAAa;gBACtB,SAAS,EAAE,KAAK;aACjB;SACF,CAAC,CAAkB,CAAC;QACrB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACpC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC;YAC3C,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,WAAW;YACrB,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,CAAC;SAC1E,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YAClC,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,QAAQ;gBACR,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAkB,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACjC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC;YACxC,OAAO,EAAE,IAAI;YACb,EAAE,EAAE,4BAA4B;YAChC,QAAQ,EAAE,UAAU;SACrB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,QAAQ,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACtC,IAAI,EAAE,gBAAgB;YACtB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,SAAS,EAAE,WAAW;aACvB;SACF,CAAC,CAAkB,CAAC;QAErB,MAAM,eAAe,GAAG,aAAa,CAAC,QAAQ,CAAQ,CAAC;QACvD,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACrC,MAAM,CAAC,eAAe,CAAC,CAAC,aAAa,CAAC;YACpC,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,eAAe;YACzB,QAAQ,EAAE,WAAW;YACrB,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,kBAAkB,CAAC,MAAM;SAChC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE3D,MAAM,WAAW,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACzC,IAAI,EAAE,uBAAuB;YAC7B,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,SAAS,EAAE,aAAa;aACzB;SACF,CAAC,CAAkB,CAAC;QAErB,MAAM,kBAAkB,GAAG,aAAa,CAAC,WAAW,CAAQ,CAAC;QAC7D,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACxC,MAAM,CAAC,kBAAkB,CAAC,CAAC,aAAa,CAAC;YACvC,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,aAAa;YACxB,OAAO,EAAE,4BAA4B;YACrC,aAAa,EAAE,yBAAyB;YACxC,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACpC,IAAI,EAAE,uBAAuB;YAC7B,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,SAAS,EAAE,aAAa;aACzB;SACF,CAAC,CAAkB,CAAC;QACrB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC;YAC1C,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,aAAa;YACxB,MAAM,EAAE,IAAI;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search-flow.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/integration/search-flow.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { afterAll, beforeAll, describe, expect, it } from "vitest";
|
|
2
|
+
import { closeTestDb, seedTestDb, setupTestDb } from "../helpers/test-db.js";
|
|
3
|
+
import { createMcpTestClient } from "../helpers/mcp-test-client.js";
|
|
4
|
+
describe("integration: search and listing flow", () => {
|
|
5
|
+
let client;
|
|
6
|
+
beforeAll(async () => {
|
|
7
|
+
await setupTestDb();
|
|
8
|
+
seedTestDb();
|
|
9
|
+
const { getDb } = await import("../../db.js");
|
|
10
|
+
const db = getDb();
|
|
11
|
+
db.prepare("INSERT INTO contacts (jid, name, notify) VALUES (?, ?, ?)").run("15550002222@s.whatsapp.net", "Bob Search", "Bob");
|
|
12
|
+
db.prepare("INSERT INTO contacts (jid, name, notify) VALUES (?, ?, ?)").run("15550003333@s.whatsapp.net", "Charlie Search", "Charlie");
|
|
13
|
+
db.prepare("INSERT INTO chats (jid, name, conversation_ts, unread_count) VALUES (?, ?, ?, ?)").run("15550002222@s.whatsapp.net", "Bob Search", 1700000010, 1);
|
|
14
|
+
db.prepare("INSERT INTO chats (jid, name, conversation_ts, unread_count) VALUES (?, ?, ?, ?)").run("15550003333@s.whatsapp.net", "Charlie Search", 1700000020, 0);
|
|
15
|
+
db.prepare("INSERT INTO messages (id, chat_jid, from_me, sender_jid, sender_name, type, text, timestamp, has_media, message_blob, transcription) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)").run("search-msg-1", "15550002222@s.whatsapp.net", 0, "15550002222@s.whatsapp.net", "Bob Search", "text", "Need coffee beans and filters", 1700000010, 0, null, null);
|
|
16
|
+
db.prepare("INSERT INTO messages (id, chat_jid, from_me, sender_jid, sender_name, type, text, timestamp, has_media, message_blob, transcription) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)").run("search-msg-2", "15550003333@s.whatsapp.net", 0, "15550003333@s.whatsapp.net", "Charlie Search", "text", "Coffee meeting tomorrow", 1700000020, 0, null, null);
|
|
17
|
+
client = await createMcpTestClient();
|
|
18
|
+
});
|
|
19
|
+
afterAll(async () => {
|
|
20
|
+
await client?.close();
|
|
21
|
+
closeTestDb();
|
|
22
|
+
});
|
|
23
|
+
it("searches messages and contacts, and honors list_chats limit", async () => {
|
|
24
|
+
const messageResults = (await client.callTool("search_messages", {
|
|
25
|
+
query: "coffee",
|
|
26
|
+
}));
|
|
27
|
+
expect(messageResults).toEqual(expect.arrayContaining([
|
|
28
|
+
expect.objectContaining({ id: "search-msg-1" }),
|
|
29
|
+
expect.objectContaining({ id: "search-msg-2" }),
|
|
30
|
+
]));
|
|
31
|
+
const contactResults = (await client.callTool("search_contacts", {
|
|
32
|
+
query: "Search",
|
|
33
|
+
}));
|
|
34
|
+
expect(contactResults).toEqual(expect.arrayContaining([
|
|
35
|
+
expect.objectContaining({ jid: "15550002222@s.whatsapp.net", name: "Bob Search" }),
|
|
36
|
+
expect.objectContaining({ jid: "15550003333@s.whatsapp.net", name: "Charlie Search" }),
|
|
37
|
+
]));
|
|
38
|
+
const chats = (await client.callTool("list_chats", { limit: 2 }));
|
|
39
|
+
expect(chats).toHaveLength(2);
|
|
40
|
+
expect(chats[0]).toMatchObject({ jid: "15550003333@s.whatsapp.net" });
|
|
41
|
+
expect(chats[1]).toMatchObject({ jid: "15550002222@s.whatsapp.net" });
|
|
42
|
+
});
|
|
43
|
+
});
|
|
44
|
+
//# sourceMappingURL=search-flow.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search-flow.test.js","sourceRoot":"","sources":["../../../src/__tests__/integration/search-flow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAC7E,OAAO,EAAE,mBAAmB,EAAsB,MAAM,+BAA+B,CAAC;AAExF,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,IAAI,MAAqB,CAAC;IAE1B,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,WAAW,EAAE,CAAC;QACpB,UAAU,EAAE,CAAC;QAEb,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;QACnB,EAAE,CAAC,OAAO,CAAC,2DAA2D,CAAC,CAAC,GAAG,CACzE,4BAA4B,EAC5B,YAAY,EACZ,KAAK,CACN,CAAC;QACF,EAAE,CAAC,OAAO,CAAC,2DAA2D,CAAC,CAAC,GAAG,CACzE,4BAA4B,EAC5B,gBAAgB,EAChB,SAAS,CACV,CAAC;QAEF,EAAE,CAAC,OAAO,CAAC,kFAAkF,CAAC,CAAC,GAAG,CAChG,4BAA4B,EAC5B,YAAY,EACZ,UAAU,EACV,CAAC,CACF,CAAC;QACF,EAAE,CAAC,OAAO,CAAC,kFAAkF,CAAC,CAAC,GAAG,CAChG,4BAA4B,EAC5B,gBAAgB,EAChB,UAAU,EACV,CAAC,CACF,CAAC;QAEF,EAAE,CAAC,OAAO,CACR,+KAA+K,CAChL,CAAC,GAAG,CACH,cAAc,EACd,4BAA4B,EAC5B,CAAC,EACD,4BAA4B,EAC5B,YAAY,EACZ,MAAM,EACN,+BAA+B,EAC/B,UAAU,EACV,CAAC,EACD,IAAI,EACJ,IAAI,CACL,CAAC;QACF,EAAE,CAAC,OAAO,CACR,+KAA+K,CAChL,CAAC,GAAG,CACH,cAAc,EACd,4BAA4B,EAC5B,CAAC,EACD,4BAA4B,EAC5B,gBAAgB,EAChB,MAAM,EACN,yBAAyB,EACzB,UAAU,EACV,CAAC,EACD,IAAI,EACJ,IAAI,CACL,CAAC;QAEF,MAAM,GAAG,MAAM,mBAAmB,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,MAAM,EAAE,KAAK,EAAE,CAAC;QACtB,WAAW,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,cAAc,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC,iBAAiB,EAAE;YAC/D,KAAK,EAAE,QAAQ;SAChB,CAAC,CAAU,CAAC;QACb,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAC5B,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC;YAC/C,MAAM,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC;SAChD,CAAC,CACH,CAAC;QAEF,MAAM,cAAc,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC,iBAAiB,EAAE;YAC/D,KAAK,EAAE,QAAQ;SAChB,CAAC,CAAU,CAAC;QACb,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAC5B,MAAM,CAAC,eAAe,CAAC;YACrB,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;YAClF,MAAM,CAAC,gBAAgB,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;SACvF,CAAC,CACH,CAAC;QAEF,MAAM,KAAK,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAU,CAAC;QAC3E,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"send-message.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/integration/send-message.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,160 @@
|
|
|
1
|
+
import { afterAll, beforeAll, describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
3
|
+
import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
|
|
4
|
+
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
5
|
+
import { createFakeBaileysSocket } from "../helpers/fake-baileys.js";
|
|
6
|
+
import { closeTestDb, seedTestDb, setupTestDb } from "../helpers/test-db.js";
|
|
7
|
+
const mockState = vi.hoisted(() => ({
|
|
8
|
+
sockets: [],
|
|
9
|
+
}));
|
|
10
|
+
vi.mock("@whiskeysockets/baileys", () => ({
|
|
11
|
+
default: vi.fn(() => {
|
|
12
|
+
const socket = createFakeBaileysSocket();
|
|
13
|
+
mockState.sockets.push(socket);
|
|
14
|
+
return socket;
|
|
15
|
+
}),
|
|
16
|
+
DisconnectReason: {
|
|
17
|
+
loggedOut: 401,
|
|
18
|
+
connectionReplaced: 440,
|
|
19
|
+
},
|
|
20
|
+
fetchLatestBaileysVersion: vi.fn(async () => ({ version: [2, 3000, 0] })),
|
|
21
|
+
downloadMediaMessage: vi.fn(),
|
|
22
|
+
getContentType: vi.fn(),
|
|
23
|
+
initAuthCreds: vi.fn(() => ({})),
|
|
24
|
+
BufferJSON: {
|
|
25
|
+
replacer: (_key, value) => value,
|
|
26
|
+
reviver: (_key, value) => value,
|
|
27
|
+
},
|
|
28
|
+
proto: {
|
|
29
|
+
Message: {
|
|
30
|
+
AppStateSyncKeyData: {
|
|
31
|
+
fromObject: (value) => value,
|
|
32
|
+
},
|
|
33
|
+
},
|
|
34
|
+
},
|
|
35
|
+
}));
|
|
36
|
+
function parseToolJson(result) {
|
|
37
|
+
const text = result.content?.find((entry) => entry.type === "text")?.text || "";
|
|
38
|
+
try {
|
|
39
|
+
return JSON.parse(text);
|
|
40
|
+
}
|
|
41
|
+
catch {
|
|
42
|
+
return text;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
describe("integration: send_message", () => {
|
|
46
|
+
const originalMinInterval = process.env.MIN_SEND_INTERVAL_MS;
|
|
47
|
+
const originalJitter = process.env.SEND_JITTER_MS;
|
|
48
|
+
let server;
|
|
49
|
+
let client;
|
|
50
|
+
let closeWhatsApp;
|
|
51
|
+
beforeAll(async () => {
|
|
52
|
+
process.env.MIN_SEND_INTERVAL_MS = "150";
|
|
53
|
+
process.env.SEND_JITTER_MS = "0";
|
|
54
|
+
await setupTestDb();
|
|
55
|
+
seedTestDb();
|
|
56
|
+
const whatsapp = await import("../../whatsapp.js");
|
|
57
|
+
closeWhatsApp = whatsapp.closeWhatsApp;
|
|
58
|
+
await whatsapp.initWhatsApp();
|
|
59
|
+
const socket = mockState.sockets.at(-1);
|
|
60
|
+
if (!socket)
|
|
61
|
+
throw new Error("Expected fake Baileys socket");
|
|
62
|
+
socket.emitConnectionOpen();
|
|
63
|
+
server = new McpServer({ name: "whatsapp-test", version: "1.0.0" });
|
|
64
|
+
const { registerTools } = await import("../../tools.js");
|
|
65
|
+
registerTools(server);
|
|
66
|
+
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
|
|
67
|
+
client = new Client({ name: "whatsapp-test-client", version: "1.0.0" });
|
|
68
|
+
await Promise.all([server.connect(serverTransport), client.connect(clientTransport)]);
|
|
69
|
+
});
|
|
70
|
+
afterAll(async () => {
|
|
71
|
+
await Promise.all([client?.close(), server?.close()]);
|
|
72
|
+
await closeWhatsApp?.();
|
|
73
|
+
closeTestDb();
|
|
74
|
+
if (originalMinInterval === undefined) {
|
|
75
|
+
delete process.env.MIN_SEND_INTERVAL_MS;
|
|
76
|
+
}
|
|
77
|
+
else {
|
|
78
|
+
process.env.MIN_SEND_INTERVAL_MS = originalMinInterval;
|
|
79
|
+
}
|
|
80
|
+
if (originalJitter === undefined) {
|
|
81
|
+
delete process.env.SEND_JITTER_MS;
|
|
82
|
+
}
|
|
83
|
+
else {
|
|
84
|
+
process.env.SEND_JITTER_MS = originalJitter;
|
|
85
|
+
}
|
|
86
|
+
});
|
|
87
|
+
it("returns preview when confirmed=false", async () => {
|
|
88
|
+
const socket = mockState.sockets.at(-1);
|
|
89
|
+
if (!socket)
|
|
90
|
+
throw new Error("Expected fake Baileys socket");
|
|
91
|
+
const before = socket.sentMessages.length;
|
|
92
|
+
const result = (await client.callTool({
|
|
93
|
+
name: "send_message",
|
|
94
|
+
arguments: {
|
|
95
|
+
jid: "15550001111@s.whatsapp.net",
|
|
96
|
+
text: "Preview only",
|
|
97
|
+
confirmed: false,
|
|
98
|
+
},
|
|
99
|
+
}));
|
|
100
|
+
const payload = parseToolJson(result);
|
|
101
|
+
expect(result.isError).toBeFalsy();
|
|
102
|
+
expect(payload).toMatchObject({
|
|
103
|
+
status: "confirmation_required",
|
|
104
|
+
jid: "15550001111@s.whatsapp.net",
|
|
105
|
+
phone: "15550001111",
|
|
106
|
+
message: "Preview only",
|
|
107
|
+
});
|
|
108
|
+
expect(socket.sentMessages.length).toBe(before);
|
|
109
|
+
});
|
|
110
|
+
it("sends when confirmed=true", async () => {
|
|
111
|
+
const socket = mockState.sockets.at(-1);
|
|
112
|
+
if (!socket)
|
|
113
|
+
throw new Error("Expected fake Baileys socket");
|
|
114
|
+
const result = (await client.callTool({
|
|
115
|
+
name: "send_message",
|
|
116
|
+
arguments: {
|
|
117
|
+
jid: "15550001111@s.whatsapp.net",
|
|
118
|
+
text: "Actually send this",
|
|
119
|
+
confirmed: true,
|
|
120
|
+
},
|
|
121
|
+
}));
|
|
122
|
+
const payload = parseToolJson(result);
|
|
123
|
+
expect(result.isError).toBeFalsy();
|
|
124
|
+
expect(payload).toMatchObject({
|
|
125
|
+
success: true,
|
|
126
|
+
to: "15550001111@s.whatsapp.net",
|
|
127
|
+
messageId: expect.any(String),
|
|
128
|
+
});
|
|
129
|
+
expect(socket.sentMessages.at(-1)).toMatchObject({
|
|
130
|
+
jid: "15550001111@s.whatsapp.net",
|
|
131
|
+
content: { text: "Actually send this" },
|
|
132
|
+
});
|
|
133
|
+
});
|
|
134
|
+
it("applies rate limiting delay for second rapid send", async () => {
|
|
135
|
+
const firstResult = (await client.callTool({
|
|
136
|
+
name: "send_message",
|
|
137
|
+
arguments: {
|
|
138
|
+
jid: "15550001111@s.whatsapp.net",
|
|
139
|
+
text: "first rapid",
|
|
140
|
+
confirmed: true,
|
|
141
|
+
},
|
|
142
|
+
}));
|
|
143
|
+
expect(firstResult.isError).toBeFalsy();
|
|
144
|
+
const started = Date.now();
|
|
145
|
+
const secondResult = (await client.callTool({
|
|
146
|
+
name: "send_message",
|
|
147
|
+
arguments: {
|
|
148
|
+
jid: "15550001111@s.whatsapp.net",
|
|
149
|
+
text: "second rapid",
|
|
150
|
+
confirmed: true,
|
|
151
|
+
},
|
|
152
|
+
}));
|
|
153
|
+
const elapsed = Date.now() - started;
|
|
154
|
+
const payload = parseToolJson(secondResult);
|
|
155
|
+
expect(secondResult.isError).toBeFalsy();
|
|
156
|
+
expect(payload.success).toBe(true);
|
|
157
|
+
expect(elapsed).toBeGreaterThanOrEqual(120);
|
|
158
|
+
});
|
|
159
|
+
});
|
|
160
|
+
//# sourceMappingURL=send-message.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"send-message.test.js","sourceRoot":"","sources":["../../../src/__tests__/integration/send-message.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACvE,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAA0B,MAAM,4BAA4B,CAAC;AAC7F,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAE7E,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IAClC,OAAO,EAAE,EAAyB;CACnC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;QAClB,MAAM,MAAM,GAAG,uBAAuB,EAAE,CAAC;QACzC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IACF,gBAAgB,EAAE;QAChB,SAAS,EAAE,GAAG;QACd,kBAAkB,EAAE,GAAG;KACxB;IACD,yBAAyB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IACzE,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC7B,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;IACvB,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IAChC,UAAU,EAAE;QACV,QAAQ,EAAE,CAAC,IAAY,EAAE,KAAc,EAAE,EAAE,CAAC,KAAK;QACjD,OAAO,EAAE,CAAC,IAAY,EAAE,KAAc,EAAE,EAAE,CAAC,KAAK;KACjD;IACD,KAAK,EAAE;QACL,OAAO,EAAE;YACP,mBAAmB,EAAE;gBACnB,UAAU,EAAE,CAAC,KAAc,EAAE,EAAE,CAAC,KAAK;aACtC;SACF;KACF;CACF,CAAC,CAAC,CAAC;AAOJ,SAAS,aAAa,CAAC,MAAqB;IAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IAChF,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,MAAM,mBAAmB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;IAC7D,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAElD,IAAI,MAAiB,CAAC;IACtB,IAAI,MAAc,CAAC;IACnB,IAAI,aAAkC,CAAC;IAEvC,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,KAAK,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,GAAG,CAAC;QAEjC,MAAM,WAAW,EAAE,CAAC;QACpB,UAAU,EAAE,CAAC;QAEb,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACnD,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC;QACvC,MAAM,QAAQ,CAAC,YAAY,EAAE,CAAC;QAE9B,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC7D,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAE5B,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACpE,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACzD,aAAa,CAAC,MAAM,CAAC,CAAC;QAEtB,MAAM,CAAC,eAAe,EAAE,eAAe,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChF,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACxE,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,aAAa,EAAE,EAAE,CAAC;QACxB,WAAW,EAAE,CAAC;QAEd,IAAI,mBAAmB,KAAK,SAAS,EAAE,CAAC;YACtC,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;QAC1C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,mBAAmB,CAAC;QACzD,CAAC;QAED,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;YACjC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,cAAc,CAAC;QAC9C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAE7D,MAAM,MAAM,GAAG,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC;QAC1C,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACpC,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,KAAK;aACjB;SACF,CAAC,CAAkB,CAAC;QAErB,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAQ,CAAC;QAE7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YAC5B,MAAM,EAAE,uBAAuB;YAC/B,GAAG,EAAE,4BAA4B;YACjC,KAAK,EAAE,aAAa;YACpB,OAAO,EAAE,cAAc;SACxB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAE7D,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACpC,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,IAAI,EAAE,oBAAoB;gBAC1B,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAkB,CAAC;QAErB,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAQ,CAAC;QAE7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC;YAC5B,OAAO,EAAE,IAAI;YACb,EAAE,EAAE,4BAA4B;YAChC,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAC9B,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;YAC/C,GAAG,EAAE,4BAA4B;YACjC,OAAO,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE;SACxC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,WAAW,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YACzC,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAkB,CAAC;QACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QAExC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,CAAC,MAAM,MAAM,CAAC,QAAQ,CAAC;YAC1C,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE;gBACT,GAAG,EAAE,4BAA4B;gBACjC,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,IAAI;aAChB;SACF,CAAC,CAAkB,CAAC;QACrB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;QAErC,MAAM,OAAO,GAAG,aAAa,CAAC,YAAY,CAAQ,CAAC;QACnD,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lock-file.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/lock-file.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
import fs from "node:fs";
|
|
2
|
+
import os from "node:os";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import { afterEach, beforeEach, describe, expect, it } from "vitest";
|
|
5
|
+
import { acquireWhatsAppLock, releaseWhatsAppLock } from "../lock.js";
|
|
6
|
+
describe("lock file", () => {
|
|
7
|
+
let tmpDir;
|
|
8
|
+
let lockFile;
|
|
9
|
+
beforeEach(() => {
|
|
10
|
+
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "wa-lock-test-"));
|
|
11
|
+
lockFile = path.join(tmpDir, ".whatsapp.lock");
|
|
12
|
+
});
|
|
13
|
+
afterEach(() => {
|
|
14
|
+
// Clean up: release if owned, then remove temp dir
|
|
15
|
+
try {
|
|
16
|
+
releaseWhatsAppLock(lockFile);
|
|
17
|
+
}
|
|
18
|
+
catch { }
|
|
19
|
+
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
20
|
+
});
|
|
21
|
+
it("acquires lock on first call and writes current PID", () => {
|
|
22
|
+
expect(acquireWhatsAppLock(lockFile)).toBe(true);
|
|
23
|
+
expect(fs.existsSync(lockFile)).toBe(true);
|
|
24
|
+
const pid = parseInt(fs.readFileSync(lockFile, "utf-8").trim(), 10);
|
|
25
|
+
expect(pid).toBe(process.pid);
|
|
26
|
+
});
|
|
27
|
+
it("two rapid calls — exactly one succeeds", () => {
|
|
28
|
+
const first = acquireWhatsAppLock(lockFile);
|
|
29
|
+
const second = acquireWhatsAppLock(lockFile);
|
|
30
|
+
expect(first).toBe(true);
|
|
31
|
+
expect(second).toBe(false);
|
|
32
|
+
});
|
|
33
|
+
it("stale lock with dead PID — new process takes over", () => {
|
|
34
|
+
fs.mkdirSync(path.dirname(lockFile), { recursive: true });
|
|
35
|
+
fs.writeFileSync(lockFile, "99999999");
|
|
36
|
+
const result = acquireWhatsAppLock(lockFile);
|
|
37
|
+
expect(result).toBe(true);
|
|
38
|
+
const pid = parseInt(fs.readFileSync(lockFile, "utf-8").trim(), 10);
|
|
39
|
+
expect(pid).toBe(process.pid);
|
|
40
|
+
});
|
|
41
|
+
it("release removes the lock file when owned by current PID", () => {
|
|
42
|
+
acquireWhatsAppLock(lockFile);
|
|
43
|
+
expect(fs.existsSync(lockFile)).toBe(true);
|
|
44
|
+
releaseWhatsAppLock(lockFile);
|
|
45
|
+
expect(fs.existsSync(lockFile)).toBe(false);
|
|
46
|
+
});
|
|
47
|
+
it("release is safe when no lock file exists", () => {
|
|
48
|
+
expect(() => releaseWhatsAppLock(lockFile)).not.toThrow();
|
|
49
|
+
});
|
|
50
|
+
it("release does not remove lock owned by different PID", () => {
|
|
51
|
+
fs.mkdirSync(path.dirname(lockFile), { recursive: true });
|
|
52
|
+
fs.writeFileSync(lockFile, "99999999");
|
|
53
|
+
releaseWhatsAppLock(lockFile);
|
|
54
|
+
// File should still exist — not our PID
|
|
55
|
+
expect(fs.existsSync(lockFile)).toBe(true);
|
|
56
|
+
});
|
|
57
|
+
it("creates parent directories if they don't exist", () => {
|
|
58
|
+
const nested = path.join(tmpDir, "deep", "nested", ".whatsapp.lock");
|
|
59
|
+
expect(acquireWhatsAppLock(nested)).toBe(true);
|
|
60
|
+
expect(fs.existsSync(nested)).toBe(true);
|
|
61
|
+
});
|
|
62
|
+
});
|
|
63
|
+
//# sourceMappingURL=lock-file.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lock-file.test.js","sourceRoot":"","sources":["../../src/__tests__/lock-file.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAEtE,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,MAAc,CAAC;IACnB,IAAI,QAAgB,CAAC;IAErB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC,CAAC,CAAC;QACjE,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,mDAAmD;QACnD,IAAI,CAAC;YACH,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QACV,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,GAAG,GAAG,QAAQ,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEvC,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,GAAG,GAAG,QAAQ,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9B,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9B,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,GAAG,EAAE,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACvC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9B,wCAAwC;QACxC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACrE,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"medium-fixes.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/medium-fixes.test.ts"],"names":[],"mappings":""}
|