fa30191394
Captures the last 10 messages from AgentAPI when task workspaces stop, allowing users to view conversation history while the task is paused. The shutdown script fetches messages, builds a payload with last 10 messages, truncates to 64KB if needed (removes old messages first, then truncates content of the last message), and posts to the log snapshot endpoint. Gracefully handles non-task workspaces (skips), older Coder versions without the endpoint (logs and continues), and empty message sets. Enabled by default via task_log_snapshot variable. Task ID is automatically resolved from data.coder_task when available. Updates coder/internal#1257
85 lines
2.0 KiB
JavaScript
85 lines
2.0 KiB
JavaScript
#!/usr/bin/env node
|
|
// Mock AgentAPI server for shutdown script tests.
|
|
// Usage: MESSAGES='[...]' node agentapi-mock-shutdown.js [port]
|
|
|
|
const http = require("http");
|
|
const port = process.argv[2] || 3284;
|
|
|
|
// Parse messages from environment or use default
|
|
let messages = [];
|
|
if (process.env.MESSAGES) {
|
|
try {
|
|
messages = JSON.parse(process.env.MESSAGES);
|
|
} catch (e) {
|
|
console.error("Failed to parse MESSAGES env var:", e.message);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
// Presets for common test scenarios
|
|
if (process.env.PRESET === "normal") {
|
|
messages = [
|
|
{ id: 1, type: "input", content: "Hello", time: "2025-01-01T00:00:00Z" },
|
|
{
|
|
id: 2,
|
|
type: "output",
|
|
content: "Hi there",
|
|
time: "2025-01-01T00:00:01Z",
|
|
},
|
|
{
|
|
id: 3,
|
|
type: "input",
|
|
content: "How are you?",
|
|
time: "2025-01-01T00:00:02Z",
|
|
},
|
|
{
|
|
id: 4,
|
|
type: "output",
|
|
content: "Good!",
|
|
time: "2025-01-01T00:00:03Z",
|
|
},
|
|
{ id: 5, type: "input", content: "Great", time: "2025-01-01T00:00:04Z" },
|
|
];
|
|
} else if (process.env.PRESET === "many") {
|
|
messages = Array.from({ length: 15 }, (_, i) => ({
|
|
id: i + 1,
|
|
type: "input",
|
|
content: `Message ${i + 1}`,
|
|
time: "2025-01-01T00:00:00Z",
|
|
}));
|
|
} else if (process.env.PRESET === "huge") {
|
|
messages = [
|
|
{
|
|
id: 1,
|
|
type: "output",
|
|
content: "x".repeat(70000),
|
|
time: "2025-01-01T00:00:00Z",
|
|
},
|
|
];
|
|
}
|
|
|
|
const server = http.createServer((req, res) => {
|
|
if (req.url === "/messages") {
|
|
res.writeHead(200, { "Content-Type": "application/json" });
|
|
res.end(JSON.stringify({ messages }));
|
|
} else if (req.url === "/status") {
|
|
res.writeHead(200, { "Content-Type": "application/json" });
|
|
res.end(JSON.stringify({ status: "stable" }));
|
|
} else {
|
|
res.writeHead(404);
|
|
res.end();
|
|
}
|
|
});
|
|
|
|
server.listen(port, () => {
|
|
console.error(`Mock AgentAPI listening on port ${port}`);
|
|
});
|
|
|
|
process.on("SIGTERM", () => {
|
|
server.close(() => process.exit(0));
|
|
});
|
|
|
|
process.on("SIGINT", () => {
|
|
server.close(() => process.exit(0));
|
|
});
|