@bytespell/shella 0.2.0 → 0.2.2

package/bundled-plugins/agent/package.json

@@ -0,0 +1,52 @@
+{
+  "name": "agent",
+  "private": true,
+  "version": "0.2.2",
+  "type": "module",
+  "main": "server.js",
+  "scripts": {
+    "dev": "vite",
+    "build": "tsc -b && vite build",
+    "lint": "eslint .",
+    "preview": "vite preview"
+  },
+  "dependencies": {
+    "@base-ui/react": "^1.1.0",
+    "@fontsource-variable/inter": "^5.2.8",
+    "@streamdown/code": "^1.0.1",
+    "@tailwindcss/typography": "^0.5.19",
+    "@tailwindcss/vite": "^4.1.17",
+    "class-variance-authority": "^0.7.1",
+    "clsx": "^2.1.1",
+    "express": "^5.0.0",
+    "lucide-react": "^0.562.0",
+    "radix-ui": "^1.4.3",
+    "react": "^19.2.0",
+    "react-dom": "^19.2.0",
+    "shadcn": "^3.7.0",
+    "streamdown": "^2.1.0",
+    "tailwind-merge": "^3.4.0",
+    "tailwindcss": "^4.1.17",
+    "tw-animate-css": "^1.4.0",
+    "use-stick-to-bottom": "^1.1.1",
+    "ws": "^8.19.0"
+  },
+  "devDependencies": {
+    "@eslint/js": "^9.39.1",
+    "@types/node": "^24.10.1",
+    "@types/react": "^19.2.5",
+    "@types/react-dom": "^19.2.3",
+    "@vitejs/plugin-react": "^5.1.1",
+    "eslint": "^9.39.1",
+    "eslint-plugin-react-hooks": "^7.0.1",
+    "eslint-plugin-react-refresh": "^0.4.24",
+    "globals": "^16.5.0",
+    "typescript": "~5.9.3",
+    "typescript-eslint": "^8.46.4",
+    "vite": "^7.2.4"
+  },
+  "shella": {
+    "displayName": "Agent",
+    "devCommand": "node server.js"
+  }
+}

package/bundled-plugins/agent/public/vite.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="31.88" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 257"><defs><linearGradient id="IconifyId1813088fe1fbc01fb466" x1="-.828%" x2="57.636%" y1="7.652%" y2="78.411%"><stop offset="0%" stop-color="#41D1FF"></stop><stop offset="100%" stop-color="#BD34FE"></stop></linearGradient><linearGradient id="IconifyId1813088fe1fbc01fb467" x1="43.376%" x2="50.316%" y1="2.242%" y2="89.03%"><stop offset="0%" stop-color="#FFEA83"></stop><stop offset="8.333%" stop-color="#FFDD35"></stop><stop offset="100%" stop-color="#FFA800"></stop></linearGradient></defs><path fill="url(#IconifyId1813088fe1fbc01fb466)" d="M255.153 37.938L134.897 252.976c-2.483 4.44-8.862 4.466-11.382.048L.875 37.958c-2.746-4.814 1.371-10.646 6.827-9.67l120.385 21.517a6.537 6.537 0 0 0 2.322-.004l117.867-21.483c5.438-.991 9.574 4.796 6.877 9.62Z"></path><path fill="url(#IconifyId1813088fe1fbc01fb467)" d="M185.432.063L96.44 17.501a3.268 3.268 0 0 0-2.634 3.014l-5.474 92.456a3.268 3.268 0 0 0 3.997 3.378l24.777-5.718c2.318-.535 4.413 1.507 3.936 3.838l-7.361 36.047c-.495 2.426 1.782 4.5 4.151 3.78l15.304-4.649c2.372-.72 4.652 1.36 4.15 3.788l-11.698 56.621c-.732 3.542 3.979 5.473 5.943 2.437l1.313-2.028l72.516-144.72c1.215-2.423-.88-5.186-3.54-4.672l-25.505 4.922c-2.396.462-4.435-1.77-3.759-4.114l16.646-57.705c.677-2.35-1.37-4.583-3.769-4.113Z"></path></svg>

package/bundled-plugins/agent/server.js

@@ -0,0 +1,240 @@
+import express from 'express';
+import { createServer } from 'http';
+import { WebSocketServer, WebSocket } from 'ws';
+import { spawn } from 'child_process';
+import path from 'path';
+import fs from 'fs';
+import os from 'os';
+import { fileURLToPath } from 'url';
+import readline from 'readline';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const app = express();
+const PORT = process.env.PORT;
+const isDev = !fs.existsSync(path.join(__dirname, 'dist', 'index.html'));
+const PI_AUTH_PATH = path.join(os.homedir(), '.pi', 'agent', 'auth.json');
+
+// Allow embedding in iframes
+app.use((req, res, next) => {
+  res.removeHeader('X-Frame-Options');
+  res.setHeader('Content-Security-Policy', 'frame-ancestors *');
+  next();
+});
+
+app.use(express.json());
+
+// Health check
+app.get('/api/health', (req, res) => {
+  res.json({ status: 'ok' });
+});
+
+// Check auth status
+app.get('/api/auth/status', (req, res) => {
+  try {
+    if (fs.existsSync(PI_AUTH_PATH)) {
+      const authData = JSON.parse(fs.readFileSync(PI_AUTH_PATH, 'utf-8'));
+      const providers = Object.keys(authData);
+      res.json({ authenticated: providers.length > 0, providers });
+    } else {
+      res.json({ authenticated: false, providers: [] });
+    }
+  } catch {
+    res.json({ authenticated: false, providers: [] });
+  }
+});
+
+async function startServer() {
+  if (isDev) {
+    const { createServer: createViteServer } = await import('vite');
+    const vite = await createViteServer({
+      server: { middlewareMode: true, hmr: { port: Number(PORT) + 1000 } },
+      appType: 'spa',
+    });
+    app.use(vite.middlewares);
+  } else {
+    app.use(express.static(path.join(__dirname, 'dist')));
+    app.get('/{*splat}', (req, res) => {
+      res.sendFile(path.join(__dirname, 'dist', 'index.html'));
+    });
+  }
+
+  const server = createServer(app);
+
+  // Single pi process shared across all clients
+  let piProcess = null;
+  let rl = null;
+  const clients = new Set();
+
+  function broadcast(message) {
+    for (const client of clients) {
+      if (client.readyState === WebSocket.OPEN) {
+        client.send(message);
+      }
+    }
+  }
+
+  function sendCommand(cmd) {
+    if (piProcess?.stdin?.writable) {
+      console.log('[agent] Sending command:', JSON.stringify(cmd));
+      piProcess.stdin.write(JSON.stringify(cmd) + '\n');
+    }
+  }
+
+  function spawnPi() {
+    if (piProcess) return;
+
+    console.log('[agent] Spawning pi --mode rpc');
+
+    piProcess = spawn('pi', ['--mode', 'rpc'], {
+      stdio: ['pipe', 'pipe', 'pipe'],
+      env: { ...process.env },
+    });
+
+    rl = readline.createInterface({
+      input: piProcess.stdout,
+      crlfDelay: Infinity
+    });
+
+    rl.on('line', (line) => {
+      try {
+        const event = JSON.parse(line);
+        // Log different event types
+        switch (event.type) {
+          case 'response':
+            console.log('[agent] Response:', event.command, event.success ? 'success' : 'failed',
+              event.data ? `data keys: ${Object.keys(event.data)}` : 'no data');
+            break;
+          case 'agent_start':
+            console.log('[agent] Agent started');
+            break;
+          case 'agent_end':
+            console.log('[agent] Agent ended:', event.stopReason || 'unknown reason');
+            break;
+          case 'tool_execution_start':
+            console.log('[agent] Tool start:', event.toolName, 'id:', event.toolCallId);
+            if (event.args) {
+              const argsStr = JSON.stringify(event.args);
+              console.log('[agent]   args:', argsStr.length > 200 ? argsStr.slice(0, 200) + '...' : argsStr);
+            }
+            break;
+          case 'tool_execution_update':
+            // Don't log updates to avoid spam, but could enable for debugging
+            break;
+          case 'tool_execution_end':
+            console.log('[agent] Tool end:', event.toolCallId, event.isError ? 'ERROR' : 'success');
+            break;
+          case 'message_start':
+            console.log('[agent] Message start:', event.message?.role || 'unknown role');
+            break;
+          case 'message_update':
+            // Don't log message updates to avoid spam
+            break;
+          case 'message_end':
+            console.log('[agent] Message end');
+            break;
+          default:
+            console.log('[agent] Event:', event.type);
+        }
+        broadcast(JSON.stringify({ type: 'rpc_event', event }));
+      } catch {
+        console.log('[agent] Non-JSON output:', line);
+      }
+    });
+
+    piProcess.stderr?.on('data', (data) => {
+      console.error('[agent] stderr:', data.toString());
+    });
+
+    piProcess.on('exit', (code, signal) => {
+      console.log(`[agent] pi exited: code=${code}, signal=${signal}`);
+      piProcess = null;
+      rl?.close();
+      rl = null;
+      broadcast(JSON.stringify({ type: 'process_exit', code, signal }));
+    });
+
+    piProcess.on('error', (err) => {
+      console.error('[agent] error:', err);
+      broadcast(JSON.stringify({
+        type: 'error',
+        message: `Failed to start pi: ${err.message}`
+      }));
+    });
+
+    sendCommand({ type: 'get_state' });
+    sendCommand({ type: 'get_messages' });
+  }
+
+  spawnPi();
+
+  const wss = new WebSocketServer({ server, path: '/ws' });
+
+  wss.on('connection', (ws) => {
+    console.log('[agent] Client connected');
+    clients.add(ws);
+
+    if (piProcess) {
+      ws.send(JSON.stringify({ type: 'ready' }));
+      sendCommand({ type: 'get_state' });
+      sendCommand({ type: 'get_messages' });
+      sendCommand({ type: 'get_session_stats' });
+    } else {
+      ws.send(JSON.stringify({ type: 'error', message: 'Pi process not running' }));
+    }
+
+    ws.on('message', (message) => {
+      try {
+        const msg = JSON.parse(message.toString());
+
+        switch (msg.type) {
+          case 'start':
+            if (!piProcess) spawnPi();
+            ws.send(JSON.stringify({ type: 'ready' }));
+            break;
+          case 'command':
+            if (msg.command) sendCommand(msg.command);
+            break;
+          case 'prompt':
+            sendCommand({ type: 'prompt', message: msg.message, images: msg.images });
+            break;
+          case 'abort':
+            sendCommand({ type: 'abort' });
+            break;
+          case 'steer':
+            sendCommand({ type: 'steer', message: msg.message });
+            break;
+          case 'extension_ui_response':
+            sendCommand(msg);
+            break;
+        }
+      } catch (err) {
+        console.error('[agent] Invalid message:', err);
+      }
+    });
+
+    ws.on('close', () => {
+      console.log('[agent] Client disconnected');
+      clients.delete(ws);
+    });
+
+    ws.on('error', (err) => {
+      console.error('[agent] WebSocket error:', err);
+      clients.delete(ws);
+    });
+  });
+
+  server.listen(PORT, () => {
+    console.log(`[agent] Running on port ${PORT}${isDev ? ' (dev mode)' : ''}`);
+  });
+
+  process.on('SIGTERM', () => {
+    console.log('[agent] Shutting down');
+    piProcess?.kill();
+    rl?.close();
+    for (const ws of clients) ws.close();
+    wss.close();
+    server.close(() => process.exit(0));
+  });
+}
+
+startServer();