cmd_start_session.rs

use std::net::SocketAddr;

use anyhow::Result;
use clap::Parser;

/// Starts a modeling session.
///
/// This command starts a server on localhost (on the configurable interface), and
/// waits to receive KCL programs over that server. It also connects to the KittyCAD
/// API and keeps the connection alive until this process is stopped. When it receives
/// a KCL program over the local server, it executes it using the long-lived KittyCAD
/// connection.
///
/// This subcommand is designed to be used with `zoo kcl snapshot --session localhost:3333`,
/// which will reuse the existing connection started by `zoo start-session localhost:3333`.
#[derive(Parser, Debug, Clone)]
#[clap(verbatim_doc_comment)]
pub struct CmdStartSession {
    /// What host/port to accept KCL programs on.
    #[clap(default_value = "0.0.0.0:3333")]
    pub listen_on: SocketAddr,
    /// How many engine connections to use in the connection pool.
    #[clap(default_value_t = 1)]
    pub num_engine_connections: u8,
}

#[async_trait::async_trait(?Send)]
impl crate::cmd::Command for CmdStartSession {
    async fn run(&self, ctx: &mut crate::context::Context) -> Result<()> {
        let args = kcl_test_server::ServerArgs {
            listen_on: self.listen_on,
            num_engine_conns: self.num_engine_connections,
            engine_address: None,
        };
        kcl_test_server::start_server(args).await?;
        writeln!(ctx.io.out, "Terminating").ok();
        Ok(())
    }
}