golem-vm-provider 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

{golem_vm_provider-0.1.0.dist-info → golem_vm_provider-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: golem-vm-provider
-Version: 0.1.0
+Version: 0.1.1
 Summary: VM on Golem Provider Node - Run your own provider node to offer VMs on the Golem Network
 Home-page: https://github.com/cryptobench/vm-on-golem
 Keywords: golem,vm,provider,cloud,decentralized

{golem_vm_provider-0.1.0.dist-info → golem_vm_provider-0.1.1.dist-info}/RECORD

@@ -6,7 +6,7 @@ provider/config.py,sha256=Q6xmU6cPRBUYkVusv-Fc1U4ekr4OYLXfOfEtaWL9miM,5000
 provider/discovery/__init__.py,sha256=VR3NRoQtZRH5Vs8FG7jnGLR7p7wn7XeZdLaBb3t8e1g,123
 provider/discovery/advertiser.py,sha256=yv7RbRf1K43qOLAEa2Olj9hhN8etl2qsBuoHok0xoVs,6784
 provider/discovery/resource_tracker.py,sha256=8dYhJxoe_jLRwisHoA0jr575YhUKmLIqSXfW88KshcQ,6000
-provider/main.py,sha256=hiHN-wWSC-RZgal6DvBD2nlHsGvVehO4M4k1jfTbREA,4547
+provider/main.py,sha256=A5qWwO0LQrfz6aaHDXPEluYgSsS1bDH_fQ9iYHk1kds,8992
 provider/network/port_verifier.py,sha256=0ku_wa8QlQKjrHQd2Ih2G0kUZX4d87WDIzmiZMR7Uqc,12183
 provider/security/ethereum.py,sha256=SDRDbcjynbVy44kNnxlDcYLL0BZ3Qnc0DvmneQ-WKLE,1383
 provider/utils/ascii_art.py,sha256=ykBFsztk57GIiz1NJ-EII5UvN74iECqQL4h9VmiW6Z8,3161
@@ -20,7 +20,7 @@ provider/vm/multipass.py,sha256=kS0HkKZT6--YEjxxdr4Ot0oF24T128hLrq-STKsuCK4,1640
 provider/vm/name_mapper.py,sha256=MrshNeJ4Dw-WBsyiIVcn9N5xyOxaBKX4Yqhyh_m5IFg,4103
 provider/vm/port_manager.py,sha256=UeAp64UQXy-yxt-xGOB4aJkctFTLxDyouuwX5twhJpI,7428
 provider/vm/proxy_manager.py,sha256=UEoxpkqtyreqB4ypvIi190Xnr6WYbiID4v50xooKezs,9130
-golem_vm_provider-0.1.0.dist-info/METADATA,sha256=7oyA0cvyCi-_pSout7w_MsCszApVc7X-zRyfrok7U4Y,10449
-golem_vm_provider-0.1.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-golem_vm_provider-0.1.0.dist-info/entry_points.txt,sha256=E4rCWo_Do_2zCG_GewNuftfVlHF_8b_OvioZre0dfeA,54
-golem_vm_provider-0.1.0.dist-info/RECORD,,
+golem_vm_provider-0.1.1.dist-info/METADATA,sha256=ZlaXgpdjp7hcUMsoR0C9RDnxUPYdzdY-WL-EIhh3Evo,10449
+golem_vm_provider-0.1.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+golem_vm_provider-0.1.1.dist-info/entry_points.txt,sha256=E4rCWo_Do_2zCG_GewNuftfVlHF_8b_OvioZre0dfeA,54
+golem_vm_provider-0.1.1.dist-info/RECORD,,

provider/main.py

@@ -1,4 +1,5 @@
 import asyncio
+import os
 from fastapi import FastAPI
 from typing import Optional
 
@@ -122,4 +123,126 @@ from .api import routes
 app.include_router(routes.router, prefix="/api/v1")
 
 # Export app for uvicorn
-__all__ = ["app"]
+__all__ = ["app", "start"]
+
+def check_requirements():
+    """Check if all requirements are met."""
+    import os
+    from pathlib import Path
+    
+    # Check if multipass is installed
+    multipass_path = os.environ.get('GOLEM_PROVIDER_MULTIPASS_BINARY_PATH', '/usr/local/bin/multipass')
+    if not Path(multipass_path).exists():
+        logger.error(f"Multipass binary not found at {multipass_path}")
+        return False
+        
+    # Check required directories
+    vm_data_dir = os.environ.get(
+        'GOLEM_PROVIDER_VM_DATA_DIR',
+        str(Path.home() / '.golem' / 'provider' / 'vms')
+    )
+    ssh_key_dir = os.environ.get(
+        'GOLEM_PROVIDER_SSH_KEY_DIR',
+        str(Path.home() / '.golem' / 'provider' / 'ssh')
+    )
+    proxy_state_dir = os.environ.get(
+        'GOLEM_PROVIDER_PROXY_STATE_DIR',
+        str(Path.home() / '.golem' / 'provider' / 'proxy')
+    )
+    
+    try:
+        # Create and secure directories
+        for directory in [vm_data_dir, ssh_key_dir, proxy_state_dir]:
+            path = Path(directory)
+            path.mkdir(parents=True, exist_ok=True)
+            if directory == ssh_key_dir:
+                path.chmod(0o700)  # Secure permissions for SSH keys
+    except Exception as e:
+        logger.error(f"Failed to create required directories: {e}")
+        return False
+        
+    return True
+
+async def verify_ports():
+    """Verify port accessibility before starting server."""
+    from .vm.port_manager import PortManager
+    from .utils.port_display import PortVerificationDisplay
+    from .config import settings
+    
+    display = PortVerificationDisplay(
+        provider_port=settings.PORT,
+        port_range_start=settings.PORT_RANGE_START,
+        port_range_end=settings.PORT_RANGE_END
+    )
+    display.print_header()
+
+    # Initialize port manager
+    logger.process("🔄 Verifying port accessibility...")
+    port_manager = PortManager(
+        start_port=settings.PORT_RANGE_START,
+        end_port=settings.PORT_RANGE_END,
+        discovery_port=settings.PORT
+    )
+    if not await port_manager.initialize():
+        logger.error("Port verification failed. Please ensure:")
+        logger.error(f"1. Port {settings.PORT} is accessible for provider access")
+        logger.error(f"2. Some ports in range {settings.PORT_RANGE_START}-{settings.PORT_RANGE_END} are accessible for VM access")
+        logger.error("3. Your firewall/router is properly configured")
+        return False
+    
+    logger.success(f"✅ Port verification successful - {len(port_manager.verified_ports)} ports available")
+    return True
+
+def start():
+    """Start the provider server."""
+    import sys
+    import asyncio
+    from pathlib import Path
+    from dotenv import load_dotenv
+    import uvicorn
+    from .utils.logging import setup_logger
+    from .config import settings
+    
+    # Configure logging with debug mode
+    logger = setup_logger(__name__, debug=True)
+    
+    try:
+        # Load environment variables from .env file
+        env_path = Path(__file__).parent.parent / '.env'
+        load_dotenv(dotenv_path=env_path)
+        
+        # Log environment variables
+        logger.info("Environment variables:")
+        for key, value in os.environ.items():
+            if key.startswith('GOLEM_PROVIDER_'):
+                logger.info(f"{key}={value}")
+
+        # Check requirements
+        if not check_requirements():
+            logger.error("Requirements check failed")
+            sys.exit(1)
+
+        # Verify ports before starting server
+        if not asyncio.run(verify_ports()):
+            logger.error("Port verification failed")
+            sys.exit(1)
+        
+        # Configure uvicorn logging
+        log_config = uvicorn.config.LOGGING_CONFIG
+        log_config["formatters"]["access"]["fmt"] = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+        
+        # Run server
+        logger.process(f"🚀 Starting provider server on {settings.HOST}:{settings.PORT}")
+        uvicorn.run(
+            "provider:app",
+            host=settings.HOST,
+            port=settings.PORT,
+            reload=settings.DEBUG,
+            log_level="info" if not settings.DEBUG else "debug",
+            log_config=log_config,
+            timeout_keep_alive=60,  # Increase keep-alive timeout
+            limit_concurrency=100,  # Limit concurrent connections
+        )
+    except Exception as e:
+        logger.error(f"Failed to start provider server: {e}")
+        sys.exit(1)