whatsapp-pi 1.0.9 → 1.0.11

package/README.md

@@ -1,98 +1,98 @@
-<p align="center">
-  <img src="https://upload.wikimedia.org/wikipedia/commons/6/6b/WhatsApp.svg" alt="WhatsApp Logo" width="100">
-</p>
-
-# WhatsApp-Pi
-
-A WhatsApp integration extension for the **[Pi Coding Agent](https://github.com/mariozechner/pi-coding-agent)**. 
-
-Pi is a powerful agentic AI coding assistant that operates in your terminal. This extension allows you to chat and pair-program with your Pi agent directly through WhatsApp, featuring message filtering, allow-listing, and reliable message delivery.
-
-## Features
-
-- **Manual WhatsApp Connection**: QR code-based authentication with session persistence
-- **Allow List**: Control which numbers can interact with Pi
-  - Add contacts with optional names for easy identification
-  - View ignored numbers (not in allow list) and add them when needed
-- **Reliable Messaging**: Queue-based message sending with retry logic
-- **TUI Integration**: Menu-driven interface for managing connections and contacts
-
-## Quick Start
-
-1. Install the extension:
-```bash
-pi install npm:whatsapp-pi
-```
-
-2. Start Pi (the extension will load automatically once installed):
-```bash
-pi
-```
-
-To automatically connect to WhatsApp on startup (if you are already authenticated):
-```bash
-pi -w
-# or
-pi --whatsapp
-```
-
-3. Use the menu to connect WhatsApp and manage allowed/blocked numbers
-
-## Development / Testing
-
-If you are developing or testing the extension locally:
-
-1. Install dependencies:
-```bash
-npm install
-```
-
-2. Run the extension:
-```bash
-pi -e whatsapp-pi.ts
-```
-
-For verbose mode (shows Baileys trace logs for debugging):
-```bash
-pi -e whatsapp-pi.ts -v
-# or
-pi -e whatsapp-pi.ts --verbose
-```
-
-## Commands
-
-- `/whatsapp` - Open the WhatsApp management menu
-  - **Allow Numbers**: Manage contacts that can interact with Pi
-  - **Blocked Numbers**: View ignored numbers (not in allow list) and add them to allow list
-
-## Project Structure
-
-```
-src/
-├── models/          # Type definitions
-├── services/        # Core services (WhatsApp, Session, MessageSender)
-└── ui/              # Menu handlers
-
-specs/               # Feature specifications
-tests/               # Unit and integration tests
-```
-
-## Documentation
-
-See `specs/` directory for detailed feature documentation:
-- `001-whatsapp-tui-integration/` - TUI menu system
-- `002-manual-whatsapp-connection/` - Connection management
-- `003-whatsapp-messaging-refactor/` - Reliable messaging
-- `004-blocked-numbers-management/` - Block list feature
-
-## Development
-
-Run tests:
-```bash
-npm test
-```
-
-Lint:
-```bash
-npm run lint
-```
+<p align="center">
+  <img src="https://upload.wikimedia.org/wikipedia/commons/6/6b/WhatsApp.svg" alt="WhatsApp Logo" width="100">
+</p>
+
+# WhatsApp-Pi
+
+A WhatsApp integration extension for the **[Pi Coding Agent](https://github.com/mariozechner/pi-coding-agent)**. 
+
+Pi is a powerful agentic AI coding assistant that operates in your terminal. This extension allows you to chat and pair-program with your Pi agent directly through WhatsApp, featuring message filtering, allow-listing, and reliable message delivery.
+
+## Features
+
+- **Manual WhatsApp Connection**: QR code-based authentication with session persistence
+- **Allow List**: Control which numbers can interact with Pi
+  - Add contacts with optional names for easy identification
+  - View ignored numbers (not in allow list) and add them when needed
+- **Reliable Messaging**: Queue-based message sending with retry logic
+- **TUI Integration**: Menu-driven interface for managing connections and contacts
+
+## Quick Start
+
+1. Install the extension:
+```bash
+pi install npm:whatsapp-pi
+```
+
+2. Start Pi (the extension will load automatically once installed):
+```bash
+pi
+```
+
+To automatically connect to WhatsApp on startup (if you are already authenticated):
+```bash
+pi -w
+# or
+pi --whatsapp
+```
+
+3. Use the menu to connect WhatsApp and manage allowed/blocked numbers
+
+## Development / Testing
+
+If you are developing or testing the extension locally:
+
+1. Install dependencies:
+```bash
+npm install
+```
+
+2. Run the extension:
+```bash
+pi -e whatsapp-pi.ts
+```
+
+For verbose mode (shows Baileys trace logs for debugging):
+```bash
+pi -e whatsapp-pi.ts -v
+# or
+pi -e whatsapp-pi.ts --verbose
+```
+
+## Commands
+
+- `/whatsapp` - Open the WhatsApp management menu
+  - **Allow Numbers**: Manage contacts that can interact with Pi
+  - **Blocked Numbers**: View ignored numbers (not in allow list) and add them to allow list
+
+## Project Structure
+
+```
+src/
+├── models/          # Type definitions
+├── services/        # Core services (WhatsApp, Session, MessageSender)
+└── ui/              # Menu handlers
+
+specs/               # Feature specifications
+tests/               # Unit and integration tests
+```
+
+## Documentation
+
+See `specs/` directory for detailed feature documentation:
+- `001-whatsapp-tui-integration/` - TUI menu system
+- `002-manual-whatsapp-connection/` - Connection management
+- `003-whatsapp-messaging-refactor/` - Reliable messaging
+- `004-blocked-numbers-management/` - Block list feature
+
+## Development
+
+Run tests:
+```bash
+npm test
+```
+
+Lint:
+```bash
+npm run lint
+```

package/package.json

@@ -1,45 +1,45 @@
-{
-  "name": "whatsapp-pi",
-  "version": "1.0.9",
-  "type": "module",
-  "description": "WhatsApp integration extension for Pi",
-  "main": "whatsapp-pi.ts",
-  "files": [
-    "whatsapp-pi.ts",
-    "src"
-  ],
-  "keywords": [
-    "pi",
-    "pi-package",
-    "pi-extension",
-    "whatsapp",
-    "baileys",
-    "agent"
-  ],
-  "author": "Rapha",
-  "license": "MIT",
-  "scripts": {
-    "test": "vitest run",
-    "typecheck": "tsc --noEmit"
-  },
-  "dependencies": {
-    "@whiskeysockets/baileys": "^6.11.0",
-    "pino": "^10.3.1",
-    "qrcode-terminal": "^0.12.0"
-  },
-  "devDependencies": {
-    "@mariozechner/pi-coding-agent": "latest",
-    "@types/node": "^20.11.0",
-    "@types/qrcode-terminal": "^0.12.2",
-    "ts-node": "^10.9.2",
-    "tsx": "^4.7.0",
-    "typescript": "^5.3.0",
-    "vitest": "^1.2.0"
-  },
-  "pi": {
-    "extensions": [
-      "./whatsapp-pi.ts"
-    ],
-    "image": "https://upload.wikimedia.org/wikipedia/commons/thumb/6/6b/WhatsApp.svg/512px-WhatsApp.svg.png"
-  }
-}
+{
+  "name": "whatsapp-pi",
+  "version": "1.0.11",
+  "type": "module",
+  "description": "WhatsApp integration extension for Pi",
+  "main": "whatsapp-pi.ts",
+  "files": [
+    "whatsapp-pi.ts",
+    "src"
+  ],
+  "keywords": [
+    "pi",
+    "pi-package",
+    "pi-extension",
+    "whatsapp",
+    "baileys",
+    "agent"
+  ],
+  "author": "Rapha",
+  "license": "MIT",
+  "scripts": {
+    "test": "vitest run",
+    "typecheck": "tsc --noEmit"
+  },
+  "dependencies": {
+    "@whiskeysockets/baileys": "^6.11.0",
+    "pino": "^10.3.1",
+    "qrcode-terminal": "^0.12.0"
+  },
+  "devDependencies": {
+    "@mariozechner/pi-coding-agent": "latest",
+    "@types/node": "^20.11.0",
+    "@types/qrcode-terminal": "^0.12.2",
+    "ts-node": "^10.9.2",
+    "tsx": "^4.7.0",
+    "typescript": "^5.3.0",
+    "vitest": "^1.2.0"
+  },
+  "pi": {
+    "extensions": [
+      "./whatsapp-pi.ts"
+    ],
+    "image": "https://upload.wikimedia.org/wikipedia/commons/thumb/6/6b/WhatsApp.svg/512px-WhatsApp.svg.png"
+  }
+}

package/src/models/whatsapp.types.ts

@@ -1,46 +1,46 @@
-export type SessionStatus = 'logged-out' | 'pairing' | 'connected' | 'disconnected';
-
-export interface WhatsAppSession {
-    id: string;
-    status: SessionStatus;
-    credentialsPath: string;
-}
-
-export interface AllowList {
-    numbers: string[];
-}
-
-export interface IncomingMessage {
-    id: string;
-    remoteJid: string;
-    pushName?: string;
-    text?: string;
-    timestamp: number;
-}
-
-export interface MessageRequest {
-    recipientJid: string;
-    text: string;
-    options?: {
-        maxRetries?: number;
-        priority?: 'high' | 'normal';
-    };
-}
-
-export interface MessageResult {
-    success: boolean;
-    messageId?: string;
-    error?: string;
-    attempts: number;
-}
-
-export class WhatsAppError extends Error {
-    constructor(public code: string, message: string) {
-        super(message);
-        this.name = 'WhatsAppError';
-    }
-}
-
-export function validatePhoneNumber(number: string): boolean {
-    return /^\+[1-9]\d{1,14}$/.test(number);
-}
+export type SessionStatus = 'logged-out' | 'pairing' | 'connected' | 'disconnected';
+
+export interface WhatsAppSession {
+    id: string;
+    status: SessionStatus;
+    credentialsPath: string;
+}
+
+export interface AllowList {
+    numbers: string[];
+}
+
+export interface IncomingMessage {
+    id: string;
+    remoteJid: string;
+    pushName?: string;
+    text?: string;
+    timestamp: number;
+}
+
+export interface MessageRequest {
+    recipientJid: string;
+    text: string;
+    options?: {
+        maxRetries?: number;
+        priority?: 'high' | 'normal';
+    };
+}
+
+export interface MessageResult {
+    success: boolean;
+    messageId?: string;
+    error?: string;
+    attempts: number;
+}
+
+export class WhatsAppError extends Error {
+    constructor(public code: string, message: string) {
+        super(message);
+        this.name = 'WhatsAppError';
+    }
+}
+
+export function validatePhoneNumber(number: string): boolean {
+    return /^\+[1-9]\d{1,14}$/.test(number);
+}

package/src/services/audio.service.ts

@@ -1,53 +1,53 @@
-import { downloadContentFromMessage } from '@whiskeysockets/baileys';
-import { exec } from 'node:child_process';
-import { promisify } from 'node:util';
-import { writeFile, mkdir } from 'node:fs/promises';
-import { join } from 'node:path';
-import { existsSync } from 'node:fs';
-
-const execAsync = promisify(exec);
-
-export class AudioService {
-    private readonly mediaDir = '/home/opc/.pi/whatsapp-medias';
-    private readonly whisperPath = '/home/opc/.local/bin/whisper';
-
-    constructor() {
-        if (!existsSync(this.mediaDir)) {
-            mkdir(this.mediaDir, { recursive: true }).catch(() => {});
-        }
-    }
-
-    async transcribe(audioMessage: any): Promise<string> {
-        try {
-            const filename = `audio_${Date.now()}`;
-            const inputPath = join(this.mediaDir, `${filename}.ogg`);
-
-            // Download audio content
-            const stream = await downloadContentFromMessage(audioMessage, 'audio');
-            let buffer = Buffer.from([]);
-            for await (const chunk of stream) {
-                buffer = Buffer.concat([buffer, chunk]);
-            }
-
-            await writeFile(inputPath, buffer);
-
-            // Transcribe using Whisper
-            // Using small model for better accuracy
-            const command = `${this.whisperPath} "${inputPath}" --model small --language pt --output_format txt --output_dir "${this.mediaDir}" --fp16 False`;
-            
-            await execAsync(command);
-
-            const txtPath = join(this.mediaDir, `${filename}.txt`);
-            if (existsSync(txtPath)) {
-                const fs = await import('node:fs/promises');
-                const text = await fs.readFile(txtPath, 'utf8');
-                return text.trim();
-            }
-
-            return '[Transcrição vazia]';
-        } catch (error) {
-            console.error('[AudioService] Transcription error:', error);
-            return `[Erro na transcrição: ${error instanceof Error ? error.message : String(error)}]`;
-        }
-    }
-}
+import { downloadContentFromMessage } from '@whiskeysockets/baileys';
+import { exec } from 'node:child_process';
+import { promisify } from 'node:util';
+import { writeFile, mkdir } from 'node:fs/promises';
+import { join } from 'node:path';
+import { existsSync } from 'node:fs';
+
+const execAsync = promisify(exec);
+
+export class AudioService {
+    private readonly mediaDir = '/home/opc/.pi/whatsapp-medias';
+    private readonly whisperPath = '/home/opc/.local/bin/whisper';
+
+    constructor() {
+        if (!existsSync(this.mediaDir)) {
+            mkdir(this.mediaDir, { recursive: true }).catch(() => {});
+        }
+    }
+
+    async transcribe(audioMessage: any): Promise<string> {
+        try {
+            const filename = `audio_${Date.now()}`;
+            const inputPath = join(this.mediaDir, `${filename}.ogg`);
+
+            // Download audio content
+            const stream = await downloadContentFromMessage(audioMessage, 'audio');
+            let buffer = Buffer.from([]);
+            for await (const chunk of stream) {
+                buffer = Buffer.concat([buffer, chunk]);
+            }
+
+            await writeFile(inputPath, buffer);
+
+            // Transcribe using Whisper
+            // Using small model for better accuracy
+            const command = `${this.whisperPath} "${inputPath}" --model small --language pt --output_format txt --output_dir "${this.mediaDir}" --fp16 False`;
+            
+            await execAsync(command);
+
+            const txtPath = join(this.mediaDir, `${filename}.txt`);
+            if (existsSync(txtPath)) {
+                const fs = await import('node:fs/promises');
+                const text = await fs.readFile(txtPath, 'utf8');
+                return text.trim();
+            }
+
+            return '[Transcrição vazia]';
+        } catch (error) {
+            console.error('[AudioService] Transcription error:', error);
+            return `[Erro na transcrição: ${error instanceof Error ? error.message : String(error)}]`;
+        }
+    }
+}

package/src/services/message.sender.ts

@@ -1,93 +1,93 @@
-import { WhatsAppService } from './whatsapp.service.js';
-import { MessageRequest, MessageResult, WhatsAppError } from '../models/whatsapp.types.js';
-
-export class MessageSender {
-    private whatsappService: WhatsAppService;
-
-    constructor(whatsappService: WhatsAppService) {
-        this.whatsappService = whatsappService;
-    }
-
-    /**
-     * Pauses execution for the specified time.
-     * @param ms Milliseconds to sleep.
-     */
-    private async sleep(ms: number) {
-        return new Promise(resolve => setTimeout(resolve, ms));
-    }
-
-    /**
-     * Waits for the WhatsApp connection to be active.
-     * @param timeoutMs Maximum time to wait in milliseconds.
-     * @throws {WhatsAppError} If connection is not established within timeout.
-     */
-    private async waitIfOffline(timeoutMs: number = 30000): Promise<void> {
-        const start = Date.now();
-        while (this.whatsappService.getStatus() !== 'connected') {
-            if (Date.now() - start > timeoutMs) {
-                throw new WhatsAppError('TIMEOUT', 'Timed out waiting for WhatsApp connection');
-            }
-            await this.sleep(1000);
-        }
-    }
-
-    /**
-     * Sends a message with retry logic and connection awareness.
-     * @param request The message recipient and content.
-     * @returns Promise resolving to a result object indicating success or failure.
-     */
-    public async send(request: MessageRequest): Promise<MessageResult> {
-        const maxRetries = request.options?.maxRetries ?? 3;
-        let attempts = 0;
-        let lastError: any = null;
-
-        while (attempts < maxRetries) {
-            attempts++;
-            try {
-                // 1. Ensure we are online
-                await this.waitIfOffline();
-                
-                // 2. Get active socket
-                const socket = this.whatsappService.getSocket();
-                if (!socket) {
-                    throw new WhatsAppError('SOCKET_NOT_INIT', 'WhatsApp socket not initialized');
-                }
-
-                // 3. Send the message
-                // Note: Branding π is applied here to ensure consistency
-                const response = await socket.sendMessage(request.recipientJid, { 
-                    text: `${request.text} π` 
-                });
-
-                return {
-                    success: true,
-                    messageId: response.key.id,
-                    attempts
-                };
-            } catch (error: any) {
-                lastError = error;
-                console.error(`[MessageSender] Attempt ${attempts} failed for ${request.recipientJid}: ${error.message}`);
-                
-                // Specific handling for non-retryable errors
-                if (error.code === 'TIMEOUT') {
-                    break;
-                }
-
-                // 4. Backoff before retry
-                if (attempts < maxRetries) {
-                    const backoff = Math.pow(2, attempts) * 1000;
-                    if (this.whatsappService.isVerbose()) {
-                        console.log(`[MessageSender] Retrying in ${backoff}ms...`);
-                    }
-                    await this.sleep(backoff);
-                }
-            }
-        }
-
-        return {
-            success: false,
-            error: lastError?.message || 'Unknown error',
-            attempts
-        };
-    }
-}
+import { WhatsAppService } from './whatsapp.service.js';
+import { MessageRequest, MessageResult, WhatsAppError } from '../models/whatsapp.types.js';
+
+export class MessageSender {
+    private whatsappService: WhatsAppService;
+
+    constructor(whatsappService: WhatsAppService) {
+        this.whatsappService = whatsappService;
+    }
+
+    /**
+     * Pauses execution for the specified time.
+     * @param ms Milliseconds to sleep.
+     */
+    private async sleep(ms: number) {
+        return new Promise(resolve => setTimeout(resolve, ms));
+    }
+
+    /**
+     * Waits for the WhatsApp connection to be active.
+     * @param timeoutMs Maximum time to wait in milliseconds.
+     * @throws {WhatsAppError} If connection is not established within timeout.
+     */
+    private async waitIfOffline(timeoutMs: number = 30000): Promise<void> {
+        const start = Date.now();
+        while (this.whatsappService.getStatus() !== 'connected') {
+            if (Date.now() - start > timeoutMs) {
+                throw new WhatsAppError('TIMEOUT', 'Timed out waiting for WhatsApp connection');
+            }
+            await this.sleep(1000);
+        }
+    }
+
+    /**
+     * Sends a message with retry logic and connection awareness.
+     * @param request The message recipient and content.
+     * @returns Promise resolving to a result object indicating success or failure.
+     */
+    public async send(request: MessageRequest): Promise<MessageResult> {
+        const maxRetries = request.options?.maxRetries ?? 3;
+        let attempts = 0;
+        let lastError: any = null;
+
+        while (attempts < maxRetries) {
+            attempts++;
+            try {
+                // 1. Ensure we are online
+                await this.waitIfOffline();
+                
+                // 2. Get active socket
+                const socket = this.whatsappService.getSocket();
+                if (!socket) {
+                    throw new WhatsAppError('SOCKET_NOT_INIT', 'WhatsApp socket not initialized');
+                }
+
+                // 3. Send the message
+                // Note: Branding π is applied here to ensure consistency
+                const response = await socket.sendMessage(request.recipientJid, { 
+                    text: `${request.text} π` 
+                });
+
+                return {
+                    success: true,
+                    messageId: response.key.id,
+                    attempts
+                };
+            } catch (error: any) {
+                lastError = error;
+                console.error(`[MessageSender] Attempt ${attempts} failed for ${request.recipientJid}: ${error.message}`);
+                
+                // Specific handling for non-retryable errors
+                if (error.code === 'TIMEOUT') {
+                    break;
+                }
+
+                // 4. Backoff before retry
+                if (attempts < maxRetries) {
+                    const backoff = Math.pow(2, attempts) * 1000;
+                    if (this.whatsappService.isVerbose()) {
+                        console.log(`[MessageSender] Retrying in ${backoff}ms...`);
+                    }
+                    await this.sleep(backoff);
+                }
+            }
+        }
+
+        return {
+            success: false,
+            error: lastError?.message || 'Unknown error',
+            attempts
+        };
+    }
+}