133 lines
4.0 KiB
Rust
133 lines
4.0 KiB
Rust
use std::env;
|
|
use std::fs;
|
|
|
|
use colored::Colorize;
|
|
use serde::Deserialize;
|
|
use toml::from_str;
|
|
|
|
use crate::data_struct::{ica, tailchat};
|
|
|
|
/// Icalingua bot 的配置
|
|
#[derive(Debug, Clone, Deserialize)]
|
|
pub struct IcaConfig {
|
|
/// icalingua 私钥
|
|
pub private_key: String,
|
|
/// icalingua 服务器地址
|
|
pub host: String,
|
|
/// bot 的 qq
|
|
pub self_id: ica::UserId,
|
|
/// 提醒的房间
|
|
#[serde(default = "default_empty_i64_vec")]
|
|
pub notice_room: Vec<ica::RoomId>,
|
|
/// 是否提醒
|
|
#[serde(default = "default_false")]
|
|
pub notice_start: bool,
|
|
/// 管理员列表
|
|
#[serde(default = "default_empty_i64_vec")]
|
|
pub admin_list: Vec<ica::UserId>,
|
|
/// 过滤列表
|
|
#[serde(default = "default_empty_i64_vec")]
|
|
pub filter_list: Vec<ica::UserId>,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Deserialize)]
|
|
pub struct TailchatConfig {
|
|
/// 服务器地址
|
|
pub host: String,
|
|
/// 机器人 App ID
|
|
pub app_id: String,
|
|
/// 机器人 App Secret
|
|
pub app_secret: String,
|
|
/// 提醒的房间
|
|
pub notice_room: Vec<(tailchat::GroupId, tailchat::ConverseId)>,
|
|
/// 是否提醒
|
|
#[serde(default = "default_false")]
|
|
pub notice_start: bool,
|
|
/// 管理员列表
|
|
#[serde(default = "default_empty_str_vec")]
|
|
pub admin_list: Vec<tailchat::UserId>,
|
|
/// 过滤列表
|
|
#[serde(default = "default_empty_str_vec")]
|
|
pub filter_list: Vec<tailchat::UserId>,
|
|
}
|
|
|
|
fn default_plugin_path() -> String { "./plugins".to_string() }
|
|
fn default_config_path() -> String { "./config".to_string() }
|
|
|
|
#[derive(Debug, Clone, Deserialize)]
|
|
pub struct PyConfig {
|
|
/// 插件路径
|
|
#[serde(default = "default_plugin_path")]
|
|
pub plugin_path: String,
|
|
/// 配置文件夹路径
|
|
#[serde(default = "default_config_path")]
|
|
pub config_path: String,
|
|
}
|
|
|
|
fn default_empty_i64_vec() -> Vec<i64> { Vec::new() }
|
|
fn default_empty_str_vec() -> Vec<String> { Vec::new() }
|
|
fn default_false() -> bool { false }
|
|
|
|
/// 主配置
|
|
#[derive(Debug, Clone, Deserialize)]
|
|
pub struct BotConfig {
|
|
/// 是否启用 icalingua
|
|
#[serde(default = "default_false")]
|
|
pub enable_ica: bool,
|
|
/// Ica 配置
|
|
pub ica: Option<IcaConfig>,
|
|
|
|
/// 是否启用 Tailchat
|
|
#[serde(default = "default_false")]
|
|
pub enable_tailchat: bool,
|
|
/// Tailchat 配置
|
|
pub tailchat: Option<TailchatConfig>,
|
|
|
|
/// 是否启用 Python 插件
|
|
#[serde(default = "default_false")]
|
|
pub enable_py: bool,
|
|
/// Python 插件配置
|
|
pub py: Option<PyConfig>,
|
|
}
|
|
|
|
impl BotConfig {
|
|
pub fn new_from_path(config_file_path: String) -> Self {
|
|
// try read config from file
|
|
let config = fs::read_to_string(&config_file_path).expect("Failed to read config file");
|
|
let ret: Self = from_str(&config).unwrap_or_else(|e| {
|
|
panic!("Failed to parse config file {}\ne:{:?}", &config_file_path, e)
|
|
});
|
|
ret
|
|
}
|
|
pub fn new_from_cli() -> Self {
|
|
// let config_file_path = env::args().nth(1).expect("No config path given");
|
|
// -c <config_file_path>
|
|
let mut config_file_path = String::new();
|
|
let mut args = env::args();
|
|
while let Some(arg) = args.next() {
|
|
if arg == "-c" {
|
|
config_file_path = args.next().unwrap_or_else(|| {
|
|
panic!("{}", "No config path given\nUsage: -c <config_file_path>".red())
|
|
});
|
|
break;
|
|
}
|
|
}
|
|
Self::new_from_path(config_file_path)
|
|
}
|
|
|
|
/// 检查是否启用 ica
|
|
pub fn check_ica(&self) -> bool { self.enable_ica }
|
|
|
|
/// 检查是否启用 Tailchat
|
|
pub fn check_tailchat(&self) -> bool { self.enable_tailchat }
|
|
|
|
/// 检查是否启用 Python 插件
|
|
pub fn check_py(&self) -> bool { self.enable_py }
|
|
|
|
pub fn ica(&self) -> IcaConfig { self.ica.clone().expect("No ica config found") }
|
|
pub fn tailchat(&self) -> TailchatConfig {
|
|
self.tailchat.clone().expect("No tailchat config found")
|
|
}
|
|
pub fn py(&self) -> PyConfig { self.py.clone().expect("No py config found") }
|
|
}
|