npm - opencode-avatar - Versions diffs - 0.3.3 → 0.3.6 - Mend

opencode-avatar 0.3.3 → 0.3.6

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.

Files changed (3) hide show

package/dist/electron.js CHANGED Viewed

@@ -405,17 +405,20 @@ import * as fs from "fs";
 import * as os from "os";
 import { fileURLToPath } from "url";
 require_main().config();
+var logFile = path.join(os.homedir(), "avatar.log");
+function log(msg) {
+  fs.appendFileSync(logFile, new Date().toISOString() + ": " + msg + `
+`);
+}
 function getConfig() {
   try {
     const configPath = path.join(os.homedir(), ".config", "opencode", "opencode-avatar.json");
     const config = JSON.parse(fs.readFileSync(configPath, "utf8"));
     if (!config.falKey) {
-      console.warn("Warning: falKey not found in config file. Avatar generation will not work. Please set falKey in ~/.config/opencode/opencode-avatar.json");
       return { falKey: null, prompt: null };
     }
     return { falKey: config.falKey, prompt: config.prompt || null };
   } catch (error) {
-    console.warn(`Warning: Failed to read config file: ${error.message}. Avatar generation will not work. Please ensure ~/.config/opencode/opencode-avatar.json exists and contains falKey.`);
     return { falKey: null, prompt: null };
   }
 }
@@ -490,38 +493,35 @@ var HTML_CONTENT = `<!DOCTYPE html>
         ctx.drawImage(srcImg, 0, 0);
-        const topLeftPixel = ctx.getImageData(0, 0, 1, 1).data;
-        const chromaR = topLeftPixel[0];
-        const chromaG = topLeftPixel[1];
-        const chromaB = topLeftPixel[2];
+        // Chroma keying: make background transparent
         const imageData = ctx.getImageData(0, 0, canvas.width, canvas.height);
         const data = imageData.data;
+        const chromaR = data[0]; // first pixel R
+        const chromaG = data[1]; // first pixel G
+        const chromaB = data[2]; // first pixel B
         const tolerance = 30;
         for (let i = 0; i < data.length; i += 4) {
           const r = data[i];
           const g = data[i + 1];
           const b = data[i + 2];
-          if (
-            Math.abs(r - chromaR) <= tolerance &&
-            Math.abs(g - chromaG) <= tolerance &&
-            Math.abs(b - chromaB) <= tolerance
-          ) {
-            data[i + 3] = 0;
+          if (Math.abs(r - chromaR) <= tolerance &&
+              Math.abs(g - chromaG) <= tolerance &&
+              Math.abs(b - chromaB) <= tolerance) {
+            data[i + 3] = 0; // set alpha to 0
           }
         }
         ctx.putImageData(imageData, 0, 0);
         img.src = canvas.toDataURL('image/png');
       };
-      srcImg.onerror = function(e) {
-        console.error('Failed to load image:', e);
-        img.src = 'avatar.svg';
-      };
+       srcImg.onerror = function(e) {
+         console.error('Failed to load image:', e);
+         log('Renderer: Avatar load failed, using fallback transparent image');
+         img.src = 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==';
+       };
     }
     ipcRenderer.on('set-avatar', (event, avatarDataUrl) => {
@@ -690,18 +690,17 @@ function startAvatarServer() {
       req.on("end", () => {
         try {
           const { avatarPath } = JSON.parse(body);
+          log("Set-avatar request with path: " + avatarPath);
           if (mainWindow && avatarPath) {
             const imageBuffer = fs.readFileSync(avatarPath);
             const base64 = imageBuffer.toString("base64");
             const dataUrl = `data:image/png;base64,${base64}`;
             mainWindow.webContents.send("set-avatar", dataUrl);
+            updateTrayIcon(avatarPath);
           }
           res.writeHead(200, { "Content-Type": "application/json" });
           res.end(JSON.stringify({ success: true }));
-        } catch (e) {
-          res.writeHead(400, { "Content-Type": "application/json" });
-          res.end(JSON.stringify({ error: "Invalid JSON" }));
-        }
+        } catch (e) {}
       });
     } else if (req.method === "POST" && req.url === "/generate-avatar") {
       let body = "";
@@ -722,6 +721,7 @@ function startAvatarServer() {
             const base64 = imageBuffer.toString("base64");
             const dataUrl = `data:image/png;base64,${base64}`;
             mainWindow.webContents.send("set-avatar", dataUrl);
+            updateTrayIcon(avatarPath);
           }
           res.writeHead(200, { "Content-Type": "application/json" });
           res.end(JSON.stringify({ success: true, avatarPath }));
@@ -780,21 +780,21 @@ function createWindow() {
   mainWindow.loadURL(`data:text/html;charset=utf-8,${encodeURIComponent(HTML_CONTENT)}`);
   mainWindow.webContents.on("did-finish-load", () => {
     if (mainWindow) {
+      log("Window finished loading");
       const avatarPath = getAvatarPath();
       try {
         const imageBuffer = fs.readFileSync(avatarPath);
         const base64 = imageBuffer.toString("base64");
         const dataUrl = `data:image/png;base64,${base64}`;
         mainWindow.webContents.send("set-avatar", dataUrl);
+        updateTrayIcon(avatarPath);
         setTimeout(() => {
           if (mainWindow && !mainWindow.isVisible()) {
             mainWindow.show();
             mainWindow.setAlwaysOnTop(true, "screen-saver");
           }
         }, 100);
-      } catch (err) {
-        const message = err instanceof Error ? err.message : String(err);
-      }
+      } catch (error) {}
     }
   });
   mainWindow.webContents.on("did-fail-load", (event, errorCode, errorDescription) => {});
@@ -809,10 +809,7 @@ function createTray() {
   let trayIcon;
   try {
     const pngPath = path.join(AVATAR_DIR, "avatar.png");
-    trayIcon = nativeImage.createFromPath(pngPath);
-    if (trayIcon.isEmpty()) {
-      trayIcon = nativeImage.createFromPath(path.join(AVATAR_DIR, "avatar.svg"));
-    }
+    trayIcon = processTrayIcon(pngPath);
   } catch (e) {
     const message = e instanceof Error ? e.message : String(e);
     trayIcon = nativeImage.createFromPath(path.join(AVATAR_DIR, "avatar.svg"));
@@ -836,6 +833,38 @@ function createTray() {
   tray.setContextMenu(contextMenu);
   tray.on("click", () => mainWindow?.isVisible() ? mainWindow.hide() : mainWindow?.show());
 }
+function processTrayIcon(pngPath) {
+  log("Processing tray icon from: " + pngPath);
+  let trayIcon = nativeImage.createFromPath(pngPath);
+  if (!trayIcon.isEmpty()) {
+    const size = trayIcon.getSize();
+    const bitmap = trayIcon.getBitmap();
+    const chromaR = bitmap[0];
+    const chromaG = bitmap[1];
+    const chromaB = bitmap[2];
+    const tolerance = 30;
+    for (let i = 0;i < bitmap.length; i += 4) {
+      const r = bitmap[i];
+      const g = bitmap[i + 1];
+      const b = bitmap[i + 2];
+      if (Math.abs(r - chromaR) <= tolerance && Math.abs(g - chromaG) <= tolerance && Math.abs(b - chromaB) <= tolerance) {
+        bitmap[i + 3] = 0;
+      }
+    }
+    trayIcon = nativeImage.createFromBitmap(bitmap, size);
+    log("Tray icon processed successfully");
+  } else {
+    log("Tray icon is empty, using fallback");
+    trayIcon = nativeImage.createFromDataURL("data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==");
+  }
+  return trayIcon;
+}
+function updateTrayIcon(avatarPath) {
+  if (tray) {
+    const trayIcon = processTrayIcon(avatarPath);
+    tray.setImage(trayIcon);
+  }
+}
 app.commandLine.appendSwitch("enable-transparent-visuals");
 app.whenReady().then(async () => {
   fs.mkdirSync(AVATAR_DIR, { recursive: true });

package/dist/index.js CHANGED Viewed

@@ -6,9 +6,9 @@ import * as http from "http";
 import * as fs from "fs";
 import * as os from "os";
 var __dirname = "/var/home/wizard/av";
-var AVATAR_DIR = __dirname;
+var PLUGIN_DIR = __dirname;
+var AVATAR_DIR = path.join(os.homedir(), ".config", "opencode");
 var DEFAULT_AVATAR = "avatar.png";
-var USER_AVATAR = path.join(os.homedir(), ".config", "opencode", "avatar.png");
 var THINKING_PROMPT = "thinking hard";
 var AVATAR_PORT = 47291;
 function getToolPrompt(toolName, toolDescription) {
@@ -93,18 +93,16 @@ function promptToFilename(prompt, toolName) {
   return "avatar_" + baseName.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "_").substring(0, 50) + ".png";
 }
 function getAvatarPath(prompt, toolName) {
+  const defaultAvatar = path.join(AVATAR_DIR, DEFAULT_AVATAR);
   if (!prompt) {
-    if (fs.existsSync(USER_AVATAR)) {
-      return USER_AVATAR;
-    }
-    return path.join(AVATAR_DIR, DEFAULT_AVATAR);
+    return defaultAvatar;
   }
   const filename = promptToFilename(prompt, toolName);
   const avatarPath = path.join(AVATAR_DIR, filename);
   if (fs.existsSync(avatarPath)) {
     return avatarPath;
   }
-  return path.join(AVATAR_DIR, DEFAULT_AVATAR);
+  return defaultAvatar;
 }
 async function startElectron(avatarPath) {
   if (isShuttingDown) {
@@ -120,10 +118,10 @@ async function startElectron(avatarPath) {
     } catch (e) {}
     electronProcess = null;
   }
-  const electronPath = path.join(AVATAR_DIR, "node_modules", ".bin", "electron");
-  const electronEntry = path.join(AVATAR_DIR, "dist", "electron.js");
+  const electronPath = path.join(PLUGIN_DIR, "node_modules", ".bin", "electron");
+  const electronEntry = path.join(PLUGIN_DIR, "dist", "electron.js");
   const child = spawn(electronPath, [electronEntry, "--avatar", avatarPath, "--avatar-port", String(AVATAR_PORT)], {
-    cwd: AVATAR_DIR,
+    cwd: PLUGIN_DIR,
     stdio: ["ignore", "pipe", "pipe"],
     detached: false
   });
@@ -166,9 +164,9 @@ process.on("SIGTERM", () => {
 process.on("uncaughtException", (err) => {
   shutdownElectron();
 });
-async function setAvatarViaHttp(prompt, toolName) {
+async function setAvatarViaHttp(prompt, toolName, force) {
   const avatarPath = getAvatarPath(prompt, toolName);
-  if (avatarPath === currentAvatar) {
+  if (!force && avatarPath === currentAvatar) {
     return;
   }
   currentAvatar = avatarPath;
@@ -236,10 +234,6 @@ var AvatarPlugin = async ({ client }) => {
         let data = "";
         res.on("data", (chunk) => data += chunk);
         res.on("end", () => {
-          if (!showToasts) {
-            isToolActive = false;
-            isThinking = false;
-          }
           if (res.statusCode === 200) {
             if (showToasts) {
               showInfoToast(`Avatar ready: ${prompt}`);
@@ -303,7 +297,7 @@ var AvatarPlugin = async ({ client }) => {
         isThinking = false;
         isToolActive = false;
         currentRequestId = null;
-        await setAvatarViaHttp();
+        await setAvatarViaHttp(undefined, undefined, true);
       }
     }
   };

package/package.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "name": "opencode-avatar",
-  "version": "0.3.3",
+  "version": "0.3.6",
   "description": "Dynamic desktop avatar plugin for OpenCode that reacts to your coding activities",
   "type": "module",
   "main": "./dist/index.js",