#!/usr/bin/env python3
"""
Main entry point for NeoCoder package.
"""
import argparse
import logging
import os
from .init_db import main as init_db_main
from .server import main as server_main
# Configure logging
logging.basicConfig(
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger("mcp_neocoder")
def main() -> None:
"""Main entry point for the NeoCoder package."""
# Register cleanup handlers early
try:
from .process_manager import register_cleanup_handlers
register_cleanup_handlers()
logger.info("Process cleanup handlers registered")
except Exception as e:
logger.error(f"Failed to register cleanup handlers: {e}")
parser = argparse.ArgumentParser(
description="NeoCoder Neo4j-Guided AI Coding Workflow (default: server if no command specified)"
)
# Create subparsers for different commands
subparsers = parser.add_subparsers(dest="command")
# Server command
server_parser = subparsers.add_parser("server", help="Run the MCP server")
server_parser.add_argument("--db-url", default=None, help="Neo4j connection URL")
server_parser.add_argument("--username", default=None, help="Neo4j username")
server_parser.add_argument("--password", default=None, help="Neo4j password")
server_parser.add_argument("--database", default=None, help="Neo4j database name")
# Init command
init_parser = subparsers.add_parser("init", help="Initialize the Neo4j database")
init_parser.add_argument("--db-url", default=None, help="Neo4j connection URL")
init_parser.add_argument("--username", default=None, help="Neo4j username")
init_parser.add_argument("--password", default=None, help="Neo4j password")
init_parser.add_argument("--database", default=None, help="Neo4j database name")
args = parser.parse_args()
# Log the environment variables for debugging
logger.info(f"NEO4J_URL from environment: {os.environ.get('NEO4J_URL', 'Not set')}")
logger.info(
f"NEO4J_USERNAME from environment: {os.environ.get('NEO4J_USERNAME', 'Not set')}"
)
logger.info(
f"NEO4J_PASSWORD from environment: {'Set' if os.environ.get('NEO4J_PASSWORD') else 'Not set'}"
)
logger.info(
f"NEO4J_DATABASE from environment: {os.environ.get('NEO4J_DATABASE', 'Not set')}"
)
# Set environment variables for database connection (only if those attributes exist)
for arg_name, env_name in [
("db_url", "NEO4J_URL"),
("username", "NEO4J_USERNAME"),
("password", "NEO4J_PASSWORD"),
("database", "NEO4J_DATABASE"),
]:
if hasattr(args, arg_name):
value = getattr(args, arg_name)
if value is not None:
os.environ[env_name] = value
# Execute the appropriate command
if args.command == "server":
logger.info("Starting NeoCoder MCP Server")
server_main()
elif args.command == "init":
logger.info("Initializing Neo4j database")
import sys
# Pass remaining arguments to init_db (excluding script name and 'init' command)
# Find 'init' in argv to be safe about where to slice
try:
init_index = sys.argv.index("init")
init_args = sys.argv[init_index + 1 :]
except ValueError:
# Fallback if 'init' not found (unlikely if parser matched it)
init_args = []
init_db_main(init_args)
else:
# Default to server if no command provided
logger.info("No command specified, starting NeoCoder MCP Server")
server_main()
if __name__ == "__main__":
main()
