@straiffi/archon 1.2.8 → 1.2.10

package/dist/server/index.js

@@ -1,4 +1,5 @@
 import express from 'express';
+import multer from 'multer';
 import { existsSync } from 'fs';
 import { createServer } from 'http';
 import { fileURLToPath } from 'url';
@@ -8,6 +9,7 @@ import cors from 'cors';
 import { adjectives, animals, uniqueNamesGenerator } from 'unique-names-generator';
 import db from './db.js';
 import { deleteChatSession, getChatSession, listChatSessions, updateChatSessionMode } from './lib/chats.js';
+import { createChatAttachment, deleteChatAttachment, getChatAttachmentById, getChatAttachmentForSession, getChatAttachmentLinkedMessageId, MAX_CHAT_ATTACHMENTS_PER_TURN, MAX_CHAT_ATTACHMENT_SIZE_BYTES, readChatAttachmentContent, validateImageAttachment, } from './lib/attachments.js';
 import { createProjectChatSession, resolveChatMode, resolveChatTool, stopProjectChatSessionResponse, submitProjectChatSessionMessage, } from './lib/chatOperations.js';
 import { resolveChatSessionTarget } from './lib/chatTargets.js';
 import { createChatMessage, getChatMessage, updateChatMessageContent } from './lib/chatMessages.js';
@@ -275,6 +277,24 @@ const asTrimmedString = (value) => {
     return typeof value === 'string' ? value.trim() : '';
 };
 const COMMIT_MESSAGE_DIFF_CHAR_LIMIT = 12_000;
+const chatAttachmentUpload = multer({
+    storage: multer.memoryStorage(),
+    limits: {
+        files: MAX_CHAT_ATTACHMENTS_PER_TURN,
+        fileSize: MAX_CHAT_ATTACHMENT_SIZE_BYTES,
+    },
+});
+const getChatAttachmentUploadErrorMessage = (error) => {
+    if (error instanceof multer.MulterError) {
+        if (error.code === 'LIMIT_FILE_COUNT') {
+            return 'You can attach up to 4 images per turn.';
+        }
+        if (error.code === 'LIMIT_FILE_SIZE') {
+            return 'Images must be 10 MB or smaller.';
+        }
+    }
+    return 'Unable to upload the selected images right now.';
+};
 const normalizeGeneratedCommitMessage = (value) => {
     const firstLine = value
         .replace(/\r/g, '\n')
@@ -871,12 +891,12 @@ const listBundleConversationTickets = (projectId, bundleId) => {
 };
 const moveBundleTicketToReview = (ticket) => {
     const laneOrder = assignTicketLaneOrder({ ...ticket, state: 'review' });
-    db.prepare('UPDATE tickets SET state = ?, tool = ?, agent_status = NULL, agent_log = NULL, streaming_response = NULL, lane_order = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?').run('review', resolveTicketTool(ticket, config.tool ?? 'opencode'), laneOrder ?? null, ticket.id);
+    db.prepare('UPDATE tickets SET state = ?, tool = ?, agent_status = NULL, streaming_response = NULL, lane_order = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?').run('review', resolveTicketTool(ticket, config.tool ?? 'opencode'), laneOrder ?? null, ticket.id);
     return getTicket(ticket.id);
 };
 const restoreBundleReviewTicketToBuild = (ticket) => {
     const laneOrder = assignTicketLaneOrder({ ...ticket, state: 'build' });
-    db.prepare('UPDATE tickets SET state = ?, agent_status = ?, agent_log = NULL, streaming_response = NULL, lane_order = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?').run('build', 'done', laneOrder ?? null, ticket.id);
+    db.prepare('UPDATE tickets SET state = ?, agent_status = ?, streaming_response = NULL, lane_order = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?').run('build', 'done', laneOrder ?? null, ticket.id);
     return getTicket(ticket.id);
 };
 const resolveBundleLatestSessionTicket = (tickets) => {
@@ -2164,6 +2184,103 @@ app.post('/projects/:id/chat-sessions/:sessionId/messages', (req, res) => {
     }
     return res.status(result.status).json(result.value);
 });
+app.post('/projects/:id/chat-sessions/:sessionId/attachments', (req, res) => {
+    const project = getProjectById(req.params.id);
+    if (!project) {
+        return res.status(404).json({ error: 'Not found' });
+    }
+    const chatSession = getProjectChatSession(project.id, req.params.sessionId);
+    if (!chatSession) {
+        return res.status(404).json({ error: 'Not found' });
+    }
+    if (chatSession.agent_status === 'running') {
+        return res.status(409).json({ error: 'Wait for the current chat response to finish before uploading images.' });
+    }
+    return chatAttachmentUpload.array('files', MAX_CHAT_ATTACHMENTS_PER_TURN)(req, res, error => {
+        if (error) {
+            return res.status(400).json({ error: getChatAttachmentUploadErrorMessage(error) });
+        }
+        const files = (req.files ?? []);
+        if (files.length === 0) {
+            return res.status(400).json({ error: 'At least one image is required.' });
+        }
+        for (const file of files) {
+            const validationError = validateImageAttachment({
+                mimeType: file.mimetype,
+                originalName: file.originalname,
+            });
+            if (validationError) {
+                return res.status(400).json({ error: validationError });
+            }
+        }
+        const createdAttachmentIds = [];
+        try {
+            const attachments = files.map(file => {
+                const attachment = createChatAttachment({
+                    chatSessionId: chatSession.id,
+                    originalName: file.originalname,
+                    mimeType: file.mimetype,
+                    buffer: file.buffer,
+                    projectId: project.id,
+                });
+                createdAttachmentIds.push(attachment.id);
+                return attachment;
+            });
+            return res.status(201).json({ attachments });
+        }
+        catch (uploadError) {
+            for (const attachmentId of createdAttachmentIds) {
+                const attachment = getChatAttachmentById(attachmentId);
+                if (!attachment) {
+                    continue;
+                }
+                deleteChatAttachment(attachment);
+            }
+            return res.status(500).json({ error: uploadError instanceof Error ? uploadError.message : 'Unable to upload the selected images right now.' });
+        }
+    });
+});
+app.delete('/projects/:id/chat-sessions/:sessionId/attachments/:attachmentId', (req, res) => {
+    const project = getProjectById(req.params.id);
+    if (!project) {
+        return res.status(404).json({ error: 'Not found' });
+    }
+    const chatSession = getProjectChatSession(project.id, req.params.sessionId);
+    if (!chatSession) {
+        return res.status(404).json({ error: 'Not found' });
+    }
+    const attachment = getChatAttachmentForSession(req.params.attachmentId, chatSession.id);
+    if (!attachment) {
+        return res.status(404).json({ error: 'Attachment not found' });
+    }
+    if (getChatAttachmentLinkedMessageId(attachment.id)) {
+        return res.status(409).json({ error: 'Sent message attachments cannot be removed.' });
+    }
+    deleteChatAttachment(attachment);
+    return res.json({ ok: true });
+});
+app.get('/projects/:id/chat-sessions/:sessionId/attachments/:attachmentId/content', (req, res) => {
+    const project = getProjectById(req.params.id);
+    if (!project) {
+        return res.status(404).json({ error: 'Not found' });
+    }
+    const chatSession = getProjectChatSession(project.id, req.params.sessionId);
+    if (!chatSession) {
+        return res.status(404).json({ error: 'Not found' });
+    }
+    const attachment = getChatAttachmentForSession(req.params.attachmentId, chatSession.id);
+    if (!attachment) {
+        return res.status(404).json({ error: 'Attachment not found' });
+    }
+    try {
+        const content = readChatAttachmentContent(attachment);
+        res.type(attachment.mime_type);
+        return res.send(content);
+    }
+    catch {
+        return res.status(404).json({ error: 'Attachment content not found' });
+    }
+});
 app.post('/projects/:id/chat-sessions/:sessionId/stop', async (req, res) => {
     const result = await stopProjectChatSessionResponse({
         projectId: req.params.id,
@@ -4640,7 +4757,7 @@ app.post('/tickets/:id/retry', (req, res) => {
     if (ticket.agent_status !== 'error') {
         return res.status(400).json({ error: 'Ticket is not in error state' });
     }
-    db.prepare('UPDATE tickets SET agent_status = NULL, agent_log = NULL, streaming_response = NULL, build_completed_at = NULL, auto_park_dismissed_at = NULL, updated_at = CURRENT_TIMESTAMP WHERE id = ?').run(req.params.id);
+    db.prepare('UPDATE tickets SET agent_status = NULL, agent_log = NULL, agent_message_id = NULL, previous_agent_log = NULL, previous_agent_message_id = NULL, streaming_response = NULL, build_completed_at = NULL, auto_park_dismissed_at = NULL, updated_at = CURRENT_TIMESTAMP WHERE id = ?').run(req.params.id);
     const updated = getTicket(req.params.id);
     io.emit('ticket:updated', updated);
     res.json(updated);