pi-to-chrome 0.1.2

package/tools/read-console.ts

@@ -0,0 +1,74 @@
+/**
+ * chrome_read_console - Read console messages from the buffer
+ *
+ * Returns messages captured via CDP event listener since chrome-start.
+ * No DevTools window needed.
+ */
+
+import type { ToolDefinition, ToolDeps } from '../core/types';
+import { Type } from '@sinclair/typebox';
+import type { ConsoleLevel } from '../console-buffer';
+
+export const readConsoleTool: ToolDefinition<{
+  level?: ConsoleLevel | 'all';
+  limit?: number;
+}> = {
+  name: 'chrome_read_console',
+  label: 'Chrome Read Console',
+  description: '读取页面的 console 日志消息，支持按级别过滤（log/warn/error/info）。',
+  promptSnippet: '读取页面的 console 日志',
+  promptGuidelines: [
+    '【排错时的第一反应】当页面行为异常或功能不工作时，先用 chrome_read_console 查看 error 级别日志，定位 JS 报错。',
+    '用 level 参数过滤：调试 JS 报错用 "error"，查警告用 "warn"，看完整日志用 "all"。',
+    '在修改代码后刷新页面，再用 chrome_read_console 确认错误是否消失。'
+  ],
+  parameters: Type.Object({
+    level: Type.Optional(Type.Union([
+      Type.Literal('log'),
+      Type.Literal('warn'),
+      Type.Literal('error'),
+      Type.Literal('info'),
+      Type.Literal('all')
+    ])),
+    limit: Type.Optional(Type.Number({ minimum: 1, maximum: 500, default: 50 }))
+  }),
+  async execute(page, params, deps?: ToolDeps) {
+    const consoleBuffer = deps?.consoleBuffer;
+    if (!consoleBuffer) throw new Error('ConsoleBuffer not available');
+
+    const level = (params.level || 'all') as ConsoleLevel | 'all';
+    const limit = params.limit ?? 50;
+
+    const messages = consoleBuffer.getMessages(level, limit);
+
+    const typeEmoji: Record<string, string> = {
+      log: '📝',
+      warn: '⚠️',
+      error: '❌',
+      info: 'ℹ️'
+    };
+
+    const lines = messages.map(m => {
+      const time = new Date(m.timestamp).toLocaleTimeString('zh-CN', { hour12: false });
+      const shortUrl = m.url.length > 60 ? m.url.slice(0, 57) + '...' : m.url;
+      return `${typeEmoji[m.type] || '📝'} [${time}] ${m.text} (${shortUrl})`;
+    });
+
+    const totalCount = consoleBuffer.count;
+    const filtered = messages.length;
+    const errorCount = messages.filter(m => m.type === 'error').length;
+    const warnCount = messages.filter(m => m.type === 'warn').length;
+
+    const header = `缓冲区 ${totalCount} 条 | 返回 ${filtered} 条 (${errorCount} errors, ${warnCount} warnings)`;
+    const fullText = `${header}\n\n${lines.join('\n')}`;
+
+    const truncated = fullText.length > 10000
+      ? fullText.slice(0, 10000) + `\n\n... (截断，用更小的 limit 缩小范围)`
+      : fullText;
+
+    return {
+      content: [{ type: 'text', text: truncated || '暂无 console 日志（只捕获连接后的输出）' }],
+      details: { count: totalCount, filtered }
+    };
+  }
+};

package/tools/take-screenshot.ts

@@ -0,0 +1,89 @@
+/**
+ * chrome_take_screenshot - Take a screenshot of the current page
+ *
+ * Automatically scales down large viewports and restores after capture.
+ */
+
+import type { ToolDefinition } from '../core/types';
+import { Type } from '@sinclair/typebox';
+
+export const takeScreenshotTool: ToolDefinition<{
+  format?: 'png' | 'jpeg';
+  quality?: number;
+}> = {
+  name: 'chrome_take_screenshot',
+  label: 'Chrome Take Screenshot',
+  description: 'Take a screenshot of the current page. Returns base64 encoded image.',
+  promptSnippet: 'Take a screenshot of the current page',
+  promptGuidelines: [
+    'Use chrome_take_screenshot when the user asks to see what the page looks like visually.'
+  ],
+  parameters: Type.Object({
+    format: Type.Optional(Type.Union([Type.Literal('png'), Type.Literal('jpeg')])),
+    quality: Type.Optional(Type.Number({ minimum: 0, maximum: 100 }))
+  }),
+  async execute(page, params) {
+    const format = params.format || 'jpeg';
+    const quality = params.quality ?? 50;
+
+    // Save original viewport
+    const originalViewport = page.viewport();
+
+    try {
+      // Check viewport size and scale if needed
+      let viewport = originalViewport;
+      const maxDimension = 2000;
+
+      if (originalViewport) {
+        const width = originalViewport.width;
+        const height = originalViewport.height;
+
+        if (width > maxDimension || height > maxDimension) {
+          const scale = maxDimension / Math.max(width, height);
+          viewport = {
+            width: Math.round(width * scale),
+            height: Math.round(height * scale),
+            deviceScaleFactor: (originalViewport.deviceScaleFactor ?? 1) * scale
+          };
+          await page.setViewport(viewport);
+        }
+      }
+
+      // Take screenshot
+      const screenshot = await page.screenshot({
+        type: format,
+        encoding: 'base64',
+        ...(format === 'jpeg' ? { quality } : {})
+      });
+
+      // Restore original viewport
+      if (originalViewport) {
+        await page.setViewport(originalViewport);
+      }
+
+      const base64Data = typeof screenshot === 'string' ? screenshot : Buffer.from(screenshot as any).toString('base64');
+      const sizeBytes = Math.round(base64Data.length * 0.75); // Approximate decoded size
+      const sizeKB = Math.round(sizeBytes / 1024);
+
+      return {
+        content: [{ type: 'text', text: `截图已获取 (${format}, ${sizeKB}KB)` }],
+        details: {
+          format,
+          data: base64Data,
+          sizeBytes
+        }
+      };
+
+    } catch (error: any) {
+      // Ensure viewport is restored even on error
+      if (originalViewport) {
+        try {
+          await page.setViewport(originalViewport);
+        } catch (viewError) {
+          console.error('[pi-to-chrome] take-screenshot: viewport 恢复失败', viewError);
+        }
+      }
+      throw error;
+    }
+  }
+};