iwa 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

iwa/web/dependencies.py

@@ -9,8 +9,42 @@ from loguru import logger
 
 from iwa.core.wallet import Wallet
 
-# Singleton wallet instance for the web app
-wallet = Wallet()
+# Singleton wallet instance (lazy-initialized or injected)
+_wallet: Optional[Wallet] = None
+
+
+def get_wallet() -> Wallet:
+    """Get the wallet instance (lazy initialization or injected).
+
+    Returns the injected wallet if set_wallet() was called,
+    otherwise creates a new Wallet on first access.
+    """
+    global _wallet
+    if _wallet is None:
+        _wallet = Wallet()
+    return _wallet
+
+
+def set_wallet(wallet: Wallet) -> None:
+    """Inject an external wallet instance.
+
+    Call this BEFORE importing routers to share a wallet
+    with an external application (e.g., Triton).
+    """
+    global _wallet
+    _wallet = wallet
+
+
+# Backwards compatibility: module-level wallet property
+# Deprecated: use get_wallet() instead
+class _WalletProxy:
+    """Proxy that redirects to get_wallet() for backwards compatibility."""
+
+    def __getattr__(self, name: str):
+        return getattr(get_wallet(), name)
+
+
+wallet = _WalletProxy()
 
 # Authentication
 api_key_header = APIKeyHeader(name="X-API-Key", auto_error=False)

iwa/web/server.py

@@ -147,11 +147,28 @@ async def root():
 
 
 def run_server(host: str = "127.0.0.1", port: int = 8000):
-    """Run the web server using uvicorn."""
+    """Run the web server using uvicorn (blocking)."""
     import uvicorn
 
     uvicorn.run(app, host=host, port=port)
 
 
+async def run_server_async(host: str = "127.0.0.1", port: int = 8000):
+    """Run the web server in an async context (non-blocking).
+
+    Use this to embed the web server in another async application.
+    The server runs until cancelled.
+
+    Args:
+        host: Bind address. Default "127.0.0.1" (localhost only for security).
+        port: Port number. Default 8000.
+    """
+    import uvicorn
+
+    config = uvicorn.Config(app, host=host, port=port, log_level="warning")
+    server = uvicorn.Server(config)
+    await server.serve()
+
+
 if __name__ == "__main__":
     run_server()

iwa/web/tests/test_wallet_injection.py

@@ -0,0 +1,204 @@
+"""Tests for wallet injection in web dependencies."""
+
+from unittest.mock import MagicMock, patch
+
+
+class TestWalletInjection:
+    """Tests for get_wallet, set_wallet, and _WalletProxy."""
+
+    def setup_method(self):
+        """Reset the global wallet state before each test."""
+        # Import fresh to reset state
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+    def test_get_wallet_lazy_initialization(self):
+        """get_wallet() creates a Wallet on first call if none injected."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            mock_wallet_cls.return_value = MagicMock(name="LazyWallet")
+
+            result = deps.get_wallet()
+
+            mock_wallet_cls.assert_called_once()
+            assert result == mock_wallet_cls.return_value
+
+    def test_get_wallet_returns_same_instance(self):
+        """get_wallet() returns the same instance on subsequent calls."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            mock_wallet_cls.return_value = MagicMock(name="SingletonWallet")
+
+            first = deps.get_wallet()
+            second = deps.get_wallet()
+
+            # Should only create once
+            mock_wallet_cls.assert_called_once()
+            assert first is second
+
+    def test_set_wallet_injects_external_wallet(self):
+        """set_wallet() allows injecting an external wallet instance."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        external_wallet = MagicMock(name="ExternalWallet")
+        deps.set_wallet(external_wallet)
+
+        result = deps.get_wallet()
+
+        assert result is external_wallet
+
+    def test_set_wallet_prevents_lazy_init(self):
+        """set_wallet() prevents Wallet() from being called."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            external_wallet = MagicMock(name="InjectedWallet")
+            deps.set_wallet(external_wallet)
+
+            result = deps.get_wallet()
+
+            # Wallet() should NOT be called
+            mock_wallet_cls.assert_not_called()
+            assert result is external_wallet
+
+    def test_set_wallet_overrides_existing(self):
+        """set_wallet() can override a previously set wallet."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        wallet1 = MagicMock(name="Wallet1")
+        wallet2 = MagicMock(name="Wallet2")
+
+        deps.set_wallet(wallet1)
+        assert deps.get_wallet() is wallet1
+
+        deps.set_wallet(wallet2)
+        assert deps.get_wallet() is wallet2
+
+
+class TestWalletProxy:
+    """Tests for _WalletProxy backwards compatibility."""
+
+    def setup_method(self):
+        """Reset the global wallet state before each test."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+    def test_wallet_proxy_forwards_attribute_access(self):
+        """wallet.attribute forwards to get_wallet().attribute."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        mock_wallet = MagicMock()
+        mock_wallet.balance_service = MagicMock(name="BalanceService")
+        deps.set_wallet(mock_wallet)
+
+        # Access through the proxy
+        result = deps.wallet.balance_service
+
+        assert result is mock_wallet.balance_service
+
+    def test_wallet_proxy_forwards_method_calls(self):
+        """wallet.method() forwards to get_wallet().method()."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        mock_wallet = MagicMock()
+        mock_wallet.get_accounts_balances.return_value = ({"0x1": {}}, {"0x1": {}})
+        deps.set_wallet(mock_wallet)
+
+        # Call method through proxy
+        result = deps.wallet.get_accounts_balances("gnosis", ["native"])
+
+        mock_wallet.get_accounts_balances.assert_called_once_with("gnosis", ["native"])
+        assert result == ({"0x1": {}}, {"0x1": {}})
+
+    def test_wallet_proxy_triggers_lazy_init(self):
+        """Accessing wallet.attr when no wallet set triggers lazy init."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            mock_instance = MagicMock()
+            mock_instance.key_storage = MagicMock(name="KeyStorage")
+            mock_wallet_cls.return_value = mock_instance
+
+            # Access through proxy should trigger lazy init
+            result = deps.wallet.key_storage
+
+            mock_wallet_cls.assert_called_once()
+            assert result is mock_instance.key_storage
+
+
+class TestIntegrationWithRouters:
+    """Integration tests for wallet injection with routers."""
+
+    def setup_method(self):
+        """Reset the global wallet state before each test."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+    def test_injected_wallet_used_by_routers(self):
+        """Routers using 'wallet' get the injected instance."""
+        import iwa.web.dependencies as deps
+
+        # Inject a mock wallet BEFORE routers access it
+        mock_wallet = MagicMock()
+        mock_wallet.key_storage = MagicMock()
+        mock_wallet.key_storage.accounts = {}
+        deps.set_wallet(mock_wallet)
+
+        # Simulate what a router does
+        from iwa.web.dependencies import wallet
+
+        # Access through the module-level wallet (proxy)
+        accounts = wallet.key_storage.accounts
+
+        assert accounts == {}
+        # Verify it's using our injected wallet
+        assert deps.get_wallet() is mock_wallet
+
+
+class TestServerAsync:
+    """Tests for run_server_async function."""
+
+    def test_run_server_async_source_has_localhost_default(self):
+        """run_server_async defaults to localhost for security (source check)."""
+        import pathlib
+
+        # Read the source file directly to verify default
+        server_path = pathlib.Path(__file__).parent.parent / "server.py"
+        source = server_path.read_text()
+
+        # Verify the function signature has localhost default
+        assert 'async def run_server_async(host: str = "127.0.0.1"' in source
+
+    def test_run_server_source_has_async_function(self):
+        """run_server_async function exists in server.py source."""
+        import pathlib
+
+        server_path = pathlib.Path(__file__).parent.parent / "server.py"
+        source = server_path.read_text()
+
+        # Verify the async function exists
+        assert "async def run_server_async" in source
+        assert "uvicorn.Config" in source
+        assert "uvicorn.Server" in source
+        assert "await server.serve()" in source

{iwa-0.1.3.dist-info → iwa-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iwa
-Version: 0.1.3
+Version: 0.1.4
 Summary: A secure, modular, and plugin-based framework for crypto agents and ops
 Requires-Python: <4.0,>=3.12
 Description-Content-Type: text/markdown

{iwa-0.1.3.dist-info → iwa-0.1.4.dist-info}/RECORD

@@ -146,9 +146,9 @@ iwa/tui/tests/test_wallets_refactor.py,sha256=71G3HLbhTtgDy3ffVbYv0MFYRgdYd-NWGB
 iwa/tui/tests/test_widgets.py,sha256=C9UgIGeWRaQ459JygFEQx-7hOi9mWrSUDDIMZH1ge50,3994
 iwa/tui/widgets/__init__.py,sha256=UzD6nJbwv9hOtkWl9I7faXm1a-rcu4xFRxrf4KBwwY4,161
 iwa/tui/widgets/base.py,sha256=Z8FigMhsfD76PkFVERqMaotd-xwXfuFZm_8TmCMOsl4,3381
-iwa/web/dependencies.py,sha256=0_dAJlRh6gKrUDRPKUe92eshFsg572yx_H0lQgSqGDA,2103
+iwa/web/dependencies.py,sha256=iTvdCSuETFLeQPtFydi21s1EKA_a80rbA57s-994fMQ,2980
 iwa/web/models.py,sha256=MSD9WPy_Nz_amWgoo2KSDTn4ZLv_AV0o0amuNtSf-68,3035
-iwa/web/server.py,sha256=4ZLVFEKoGs_NoCcXMeyYzDNdxUXazjwHQaX7CR1pwHE,5239
+iwa/web/server.py,sha256=BcaLtmyPSsVehUtbWf4fbEGJ4E0bDlY_PzVq479HoZM,5786
 iwa/web/routers/accounts.py,sha256=VhCHrwzRWqZcSW-tTEqFWT5hFl-IYEWpqXeuN8xM3-4,3922
 iwa/web/routers/state.py,sha256=wsBAOIWZeMWjMwLiiWVhuEXHAceI0IIq6CPpmh7SbIc,2469
 iwa/web/routers/swap.py,sha256=8xycAytquR29ELxW3vx428W8s9bI_w_x2kpRhhJ0KXY,22630
@@ -162,11 +162,12 @@ iwa/web/routers/olas/staking.py,sha256=jktJ2C1Q9X4aC0tWJByN3sHpEXY0EIvr3rr4N0MtX
 iwa/web/static/app.js,sha256=VCm9zPJERb9pX6zbFQ_7D47UnztGdibrkZ1dmwhsvdc,114949
 iwa/web/static/index.html,sha256=Qg06MFK__2KxwWTr8hhjX_GwsoN53QsLCTypu4fBR-Q,28516
 iwa/web/static/style.css,sha256=7i6T96pS7gXSLDZfyp_87gRlyB9rpsFWJEHJ-dRY1ug,24371
+iwa/web/tests/test_wallet_injection.py,sha256=YjoQh1FMwroswF6O_XZOwYXzanv-JYElNTDNKxAS77g,6751
 iwa/web/tests/test_web_endpoints.py,sha256=vA25YghHNB23sbmhD4ciesn_f_okSq0tjlkrSiKZ0rs,24007
 iwa/web/tests/test_web_olas.py,sha256=GunKEAzcbzL7FoUGMtEl8wqiqwYwA5lB9sOhfCNj0TA,16312
 iwa/web/tests/test_web_swap.py,sha256=7A4gBJFL01kIXPtW1E1J17SCsVc_0DmUn-R8kKrnnVA,2974
 iwa/web/tests/test_web_swap_coverage.py,sha256=zGNrzlhZ_vWDCvWmLcoUwFgqxnrp_ACbo49AtWBS_Kw,5584
-iwa-0.1.3.dist-info/licenses/LICENSE,sha256=eIubm_IlBHPYRQlLNZKbBNKhJUUP3JH0A2miZUhAVfI,1078
+iwa-0.1.4.dist-info/licenses/LICENSE,sha256=eIubm_IlBHPYRQlLNZKbBNKhJUUP3JH0A2miZUhAVfI,1078
 tests/legacy_cow.py,sha256=oOkZvIxL70ReEoD9oHQbOD5GpjIr6AGNHcOCgfPlerU,8389
 tests/legacy_safe.py,sha256=AssM2g13E74dNGODu_H0Q0y412lgqsrYnEzI97nm_Ts,2972
 tests/legacy_transaction_retry_logic.py,sha256=D9RqZ7DBu61Xr2djBAodU2p9UE939LL-DnQXswX5iQk,1497
@@ -224,8 +225,8 @@ tests/test_utils.py,sha256=vkP49rYNI8BRzLpWR3WnKdDr8upeZjZcs7Rx0pjbQMo,1292
 tests/test_workers.py,sha256=MInwdkFY5LdmFB3o1odIaSD7AQZb3263hNafO1De5PE,2793
 tools/create_and_stake_service.py,sha256=1xwy_bJQI1j9yIQ968Oc9Db_F6mk1659LuuZntTASDE,3742
 tools/verify_drain.py,sha256=PkMjblyOOAuQge88FwfEzRtCYeEtJxXhPBmtQYCoQ-8,6743
-iwa-0.1.3.dist-info/METADATA,sha256=uKemLrcWBfz_raFNxn3okTiQWiwF-lLQok838PpZXtI,7336
-iwa-0.1.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-iwa-0.1.3.dist-info/entry_points.txt,sha256=nwB6kscrfA7M00pYmL2j-sBH6eF6h2ga9IK1BZxdiyQ,241
-iwa-0.1.3.dist-info/top_level.txt,sha256=kedS9cRUbm4JE2wYeabIXilhHjN8KCw0IGbqqqsw0Bs,16
-iwa-0.1.3.dist-info/RECORD,,
+iwa-0.1.4.dist-info/METADATA,sha256=a0L36sHev8jqTBgB7MfGHSz46nAvG9J3OrkHNE1FpvM,7336
+iwa-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+iwa-0.1.4.dist-info/entry_points.txt,sha256=nwB6kscrfA7M00pYmL2j-sBH6eF6h2ga9IK1BZxdiyQ,241
+iwa-0.1.4.dist-info/top_level.txt,sha256=kedS9cRUbm4JE2wYeabIXilhHjN8KCw0IGbqqqsw0Bs,16
+iwa-0.1.4.dist-info/RECORD,,