autokap 1.0.7 → 1.0.8

package/dist/ws-metrics-server.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Lightweight internal HTTP server that exposes WS server metrics
+ * for the monitoring cron running in the Next.js process.
+ *
+ * Binds to 127.0.0.1 only — not exposed externally.
+ */
+import { type Server } from 'node:http';
+import type { WebSocketServer } from 'ws';
+export declare function startMetricsServer(wss: WebSocketServer): Server;

package/dist/ws-metrics-server.js

@@ -0,0 +1,31 @@
+/**
+ * Lightweight internal HTTP server that exposes WS server metrics
+ * for the monitoring cron running in the Next.js process.
+ *
+ * Binds to 127.0.0.1 only — not exposed externally.
+ */
+import { createServer } from 'node:http';
+import { browserPool } from './browser-pool.js';
+import { logger } from './logger.js';
+const METRICS_PORT = parseInt(process.env.METRICS_PORT ?? '3002', 10);
+export function startMetricsServer(wss) {
+    const server = createServer((req, res) => {
+        if (req.method === 'GET' && req.url === '/metrics') {
+            const poolStats = browserPool.getStats();
+            const payload = {
+                browserPool: poolStats,
+                wsConnections: wss.clients.size,
+            };
+            res.writeHead(200, { 'Content-Type': 'application/json' });
+            res.end(JSON.stringify(payload));
+            return;
+        }
+        res.writeHead(404);
+        res.end();
+    });
+    server.listen(METRICS_PORT, '127.0.0.1', () => {
+        logger.info(`[Metrics] Internal metrics server on 127.0.0.1:${METRICS_PORT}/metrics`);
+    });
+    return server;
+}
+//# sourceMappingURL=ws-metrics-server.js.map

package/dist/ws-server.js

@@ -10,15 +10,30 @@ import { config as loadEnv } from 'dotenv';
 import { WebSocketServer } from 'ws';
 import { handleConnection } from './ws-handler.js';
 import { logger } from './logger.js';
+import { MAX_CONNECTIONS_PER_IP, getIpConnectionCount, incrementIpConnections, decrementIpConnections, } from './ws-connection-limits.js';
+import { startMetricsServer } from './ws-metrics-server.js';
+import { validateRequiredEnv } from './env-validation.js';
 loadEnv();
+validateRequiredEnv();
 const PORT = parseInt(process.env.WS_PORT ?? '3001', 10);
+function getClientIp(req) {
+    return req.headers['x-real-ip'] ?? req.socket.remoteAddress ?? 'unknown';
+}
 const wss = new WebSocketServer({
     port: PORT,
     path: '/ws',
     maxPayload: 10 * 1024 * 1024, // 10 MB max message (screenshots can be large)
 });
 wss.on('connection', (ws, req) => {
-    const ip = req.headers['x-real-ip'] ?? req.socket.remoteAddress ?? 'unknown';
+    const ip = getClientIp(req);
+    // Per-IP rate limit
+    const ipCount = getIpConnectionCount(ip);
+    if (ipCount >= MAX_CONNECTIONS_PER_IP) {
+        logger.warn(`[WS] Rejecting connection from ${ip} — IP limit reached (${ipCount})`);
+        ws.close(4029, 'Too many connections');
+        return;
+    }
+    incrementIpConnections(ip);
     logger.info(`[WS] New connection from ${ip}`);
     handleConnection(ws, req).catch(err => {
         logger.error(`[WS] Unhandled error: ${err.message}`);
@@ -28,16 +43,40 @@ wss.on('connection', (ws, req) => {
         catch { /* ignore */ }
     });
     ws.on('close', () => {
+        decrementIpConnections(ip);
         logger.info(`[WS] Connection closed from ${ip}`);
     });
 });
 wss.on('error', (err) => {
     logger.error(`[WS] Server error: ${err.message}`);
 });
+// Heartbeat — detect dead connections (NAT timeout, network drops)
+const PING_INTERVAL_MS = 30_000;
+const PONG_TIMEOUT_MS = 10_000;
+const aliveMap = new WeakMap();
+const heartbeatInterval = setInterval(() => {
+    for (const ws of wss.clients) {
+        if (aliveMap.get(ws) === false) {
+            logger.warn('[WS] Terminating unresponsive connection');
+            ws.terminate();
+            continue;
+        }
+        aliveMap.set(ws, false);
+        ws.ping();
+    }
+}, PING_INTERVAL_MS);
+wss.on('connection', (ws) => {
+    aliveMap.set(ws, true);
+    ws.on('pong', () => { aliveMap.set(ws, true); });
+});
+heartbeatInterval.unref();
 logger.success(`AutoKap WS server listening on port ${PORT} (path: /ws)`);
+// Internal metrics endpoint for the monitoring cron
+const metricsServer = startMetricsServer(wss);
 // Graceful shutdown
 process.on('SIGTERM', () => {
     logger.info('[WS] SIGTERM received, closing...');
+    metricsServer.close();
     wss.close(() => {
         logger.info('[WS] Server closed');
         process.exit(0);
@@ -45,6 +84,7 @@ process.on('SIGTERM', () => {
 });
 process.on('SIGINT', () => {
     logger.info('[WS] SIGINT received, closing...');
+    metricsServer.close();
     wss.close(() => {
         process.exit(0);
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "autokap",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "AI-powered CLI tool for capturing clean screenshots of websites",
   "type": "module",
   "main": "./dist/index.js",
@@ -14,14 +14,22 @@
       "types": "./dist/browser.d.ts",
       "default": "./dist/browser.js"
     },
-    "./agent": {
-      "types": "./dist/agent.d.ts",
-      "default": "./dist/agent.js"
-    },
     "./cookie-dismiss": {
       "types": "./dist/cookie-dismiss.d.ts",
       "default": "./dist/cookie-dismiss.js"
     },
+    "./dom-serializer": {
+      "types": "./dist/dom-serializer.d.ts",
+      "default": "./dist/dom-serializer.js"
+    },
+    "./dom-css-purger": {
+      "types": "./dist/dom-css-purger.d.ts",
+      "default": "./dist/dom-css-purger.js"
+    },
+    "./dom-font-inliner": {
+      "types": "./dist/dom-font-inliner.d.ts",
+      "default": "./dist/dom-font-inliner.js"
+    },
     "./logger": {
       "types": "./dist/logger.d.ts",
       "default": "./dist/logger.js"
@@ -30,6 +38,22 @@
       "types": "./dist/types.d.ts",
       "default": "./dist/types.js"
     },
+    "./execution-schema": {
+      "types": "./dist/execution-schema.d.ts",
+      "default": "./dist/execution-schema.js"
+    },
+    "./execution-types": {
+      "types": "./dist/execution-types.d.ts",
+      "default": "./dist/execution-types.js"
+    },
+    "./alt-text": {
+      "types": "./dist/alt-text.d.ts",
+      "default": "./dist/alt-text.js"
+    },
+    "./llm-provider": {
+      "types": "./dist/llm-provider.d.ts",
+      "default": "./dist/llm-provider.js"
+    },
     "./mockup": {
       "types": "./dist/mockup.d.ts",
       "default": "./dist/mockup.js"
@@ -50,42 +74,18 @@
       "types": "./dist/status-bar-l10n.d.ts",
       "default": "./dist/status-bar-l10n.js"
     },
-    "./element-capture": {
-      "types": "./dist/element-capture.d.ts",
-      "default": "./dist/element-capture.js"
-    },
-    "./prompts": {
-      "types": "./dist/prompts.d.ts",
-      "default": "./dist/prompts.js"
-    },
     "./session-profile": {
       "types": "./dist/session-profile.d.ts",
       "default": "./dist/session-profile.js"
     },
-    "./tools": {
-      "types": "./dist/tools.d.ts",
-      "default": "./dist/tools.js"
-    },
     "./security": {
       "types": "./dist/security.d.ts",
       "default": "./dist/security.js"
     },
-    "./video-agent": {
-      "types": "./dist/video-agent.d.ts",
-      "default": "./dist/video-agent.js"
-    },
-    "./video-planner": {
-      "types": "./dist/video-planner.d.ts",
-      "default": "./dist/video-planner.js"
-    },
     "./mouse-animation": {
       "types": "./dist/mouse-animation.d.ts",
       "default": "./dist/mouse-animation.js"
     },
-    "./clip-orchestrator": {
-      "types": "./dist/clip-orchestrator.d.ts",
-      "default": "./dist/clip-orchestrator.js"
-    },
     "./clip-postprocess": {
       "types": "./dist/clip-postprocess.d.ts",
       "default": "./dist/clip-postprocess.js"
@@ -94,6 +94,10 @@
       "types": "./dist/capture-alt-text.d.ts",
       "default": "./dist/capture-alt-text.js"
     },
+    "./version": {
+      "types": "./dist/version.d.ts",
+      "default": "./dist/version.js"
+    },
     "./capture-encryption": {
       "types": "./dist/capture-encryption.d.ts",
       "default": "./dist/capture-encryption.js"
@@ -122,10 +126,18 @@
       "types": "./dist/capture-request-plan.d.ts",
       "default": "./dist/capture-request-plan.js"
     },
+    "./capture-viewport-lock": {
+      "types": "./dist/capture-viewport-lock.d.ts",
+      "default": "./dist/capture-viewport-lock.js"
+    },
     "./capture-page-identity": {
       "types": "./dist/capture-page-identity.d.ts",
       "default": "./dist/capture-page-identity.js"
     },
+    "./capture-transition-engine": {
+      "types": "./dist/capture-transition-engine.d.ts",
+      "default": "./dist/capture-transition-engine.js"
+    },
     "./capture-variant-state": {
       "types": "./dist/capture-variant-state.d.ts",
       "default": "./dist/capture-variant-state.js"
@@ -166,16 +178,20 @@
       "types": "./dist/capture-studio-sync.d.ts",
       "default": "./dist/capture-studio-sync.js"
     },
-    "./hybrid-navigator": {
-      "types": "./dist/hybrid-navigator.d.ts",
-      "default": "./dist/hybrid-navigator.js"
+    "./cli-contract": {
+      "types": "./dist/cli-contract.d.ts",
+      "default": "./dist/cli-contract.js"
+    },
+    "./provider-config": {
+      "types": "./dist/provider-config.d.ts",
+      "default": "./dist/provider-config.js"
     }
   },
   "bin": {
-    "autokap": "./dist/cli.js"
+    "autokap": "dist/cli.js"
   },
   "scripts": {
-    "postinstall": "npx playwright install chromium 2>/dev/null || true"
+    "postinstall": "node -e \"try{require('child_process').execSync('npx playwright install chromium',{stdio:'inherit'})}catch(e){console.warn('AutoKap: Playwright Chromium install failed. Run `npx playwright install chromium` manually.',e.message||'')}\""
   },
   "engines": {
     "node": ">=20"
@@ -191,6 +207,7 @@
     "assets/frames/MacBook*",
     "assets/frames/Status*",
     "assets/icons",
+    "assets/cursors",
     "assets/skill"
   ],
   "keywords": [