from __future__ import annotations import asyncio import logging from pathlib import Path from typing import TYPE_CHECKING import socketio from fastapi import FastAPI, HTTPException from fastapi.middleware.cors import CORSMiddleware from fastapi.responses import FileResponse, HTMLResponse from fastapi.staticfiles import StaticFiles from pydantic import BaseModel if TYPE_CHECKING: import uvicorn from src.core.client import Twitch from src.web.gui_manager import WebGUIManager logger = logging.getLogger("TwitchDrops") # Create FastAPI app app = FastAPI(title="Twitch Drops Miner Web", version="1.0.0") # Add CORS middleware app.add_middleware( CORSMiddleware, allow_origins=["*"], # In production, specify exact origins allow_credentials=True, allow_methods=["*"], allow_headers=["*"], ) # Create Socket.IO server sio = socketio.AsyncServer( async_mode="asgi", cors_allowed_origins="*", logger=False, engineio_logger=False ) # Wrap with ASGI app socket_app = socketio.ASGIApp(sio, app) # Global references (set by main.py) gui_manager: WebGUIManager | None = None twitch_client: Twitch | None = None _server_instance: uvicorn.Server | None = None def set_managers(gui: WebGUIManager, twitch: Twitch): """Called by main.py to set up references""" global gui_manager, twitch_client gui_manager = gui twitch_client = twitch gui.set_socketio(sio) # Pydantic models for API class LoginRequest(BaseModel): username: str password: str token: str = "" class ChannelSelectRequest(BaseModel): channel_id: int class SettingsUpdate(BaseModel): games_to_watch: list[str] | None = None dark_mode: bool | None = None language: str | None = None proxy: str | None = None connection_quality: int | None = None minimum_refresh_interval_minutes: int | None = None inventory_filters: dict | None = None mining_benefits: dict[str, bool] | None = None class ProxyVerifyRequest(BaseModel): proxy: str # ==================== REST API Endpoints ==================== @app.get("/", response_class=HTMLResponse) async def serve_index(): """Serve the main web interface""" # Web files are in project_root/web/, we're in project_root/src/web/ web_dir = Path(__file__).parent.parent.parent / "web" index_file = web_dir / "index.html" logger.debug( f"Looking for web files: __file__={__file__}, web_dir={web_dir}, index_file={index_file}, exists={index_file.exists()}" ) if index_file.exists(): return FileResponse(index_file) return HTMLResponse( content=f"

Twitch Drops Miner

Web interface files not found. Please check installation.

Debug: Looking for {index_file}

", status_code=500, ) @app.get("/api/status") async def get_status(): """Get current application status""" if not gui_manager or not twitch_client: raise HTTPException(status_code=503, detail="GUI not initialized") return { "status": gui_manager.status.get(), "login": gui_manager.login.get_status(), "manual_mode": twitch_client.get_manual_mode_info(), } @app.get("/api/channels") async def get_channels(): """Get list of tracked channels""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") return {"channels": gui_manager.channels.get_channels()} @app.post("/api/channels/select") async def select_channel(request: ChannelSelectRequest): """Select a channel to watch""" if not gui_manager or not twitch_client: raise HTTPException(status_code=503, detail="GUI not initialized") # Validate channel exists channel = twitch_client.channels.get(request.channel_id) if not channel: raise HTTPException(status_code=404, detail="Channel not found") # Validate channel has a game if not channel.game: raise HTTPException(status_code=400, detail="Channel is not playing any game") # Warn if channel has no drops (shouldn't happen if GUI is filtering correctly) if not any(campaign.can_earn(channel) for campaign in twitch_client.inventory): logger.warning(f"User selected channel {channel.name} but it has no available drops") gui_manager.select_channel(request.channel_id) # Trigger channel switch to apply the selection from src.config import State twitch_client.change_state(State.CHANNEL_SWITCH) return {"success": True} @app.get("/api/campaigns") async def get_campaigns(): """Get campaign inventory""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") return {"campaigns": gui_manager.inv.get_campaigns()} @app.get("/api/console") async def get_console_history(): """Get console output history""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") return {"lines": gui_manager.output.get_history()} @app.get("/api/settings") async def get_settings(): """Get current settings""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") return gui_manager.settings.get_settings() @app.get("/api/languages") async def get_languages(): """Get available languages""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") return gui_manager.settings.get_languages() @app.get("/api/translations") async def get_translations(): """Get translations for current language""" from src.i18n.translator import _ # Return the full Translation object return _.t @app.post("/api/settings") async def update_settings(settings: SettingsUpdate): """Update application settings""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") settings_dict = settings.dict(exclude_unset=True) gui_manager.settings.update_settings(settings_dict) return {"success": True, "settings": gui_manager.settings.get_settings()} @app.post("/api/settings/verify-proxy") async def verify_proxy(request: ProxyVerifyRequest): """Verify proxy connectivity""" import time import aiohttp proxy_url = request.proxy.strip() if not proxy_url: return {"success": False, "message": "Proxy URL is empty"} try: start_time = time.time() # Test connection to Twitch async with ( aiohttp.ClientSession() as session, session.get("https://www.twitch.tv", proxy=proxy_url, timeout=10) as response, ): # Just checking if we can connect and get a response if response.status < 500: latency = round((time.time() - start_time) * 1000) return { "success": True, "message": f"Connected! ({latency}ms)", "latency": latency, } else: return { "success": False, "message": f"Proxy reachable but returned {response.status}", } except Exception as e: return {"success": False, "message": f"Connection failed: {str(e)}"} @app.get("/api/version") async def get_version(): """Get current application version and check for updates""" import aiohttp from src.version import __version__ current_version = __version__ latest_version = None update_available = False download_url = None try: # Check GitHub API for latest release async with ( aiohttp.ClientSession() as session, session.get( "https://api.github.com/repos/rangermix/TwitchDropsMiner/releases/latest", timeout=5 ) as response, ): if response.status == 200: data = await response.json() latest_version = data.get("tag_name", "").lstrip("v") download_url = data.get("html_url") # Compare versions (simple string comparison works for semantic versioning) if latest_version and latest_version > current_version: update_available = True except Exception as e: logger.warning(f"Failed to check for updates: {str(e)}") return { "current_version": current_version, "latest_version": latest_version, "update_available": update_available, "download_url": download_url or "https://github.com/rangermix/TwitchDropsMiner/releases", } @app.post("/api/login") async def submit_login(login_data: LoginRequest): """Submit login credentials""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") gui_manager.login.submit_login(login_data.username, login_data.password, login_data.token) return {"success": True} @app.post("/api/oauth/confirm") async def confirm_oauth(): """Confirm OAuth code has been entered by user""" if not gui_manager: raise HTTPException(status_code=503, detail="GUI not initialized") # Just set the event to signal the user has acknowledged the code gui_manager.login._login_event.set() return {"success": True} @app.post("/api/reload") async def trigger_reload(): """Trigger application reload""" if not twitch_client: raise HTTPException(status_code=503, detail="Twitch client not initialized") from src.config import State twitch_client.change_state(State.INVENTORY_FETCH) return {"success": True} @app.post("/api/close") async def trigger_close(): """Trigger application shutdown""" if not twitch_client: raise HTTPException(status_code=503, detail="Twitch client not initialized") twitch_client.close() return {"success": True} @app.post("/api/mode/exit-manual") async def exit_manual_mode(): """Exit manual mode and return to automatic channel selection""" if not twitch_client: raise HTTPException(status_code=503, detail="Twitch client not initialized") if not twitch_client.is_manual_mode(): return {"success": False, "message": "Not in manual mode"} twitch_client.exit_manual_mode("User requested") return {"success": True} # ==================== Socket.IO Events ==================== @sio.event async def connect(sid, environ): """Client connected""" logger.info(f"Web client connected: {sid}") # Send initial state to new client if gui_manager and twitch_client: await sio.emit( "initial_state", { "status": gui_manager.status.get(), "channels": gui_manager.channels.get_channels(), "campaigns": gui_manager.inv.get_campaigns(), "console": gui_manager.output.get_history(), "settings": gui_manager.settings.get_settings(), "login": gui_manager.login.get_status(), "manual_mode": twitch_client.get_manual_mode_info(), "current_drop": gui_manager.progress.get_current_drop(), "wanted_items": gui_manager.get_wanted_game_tree(), }, room=sid, ) @sio.event async def disconnect(sid): """Client disconnected""" logger.info(f"Web client disconnected: {sid}") @sio.event async def request_login(sid): """Client requested login form submission""" logger.info(f"Login request from client: {sid}") # The actual login data comes via REST API @sio.event async def request_reload(sid): """Client requested application reload""" if twitch_client: from src.config import State twitch_client.change_state(State.INVENTORY_FETCH) @sio.event async def get_wanted_items(sid): """Client requested wanted items list""" if gui_manager: await sio.emit("wanted_items_update", gui_manager.get_wanted_game_tree(), to=sid) # Mount static files (CSS, JS, images) # Web files are in project_root/web/, we're in project_root/src/web/ web_dir = Path(__file__).parent.parent.parent / "web" if web_dir.exists(): static_dir = web_dir / "static" if static_dir.exists(): app.mount("/static", StaticFiles(directory=static_dir), name="static") # Development server runner async def run_server(host: str = "0.0.0.0", port: int = 8080): """Run the web server (used for development/testing)""" global _server_instance import uvicorn config = uvicorn.Config(socket_app, host=host, port=port, log_level="info", access_log=False) server = uvicorn.Server(config) _server_instance = server try: await server.serve() finally: _server_instance = None async def shutdown_server(): """Gracefully shutdown the web server""" if _server_instance: logger.info("Setting server.should_exit = True") _server_instance.should_exit = True # Give the server a moment to process the shutdown signal # The uvicorn server checks should_exit periodically await asyncio.sleep(0.1) if __name__ == "__main__": # For standalone testing asyncio.run(run_server())