EasyTier/easytier/src/web_client/mod.rs

use std::sync::Arc;

use crate::{
    common::scoped_task::ScopedTask, instance_manager::NetworkInstanceManager,
    tunnel::TunnelConnector,
};

pub mod controller;
pub mod session;

pub struct WebClient {
    controller: Arc<controller::Controller>,
    tasks: ScopedTask<()>,
}

impl WebClient {
    pub fn new<T: TunnelConnector + 'static, S: ToString, H: ToString>(
        connector: T,
        token: S,
        hostname: H,
        manager: Arc<NetworkInstanceManager>,
    ) -> Self {
        let controller = Arc::new(controller::Controller::new(
            token.to_string(),
            hostname.to_string(),
            manager,
        ));

        let controller_clone = controller.clone();
        let tasks = ScopedTask::from(tokio::spawn(async move {
            Self::routine(controller_clone, Box::new(connector)).await;
        }));

        WebClient { controller, tasks }
    }

    async fn routine(
        controller: Arc<controller::Controller>,
        mut connector: Box<dyn TunnelConnector>,
    ) {
        loop {
            let conn = match connector.connect().await {
                Ok(conn) => conn,
                Err(e) => {
                    println!(
                        "Failed to connect to the server ({}), retrying in 5 seconds...",
                        e
                    );
                    tokio::time::sleep(std::time::Duration::from_secs(1)).await;
                    continue;
                }
            };

            println!("Successfully connected to {:?}", conn.info());

            let mut session = session::Session::new(conn, controller.clone());
            session.wait().await;
        }
    }
}