ownpg 0.1.0

OwnPG serves PostgreSQL DBA tools to AI clients over the Model Context Protocol, one database and one schema per run
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

use std::time::Duration;

use ownpg_core::config::{FlagLayer, HttpFlags, Sources, resolve_bind};
use ownpg_core::server::http::{LIVE_PATH, READY_PATH, probe};
use ownpg_core::{Error, ExitClass, Result};

use crate::cli::HealthArgs;
use crate::context::Process;
use crate::output::{emit, report_error, stdout_error};

pub(crate) fn run(args: &HealthArgs, process: &Process) -> Result<ExitClass> {
    match check(args, process) {
        Ok(answer) => {
            emit(|out| writeln!(out, "{answer}")).map_err(stdout_error)?;
            Ok(ExitClass::Success)
        }
        Err(error) => {
            report_error(&error);
            Ok(ExitClass::Runtime)
        }
    }
}

fn check(args: &HealthArgs, process: &Process) -> Result<String> {
    let flags = FlagLayer {
        http: HttpFlags {
            enabled: true,
            bind: args.bind.clone(),
            auth: None,
        },
        ..FlagLayer::default()
    };
    let bind = resolve_bind(
        &flags,
        &Sources {
            env: &process.env,
            paths: process.paths.clone(),
            keychain: None,
        },
    )?;
    let path = if args.live { LIVE_PATH } else { READY_PATH };
    let runtime = tokio::runtime::Builder::new_current_thread()
        .enable_all()
        .build()
        .map_err(|error| Error::ProtocolFailed {
            detail: format!("the runtime could not start: {error}"),
        })?;
    runtime.block_on(probe(bind, path, Duration::from_secs(args.timeout)))
}