iwa 0.0.0__py3-none-any.whl → 0.0.1a2__py3-none-any.whl

iwa/tools/restore_backup.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+"""Validate and restore wallet backup."""
+
+import json
+import shutil
+import sys
+from pathlib import Path
+
+PROJECT_ROOT = Path(__file__).parent.parent.parent.parent
+DATA_DIR = PROJECT_ROOT / "data"
+WALLET_PATH = DATA_DIR / "wallet.json"
+BACKUP_DIR = DATA_DIR / "backup"
+
+
+def validate_wallet_backup(backup_path: Path) -> tuple[bool, str, int]:
+    """Validate that backup file is a valid wallet using Pydantic models.
+
+    Returns:
+        Tuple of (is_valid, message, account_count)
+
+    """
+    from iwa.core.keys import EncryptedAccount, StoredSafeAccount
+    from iwa.core.models import EthereumAddress
+
+    try:
+        with open(backup_path, "r", encoding="utf-8") as f:
+            data = json.load(f)
+    except json.JSONDecodeError as e:
+        return False, f"Invalid JSON: {e}", 0
+
+    if not isinstance(data, dict) or "accounts" not in data:
+        return False, "Missing 'accounts' key", 0
+
+    accounts = data.get("accounts", {})
+    if not isinstance(accounts, dict):
+        return False, "'accounts' must be a dictionary", 0
+
+    # Validate each account using Pydantic models
+    for address, account_data in accounts.items():
+        try:
+            # Validate address
+            EthereumAddress(address)
+
+            # Validate account structure
+            if "signers" in account_data:
+                StoredSafeAccount(**account_data)
+            else:
+                EncryptedAccount(**account_data)
+        except Exception as e:
+            return False, f"Invalid account {address[:10]}...: {e}", 0
+
+    return True, "Valid wallet structure", len(accounts)
+
+
+def restore_backup(backup_name: str) -> int:
+    """Restore wallet from backup with validation."""
+    backup_path = BACKUP_DIR / backup_name
+
+    if not backup_path.exists():
+        print(f"Error: Backup file not found: {backup_path}")
+        return 1
+
+    is_valid, message, num_accounts = validate_wallet_backup(backup_path)
+    if not is_valid:
+        print(f"Error: {message}")
+        return 1
+
+    print(f"Backup validated: {num_accounts} account(s) found")
+
+    # Restore
+    shutil.copy2(backup_path, WALLET_PATH)
+    print(f"Restored wallet from {backup_name}")
+    return 0
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: restore_backup.py <backup_filename>")
+        print("Example: restore_backup.py wallet.json.20260102_101400.bkp")
+        sys.exit(1)
+
+    sys.exit(restore_backup(sys.argv[1]))

iwa/tui/__init__.py

@@ -0,0 +1 @@
+"""TUI package for IWA."""

iwa/tui/app.py

@@ -0,0 +1,174 @@
+"""Main TUI Application module."""
+
+from loguru import logger
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.widgets import Footer, Header, TabbedContent, TabPane
+
+from iwa.core.wallet import Wallet
+from iwa.tui.rpc import RPCView
+from iwa.tui.screens.wallets import WalletsScreen
+
+
+class IwaApp(App):
+    """Iwa TUI Application."""
+
+    # ... (keep constants) ...
+    TITLE = "Iwa"
+
+    CSS = """
+    .header {
+        margin: 1 0;
+        text-style: bold;
+    }
+    .form-row {
+        height: 5;
+        margin: 1 0;
+        align: left middle;
+    }
+    .label {
+        margin: 1 1;
+        width: auto;
+    }
+    Input {
+        width: 15;
+        height: 3;
+        margin-right: 1;
+    }
+    Select {
+        width: 30;
+        height: 3;
+        margin-right: 1;
+    }
+    Button {
+        margin-right: 1;
+    }
+    #chain_row {
+        height: 3;
+        margin: 1 0;
+        align: left middle;
+    }
+    #chain_row Label {
+        margin: 1 1;
+        height: 1;
+    }
+    #tokens_row {
+        height: 3;
+        margin: 0 0;
+    }
+    #tokens_row Label {
+        margin: 1 1;
+        height: 1;
+    }
+    #token_toggles {
+        height: 3;
+        margin: 0 0;
+    }
+    #accounts_table {
+        margin-top: 1;
+        height: 1fr;
+        min-height: 10;
+    }
+    #tx_table {
+        height: 10;
+    }
+    Tab {
+        width: 20;
+    }
+    .btn-group {
+        width: 100%;
+        height: auto;
+        align: center middle;
+        margin-top: 1;
+        margin-bottom: 1;
+    }
+    .create-btn {
+        margin-left: 1;
+        margin-right: 1;
+        width: auto;
+        min-width: 20;
+    }
+    """
+
+    BINDINGS = [
+        Binding("q", "quit", "Quit"),
+        Binding("r", "refresh", "Refresh"),
+    ]
+
+    def __init__(self):
+        """Initialize the App."""
+        super().__init__()
+        # Configure logger for TUI
+        logger.remove()
+        logger.add("iwa.log", rotation="10 MB", level="INFO")
+
+        self.wallet = Wallet()
+
+        # Use PluginService from wallet
+        self.plugins = self.wallet.plugin_service.get_all_plugins()
+
+    def compose(self) -> ComposeResult:
+        """Compose the application layout.
+
+        Sets up the Header, TabbedContent with Wallets, Plugins, and RPC views,
+        and the Footer.
+
+        Returns:
+            ComposeResult: The widgets to be shown in the app.
+
+        """
+        yield Header(show_clock=True)
+
+        with TabbedContent(initial="wallets-tab"):
+            # Wallets first (default)
+            with TabPane("Wallets", id="wallets-tab"):
+                yield WalletsScreen(self.wallet)
+
+            # Plugin tabs (Olas)
+            for _name, plugin in self.plugins.items():
+                view = plugin.get_tui_view(wallet=self.wallet)
+                if view:
+                    with TabPane(plugin.name.capitalize(), id=f"{plugin.name.lower()}-tab"):
+                        yield view
+
+            # RPC Status last
+            with TabPane("RPC Status", id="rpc-tab"):
+                yield RPCView()
+
+        yield Footer()
+
+    def action_refresh(self) -> None:
+        """Action handler for the 'refresh' key binding.
+
+        Triggers a refresh of the currently active view. Currently specifically
+        targets the WalletsScreen to reload accounts and balances.
+        """
+        # Ideally, propagate refresh to active tab
+        # For now, just refresh wallets view explicitly if it's there
+        try:
+            wallets_screen = self.query_one(WalletsScreen)
+            wallets_screen.refresh_accounts()
+        except Exception:
+            pass
+
+    def copy_to_clipboard(self, text: str) -> None:
+        """Copy the provided text to the system clipboard.
+
+        Uses `pyperclip` to handle cross-platform clipboard operations.
+        Logs an error if the copy operation fails.
+
+        Args:
+            text: The string to copy.
+
+        """
+        try:
+            import pyperclip
+
+            pyperclip.copy(str(text))
+        except Exception as e:
+            logger.error(f"Failed to copy to clipboard: {e}")
+
+
+if __name__ == "__main__":
+    app = IwaApp()
+    app.run()

iwa/tui/modals/__init__.py

@@ -0,0 +1,5 @@
+"""Modals for the IWA TUI."""
+
+from .base import CreateEOAModal, CreateSafeModal
+
+__all__ = ["CreateEOAModal", "CreateSafeModal"]

iwa/tui/modals/base.py

@@ -0,0 +1,406 @@
+"""Modal screens for the IWA TUI."""
+
+from typing import List, Tuple
+
+from textual.app import ComposeResult
+from textual.containers import Horizontal, Vertical
+from textual.screen import ModalScreen
+from textual.widgets import (
+    Button,
+    Input,
+    Label,
+    Select,
+    SelectionList,
+)
+
+from iwa.core.chain import ChainInterfaces
+
+
+class CreateEOAModal(ModalScreen):
+    """Modal screen for creating a new EOA wallet."""
+
+    CSS = """
+    CreateEOAModal {
+        align: center middle;
+    }
+    #dialog {
+        padding: 1 2;
+        width: 60;
+        height: auto;
+        border: thick $background 80%;
+        background: $surface;
+    }
+    #dialog Label {
+        width: 100%;
+        margin-bottom: 1;
+    }
+    .header {
+        text-align: center;
+        text-style: bold;
+        margin-bottom: 2;
+    }
+    #tag_input {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    #btn_row {
+        height: 3;
+        width: 100%;
+        align: center middle;
+    }
+    Button {
+        margin: 0 1;
+    }
+    """
+
+    def compose(self) -> ComposeResult:
+        """Compose the modal UI."""
+        with Vertical(id="dialog"):
+            yield Label("Create New EOA Wallet", classes="header")
+            yield Label("Tag (Name):")
+            yield Input(placeholder="e.g. My EOA", id="tag_input")
+            with Horizontal(id="btn_row"):
+                yield Button("Cancel", id="cancel")
+                yield Button("Create", variant="primary", id="create")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press."""
+        if event.button.id == "create":
+            tag = self.query_one("#tag_input").value
+            self.dismiss(tag)
+        elif event.button.id == "cancel":
+            self.dismiss(None)
+
+
+class CreateSafeModal(ModalScreen):
+    """Modal screen for creating a new Safe wallet."""
+
+    CSS = """
+    CreateSafeModal {
+        align: center middle;
+    }
+    #dialog {
+        padding: 1 2;
+        width: 70;
+        height: auto;
+        max-height: 90%;
+        border: thick $background 80%;
+        background: $surface;
+        overflow-y: auto;
+    }
+    #dialog Label {
+        width: 100%;
+        margin-bottom: 1;
+    }
+    .header {
+        text-align: center;
+        text-style: bold;
+        margin-bottom: 2;
+    }
+    #tag_input {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    #threshold_input {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    SelectionList {
+        height: 8;
+        margin-bottom: 2;
+        border: solid $secondary;
+    }
+    #btn_row {
+        height: 3;
+        width: 100%;
+        align: center middle;
+    }
+    Button {
+        margin: 0 1;
+    }
+    """
+
+    def __init__(self, existing_accounts: List[Tuple[str, str]]):
+        """Init with list of (tag, address) tuples."""
+        super().__init__()
+        self.existing_accounts = existing_accounts
+
+    def compose(self) -> ComposeResult:
+        """Compose the modal UI."""
+        with Vertical(id="dialog"):
+            yield Label("Create New Safe Wallet", classes="header")
+
+            yield Label("Tag (Name):")
+            yield Input(placeholder="e.g. My Safe", id="tag_input")
+
+            yield Label("Threshold (Min signatures):")
+            yield Input(placeholder="1", id="threshold_input", type="integer")
+
+            yield Label("Owners (select multiple):")
+            options = [(f"{tag} ({addr})", addr) for tag, addr in self.existing_accounts]
+            yield SelectionList[str](*options, id="owners_list")
+
+            yield Label("Chains (select multiple):")
+            chain_options = [(name.title(), name) for name, _ in ChainInterfaces().items()]
+            yield SelectionList[str](*chain_options, id="chains_list")
+
+            with Horizontal(id="btn_row"):
+                yield Button("Cancel", id="cancel")
+                yield Button("Create", variant="primary", id="create")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press."""
+        if event.button.id == "create":
+            tag = self.query_one("#tag_input", Input).value
+            threshold = int(self.query_one("#threshold_input", Input).value or "1")
+            owners = self.query_one("#owners_list", SelectionList).selected
+            chains = self.query_one("#chains_list", SelectionList).selected
+            self.dismiss({"tag": tag, "threshold": threshold, "owners": owners, "chains": chains})
+        elif event.button.id == "cancel":
+            self.dismiss(None)
+
+
+class StakeServiceModal(ModalScreen):
+    """Modal screen for selecting a staking contract."""
+
+    CSS = """
+    StakeServiceModal {
+        align: center middle;
+    }
+    #dialog {
+        padding: 1 2;
+        width: 60;
+        height: auto;
+        border: thick $background 80%;
+        background: $surface;
+    }
+    #dialog Label {
+        width: 100%;
+        margin-bottom: 1;
+    }
+    .header {
+        text-align: center;
+        text-style: bold;
+        margin-bottom: 2;
+    }
+    Select {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    #btn_row {
+        height: 3;
+        width: 100%;
+        align: center middle;
+    }
+    Button {
+        margin: 0 1;
+    }
+    """
+
+    def __init__(self, contracts: List[Tuple[str, str]]):
+        """Init with list of (name, address) tuples."""
+        super().__init__()
+        self.contracts = contracts
+
+    def compose(self) -> ComposeResult:
+        """Compose the modal UI."""
+        with Vertical(id="dialog"):
+            yield Label("Stake Service", classes="header")
+            yield Label("Select Staking Contract:")
+            options = [(name, addr) for name, addr in self.contracts]
+            yield Select(options, prompt="Select a contract...", id="contract_select")
+            with Horizontal(id="btn_row"):
+                yield Button("Cancel", id="cancel")
+                yield Button("Stake", variant="primary", id="stake")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press."""
+        if event.button.id == "stake":
+            contract_address = self.query_one("#contract_select", Select).value
+            if contract_address == Select.BLANK:
+                return
+            self.dismiss(contract_address)
+        elif event.button.id == "cancel":
+            self.dismiss(None)
+
+
+class CreateServiceModal(ModalScreen):
+    """Modal screen for creating a new Olas service."""
+
+    CSS = """
+    CreateServiceModal {
+        align: center middle;
+    }
+    #dialog {
+        padding: 1 2;
+        width: 65;
+        height: auto;
+        border: thick $background 80%;
+        background: $surface;
+    }
+    #dialog Label {
+        width: 100%;
+        margin-bottom: 1;
+    }
+    .header {
+        text-align: center;
+        text-style: bold;
+        margin-bottom: 2;
+    }
+    Input {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    Select {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    #btn_row {
+        height: 3;
+        width: 100%;
+        align: center middle;
+    }
+    Button {
+        margin: 0 1;
+    }
+    """
+
+    def __init__(
+        self,
+        chains: List[str],
+        default_chain: str = "gnosis",
+        staking_contracts: List[Tuple[str, str]] = None,
+    ):
+        """Init with list of available chains and staking contracts."""
+        super().__init__()
+        self.chains = chains
+        self.default_chain = default_chain
+        self.staking_contracts = staking_contracts or []
+
+    def compose(self) -> ComposeResult:
+        """Compose the modal UI."""
+        with Vertical(id="dialog"):
+            yield Label("Create New Olas Service", classes="header")
+
+            yield Label("Service Name:")
+            yield Input(placeholder="e.g. My Trader", id="name_input")
+
+            yield Label("Chain:")
+            chain_options = [(c.title(), c) for c in self.chains]
+            yield Select(chain_options, value=self.default_chain, id="chain_select")
+
+            yield Label("Agent Type:")
+            agent_options = [("Trader", "trader")]
+            yield Select(agent_options, value="trader", id="agent_type_select")
+
+            yield Label("Staking Contract:")
+            contract_options = [("None (don't stake)", "")]
+            contract_options.extend([(name, addr) for name, addr in self.staking_contracts])
+            yield Select(contract_options, value="", id="staking_select")
+
+            with Horizontal(id="btn_row"):
+                yield Button("Cancel", id="cancel")
+                yield Button("Create", variant="primary", id="create")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press."""
+        if event.button.id == "create":
+            name = self.query_one("#name_input", Input).value
+            chain = self.query_one("#chain_select", Select).value
+            agent_type = self.query_one("#agent_type_select", Select).value
+            staking_contract = self.query_one("#staking_select", Select).value
+            if not name or chain == Select.BLANK:
+                return
+            self.dismiss(
+                {
+                    "name": name,
+                    "chain": chain,
+                    "agent_type": agent_type if agent_type != Select.BLANK else "trader",
+                    "staking_contract": staking_contract
+                    if staking_contract != Select.BLANK
+                    else None,
+                }
+            )
+        elif event.button.id == "cancel":
+            self.dismiss(None)
+
+
+class FundServiceModal(ModalScreen):
+    """Modal screen for funding Olas service accounts."""
+
+    CSS = """
+    FundServiceModal {
+        align: center middle;
+    }
+    #dialog {
+        padding: 1 2;
+        width: 60;
+        height: auto;
+        border: thick $background 80%;
+        background: $surface;
+    }
+    #dialog Label {
+        width: 100%;
+        margin-bottom: 1;
+    }
+    .header {
+        text-align: center;
+        text-style: bold;
+        margin-bottom: 2;
+    }
+    .desc {
+        color: $text-muted;
+        margin-bottom: 2;
+    }
+    Input {
+        width: 100%;
+        margin-bottom: 2;
+    }
+    #btn_row {
+        height: 3;
+        width: 100%;
+        align: center middle;
+    }
+    Button {
+        margin: 0 1;
+    }
+    """
+
+    def __init__(self, service_key: str, native_symbol: str = "xDAI"):
+        """Init with service key and native currency symbol."""
+        super().__init__()
+        self.service_key = service_key
+        self.native_symbol = native_symbol
+
+    def compose(self) -> ComposeResult:
+        """Compose the modal UI."""
+        with Vertical(id="dialog"):
+            yield Label("Fund Service", classes="header")
+            yield Label(f"Send {self.native_symbol} from master wallet:", classes="desc")
+            yield Label(f"Agent Amount ({self.native_symbol}):")
+            yield Input(placeholder="0.0", id="agent_amount", type="number")
+            yield Label(f"Safe Amount ({self.native_symbol}):")
+            yield Input(placeholder="0.0", id="safe_amount", type="number")
+            with Horizontal(id="btn_row"):
+                yield Button("Cancel", id="cancel")
+                yield Button("Fund", variant="primary", id="fund")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press."""
+        if event.button.id == "fund":
+            try:
+                agent_amount = float(self.query_one("#agent_amount", Input).value or "0")
+                safe_amount = float(self.query_one("#safe_amount", Input).value or "0")
+            except ValueError:
+                return
+            if agent_amount <= 0 and safe_amount <= 0:
+                return
+            self.dismiss(
+                {
+                    "service_key": self.service_key,
+                    "agent_amount": agent_amount,
+                    "safe_amount": safe_amount,
+                }
+            )
+        elif event.button.id == "cancel":
+            self.dismiss(None)