@musashishao/agent-kit 1.2.2

package/.agent/mcp/templates/custom/src/index.ts.template

@@ -0,0 +1,133 @@
+#!/usr/bin/env node
+/**
+ * {{SERVER_NAME}} - Custom MCP Server
+ * 
+ * A minimal template to build your own MCP server.
+ * 
+ * Generated by Agent Kit MCP Generator
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+
+const server = new McpServer({
+  name: "{{SERVER_NAME}}",
+  version: "1.0.0",
+});
+
+// ============================================================================
+// Your Tools Here
+// ============================================================================
+
+/**
+ * Example tool - Hello World
+ * Replace this with your own tools
+ */
+server.tool(
+  "hello",
+  "A simple hello world tool",
+  {
+    name: z.string().optional().describe("Name to greet"),
+  },
+  async ({ name }) => {
+    const greeting = name ? `Hello, ${name}!` : "Hello, World!";
+    
+    return {
+      content: [{
+        type: "text",
+        text: greeting,
+      }],
+    };
+  }
+);
+
+/**
+ * Example tool with complex input
+ */
+server.tool(
+  "calculate",
+  "Perform a calculation",
+  {
+    operation: z.enum(["add", "subtract", "multiply", "divide"])
+      .describe("The operation to perform"),
+    a: z.number().describe("First number"),
+    b: z.number().describe("Second number"),
+  },
+  async ({ operation, a, b }) => {
+    let result: number;
+    
+    switch (operation) {
+      case "add":
+        result = a + b;
+        break;
+      case "subtract":
+        result = a - b;
+        break;
+      case "multiply":
+        result = a * b;
+        break;
+      case "divide":
+        if (b === 0) {
+          return {
+            content: [{
+              type: "text",
+              text: JSON.stringify({ error: "Cannot divide by zero" }),
+            }],
+          };
+        }
+        result = a / b;
+        break;
+    }
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify({ 
+          operation, 
+          a, 
+          b, 
+          result 
+        }),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Your Resources Here
+// ============================================================================
+
+/**
+ * Example resource
+ */
+server.resource(
+  "server-info",
+  "info://server",
+  async (uri) => ({
+    contents: [{
+      uri: uri.href,
+      mimeType: "application/json",
+      text: JSON.stringify({
+        name: "{{SERVER_NAME}}",
+        version: "1.0.0",
+        timestamp: new Date().toISOString(),
+      }),
+    }],
+  })
+);
+
+// ============================================================================
+// Start Server
+// ============================================================================
+
+async function main() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error("{{SERVER_NAME}} MCP Server running on stdio");
+}
+
+main().catch((error) => {
+  console.error("Failed to start MCP server:", error);
+  process.exit(1);
+});

package/.agent/mcp/templates/database/src/index.ts.template

@@ -0,0 +1,273 @@
+#!/usr/bin/env node
+/**
+ * {{SERVER_NAME}} - Database MCP Server
+ * 
+ * Query and manage databases from your AI assistant.
+ * Supports: SQLite, PostgreSQL (via connection string)
+ * 
+ * Generated by Agent Kit MCP Generator
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+import * as fs from "fs";
+import * as path from "path";
+
+// Configuration
+const DB_TYPE = process.env.DB_TYPE || "sqlite";
+const DB_PATH = process.env.DB_PATH || "./database.db";
+const DB_URL = process.env.DATABASE_URL || "";
+
+const server = new McpServer({
+  name: "{{SERVER_NAME}}",
+  version: "1.0.0",
+});
+
+// ============================================================================
+// SQLite Implementation
+// ============================================================================
+
+interface QueryResult {
+  success: boolean;
+  rows?: Record<string, unknown>[];
+  rowCount?: number;
+  error?: string;
+}
+
+async function executeSqliteQuery(sql: string, params: unknown[] = []): Promise<QueryResult> {
+  // Dynamic import for sqlite3
+  try {
+    const sqlite3 = await import("better-sqlite3");
+    const db = new sqlite3.default(DB_PATH);
+    
+    try {
+      // Check if it's a SELECT query
+      const isSelect = sql.trim().toUpperCase().startsWith("SELECT");
+      
+      if (isSelect) {
+        const stmt = db.prepare(sql);
+        const rows = stmt.all(...params) as Record<string, unknown>[];
+        return { success: true, rows, rowCount: rows.length };
+      } else {
+        const stmt = db.prepare(sql);
+        const result = stmt.run(...params);
+        return { 
+          success: true, 
+          rowCount: result.changes,
+          rows: [{ changes: result.changes, lastInsertRowid: result.lastInsertRowid }]
+        };
+      }
+    } finally {
+      db.close();
+    }
+  } catch (error) {
+    return { 
+      success: false, 
+      error: `Database error: ${error instanceof Error ? error.message : "Unknown error"}` 
+    };
+  }
+}
+
+// ============================================================================
+// Tool: db_query
+// ============================================================================
+
+server.tool(
+  "db_query",
+  "Execute a read-only SQL query (SELECT only for safety)",
+  {
+    sql: z.string().describe("SQL SELECT query to execute"),
+    limit: z.number().optional().describe("Max rows to return (default: 100)"),
+  },
+  async ({ sql, limit = 100 }) => {
+    // Security: Only allow SELECT queries
+    const trimmedSql = sql.trim().toUpperCase();
+    if (!trimmedSql.startsWith("SELECT")) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ 
+            error: "Only SELECT queries are allowed. Use db_execute for writes." 
+          }),
+        }],
+      };
+    }
+    
+    // Add LIMIT if not present
+    let safeSql = sql;
+    if (!trimmedSql.includes("LIMIT")) {
+      safeSql = `${sql} LIMIT ${limit}`;
+    }
+    
+    const result = await executeSqliteQuery(safeSql);
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify(result, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Tool: db_schema
+// ============================================================================
+
+server.tool(
+  "db_schema",
+  "Get database schema (tables and columns)",
+  {},
+  async () => {
+    const result = await executeSqliteQuery(`
+      SELECT 
+        m.name as table_name,
+        p.name as column_name,
+        p.type as column_type
+      FROM sqlite_master m
+      JOIN pragma_table_info(m.name) p
+      WHERE m.type = 'table' AND m.name NOT LIKE 'sqlite_%'
+      ORDER BY m.name, p.cid
+    `);
+    
+    if (!result.success) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ error: result.error }),
+        }],
+      };
+    }
+    
+    // Group by table
+    const tables: Record<string, Array<{ name: string; type: string }>> = {};
+    for (const row of result.rows || []) {
+      const tableName = row.table_name as string;
+      if (!tables[tableName]) {
+        tables[tableName] = [];
+      }
+      tables[tableName].push({
+        name: row.column_name as string,
+        type: row.column_type as string,
+      });
+    }
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify({ tables }, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Tool: db_tables
+// ============================================================================
+
+server.tool(
+  "db_tables",
+  "List all tables in the database",
+  {},
+  async () => {
+    const result = await executeSqliteQuery(`
+      SELECT name, type 
+      FROM sqlite_master 
+      WHERE type IN ('table', 'view') AND name NOT LIKE 'sqlite_%'
+      ORDER BY name
+    `);
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify(result, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Tool: db_describe
+// ============================================================================
+
+server.tool(
+  "db_describe",
+  "Describe a specific table's structure",
+  {
+    table: z.string().describe("Table name to describe"),
+  },
+  async ({ table }) => {
+    // Validate table name (prevent SQL injection)
+    if (!/^[a-zA-Z_][a-zA-Z0-9_]*$/.test(table)) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ error: "Invalid table name" }),
+        }],
+      };
+    }
+    
+    const result = await executeSqliteQuery(`PRAGMA table_info(${table})`);
+    
+    if (!result.success || !result.rows?.length) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ error: `Table '${table}' not found or empty` }),
+        }],
+      };
+    }
+    
+    // Get row count
+    const countResult = await executeSqliteQuery(`SELECT COUNT(*) as count FROM ${table}`);
+    const rowCount = countResult.rows?.[0]?.count || 0;
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify({
+          table,
+          columns: result.rows,
+          rowCount,
+        }, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Resource: Database Info
+// ============================================================================
+
+server.resource(
+  "db-info",
+  "db://info",
+  async (uri) => ({
+    contents: [{
+      uri: uri.href,
+      mimeType: "application/json",
+      text: JSON.stringify({
+        type: DB_TYPE,
+        path: DB_PATH,
+        exists: fs.existsSync(DB_PATH),
+        timestamp: new Date().toISOString(),
+      }),
+    }],
+  })
+);
+
+// ============================================================================
+// Start Server
+// ============================================================================
+
+async function main() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error("{{SERVER_NAME}} MCP Server running on stdio");
+}
+
+main().catch((error) => {
+  console.error("Failed to start MCP server:", error);
+  process.exit(1);
+});

package/.agent/mcp/templates/web-scraper/src/index.ts.template

@@ -0,0 +1,239 @@
+#!/usr/bin/env node
+/**
+ * {{SERVER_NAME}} - Web Scraper MCP Server
+ * 
+ * Extract data from websites for your AI assistant.
+ * 
+ * Generated by Agent Kit MCP Generator
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+
+const server = new McpServer({
+  name: "{{SERVER_NAME}}",
+  version: "1.0.0",
+});
+
+// ============================================================================
+// Helper Functions
+// ============================================================================
+
+interface FetchResult {
+  success: boolean;
+  html?: string;
+  text?: string;
+  title?: string;
+  error?: string;
+}
+
+async function fetchPage(url: string): Promise<FetchResult> {
+  try {
+    const response = await fetch(url, {
+      headers: {
+        "User-Agent": "Mozilla/5.0 (compatible; MCPBot/1.0)",
+      },
+    });
+    
+    if (!response.ok) {
+      return { 
+        success: false, 
+        error: `HTTP ${response.status}: ${response.statusText}` 
+      };
+    }
+    
+    const html = await response.text();
+    
+    // Extract title
+    const titleMatch = html.match(/<title[^>]*>([^<]+)<\/title>/i);
+    const title = titleMatch ? titleMatch[1].trim() : undefined;
+    
+    // Extract text (basic - remove tags)
+    const text = html
+      .replace(/<script[^>]*>[\s\S]*?<\/script>/gi, "")
+      .replace(/<style[^>]*>[\s\S]*?<\/style>/gi, "")
+      .replace(/<[^>]+>/g, " ")
+      .replace(/\s+/g, " ")
+      .trim();
+    
+    return { success: true, html, text, title };
+  } catch (error) {
+    return { 
+      success: false, 
+      error: `Fetch error: ${error instanceof Error ? error.message : "Unknown"}` 
+    };
+  }
+}
+
+// ============================================================================
+// Tool: scrape_url
+// ============================================================================
+
+server.tool(
+  "scrape_url",
+  "Fetch and extract text content from a URL",
+  {
+    url: z.string().url().describe("URL to scrape"),
+    maxLength: z.number().optional().describe("Max text length (default: 5000)"),
+  },
+  async ({ url, maxLength = 5000 }) => {
+    const result = await fetchPage(url);
+    
+    if (!result.success) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ error: result.error }),
+        }],
+      };
+    }
+    
+    const truncatedText = result.text?.slice(0, maxLength);
+    const truncated = (result.text?.length || 0) > maxLength;
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify({
+          url,
+          title: result.title,
+          text: truncatedText,
+          truncated,
+          originalLength: result.text?.length,
+        }, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Tool: extract_links
+// ============================================================================
+
+server.tool(
+  "extract_links",
+  "Extract all links from a webpage",
+  {
+    url: z.string().url().describe("URL to extract links from"),
+    filter: z.string().optional().describe("Filter links containing this string"),
+  },
+  async ({ url, filter }) => {
+    const result = await fetchPage(url);
+    
+    if (!result.success || !result.html) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ error: result.error }),
+        }],
+      };
+    }
+    
+    // Extract href attributes
+    const linkRegex = /href=["']([^"']+)["']/gi;
+    const links: string[] = [];
+    let match;
+    
+    while ((match = linkRegex.exec(result.html)) !== null) {
+      let href = match[1];
+      
+      // Convert relative URLs to absolute
+      if (href.startsWith("/")) {
+        const urlObj = new URL(url);
+        href = `${urlObj.origin}${href}`;
+      }
+      
+      // Filter if specified
+      if (filter && !href.includes(filter)) continue;
+      
+      // Skip non-http links
+      if (!href.startsWith("http")) continue;
+      
+      if (!links.includes(href)) {
+        links.push(href);
+      }
+    }
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify({
+          url,
+          links: links.slice(0, 50), // Limit to 50
+          count: links.length,
+        }, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Tool: extract_metadata
+// ============================================================================
+
+server.tool(
+  "extract_metadata",
+  "Extract metadata (title, description, og tags) from a webpage",
+  {
+    url: z.string().url().describe("URL to extract metadata from"),
+  },
+  async ({ url }) => {
+    const result = await fetchPage(url);
+    
+    if (!result.success || !result.html) {
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ error: result.error }),
+        }],
+      };
+    }
+    
+    const html = result.html;
+    
+    // Extract meta tags
+    const getMeta = (name: string): string | undefined => {
+      const match = html.match(
+        new RegExp(`<meta[^>]*(?:name|property)=["']${name}["'][^>]*content=["']([^"']+)["']`, "i")
+      ) || html.match(
+        new RegExp(`<meta[^>]*content=["']([^"']+)["'][^>]*(?:name|property)=["']${name}["']`, "i")
+      );
+      return match?.[1];
+    };
+    
+    const metadata = {
+      url,
+      title: result.title,
+      description: getMeta("description"),
+      ogTitle: getMeta("og:title"),
+      ogDescription: getMeta("og:description"),
+      ogImage: getMeta("og:image"),
+      ogType: getMeta("og:type"),
+      twitterCard: getMeta("twitter:card"),
+      twitterTitle: getMeta("twitter:title"),
+    };
+    
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify(metadata, null, 2),
+      }],
+    };
+  }
+);
+
+// ============================================================================
+// Start Server
+// ============================================================================
+
+async function main() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error("{{SERVER_NAME}} MCP Server running on stdio");
+}
+
+main().catch((error) => {
+  console.error("Failed to start MCP server:", error);
+  process.exit(1);
+});