aitbc/cli/aitbc_cli.py

#!/usr/bin/env python3
"""Compatibility wrapper for the AITBC CLI entrypoint."""

from __future__ import annotations

import importlib.util
import sys
from pathlib import Path
from types import ModuleType

REPO_ROOT = Path(__file__).resolve().parents[1]
CLI_DIR = Path(__file__).resolve().parent
sys.path.insert(0, str(REPO_ROOT))
sys.path.insert(0, str(CLI_DIR))

# Ensure we don't pick up hermes-agent's cli module
hermes_cli_path = "/usr/local/lib/hermes-agent"
if hermes_cli_path in sys.path:
    sys.path.remove(hermes_cli_path)

from aitbc.constants import BLOCKCHAIN_RPC_PORT

DEFAULT_RPC_URL = f"http://localhost:{BLOCKCHAIN_RPC_PORT}"
_CLI_MODULE: ModuleType | None = None


def _load_cli_module() -> ModuleType:
    global _CLI_MODULE
    if _CLI_MODULE is not None:
        return _CLI_MODULE

    # Try the new unified_cli.py location first
    cli_path = CLI_DIR / "unified_cli.py"
    if not cli_path.exists():
        # Fallback to old location
        cli_path = REPO_ROOT / "aitbc_cli" / "core" / "main.py"
    
    spec = importlib.util.spec_from_file_location("aitbc_cli_unified", cli_path)
    if spec is None or spec.loader is None:
        raise ImportError(f"Unable to load CLI entrypoint from {cli_path}")

    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    _CLI_MODULE = module
    return module


def main(argv=None):
    # Directly execute unified_cli without circular import
    import sys
    sys.path.insert(0, str(CLI_DIR))
    
    # Import unified_cli module directly
    import unified_cli
    
    # Create a mock core dict for the CLI
    from aitbc.constants import BLOCKCHAIN_RPC_PORT
    
    # Stub handler functions for all parser handlers
    def stub_handler(*args, **kwargs):
        return None
    
    core = {
        "DEFAULT_RPC_URL": f"http://localhost:{BLOCKCHAIN_RPC_PORT}",
        "DEFAULT_COORDINATOR_URL": "http://localhost:9001",
        "DEFAULT_GPU_URL": "http://localhost:8101",
        "DEFAULT_MARKETPLACE_URL": "http://localhost:8001",
        "DEFAULT_TRADING_URL": "http://localhost:8104",
        "DEFAULT_GOVERNANCE_URL": "http://localhost:8105",
        "CLI_VERSION": "0.0.0",
        # Add stub functions for core operations
        "create_wallet": lambda *args, **kwargs: {"status": "stub"},
        "list_wallets": lambda *args, **kwargs: [],
        "get_balance": lambda *args, **kwargs: 0,
        "get_transactions": lambda *args, **kwargs: [],
        "send_transaction": lambda *args, **kwargs: {"status": "stub"},
        "import_wallet": lambda *args, **kwargs: {"status": "stub"},
        "export_wallet": lambda *args, **kwargs: {"status": "stub"},
        "delete_wallet": lambda *args, **kwargs: {"status": "stub"},
        "rename_wallet": lambda *args, **kwargs: {"status": "stub"},
        "send_batch_transactions": lambda *args, **kwargs: {"status": "stub"},
        "get_chain_info": lambda *args, **kwargs: {},
        "get_blockchain_analytics": lambda *args, **kwargs: {},
        "marketplace_operations": lambda *args, **kwargs: {},
        "ai_operations": lambda *args, **kwargs: {},
        "mining_operations": lambda *args, **kwargs: {},
        "agent_operations": lambda *args, **kwargs: {},
        "hermes_training_operations": lambda *args, **kwargs: {},
        "workflow_operations": lambda *args, **kwargs: {},
        "resource_operations": lambda *args, **kwargs: {},
        "simulate_blockchain": lambda *args, **kwargs: None,
        "simulate_wallets": lambda *args, **kwargs: None,
        "simulate_price": lambda *args, **kwargs: None,
        "simulate_network": lambda *args, **kwargs: None,
        "simulate_ai_jobs": lambda *args, **kwargs: None,
        # Add stub handlers for all parser handlers
        "handle_market_gpu_register": stub_handler,
        "handle_market_gpu_list": stub_handler,
        "handle_market_listings": stub_handler,
        "handle_market_create": stub_handler,
        "handle_market_get": stub_handler,
        "handle_market_delete": stub_handler,
        "handle_market_buy": stub_handler,
        "handle_market_sell": stub_handler,
        "handle_market_orders": stub_handler,
        "handle_market_list_plugins": stub_handler,
    }
    
    return unified_cli.run_cli(argv, core)


if __name__ == "__main__":
    raise SystemExit(main())