|
|
#!/usr/bin/env python3 |
|
|
""" |
|
|
Post-installation script to automatically configure Kilocode MCP settings. |
|
|
""" |
|
|
|
|
|
import json |
|
|
import os |
|
|
import sys |
|
|
from pathlib import Path |
|
|
|
|
|
|
|
|
def get_kilocode_config_path(): |
|
|
"""Find the Kilocode MCP settings file.""" |
|
|
# Common paths for Kilocode/VSCodium |
|
|
possible_paths = [ |
|
|
Path.home() / ".config/VSCodium/User/globalStorage/kilocode.kilo-code/settings/mcp_settings.json", |
|
|
Path.home() / ".config/Code/User/globalStorage/kilocode.kilo-code/settings/mcp_settings.json", |
|
|
Path.home() / "Library/Application Support/VSCodium/User/globalStorage/kilocode.kilo-code/settings/mcp_settings.json", |
|
|
Path.home() / "Library/Application Support/Code/User/globalStorage/kilocode.kilo-code/settings/mcp_settings.json", |
|
|
] |
|
|
|
|
|
for path in possible_paths: |
|
|
if path.exists(): |
|
|
return path |
|
|
|
|
|
return None |
|
|
|
|
|
|
|
|
def add_to_kilocode_config(): |
|
|
"""Add the image-recognition server to Kilocode's MCP settings.""" |
|
|
config_path = get_kilocode_config_path() |
|
|
|
|
|
if not config_path: |
|
|
print("⚠️ Kilocode MCP settings file not found.") |
|
|
print(" You'll need to manually add the server configuration.") |
|
|
print("\n Add this to your MCP settings:") |
|
|
print(json.dumps({ |
|
|
"image-recognition": { |
|
|
"command": "uvx", |
|
|
"args": ["image-recognition-mcp"], |
|
|
"env": { |
|
|
"OPENAI_API_KEY": "your-openai-api-key-here" |
|
|
} |
|
|
} |
|
|
}, indent=2)) |
|
|
return False |
|
|
|
|
|
try: |
|
|
# Read existing config |
|
|
with open(config_path, 'r') as f: |
|
|
config = json.load(f) |
|
|
|
|
|
# Ensure mcpServers exists |
|
|
if "mcpServers" not in config: |
|
|
config["mcpServers"] = {} |
|
|
|
|
|
# Check if already configured |
|
|
if "image-recognition" in config["mcpServers"]: |
|
|
print("✅ Image Recognition MCP server already configured in Kilocode") |
|
|
return True |
|
|
|
|
|
# Add our server |
|
|
config["mcpServers"]["image-recognition"] = { |
|
|
"command": "uvx", |
|
|
"args": ["image-recognition-mcp"], |
|
|
"env": { |
|
|
"OPENAI_API_KEY": "your-openai-api-key-here" |
|
|
} |
|
|
} |
|
|
|
|
|
# Backup original config |
|
|
backup_path = config_path.with_suffix('.json.backup') |
|
|
with open(backup_path, 'w') as f: |
|
|
json.dump(config, f, indent=2) |
|
|
|
|
|
# Write updated config |
|
|
with open(config_path, 'w') as f: |
|
|
json.dump(config, f, indent=2) |
|
|
|
|
|
print("✅ Successfully added Image Recognition MCP server to Kilocode!") |
|
|
print(f" Config file: {config_path}") |
|
|
print(f" Backup saved: {backup_path}") |
|
|
print("\n⚠️ Remember to:") |
|
|
print(" 1. Replace 'your-openai-api-key-here' with your actual OpenAI API key") |
|
|
print(" 2. Restart Kilocode to load the new server") |
|
|
|
|
|
return True |
|
|
|
|
|
except Exception as e: |
|
|
print(f"❌ Error updating Kilocode config: {e}") |
|
|
return False |
|
|
|
|
|
|
|
|
def main(): |
|
|
"""Main entry point for the installer.""" |
|
|
print("🚀 Image Recognition MCP Server - Post-Install Configuration") |
|
|
print("=" * 60) |
|
|
|
|
|
if add_to_kilocode_config(): |
|
|
print("\n✨ Installation complete!") |
|
|
else: |
|
|
print("\n⚠️ Manual configuration required.") |
|
|
print(" See PUBLISHING.md for instructions.") |
|
|
|
|
|
return 0 |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
sys.exit(main()) |
