crabtalk 0.0.18

Run autonomous agents with built-in LLM inference
Documentation 
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

//! System service management (install/uninstall daemon as launchd/systemd service).

use crate::cmd::attach::setup_provider;
use anyhow::Result;
use wcore::paths::{CONFIG_DIR, LOGS_DIR};

#[cfg(target_os = "macos")]
const LAUNCHD_TEMPLATE: &str = include_str!("launchd.plist");
#[cfg(target_os = "linux")]
const SYSTEMD_TEMPLATE: &str = include_str!("systemd.service");
#[cfg(target_os = "windows")]
const SCHTASKS_TEMPLATE: &str = include_str!("schtasks.xml");

const LABEL: &str = "ai.crabtalk.crabtalk";

/// Check if providers are configured; scaffold config and prompt if needed.
fn ensure_providers() -> Result<()> {
    let config_path = CONFIG_DIR.join(wcore::paths::CONFIG_FILE);
    if !config_path.exists() {
        ::daemon::config::scaffold_config_dir(&CONFIG_DIR)?;
    }

    let config = ::daemon::DaemonConfig::load(&config_path)?;
    if config.provider.is_empty() {
        setup_provider(&config_path)?;
    }
    Ok(())
}

fn render_daemon_template(template: &str, verbose: u8) -> Result<String> {
    let binary = std::env::current_exe()?;
    let path_env = std::env::var("PATH").unwrap_or_default();
    Ok(template
        .replace("{label}", LABEL)
        .replace("{description}", "Crabtalk Daemon")
        .replace("{log_name}", "daemon")
        .replace("{binary}", &binary.display().to_string())
        .replace("-v", &crabtalk_command::verbose_flag(verbose))
        .replace("{logs_dir}", &LOGS_DIR.display().to_string())
        .replace("{config_dir}", &CONFIG_DIR.display().to_string())
        .replace("{path}", &path_env))
}

#[cfg(target_os = "macos")]
pub fn install(verbose: u8, force: bool) -> Result<()> {
    if !force && crabtalk_command::is_installed(LABEL) {
        println!("daemon is already running");
        return Ok(());
    }
    ensure_providers()?;
    let rendered = render_daemon_template(LAUNCHD_TEMPLATE, verbose)?;
    crabtalk_command::install(&rendered, LABEL)
}

#[cfg(target_os = "macos")]
pub fn uninstall() -> Result<()> {
    crabtalk_command::uninstall(LABEL)
}

#[cfg(target_os = "linux")]
pub fn install(verbose: u8, force: bool) -> Result<()> {
    if !force && crabtalk_command::is_installed(LABEL) {
        println!("daemon is already running");
        return Ok(());
    }
    ensure_providers()?;
    let rendered = render_daemon_template(SYSTEMD_TEMPLATE, verbose)?;
    crabtalk_command::install(&rendered, LABEL)
}

#[cfg(target_os = "linux")]
pub fn uninstall() -> Result<()> {
    crabtalk_command::uninstall(LABEL)
}

#[cfg(target_os = "windows")]
pub fn install(verbose: u8, force: bool) -> Result<()> {
    if !force && crabtalk_command::is_installed(LABEL) {
        println!("daemon is already running");
        return Ok(());
    }
    ensure_providers()?;
    let rendered = render_daemon_template(SCHTASKS_TEMPLATE, verbose)?;
    crabtalk_command::install(&rendered, LABEL)
}

#[cfg(target_os = "windows")]
pub fn uninstall() -> Result<()> {
    crabtalk_command::uninstall(LABEL)
}

#[cfg(not(any(target_os = "macos", target_os = "linux", target_os = "windows")))]
pub fn install(_verbose: u8, _force: bool) -> Result<()> {
    anyhow::bail!("daemon start is not supported on this platform")
}

#[cfg(not(any(target_os = "macos", target_os = "linux", target_os = "windows")))]
pub fn uninstall() -> Result<()> {
    anyhow::bail!("daemon stop is not supported on this platform")
}