terrazzo-terminal 0.2.7

use std::sync::Arc;

use server_fn::Http;
use server_fn::ServerFnError;
use server_fn::codec::Json;
use terrazzo::server;

use crate::api::client_address::ClientAddress;
use crate::portforward::schema::PortForward;

/// Stores the [PortForward]s on the designated remote terrazzo server.
#[server(protocol = Http<Json, Json>)]
#[cfg_attr(feature = "server", nameth::nameth)]
pub async fn store_port_forwards(
    remote: ClientAddress,
    port_forwards: Arc<Vec<PortForward>>,
) -> Result<(), ServerFnError> {
    #[cfg(debug_assertions)]
    tokio::time::sleep(std::time::Duration::from_secs(1)).await;
    Ok(backend::STORE_PORT_FORWARDS_FN
        .call(remote, port_forwards)
        .await?)
}

/// Loads the [PortForward]s from the designated remote terrazzo server.
#[server(protocol = Http<Json, Json>)]
#[cfg_attr(feature = "server", nameth::nameth)]
pub async fn load_port_forwards(
    remote: ClientAddress,
) -> Result<Box<Vec<PortForward>>, ServerFnError> {
    Ok(backend::LOAD_PORT_FORWARDS_FN.call(remote, ()).await?)
}

#[cfg(feature = "server")]
mod backend {
    use std::future::ready;
    use std::sync::Arc;
    use std::sync::Mutex;

    use crate::backend::client_service::remote_fn_service;
    use crate::portforward::engine::PreparedPortForwards;
    use crate::portforward::engine::RunningPortForward;
    use crate::portforward::schema::PortForward;

    static STATE: Mutex<Option<Box<[RunningPortForward]>>> = Mutex::new(None);

    remote_fn_service::unary::declare_remote_fn!(
        /// Stores the port forwards on the designated remote terrazzo server.
        ///
        /// As a side effect, it runs the port forward engine as necessary to activate the new
        /// port forwards configuration.
        ///
        /// - First, port forwards that are deleted or changed are stopped.
        /// - Then, new and changed port forwards are started.
        STORE_PORT_FORWARDS_FN,
        super::STORE_PORT_FORWARDS,
        Arc<Vec<PortForward>>,
        (),
        |server, port_forwards| {
            let server = server.clone();
            async move {
                let (stopping, pending) = {
                    let mut lock = STATE.lock().expect(super::STORE_PORT_FORWARDS);
                    let PreparedPortForwards {
                        running,
                        stopping,
                        pending,
                    } = engine::prepare(lock.take().unwrap_or_default(), port_forwards);
                    *lock = Some(running);
                    (stopping, pending)
                };

                for stopping in stopping {
                    let () = stopping.stop().await;
                }

                use super::super::engine;
                let () = engine::process(&server, pending).await;
                Ok::<(), tonic::Status>(())
            }
        }
    );

    remote_fn_service::unary::declare_remote_fn!(
        /// Loads the port forwards from the designated remote terrazzo server.
        LOAD_PORT_FORWARDS_FN,
        super::LOAD_PORT_FORWARDS,
        (),
        Box<Vec<PortForward>>,
        |_server, ()| {
            let state = STATE.lock().expect(super::LOAD_PORT_FORWARDS);
            let state = state
                .as_deref()
                .unwrap_or_default()
                .iter()
                .map(|running| running.port_forward.clone())
                .collect::<Vec<_>>();
            ready(Ok::<_, tonic::Status>(state.into()))
        }
    );
}