#!/usr/bin/env python3
"""
Main entry point for SearchEngine operations.

Provides a unified CLI for common operations:
- serve: Start API service (search + admin routes)
- serve-indexer: Start dedicated Indexer API service
- search: Test search from command line
"""

import sys
import os
import argparse
import json
import uvicorn

# Add parent directory to path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from config import ConfigLoader
from config.env_config import DB_CONFIG
from utils import ESClient
from search import Searcher
from suggestion import SuggestionIndexBuilder
from utils.db_connector import create_db_connection


def cmd_serve(args):
    """Start API service."""
    os.environ['ES_HOST'] = args.es_host

    print("Starting API service (multi-tenant)...")
    print(f"  Host: {args.host}:{args.port} (search + indexer routes)")
    print(f"  Elasticsearch: {args.es_host}")

    uvicorn.run(
        "api.app:app",
        host=args.host,
        port=args.port,
        reload=args.reload
    )


def cmd_serve_indexer(args):
    """Start dedicated Indexer API service (no search endpoints)."""
    os.environ['ES_HOST'] = args.es_host

    print("Starting Indexer API service...")
    print(f"  Host: {args.host}:{args.port} (indexer only)")
    print(f"  Elasticsearch: {args.es_host}")

    uvicorn.run(
        "api.indexer_app:app",
        host=args.host,
        port=args.port,
        reload=args.reload
    )

def cmd_search(args):
    """Test search from command line."""
    # Load config
    config_loader = ConfigLoader("config/config.yaml")
    config = config_loader.load_config()

    # Initialize ES and searcher
    es_client = ESClient(hosts=[args.es_host])
    if not es_client.ping():
        print(f"ERROR: Cannot connect to Elasticsearch at {args.es_host}")
        return 1

    from query import QueryParser
    query_parser = QueryParser(config)
    searcher = Searcher(es_client, config, query_parser)

    # Execute search
    print(f"Searching for: '{args.query}' (tenant: {args.tenant_id})")
    result = searcher.search(
        query=args.query,
        tenant_id=args.tenant_id,
        size=args.size
    )

    # Display results
    print(f"\nFound {result.total} results in {result.took_ms}ms")
    print(f"Max score: {result.max_score}")

    if args.json:
        print(json.dumps(result.to_dict(), indent=2, ensure_ascii=False))
    else:
        print(f"\nTop {len(result.hits)} results:")
        for i, hit in enumerate(result.hits, 1):
            source = hit['_source']
            score = hit['_score']
            print(f"\n{i}. Score: {score:.4f}")
            print(f"   ID: {source.get('skuId', 'N/A')}")
            print(f"   Name: {source.get('name', 'N/A')}")
            print(f"   Category: {source.get('categoryName', 'N/A')}")
            print(f"   Brand: {source.get('brandName', 'N/A')}")

    return 0


def cmd_build_suggestions(args):
    """Build suggestion index for a tenant."""
    es_client = ESClient(hosts=[args.es_host])
    if not es_client.ping():
        print(f"ERROR: Cannot connect to Elasticsearch at {args.es_host}")
        return 1

    db_engine = create_db_connection(
        host=DB_CONFIG["host"],
        port=DB_CONFIG["port"],
        database=DB_CONFIG["database"],
        username=DB_CONFIG["username"],
        password=DB_CONFIG["password"],
    )
    builder = SuggestionIndexBuilder(es_client=es_client, db_engine=db_engine)
    result = builder.rebuild_tenant_index(
        tenant_id=args.tenant_id,
        days=args.days,
        recreate=args.recreate,
        batch_size=args.batch_size,
        min_query_len=args.min_query_len,
    )
    print(json.dumps(result, indent=2, ensure_ascii=False))
    return 0


def main():
    """Main CLI entry point."""
    parser = argparse.ArgumentParser(
        description='SearchEngine - E-Commerce Search SaaS',
        formatter_class=argparse.RawDescriptionHelpFormatter
    )

    subparsers = parser.add_subparsers(dest='command', help='Command to execute')

    # Serve command
    serve_parser = subparsers.add_parser('serve', help='Start API service (multi-tenant)')
    serve_parser.add_argument('--host', default='0.0.0.0', help='Host to bind to')
    serve_parser.add_argument('--port', type=int, default=6002, help='Port to bind to')
    serve_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
    serve_parser.add_argument('--reload', action='store_true', help='Enable auto-reload')

    # Serve-indexer command
    serve_indexer_parser = subparsers.add_parser(
        'serve-indexer',
        help='Start dedicated Indexer API service (indexer routes only)'
    )
    serve_indexer_parser.add_argument('--host', default='0.0.0.0', help='Host to bind to')
    serve_indexer_parser.add_argument('--port', type=int, default=6004, help='Port to bind to')
    serve_indexer_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
    serve_indexer_parser.add_argument('--reload', action='store_true', help='Enable auto-reload')

    # Search command
    search_parser = subparsers.add_parser('search', help='Test search from command line')
    search_parser.add_argument('query', help='Search query')
    search_parser.add_argument('--tenant-id', required=True, help='Tenant ID (required)')
    search_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
    search_parser.add_argument('--size', type=int, default=10, help='Number of results')
    search_parser.add_argument('--no-translation', action='store_true', help='Disable translation')
    search_parser.add_argument('--no-embedding', action='store_true', help='Disable embeddings')
    search_parser.add_argument('--json', action='store_true', help='Output JSON')

    # Suggestion build command
    suggest_build_parser = subparsers.add_parser(
        'build-suggestions',
        help='Build tenant suggestion index (full rebuild)'
    )
    suggest_build_parser.add_argument('--tenant-id', required=True, help='Tenant ID')
    suggest_build_parser.add_argument('--es-host', default='http://localhost:9200', help='Elasticsearch host')
    suggest_build_parser.add_argument('--days', type=int, default=30, help='Query log lookback days')
    suggest_build_parser.add_argument('--batch-size', type=int, default=500, help='Product scan batch size')
    suggest_build_parser.add_argument('--min-query-len', type=int, default=1, help='Minimum query length')
    suggest_build_parser.add_argument(
        '--recreate',
        action='store_true',
        help='Delete and recreate suggestion index before build'
    )

    args = parser.parse_args()

    if not args.command:
        parser.print_help()
        return 1

    # Execute command
    if args.command == 'serve':
        return cmd_serve(args)
    elif args.command == 'serve-indexer':
        return cmd_serve_indexer(args)
    elif args.command == 'search':
        return cmd_search(args)
    elif args.command == 'build-suggestions':
        return cmd_build_suggestions(args)
    else:
        print(f"Unknown command: {args.command}")
        return 1


if __name__ == "__main__":
    sys.exit(main())