opc-agent 1.3.2 → 2.0.0

package/src/tools/builtin/index.ts

@@ -0,0 +1,28 @@
+import type { MCPTool } from '../mcp';
+import { fileTool } from './file';
+import { webTool } from './web';
+import { shellTool } from './shell';
+import { datetimeTool } from './datetime';
+
+export { fileTool, webTool, shellTool, datetimeTool };
+
+const ALL_BUILTIN_TOOLS: MCPTool[] = [fileTool, webTool, shellTool, datetimeTool];
+
+const BUILTIN_MAP = new Map<string, MCPTool>(
+  ALL_BUILTIN_TOOLS.map(t => [t.name, t])
+);
+
+/**
+ * Get all built-in tools.
+ */
+export function getBuiltinTools(): MCPTool[] {
+  return [...ALL_BUILTIN_TOOLS];
+}
+
+/**
+ * Get specific built-in tools by name. If no names given, returns all.
+ */
+export function getBuiltinToolsByName(names?: string[]): MCPTool[] {
+  if (!names || names.length === 0) return getBuiltinTools();
+  return names.map(n => BUILTIN_MAP.get(n)).filter((t): t is MCPTool => !!t);
+}

package/src/tools/builtin/shell.ts

@@ -0,0 +1,43 @@
+import { execSync } from 'child_process';
+import * as path from 'path';
+import type { MCPTool, MCPToolResult } from '../mcp';
+
+export const shellTool: MCPTool = {
+  name: 'shell_exec',
+  description: 'Execute a shell command (sandboxed to workspace)',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      command: { type: 'string' },
+      timeout: { type: 'number', default: 30000 },
+    },
+    required: ['command'],
+  },
+  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
+    const command = input.command as string;
+    const timeout = (input.timeout as number) || 30000;
+    const workspace = process.cwd();
+
+    // Block path traversal attempts
+    if (command.includes('..')) {
+      return { content: 'Commands with ".." are not allowed for security', isError: true };
+    }
+
+    try {
+      const output = execSync(command, {
+        cwd: workspace,
+        timeout,
+        encoding: 'utf-8',
+        maxBuffer: 1024 * 1024,
+        stdio: ['pipe', 'pipe', 'pipe'],
+      });
+      const result = (output || '').slice(0, 5000);
+      return { content: result || '(no output)', isError: false };
+    } catch (err: any) {
+      const stderr = err.stderr ? String(err.stderr).slice(0, 2500) : '';
+      const stdout = err.stdout ? String(err.stdout).slice(0, 2500) : '';
+      const output = [stdout, stderr].filter(Boolean).join('\n') || err.message;
+      return { content: `Command failed: ${output.slice(0, 5000)}`, isError: true };
+    }
+  },
+};

package/src/tools/builtin/web.ts

@@ -0,0 +1,35 @@
+import type { MCPTool, MCPToolResult } from '../mcp';
+
+export const webTool: MCPTool = {
+  name: 'web_fetch',
+  description: 'Fetch content from a URL',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      url: { type: 'string' },
+      method: { type: 'string', enum: ['GET', 'POST'], default: 'GET' },
+      maxLength: { type: 'number', default: 5000 },
+    },
+    required: ['url'],
+  },
+  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
+    const url = input.url as string;
+    const method = (input.method as string) || 'GET';
+    const maxLength = (input.maxLength as number) || 5000;
+
+    try {
+      const response = await fetch(url, { method, signal: AbortSignal.timeout(15000) });
+      const text = await response.text();
+      const truncated = text.length > maxLength ? text.slice(0, maxLength) + '\n...[truncated]' : text;
+      return {
+        content: `Status: ${response.status}\n\n${truncated}`,
+        isError: false,
+      };
+    } catch (err) {
+      return {
+        content: `Fetch error: ${err instanceof Error ? err.message : String(err)}`,
+        isError: true,
+      };
+    }
+  },
+};

package/src/tools/calculator.ts

@@ -1,73 +1,73 @@
-import type { MCPTool, MCPToolResult } from './mcp';
-
-/**
- * Calculator Tool — v0.8.0
- * Safe math expression evaluation as an LLM function tool.
- */
-export const CalculatorTool: MCPTool = {
-  name: 'calculator',
-  description: 'Evaluate a mathematical expression. Supports basic arithmetic, powers, sqrt, abs, min, max, round, ceil, floor, PI, E.',
-  inputSchema: {
-    type: 'object',
-    properties: {
-      expression: {
-        type: 'string',
-        description: 'Mathematical expression to evaluate, e.g. "2 + 3 * 4" or "sqrt(144) + PI"',
-      },
-    },
-    required: ['expression'],
-  },
-
-  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
-    const expr = String(input.expression ?? '');
-    try {
-      const result = safeEval(expr);
-      return { content: String(result) };
-    } catch (err) {
-      return { content: `Error: ${(err as Error).message}`, isError: true };
-    }
-  },
-};
-
-/** Safe math evaluator — no eval(), no arbitrary code */
-function safeEval(expr: string): number {
-  // Whitelist: digits, operators, parens, dots, commas, spaces, and known functions
-  const sanitized = expr.replace(/\s+/g, '');
-  const allowed = /^[0-9+\-*/().,%^a-zA-Z_]+$/;
-  if (!allowed.test(sanitized)) {
-    throw new Error('Invalid characters in expression');
-  }
-
-  // Replace known math functions/constants
-  const prepared = sanitized
-    .replace(/\bPI\b/gi, String(Math.PI))
-    .replace(/\bE\b/g, String(Math.E))
-    .replace(/\bsqrt\b/gi, 'Math.sqrt')
-    .replace(/\babs\b/gi, 'Math.abs')
-    .replace(/\bmin\b/gi, 'Math.min')
-    .replace(/\bmax\b/gi, 'Math.max')
-    .replace(/\bround\b/gi, 'Math.round')
-    .replace(/\bceil\b/gi, 'Math.ceil')
-    .replace(/\bfloor\b/gi, 'Math.floor')
-    .replace(/\bpow\b/gi, 'Math.pow')
-    .replace(/\blog\b/gi, 'Math.log')
-    .replace(/\blog10\b/gi, 'Math.log10')
-    .replace(/\bsin\b/gi, 'Math.sin')
-    .replace(/\bcos\b/gi, 'Math.cos')
-    .replace(/\btan\b/gi, 'Math.tan')
-    .replace(/\^/g, '**');
-
-  // Block anything that isn't math
-  if (/[a-zA-Z_]/.test(prepared.replace(/Math\.\w+/g, ''))) {
-    throw new Error('Unsupported function or variable in expression');
-  }
-
-  // Use Function constructor with restricted scope
-  const fn = new Function('Math', `"use strict"; return (${prepared});`);
-  const result = fn(Math);
-
-  if (typeof result !== 'number' || !isFinite(result)) {
-    throw new Error('Expression did not evaluate to a finite number');
-  }
-  return result;
-}
+import type { MCPTool, MCPToolResult } from './mcp';
+
+/**
+ * Calculator Tool — v0.8.0
+ * Safe math expression evaluation as an LLM function tool.
+ */
+export const CalculatorTool: MCPTool = {
+  name: 'calculator',
+  description: 'Evaluate a mathematical expression. Supports basic arithmetic, powers, sqrt, abs, min, max, round, ceil, floor, PI, E.',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      expression: {
+        type: 'string',
+        description: 'Mathematical expression to evaluate, e.g. "2 + 3 * 4" or "sqrt(144) + PI"',
+      },
+    },
+    required: ['expression'],
+  },
+
+  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
+    const expr = String(input.expression ?? '');
+    try {
+      const result = safeEval(expr);
+      return { content: String(result) };
+    } catch (err) {
+      return { content: `Error: ${(err as Error).message}`, isError: true };
+    }
+  },
+};
+
+/** Safe math evaluator — no eval(), no arbitrary code */
+function safeEval(expr: string): number {
+  // Whitelist: digits, operators, parens, dots, commas, spaces, and known functions
+  const sanitized = expr.replace(/\s+/g, '');
+  const allowed = /^[0-9+\-*/().,%^a-zA-Z_]+$/;
+  if (!allowed.test(sanitized)) {
+    throw new Error('Invalid characters in expression');
+  }
+
+  // Replace known math functions/constants
+  const prepared = sanitized
+    .replace(/\bPI\b/gi, String(Math.PI))
+    .replace(/\bE\b/g, String(Math.E))
+    .replace(/\bsqrt\b/gi, 'Math.sqrt')
+    .replace(/\babs\b/gi, 'Math.abs')
+    .replace(/\bmin\b/gi, 'Math.min')
+    .replace(/\bmax\b/gi, 'Math.max')
+    .replace(/\bround\b/gi, 'Math.round')
+    .replace(/\bceil\b/gi, 'Math.ceil')
+    .replace(/\bfloor\b/gi, 'Math.floor')
+    .replace(/\bpow\b/gi, 'Math.pow')
+    .replace(/\blog\b/gi, 'Math.log')
+    .replace(/\blog10\b/gi, 'Math.log10')
+    .replace(/\bsin\b/gi, 'Math.sin')
+    .replace(/\bcos\b/gi, 'Math.cos')
+    .replace(/\btan\b/gi, 'Math.tan')
+    .replace(/\^/g, '**');
+
+  // Block anything that isn't math
+  if (/[a-zA-Z_]/.test(prepared.replace(/Math\.\w+/g, ''))) {
+    throw new Error('Unsupported function or variable in expression');
+  }
+
+  // Use Function constructor with restricted scope
+  const fn = new Function('Math', `"use strict"; return (${prepared});`);
+  const result = fn(Math);
+
+  if (typeof result !== 'number' || !isFinite(result)) {
+    throw new Error('Expression did not evaluate to a finite number');
+  }
+  return result;
+}

package/src/tools/datetime.ts

@@ -1,149 +1,149 @@
-import type { MCPTool, MCPToolResult } from './mcp';
-
-/**
- * DateTime Tool — v0.8.0
- * Date/time operations as an LLM function tool.
- */
-export const DateTimeTool: MCPTool = {
-  name: 'datetime',
-  description: 'Get current date/time, format dates, calculate date differences, or add/subtract time.',
-  inputSchema: {
-    type: 'object',
-    properties: {
-      operation: {
-        type: 'string',
-        enum: ['now', 'format', 'diff', 'add', 'parse'],
-        description: 'Operation to perform',
-      },
-      date: {
-        type: 'string',
-        description: 'Date string (ISO 8601 or common format)',
-      },
-      date2: {
-        type: 'string',
-        description: 'Second date for diff operation',
-      },
-      format: {
-        type: 'string',
-        description: 'Output format: iso, date, time, datetime, unix, relative',
-      },
-      amount: {
-        type: 'number',
-        description: 'Amount to add (can be negative)',
-      },
-      unit: {
-        type: 'string',
-        enum: ['milliseconds', 'seconds', 'minutes', 'hours', 'days', 'weeks', 'months', 'years'],
-        description: 'Unit for add operation',
-      },
-      timezone: {
-        type: 'string',
-        description: 'IANA timezone (e.g. Asia/Shanghai, America/New_York)',
-      },
-    },
-    required: ['operation'],
-  },
-
-  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
-    try {
-      const op = String(input.operation);
-      const tz = input.timezone as string | undefined;
-
-      switch (op) {
-        case 'now': {
-          const now = new Date();
-          const fmt = (input.format as string) ?? 'iso';
-          return { content: formatDate(now, fmt, tz) };
-        }
-
-        case 'format': {
-          const d = parseDate(input.date as string);
-          const fmt = (input.format as string) ?? 'iso';
-          return { content: formatDate(d, fmt, tz) };
-        }
-
-        case 'diff': {
-          const d1 = parseDate(input.date as string);
-          const d2 = input.date2 ? parseDate(input.date2 as string) : new Date();
-          const diffMs = d2.getTime() - d1.getTime();
-          const days = Math.floor(Math.abs(diffMs) / 86400000);
-          const hours = Math.floor((Math.abs(diffMs) % 86400000) / 3600000);
-          const mins = Math.floor((Math.abs(diffMs) % 3600000) / 60000);
-          const sign = diffMs < 0 ? '-' : '';
-          return {
-            content: JSON.stringify({
-              milliseconds: diffMs,
-              readable: `${sign}${days}d ${hours}h ${mins}m`,
-              days: diffMs / 86400000,
-            }),
-          };
-        }
-
-        case 'add': {
-          const d = input.date ? parseDate(input.date as string) : new Date();
-          const amount = Number(input.amount ?? 0);
-          const unit = String(input.unit ?? 'days');
-          const result = addTime(d, amount, unit);
-          const fmt = (input.format as string) ?? 'iso';
-          return { content: formatDate(result, fmt, tz) };
-        }
-
-        case 'parse': {
-          const d = parseDate(input.date as string);
-          return {
-            content: JSON.stringify({
-              iso: d.toISOString(),
-              unix: d.getTime(),
-              year: d.getFullYear(),
-              month: d.getMonth() + 1,
-              day: d.getDate(),
-              weekday: ['Sunday', 'Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday'][d.getDay()],
-            }),
-          };
-        }
-
-        default:
-          return { content: `Unknown operation: ${op}`, isError: true };
-      }
-    } catch (err) {
-      return { content: `Error: ${(err as Error).message}`, isError: true };
-    }
-  },
-};
-
-function parseDate(s: string): Date {
-  const d = new Date(s);
-  if (isNaN(d.getTime())) throw new Error(`Invalid date: ${s}`);
-  return d;
-}
-
-function formatDate(d: Date, fmt: string, tz?: string): string {
-  if (tz) {
-    const localeStr = d.toLocaleString('en-US', { timeZone: tz });
-    d = new Date(localeStr);
-  }
-  switch (fmt) {
-    case 'iso': return d.toISOString();
-    case 'date': return d.toISOString().split('T')[0];
-    case 'time': return d.toTimeString().split(' ')[0];
-    case 'datetime': return d.toISOString().replace('T', ' ').replace(/\.\d+Z$/, '');
-    case 'unix': return String(d.getTime());
-    default: return d.toISOString();
-  }
-}
-
-function addTime(d: Date, amount: number, unit: string): Date {
-  const result = new Date(d);
-  switch (unit) {
-    case 'milliseconds': result.setTime(result.getTime() + amount); break;
-    case 'seconds': result.setTime(result.getTime() + amount * 1000); break;
-    case 'minutes': result.setTime(result.getTime() + amount * 60000); break;
-    case 'hours': result.setTime(result.getTime() + amount * 3600000); break;
-    case 'days': result.setDate(result.getDate() + amount); break;
-    case 'weeks': result.setDate(result.getDate() + amount * 7); break;
-    case 'months': result.setMonth(result.getMonth() + amount); break;
-    case 'years': result.setFullYear(result.getFullYear() + amount); break;
-    default: throw new Error(`Unknown unit: ${unit}`);
-  }
-  return result;
-}
+import type { MCPTool, MCPToolResult } from './mcp';
+
+/**
+ * DateTime Tool — v0.8.0
+ * Date/time operations as an LLM function tool.
+ */
+export const DateTimeTool: MCPTool = {
+  name: 'datetime',
+  description: 'Get current date/time, format dates, calculate date differences, or add/subtract time.',
+  inputSchema: {
+    type: 'object',
+    properties: {
+      operation: {
+        type: 'string',
+        enum: ['now', 'format', 'diff', 'add', 'parse'],
+        description: 'Operation to perform',
+      },
+      date: {
+        type: 'string',
+        description: 'Date string (ISO 8601 or common format)',
+      },
+      date2: {
+        type: 'string',
+        description: 'Second date for diff operation',
+      },
+      format: {
+        type: 'string',
+        description: 'Output format: iso, date, time, datetime, unix, relative',
+      },
+      amount: {
+        type: 'number',
+        description: 'Amount to add (can be negative)',
+      },
+      unit: {
+        type: 'string',
+        enum: ['milliseconds', 'seconds', 'minutes', 'hours', 'days', 'weeks', 'months', 'years'],
+        description: 'Unit for add operation',
+      },
+      timezone: {
+        type: 'string',
+        description: 'IANA timezone (e.g. Asia/Shanghai, America/New_York)',
+      },
+    },
+    required: ['operation'],
+  },
+
+  async execute(input: Record<string, unknown>): Promise<MCPToolResult> {
+    try {
+      const op = String(input.operation);
+      const tz = input.timezone as string | undefined;
+
+      switch (op) {
+        case 'now': {
+          const now = new Date();
+          const fmt = (input.format as string) ?? 'iso';
+          return { content: formatDate(now, fmt, tz) };
+        }
+
+        case 'format': {
+          const d = parseDate(input.date as string);
+          const fmt = (input.format as string) ?? 'iso';
+          return { content: formatDate(d, fmt, tz) };
+        }
+
+        case 'diff': {
+          const d1 = parseDate(input.date as string);
+          const d2 = input.date2 ? parseDate(input.date2 as string) : new Date();
+          const diffMs = d2.getTime() - d1.getTime();
+          const days = Math.floor(Math.abs(diffMs) / 86400000);
+          const hours = Math.floor((Math.abs(diffMs) % 86400000) / 3600000);
+          const mins = Math.floor((Math.abs(diffMs) % 3600000) / 60000);
+          const sign = diffMs < 0 ? '-' : '';
+          return {
+            content: JSON.stringify({
+              milliseconds: diffMs,
+              readable: `${sign}${days}d ${hours}h ${mins}m`,
+              days: diffMs / 86400000,
+            }),
+          };
+        }
+
+        case 'add': {
+          const d = input.date ? parseDate(input.date as string) : new Date();
+          const amount = Number(input.amount ?? 0);
+          const unit = String(input.unit ?? 'days');
+          const result = addTime(d, amount, unit);
+          const fmt = (input.format as string) ?? 'iso';
+          return { content: formatDate(result, fmt, tz) };
+        }
+
+        case 'parse': {
+          const d = parseDate(input.date as string);
+          return {
+            content: JSON.stringify({
+              iso: d.toISOString(),
+              unix: d.getTime(),
+              year: d.getFullYear(),
+              month: d.getMonth() + 1,
+              day: d.getDate(),
+              weekday: ['Sunday', 'Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday'][d.getDay()],
+            }),
+          };
+        }
+
+        default:
+          return { content: `Unknown operation: ${op}`, isError: true };
+      }
+    } catch (err) {
+      return { content: `Error: ${(err as Error).message}`, isError: true };
+    }
+  },
+};
+
+function parseDate(s: string): Date {
+  const d = new Date(s);
+  if (isNaN(d.getTime())) throw new Error(`Invalid date: ${s}`);
+  return d;
+}
+
+function formatDate(d: Date, fmt: string, tz?: string): string {
+  if (tz) {
+    const localeStr = d.toLocaleString('en-US', { timeZone: tz });
+    d = new Date(localeStr);
+  }
+  switch (fmt) {
+    case 'iso': return d.toISOString();
+    case 'date': return d.toISOString().split('T')[0];
+    case 'time': return d.toTimeString().split(' ')[0];
+    case 'datetime': return d.toISOString().replace('T', ' ').replace(/\.\d+Z$/, '');
+    case 'unix': return String(d.getTime());
+    default: return d.toISOString();
+  }
+}
+
+function addTime(d: Date, amount: number, unit: string): Date {
+  const result = new Date(d);
+  switch (unit) {
+    case 'milliseconds': result.setTime(result.getTime() + amount); break;
+    case 'seconds': result.setTime(result.getTime() + amount * 1000); break;
+    case 'minutes': result.setTime(result.getTime() + amount * 60000); break;
+    case 'hours': result.setTime(result.getTime() + amount * 3600000); break;
+    case 'days': result.setDate(result.getDate() + amount); break;
+    case 'weeks': result.setDate(result.getDate() + amount * 7); break;
+    case 'months': result.setMonth(result.getMonth() + amount); break;
+    case 'years': result.setFullYear(result.getFullYear() + amount); break;
+    default: throw new Error(`Unknown unit: ${unit}`);
+  }
+  return result;
+}