whatsapp-pi 1.0.15 → 1.0.18

package/README.md

@@ -6,6 +6,8 @@
 
 A WhatsApp integration extension for the **[Pi Coding Agent](https://github.com/mariozechner/pi-coding-agent)**. 
 
+[![GitHub](https://img.shields.io/badge/github-repo-black.svg?style=flat-square&logo=github)](https://github.com/RaphaCastelloes/whatsapp-pi)
+
 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
@@ -40,10 +42,12 @@ pi --whatsapp
 
 ## Development / Testing
 
-If you are developing or testing the extension locally:
+If you are developing or testing the extension locally, you can clone the repository from [GitHub](https://github.com/RaphaCastelloes/whatsapp-pi):
 
-1. Install dependencies:
+1. Clone and install dependencies:
 ```bash
+git clone https://github.com/RaphaCastelloes/whatsapp-pi.git
+cd whatsapp-pi
 npm install
 ```
 

package/package.json

@@ -1,9 +1,17 @@
 {
   "name": "whatsapp-pi",
-  "version": "1.0.15",
+  "version": "1.0.18",
   "type": "module",
   "description": "WhatsApp integration extension for Pi",
   "main": "whatsapp-pi.ts",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/RaphaCastelloes/whatsapp-pi.git"
+  },
+  "bugs": {
+    "url": "https://github.com/RaphaCastelloes/whatsapp-pi/issues"
+  },
+  "homepage": "https://github.com/RaphaCastelloes/whatsapp-pi#readme",
   "files": [
     "whatsapp-pi.ts",
     "src"

package/src/services/session.manager.ts

@@ -21,6 +21,8 @@ export class SessionManager {
     private allowList: Contact[] = [];
     private blockList: Contact[] = [];
     private ignoredNumbers: Contact[] = [];
+    private openaiKey: string = '';
+    private visionModel: string = 'gpt-4o';
 
     public async ensureInitialized() {
         try {
@@ -54,6 +56,8 @@ export class SessionManager {
             this.blockList = (config.blockList || []).map(cleanContact).filter(Boolean) as Contact[];
             this.ignoredNumbers = (config.ignoredNumbers || []).map(cleanContact).filter(Boolean) as Contact[];
             this.status = config.status || 'logged-out';
+            this.openaiKey = config.openaiKey || '';
+            this.visionModel = config.visionModel || 'gpt-4o';
         } catch (error) {
             // File not found is fine
         }
@@ -65,7 +69,9 @@ export class SessionManager {
                 allowList: this.allowList,
                 blockList: this.blockList,
                 ignoredNumbers: this.ignoredNumbers,
-                status: this.status
+                status: this.status,
+                openaiKey: this.openaiKey,
+                visionModel: this.visionModel
             };
             await writeFile(this.configPath, JSON.stringify(config, null, 2));
         } catch (error) {
@@ -195,6 +201,24 @@ export class SessionManager {
         await this.saveConfig();
     }
 
+    getOpenaiKey(): string {
+        return this.openaiKey;
+    }
+
+    async setOpenaiKey(key: string) {
+        this.openaiKey = key;
+        await this.saveConfig();
+    }
+
+    getVisionModel(): string {
+        return this.visionModel;
+    }
+
+    async setVisionModel(model: string) {
+        this.visionModel = model;
+        await this.saveConfig();
+    }
+
     getAuthDir(): string {
         return this.authDir;
     }

package/whatsapp-pi.ts

@@ -120,13 +120,38 @@ export default function (pi: ExtensionAPI) {
         }
 
         // Handle media types
+        let imageBuffer: Buffer | undefined;
+        let imageMimeType: string | undefined;
+
         if (msg.message.audioMessage) {
             console.log(`[WhatsApp-Pi] Transcribing audio from ${pushName}...`);
             const transcription = await audioService.transcribe(msg.message.audioMessage);
             text = `[Áudio Transcrito]: ${transcription}`;
+        } else if (msg.message.imageMessage) {
+            console.log(`[WhatsApp-Pi] Downloading image from ${pushName}...`);
+            try {
+                const { downloadContentFromMessage } = await import('@whiskeysockets/baileys');
+                const stream = await downloadContentFromMessage(msg.message.imageMessage, 'image');
+                let buffer = Buffer.from([]);
+                for await (const chunk of stream) {
+                    buffer = Buffer.concat([buffer, chunk]);
+                }
+                imageBuffer = buffer;
+                
+                // Normalize mime type for Cloud Code Assist / Gemini
+                let rawMime = msg.message.imageMessage.mimetype || 'image/jpeg';
+                imageMimeType = rawMime.toLowerCase().split(';')[0].trim();
+                if (imageMimeType === 'image/jpg') imageMimeType = 'image/jpeg';
+                
+                console.log(`[WhatsApp-Pi] Image downloaded. MIME: ${imageMimeType} (original: ${rawMime}), Size: ${imageBuffer.length} bytes`);
+                
+                text = msg.message.imageMessage.caption || "[Image]";
+            } catch (e) {
+                console.error(`[WhatsApp-Pi] Failed to download image:`, e);
+                text = "[Image (download failed)]";
+            }
         } else if (!text) {
-            if (msg.message.imageMessage) text = "[Image]";
-            else if (msg.message.videoMessage) text = "[Video]";
+            if (msg.message.videoMessage) text = "[Video]";
             else if (msg.message.stickerMessage) text = "[Sticker]";
             else if (msg.message.documentMessage) text = "[Document]";
             else if (msg.message.contactMessage || msg.message.contactsArrayMessage) text = "[Contact]";
@@ -137,6 +162,16 @@ export default function (pi: ExtensionAPI) {
         // Always log to console so it appears in the TUI log pane
         console.log(`[WhatsApp-Pi] ${pushName} (+${sender}): ${text}`);
 
+        // Use a standard delivery for ALL messages to ensure TUI consistency
+        if (imageBuffer && imageMimeType) {
+            pi.sendUserMessage([
+                { type: "text", text: `Mensagem de ${pushName} (+${sender}): ${text}` },
+                { type: "image", data: imageBuffer.toString('base64'), mimeType: imageMimeType }
+            ], { deliverAs: "followUp" });
+        } else {
+            pi.sendUserMessage(`Mensagem de ${pushName} (+${sender}): ${text}`, { deliverAs: "followUp" });
+        }
+
         // Handle commands
         if (text.trim().toLowerCase().startsWith('/compact')) {
             console.log(`[WhatsApp-Pi] Session compact requested by ${pushName}.`);
@@ -157,8 +192,7 @@ export default function (pi: ExtensionAPI) {
             return;
         }
 
-        // Use a standard delivery for ALL messages to ensure TUI consistency
-        pi.sendUserMessage(`Mensagem de ${pushName} (+${sender}): ${text}`, { deliverAs: "followUp" });
+        
     });
 
     // Register commands