@onmars/lunar-core 0.2.0 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@onmars/lunar-core",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "type": "module",
   "main": "src/index.ts",
   "types": "src/index.ts",

package/src/__tests__/router.test.ts

@@ -1190,6 +1190,150 @@ describe('Router', () => {
     })
   })
 
+  // ─── Attachment download and injection ──────────────────────
+
+  describe('image/file attachment handling', () => {
+    it('downloads image attachment and prepends path to prompt', async () => {
+      const spy = createSpyAgent()
+      registry.registerAgent(spy, true)
+
+      const channels: import('../types/moon').ChannelPersona[] = [
+        { id: 'chan-1', name: 'orbit', platform: 'discord' },
+      ]
+      const router = new Router(registry, sessions, mockConfig, moonLoader, channels)
+
+      // Mock global fetch to return fake image data
+      const originalFetch = globalThis.fetch
+      globalThis.fetch = (async () =>
+        new Response(new Uint8Array([0x89, 0x50, 0x4e, 0x47]), {
+          status: 200,
+        })) as unknown as typeof fetch
+
+      try {
+        const message: IncomingMessage = {
+          id: 'msg-1',
+          channelId: 'chan-1',
+          text: 'What is this?',
+          sender: { id: 'user-123', name: 'Test User' },
+          timestamp: new Date(),
+          attachments: [
+            {
+              type: 'image',
+              url: 'https://cdn.discord.com/attachments/test/image.png',
+              filename: 'photo.png',
+              mimeType: 'image/png',
+            },
+          ],
+        }
+
+        await router.route('discord', message)
+
+        expect(spy.capturedInput).not.toBeNull()
+        expect(spy.capturedInput.prompt).toContain('[media attached:')
+        expect(spy.capturedInput.prompt).toContain('.png')
+        expect(spy.capturedInput.prompt).toContain('What is this?')
+      } finally {
+        globalThis.fetch = originalFetch
+      }
+    })
+
+    it('handles multiple attachments', async () => {
+      const spy = createSpyAgent()
+      registry.registerAgent(spy, true)
+
+      const channels: import('../types/moon').ChannelPersona[] = [
+        { id: 'chan-1', name: 'orbit', platform: 'discord' },
+      ]
+      const router = new Router(registry, sessions, mockConfig, moonLoader, channels)
+
+      const originalFetch = globalThis.fetch
+      globalThis.fetch = (async () =>
+        new Response(new Uint8Array([0x00, 0x01]), { status: 200 })) as unknown as typeof fetch
+
+      try {
+        const message: IncomingMessage = {
+          id: 'msg-1',
+          channelId: 'chan-1',
+          text: 'Check these',
+          sender: { id: 'user-123', name: 'Test User' },
+          timestamp: new Date(),
+          attachments: [
+            { type: 'image', url: 'https://example.com/a.png', filename: 'a.png', mimeType: 'image/png' },
+            { type: 'file', url: 'https://example.com/b.pdf', filename: 'b.pdf', mimeType: 'application/pdf' },
+          ],
+        }
+
+        await router.route('discord', message)
+
+        expect(spy.capturedInput.prompt).toContain('.png')
+        expect(spy.capturedInput.prompt).toContain('.pdf')
+        expect(spy.capturedInput.prompt).toContain('Check these')
+      } finally {
+        globalThis.fetch = originalFetch
+      }
+    })
+
+    it('skips audio attachments (handled by STT)', async () => {
+      const spy = createSpyAgent()
+      registry.registerAgent(spy, true)
+
+      const channels: import('../types/moon').ChannelPersona[] = [
+        { id: 'chan-1', name: 'orbit', platform: 'discord' },
+      ]
+      const router = new Router(registry, sessions, mockConfig, moonLoader, channels)
+
+      const message: IncomingMessage = {
+        id: 'msg-1',
+        channelId: 'chan-1',
+        text: 'Hello',
+        sender: { id: 'user-123', name: 'Test User' },
+        timestamp: new Date(),
+        attachments: [
+          { type: 'audio', url: 'https://example.com/voice.ogg', filename: 'voice.ogg' },
+        ],
+      }
+
+      await router.route('discord', message)
+
+      // Audio should not be downloaded by the image handler (no fetch mock needed)
+      expect(spy.capturedInput.prompt).toBe('Hello')
+    })
+
+    it('continues gracefully when attachment download fails', async () => {
+      const spy = createSpyAgent()
+      registry.registerAgent(spy, true)
+
+      const channels: import('../types/moon').ChannelPersona[] = [
+        { id: 'chan-1', name: 'orbit', platform: 'discord' },
+      ]
+      const router = new Router(registry, sessions, mockConfig, moonLoader, channels)
+
+      const originalFetch = globalThis.fetch
+      globalThis.fetch = (async () =>
+        new Response(null, { status: 404 })) as unknown as typeof fetch
+
+      try {
+        const message: IncomingMessage = {
+          id: 'msg-1',
+          channelId: 'chan-1',
+          text: 'Check this',
+          sender: { id: 'user-123', name: 'Test User' },
+          timestamp: new Date(),
+          attachments: [
+            { type: 'image', url: 'https://example.com/broken.png', filename: 'broken.png' },
+          ],
+        }
+
+        await router.route('discord', message)
+
+        // Should still send the text without attachment reference
+        expect(spy.capturedInput.prompt).toBe('Check this')
+      } finally {
+        globalThis.fetch = originalFetch
+      }
+    })
+  })
+
   // ─── Multi-agent routing (Phase B) ────────────────────────
 
   describe('multi-agent routing — channelPersona.agentId directs to correct agent', () => {

package/src/lib/router.ts

@@ -1,4 +1,5 @@
-import { existsSync, readFileSync } from 'node:fs'
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs'
+import { randomUUID } from 'node:crypto'
 import { resolve } from 'node:path'
 import type { AgentEvent, AgentInput } from '../types/agent'
 import type { IncomingMessage, OutgoingMessage } from '../types/channel'
@@ -316,6 +317,37 @@ export class Router {
       }
     }
 
+    // ─── Image/File attachments: download and save ───────────────
+    const nonAudioAttachments =
+      message.attachments?.filter((a) => a.type !== 'audio') ?? []
+
+    if (nonAudioAttachments.length > 0) {
+      const mediaDir = resolve(this.config.dataDir, 'media', 'inbound')
+      mkdirSync(mediaDir, { recursive: true })
+
+      for (const att of nonAudioAttachments) {
+        try {
+          const buffer = await this.downloadAttachment(att.url)
+          const ext =
+            att.filename?.split('.').pop() ?? (att.type === 'image' ? 'png' : 'bin')
+          const filename = `${randomUUID()}.${ext}`
+          const filepath = resolve(mediaDir, filename)
+          writeFileSync(filepath, buffer)
+
+          const label = att.type === 'image' ? 'image' : att.type
+          effectiveText = `[media attached: ${filepath} (${att.mimeType ?? label}) | ${filepath}]\n${effectiveText}`
+
+          log.info(
+            { type: att.type, filename: att.filename, size: buffer.length, path: filepath },
+            'Attachment saved',
+          )
+        } catch (err) {
+          const errMsg = err instanceof Error ? err.message : String(err)
+          log.warn({ error: errMsg, url: att.url }, 'Failed to download attachment — skipping')
+        }
+      }
+    }
+
     // Build system prompt from moon + channel persona
     let systemPrompt = moon ? this.moonLoader.buildSystemPrompt(moon, channelPersona) : undefined