use crate::package_manager::component::ComponentConfig; use crate::package_manager::os::detect_os; use crate::package_manager::{InstallMode, OsType}; use anyhow::Result; use log::trace; use rand::distr::Alphanumeric; use sha2::{Digest, Sha256}; use std::collections::HashMap; use std::path::PathBuf; pub struct PackageManager { pub mode: InstallMode, pub os_type: OsType, pub base_path: PathBuf, pub tenant: String, pub components: HashMap, } impl PackageManager { pub fn new(mode: InstallMode, tenant: Option) -> Result { let os_type = detect_os(); let base_path = if mode == InstallMode::Container { PathBuf::from("/opt/gbo") } else { std::env::current_dir()?.join("botserver-stack") }; let tenant = tenant.unwrap_or_else(|| "default".to_string()); let mut pm = PackageManager { mode, os_type, base_path, tenant, components: HashMap::new(), }; pm.register_components(); Ok(pm) } fn register_components(&mut self) { self.register_tables(); self.register_cache(); self.register_drive(); self.register_llm(); self.register_email(); self.register_proxy(); self.register_directory(); self.register_alm(); self.register_alm_ci(); self.register_dns(); self.register_webmail(); self.register_meeting(); self.register_table_editor(); self.register_doc_editor(); self.register_desktop(); self.register_devtools(); self.register_bot(); self.register_system(); self.register_vector_db(); self.register_host(); } fn register_drive(&mut self) { let drive_password = self.generate_secure_password(16); let farm_password = std::env::var("FARM_PASSWORD").unwrap_or_else(|_| self.generate_secure_password(32)); let encrypted_drive_password = self.encrypt_password(&drive_password, &farm_password); self.components.insert("drive".to_string(), ComponentConfig { name: "drive".to_string(), required: true, ports: vec![9000, 9001], dependencies: vec![], linux_packages: vec![], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://dl.min.io/server/minio/release/linux-amd64/minio".to_string()), binary_name: Some("minio".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "wget https://dl.min.io/client/mc/release/linux-amd64/mc -O {{BIN_PATH}}/mc".to_string(), "chmod +x {{BIN_PATH}}/mc".to_string(), format!("{{{{BIN_PATH}}}}/mc alias set mc http://localhost:9000 gbdriveuser {}", drive_password), "{{BIN_PATH}}/mc mb mc/default.gbai".to_string(), format!("{{{{BIN_PATH}}}}/mc admin user add mc gbdriveuser {}", drive_password), "{{BIN_PATH}}/mc admin policy attach mc readwrite --user=gbdriveuser".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![ "wget https://dl.min.io/client/mc/release/darwin-amd64/mc -O {{BIN_PATH}}/mc".to_string(), "chmod +x {{BIN_PATH}}/mc".to_string() ], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::from([ ("MINIO_ROOT_USER".to_string(), "gbdriveuser".to_string()), ("MINIO_ROOT_PASSWORD".to_string(), drive_password) ]), exec_cmd: "nohup {{BIN_PATH}}/minio server {{DATA_PATH}} --address :9000 --console-address :9001 > {{LOGS_PATH}}/minio.log 2>&1 &".to_string(), }); self.update_drive_credentials_in_database(&encrypted_drive_password) .ok(); } fn update_drive_credentials_in_database(&self, encrypted_drive_password: &str) -> Result<()> { use crate::shared::models::schema::bots::dsl::*; use diesel::pg::PgConnection; use diesel::prelude::*; use uuid::Uuid; let database_url = std::env::var("DATABASE_URL") .unwrap_or_else(|_| "postgres://gbuser:@localhost:5432/botserver".to_string()); if let Ok(mut conn) = PgConnection::establish(&database_url) { let system_bot_id = Uuid::parse_str("00000000-0000-0000-0000-000000000000")?; diesel::update(bots) .filter(bot_id.eq(system_bot_id)) .set(config.eq(serde_json::json!({ "encrypted_drive_password": encrypted_drive_password, }))) .execute(&mut conn)?; trace!("Updated drive credentials in database for system bot"); } Ok(()) } fn register_tables(&mut self) { let db_password = std::env::var("DATABASE_URL") .ok() .and_then(|url| { if let Some(stripped) = url.strip_prefix("postgres://gbuser:") { if let Some(at_pos) = stripped.find('@') { Some(stripped[..at_pos].to_string()) } else { None } } else { None } }) .unwrap_or_else(|| self.generate_secure_password(16)); self.components.insert("tables".to_string(), ComponentConfig { name: "tables".to_string(), required: true, ports: vec![5432], dependencies: vec![], linux_packages: vec![], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/theseus-rs/postgresql-binaries/releases/download/18.0.0/postgresql-18.0.0-x86_64-unknown-linux-gnu.tar.gz".to_string()), binary_name: None, pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "chmod +x ./bin/*".to_string(), format!("if [ ! -d \"{{{{DATA_PATH}}}}/pgdata\" ]; then PG_PASSWORD={} ./bin/initdb -D {{{{DATA_PATH}}}}/pgdata -U gbuser --pwfile=<(echo $PG_PASSWORD); fi", db_password).to_string(), "echo \"data_directory = '{{DATA_PATH}}/pgdata'\" > {{CONF_PATH}}/postgresql.conf".to_string(), "echo \"ident_file = '{{CONF_PATH}}/pg_ident.conf'\" >> {{CONF_PATH}}/postgresql.conf".to_string(), "echo \"port = 5432\" >> {{CONF_PATH}}/postgresql.conf".to_string(), "echo \"listen_addresses = '*'\" >> {{CONF_PATH}}/postgresql.conf".to_string(), "echo \"log_directory = '{{LOGS_PATH}}'\" >> {{CONF_PATH}}/postgresql.conf".to_string(), "echo \"logging_collector = on\" >> {{CONF_PATH}}/postgresql.conf".to_string(), "echo \"host all all all md5\" > {{CONF_PATH}}/pg_hba.conf".to_string(), "touch {{CONF_PATH}}/pg_ident.conf".to_string(), format!("./bin/pg_ctl -D {{{{DATA_PATH}}}}/pgdata -l {{{{LOGS_PATH}}}}/postgres.log start; for i in 1 2 3 4 5 6 7 8 9 10; do ./bin/pg_isready -h localhost -p 5432 >/dev/null 2>&1 && break; echo 'Waiting for PostgreSQL to start...' >&2; sleep 1; done; ./bin/pg_isready -h localhost -p 5432"), format!("PGPASSWORD={} ./bin/psql -h localhost -U gbuser -d postgres -c \"CREATE DATABASE botserver WITH OWNER gbuser\" 2>&1 | grep -v 'already exists' || true", db_password) ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![ "chmod +x ./bin/*".to_string(), "if [ ! -d \"{{DATA_PATH}}/pgdata\" ]; then ./bin/initdb -D {{DATA_PATH}}/pgdata -U postgres; fi".to_string(), ], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "./bin/pg_ctl -D {{DATA_PATH}}/pgdata -l {{LOGS_PATH}}/postgres.log start".to_string(), }); } fn register_cache(&mut self) { self.components.insert("cache".to_string(), ComponentConfig { name: "cache".to_string(), required: true, ports: vec![6379], dependencies: vec![], linux_packages: vec!["curl".to_string(), "gnupg".to_string(), "lsb-release".to_string()], macos_packages: vec!["redis".to_string()], windows_packages: vec![], download_url: None, binary_name: Some("valkey-server".to_string()), pre_install_cmds_linux: vec![ "sudo bash -c 'if [ ! -f /usr/share/keyrings/valkey.gpg ]; then curl -fsSL https://packages.redis.io/gpg | gpg --dearmor -o /usr/share/keyrings/valkey.gpg; fi'".to_string(), "sudo bash -c 'if [ ! -f /etc/apt/sources.list.d/valkey.list ]; then echo \"deb [signed-by=/usr/share/keyrings/valkey.gpg] https://packages.redis.io/deb $(lsb_release -cs) main\" | tee /etc/apt/sources.list.d/valkey.list; fi'".to_string(), "sudo apt-get update && sudo apt-get install -y valkey".to_string() ], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "valkey-server --port 6379 --dir {{DATA_PATH}}".to_string(), }); } fn register_llm(&mut self) { self.components.insert("llm".to_string(), ComponentConfig { name: "llm".to_string(), required: true, ports: vec![8081, 8082], dependencies: vec![], linux_packages: vec!["unzip".to_string()], macos_packages: vec!["unzip".to_string()], windows_packages: vec![], download_url: Some("https://github.com/ggml-org/llama.cpp/releases/download/b6148/llama-b6148-bin-ubuntu-x64.zip".to_string()), binary_name: Some("llama-server".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "wget -q https://huggingface.co/bartowski/DeepSeek-R1-Distill-Qwen-1.5B-GGUF/resolve/main/DeepSeek-R1-Distill-Qwen-1.5B-Q3_K_M.gguf -P {{DATA_PATH}}".to_string(), "wget -q https://huggingface.co/CompendiumLabs/bge-small-en-v1.5-gguf/resolve/main/bge-small-en-v1.5-f32.gguf -P {{DATA_PATH}}".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![ "wget -q https://huggingface.co/bartowski/DeepSeek-R1-Distill-Qwen-1.5B-GGUF/resolve/main/DeepSeek-R1-Distill-Qwen-1.5B-Q3_K_M.gguf -P {{DATA_PATH}}".to_string(), "wget -q https://huggingface.co/CompendiumLabs/bge-small-en-v1.5-gguf/resolve/main/bge-small-en-v1.5-f32.gguf -P {{DATA_PATH}}".to_string() ], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "nohup {{BIN_PATH}}/llama-server -m {{DATA_PATH}}/DeepSeek-R1-Distill-Qwen-1.5B-Q3_K_M.gguf --port 8081 > {{LOGS_PATH}}/llm-main.log 2>&1 & nohup {{BIN_PATH}}/llama-server -m {{DATA_PATH}}/bge-small-en-v1.5-f32.gguf --port 8082 --embedding > {{LOGS_PATH}}/llm-embed.log 2>&1 &".to_string(), }); } fn register_email(&mut self) { self.components.insert("email".to_string(), ComponentConfig { name: "email".to_string(), required: false, ports: vec![25, 80, 110, 143, 465, 587, 993, 995, 4190], dependencies: vec![], linux_packages: vec!["libcap2-bin".to_string(), "resolvconf".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/stalwartlabs/stalwart/releases/download/v0.13.1/stalwart-x86_64-unknown-linux-gnu.tar.gz".to_string()), binary_name: Some("stalwart".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "setcap 'cap_net_bind_service=+ep' {{BIN_PATH}}/stalwart".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/stalwart --config {{CONF_PATH}}/config.toml".to_string(), }); } fn register_proxy(&mut self) { self.components.insert("proxy".to_string(), ComponentConfig { name: "proxy".to_string(), required: false, ports: vec![80, 443], dependencies: vec![], linux_packages: vec!["libcap2-bin".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/caddyserver/caddy/releases/download/v2.10.0-beta.3/caddy_2.10.0-beta.3_linux_amd64.tar.gz".to_string()), binary_name: Some("caddy".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "setcap 'cap_net_bind_service=+ep' {{BIN_PATH}}/caddy".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::from([ ("XDG_DATA_HOME".to_string(), "{{DATA_PATH}}".to_string()) ]), exec_cmd: "{{BIN_PATH}}/caddy run --config {{CONF_PATH}}/Caddyfile".to_string(), }); } fn register_directory(&mut self) { self.components.insert("directory".to_string(), ComponentConfig { name: "directory".to_string(), required: false, ports: vec![8080], dependencies: vec![], linux_packages: vec!["libcap2-bin".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/zitadel/zitadel/releases/download/v2.71.2/zitadel-linux-amd64.tar.gz".to_string()), binary_name: Some("zitadel".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "setcap 'cap_net_bind_service=+ep' {{BIN_PATH}}/zitadel".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/zitadel start --config {{CONF_PATH}}/zitadel.yaml".to_string(), }); } fn register_alm(&mut self) { self.components.insert("alm".to_string(), ComponentConfig { name: "alm".to_string(), required: false, ports: vec![3000], dependencies: vec![], linux_packages: vec!["git".to_string(), "git-lfs".to_string()], macos_packages: vec!["git".to_string(), "git-lfs".to_string()], windows_packages: vec![], download_url: Some("https://codeberg.org/forgejo/forgejo/releases/download/v10.0.2/forgejo-10.0.2-linux-amd64".to_string()), binary_name: Some("forgejo".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::from([ ("USER".to_string(), "alm".to_string()), ("HOME".to_string(), "{{DATA_PATH}}".to_string()) ]), exec_cmd: "{{BIN_PATH}}/forgejo web --work-path {{DATA_PATH}}".to_string(), }); } fn register_alm_ci(&mut self) { self.components.insert("alm-ci".to_string(), ComponentConfig { name: "alm-ci".to_string(), required: false, ports: vec![], dependencies: vec!["alm".to_string()], linux_packages: vec!["git".to_string(), "curl".to_string(), "gnupg".to_string(), "ca-certificates".to_string(), "build-essential".to_string()], macos_packages: vec!["git".to_string(), "node".to_string()], windows_packages: vec![], download_url: Some("https://code.forgejo.org/forgejo/runner/releases/download/v6.3.1/forgejo-runner-6.3.1-linux-amd64".to_string()), binary_name: Some("forgejo-runner".to_string()), pre_install_cmds_linux: vec![ "curl -fsSL https://deb.nodesource.com/setup_22.x | bash -".to_string(), "apt-get install -y nodejs".to_string() ], post_install_cmds_linux: vec![ "npm install -g pnpm@latest".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![ "npm install -g pnpm@latest".to_string() ], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/forgejo-runner daemon --config {{CONF_PATH}}/config.yaml".to_string(), }); } fn register_dns(&mut self) { self.components.insert("dns".to_string(), ComponentConfig { name: "dns".to_string(), required: false, ports: vec![53], dependencies: vec![], linux_packages: vec![], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/coredns/coredns/releases/download/v1.12.4/coredns_1.12.4_linux_amd64.tgz".to_string()), binary_name: Some("coredns".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![ "setcap cap_net_bind_service=+ep {{BIN_PATH}}/coredns".to_string() ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/coredns -conf {{CONF_PATH}}/Corefile".to_string(), }); } fn register_webmail(&mut self) { self.components.insert("webmail".to_string(), ComponentConfig { name: "webmail".to_string(), required: false, ports: vec![8080], dependencies: vec!["email".to_string()], linux_packages: vec!["ca-certificates".to_string(), "apt-transport-https".to_string(), "php8.1".to_string(), "php8.1-fpm".to_string()], macos_packages: vec!["php".to_string()], windows_packages: vec![], download_url: Some("https://github.com/roundcube/roundcubemail/releases/download/1.6.6/roundcubemail-1.6.6-complete.tar.gz".to_string()), binary_name: None, pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "php -S 0.0.0.0:8080 -t {{DATA_PATH}}/roundcubemail".to_string(), }); } fn register_meeting(&mut self) { self.components.insert("meeting".to_string(), ComponentConfig { name: "meeting".to_string(), required: false, ports: vec![7880, 3478], dependencies: vec![], linux_packages: vec!["coturn".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/livekit/livekit/releases/download/v1.8.4/livekit_1.8.4_linux_amd64.tar.gz".to_string()), binary_name: Some("livekit-server".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/livekit-server --config {{CONF_PATH}}/config.yaml".to_string(), }); } fn register_table_editor(&mut self) { self.components.insert( "table_editor".to_string(), ComponentConfig { name: "table_editor".to_string(), required: false, ports: vec![5757], dependencies: vec!["tables".to_string()], linux_packages: vec!["curl".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: Some("http://get.nocodb.com/linux-x64".to_string()), binary_name: Some("nocodb".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/nocodb".to_string(), }, ); } fn register_doc_editor(&mut self) { self.components.insert( "doc_editor".to_string(), ComponentConfig { name: "doc_editor".to_string(), required: false, ports: vec![9980], dependencies: vec![], linux_packages: vec!["gnupg".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: None, binary_name: Some("coolwsd".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "coolwsd --config-file={{CONF_PATH}}/coolwsd.xml".to_string(), }, ); } fn register_desktop(&mut self) { self.components.insert( "desktop".to_string(), ComponentConfig { name: "desktop".to_string(), required: false, ports: vec![3389], dependencies: vec![], linux_packages: vec!["xvfb".to_string(), "xrdp".to_string(), "xfce4".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: None, binary_name: None, pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "xrdp --nodaemon".to_string(), }, ); } fn register_devtools(&mut self) { self.components.insert( "devtools".to_string(), ComponentConfig { name: "devtools".to_string(), required: false, ports: vec![], dependencies: vec![], linux_packages: vec!["xclip".to_string(), "git".to_string(), "curl".to_string()], macos_packages: vec!["git".to_string()], windows_packages: vec![], download_url: None, binary_name: None, pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "".to_string(), }, ); } fn register_bot(&mut self) { self.components.insert( "bot".to_string(), ComponentConfig { name: "bot".to_string(), required: false, ports: vec![3000], dependencies: vec![], linux_packages: vec![ "curl".to_string(), "gnupg".to_string(), "ca-certificates".to_string(), "git".to_string(), ], macos_packages: vec!["node".to_string()], windows_packages: vec![], download_url: None, binary_name: None, pre_install_cmds_linux: vec![ "curl -fsSL https://deb.nodesource.com/setup_22.x | bash -".to_string(), "apt-get install -y nodejs".to_string(), ], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::from([("DISPLAY".to_string(), ":99".to_string())]), exec_cmd: "".to_string(), }, ); } fn register_system(&mut self) { self.components.insert( "system".to_string(), ComponentConfig { name: "system".to_string(), required: false, ports: vec![8000], dependencies: vec![], linux_packages: vec!["curl".to_string(), "unzip".to_string(), "git".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: None, binary_name: None, pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "".to_string(), }, ); } fn register_vector_db(&mut self) { self.components.insert("vector_db".to_string(), ComponentConfig { name: "vector_db".to_string(), required: false, ports: vec![6333], dependencies: vec![], linux_packages: vec![], macos_packages: vec![], windows_packages: vec![], download_url: Some("https://github.com/qdrant/qdrant/releases/latest/download/qdrant-x86_64-unknown-linux-gnu.tar.gz".to_string()), binary_name: Some("qdrant".to_string()), pre_install_cmds_linux: vec![], post_install_cmds_linux: vec![], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "{{BIN_PATH}}/qdrant --storage-path {{DATA_PATH}}".to_string(), }); } fn register_host(&mut self) { self.components.insert( "host".to_string(), ComponentConfig { name: "host".to_string(), required: false, ports: vec![], dependencies: vec![], linux_packages: vec!["sshfs".to_string(), "bridge-utils".to_string()], macos_packages: vec![], windows_packages: vec![], download_url: None, binary_name: None, pre_install_cmds_linux: vec![ "echo 'net.ipv4.ip_forward=1' | tee -a /etc/sysctl.conf".to_string(), "sysctl -p".to_string(), ], post_install_cmds_linux: vec![ "lxd init --auto".to_string(), "lxc storage create default dir".to_string(), "lxc profile device add default root disk path=/ pool=default".to_string(), ], pre_install_cmds_macos: vec![], post_install_cmds_macos: vec![], pre_install_cmds_windows: vec![], post_install_cmds_windows: vec![], env_vars: HashMap::new(), exec_cmd: "".to_string(), }, ); } pub fn start(&self, component: &str) -> Result { if let Some(component) = self.components.get(component) { let bin_path = self.base_path.join("bin").join(&component.name); let data_path = self.base_path.join("data").join(&component.name); let conf_path = self.base_path.join("conf").join(&component.name); let logs_path = self.base_path.join("logs").join(&component.name); // For PostgreSQL, check if it's already running if component.name == "tables" { let check_cmd = format!( "./bin/pg_ctl -D {} status", data_path.join("pgdata").display() ); let check_output = std::process::Command::new("sh") .current_dir(&bin_path) .arg("-c") .arg(&check_cmd) .output(); if let Ok(output) = check_output { if output.status.success() { trace!( "Component {} is already running, skipping start", component.name ); // Return a dummy child process handle - PostgreSQL is already running return Ok(std::process::Command::new("sh") .arg("-c") .arg("echo 'Already running'") .spawn()?); } } } let rendered_cmd = component .exec_cmd .replace("{{BIN_PATH}}", &bin_path.to_string_lossy()) .replace("{{DATA_PATH}}", &data_path.to_string_lossy()) .replace("{{CONF_PATH}}", &conf_path.to_string_lossy()) .replace("{{LOGS_PATH}}", &logs_path.to_string_lossy()); trace!( "Starting component {} with command: {}", component.name, rendered_cmd ); let child = std::process::Command::new("sh") .current_dir(&bin_path) .arg("-c") .arg(&rendered_cmd) .spawn(); // Handle "already running" errors gracefully match child { Ok(c) => Ok(c), Err(e) => { let err_msg = e.to_string(); if err_msg.contains("already running") || component.name == "tables" { trace!( "Component {} may already be running, continuing anyway", component.name ); Ok(std::process::Command::new("sh") .arg("-c") .arg("echo 'Already running'") .spawn()?) } else { Err(e.into()) } } } } else { Err(anyhow::anyhow!("Component {} not found", component)) } } fn generate_secure_password(&self, length: usize) -> String { // Use the non-deprecated `rng` function to obtain a thread-local RNG. let mut rng = rand::rng(); // Generate `length` alphanumeric characters. (0..length) .map(|_| { // `Alphanumeric` implements the `Distribution` trait. // Use the fully qualified `rand::Rng::sample` method to avoid needing an explicit import. let byte = rand::Rng::sample(&mut rng, Alphanumeric); char::from(byte) }) .collect() } fn encrypt_password(&self, password: &str, key: &str) -> String { let mut hasher = Sha256::new(); hasher.update(key.as_bytes()); hasher.update(password.as_bytes()); format!("{:x}", hasher.finalize()) } }