request-vm-on-golem 0.1.40__tar.gz → 0.1.44__tar.gz

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: request-vm-on-golem
-Version: 0.1.40
+Version: 0.1.44
 Summary: VM on Golem Requestor CLI - Create and manage virtual machines on the Golem Network
 Keywords: golem,vm,cloud,decentralized,cli
 Author: Phillip Jensen
@@ -25,6 +25,8 @@ Requires-Dist: colorlog (>=6.8.0,<7.0.0)
 Requires-Dist: cryptography (>=3.4.7,<4.0.0)
 Requires-Dist: fastapi (>=0.103.0,<0.104.0)
 Requires-Dist: golem-base-sdk (==0.1.0)
+Requires-Dist: golem-faucet (>=0.1.0,<0.2.0)
+Requires-Dist: golem-streaming-abi (>=0.1.0,<0.2.0)
 Requires-Dist: httptools (>=0.6.0,<0.7.0)
 Requires-Dist: pydantic (>=2.4.0,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.1.0,<3.0.0)
@@ -119,6 +121,79 @@ The SSH connection process:
 2. The provider's proxy system forwards your SSH connection to the VM
 3. All traffic is securely routed through the allocated port
 
+## Streaming Payments (Native ETH on L2)
+
+This requestor integrates with an on‑chain StreamPayment contract to enable “pay‑as‑you‑go” rentals using native ETH (no ERC20 approvals when the token address is zero).
+
+Flow:
+
+1. Fetch provider info (preferred addresses):
+   - `GET http://{provider}:7466/api/v1/provider/info` → `provider_id`, `stream_payment_address`, `glm_token_address` (zero address means native ETH).
+2. Compute `ratePerSecond` from provider pricing and requested VM resources.
+3. Ensure `deposit >= ratePerSecond * 3600` (≥ 1 hour runway recommended/minimum).
+4. Create a stream (`createStream(0x000...0, provider_id, deposit, ratePerSecond)` plus `value=deposit`), capture `stream_id`. For ERC20 mode use a token address and approve first.
+5. Create VM: `POST /api/v1/vms` with `stream_id` included.
+6. Top‑up over time with `topUp(stream_id, amount)` to extend stopTime and keep the VM running indefinitely.
+7. On stop/destroy: the requestor will best‑effort `withdraw` / `terminate` to settle.
+
+CLI helpers
+
+- Open a stream for a planned VM (computes rate from provider pricing):
+
+```bash
+poetry run golem vm stream open \
+  --provider-id 0xProvider \
+  --cpu 2 --memory 4 --storage 20 \
+  --hours 1
+# prints { stream_id, rate_per_second_wei, deposit_wei }
+```
+
+- Top up an existing stream:
+
+```bash
+# Add 3 hours at prior rate
+poetry run golem vm stream topup --stream-id 123 --hours 3
+
+# Or specify exact GLM amount
+poetry run golem vm stream topup --stream-id 123 --glm 25.0
+```
+
+- Create a VM and attach an existing stream (no auto-streams are created by the requestor):
+
+```bash
+poetry run golem vm create my-vm \
+  --provider-id 0xProvider \
+  --cpu 2 --memory 4 --storage 20 \
+  --stream-id 123
+```
+
+Environment (env prefix `GOLEM_REQUESTOR_`):
+
+- `polygon_rpc_url` — EVM RPC URL (default L2 RPC)
+- `stream_payment_address` — StreamPayment address (fallback if provider doesn’t advertise)
+- `glm_token_address` — Token address; set to zero address to use native ETH
+- `provider_eth_address` — optional dev helper; in production always use `/provider/info`
+- `network` — Target network for discovery filtering: `testnet` (default) or `mainnet`
+
+Efficiency tips:
+
+- Batch top‑ups (e.g., add several hours at once) to reduce on‑chain calls.
+- Withdrawals are typically executed by providers; requestors don’t need to withdraw.
+- The CLI `vm stream open` will prefer the provider’s advertised contract/token addresses to prevent mismatches.
+
+## Faucet (L2 ETH)
+
+- Request L2 test ETH to cover stream transactions:
+
+```bash
+poetry run golem wallet faucet
+```
+
+- Defaults:
+  - Faucet: `https://l2.holesky.golemdb.io/faucet`
+  - CAPTCHA: `https://cap.gobas.me/05381a2cef5e`
+  - Override with env: `GOLEM_REQUESTOR_l2_faucet_url`, `GOLEM_REQUESTOR_captcha_url`, `GOLEM_REQUESTOR_captcha_api_key`.
+
 ## Installation
 
 ```bash
@@ -159,6 +234,27 @@ Alternatively, you can prepend the environment variables directly to the command
 GOLEM_REQUESTOR_ENVIRONMENT="development" GOLEM_REQUESTOR_FORCE_LOCALHOST="true" poetry run golem vm providers
 ```
 
+### Mode vs. Network
+
+- Development Mode (`GOLEM_REQUESTOR_ENVIRONMENT=development`)
+  - Improves local workflows: prefixes central discovery URL with `DEVMODE-` and, when using the central driver, maps provider IPs to `localhost` for easier testing.
+  - Does not determine chain selection.
+
+- Network Selection (`--network` or `GOLEM_REQUESTOR_NETWORK`)
+  - Filters Golem Base discovery results by `golem_network=testnet|mainnet`.
+  - Combine with the appropriate RPC envs (`GOLEM_REQUESTOR_GOLEM_BASE_RPC_URL`, `GOLEM_REQUESTOR_GOLEM_BASE_WS_URL`) and any contract addresses.
+  - Independent from dev ergonomics.
+
+Examples:
+- List providers on mainnet without changing env:
+  ```bash
+  poetry run golem vm providers --network mainnet
+  ```
+- Create a VM while targeting testnet:
+  ```bash
+  poetry run golem vm create my-vm --provider-id 0xProvider --cpu 2 --memory 4 --storage 20 --network testnet
+  ```
+
 ## Usage
 
 ### Provider Discovery
@@ -276,6 +372,7 @@ export GOLEM_REQUESTOR_DB_PATH="/path/to/database.db"
 # Environment Mode (defaults to "production")
 export GOLEM_REQUESTOR_ENVIRONMENT="development"  # Optional: Switch to development mode
 export GOLEM_REQUESTOR_FORCE_LOCALHOST="true"    # Optional: Force localhost in development mode
+export GOLEM_REQUESTOR_NETWORK="testnet"         # Or "mainnet"; filters Golem Base results by annotation
 ```
 
 2. Directory Structure:

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/README.md

@@ -80,6 +80,79 @@ The SSH connection process:
 2. The provider's proxy system forwards your SSH connection to the VM
 3. All traffic is securely routed through the allocated port
 
+## Streaming Payments (Native ETH on L2)
+
+This requestor integrates with an on‑chain StreamPayment contract to enable “pay‑as‑you‑go” rentals using native ETH (no ERC20 approvals when the token address is zero).
+
+Flow:
+
+1. Fetch provider info (preferred addresses):
+   - `GET http://{provider}:7466/api/v1/provider/info` → `provider_id`, `stream_payment_address`, `glm_token_address` (zero address means native ETH).
+2. Compute `ratePerSecond` from provider pricing and requested VM resources.
+3. Ensure `deposit >= ratePerSecond * 3600` (≥ 1 hour runway recommended/minimum).
+4. Create a stream (`createStream(0x000...0, provider_id, deposit, ratePerSecond)` plus `value=deposit`), capture `stream_id`. For ERC20 mode use a token address and approve first.
+5. Create VM: `POST /api/v1/vms` with `stream_id` included.
+6. Top‑up over time with `topUp(stream_id, amount)` to extend stopTime and keep the VM running indefinitely.
+7. On stop/destroy: the requestor will best‑effort `withdraw` / `terminate` to settle.
+
+CLI helpers
+
+- Open a stream for a planned VM (computes rate from provider pricing):
+
+```bash
+poetry run golem vm stream open \
+  --provider-id 0xProvider \
+  --cpu 2 --memory 4 --storage 20 \
+  --hours 1
+# prints { stream_id, rate_per_second_wei, deposit_wei }
+```
+
+- Top up an existing stream:
+
+```bash
+# Add 3 hours at prior rate
+poetry run golem vm stream topup --stream-id 123 --hours 3
+
+# Or specify exact GLM amount
+poetry run golem vm stream topup --stream-id 123 --glm 25.0
+```
+
+- Create a VM and attach an existing stream (no auto-streams are created by the requestor):
+
+```bash
+poetry run golem vm create my-vm \
+  --provider-id 0xProvider \
+  --cpu 2 --memory 4 --storage 20 \
+  --stream-id 123
+```
+
+Environment (env prefix `GOLEM_REQUESTOR_`):
+
+- `polygon_rpc_url` — EVM RPC URL (default L2 RPC)
+- `stream_payment_address` — StreamPayment address (fallback if provider doesn’t advertise)
+- `glm_token_address` — Token address; set to zero address to use native ETH
+- `provider_eth_address` — optional dev helper; in production always use `/provider/info`
+- `network` — Target network for discovery filtering: `testnet` (default) or `mainnet`
+
+Efficiency tips:
+
+- Batch top‑ups (e.g., add several hours at once) to reduce on‑chain calls.
+- Withdrawals are typically executed by providers; requestors don’t need to withdraw.
+- The CLI `vm stream open` will prefer the provider’s advertised contract/token addresses to prevent mismatches.
+
+## Faucet (L2 ETH)
+
+- Request L2 test ETH to cover stream transactions:
+
+```bash
+poetry run golem wallet faucet
+```
+
+- Defaults:
+  - Faucet: `https://l2.holesky.golemdb.io/faucet`
+  - CAPTCHA: `https://cap.gobas.me/05381a2cef5e`
+  - Override with env: `GOLEM_REQUESTOR_l2_faucet_url`, `GOLEM_REQUESTOR_captcha_url`, `GOLEM_REQUESTOR_captcha_api_key`.
+
 ## Installation
 
 ```bash
@@ -120,6 +193,27 @@ Alternatively, you can prepend the environment variables directly to the command
 GOLEM_REQUESTOR_ENVIRONMENT="development" GOLEM_REQUESTOR_FORCE_LOCALHOST="true" poetry run golem vm providers
 ```
 
+### Mode vs. Network
+
+- Development Mode (`GOLEM_REQUESTOR_ENVIRONMENT=development`)
+  - Improves local workflows: prefixes central discovery URL with `DEVMODE-` and, when using the central driver, maps provider IPs to `localhost` for easier testing.
+  - Does not determine chain selection.
+
+- Network Selection (`--network` or `GOLEM_REQUESTOR_NETWORK`)
+  - Filters Golem Base discovery results by `golem_network=testnet|mainnet`.
+  - Combine with the appropriate RPC envs (`GOLEM_REQUESTOR_GOLEM_BASE_RPC_URL`, `GOLEM_REQUESTOR_GOLEM_BASE_WS_URL`) and any contract addresses.
+  - Independent from dev ergonomics.
+
+Examples:
+- List providers on mainnet without changing env:
+  ```bash
+  poetry run golem vm providers --network mainnet
+  ```
+- Create a VM while targeting testnet:
+  ```bash
+  poetry run golem vm create my-vm --provider-id 0xProvider --cpu 2 --memory 4 --storage 20 --network testnet
+  ```
+
 ## Usage
 
 ### Provider Discovery
@@ -237,6 +331,7 @@ export GOLEM_REQUESTOR_DB_PATH="/path/to/database.db"
 # Environment Mode (defaults to "production")
 export GOLEM_REQUESTOR_ENVIRONMENT="development"  # Optional: Switch to development mode
 export GOLEM_REQUESTOR_FORCE_LOCALHOST="true"    # Optional: Force localhost in development mode
+export GOLEM_REQUESTOR_NETWORK="testnet"         # Or "mainnet"; filters Golem Base results by annotation
 ```
 
 2. Directory Structure:

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "request-vm-on-golem"
-version = "0.1.40"
+version = "0.1.44"
 description = "VM on Golem Requestor CLI - Create and manage virtual machines on the Golem Network"
 authors = ["Phillip Jensen <phillip+vm-on-golem@golemgrid.com>"]
 readme = "README.md"
@@ -39,10 +39,12 @@ golem-base-sdk = "==0.1.0"
 httptools = "^0.6.0"
 pydantic-settings = "^2.1.0"
 web3 = "==7.13.0"
+golem-streaming-abi = ">=0.1.0,<0.2.0"
+golem-faucet = ">=0.1.0,<0.2.0"
 
 [tool.poetry.group.dev.dependencies]
-pytest = "^7.0.0"
-pytest-asyncio = "^0.18.0"
+pytest = "^8.2.0"
+pytest-asyncio = "^1.1.0"
 pytest-cov = "^3.0.0"
 httpx = "^0.23.0"
 black = "^22.3.0"

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/requestor/cli/commands.py

@@ -75,9 +75,17 @@ def print_version(ctx, param, value):
 @click.group()
 @click.option('--version', is_flag=True, callback=print_version,
               expose_value=False, is_eager=True, help="Show the version and exit.")
-def cli():
+@click.option('--network', type=click.Choice(['testnet', 'mainnet']), default=None,
+              help="Override network for discovery filtering ('testnet' or 'mainnet')")
+def cli(network: str | None):
     """VM on Golem management CLI"""
     ensure_config()
+    # Allow on-demand override without touching env
+    if network:
+        try:
+            config.network = network
+        except Exception:
+            pass
     pass
 
 
@@ -94,10 +102,14 @@ def vm():
 @click.option('--country', help='Preferred provider country')
 @click.option('--driver', type=click.Choice(['central', 'golem-base']), default=None, help='Discovery driver to use')
 @click.option('--json', 'as_json', is_flag=True, help='Output in JSON format')
+@click.option('--network', type=click.Choice(['testnet', 'mainnet']), default=None,
+              help='Override network filter for this command')
 @async_command
-async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Optional[int], country: Optional[str], driver: Optional[str], as_json: bool):
+async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Optional[int], country: Optional[str], driver: Optional[str], as_json: bool, network: Optional[str] = None):
     """List available providers matching requirements."""
     try:
+        if network:
+            config.network = network
         # Log search criteria if any
         if any([cpu, memory, storage, country]):
             logger.command("🔍 Searching for providers with criteria:")
@@ -112,7 +124,7 @@ async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Opt
 
         # Determine the discovery driver being used
         discovery_driver = driver or config.discovery_driver
-        logger.process(f"Querying discovery service via {discovery_driver}")
+        logger.process(f"Querying discovery service via {discovery_driver} (network={config.network})")
 
         # Initialize provider service
         provider_service = ProviderService()
@@ -173,11 +185,15 @@ async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Opt
 @click.option('--cpu', type=int, required=True, help='Number of CPU cores')
 @click.option('--memory', type=int, required=True, help='Memory in GB')
 @click.option('--storage', type=int, required=True, help='Disk in GB')
+@click.option('--stream-id', type=int, default=None, help='Optional StreamPayment stream id to fund this VM')
 @click.option('--yes', is_flag=True, help='Do not prompt for confirmation')
+@click.option('--network', type=click.Choice(['testnet', 'mainnet']), default=None, help='Override network for discovery during creation')
 @async_command
-async def create_vm(name: str, provider_id: str, cpu: int, memory: int, storage: int, yes: bool):
+async def create_vm(name: str, provider_id: str, cpu: int, memory: int, storage: int, stream_id: int | None, yes: bool, network: Optional[str] = None):
     """Create a new VM on a specific provider."""
     try:
+        if network:
+            config.network = network
         # Show configuration details
         click.echo("\n" + "─" * 60)
         click.echo(click.style("  VM Configuration", fg="blue", bold=True))
@@ -232,7 +248,8 @@ async def create_vm(name: str, provider_id: str, cpu: int, memory: int, storage:
                         memory=memory,
                         storage=storage,
                         provider_ip=provider_ip,
-                        ssh_key=key_pair.public_key_content
+                        ssh_key=key_pair.public_key_content,
+                        stream_id=stream_id
                     )
 
                     # Get access info from config
@@ -281,6 +298,117 @@ async def create_vm(name: str, provider_id: str, cpu: int, memory: int, storage:
         raise click.Abort()
 
 
+@vm.group(name='stream')
+def vm_stream():
+    """Streaming payments helpers"""
+    pass
+
+
+@vm_stream.command('open')
+@click.option('--provider-id', required=True, help='Provider ID to use')
+@click.option('--cpu', type=int, required=True, help='CPU cores for rate calc')
+@click.option('--memory', type=int, required=True, help='Memory (GB) for rate calc')
+@click.option('--storage', type=int, required=True, help='Storage (GB) for rate calc')
+@click.option('--hours', type=int, default=1, help='Deposit coverage in hours (default 1)')
+@click.option('--network', type=click.Choice(['testnet', 'mainnet']), default=None, help='Override network for discovery during stream open')
+@async_command
+async def stream_open(provider_id: str, cpu: int, memory: int, storage: int, hours: int, network: Optional[str] = None):
+    """Create a GLM stream for a planned VM rental."""
+    from ..payments.blockchain_service import StreamPaymentClient, StreamPaymentConfig
+    try:
+        if network:
+            config.network = network
+        provider_service = ProviderService()
+        async with provider_service:
+            provider = await provider_service.verify_provider(provider_id)
+            est = provider_service.compute_estimate(provider, (cpu, memory, storage))
+            if not est or est.get('glm_per_month') is None:
+                raise RequestorError('Provider does not advertise GLM pricing; cannot compute ratePerSecond')
+            glm_month = est['glm_per_month']
+            glm_per_second = float(glm_month) / (730.0 * 3600.0)
+            rate_per_second_wei = int(glm_per_second * (10**18))
+
+            provider_ip = 'localhost' if config.environment == "development" else provider.get('ip_address')
+            if not provider_ip and config.environment == "production":
+                raise RequestorError("Provider IP address not found in advertisement")
+            provider_url = config.get_provider_url(provider_ip)
+            async with ProviderClient(provider_url) as client:
+                info = await client.get_provider_info()
+                recipient = info['provider_id']
+
+            deposit_wei = rate_per_second_wei * int(hours) * 3600
+            # Prefer provider-advertised contract addresses to avoid mismatches
+            spc = StreamPaymentConfig(
+                rpc_url=config.polygon_rpc_url,
+                contract_address=info.get('stream_payment_address') or config.stream_payment_address,
+                glm_token_address=info.get('glm_token_address') or config.glm_token_address,
+                private_key=config.ethereum_private_key,
+            )
+            sp = StreamPaymentClient(spc)
+            stream_id = sp.create_stream(recipient, deposit_wei, rate_per_second_wei)
+            click.echo(json.dumps({"stream_id": stream_id, "rate_per_second_wei": rate_per_second_wei, "deposit_wei": deposit_wei}, indent=2))
+    except Exception as e:
+        logger.error(f"Failed to open stream: {e}")
+        raise click.Abort()
+
+
+@vm_stream.command('topup')
+@click.option('--stream-id', type=int, required=True)
+@click.option('--glm', type=float, required=False, help='GLM amount to add')
+@click.option('--hours', type=int, required=False, help='Hours of coverage to add at prior rate')
+@async_command
+async def stream_topup(stream_id: int, glm: float | None, hours: int | None):
+    """Top up a stream. Provide either --glm or --hours (using prior rate)."""
+    from ..payments.blockchain_service import StreamPaymentClient, StreamPaymentConfig
+    try:
+        spc = StreamPaymentConfig(
+            rpc_url=config.polygon_rpc_url,
+            contract_address=config.stream_payment_address,
+            glm_token_address=config.glm_token_address,
+            private_key=config.ethereum_private_key,
+        )
+        sp = StreamPaymentClient(spc)
+        add_wei: int
+        if glm is not None:
+            add_wei = int(float(glm) * (10**18))
+        elif hours is not None:
+            # naive: use last known rate by reading on-chain stream
+            rate = sp.contract.functions.streams(int(stream_id)).call()[5]  # ratePerSecond
+            add_wei = int(rate) * int(hours) * 3600
+        else:
+            raise RequestorError('Provide either --glm or --hours')
+        tx = sp.top_up(stream_id, add_wei)
+        click.echo(json.dumps({"stream_id": stream_id, "topped_up_wei": add_wei, "tx": tx}, indent=2))
+    except Exception as e:
+        logger.error(f"Failed to top up stream: {e}")
+        raise click.Abort()
+
+
+@cli.group()
+def wallet():
+    """Wallet utilities (funding, balance)."""
+    pass
+
+
+@wallet.command('faucet')
+@async_command
+async def wallet_faucet():
+    """Request L2 faucet funds for the requestor's payment address."""
+    try:
+        from ..security.faucet import L2FaucetService
+        from eth_account import Account
+        acct = Account.from_key(config.ethereum_private_key)
+        service = L2FaucetService(config)
+        tx = await service.request_funds(acct.address)
+        if tx:
+            click.echo(json.dumps({"address": acct.address, "tx": tx}, indent=2))
+        else:
+            click.echo(json.dumps({"address": acct.address, "tx": None}, indent=2))
+    except Exception as e:
+        logger.error(f"Faucet request failed: {e}")
+        raise click.Abort()
+
+
 @vm.command(name='ssh')
 @click.argument('name')
 @async_command

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/requestor/config.py

@@ -3,6 +3,7 @@ from typing import Optional, Dict
 import os
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from pydantic import Field, field_validator, ValidationInfo
+import os
 
 
 def ensure_config() -> None:
@@ -47,6 +48,11 @@ class RequestorConfig(BaseSettings):
         default="production",
         description="Environment mode: 'development' or 'production'"
     )
+    # Network (for discovery filtering and defaults)
+    network: str = Field(
+        default="mainnet",
+        description="Target network: 'testnet' or 'mainnet'"
+    )
     
     # Development Settings
     force_localhost: bool = Field(
@@ -94,6 +100,51 @@ class RequestorConfig(BaseSettings):
         description="Private key for Golem Base"
     )
     
+    # Payments (EVM RPC)
+    polygon_rpc_url: str = Field(
+        default="https://l2.holesky.golemdb.io/rpc",
+        description="EVM RPC URL for streaming payments (L2 by default)"
+    )
+    stream_payment_address: str = Field(
+        default="0x0000000000000000000000000000000000000000",
+        description="Deployed StreamPayment contract address"
+    )
+    glm_token_address: str = Field(
+        default="0x0000000000000000000000000000000000000000",
+        description="Token address (0x0 means native ETH)"
+    )
+    # Faucet settings (L2 payments)
+    l2_faucet_url: str = Field(
+        default="https://l2.holesky.golemdb.io/faucet",
+        description="L2 faucet base URL (no trailing /api)"
+    )
+    captcha_url: str = Field(
+        default="https://cap.gobas.me",
+        description="CAPTCHA base URL"
+    )
+    captcha_api_key: str = Field(
+        default="05381a2cef5e",
+        description="CAPTCHA API key path segment"
+    )
+    provider_eth_address: str = Field(
+        default="",
+        description="Optional provider Ethereum address for test/dev streaming"
+    )
+
+    @field_validator("polygon_rpc_url", mode='before')
+    @classmethod
+    def prefer_alt_env(cls, v: str) -> str:
+        # Accept alt aliases
+        for key in (
+            "GOLEM_REQUESTOR_l2_rpc_url",
+            "GOLEM_REQUESTOR_L2_RPC_URL",
+            "GOLEM_REQUESTOR_kaolin_rpc_url",
+            "GOLEM_REQUESTOR_KAOLIN_RPC_URL",
+        ):
+            if os.environ.get(key):
+                return os.environ[key]
+        return v
+
     # Base Directory
     base_dir: Path = Field(
         default_factory=lambda: Path.home() / ".golem" / "requestor",

request_vm_on_golem-0.1.44/requestor/payments/blockchain_service.py

@@ -0,0 +1,161 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Optional, Any, Dict
+
+from web3 import Web3
+from eth_account import Account
+from golem_streaming_abi import STREAM_PAYMENT_ABI, ERC20_ABI
+
+
+@dataclass
+class StreamPaymentConfig:
+    rpc_url: str
+    contract_address: str
+    glm_token_address: str
+    private_key: str
+
+
+class StreamPaymentClient:
+    def __init__(self, cfg: StreamPaymentConfig):
+        self.web3 = Web3(Web3.HTTPProvider(cfg.rpc_url))
+        self.account = Account.from_key(cfg.private_key)
+        self.web3.eth.default_account = self.account.address
+
+        self.contract = self.web3.eth.contract(
+            address=Web3.to_checksum_address(cfg.contract_address), abi=STREAM_PAYMENT_ABI
+        )
+        self.token_address = Web3.to_checksum_address(cfg.glm_token_address)
+        self.native_eth = self.token_address.lower() == Web3.to_checksum_address("0x0000000000000000000000000000000000000000").lower()
+        self.erc20 = None
+        if not self.native_eth:
+            self.erc20 = self.web3.eth.contract(
+                address=self.token_address, abi=ERC20_ABI
+            )
+
+    def _send(self, fn) -> Dict[str, Any]:
+        base = {
+            "from": self.account.address,
+            "nonce": self.web3.eth.get_transaction_count(self.account.address),
+        }
+        # Fill chainId
+        try:
+            base["chainId"] = getattr(self.web3.eth, "chain_id", None) or self.web3.eth.chain_id
+        except Exception:
+            pass
+        # Try gas estimation and fee fields
+        try:
+            tx_preview = fn.build_transaction(base)
+            gas = self.web3.eth.estimate_gas(tx_preview)
+            base["gas"] = gas
+        except Exception:
+            pass
+        try:
+            # Prefer EIP-1559 if available
+            max_fee = getattr(self.web3.eth, "max_priority_fee", None)
+            if max_fee is not None:
+                base.setdefault("maxPriorityFeePerGas", max_fee)
+            base.setdefault("maxFeePerGas", getattr(self.web3.eth, "gas_price", lambda: None)() or self.web3.eth.gas_price)
+        except Exception:
+            try:
+                base.setdefault("gasPrice", self.web3.eth.gas_price)
+            except Exception:
+                pass
+
+        tx = fn.build_transaction(base)
+        # In production, sign and send raw; in tests, Account may be a dummy without signer
+        if hasattr(self.account, "sign_transaction"):
+            signed = self.account.sign_transaction(tx)
+            tx_hash = self.web3.eth.send_raw_transaction(signed.rawTransaction)
+        else:
+            tx_hash = self.web3.eth.send_transaction(tx)
+        receipt = self.web3.eth.wait_for_transaction_receipt(tx_hash)
+        return {"transactionHash": tx_hash.hex(), "status": receipt.status, "logs": receipt.logs}
+
+    def create_stream(self, provider_address: str, deposit_wei: int, rate_per_second_wei: int) -> int:
+        tx_value = None
+        token_param = self.token_address if not self.native_eth else Web3.to_checksum_address("0x0000000000000000000000000000000000000000")
+
+        if not self.native_eth:
+            # Approve deposit for the StreamPayment contract (only if needed)
+            try:
+                allowance = self.erc20.functions.allowance(self.account.address, self.contract.address).call()
+            except Exception:
+                allowance = 0
+            if int(allowance) < int(deposit_wei):
+                approve = self.erc20.functions.approve(self.contract.address, int(deposit_wei))
+                self._send(approve)
+        else:
+            tx_value = int(deposit_wei)
+
+        # Create stream
+        fn = self.contract.functions.createStream(
+            token_param,
+            Web3.to_checksum_address(provider_address),
+            int(deposit_wei),
+            int(rate_per_second_wei),
+        )
+        # Include ETH value if native
+        if tx_value is not None:
+            # Build/send with value field
+            base = {
+                "from": self.account.address,
+                "nonce": self.web3.eth.get_transaction_count(self.account.address),
+                "value": tx_value,
+            }
+            tx = fn.build_transaction(base)
+            signed = self.account.sign_transaction(tx)
+            tx_hash = self.web3.eth.send_raw_transaction(signed.rawTransaction)
+            receipt = self.web3.eth.wait_for_transaction_receipt(tx_hash)
+            tx_receipt = {"transactionHash": tx_hash.hex(), "status": receipt.status, "logs": receipt.logs}
+        else:
+            tx_receipt = self._send(fn)
+
+        # Try to parse StreamCreated event for streamId
+        try:
+            for log in tx_receipt["logs"]:
+                # very naive filter: topic0 = keccak256(StreamCreated(...))
+                # When ABI is attached to contract, use contract.events
+                ev = self.contract.events.StreamCreated().process_log(log)
+                return int(ev["args"]["streamId"])
+        except Exception:
+            pass
+        # As a fallback, cannot easily fetch return value from a tx; caller should query later
+        raise RuntimeError("create_stream: could not parse streamId from receipt")
+
+    def withdraw(self, stream_id: int) -> str:
+        fn = self.contract.functions.withdraw(int(stream_id))
+        receipt = self._send(fn)
+        return receipt["transactionHash"]
+
+    def terminate(self, stream_id: int) -> str:
+        fn = self.contract.functions.terminate(int(stream_id))
+        receipt = self._send(fn)
+        return receipt["transactionHash"]
+
+    def top_up(self, stream_id: int, amount_wei: int) -> str:
+        if not self.native_eth:
+            # Approve first (only if needed)
+            try:
+                allowance = self.erc20.functions.allowance(self.account.address, self.contract.address).call()
+            except Exception:
+                allowance = 0
+            if int(allowance) < int(amount_wei):
+                approve = self.erc20.functions.approve(self.contract.address, int(amount_wei))
+                self._send(approve)
+            fn = self.contract.functions.topUp(int(stream_id), int(amount_wei))
+            receipt = self._send(fn)
+            return receipt["transactionHash"]
+        else:
+            # Native ETH mode: send value along with call
+            fn = self.contract.functions.topUp(int(stream_id), int(amount_wei))
+            base = {
+                "from": self.account.address,
+                "nonce": self.web3.eth.get_transaction_count(self.account.address),
+                "value": int(amount_wei),
+            }
+            tx = fn.build_transaction(base)
+            signed = self.account.sign_transaction(tx)
+            tx_hash = self.web3.eth.send_raw_transaction(signed.rawTransaction)
+            self.web3.eth.wait_for_transaction_receipt(tx_hash)
+            return tx_hash.hex()

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/requestor/provider/client.py

@@ -21,26 +21,37 @@ class ProviderClient:
         cpu: int,
         memory: int,
         storage: int,
-        ssh_key: str
+        ssh_key: str,
+        stream_id: int | None = None,
     ) -> Dict:
         """Create a VM on the provider."""
+        payload = {
+            "name": name,
+            "resources": {
+                "cpu": cpu,
+                "memory": memory,
+                "storage": storage
+            },
+            "ssh_key": ssh_key
+        }
+        if stream_id is not None:
+            payload["stream_id"] = int(stream_id)
         async with self.session.post(
             f"{self.provider_url}/api/v1/vms",
-            json={
-                "name": name,
-                "resources": {
-                    "cpu": cpu,
-                    "memory": memory,
-                    "storage": storage
-                },
-                "ssh_key": ssh_key
-            }
+            json=payload
         ) as response:
             if not response.ok:
                 error_text = await response.text()
                 raise Exception(f"Failed to create VM: {error_text}")
             return await response.json()
 
+    async def get_provider_info(self) -> Dict:
+        async with self.session.get(f"{self.provider_url}/api/v1/provider/info") as response:
+            if not response.ok:
+                error_text = await response.text()
+                raise Exception(f"Failed to fetch provider info: {error_text}")
+            return await response.json()
+
     async def add_ssh_key(self, vm_id: str, key: str) -> None:
         """Add SSH key to VM."""
         async with self.session.post(

request_vm_on_golem-0.1.44/requestor/security/faucet.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+from typing import Optional, List, Tuple
+
+from web3 import Web3
+from golem_faucet import PowFaucetClient
+from ..utils.logging import setup_logger
+from ..config import RequestorConfig
+
+logger = setup_logger(__name__)
+
+
+class L2FaucetService:
+    def __init__(self, config: RequestorConfig):
+        self.cfg = config
+        self.web3 = Web3(Web3.HTTPProvider(config.polygon_rpc_url))
+        self.client = PowFaucetClient(
+            faucet_url=getattr(config, 'l2_faucet_url', 'https://l2.holesky.golemdb.io/faucet'),
+            captcha_base_url=getattr(config, 'captcha_url', 'https://cap.gobas.me'),
+            captcha_api_key=getattr(config, 'captcha_api_key', '05381a2cef5e'),
+        )
+
+    def _balance_eth(self, address: str) -> float:
+        try:
+            wei = self.web3.eth.get_balance(Web3.to_checksum_address(address))
+            return float(self.web3.from_wei(wei, 'ether'))
+        except Exception as e:
+            logger.warning(f"balance check failed: {e}")
+            return 0.0
+
+    async def request_funds(self, address: str) -> Optional[str]:
+        bal = self._balance_eth(address)
+        if bal > 0.01:
+            logger.info(f"Sufficient L2 funds ({bal} ETH), skipping faucet.")
+            return None
+        chall = await self.client.get_challenge()
+        if not chall:
+            logger.error("could not fetch faucet challenge")
+            return None
+        token = chall.get('token')
+        challenge_list = chall.get('challenge') or []
+        solutions: List[Tuple[str, str, int]] = []
+        for salt, target in challenge_list:
+            nonce = PowFaucetClient.solve_challenge(salt, target)
+            solutions.append((salt, target, nonce))
+        redeemed = await self.client.redeem(token, solutions)
+        if not redeemed:
+            logger.error("failed to redeem challenge")
+            return None
+        tx = await self.client.request_funds(address, redeemed)
+        if tx:
+            logger.success(f"L2 faucet sent tx: {tx}")
+        return tx
+

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/requestor/services/provider_service.py

@@ -93,6 +93,9 @@ class ProviderService:
         """Find providers using Golem Base."""
         try:
             query = 'golem_type="provider"'
+            # Filter by advertised network to avoid cross-network results
+            if config.network:
+                query += f' && golem_network="{config.network}"'
             if cpu:
                 query += f' && golem_cpu>={cpu}'
             if memory:

{request_vm_on_golem-0.1.40 → request_vm_on_golem-0.1.44}/requestor/services/vm_service.py

@@ -14,11 +14,13 @@ class VMService:
         self,
         db_service: DatabaseService,
         ssh_service: SSHService,
-        provider_client: Optional[ProviderClient] = None
+        provider_client: Optional[ProviderClient] = None,
+        blockchain_client: Optional[object] = None,
     ):
         self.db = db_service
         self.ssh_service = ssh_service
         self.provider_client = provider_client
+        self.blockchain_client = blockchain_client
 
     async def create_vm(
         self,
@@ -27,7 +29,8 @@ class VMService:
         memory: int,
         storage: int,
         provider_ip: str,
-        ssh_key: str
+        ssh_key: str,
+        stream_id: int | None = None,
     ) -> Dict:
         """Create a new VM with validation and error handling."""
         try:
@@ -42,18 +45,23 @@ class VMService:
                 cpu=cpu,
                 memory=memory,
                 storage=storage,
-                ssh_key=ssh_key
+                ssh_key=ssh_key,
+                stream_id=stream_id
             )
 
             # Get VM access info
             access_info = await self.provider_client.get_vm_access(vm['id'])
 
+            # Preserve any provided stream_id; do not auto-create streams here
+            # Stream creation should be explicit via CLI `vm stream open` command.
+
             # Save VM details to database
             config = {
                 'cpu': cpu,
                 'memory': memory,
                 'storage': storage,
-                'ssh_port': access_info['ssh_port']
+                'ssh_port': access_info['ssh_port'],
+                **({"stream_id": stream_id} if stream_id is not None else {}),
             }
             await self.db.save_vm(
                 name=name,
@@ -88,6 +96,15 @@ class VMService:
                 if "Not Found" not in str(e):
                     raise
 
+            # Attempt to terminate stream if present
+            try:
+                stream_id = vm.get('config', {}).get('stream_id')
+                if stream_id is not None and self.blockchain_client:
+                    self.blockchain_client.terminate(stream_id)
+            except Exception:
+                # Best-effort: do not block deletion on chain failure
+                pass
+
             # Remove from database
             await self.db.delete_vm(name)
 
@@ -125,6 +142,14 @@ class VMService:
             # Update status in database
             await self.db.update_vm_status(name, "stopped")
 
+            # Best-effort withdraw on stop
+            try:
+                stream_id = vm.get('config', {}).get('stream_id')
+                if stream_id is not None and self.blockchain_client:
+                    self.blockchain_client.withdraw(stream_id)
+            except Exception:
+                pass
+
         except Exception as e:
             raise VMError(f"Failed to stop VM: {str(e)}")