use serde::Deserialize;
use std::path::PathBuf;
use anyhow::{Result, Context};
#[derive(Debug, Deserialize, Clone)]
pub struct Config {
pub rcon: RconConfig,
pub ai: Option<AiConfig>,
pub ollama: Option<OllamaConfig>,
#[serde(default)]
pub backup: BackupConfig,
#[serde(default)]
pub notification: NotificationConfig,
}
#[derive(Debug, Deserialize, Clone)]
pub struct RconConfig {
#[serde(default = "default_rcon_host")]
pub host: String,
#[serde(default = "default_rcon_port")]
pub port: u16,
pub password: String,
}
fn default_rcon_host() -> String { "127.0.0.1".to_string() }
fn default_rcon_port() -> u16 { 25575 }
#[derive(Debug, Deserialize, Clone)]
pub struct AiConfig {
pub api_url: String,
pub api_key: String,
#[serde(default = "default_model")]
pub model: String,
#[serde(default = "default_trigger")]
pub trigger: String,
#[serde(default = "default_max_tokens")]
pub max_tokens: u32,
#[serde(default = "default_temperature")]
pub temperature: f32,
}
fn default_model() -> String { "gpt-3.5-turbo".to_string() }
fn default_trigger() -> String { "!".to_string() }
fn default_max_tokens() -> u32 { 150 }
fn default_temperature() -> f32 { 0.7 }
#[derive(Debug, Deserialize, Clone)]
pub struct OllamaConfig {
#[serde(default = "default_ollama_enabled")]
pub enabled: bool,
#[serde(default = "default_ollama_url")]
pub url: String,
#[serde(default = "default_ollama_model")]
pub model: String,
}
fn default_ollama_enabled() -> bool { false }
fn default_ollama_url() -> String { "http://localhost:11434/api/generate".to_string() }
fn default_ollama_model() -> String { "qwen:0.5b".to_string() }
#[derive(Debug, Deserialize, Clone)]
pub struct BackupConfig {
#[serde(default = "default_world_dir")]
pub world_dir: String,
#[serde(default = "default_backup_dest")]
pub backup_dest: String,
#[serde(default = "default_retain_days")]
pub retain_days: u32,
}
fn default_world_dir() -> String { "world".to_string() }
fn default_backup_dest() -> String { "../backups".to_string() }
fn default_retain_days() -> u32 { 7 }
impl Default for BackupConfig {
fn default() -> Self {
Self {
world_dir: default_world_dir(),
backup_dest: default_backup_dest(),
retain_days: default_retain_days(),
}
}
}
#[derive(Debug, Deserialize, Clone)]
pub struct NotificationConfig {
#[serde(default)]
pub telegram_bot_token: String,
#[serde(default)]
pub telegram_chat_id: String,
#[serde(default = "default_termux_notify")]
pub termux_notify: bool,
}
fn default_termux_notify() -> bool { true }
impl Default for NotificationConfig {
fn default() -> Self {
Self {
telegram_bot_token: String::new(),
telegram_chat_id: String::new(),
termux_notify: default_termux_notify(),
}
}
}
impl Config {
pub fn load(path: &PathBuf) -> Result<Self> {
let content = std::fs::read_to_string(path)
.with_context(|| format!("Failed to read config file: {:?}", path))?;
let mut config: Config = toml::from_str(&content)
.with_context(|| "Failed to parse config file")?;
if let Some(ref ai) = config.ai {
if ai.api_key.is_empty() || ai.api_url.is_empty() {
config.ai = None;
}
}
Ok(config)
}
pub fn load_from_str(content: &str) -> Result<Self> {
let mut config: Config = toml::from_str(content)
.with_context(|| "Failed to parse config content")?;
if let Some(ref ai) = config.ai {
if ai.api_key.is_empty() || ai.api_url.is_empty() {
config.ai = None;
}
}
Ok(config)
}
}