@mediagato/modelreins-channel 1.0.0

package/modelreins-channel.mjs

@@ -0,0 +1,261 @@
+#!/usr/bin/env node
+/**
+ * ModelReins MCP Channel — bridges a Claude Code session to ModelReins.
+ *
+ * When installed in .mcp.json, this turns any Claude Code session into a
+ * ModelReins worker with full local access (files, git, SSH, env vars).
+ *
+ * The channel:
+ *   1. Registers with ModelReins as a worker (PUT /presence)
+ *   2. Polls for assigned jobs (GET /jobs?assigned_to=<name>&status=pending)
+ *   3. Pushes job prompts into Claude's session via MCP notifications
+ *   4. Captures Claude's responses and posts them back (POST /jobs/{id}/output)
+ *
+ * Config (env vars or passed via args):
+ *   MODELREINS_URL     — ModelReins server URL (default: http://localhost:8484)
+ *   MODELREINS_TOKEN   — Auth token
+ *   MODELREINS_WORKER  — Worker name (default: channel-<hostname>)
+ *
+ * .mcp.json entry:
+ *   {
+ *     "mcpServers": {
+ *       "modelreins": {
+ *         "command": "node",
+ *         "args": ["path/to/modelreins-channel.mjs"],
+ *         "env": {
+ *           "MODELREINS_URL": "https://app.modelreins.com",
+ *           "MODELREINS_TOKEN": "your-token",
+ *           "MODELREINS_WORKER": "my-laptop"
+ *         }
+ *       }
+ *     }
+ *   }
+ */
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import {
+    ListToolsRequestSchema,
+    CallToolRequestSchema,
+} from '@modelcontextprotocol/sdk/types.js';
+import { hostname } from 'os';
+
+// ── Config ──────────────────────────────────────────────────────────────────
+
+const BASE_URL = process.env.MODELREINS_URL || 'http://localhost:8484';
+const TOKEN = process.env.MODELREINS_TOKEN || '';
+const WORKER_NAME = process.env.MODELREINS_WORKER || `channel-${hostname().toLowerCase()}`;
+const POLL_MS = parseInt(process.env.MODELREINS_POLL_MS || '5000', 10);
+const WORKER_MODEL = process.env.MODELREINS_WORKER_MODEL || 'claude-session';
+const WORKER_TAGS = process.env.MODELREINS_WORKER_TAGS || 'local,interactive,full-access';
+
+// ── HTTP helpers ────────────────────────────────────────────────────────────
+
+async function api(method, path, body = null) {
+    const url = `${BASE_URL}${path}`;
+    const opts = {
+        method,
+        headers: {
+            'Authorization': `Bearer ${TOKEN}`,
+            'Content-Type': 'application/json',
+        },
+    };
+    if (body) opts.body = JSON.stringify(body);
+    try {
+        const res = await fetch(url, opts);
+        const data = await res.json();
+        return { status: res.status, data };
+    } catch (e) {
+        return { status: 0, data: { error: e.message } };
+    }
+}
+
+// ── State ───────────────────────────────────────────────────────────────────
+
+let currentJobId = null;
+let mcp = null;
+
+// ── MCP Server ──────────────────────────────────────────────────────────────
+
+mcp = new Server(
+    { name: 'modelreins', version: '1.0.0' },
+    {
+        capabilities: {
+            experimental: { 'claude/channel': {} },
+            tools: {},
+        },
+        instructions: [
+            `You are connected to ModelReins (${BASE_URL}) as worker "${WORKER_NAME}".`,
+            'Jobs from the ModelReins dashboard arrive as <channel source="modelreins" job_id="..." priority="..."> tags.',
+            'When you receive a job, complete the task described in the prompt using your full local capabilities.',
+            'When done, use the modelreins_complete tool to report the result back.',
+            'If you cannot complete a job, use modelreins_complete with success=false and explain why.',
+            'Between jobs, you remain a normal Claude Code session — the user can still interact with you directly.',
+        ].join('\n'),
+    }
+);
+
+// Tool: report job completion back to ModelReins
+mcp.setRequestHandler(ListToolsRequestSchema, async () => ({
+    tools: [
+        {
+            name: 'modelreins_complete',
+            description: 'Report job completion back to ModelReins dashboard. Call this when you finish a dispatched task.',
+            inputSchema: {
+                type: 'object',
+                properties: {
+                    job_id: { type: 'string', description: 'The job ID from the <channel> tag' },
+                    summary: { type: 'string', description: 'Brief summary of what was done' },
+                    success: { type: 'boolean', description: 'Whether the task succeeded' },
+                },
+                required: ['job_id', 'summary', 'success'],
+            },
+        },
+        {
+            name: 'modelreins_dispatch',
+            description: 'Dispatch a new job to another ModelReins worker from this session.',
+            inputSchema: {
+                type: 'object',
+                properties: {
+                    prompt: { type: 'string', description: 'The task prompt' },
+                    assigned_to: { type: 'string', description: 'Worker name or "auto"' },
+                    model_preference: { type: 'string', description: 'Model: auto/haiku/sonnet/opus' },
+                },
+                required: ['prompt'],
+            },
+        },
+        {
+            name: 'modelreins_status',
+            description: 'Check ModelReins fleet status — see active workers, running jobs, and schedules.',
+            inputSchema: {
+                type: 'object',
+                properties: {},
+            },
+        },
+    ],
+}));
+
+mcp.setRequestHandler(CallToolRequestSchema, async (req) => {
+    const { name, arguments: args } = req.params;
+
+    if (name === 'modelreins_complete') {
+        const { job_id, summary, success } = args;
+        // Post completion output
+        await api('POST', `/jobs/${job_id}/output`, {
+            stream: 'stdout',
+            content: summary,
+        });
+        // Mark job done or failed
+        await api('PUT', `/jobs/${job_id}`, {
+            status: success ? 'done' : 'failed',
+            exit_code: success ? 0 : 1,
+            error: success ? null : summary,
+        });
+        currentJobId = null;
+        return { content: [{ type: 'text', text: `Job #${job_id} ${success ? 'completed' : 'failed'}. Result posted to ModelReins.` }] };
+    }
+
+    if (name === 'modelreins_dispatch') {
+        const { prompt, assigned_to, model_preference } = args;
+        const res = await api('POST', '/dispatch', {
+            prompt,
+            assigned_to: assigned_to || 'auto',
+            model_preference: model_preference || null,
+        });
+        if (res.data?.success) {
+            return { content: [{ type: 'text', text: `Dispatched job #${res.data.id} to ${res.data.assigned_to}` }] };
+        }
+        return { content: [{ type: 'text', text: `Dispatch failed: ${JSON.stringify(res.data)}` }] };
+    }
+
+    if (name === 'modelreins_status') {
+        const [workers, jobs, schedules] = await Promise.all([
+            api('GET', '/workers'),
+            api('GET', '/jobs?limit=5'),
+            api('GET', '/schedules'),
+        ]);
+
+        const lines = ['## ModelReins Fleet Status\n'];
+
+        // Workers
+        const wList = workers.data?.workers || [];
+        lines.push(`### Workers (${wList.length})`);
+        for (const w of wList) {
+            lines.push(`- ${w.instance}: ${w.status} (${w.model || '?'})`);
+        }
+
+        // Recent jobs
+        const jList = workers.data?.jobs || jobs.data?.jobs || [];
+        lines.push(`\n### Recent Jobs`);
+        for (const j of jList.slice(0, 5)) {
+            lines.push(`- #${j.id} ${j.status} → ${j.assigned_to} (${j.prompt?.substring(0, 60)}...)`);
+        }
+
+        // Schedules
+        const sList = schedules.data?.schedules || [];
+        if (sList.length > 0) {
+            lines.push(`\n### Schedules (${sList.length})`);
+            for (const s of sList) {
+                lines.push(`- ${s.name}: ${s.human_schedule || s.cron_expr} → ${s.assigned_to || 'auto'} (runs: ${s.run_count})`);
+            }
+        }
+
+        return { content: [{ type: 'text', text: lines.join('\n') }] };
+    }
+
+    throw new Error(`Unknown tool: ${name}`);
+});
+
+// ── Connect ─────────────────────────────────────────────────────────────────
+
+await mcp.connect(new StdioServerTransport());
+
+// ── Heartbeat + Job Polling Loop ────────────────────────────────────────────
+
+async function heartbeat() {
+    await api('PUT', '/presence', {
+        instance: WORKER_NAME,
+        status: currentJobId ? 'busy' : 'active',
+        project: 'channel',
+        worker_type: 'channel',
+        model: WORKER_MODEL,
+        details: currentJobId ? `working on #${currentJobId}` : 'idle, polling',
+    });
+}
+
+async function pollJobs() {
+    if (currentJobId) return; // already working on something
+
+    const res = await api('GET', `/jobs?status=pending&assigned_to=${WORKER_NAME}&limit=1`);
+    const jobs = res.data?.jobs || (Array.isArray(res.data) ? res.data : []);
+
+    if (jobs.length > 0) {
+        const job = jobs[0];
+        currentJobId = job.id;
+
+        // Claim the job
+        await api('PUT', `/jobs/${job.id}`, { status: 'running' });
+
+        // Push into Claude's session
+        await mcp.notification({
+            method: 'notifications/claude/channel',
+            params: {
+                content: job.prompt,
+                meta: {
+                    job_id: String(job.id),
+                    priority: String(job.priority || 0),
+                    model_preference: job.model_preference || '',
+                    working_dir: job.working_dir || '',
+                    source: 'modelreins-dashboard',
+                },
+            },
+        });
+    }
+}
+
+// Main loop
+setInterval(heartbeat, 30000);
+setInterval(pollJobs, POLL_MS);
+
+// Initial heartbeat
+heartbeat();

package/package.json

@@ -0,0 +1,19 @@
+{
+  "name": "@mediagato/modelreins-channel",
+  "version": "1.0.0",
+  "description": "ModelReins MCP Channel — turn any Claude Code session into a ModelReins worker",
+  "type": "module",
+  "main": "modelreins-channel.mjs",
+  "bin": {
+    "modelreins-channel": "./modelreins-channel.mjs"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.0"
+  },
+  "keywords": ["modelreins", "mcp", "claude-code", "channel", "ai-orchestration"],
+  "license": "BSL-1.1",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/mediagato/modelreins"
+  }
+}