onkol 0.5.1 → 0.5.2

package/dist/plugin/discord-client.js

@@ -1,4 +1,10 @@
 import { Client, GatewayIntentBits } from 'discord.js';
+import { mkdirSync, writeFileSync } from 'fs';
+import { join } from 'path';
+import { tmpdir } from 'os';
+const IMAGE_EXTENSIONS = new Set(['.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp', '.svg']);
+const ATTACHMENT_DIR = join(tmpdir(), 'onkol-attachments');
+mkdirSync(ATTACHMENT_DIR, { recursive: true });
 export function shouldForwardMessage(messageChannelId, authorId, isBot, targetChannelId, allowedUsers) {
     if (isBot)
         return false;
@@ -8,17 +14,38 @@ export function shouldForwardMessage(messageChannelId, authorId, isBot, targetCh
         return false;
     return true;
 }
-// When a message is too long, Discord auto-converts it to a .txt file attachment
-// with empty message content. This fetches the text from those attachments.
-async function resolveTextAttachments(message) {
+function isTextAttachment(a) {
+    return a.contentType?.startsWith('text/') === true || a.name?.endsWith('.txt') === true;
+}
+function isImageAttachment(a) {
+    if (a.contentType?.startsWith('image/'))
+        return true;
+    const name = (a.name || '').toLowerCase();
+    return IMAGE_EXTENSIONS.has(name.slice(name.lastIndexOf('.')));
+}
+// Resolve all attachments: text gets inlined, images get downloaded to temp files
+async function resolveAttachments(message) {
     let content = message.content;
-    const textAttachments = message.attachments.filter((a) => a.contentType?.startsWith('text/') || a.name?.endsWith('.txt'));
-    for (const attachment of textAttachments.values()) {
+    for (const attachment of message.attachments.values()) {
         try {
-            const res = await fetch(attachment.url);
-            if (res.ok) {
-                const text = await res.text();
-                content = content ? `${content}\n\n${text}` : text;
+            if (isTextAttachment(attachment)) {
+                const res = await fetch(attachment.url);
+                if (res.ok) {
+                    const text = await res.text();
+                    content = content ? `${content}\n\n${text}` : text;
+                }
+            }
+            else if (isImageAttachment(attachment)) {
+                const res = await fetch(attachment.url);
+                if (res.ok) {
+                    const buffer = Buffer.from(await res.arrayBuffer());
+                    const filename = `${message.id}-${attachment.name || 'image.png'}`;
+                    const filepath = join(ATTACHMENT_DIR, filename);
+                    writeFileSync(filepath, buffer);
+                    const note = `[User sent an image: ${attachment.name || 'image'}. Saved to ${filepath} — use the Read tool to view it.]`;
+                    content = content ? `${content}\n\n${note}` : note;
+                    console.error(`[discord-filtered] Downloaded image: ${filepath} (${buffer.length} bytes)`);
+                }
             }
         }
         catch (err) {
@@ -37,7 +64,7 @@ export function createDiscordClient(config, onMessage) {
     });
     client.on('messageCreate', async (message) => {
         if (shouldForwardMessage(message.channel.id, message.author.id, message.author.bot, config.channelId, config.allowedUsers)) {
-            const content = await resolveTextAttachments(message);
+            const content = await resolveAttachments(message);
             if (content) {
                 onMessage(content, message);
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "onkol",
-  "version": "0.5.1",
+  "version": "0.5.2",
   "description": "Decentralized on-call agent system powered by Claude Code",
   "type": "module",
   "bin": {

package/src/plugin/discord-client.ts

@@ -1,4 +1,11 @@
 import { Client, GatewayIntentBits, type Message, type Attachment } from 'discord.js'
+import { mkdirSync, writeFileSync } from 'fs'
+import { join } from 'path'
+import { tmpdir } from 'os'
+
+const IMAGE_EXTENSIONS = new Set(['.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp', '.svg'])
+const ATTACHMENT_DIR = join(tmpdir(), 'onkol-attachments')
+mkdirSync(ATTACHMENT_DIR, { recursive: true })
 
 export interface DiscordClientConfig {
   botToken: string
@@ -19,24 +26,45 @@ export function shouldForwardMessage(
   return true
 }
 
-// When a message is too long, Discord auto-converts it to a .txt file attachment
-// with empty message content. This fetches the text from those attachments.
-async function resolveTextAttachments(message: Message): Promise<string> {
+function isTextAttachment(a: Attachment): boolean {
+  return a.contentType?.startsWith('text/') === true || a.name?.endsWith('.txt') === true
+}
+
+function isImageAttachment(a: Attachment): boolean {
+  if (a.contentType?.startsWith('image/')) return true
+  const name = (a.name || '').toLowerCase()
+  return IMAGE_EXTENSIONS.has(name.slice(name.lastIndexOf('.')))
+}
+
+// Resolve all attachments: text gets inlined, images get downloaded to temp files
+async function resolveAttachments(message: Message): Promise<string> {
   let content = message.content
-  const textAttachments = message.attachments.filter(
-    (a: Attachment) => a.contentType?.startsWith('text/') || a.name?.endsWith('.txt')
-  )
-  for (const attachment of textAttachments.values()) {
+
+  for (const attachment of message.attachments.values()) {
     try {
-      const res = await fetch(attachment.url)
-      if (res.ok) {
-        const text = await res.text()
-        content = content ? `${content}\n\n${text}` : text
+      if (isTextAttachment(attachment)) {
+        const res = await fetch(attachment.url)
+        if (res.ok) {
+          const text = await res.text()
+          content = content ? `${content}\n\n${text}` : text
+        }
+      } else if (isImageAttachment(attachment)) {
+        const res = await fetch(attachment.url)
+        if (res.ok) {
+          const buffer = Buffer.from(await res.arrayBuffer())
+          const filename = `${message.id}-${attachment.name || 'image.png'}`
+          const filepath = join(ATTACHMENT_DIR, filename)
+          writeFileSync(filepath, buffer)
+          const note = `[User sent an image: ${attachment.name || 'image'}. Saved to ${filepath} — use the Read tool to view it.]`
+          content = content ? `${content}\n\n${note}` : note
+          console.error(`[discord-filtered] Downloaded image: ${filepath} (${buffer.length} bytes)`)
+        }
       }
     } catch (err) {
       console.error(`[discord-filtered] Failed to fetch attachment ${attachment.name}: ${err}`)
     }
   }
+
   return content
 }
 
@@ -62,7 +90,7 @@ export function createDiscordClient(
         config.allowedUsers
       )
     ) {
-      const content = await resolveTextAttachments(message)
+      const content = await resolveAttachments(message)
       if (content) {
         onMessage(content, message)
       }