Merge pull request #27 from Jpisnice/fixes

Fixes

Author: Jpisnice

package.json

@@ -33,7 +33,7 @@
     "claude",
     "copilot"
   ],
-  "author": "Janardhan Pollle <[email protected]>",
+  "author": "Janardhan Pollle",
   "license": "MIT",
   "repository": {
     "type": "git",

src/cli/args.ts

@@ -0,0 +1,18 @@
+export type CliOptions = {
+  githubApiKey: string | null
+}
+
+export function parseArgs(argv = process.argv.slice(2)): CliOptions {
+  const githubApiKeyIndex = argv.findIndex(
+    (arg) => arg === "--github-api-key" || arg === "-g"
+  )
+  let githubApiKey: string | null = null
+
+  if (githubApiKeyIndex !== -1 && argv[githubApiKeyIndex + 1]) {
+    githubApiKey = argv[githubApiKeyIndex + 1]
+  } else if (process.env.GITHUB_PERSONAL_ACCESS_TOKEN) {
+    githubApiKey = process.env.GITHUB_PERSONAL_ACCESS_TOKEN
+  }
+
+  return { githubApiKey }
+}

src/index.ts

@@ -1,342 +1,8 @@
 #!/usr/bin/env node
-/**
- * Shadcn UI v4 MCP Server
- *
- * A Model Context Protocol server for shadcn/ui v4 components.
- * Provides AI assistants with access to component source code, demos, blocks, and metadata.
- *
- * Usage:
- *   npx shadcn-ui-mcp-server
- *   npx shadcn-ui-mcp-server --github-api-key YOUR_TOKEN
- *   npx shadcn-ui-mcp-server -g YOUR_TOKEN
- */
-import { Server } from "@modelcontextprotocol/sdk/server/index.js"
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
-import { setupHandlers } from "./handler.js"
-import {
-  validateFrameworkSelection,
-  getAxiosImplementation,
-} from "./utils/framework.js"
-import { z } from "zod"
-import { toolHandlers, toolSchemas } from "./tools/index.js"
-import { logError, logInfo, logWarning } from "./utils/logger.js"
+import { start } from "./server/index.js"
+import { logError } from "./utils/logger.js"
 
-/**
- * Parse command line arguments
- */
-async function parseArgs() {
-  const args = process.argv.slice(2)
-
-  // Help flag
-  if (args.includes("--help") || args.includes("-h")) {
-    console.log(`
-Shadcn UI v4 MCP Server
-
-Usage:
-  npx shadcn-ui-mcp-server [options]
-
-Options:
-  --github-api-key, -g <token>    GitHub Personal Access Token for API access
-  --framework, -f <framework>     Framework to use: 'react' or 'svelte' (default: react)
-  --help, -h                      Show this help message
-  --version, -v                   Show version information
-
-Examples:
-  npx shadcn-ui-mcp-server
-  npx shadcn-ui-mcp-server --github-api-key ghp_your_token_here
-  npx shadcn-ui-mcp-server -g ghp_your_token_here
-  npx shadcn-ui-mcp-server --framework svelte
-  npx shadcn-ui-mcp-server -f react
-
-Environment Variables:
-  GITHUB_PERSONAL_ACCESS_TOKEN    Alternative way to provide GitHub token
-  FRAMEWORK                       Framework to use: 'react' or 'svelte' or 'vue' (default: react)
-  LOG_LEVEL                       Log level (debug, info, warn, error) - default: info
-
-For more information, visit: https://github.com/Jpisnice/shadcn-ui-mcp-server
-`)
-    process.exit(0)
-  }
-
-  // Version flag
-  if (args.includes("--version") || args.includes("-v")) {
-    // Read version from package.json
-    try {
-      const fs = await import("fs")
-      const path = await import("path")
-      const { fileURLToPath } = await import("url")
-
-      const __filename = fileURLToPath(import.meta.url)
-      const __dirname = path.dirname(__filename)
-      const packagePath = path.join(__dirname, "..", "package.json")
-
-      const packageContent = fs.readFileSync(packagePath, "utf8")
-      const packageJson = JSON.parse(packageContent)
-      console.log(`shadcn-ui-mcp-server v${packageJson.version}`)
-    } catch (error) {
-      console.log("shadcn-ui-mcp-server v1.0.2")
-    }
-    process.exit(0)
-  }
-
-  // GitHub API key
-  const githubApiKeyIndex = args.findIndex(
-    (arg) => arg === "--github-api-key" || arg === "-g"
-  )
-  let githubApiKey = null
-
-  if (githubApiKeyIndex !== -1 && args[githubApiKeyIndex + 1]) {
-    githubApiKey = args[githubApiKeyIndex + 1]
-  } else if (process.env.GITHUB_PERSONAL_ACCESS_TOKEN) {
-    githubApiKey = process.env.GITHUB_PERSONAL_ACCESS_TOKEN
-  }
-
-  return { githubApiKey }
-}
-
-/**
- * Main function to start the MCP server
- */
-async function main() {
-  try {
-    logInfo("Starting Shadcn UI v4 MCP Server...")
-
-    const { githubApiKey } = await parseArgs()
-
-    // Validate and log framework selection
-    validateFrameworkSelection()
-
-    // Get the appropriate axios implementation based on framework
-    const axios = await getAxiosImplementation()
-
-    // Configure GitHub API key if provided
-    if (githubApiKey) {
-      axios.setGitHubApiKey(githubApiKey)
-      logInfo("GitHub API configured with token")
-    } else {
-      logWarning(
-        "No GitHub API key provided. Rate limited to 60 requests/hour."
-      )
-    }
-
-    // Initialize the MCP server with metadata and capabilities
-    // Following MCP SDK 1.16.0 best practices
-    const server = new Server(
-      {
-        name: "shadcn-ui-mcp-server",
-        version: "1.0.2",
-      },
-      {
-        capabilities: {
-          resources: {
-            get_components: {
-              description:
-                "List of available shadcn/ui components that can be used in the project",
-              uri: "resource:get_components",
-              contentType: "text/plain",
-            },
-            get_install_script_for_component: {
-              description:
-                "Generate installation script for a specific shadcn/ui component based on package manager",
-              uriTemplate:
-                "resource-template:get_install_script_for_component?packageManager={packageManager}&component={component}",
-              contentType: "text/plain",
-            },
-            get_installation_guide: {
-              description:
-                "Get the installation guide for shadcn/ui based on build tool and package manager",
-              uriTemplate:
-                "resource-template:get_installation_guide?buildTool={buildTool}&packageManager={packageManager}",
-              contentType: "text/plain",
-            },
-          },
-          prompts: {
-            component_usage: {
-              description: "Get usage examples for a specific component",
-              arguments: {
-                componentName: {
-                  type: "string",
-                  description: "Name of the component to get usage for",
-                },
-              },
-            },
-            component_search: {
-              description: "Search for components by name or description",
-              arguments: {
-                query: {
-                  type: "string",
-                  description: "Search query",
-                },
-              },
-            },
-            component_comparison: {
-              description: "Compare two components side by side",
-              arguments: {
-                component1: {
-                  type: "string",
-                  description: "First component name",
-                },
-                component2: {
-                  type: "string",
-                  description: "Second component name",
-                },
-              },
-            },
-            component_recommendation: {
-              description: "Get component recommendations based on use case",
-              arguments: {
-                useCase: {
-                  type: "string",
-                  description: "Use case description",
-                },
-              },
-            },
-            component_tutorial: {
-              description: "Get a step-by-step tutorial for using a component",
-              arguments: {
-                componentName: {
-                  type: "string",
-                  description: "Name of the component for tutorial",
-                },
-              },
-            },
-          },
-          tools: {
-            get_component: {
-              description:
-                "Get the source code for a specific shadcn/ui v4 component",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  componentName: {
-                    type: "string",
-                    description:
-                      'Name of the shadcn/ui component (e.g., "accordion", "button")',
-                  },
-                },
-                required: ["componentName"],
-              },
-            },
-            get_component_demo: {
-              description:
-                "Get demo code illustrating how a shadcn/ui v4 component should be used",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  componentName: {
-                    type: "string",
-                    description:
-                      'Name of the shadcn/ui component (e.g., "accordion", "button")',
-                  },
-                },
-                required: ["componentName"],
-              },
-            },
-            list_components: {
-              description: "Get all available shadcn/ui v4 components",
-              inputSchema: {
-                type: "object",
-                properties: {},
-              },
-            },
-            get_component_metadata: {
-              description: "Get metadata for a specific shadcn/ui v4 component",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  componentName: {
-                    type: "string",
-                    description:
-                      'Name of the shadcn/ui component (e.g., "accordion", "button")',
-                  },
-                },
-                required: ["componentName"],
-              },
-            },
-            get_directory_structure: {
-              description:
-                "Get the directory structure of the shadcn-ui v4 repository",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  path: {
-                    type: "string",
-                    description:
-                      "Path within the repository (default: v4 registry)",
-                  },
-                  owner: {
-                    type: "string",
-                    description: 'Repository owner (default: "shadcn-ui")',
-                  },
-                  repo: {
-                    type: "string",
-                    description: 'Repository name (default: "ui")',
-                  },
-                  branch: {
-                    type: "string",
-                    description: 'Branch name (default: "main")',
-                  },
-                },
-              },
-            },
-            get_block: {
-              description:
-                "Get source code for a specific shadcn/ui v4 block (e.g., calendar-01, dashboard-01)",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  blockName: {
-                    type: "string",
-                    description:
-                      'Name of the block (e.g., "calendar-01", "dashboard-01", "login-02")',
-                  },
-                  includeComponents: {
-                    type: "boolean",
-                    description:
-                      "Whether to include component files for complex blocks (default: true)",
-                  },
-                },
-                required: ["blockName"],
-              },
-            },
-            list_blocks: {
-              description:
-                "Get all available shadcn/ui v4 blocks with categorization",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  category: {
-                    type: "string",
-                    description:
-                      "Filter by category (calendar, dashboard, login, sidebar, products)",
-                  },
-                },
-              },
-            },
-          },
-        },
-      }
-    )
-
-    // Set up request handlers and register components (tools, resources, etc.)
-    setupHandlers(server)
-
-    // Start server using stdio transport
-    const transport = new StdioServerTransport()
-
-    logInfo("Transport initialized: stdio")
-
-    await server.connect(transport)
-
-    logInfo("Server started successfully")
-  } catch (error) {
-    logError("Failed to start server", error)
-    process.exit(1)
-  }
-}
-
-// Start the server
-main().catch((error) => {
+start().catch((error:Error) => {
   logError("Unhandled startup error", error)
   process.exit(1)
-})
+})