Files
aitbc/cli/main.py
aitbc1 c0952c2525 refactor: flatten CLI directory structure - remove 'box in a box'
BEFORE:
/opt/aitbc/cli/
├── aitbc_cli/                    # Python package (box in a box)
│   ├── commands/
│   ├── main.py
│   └── ...
├── setup.py

AFTER:
/opt/aitbc/cli/                    # Flat structure
├── commands/                      # Direct access
├── main.py                        # Direct access
├── auth/
├── config/
├── core/
├── models/
├── utils/
├── plugins.py
└── setup.py

CHANGES MADE:
- Moved all files from aitbc_cli/ to cli/ root
- Fixed all relative imports (from . to absolute imports)
- Updated setup.py entry point: aitbc_cli.main → main
- Added CLI directory to Python path in entry script
- Simplified deployment.py to remove dependency on deleted core.deployment
- Fixed import paths in all command files
- Recreated virtual environment with new structure

BENEFITS:
- Eliminated 'box in a box' nesting
- Simpler directory structure
- Direct access to all modules
- Cleaner imports
- Easier maintenance and development
- CLI works with both 'python main.py' and 'aitbc' commands
2026-03-26 09:12:02 +01:00

316 lines
9.5 KiB
Python
Executable File

#!/usr/bin/env python3
"""
AITBC CLI - Main entry point for the AITBC Command Line Interface
"""
import click
import sys
from typing import Optional
from __init__ 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()