forked from projectmoon/tenebrous-dicebot
71 lines
1.8 KiB
Rust
71 lines
1.8 KiB
Rust
use chronicle_dicebot::bot::run_bot;
|
|
use chronicle_dicebot::bot::Config;
|
|
use env_logger::Env;
|
|
use std::fs;
|
|
use std::path::PathBuf;
|
|
|
|
fn read_config<P: Into<PathBuf>>(config_path: P) -> Result<Config, Box<dyn std::error::Error>> {
|
|
let config_path = config_path.into();
|
|
let config = {
|
|
let contents = fs::read_to_string(&config_path)?;
|
|
deserialize_config(&contents)?
|
|
};
|
|
|
|
Ok(config)
|
|
}
|
|
|
|
fn deserialize_config(contents: &str) -> Result<Config, Box<dyn std::error::Error>> {
|
|
let config = { toml::from_str(&contents)? };
|
|
Ok(config)
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
env_logger::from_env(Env::default().default_filter_or("chronicle_dicebot=info")).init();
|
|
|
|
let config_path = std::env::args()
|
|
.skip(1)
|
|
.next()
|
|
.expect("Need a config as an argument");
|
|
|
|
let cfg = read_config(config_path)?;
|
|
|
|
run_bot(cfg).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use indoc::indoc;
|
|
|
|
#[test]
|
|
fn deserialize_config_without_bot_section_test() {
|
|
let contents = indoc! {"
|
|
[matrix]
|
|
home_server = 'https://matrix.example.com'
|
|
username = 'username'
|
|
password = 'password'
|
|
"};
|
|
|
|
let cfg: Result<_, _> = deserialize_config(contents);
|
|
assert_eq!(true, cfg.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn deserialize_config_without_oldest_message_setting_test() {
|
|
let contents = indoc! {"
|
|
[matrix]
|
|
home_server = 'https://matrix.example.com'
|
|
username = 'username'
|
|
password = 'password'
|
|
|
|
[bot]
|
|
not_a_real_setting = 2
|
|
"};
|
|
|
|
let cfg: Result<_, _> = deserialize_config(contents);
|
|
assert_eq!(true, cfg.is_ok());
|
|
}
|
|
}
|