"""Factory helpers for agent wallet backends."""

from __future__ import annotations

from pathlib import Path

from agent_wallet.bootstrap import ensure_solana_wallet_ready, ensure_wallet_pin
from agent_wallet.encrypted_storage import load_wallet_secret_material
from agent_wallet.config import (
    normalize_btc_network,
    normalize_solana_network,
    resolve_runtime_solana_rpc_config,
    resolve_runtime_solana_swap_config,
    resolve_solana_private_key,
    settings,
)
from agent_wallet.wallet_layer.base import AgentWalletBackend, WalletBackendError
from agent_wallet.wallet_layer.wdk_evm import WdkEvmLocalWalletBackend
from agent_wallet.wallet_layer.solana import SolanaLocalKeypairSigner, SolanaWalletBackend
from agent_wallet.wallet_layer.wdk_btc import WdkBtcLocalWalletBackend


def _load_keypair_material() -> str | None:
    secret = resolve_solana_private_key()
    if secret:
        return secret

    ensured = ensure_solana_wallet_ready()
    keypair_path = settings.solana_agent_keypair_path.strip() or (ensured["path"] if ensured else "")
    if not keypair_path:
        return None

    path = Path(keypair_path).expanduser()
    if not path.exists():
        raise WalletBackendError(
            f"Configured Solana keypair path does not exist: {path}"
        )
    secret_material, _ = load_wallet_secret_material(path)
    return secret_material.strip()


def create_wallet_backend() -> AgentWalletBackend | None:
    """Build the configured wallet backend instance."""
    backend = settings.agent_wallet_backend.strip().lower()
    if not backend or backend == "none":
        return None

    if backend in {"solana", "solana_local", "solana-local"}:
        solana_network = normalize_solana_network(settings.solana_network)
        secret_material = _load_keypair_material()
        signer = (
            SolanaLocalKeypairSigner.from_secret_material(secret_material)
            if secret_material
            else None
        )
        keypair_path = settings.solana_agent_keypair_path.strip()
        if signer and keypair_path:
            ensure_wallet_pin(
                Path(keypair_path).expanduser(),
                address=signer.address,
                network=solana_network,
            )
        configured_address = settings.solana_agent_public_key.strip() or None
        rpc_config = resolve_runtime_solana_rpc_config(
            solana_network,
            settings.solana_rpc_url,
            settings.solana_rpc_urls,
        )
        swap_config = resolve_runtime_solana_swap_config(solana_network)
        return SolanaWalletBackend(
            rpc_url=rpc_config["rpc_urls"],
            commitment=settings.solana_commitment,
            network=solana_network,
            signer=signer,
            address=configured_address,
            sign_only=settings.agent_wallet_sign_only,
            rpc_provider_mode=str(rpc_config["mode"]),
            rpc_provider=str(rpc_config["provider"]),
            rpc_transport=str(rpc_config["transport"]),
            swap_provider=str(swap_config["provider"]),
            swap_transport=str(swap_config["transport"]),
        )

    if backend in {"wdk_btc_local", "wdk-btc-local", "btc_local", "btc-local"}:
        return WdkBtcLocalWalletBackend(
            service_url=settings.wdk_btc_service_url,
            wallet_id=settings.wdk_btc_wallet_id,
            network=normalize_btc_network(settings.solana_network),
            account_index=settings.wdk_btc_account_index,
            sign_only=settings.agent_wallet_sign_only,
        )

    if backend in {"wdk_evm_local", "wdk-evm-local", "evm_local", "evm-local"}:
        return WdkEvmLocalWalletBackend(
            service_url=settings.wdk_evm_service_url,
            wallet_id=settings.wdk_evm_wallet_id,
            network=settings.solana_network,
            account_index=settings.wdk_evm_account_index,
            sign_only=settings.agent_wallet_sign_only,
        )

    raise WalletBackendError(
        f"Unsupported agent wallet backend: {backend}. "
        "Supported values: none, solana_local, wdk_btc_local, wdk_evm_local."
    )