Function main

vmm/src/vmm-cli.py:1504–2020

Parse arguments and dispatch to the appropriate command handler.

()

def main():
    """Parse arguments and dispatch to the appropriate command handler."""
    parser = argparse.ArgumentParser(description="dstack-vmm CLI - Manage VMs")

    # Load config file defaults
    config = load_config()

    # Discover running VMM instances
    instances = discover_vmm_instances()

    # Priority: command line > environment variable > config file > auto-discovery > default
    default_url = os.environ.get(
        "DSTACK_VMM_URL", config.get("url", "http://localhost:8080")
    )
    default_auth_user = os.environ.get("DSTACK_VMM_AUTH_USER", config.get("auth_user"))
    default_auth_password = os.environ.get(
        "DSTACK_VMM_AUTH_PASSWORD", config.get("auth_password")
    )

    parser.add_argument(
        "--url",
        default=default_url,
        help="dstack-vmm API URL (can also be set via DSTACK_VMM_URL env var or config file)",
    )

    # Basic authentication arguments
    parser.add_argument(
        "--auth-user",
        default=default_auth_user,
        help="Basic auth username (can also be set via DSTACK_VMM_AUTH_USER env var or config file)",
    )
    parser.add_argument(
        "--auth-password",
        default=default_auth_password,
        help="Basic auth password (can also be set via DSTACK_VMM_AUTH_PASSWORD env var or config file)",
    )

    subparsers = parser.add_subparsers(dest="command", help="Commands")

    # VMM discovery commands (vmm ls / vmm switch)
    vmm_parser = subparsers.add_parser(
        "vmm", help="VMM instance management (ls, switch)"
    )
    vmm_subparsers = vmm_parser.add_subparsers(
        dest="vmm_command", help="VMM sub-commands"
    )

    vmm_ls_parser = vmm_subparsers.add_parser(
        "ls", help="List all running VMM instances on this host"
    )
    vmm_ls_parser.add_argument(
        "--json", action="store_true", help="Output in JSON format"
    )

    vmm_switch_parser = vmm_subparsers.add_parser(
        "switch", help="Switch active VMM instance"
    )
    vmm_switch_parser.add_argument(