portforge 0.2.1

Modern cross-platform port inspector & manager for developers
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use crate::config::PortForgeConfig;
use crate::error::{PortForgeError, Result};
use crate::web::handlers;
use axum::{
    Router,
    routing::{get, post},
};
use std::sync::Arc;
use tokio::sync::Mutex;
use tower_http::cors::{Any, CorsLayer};

/// Shared application state for the web server.
pub struct AppState {
    pub config: PortForgeConfig,
    pub entries: Vec<crate::models::PortEntry>,
}

pub type SharedState = Arc<Mutex<AppState>>;

/// Start the web dashboard server.
pub async fn start_server(bind: &str, port: u16, config: PortForgeConfig) -> Result<()> {
    let state: SharedState = Arc::new(Mutex::new(AppState {
        config: config.clone(),
        entries: Vec::new(),
    }));

    // Initial scan
    {
        let mut state_lock = state.lock().await;
        match crate::scanner::scan_ports(&state_lock.config, false).await {
            Ok(entries) => state_lock.entries = entries,
            Err(e) => tracing::warn!("Initial scan failed: {}", e),
        }
    }

    // Background scanner task
    let scanner_state = state.clone();
    let scanner_config = config.clone();
    tokio::spawn(async move {
        let mut interval = tokio::time::interval(std::time::Duration::from_secs(
            scanner_config.general.refresh_interval.max(2),
        ));
        loop {
            interval.tick().await;
            if let Ok(entries) = crate::scanner::scan_ports(&scanner_config, false).await {
                let mut state_lock = scanner_state.lock().await;
                state_lock.entries = entries;
            }
        }
    });

    let cors = CorsLayer::new()
        .allow_origin(Any)
        .allow_methods(Any)
        .allow_headers(Any);

    let app = Router::new()
        // Pages
        .route("/", get(handlers::dashboard_page))
        // API endpoints
        .route("/api/ports", get(handlers::api_ports))
        .route("/api/ports/{port}", get(handlers::api_port_detail))
        .route("/api/ports/{port}/kill", post(handlers::api_kill_port))
        .route("/api/stats", get(handlers::api_stats))
        // HTMX partials
        .route("/partials/table", get(handlers::partial_table))
        .route("/partials/stats", get(handlers::partial_stats))
        // Static assets
        .route("/static/{*path}", get(handlers::static_asset))
        .layer(cors)
        .with_state(state);

    let addr = format!("{}:{}", bind, port);
    let listener = tokio::net::TcpListener::bind(&addr)
        .await
        .map_err(|e| PortForgeError::WebError(format!("Failed to bind to {}: {}", addr, e)))?;

    tracing::info!("Web dashboard running at http://{}", addr);

    axum::serve(listener, app)
        .await
        .map_err(|e| PortForgeError::WebError(e.to_string()))?;

    Ok(())
}