@gemini-designer/mcp-server 0.1.2 → 0.1.29

package/dist/version.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.js","sourceRoot":"","sources":["../src/version.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC/C,MAAM,GAAG,GAAG,OAAO,CAAC,iBAAiB,CAAyB,CAAC;AAE/D,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC"}

package/package.json

@@ -1,56 +1,56 @@
 {
-    "name": "@gemini-designer/mcp-server",
-    "version": "0.1.2",
-    "description": "MCP server for AI-powered UI design and code generation with Gemini",
-    "type": "module",
-    "main": "dist/index.js",
-    "bin": {
-        "gemini-designer-mcp": "dist/index.js"
-    },
-    "scripts": {
-        "dev": "tsx watch src/index.ts",
-        "build": "tsc",
-        "start": "node dist/index.js",
-        "test": "vitest",
-        "test:coverage": "vitest --coverage",
-        "lint": "eslint src/",
-        "lint:fix": "eslint src/ --fix",
-        "format": "prettier --write src/",
-        "typecheck": "tsc --noEmit"
-    },
-    "keywords": [
-        "mcp",
-        "model-context-protocol",
-        "gemini",
-        "design-to-code",
-        "ai",
-        "frontend",
-        "ui-generation"
-    ],
-    "author": "",
-    "license": "MIT",
-    "engines": {
-        "node": ">=20.0.0"
-    },
-    "repository": {
-        "type": "git",
-        "url": "https://github.com/yourusername/gemini-designer-mcp"
-    },
-    "homepage": "https://gemini-designer.dev",
-    "dependencies": {
-        "@google/generative-ai": "^0.21.0",
-        "@modelcontextprotocol/sdk": "^1.12.0",
-        "prettier": "^3.4.0",
-        "typescript": "^5.7.0",
-        "zod": "^3.24.0"
-    },
-    "devDependencies": {
-        "@types/node": "^22.10.0",
-        "@typescript-eslint/eslint-plugin": "^8.18.0",
-        "@typescript-eslint/parser": "^8.18.0",
-        "@vitest/coverage-v8": "^2.1.0",
-        "eslint": "^9.16.0",
-        "tsx": "^4.19.0",
-        "vitest": "^2.1.0"
-    }
-}
+  "name": "@gemini-designer/mcp-server",
+  "version": "0.1.29",
+  "description": "MCP server for AI-powered UI design and code generation with Gemini",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "gemini-designer-mcp": "dist/index.js"
+  },
+  "scripts": {
+    "dev": "tsx watch src/index.ts",
+    "build": "tsc",
+    "start": "node dist/index.js",
+    "test": "vitest run",
+    "test:coverage": "vitest run --coverage",
+    "lint": "eslint src/",
+    "lint:fix": "eslint src/ --fix",
+    "format": "prettier --write src/",
+    "typecheck": "tsc --noEmit"
+  },
+  "keywords": [
+    "mcp",
+    "model-context-protocol",
+    "gemini",
+    "design-to-code",
+    "ai",
+    "frontend",
+    "ui-generation"
+  ],
+  "author": "",
+  "license": "MIT",
+  "engines": {
+    "node": ">=20.0.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/yourusername/gemini-designer-mcp"
+  },
+  "homepage": "https://gemini-designer.dev",
+  "dependencies": {
+    "@google/generative-ai": "^0.21.0",
+    "@modelcontextprotocol/sdk": "^1.12.0",
+    "prettier": "^3.4.0",
+    "typescript": "^5.7.0",
+    "zod": "^3.24.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.0",
+    "@typescript-eslint/eslint-plugin": "^8.18.0",
+    "@typescript-eslint/parser": "^8.18.0",
+    "@vitest/coverage-v8": "^2.1.0",
+    "eslint": "^9.16.0",
+    "tsx": "^4.19.0",
+    "vitest": "^2.1.0"
+  }
+}

package/src/__tests__/builder.test.ts

@@ -6,26 +6,26 @@ import { describe, it, expect } from 'vitest';
 import { estimateTokens } from '../context/builder.js';
 
 describe('estimateTokens', () => {
-    it('should estimate tokens based on character count', () => {
-        // ~4 chars per token
-        expect(estimateTokens('hello')).toBe(2); // 5 chars → ~2 tokens
-        expect(estimateTokens('')).toBe(0);
-    });
+  it('should estimate tokens based on character count', () => {
+    // ~4 chars per token
+    expect(estimateTokens('hello')).toBe(2); // 5 chars → ~2 tokens
+    expect(estimateTokens('')).toBe(0);
+  });
 
-    it('should handle longer text', () => {
-        const text = 'a'.repeat(100);
-        expect(estimateTokens(text)).toBe(25); // 100 chars → 25 tokens
-    });
+  it('should handle longer text', () => {
+    const text = 'a'.repeat(100);
+    expect(estimateTokens(text)).toBe(25); // 100 chars → 25 tokens
+  });
 
-    it('should round up for partial tokens', () => {
-        const text = 'ab'; // 2 chars → 0.5 → rounds to 1
-        expect(estimateTokens(text)).toBe(1);
-    });
+  it('should round up for partial tokens', () => {
+    const text = 'ab'; // 2 chars → 0.5 → rounds to 1
+    expect(estimateTokens(text)).toBe(1);
+  });
 
-    it('should handle code snippets', () => {
-        const code = 'const Button = () => <button>Click</button>;';
-        const tokens = estimateTokens(code);
-        expect(tokens).toBeGreaterThan(0);
-        expect(tokens).toBe(Math.ceil(code.length / 4));
-    });
+  it('should handle code snippets', () => {
+    const code = 'const Button = () => <button>Click</button>;';
+    const tokens = estimateTokens(code);
+    expect(tokens).toBeGreaterThan(0);
+    expect(tokens).toBe(Math.ceil(code.length / 4));
+  });
 });

package/src/__tests__/config.test.ts

@@ -5,48 +5,80 @@
 import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
 
 describe('Config', () => {
-    const originalEnv = process.env;
+  const originalEnv = process.env;
 
-    beforeEach(() => {
-        vi.resetModules();
-        process.env = { ...originalEnv };
+  beforeEach(() => {
+    vi.resetModules();
+    process.env = { ...originalEnv };
+  });
+
+  afterEach(() => {
+    process.env = originalEnv;
+  });
+
+  describe('loadConfigFromEnv', () => {
+    it('loads GEMINI_API_KEY and falls back to gemini provider when LiteLLM is not configured', async () => {
+      process.env.GEMINI_API_KEY = 'test-key';
+
+      // Dynamic import to get fresh module
+      const { loadConfig } = await import('../config/index.js');
+
+      const cfg = loadConfig();
+      expect(cfg.mode).toBe('local');
+      expect(cfg.apiKey).toBe('test-key');
+      expect(cfg.localProvider).toBe('gemini');
     });
 
-    afterEach(() => {
-        process.env = originalEnv;
+    it('falls back to litellm provider when GEMINI_API_KEY is missing but LITELLM_ENDPOINT is configured', async () => {
+      process.env.LITELLM_ENDPOINT = 'http://localhost:4000';
+
+      const { loadConfig } = await import('../config/index.js');
+      const cfg = loadConfig();
+      expect(cfg.mode).toBe('local');
+      expect(cfg.localProvider).toBe('litellm');
     });
 
-    describe('loadConfigFromEnv', () => {
-        it('should load API key from environment', async () => {
-            process.env.GEMINI_API_KEY = 'test-key';
+    it('throws when remote mode is selected without a remote endpoint', async () => {
+      process.env.GEMINI_DESIGNER_MODE = 'remote';
+      process.env.GEMINI_DESIGNER_REMOTE_API_KEY = 'gd_test_key';
+
+      const { loadConfig } = await import('../config/index.js');
+      const spy = vi.spyOn(console, 'error').mockImplementation(() => {});
+      expect(() => loadConfig()).toThrow(/remote endpoint/i);
+      spy.mockRestore();
+    });
 
-            // Dynamic import to get fresh module
-            const { loadConfig } = await import('../config/index.js');
+    it('throws when remote endpoint is plain HTTP (non-localhost)', async () => {
+      process.env.GEMINI_DESIGNER_MODE = 'remote';
+      process.env.GEMINI_DESIGNER_REMOTE_API_KEY = 'gd_test_key';
+      process.env.GEMINI_DESIGNER_REMOTE_ENDPOINT = 'http://example.com';
 
-            // This would throw without the key, so we test it exists
-            expect(process.env.GEMINI_API_KEY).toBe('test-key');
-        });
+      const { loadConfig } = await import('../config/index.js');
+      const spy = vi.spyOn(console, 'error').mockImplementation(() => {});
+      expect(() => loadConfig()).toThrow(/insecure remote endpoint/i);
+      spy.mockRestore();
+    });
 
-        it('should respect mode environment variable', () => {
-            process.env.GEMINI_DESIGNER_MODE = 'remote';
-            expect(process.env.GEMINI_DESIGNER_MODE).toBe('remote');
-        });
+    it('allows remote endpoint over HTTP for localhost (dev)', async () => {
+      process.env.GEMINI_DESIGNER_MODE = 'remote';
+      process.env.GEMINI_DESIGNER_REMOTE_API_KEY = 'gd_test_key';
+      process.env.GEMINI_DESIGNER_REMOTE_ENDPOINT = 'http://localhost:3000';
 
-        it('should respect debug environment variable', () => {
-            process.env.GEMINI_DESIGNER_DEBUG = 'true';
-            expect(process.env.GEMINI_DESIGNER_DEBUG).toBe('true');
-        });
+      const { loadConfig } = await import('../config/index.js');
+      const cfg = loadConfig();
+      expect(cfg.mode).toBe('remote');
+      expect(cfg.remoteEndpoint).toBe('http://localhost:3000');
     });
+  });
 
-    describe('Config schema', () => {
-        it('should have correct default framework', () => {
-            // The default is 'react' in the schema
-            expect(true).toBe(true); // Placeholder for schema validation
-        });
+  describe('Config defaults', () => {
+    it('defaults framework to react and accessibility to wcag-aa', async () => {
+      process.env.GEMINI_API_KEY = 'test-key';
 
-        it('should have correct default accessibility', () => {
-            // The default is 'wcag-aa' in the schema
-            expect(true).toBe(true); // Placeholder for schema validation
-        });
+      const { loadConfig } = await import('../config/index.js');
+      const cfg = loadConfig();
+      expect(cfg.defaultFramework).toBe('react');
+      expect(cfg.accessibility).toBe('wcag-aa');
     });
+  });
 });

package/src/__tests__/filter.test.ts

@@ -3,107 +3,113 @@
  */
 
 import { describe, it, expect } from 'vitest';
-import { isSensitiveFile, isPathAllowed, isUIRelevant, sanitizeContent } from '../context/filter.js';
+import {
+  isSensitiveFile,
+  isPathAllowed,
+  isUIRelevant,
+  sanitizeContent,
+} from '../context/filter.js';
 
 describe('isSensitiveFile', () => {
-    it('should flag .env files as sensitive', () => {
-        expect(isSensitiveFile('.env')).toBe(true);
-        expect(isSensitiveFile('.env.local')).toBe(true);
-        expect(isSensitiveFile('/path/to/.env.production')).toBe(true);
-    });
-
-    it('should flag secret files as sensitive', () => {
-        expect(isSensitiveFile('secrets.json')).toBe(true);
-        expect(isSensitiveFile('/config/secrets.yaml')).toBe(true);
-    });
-
-    it('should flag private keys as sensitive', () => {
-        expect(isSensitiveFile('server.key')).toBe(true);
-        expect(isSensitiveFile('certificate.pem')).toBe(true);
-        expect(isSensitiveFile('id_rsa')).toBe(true);
-    });
-
-    it('should flag backend directories as sensitive', () => {
-        expect(isSensitiveFile('/src/api/routes.ts')).toBe(true);
-        expect(isSensitiveFile('/server/index.ts')).toBe(true);
-        expect(isSensitiveFile('/backend/handlers.ts')).toBe(true);
-    });
-
-    it('should not flag UI files as sensitive', () => {
-        expect(isSensitiveFile('Button.tsx')).toBe(false);
-        expect(isSensitiveFile('/components/Card.tsx')).toBe(false);
-        expect(isSensitiveFile('styles.css')).toBe(false);
-    });
+  it('should flag .env files as sensitive', () => {
+    expect(isSensitiveFile('.env')).toBe(true);
+    expect(isSensitiveFile('.env.local')).toBe(true);
+    expect(isSensitiveFile('/path/to/.env.production')).toBe(true);
+  });
+
+  it('should flag secret files as sensitive', () => {
+    expect(isSensitiveFile('secrets.json')).toBe(true);
+    expect(isSensitiveFile('/config/secrets.yaml')).toBe(true);
+  });
+
+  it('should flag private keys as sensitive', () => {
+    expect(isSensitiveFile('server.key')).toBe(true);
+    expect(isSensitiveFile('certificate.pem')).toBe(true);
+    expect(isSensitiveFile('id_rsa')).toBe(true);
+  });
+
+  it('should flag backend directories as sensitive', () => {
+    expect(isSensitiveFile('/src/api/routes.ts')).toBe(true);
+    expect(isSensitiveFile('/server/index.ts')).toBe(true);
+    expect(isSensitiveFile('/backend/handlers.ts')).toBe(true);
+  });
+
+  it('should not flag UI files as sensitive', () => {
+    expect(isSensitiveFile('Button.tsx')).toBe(false);
+    expect(isSensitiveFile('/components/Card.tsx')).toBe(false);
+    expect(isSensitiveFile('styles.css')).toBe(false);
+  });
 });
 
 describe('isPathAllowed', () => {
-    it('should allow paths within allowed directories', () => {
-        expect(isPathAllowed('/project/src/file.ts', ['/project'])).toBe(true);
-        expect(isPathAllowed('/project/components/Button.tsx', ['/project'])).toBe(true);
-    });
-
-    it('should reject paths outside allowed directories', () => {
-        expect(isPathAllowed('/etc/passwd', ['/project'])).toBe(false);
-        expect(isPathAllowed('/home/user/secrets', ['/project'])).toBe(false);
-    });
-
-    it('should support multiple allowed paths', () => {
-        const allowed = ['/project', '/shared'];
-        expect(isPathAllowed('/project/file.ts', allowed)).toBe(true);
-        expect(isPathAllowed('/shared/utils.ts', allowed)).toBe(true);
-        expect(isPathAllowed('/other/file.ts', allowed)).toBe(false);
-    });
+  it('should allow paths within allowed directories', () => {
+    expect(isPathAllowed('/project/src/file.ts', ['/project'])).toBe(true);
+    expect(isPathAllowed('/project/components/Button.tsx', ['/project'])).toBe(true);
+  });
+
+  it('should reject paths outside allowed directories', () => {
+    expect(isPathAllowed('/etc/passwd', ['/project'])).toBe(false);
+    expect(isPathAllowed('/home/user/secrets', ['/project'])).toBe(false);
+  });
+
+  it('should support multiple allowed paths', () => {
+    const allowed = ['/project', '/shared'];
+    expect(isPathAllowed('/project/file.ts', allowed)).toBe(true);
+    expect(isPathAllowed('/shared/utils.ts', allowed)).toBe(true);
+    expect(isPathAllowed('/other/file.ts', allowed)).toBe(false);
+  });
 });
 
 describe('isUIRelevant', () => {
-    it('should identify CSS files as UI relevant', () => {
-        expect(isUIRelevant('styles.css')).toBe(true);
-        expect(isUIRelevant('theme.scss')).toBe(true);
-        expect(isUIRelevant('variables.less')).toBe(true);
-    });
-
-    it('should identify component files as UI relevant', () => {
-        expect(isUIRelevant('Button.tsx')).toBe(true);
-        expect(isUIRelevant('Card.jsx')).toBe(true);
-        expect(isUIRelevant('Header.vue')).toBe(true);
-        expect(isUIRelevant('Nav.svelte')).toBe(true);
-    });
-
-    it('should identify design token files as UI relevant', () => {
-        expect(isUIRelevant('tokens.json')).toBe(true);
-        expect(isUIRelevant('design-tokens.ts')).toBe(true);
-        expect(isUIRelevant('theme.ts')).toBe(true);
-    });
-
-    it('should not identify backend files as UI relevant', () => {
-        expect(isUIRelevant('database.ts')).toBe(false);
-        expect(isUIRelevant('server.ts')).toBe(false);
-    });
+  it('should identify CSS files as UI relevant', () => {
+    expect(isUIRelevant('styles.css')).toBe(true);
+    expect(isUIRelevant('theme.scss')).toBe(true);
+    expect(isUIRelevant('variables.less')).toBe(true);
+  });
+
+  it('should identify component files as UI relevant', () => {
+    expect(isUIRelevant('Button.tsx')).toBe(true);
+    expect(isUIRelevant('Card.jsx')).toBe(true);
+    expect(isUIRelevant('Header.vue')).toBe(true);
+    expect(isUIRelevant('Nav.svelte')).toBe(true);
+  });
+
+  it('should identify design token files as UI relevant', () => {
+    expect(isUIRelevant('tokens.json')).toBe(true);
+    expect(isUIRelevant('design-tokens.ts')).toBe(true);
+    expect(isUIRelevant('theme.ts')).toBe(true);
+  });
+
+  it('should not identify backend files as UI relevant', () => {
+    expect(isUIRelevant('database.ts')).toBe(false);
+    expect(isUIRelevant('server.ts')).toBe(false);
+  });
 });
 
 describe('sanitizeContent', () => {
-    it('should redact API key patterns', () => {
-        const content = 'const apiKey = "sk-abc123xyz";';
-        expect(sanitizeContent(content)).toContain('[REDACTED]');
-    });
-
-    it('should redact Bearer tokens', () => {
-        const content = 'Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ';
-        expect(sanitizeContent(content)).toContain('[REDACTED]');
-    });
-
-    it('should redact AWS keys', () => {
-        const content = 'aws_access_key = "AKIAIOSFODNN7EXAMPLE"';
-        expect(sanitizeContent(content)).toContain('[REDACTED]');
-    });
-
-    it('should redact database connection strings', () => {
-        const content = 'DATABASE_URL=postgresql://user:password@localhost:5432/db';
-        expect(sanitizeContent(content)).toContain('[REDACTED]');
-    });
-
-    it('should preserve non-sensitive content', () => {
-        const content = 'const Button = () => <button>Click me</button>;';
-        expect(sanitizeContent(content)).toBe(content);
-    });
+  it('should redact API key patterns', () => {
+    const content = 'const apiKey = "sk-abc123xyz";';
+    expect(sanitizeContent(content)).toContain('[REDACTED]');
+  });
+
+  it('should redact Bearer tokens', () => {
+    const content =
+      'Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyfQ';
+    expect(sanitizeContent(content)).toContain('[REDACTED]');
+  });
+
+  it('should redact AWS keys', () => {
+    const content = 'aws_access_key = "AKIAIOSFODNN7EXAMPLE"';
+    expect(sanitizeContent(content)).toContain('[REDACTED]');
+  });
+
+  it('should redact database connection strings', () => {
+    const content = 'DATABASE_URL=postgresql://user:password@localhost:5432/db';
+    expect(sanitizeContent(content)).toContain('[REDACTED]');
+  });
+
+  it('should preserve non-sensitive content', () => {
+    const content = 'const Button = () => <button>Click me</button>;';
+    expect(sanitizeContent(content)).toBe(content);
+  });
 });