Code-MCP

import { z } from "zod"; /** * Utility Tools - Practical helper tools that generate real output */ // ============================================ // Git Helper Tool // ============================================ export const gitHelperSchema = { name: "git_helper", description: "Generates git commands and workflows for common scenarios", inputSchema: z.object({ scenario: z .enum([ "undo-commit", "squash-commits", "rebase-branch", "cherry-pick", "stash-work", "create-branch", "merge-conflict", "reset-file", "amend-commit", "bisect-bug", "clean-branches", "tag-release", ]) .describe("The git scenario you need help with"), details: z .string() .optional() .describe("Additional context like branch names, commit count"), }), }; const gitScenarios: Record<string, (details?: string) => string> = { "undo-commit": () => `# Undo Last Commit (Keep Changes) git reset --soft HEAD~1 # Undo Last Commit (Discard Changes) git reset --hard HEAD~1 # Undo Commit Already Pushed (Safe) git revert HEAD git push # View what was undone git reflog`, "squash-commits": (d) => `# Squash Last ${d || "N"} Commits Interactively git rebase -i HEAD~${d || "N"} # In editor, change 'pick' to 'squash' (or 's') for commits to squash # Example: # pick abc123 First commit # squash def456 Second commit # squash ghi789 Third commit # After saving, edit the combined commit message # Force push if already pushed git push --force-with-lease`, "rebase-branch": (d) => `# Rebase current branch onto ${d || "main"} git fetch origin git rebase origin/${d || "main"} # If conflicts occur: # 1. Fix conflicts in files # 2. git add <fixed-files> # 3. git rebase --continue # To abort rebase git rebase --abort # Force push after rebase git push --force-with-lease`, "cherry-pick": (d) => `# Cherry-pick specific commit git cherry-pick ${d || "<commit-hash>"} # Cherry-pick multiple commits git cherry-pick <hash1> <hash2> <hash3> # Cherry-pick range (exclusive start) git cherry-pick A..B # Cherry-pick without committing git cherry-pick -n ${d || "<commit-hash>"} # If conflicts: git cherry-pick --continue # after fixing git cherry-pick --abort # to cancel`, "stash-work": () => `# Stash current changes git stash # Stash with a message git stash push -m "WIP: feature description" # Stash including untracked files git stash -u # List all stashes git stash list # Apply most recent stash (keep in stash) git stash apply # Apply and remove from stash git stash pop # Apply specific stash git stash apply stash@{2} # Drop a stash git stash drop stash@{0}`, "create-branch": (d) => `# Create and switch to new branch git checkout -b ${d || "feature/new-feature"} # Or with newer git: git switch -c ${d || "feature/new-feature"} # Create branch from specific commit git checkout -b ${d || "feature/new-feature"} <commit-hash> # Create branch from remote git checkout -b ${d || "feature/new-feature"} origin/main # Push new branch to remote git push -u origin ${d || "feature/new-feature"}`, "merge-conflict": () => `# When you have merge conflicts: # 1. See conflicted files git status # 2. Open files and look for conflict markers: # <<<<<<< HEAD # (your changes) # ======= # (their changes) # >>>>>>> branch-name # 3. Edit files to resolve conflicts # 4. Mark as resolved git add <resolved-file> # 5. Complete the merge git commit # Abort merge if needed git merge --abort # Use a merge tool git mergetool`, "reset-file": (d) => `# Discard changes to specific file git checkout -- ${d || "<file-path>"} # Or with newer git: git restore ${d || "<file-path>"} # Unstage a file (keep changes) git reset HEAD ${d || "<file-path>"} # Or with newer git: git restore --staged ${d || "<file-path>"} # Reset file to specific commit git checkout <commit-hash> -- ${d || "<file-path>"}`, "amend-commit": () => `# Amend last commit message git commit --amend -m "New message" # Amend last commit with staged changes git add <files> git commit --amend --no-edit # Amend and change message git add <files> git commit --amend -m "Updated message" # Push amended commit (if already pushed) git push --force-with-lease`, "bisect-bug": () => `# Start bisecting to find bug git bisect start # Mark current commit as bad git bisect bad # Mark known good commit git bisect good <commit-hash> # Git will checkout commits for you to test # After testing each: git bisect good # if no bug git bisect bad # if bug present # When found, git shows the bad commit # Exit bisect mode git bisect reset`, "clean-branches": () => `# Delete local branch git branch -d feature-branch # safe delete git branch -D feature-branch # force delete # Delete remote branch git push origin --delete feature-branch # Prune remote-tracking branches git fetch --prune # List merged branches (safe to delete) git branch --merged main # Delete all merged branches except main/master git branch --merged | grep -v "main\\|master" | xargs git branch -d`, "tag-release": (d) => `# Create annotated tag git tag -a ${d || "v1.0.0"} -m "Release ${d || "v1.0.0"}" # Create tag for specific commit git tag -a ${d || "v1.0.0"} <commit-hash> -m "Release ${d || "v1.0.0"}" # Push tag to remote git push origin ${d || "v1.0.0"} # Push all tags git push origin --tags # List tags git tag -l # Delete tag git tag -d ${d || "v1.0.0"} git push origin --delete ${d || "v1.0.0"}`, }; export function gitHelperHandler(args: { scenario: string; details?: string }) { const { scenario, details } = args; const generator = gitScenarios[scenario]; if (!generator) { return { content: [{ type: "text", text: `Unknown scenario: ${scenario}` }], }; } const commands = generator(details); return { content: [ { type: "text", text: `# Git: ${scenario}\n\n\`\`\`bash\n${commands}\n\`\`\`\n\n💡 **Tip**: Always use \`--force-with-lease\` instead of \`--force\` when pushing rewritten history.`, }, ], }; } // ============================================ // Package.json Generator // ============================================ export const packageJsonSchema = { name: "generate_package_json", description: "Generates a complete package.json with scripts, dependencies, and configuration", inputSchema: z.object({ name: z.string().describe("Project name"), type: z .enum(["node-api", "react", "next", "express", "cli", "library"]) .describe("Project type"), typescript: z.boolean().optional().default(true), features: z .array(z.string()) .optional() .describe("Additional features like testing, linting"), }), }; export function packageJsonHandler(args: { name: string; type: string; typescript?: boolean; features?: string[]; }) { const { name, type, typescript = true, features = [] } = args; const base = { name, version: "1.0.0", description: "", main: typescript ? "dist/index.js" : "src/index.js", types: typescript ? "dist/index.d.ts" : undefined, type: "module", scripts: {} as Record<string, string>, keywords: [], author: "", license: "MIT", dependencies: {} as Record<string, string>, devDependencies: {} as Record<string, string>, }; // Type-specific config const configs: Record< string, { scripts: Record<string, string>; deps: Record<string, string>; devDeps: Record<string, string>; } > = { "node-api": { scripts: { dev: "tsx watch src/index.ts", build: "tsc", start: "node dist/index.js", }, deps: { express: "^4.18.2", cors: "^2.8.5", dotenv: "^16.3.1" }, devDeps: { "@types/express": "^4.17.21", "@types/cors": "^2.8.17" }, }, express: { scripts: { dev: "tsx watch src/server.ts", build: "tsc", start: "node dist/server.js", }, deps: { express: "^4.18.2", cors: "^2.8.5", helmet: "^7.1.0", "express-rate-limit": "^7.1.5", }, devDeps: { "@types/express": "^4.17.21", "@types/cors": "^2.8.17" }, }, react: { scripts: { dev: "vite", build: "tsc && vite build", preview: "vite preview", }, deps: { react: "^18.2.0", "react-dom": "^18.2.0" }, devDeps: { "@vitejs/plugin-react": "^4.2.1", vite: "^5.0.10" }, }, next: { scripts: { dev: "next dev", build: "next build", start: "next start", lint: "next lint", }, deps: { next: "^14.1.0", react: "^18.2.0", "react-dom": "^18.2.0" }, devDeps: { "@types/react": "^18.2.48", "@types/react-dom": "^18.2.18" }, }, cli: { scripts: { dev: "tsx src/cli.ts", build: "tsc && chmod +x dist/cli.js", start: "node dist/cli.js", }, deps: { commander: "^12.0.0", chalk: "^5.3.0", ora: "^8.0.1" }, devDeps: {}, }, library: { scripts: { dev: "tsup src/index.ts --watch", build: "tsup src/index.ts --format cjs,esm --dts", prepublishOnly: "npm run build", }, deps: {}, devDeps: { tsup: "^8.0.1" }, }, }; const config = configs[type]; base.scripts = { ...base.scripts, ...config.scripts }; base.dependencies = { ...base.dependencies, ...config.deps }; base.devDependencies = { ...base.devDependencies, ...config.devDeps }; // TypeScript if (typescript) { base.devDependencies.typescript = "^5.3.3"; base.devDependencies.tsx = "^4.7.0"; base.devDependencies["@types/node"] = "^20.11.5"; } // Features if (features.includes("testing") || features.includes("test")) { base.scripts.test = "vitest"; base.scripts["test:coverage"] = "vitest --coverage"; base.devDependencies.vitest = "^1.2.1"; base.devDependencies["@vitest/coverage-v8"] = "^1.2.1"; } if (features.includes("linting") || features.includes("lint")) { base.scripts.lint = "eslint src/"; base.scripts["lint:fix"] = "eslint src/ --fix"; base.devDependencies.eslint = "^8.56.0"; base.devDependencies["@typescript-eslint/eslint-plugin"] = "^6.19.1"; base.devDependencies["@typescript-eslint/parser"] = "^6.19.1"; } if (features.includes("prettier") || features.includes("format")) { base.scripts.format = "prettier --write src/"; base.devDependencies.prettier = "^3.2.4"; } // Clean undefined values const clean = JSON.parse(JSON.stringify(base)); return { content: [ { type: "text", text: `# Generated package.json for ${type} project\n\n\`\`\`json\n${JSON.stringify(clean, null, 2)}\n\`\`\`\n\n## Quick Start\n\`\`\`bash\nnpm install\nnpm run dev\n\`\`\``, }, ], }; } // ============================================ // Dockerfile Generator // ============================================ export const dockerfileSchema = { name: "generate_dockerfile", description: "Generates optimized Dockerfile for different project types", inputSchema: z.object({ type: z .enum(["node", "python", "go", "rust", "static"]) .describe("Project type"), port: z.number().optional().default(3000), multiStage: z.boolean().optional().default(true), entryPoint: z .string() .optional() .describe("Entry point file (e.g., 'dist/index.js' or 'dist/server.js')"), }), }; export function dockerfileHandler(args: { type: string; port?: number; multiStage?: boolean; entryPoint?: string; }) { const { type, port = 3000, multiStage = true, entryPoint } = args; // Defaults const defaults: Record<string, string> = { node: "dist/index.js", python: "main.py", go: "./main", rust: "./app", static: "nginx", }; const entry = entryPoint || defaults[type]; const dockerfiles: Record<string, string> = { node: multiStage ? `# Build stage FROM node:20-alpine AS builder WORKDIR /app COPY package*.json ./ RUN npm ci COPY . . RUN npm run build # Production stage FROM node:20-alpine AS production WORKDIR /app ENV NODE_ENV=production COPY package*.json ./ RUN npm ci --only=production && npm cache clean --force COPY --from=builder /app/dist ./dist # Non-root user for security RUN addgroup -g 1001 -S nodejs && adduser -S nodejs -u 1001 USER nodejs EXPOSE ${port} CMD ["node", "${entry}"]` : `FROM node:20-alpine WORKDIR /app COPY package*.json ./ RUN npm ci COPY . . RUN npm run build EXPOSE ${port} CMD ["node", "${entry}"]`, python: multiStage ? `# Build stage FROM python:3.12-slim AS builder WORKDIR /app ENV PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 RUN pip install --upgrade pip COPY requirements.txt . RUN pip wheel --no-cache-dir --wheel-dir /wheels -r requirements.txt # Production stage FROM python:3.12-slim WORKDIR /app ENV PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 COPY --from=builder /wheels /wheels RUN pip install --no-cache /wheels/* COPY . . RUN adduser --system --no-create-home appuser USER appuser EXPOSE ${port} CMD ["python", "-m", "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "${port}"]` : `FROM python:3.12-slim WORKDIR /app COPY requirements.txt . RUN pip install -r requirements.txt COPY . . EXPOSE ${port} CMD ["python", "main.py"]`, go: `# Build stage FROM golang:1.22-alpine AS builder WORKDIR /app COPY go.mod go.sum ./ RUN go mod download COPY . . RUN CGO_ENABLED=0 GOOS=linux go build -ldflags="-w -s" -o main . # Production stage FROM alpine:3.19 RUN apk --no-cache add ca-certificates WORKDIR /app COPY --from=builder /app/main . RUN adduser -D -g '' appuser USER appuser EXPOSE ${port} CMD ["./main"]`, rust: `# Build stage FROM rust:1.75-alpine AS builder WORKDIR /app RUN apk add --no-cache musl-dev COPY Cargo.toml Cargo.lock ./ RUN mkdir src && echo "fn main() {}" > src/main.rs RUN cargo build --release && rm -rf src COPY . . RUN cargo build --release # Production stage FROM alpine:3.19 RUN apk --no-cache add ca-certificates WORKDIR /app COPY --from=builder /app/target/release/app . RUN adduser -D appuser USER appuser EXPOSE ${port} CMD ["./app"]`, static: `FROM nginx:alpine COPY dist/ /usr/share/nginx/html/ COPY nginx.conf /etc/nginx/nginx.conf EXPOSE 80 CMD ["nginx", "-g", "daemon off;"]`, }; const dockerfile = dockerfiles[type]; return { content: [ { type: "text", text: `# Dockerfile for ${type} (port ${port})\n\n\`\`\`dockerfile\n${dockerfile}\n\`\`\`\n\n## Build & Run\n\`\`\`bash\ndocker build -t myapp .\ndocker run -p ${port}:${port} myapp\n\`\`\``, }, ], }; } // ============================================ // Environment Variables Template // ============================================ export const envTemplateSchema = { name: "generate_env_template", description: "Generates .env.example with common environment variables", inputSchema: z.object({ type: z .enum(["api", "fullstack", "database", "auth", "all"]) .describe("Type of environment variables needed"), database: z .enum(["postgres", "mysql", "mongodb", "redis", "none"]) .optional(), }), }; export function envTemplateHandler(args: { type: string; database?: string }) { const { type, database } = args; const sections: Record<string, string> = { server: `# Server NODE_ENV=development PORT=3000 HOST=localhost`, database_postgres: `# PostgreSQL Database DATABASE_URL=postgresql://user:password@localhost:5432/mydb DB_HOST=localhost DB_PORT=5432 DB_NAME=mydb DB_USER=user DB_PASSWORD=password`, database_mysql: `# MySQL Database DATABASE_URL=mysql://user:password@localhost:3306/mydb DB_HOST=localhost DB_PORT=3306 DB_NAME=mydb DB_USER=user DB_PASSWORD=password`, database_mongodb: `# MongoDB MONGODB_URI=mongodb://localhost:27017/mydb MONGODB_DB=mydb`, database_redis: `# Redis REDIS_URL=redis://localhost:6379 REDIS_HOST=localhost REDIS_PORT=6379 REDIS_PASSWORD=`, auth: `# Authentication JWT_SECRET=your-super-secret-key-change-in-production JWT_EXPIRES_IN=24h REFRESH_TOKEN_SECRET=another-secret-key REFRESH_TOKEN_EXPIRES_IN=7d # OAuth (optional) GOOGLE_CLIENT_ID= GOOGLE_CLIENT_SECRET= GITHUB_CLIENT_ID= GITHUB_CLIENT_SECRET=`, email: `# Email SMTP_HOST=smtp.example.com SMTP_PORT=587 SMTP_USER= SMTP_PASS= EMAIL_FROM=noreply@example.com`, storage: `# File Storage AWS_ACCESS_KEY_ID= AWS_SECRET_ACCESS_KEY= AWS_REGION=us-east-1 AWS_BUCKET=my-bucket # Or Cloudflare R2 R2_ACCOUNT_ID= R2_ACCESS_KEY_ID= R2_SECRET_ACCESS_KEY= R2_BUCKET=`, api_keys: `# External APIs STRIPE_SECRET_KEY=sk_test_... STRIPE_WEBHOOK_SECRET=whsec_... OPENAI_API_KEY=sk-... SENDGRID_API_KEY=SG...`, monitoring: `# Monitoring & Logging SENTRY_DSN= LOG_LEVEL=info ENABLE_METRICS=true`, }; let envContent = "# Environment Variables\n# Copy to .env and fill in your values\n\n"; // Always add server envContent += sections.server + "\n\n"; // Add database if specified if (database && database !== "none") { envContent += (sections[`database_${database}`] || "") + "\n\n"; } // Type-specific sections if (type === "auth" || type === "all" || type === "fullstack") { envContent += sections.auth + "\n\n"; } if (type === "all" || type === "fullstack") { envContent += sections.email + "\n\n"; envContent += sections.storage + "\n\n"; envContent += sections.api_keys + "\n\n"; envContent += sections.monitoring + "\n\n"; } if (type === "api") { envContent += sections.api_keys + "\n\n"; } return { content: [ { type: "text", text: `# Generated .env.example\n\n\`\`\`env\n${envContent.trim()}\n\`\`\`\n\n⚠️ **Never commit .env to git!** Add it to .gitignore.`, }, ], }; } // Export all export const utilityTools = { gitHelperSchema, gitHelperHandler, packageJsonSchema, packageJsonHandler, dockerfileSchema, dockerfileHandler, envTemplateSchema, envTemplateHandler, };