aitbc/cli/core/main.py

#!/usr/bin/env python3
"""
AITBC CLI - Main entry point for the AITBC Command Line Interface
"""

import click
import sys
from typing import Optional

from . import __version__
from config import get_config


def with_role(role: str):
    """Decorator to set role for command groups"""
    def decorator(func):
        @click.pass_context
        def wrapper(ctx, *args, **kwargs):
            ctx.parent.detected_role = role
            return func(ctx, *args, **kwargs)
        return wrapper
    return decorator
from utils import output, setup_logging
from commands.client import client
from commands.miner import miner
from commands.wallet import wallet
from commands.auth import auth
from commands.blockchain import blockchain
from commands.marketplace import marketplace
from commands.simulate import simulate
from commands.admin import admin
from commands.config import config
from commands.monitor import monitor
from commands.governance import governance
from commands.exchange import exchange
from commands.oracle import oracle
from commands.market_maker import market_maker
from commands.multisig import multisig
from commands.genesis_protection import genesis_protection
from commands.transfer_control import transfer_control
from commands.agent import agent
from commands.multimodal import multimodal
from commands.optimize import optimize
# from commands.openclaw import openclaw  # Temporarily disabled due to naming conflict
from commands.marketplace_advanced import advanced  # Re-enabled after fixing registration issues
from commands.swarm import swarm
from commands.chain import chain
from commands.genesis import genesis
from commands.keystore import keystore
from commands.test_cli import test
from commands.node import node
from commands.analytics import analytics
from commands.agent_comm import agent_comm
from commands.deployment import deploy
from commands.cross_chain import cross_chain
from commands.compliance import compliance
from commands.surveillance import surveillance
from commands.regulatory import regulatory
from commands.ai_trading import ai_trading
from commands.advanced_analytics import advanced_analytics_group
from commands.ai_surveillance import ai_surveillance_group

# AI provider commands
from commands.ai import ai_group

# Enterprise integration (optional)
try:
    from commands.enterprise_integration import enterprise_integration_group
except ImportError:
    enterprise_integration_group = None

from commands.sync import sync
from commands.explorer import explorer
from .plugins import plugin, load_plugins


@click.group()
@click.option(
    "--url",
    default=None,
    help="Coordinator API URL (overrides config)"
)
@click.option(
    "--api-key",
    default=None,
    help="API key (overrides config)"
)
@click.option(
    "--output",
    type=click.Choice(["table", "json", "yaml"]),
    default="table",
    help="Output format"
)
@click.option(
    "--verbose", "-v",
    count=True,
    help="Increase verbosity (use -v, -vv, -vvv)"
)
@click.option(
    "--debug",
    is_flag=True,
    help="Enable debug mode"
)
@click.option(
    "--config-file",
    default=None,
    help="Path to config file"
)
@click.option(
    "--test-mode",
    is_flag=True,
    help="Enable test mode (uses mock data and test endpoints)"
)
@click.option(
    "--dry-run",
    is_flag=True,
    help="Dry run mode (show what would be done without executing)"
)
@click.option(
    "--timeout",
    type=int,
    default=30,
    help="Request timeout in seconds (useful for testing)"
)
@click.option(
    "--no-verify",
    is_flag=True,
    help="Skip SSL certificate verification (testing only)"
)
@click.version_option(version=__version__, prog_name="aitbc")
@click.pass_context
def cli(ctx, url: Optional[str], api_key: Optional[str], output: str, 
        verbose: int, debug: bool, config_file: Optional[str], test_mode: bool,
        dry_run: bool, timeout: int, no_verify: bool):
    """
    AITBC CLI - Command Line Interface for AITBC Network
    
    Manage jobs, mining, wallets, blockchain operations, marketplaces, and AI services.
    
    CORE COMMANDS:
      client          Submit and manage AI compute jobs
      miner           GPU mining operations and status
      wallet          Wallet management and transactions
      marketplace     GPU marketplace and trading
      blockchain      Blockchain operations and queries
      exchange        Real exchange integration (Binance, Coinbase, etc.)
      explorer        Blockchain explorer and analytics
      
    ADVANCED FEATURES:
      analytics       Chain performance monitoring and predictions
      ai-trading      AI-powered trading strategies
      surveillance    Market surveillance and compliance
      compliance      Regulatory compliance and reporting
      governance      Network governance and proposals
      
    DEVELOPMENT TOOLS:
      admin           Administrative operations
      config          Configuration management
      monitor         System monitoring and health
      test            CLI testing and validation
      deploy          Deployment and infrastructure management
      
    SPECIALIZED SERVICES:
      agent           AI agent operations
      multimodal      Multi-modal AI processing
      oracle          Price discovery and data feeds
      market-maker    Automated market making
      genesis-protection Advanced security features
      
    Use 'aitbc <command> --help' for detailed help on any command.
    
    Examples:
      aitbc client submit --prompt "Generate an image" --model llama2
      aitbc miner status
      aitbc wallet create --type hd
      aitbc marketplace list
      aitbc exchange create-pair --pair AITBC/BTC --base-asset AITBC --quote-asset BTC
      aitbc analytics summary
      aitbc explorer status
      aitbc explorer block 12345
      aitbc explorer transaction 0x123...
      aitbc explorer search --address 0xabc...
    """
    # Ensure context object exists
    ctx.ensure_object(dict)
    
    # Setup logging based on verbosity
    log_level = setup_logging(verbose, debug)
    
    # Detect role from command name (before config is loaded)
    role = None
    
    # Check invoked_subcommand first
    if ctx.invoked_subcommand:
        if ctx.invoked_subcommand == 'client':
            role = 'client'
        elif ctx.invoked_subcommand == 'miner':
            role = 'miner'
        elif ctx.invoked_subcommand == 'blockchain':
            role = 'blockchain'
        elif ctx.invoked_subcommand == 'admin':
            role = 'admin'
    
    # Also check if role was already set by command group
    if not role:
        role = getattr(ctx, 'detected_role', None)
    
    # Load configuration with role
    config = get_config(config_file, role=role)
    
    # Override config with command line options
    if url:
        config.coordinator_url = url
    if api_key:
        config.api_key = api_key
    
    # Store in context for subcommands
    ctx.obj['config'] = config
    ctx.obj['output_format'] = output
    ctx.obj['log_level'] = log_level
    ctx.obj['test_mode'] = test_mode
    ctx.obj['dry_run'] = dry_run
    ctx.obj['timeout'] = timeout
    ctx.obj['no_verify'] = no_verify
    
    # Apply test mode settings
    if test_mode:
        config.coordinator_url = config.coordinator_url or "http://localhost:8000"
        config.api_key = config.api_key or "test-api-key"


# Add command groups
cli.add_command(client)
cli.add_command(miner)
cli.add_command(wallet)
cli.add_command(plugin)
# cli.add_command(openclaw)  # Temporarily disabled due to naming conflict
cli.add_command(advanced)  # Re-enabled after fixing registration issues
cli.add_command(auth)
cli.add_command(blockchain)
cli.add_command(marketplace)
cli.add_command(simulate)
cli.add_command(admin)
cli.add_command(config)
cli.add_command(monitor)
cli.add_command(governance)
cli.add_command(exchange)
cli.add_command(oracle)
cli.add_command(market_maker)
cli.add_command(multisig)
cli.add_command(genesis_protection)
cli.add_command(transfer_control)
cli.add_command(agent)
cli.add_command(multimodal)
cli.add_command(optimize)
cli.add_command(ai_group)
# cli.add_command(openclaw)  # Temporarily disabled
cli.add_command(swarm)
cli.add_command(chain)
cli.add_command(genesis)
cli.add_command(keystore)
cli.add_command(test)
cli.add_command(node)
cli.add_command(analytics)
cli.add_command(agent_comm)
cli.add_command(deploy)
cli.add_command(cross_chain)
cli.add_command(compliance)
cli.add_command(surveillance)
cli.add_command(regulatory)
cli.add_command(ai_trading)
cli.add_command(advanced_analytics_group)
cli.add_command(ai_surveillance_group)
if enterprise_integration_group is not None:
    cli.add_command(enterprise_integration_group)
cli.add_command(sync)
cli.add_command(explorer)
cli.add_command(plugin)
load_plugins(cli)


@cli.command()
@click.pass_context
def version(ctx):
    """Show version information"""
    output(f"AITBC CLI version {__version__}", ctx.obj['output_format'])


@cli.command()
@click.pass_context
def config_show(ctx):
    """Show current configuration"""
    config = ctx.obj['config']
    output({
        "coordinator_url": config.coordinator_url,
        "api_key": "***REDACTED***" if config.api_key else None,
        "output_format": ctx.obj['output_format'],
        "config_file": config.config_file
    }, ctx.obj['output_format'])


def main():
    """Main entry point"""
    try:
        cli()
    except KeyboardInterrupt:
        click.echo("\nAborted by user", err=True)
        sys.exit(1)
    except Exception as e:
        click.echo(f"Error: {e}", err=True)
        sys.exit(1)


if __name__ == "__main__":
    main()