@sharnix/agent 1.0.0

package/index.js

@@ -0,0 +1,244 @@
+#!/usr/bin/env node
+'use strict';
+
+const WebSocket = require('ws');
+const http = require('http');
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { randomBytes, createHash } = require('crypto');
+
+const RELAY_BASE = (process.env.SHARNIX_URL || 'https://relay.sharnix.com').replace(/\/$/, '');
+const WS_BASE = RELAY_BASE.replace(/^http/, 'ws');
+const API_KEY = process.env.SHARNIX_API_KEY || '';
+
+// ── CLI args ──────────────────────────────────────────────────────────────────
+const args = process.argv.slice(2);
+const get = (flag) => {
+  const i = args.findIndex((a) => a === flag);
+  return i !== -1 ? args[i + 1] : null;
+};
+const has = (flag) => args.includes(flag);
+
+const port = parseInt(get('--port') || get('-p') || '3000', 10);
+const label = get('--label') || get('-l') || null;
+const agentName = get('--name') || 'local-dev';
+
+if (has('--help') || has('-h')) {
+  console.log(`
+  sharnix — tunnel your local app through Sharnix
+
+  Usage:
+    npx @sharnix/agent --port <port> [options]
+    SHARNIX_API_KEY=shx_... npx @sharnix/agent --port 3000
+
+  Options:
+    --port, -p <n>    Local port to forward (default: 3000)
+    --label, -l <s>   Human-readable label for this tunnel
+    --name <s>        Agent name used on first setup (default: local-dev)
+    --help, -h        Show this message
+
+  Environment:
+    SHARNIX_API_KEY   API key from relay.sharnix.com/app/settings (required)
+    SHARNIX_URL       Override relay base URL
+`);
+  process.exit(0);
+}
+
+if (!API_KEY) {
+  console.error('\n  Error: SHARNIX_API_KEY is not set.\n');
+  console.error('  Get your key at: https://relay.sharnix.com/app/settings\n');
+  process.exit(1);
+}
+
+if (isNaN(port) || port < 1 || port > 65535) {
+  console.error(`\n  Error: invalid port "${get('--port') || get('-p')}"\n`);
+  process.exit(1);
+}
+
+// ── Config persistence ────────────────────────────────────────────────────────
+const CONFIG_DIR = path.join(os.homedir(), '.sharnix');
+if (!fs.existsSync(CONFIG_DIR)) fs.mkdirSync(CONFIG_DIR, { recursive: true });
+
+function loadJson(file) {
+  try { return JSON.parse(fs.readFileSync(file, 'utf8')); } catch { return null; }
+}
+function saveJson(file, data) {
+  fs.writeFileSync(file, JSON.stringify(data, null, 2), { mode: 0o600 });
+}
+
+// Stable tunnel ID per working directory — same tunnel across restarts
+function getOrCreateTunnelId() {
+  const cwdHash = createHash('sha256').update(process.cwd()).digest('hex').slice(0, 16);
+  const f = path.join(CONFIG_DIR, `tunnel-${cwdHash}.json`);
+  const existing = loadJson(f);
+  if (existing?.tunnelId) return existing.tunnelId;
+  const tunnelId = randomBytes(16).toString('hex');
+  saveJson(f, { tunnelId, cwd: process.cwd(), createdAt: new Date().toISOString() });
+  return tunnelId;
+}
+
+// ── REST API helper ───────────────────────────────────────────────────────────
+async function api(method, urlPath, body) {
+  const res = await fetch(`${RELAY_BASE}${urlPath}`, {
+    method,
+    headers: { 'Content-Type': 'application/json', 'Authorization': `Bearer ${API_KEY}` },
+    body: body ? JSON.stringify(body) : undefined,
+  });
+  const data = await res.json().catch(() => ({}));
+  if (!res.ok) throw new Error(data.error || `HTTP ${res.status}`);
+  return data;
+}
+
+// ── Bootstrap: create agent credentials once, keyed to this API key ───────────
+async function bootstrap() {
+  const credsFile = path.join(CONFIG_DIR, 'agent.json');
+  const stored = loadJson(credsFile);
+  if (stored?.agentId && stored?.secret && stored?.apiKeyPrefix === API_KEY.slice(0, 12)) {
+    return stored;
+  }
+
+  process.stdout.write('  Setting up agent…');
+  const orgs = await api('GET', '/api/v1/orgs');
+  if (!orgs.length) throw new Error('No workspaces found. Sign up at relay.sharnix.com first.');
+  const orgSlug = orgs[0].slug;
+
+  const agent = await api('POST', `/api/v1/orgs/${orgSlug}/agents`, { name: agentName });
+  const creds = {
+    agentId: agent.agent_id,
+    secret: agent.secret,
+    orgSlug,
+    apiKeyPrefix: API_KEY.slice(0, 12),
+  };
+  saveJson(credsFile, creds);
+  process.stdout.write(` done (${agent.agent_id})\n`);
+  return creds;
+}
+
+// ── WebSocket tunnel ──────────────────────────────────────────────────────────
+const MSG = {
+  REGISTER: 'REGISTER', RESPONSE: 'RESPONSE',
+  HEARTBEAT: 'HEARTBEAT', REQUEST: 'REQUEST', ACK: 'ACK',
+};
+
+let ws = null;
+let heartbeatTimer = null;
+let reconnectTimer = null;
+let isShuttingDown = false;
+let creds = null;
+const tunnelId = getOrCreateTunnelId();
+
+function connect() {
+  if (isShuttingDown) return;
+  const wsUrl = `${WS_BASE}?agentId=${encodeURIComponent(creds.agentId)}&secret=${encodeURIComponent(creds.secret)}`;
+  ws = new WebSocket(wsUrl);
+
+  ws.on('open', () => {
+    clearTimeout(reconnectTimer);
+    ws.send(JSON.stringify({ type: MSG.REGISTER, tunnelId, label: label || agentName }));
+    startHeartbeat();
+    const tunnelUrl = `${RELAY_BASE}/p/${tunnelId}`;
+    console.log(`\n  Connected!\n`);
+    console.log(`  Tunnel URL  : ${tunnelUrl}`);
+    console.log(`  Forwarding  : localhost:${port}`);
+    console.log(`\n  Create a share link:`);
+    console.log(`    Dashboard  → relay.sharnix.com/app/tunnels/${tunnelId}`);
+    console.log(`    MCP agent  → "create a share link for my tunnel"\n`);
+    console.log(`  Press Ctrl+C to disconnect.\n`);
+  });
+
+  ws.on('message', async (data) => {
+    let msg;
+    try { msg = JSON.parse(data); } catch { return; }
+    if (msg.type === MSG.REQUEST) await forwardRequest(msg);
+  });
+
+  ws.on('close', (code) => {
+    clearInterval(heartbeatTimer);
+    if (!isShuttingDown) {
+      process.stdout.write(`\r  Disconnected (${code}). Reconnecting in 5s…`);
+      reconnectTimer = setTimeout(connect, 5000);
+    }
+  });
+
+  ws.on('error', (err) => {
+    if (!isShuttingDown) console.error(`\n  Tunnel error: ${err.message}`);
+  });
+}
+
+function startHeartbeat() {
+  clearInterval(heartbeatTimer);
+  heartbeatTimer = setInterval(() => {
+    if (ws?.readyState === WebSocket.OPEN) {
+      ws.send(JSON.stringify({ type: MSG.HEARTBEAT, agentId: creds.agentId }));
+    }
+  }, 20000);
+}
+
+async function forwardRequest(msg) {
+  const { requestId, payload } = msg;
+
+  const headers = { ...payload.headers };
+  delete headers['host'];
+  delete headers['connection'];
+  delete headers['transfer-encoding'];
+
+  const options = {
+    hostname: '127.0.0.1',
+    port,
+    path: payload.path || '/',
+    method: payload.method || 'GET',
+    headers,
+  };
+
+  const proxyReq = http.request(options, (proxyRes) => {
+    const chunks = [];
+    proxyRes.on('data', (c) => chunks.push(c));
+    proxyRes.on('end', () => {
+      sendResponse(requestId, {
+        statusCode: proxyRes.statusCode,
+        headers: proxyRes.headers,
+        body: Buffer.concat(chunks).toString('base64'),
+        bodyEncoding: 'base64',
+      });
+    });
+    proxyRes.on('error', (err) => {
+      sendResponse(requestId, { statusCode: 502, headers: { 'content-type': 'text/plain' }, body: err.message });
+    });
+  });
+
+  proxyReq.setTimeout(25000, () => {
+    proxyReq.destroy();
+    sendResponse(requestId, { statusCode: 504, headers: { 'content-type': 'text/plain' }, body: 'Gateway timeout' });
+  });
+
+  proxyReq.on('error', (err) => {
+    sendResponse(requestId, { statusCode: 502, headers: { 'content-type': 'text/plain' }, body: err.message });
+  });
+
+  if (payload.body) proxyReq.write(Buffer.from(payload.body, 'base64'));
+  proxyReq.end();
+}
+
+function sendResponse(requestId, response) {
+  if (ws?.readyState === WebSocket.OPEN) {
+    ws.send(JSON.stringify({ type: MSG.RESPONSE, requestId, tunnelId, payload: response }));
+  }
+}
+
+// ── Graceful shutdown ─────────────────────────────────────────────────────────
+process.on('SIGINT', () => {
+  isShuttingDown = true;
+  console.log('\n  Disconnecting…');
+  clearInterval(heartbeatTimer);
+  clearTimeout(reconnectTimer);
+  if (ws) ws.close();
+  process.exit(0);
+});
+
+// ── Main ──────────────────────────────────────────────────────────────────────
+console.log(`\n  Sharnix Agent  →  localhost:${port}\n`);
+
+bootstrap()
+  .then((c) => { creds = c; connect(); })
+  .catch((err) => { console.error(`\n  Error: ${err.message}\n`); process.exit(1); });

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "@sharnix/agent",
+  "version": "1.0.0",
+  "description": "Tunnel your local app through Sharnix — share previews with one command",
+  "keywords": ["tunnel", "preview", "sharing", "localhost", "sharnix"],
+  "homepage": "https://relay.sharnix.com",
+  "license": "MIT",
+  "engines": { "node": ">=18" },
+  "main": "./index.js",
+  "bin": {
+    "sharnix-agent": "./index.js",
+    "sharnix": "./index.js"
+  },
+  "files": ["index.js"],
+  "dependencies": {
+    "ws": "^8.18.0"
+  }
+}