opencode-avatar 0.3.11 → 0.3.12

package/dist/electron.js

@@ -537,8 +537,10 @@ var httpServer = null;
 var ongoingGenerations = new Map;
 var lastHeartbeat = Date.now();
 var HEARTBEAT_TIMEOUT = 1000;
-function promptToFilename(prompt) {
-  return "avatar_" + prompt.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "_").substring(0, 50) + ".png";
+function promptToFilename(prompt, agentBase) {
+  const base = agentBase || "avatar";
+  const action = prompt.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "_").substring(0, 50);
+  return `${base}-${action}.png`;
 }
 function getAvatarPort() {
   const args = process.argv;
@@ -620,13 +622,14 @@ async function downloadImage(url, outputPath) {
   const buffer = Buffer.from(await response.arrayBuffer());
   fs.writeFileSync(outputPath, buffer);
 }
-async function generateAvatarForPrompt(prompt) {
+async function generateAvatarForPrompt(prompt, agentBase) {
   const config = getConfig();
+  const base = agentBase || "avatar";
   if (!config.falKey) {
     console.warn("falKey is not set. Cannot generate avatar. Using default avatar.");
-    return path.join(AVATAR_DIR, "avatar.png");
+    return path.join(AVATAR_DIR, `${base}.png`);
   }
-  const cachedFilename = promptToFilename(prompt);
+  const cachedFilename = promptToFilename(prompt, base);
   const cachedPath = path.join(AVATAR_DIR, cachedFilename);
   if (fs.existsSync(cachedPath)) {
     return cachedPath;
@@ -640,7 +643,10 @@ async function generateAvatarForPrompt(prompt) {
       if (fs.existsSync(cachedPath)) {
         return;
       }
-      const sourceAvatar = path.join(AVATAR_DIR, "avatar.png");
+      let sourceAvatar = path.join(AVATAR_DIR, `${base}.png`);
+      if (!fs.existsSync(sourceAvatar)) {
+        sourceAvatar = path.join(AVATAR_DIR, "avatar.png");
+      }
       const uploadedUrl = await uploadFile(sourceAvatar, config.falKey);
       let fullPrompt = `make a character variant: ${prompt}. The action should be literal - the character should actually be performing the action (writing means writing, typing means typing, etc.), not shown as some abstract Terminal visualization or text output. Maintain the character's original essence and physicality. Keep the background as a solid green screen color. Do not let the green screen color appear in reflections or on the subject.`;
       if (config.prompt) {
@@ -702,13 +708,13 @@ function startAvatarServer() {
       });
       req.on("end", async () => {
         try {
-          const { prompt } = JSON.parse(body);
+          const { prompt, agentBase } = JSON.parse(body);
           if (!prompt) {
             res.writeHead(400, { "Content-Type": "application/json" });
             res.end(JSON.stringify({ error: "Missing prompt" }));
             return;
           }
-          const avatarPath = await generateAvatarForPrompt(prompt);
+          const avatarPath = await generateAvatarForPrompt(prompt, agentBase);
           if (mainWindow) {
             const imageBuffer = fs.readFileSync(avatarPath);
             const base64 = imageBuffer.toString("base64");

package/dist/index.js

@@ -11,6 +11,19 @@ var AVATAR_DIR = path.join(os.homedir(), ".config", "opencode");
 var DEFAULT_AVATAR = "avatar.png";
 var THINKING_PROMPT = "thinking hard";
 var AVATAR_PORT = 47291;
+function normalizeAgentName(name) {
+  return name.toLowerCase().replace(/\s+/g, "_");
+}
+function getAgentAvatarBase(agentName) {
+  if (!agentName)
+    return "avatar";
+  const normalized = normalizeAgentName(agentName);
+  const agentAvatarPath = path.join(AVATAR_DIR, `avatar-${normalized}.png`);
+  if (fs.existsSync(agentAvatarPath)) {
+    return `avatar-${normalized}`;
+  }
+  return "avatar";
+}
 function getToolPrompt(toolName, toolDescription) {
   if (toolDescription) {
     const shortDesc = toolDescription.split(".")[0].substring(0, 50);
@@ -26,6 +39,8 @@ var isShuttingDown = false;
 var idleTriggered = false;
 var currentRequestId = null;
 var heartbeatInterval = null;
+var currentAgentName = null;
+var currentAgentBase = "avatar";
 function sendHeartbeat() {
   const req = http.request({
     hostname: "localhost",
@@ -88,21 +103,30 @@ async function sendShutdownCommand() {
     req.end();
   });
 }
-function promptToFilename(prompt, toolName) {
+function promptToFilename(prompt, toolName, agentBase) {
+  const base = agentBase || "avatar";
   const baseName = toolName || prompt;
-  return "avatar_" + baseName.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "_").substring(0, 50) + ".png";
+  const action = baseName.toLowerCase().replace(/[^a-z0-9\s]/g, "").replace(/\s+/g, "_").substring(0, 50);
+  return `${base}-${action}.png`;
 }
-function getAvatarPath(prompt, toolName) {
-  const defaultAvatar = path.join(AVATAR_DIR, DEFAULT_AVATAR);
+function getAvatarPath(prompt, toolName, agentBase) {
+  const base = agentBase || currentAgentBase;
+  const defaultAvatar = path.join(AVATAR_DIR, `${base}.png`);
   if (!prompt) {
+    if (!fs.existsSync(defaultAvatar)) {
+      return path.join(AVATAR_DIR, DEFAULT_AVATAR);
+    }
     return defaultAvatar;
   }
-  const filename = promptToFilename(prompt, toolName);
+  const filename = promptToFilename(prompt, toolName, base);
   const avatarPath = path.join(AVATAR_DIR, filename);
   if (fs.existsSync(avatarPath)) {
     return avatarPath;
   }
-  return defaultAvatar;
+  if (fs.existsSync(defaultAvatar)) {
+    return defaultAvatar;
+  }
+  return path.join(AVATAR_DIR, DEFAULT_AVATAR);
 }
 async function startElectron(avatarPath) {
   if (isShuttingDown) {
@@ -256,7 +280,7 @@ var AvatarPlugin = async ({ client }) => {
         }
         reject(err);
       });
-      req.write(JSON.stringify({ prompt }));
+      req.write(JSON.stringify({ prompt, agentBase: currentAgentBase }));
       req.end();
     });
   }
@@ -292,6 +316,16 @@ var AvatarPlugin = async ({ client }) => {
       });
     },
     event: async ({ event }) => {
+      if (event.type === "message.updated") {
+        const message = event.properties?.info;
+        if (message?.role === "user" && message?.agent) {
+          const agentName = message.agent;
+          if (currentAgentName !== agentName) {
+            currentAgentName = agentName;
+            currentAgentBase = getAgentAvatarBase(currentAgentName);
+          }
+        }
+      }
       if (event.type === "session.idle" && (isThinking || isToolActive)) {
         idleTriggered = true;
         isThinking = false;

package/package.json

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