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

pub mod config;
mod db;

use anyhow::Context;
use clap::Parser;
use std::{fs, path::PathBuf};

use self::config::{Config, EnvOverride};
use crate::{extension::*, job::JobRegistry};

#[derive(Parser)]
#[clap(long_about = None)]
struct Cli {
    #[clap(short, long, env = "CALA_CONFIG", value_name = "FILE")]
    config: Option<PathBuf>,
    #[clap(
        long,
        env = "CALA_HOME",
        default_value = ".cala",
        value_name = "DIRECTORY"
    )]
    cala_home: String,
    #[clap(env = "PG_CON")]
    pg_con: String,
}

pub async fn run<Q: QueryExtensionMarker, M: MutationExtensionMarker>(
    job_registration: impl FnOnce(&mut JobRegistry),
) -> anyhow::Result<()> {
    let cli = Cli::parse();

    let config = Config::load_config(cli.config, EnvOverride { db_con: cli.pg_con })?;

    run_cmd::<Q, M>(&cli.cala_home, config, job_registration).await?;

    Ok(())
}

async fn run_cmd<Q: QueryExtensionMarker, M: MutationExtensionMarker>(
    cala_home: &str,
    config: Config,
    job_registration: impl FnOnce(&mut JobRegistry),
) -> anyhow::Result<()> {
    use cala_ledger::{CalaLedger, CalaLedgerConfig};
    cala_tracing::init_tracer(config.tracing)?;
    store_server_pid(cala_home, std::process::id())?;
    let pool = db::init_pool(&config.db).await?;
    let ledger_config = CalaLedgerConfig::builder().pool(pool.clone()).build()?;
    let ledger = CalaLedger::init(ledger_config).await?;
    let mut registry = JobRegistry::new(&ledger);
    job_registration(&mut registry);
    let app = crate::app::CalaApp::run(pool, config.app, ledger, registry).await?;
    crate::server::run::<Q, M>(config.server, app).await?;
    Ok(())
}

pub fn store_server_pid(cala_home: &str, pid: u32) -> anyhow::Result<()> {
    create_cala_dir(cala_home)?;
    let _ = fs::remove_file(format!("{cala_home}/server-pid"));
    fs::write(format!("{cala_home}/server-pid"), pid.to_string()).context("Writing PID file")?;
    Ok(())
}

fn create_cala_dir(cala_home: &str) -> anyhow::Result<()> {
    let _ = fs::create_dir(cala_home);
    Ok(())
}