Add HelloWorld debug agent and per-chat handler

2025-06-04 21:51:05 -06:00
parent dec4922d42
commit 8d2e8f4854
14 changed files with 582 additions and 35 deletions
--- a/whatsapp-router/src/index.ts
+++ b/whatsapp-router/src/index.ts
@@ -0,0 +1,149 @@
+import express from 'express';
+import axios from 'axios';
+import { WhatsAppMessage } from './types';
+import { getHandler, Handler } from './chatHandlers';
+
+const app = express();
+
+const port = Number(process.env.PORT) || 3001;
+const agentUrl = process.env.LLM_AGENT_URL as string | undefined;
+const openWaUrl = process.env.OPEN_WA_URL as string | undefined;
+
+const config = {
+  API_URL: openWaUrl || '',
+  MAX_ATTEMPTS: parseInt(process.env.MAX_ATTEMPTS || '100', 10),
+  RETRY_MS: parseInt(process.env.RETRY_MS || '2000', 10)
+};
+
+function log(level: keyof Console | 'info' | 'warn' | 'error' | 'debug', ...args: unknown[]) {
+  const logger = (console as any)[level] as ((...args: unknown[]) => void) | undefined;
+  if (logger) logger(...args);
+  else console.log(...args);
+}
+
+async function waitForGateway() {
+  for (let i = 1; i <= config.MAX_ATTEMPTS; i++) {
+    try {
+      await axios.get(`${config.API_URL}/api-docs`);
+      log('info', '🟢  nucleo-whatsapp ready');
+      return;
+    } catch {
+      log('warn', `Gateway not responding (attempt ${i}/${config.MAX_ATTEMPTS})…`);
+      await new Promise(r => setTimeout(r, config.RETRY_MS));
+    }
+  }
+  throw new Error('nucleo-whatsapp did not respond in time');
+}
+
+async function clearWebhooks() {
+  try {
+    const { data } = await axios.post(`${config.API_URL}/listWebhooks`);
+    const hooks = data?.response || [];
+    if (!hooks.length) {
+      log('info', 'No existing webhooks to remove');
+      return;
+    }
+
+    log('info', `Removing ${hooks.length} webhooks…`);
+    const results = await Promise.allSettled(
+      hooks.map((h: any) => axios.post(`${config.API_URL}/removeWebhook`, { args: { webhookId: h.id } }))
+    );
+
+    results.forEach((r: PromiseSettledResult<any>, i: number) => {
+      const id = hooks[i].id;
+      if (r.status === 'fulfilled' && r.value?.data?.response === true) {
+        log('debug', `✔️  Removed webhook ${id}`);
+      } else {
+        log('warn', `⚠️  Failed to remove webhook ${id}`);
+      }
+    });
+
+    const ok = results.filter((r: PromiseSettledResult<any>) => r.status === 'fulfilled' && (r as PromiseFulfilledResult<any>).value?.data?.response === true).length;
+    log('info', `Cleanup OK (${ok}/${hooks.length} removed)`);
+  } catch (e: any) {
+    log('error', 'Failed cleaning webhooks:', e.response?.data || e.message);
+  }
+}
+
+async function registerWebhook() {
+  const url = `http://whatsapp-router:${port}/webhook`;
+  const eventConfig = {
+    onAck: false,
+    onAddedToGroup: true,
+    onAnyMessage: true,
+    onBattery: true,
+    onBroadcast: true,
+    onButton: true,
+    onCallState: false,
+    onChatDeleted: true,
+    onChatOpened: true,
+    onChatState: true,
+    onContactAdded: true,
+    onGlobalParticipantsChanged: true,
+    onGroupApprovalRequest: true,
+    onGroupChange: true,
+    onIncomingCall: false,
+    onLabel: true,
+    onLogout: true,
+    onMessage: false,
+    onMessageDeleted: true,
+    onNewProduct: true,
+    onOrder: true,
+    onPlugged: false,
+    onPollVote: true,
+    onReaction: true,
+    onRemovedFromGroup: false,
+    onStateChanged: false,
+    onStory: false,
+  };
+
+  const events = Object.entries(eventConfig)
+    .filter(([_, enabled]) => enabled)
+    .map(([event]) => event);
+
+  const { data } = await axios.post(`${config.API_URL}/registerWebhook`, {
+    args: { url, events }
+  });
+
+  log('info', '✔️  Webhook registered:', data);
+}
+
+app.use(express.json());
+
+app.post('/webhook', async (req: express.Request, res: express.Response) => {
+  const { message, text, from } = req.body as {
+    message?: WhatsAppMessage;
+    text?: string;
+    from?: string;
+  };
+  const incoming = message || text;
+  try {
+    if (!incoming) return res.sendStatus(200);
+    if (!openWaUrl) throw new Error('Service URLs not configured');
+    const chatId = (message && message.chatId) || from;
+    const handler = getHandler(chatId, agentUrl);
+    if (!handler) throw new Error('No handler configured');
+    let reply: string;
+    if (typeof handler === 'string') {
+      const agentRes = await axios.post(handler, { message: incoming });
+      reply = agentRes.data.reply || agentRes.data;
+    } else {
+      reply = await handler(incoming);
+    }
+    await axios.post(`${openWaUrl}/sendText`, { args: { to: from, content: reply } });
+  } catch (err: any) {
+    console.error('Error processing message', err.message);
+  }
+  res.sendStatus(200);
+});
+
+app.listen(port, async () => {
+  console.log(`WhatsApp router listening on ${port}`);
+  try {
+    await waitForGateway();
+    await clearWebhooks();
+    await registerWebhook();
+  } catch (err: any) {
+    log('error', 'Webhook setup failed:', err.message);
+  }
+});