htcli 1.1.0__py3-none-any.whl

src/htcli/commands/overwatch/prompts.py

@@ -0,0 +1,359 @@
+"""
+Overwatch command prompting logic.
+
+Handles user interaction and input validation for overwatch operations.
+Uses HTCLI UI components for proper validation.
+"""
+
+from typing import Optional, Tuple
+
+from ...dependencies import get_client
+from ...ui.colors import info
+from ...ui.display import HTCLIConsole, print_error
+from ...ui.prompts import HTCLIPrompt, amount_prompt
+from ...utils.blockchain.peer_id import validate_peer_id_format
+from ...utils.wallet.crypto import format_address_display
+
+console = HTCLIConsole()
+prompt = HTCLIPrompt()
+
+
+def prompt_overwatch_node_id() -> int:
+    """Prompt for overwatch node ID."""
+    return prompt.integer_prompt(
+        "Enter the Overwatch Node ID",
+        min_value=1,
+    )
+
+
+def prompt_overwatch_register(
+    hotkey: Optional[str] = None,
+    stake_amount: Optional[float] = None,
+    coldkey_name: Optional[str] = None,
+) -> Tuple[str, int]:
+    """
+    Collect overwatch node registration parameters from user.
+
+    Returns:
+        Tuple of (hotkey_address, stake_amount_in_wei)
+    """
+    console.print(info("Overwatch Node Registration"))
+    console.print("Register a new overwatch node on the network.\n")
+
+    client = get_client()
+
+    # Get coldkey address if provided (for hotkey disambiguation)
+    coldkey_address = None
+    if coldkey_name:
+        try:
+            from ...utils.wallet.crypto import get_wallet_info_by_name
+            coldkey_info = get_wallet_info_by_name(coldkey_name, is_hotkey=False)
+            coldkey_address = coldkey_info.get("evm_address") or coldkey_info.get("ss58_address") or coldkey_info.get("address")
+        except Exception:
+            # Coldkey not found yet, will be resolved later - that's okay
+            pass
+
+    # Get hotkey - try to resolve first with coldkey context
+    if hotkey is not None:
+        try:
+            hotkey = client.offchain.wallet.resolve_hotkey_address(
+                hotkey,
+                owner_coldkey_name=coldkey_name,
+                owner_address=coldkey_address,
+            )
+            console.print(f"[htcli.success]✅ Using hotkey: {format_address_display(hotkey, truncate=False)}[/]")
+        except (ValueError, Exception) as e:
+            print_error(f"Invalid hotkey: {str(e)}")
+            hotkey = None
+
+    # Get hotkey if not provided or resolution failed
+    if hotkey is None:
+        from ...utils import retrieve_wallet_with_validation
+        from ...utils.wallet.crypto import get_wallet_info_by_name
+
+        console.print(info("\nHotkey Selection (used for overwatch node):"))
+        hotkey_wallet, hotkey_keypair = retrieve_wallet_with_validation(
+            wallet_type="hotkey", purpose="register the overwatch node"
+        )
+        # Get wallet info to retrieve the EVM address with coldkey context
+        hotkey_wallet_info = get_wallet_info_by_name(
+            hotkey_wallet,
+            is_hotkey=True,
+            owner_coldkey_name=coldkey_name,
+            owner_address=coldkey_address,
+        )
+        hotkey = (
+            hotkey_wallet_info.get("evm_address")
+            or hotkey_wallet_info.get("address")
+            or hotkey_wallet_info.get("ss58_address")
+        )
+        if not hotkey:
+            raise ValueError(f"Could not determine address for hotkey wallet '{hotkey_wallet}'")
+        console.print(f"[htcli.success]✅ Using hotkey: {format_address_display(hotkey, truncate=False)}[/]")
+
+    # Get stake amount - query blockchain for actual minimum
+    if stake_amount is None:
+        # Query the actual minimum stake balance from the blockchain
+        min_stake_wei = 0
+        try:
+            min_stake_result = client.substrate.query(
+                module="Network", storage_function="OverwatchMinStakeBalance"
+            )
+            min_stake_wei = min_stake_result.value if min_stake_result and min_stake_result.value else 0
+        except Exception:
+            # Fallback to a reasonable default if query fails
+            min_stake_wei = 0
+        
+        # Convert wei to TENSOR for the prompt
+        # Default fallback is 100 TENSOR (matches blockchain's DefaultOverwatchMinStakeBalance)
+        min_stake_tensor = min_stake_wei / 1e18 if min_stake_wei > 0 else 100.0
+        
+        console.print(f"[htcli.info]ℹ️  Minimum stake required: {min_stake_tensor:,.2f}[/]")
+        
+        stake_amount = amount_prompt(
+            "Enter initial stake amount",
+            currency="TENSOR",
+            min_amount=min_stake_tensor,
+        )
+
+    # Convert TENSOR to WEI
+    stake_amount_wei = int(stake_amount * 1e18)
+
+    return hotkey, stake_amount_wei
+
+
+def prompt_overwatch_remove() -> int:
+    """
+    Collect overwatch node removal parameters from user.
+
+    Returns:
+        Overwatch node ID to remove
+    """
+    console.print(info("Overwatch Node Removal"))
+    console.print("Remove your overwatch node from the network.\n")
+
+    return prompt.integer_prompt(
+        "Enter the Overwatch Node ID to remove",
+        min_value=1,
+    )
+
+
+def prompt_overwatch_peer_id(
+    subnet_id: Optional[int] = None,
+    overwatch_node_id: Optional[int] = None,
+    peer_id: Optional[str] = None,
+) -> Tuple[int, int, str]:
+    """
+    Collect parameters for setting overwatch node peer ID.
+
+    Returns:
+        Tuple of (subnet_id, overwatch_node_id, peer_id)
+    """
+    console.print(info("Set Overwatch Peer ID"))
+    console.print("Set your peer ID for a specific subnet.\n")
+
+    # Get subnet ID
+    if subnet_id is None:
+        subnet_id = prompt.integer_prompt(
+            "Enter the Subnet ID",
+            min_value=0,
+        )
+
+    # Get overwatch node ID
+    if overwatch_node_id is None:
+        overwatch_node_id = prompt.integer_prompt(
+            "Enter your Overwatch Node ID",
+            min_value=1,
+        )
+
+    # Validate peer_id if provided
+    if peer_id is not None:
+        if not peer_id or not isinstance(peer_id, str):
+            print_error("Invalid peer ID: must be a non-empty string")
+            peer_id = None
+        elif not validate_peer_id_format(peer_id):
+            print_error(
+                "Invalid peer ID format: must be a valid base58-encoded multihash "
+                "(e.g., starting with 'Qm', '12D3KooW', or '1')"
+            )
+            peer_id = None
+
+    # Get peer ID
+    if peer_id is None:
+        peer_id = prompt.text_prompt(
+            "Enter the Peer ID for this subnet",
+            validator=lambda x: (
+                True
+                if validate_peer_id_format(x)
+                else "Invalid peer ID format. Must be a valid base58-encoded multihash."
+            ),
+        )
+
+    return subnet_id, overwatch_node_id, peer_id
+
+
+def prompt_overwatch_add_stake(
+    overwatch_node_id: Optional[int] = None,
+    hotkey: Optional[str] = None,
+    stake_amount: Optional[float] = None,
+    coldkey_name: Optional[str] = None,
+) -> Tuple[int, str, int]:
+    """
+    Collect parameters for adding stake to an overwatch node.
+
+    Returns:
+        Tuple of (overwatch_node_id, hotkey_address, stake_amount_in_wei)
+    """
+    console.print(info("Add Overwatch Stake"))
+    console.print("Add stake to your overwatch node.\n")
+
+    client = get_client()
+
+    # Get overwatch node ID
+    if overwatch_node_id is None:
+        overwatch_node_id = prompt.integer_prompt(
+            "Enter the Overwatch Node ID",
+            min_value=1,
+        )
+
+    # Get coldkey address if provided (for hotkey disambiguation)
+    coldkey_address = None
+    if coldkey_name:
+        try:
+            from ...utils.wallet.crypto import get_wallet_info_by_name
+            coldkey_info = get_wallet_info_by_name(coldkey_name, is_hotkey=False)
+            coldkey_address = coldkey_info.get("evm_address") or coldkey_info.get("ss58_address") or coldkey_info.get("address")
+        except Exception:
+            # Coldkey not found yet, will be resolved later - that's okay
+            pass
+
+    # Get hotkey - try to resolve first with coldkey context
+    if hotkey is not None:
+        try:
+            hotkey = client.offchain.wallet.resolve_hotkey_address(
+                hotkey,
+                owner_coldkey_name=coldkey_name,
+                owner_address=coldkey_address,
+            )
+            console.print(f"[htcli.success]✅ Using hotkey: {format_address_display(hotkey, truncate=False)}[/]")
+        except (ValueError, Exception) as e:
+            print_error(f"Invalid hotkey: {str(e)}")
+            hotkey = None
+
+    # Get hotkey if not provided
+    if hotkey is None:
+        from ...utils import retrieve_wallet_with_validation
+        from ...utils.wallet.crypto import get_wallet_info_by_name
+
+        console.print(info("\nHotkey Selection (must match overwatch node):"))
+        hotkey_wallet, hotkey_keypair = retrieve_wallet_with_validation(
+            wallet_type="hotkey", purpose="identify the overwatch node"
+        )
+        # Get wallet info to retrieve the EVM address with coldkey context
+        hotkey_wallet_info = get_wallet_info_by_name(
+            hotkey_wallet,
+            is_hotkey=True,
+            owner_coldkey_name=coldkey_name,
+            owner_address=coldkey_address,
+        )
+        hotkey = (
+            hotkey_wallet_info.get("evm_address")
+            or hotkey_wallet_info.get("address")
+            or hotkey_wallet_info.get("ss58_address")
+        )
+        if not hotkey:
+            raise ValueError(f"Could not determine address for hotkey wallet '{hotkey_wallet}'")
+        console.print(f"[htcli.success]✅ Using hotkey: {format_address_display(hotkey, truncate=False)}[/]")
+
+    # Get stake amount
+    if stake_amount is None:
+        stake_amount = amount_prompt(
+            "Enter stake amount to add",
+            currency="TENSOR",
+            min_amount=0.001,  # Very small minimum
+        )
+
+    # Convert TENSOR to WEI
+    stake_amount_wei = int(stake_amount * 1e18)
+
+    return overwatch_node_id, hotkey, stake_amount_wei
+
+
+def prompt_overwatch_remove_stake(
+    hotkey: Optional[str] = None,
+    stake_amount: Optional[float] = None,
+    coldkey_name: Optional[str] = None,
+) -> Tuple[str, int]:
+    """
+    Collect parameters for removing stake from an overwatch node.
+
+    Returns:
+        Tuple of (hotkey_address, stake_amount_in_wei)
+    """
+    console.print(info("Remove Overwatch Stake"))
+    console.print("Remove stake from your overwatch node.\n")
+    console.print("[htcli.warning]⚠️  Removed stake will enter unbonding period before becoming available.[/]\n")
+
+    client = get_client()
+
+    # Get coldkey address if provided (for hotkey disambiguation)
+    coldkey_address = None
+    if coldkey_name:
+        try:
+            from ...utils.wallet.crypto import get_wallet_info_by_name
+            coldkey_info = get_wallet_info_by_name(coldkey_name, is_hotkey=False)
+            coldkey_address = coldkey_info.get("evm_address") or coldkey_info.get("ss58_address") or coldkey_info.get("address")
+        except Exception:
+            # Coldkey not found yet, will be resolved later - that's okay
+            pass
+
+    # Get hotkey - try to resolve first with coldkey context
+    if hotkey is not None:
+        try:
+            hotkey = client.offchain.wallet.resolve_hotkey_address(
+                hotkey,
+                owner_coldkey_name=coldkey_name,
+                owner_address=coldkey_address,
+            )
+            console.print(f"[htcli.success]✅ Using hotkey: {format_address_display(hotkey, truncate=False)}[/]")
+        except (ValueError, Exception) as e:
+            print_error(f"Invalid hotkey: {str(e)}")
+            hotkey = None
+
+    # Get hotkey if not provided
+    if hotkey is None:
+        from ...utils import retrieve_wallet_with_validation
+        from ...utils.wallet.crypto import get_wallet_info_by_name
+
+        console.print(info("\nHotkey Selection (must match overwatch node):"))
+        hotkey_wallet, hotkey_keypair = retrieve_wallet_with_validation(
+            wallet_type="hotkey", purpose="identify the overwatch node"
+        )
+        # Get wallet info to retrieve the EVM address with coldkey context
+        hotkey_wallet_info = get_wallet_info_by_name(
+            hotkey_wallet,
+            is_hotkey=True,
+            owner_coldkey_name=coldkey_name,
+            owner_address=coldkey_address,
+        )
+        hotkey = (
+            hotkey_wallet_info.get("evm_address")
+            or hotkey_wallet_info.get("address")
+            or hotkey_wallet_info.get("ss58_address")
+        )
+        if not hotkey:
+            raise ValueError(f"Could not determine address for hotkey wallet '{hotkey_wallet}'")
+        console.print(f"[htcli.success]✅ Using hotkey: {format_address_display(hotkey, truncate=False)}[/]")
+
+    # Get stake amount
+    if stake_amount is None:
+        stake_amount = amount_prompt(
+            "Enter stake amount to remove",
+            currency="TENSOR",
+            min_amount=0.001,  # Very small minimum
+        )
+
+    # Convert TENSOR to WEI
+    stake_amount_wei = int(stake_amount * 1e18)
+
+    return hotkey, stake_amount_wei