81 lines
1.8 KiB
Rust
81 lines
1.8 KiB
Rust
#[macro_use]
|
|
extern crate anyhow;
|
|
#[macro_use]
|
|
extern crate crossterm;
|
|
#[macro_use]
|
|
extern crate log;
|
|
#[macro_use]
|
|
extern crate serde;
|
|
|
|
mod config;
|
|
mod mail;
|
|
mod ui;
|
|
|
|
use std::fs::File;
|
|
use std::io::Read;
|
|
use std::path::PathBuf;
|
|
|
|
use anyhow::Result;
|
|
use futures::future::TryFutureExt;
|
|
use structopt::StructOpt;
|
|
use tokio::sync::{mpsc, oneshot};
|
|
|
|
use crate::config::Config;
|
|
|
|
type ExitSender = oneshot::Sender<()>;
|
|
|
|
#[derive(Debug, StructOpt)]
|
|
struct Opt {
|
|
/// The path to the log file. By default, does not log.
|
|
#[structopt(long = "log-file")]
|
|
log_file: Option<PathBuf>,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
let opt = Opt::from_args();
|
|
|
|
setup_logger(&opt)?;
|
|
|
|
let config: Config = {
|
|
let mut config_file = File::open("config.toml")?;
|
|
let mut contents = Vec::new();
|
|
config_file.read_to_end(&mut contents)?;
|
|
toml::from_slice(&contents)?
|
|
};
|
|
|
|
let (exit_tx, exit_rx) = oneshot::channel::<()>();
|
|
let (mail_tx, mail_rx) = mpsc::unbounded_channel();
|
|
|
|
tokio::spawn(mail::run_mail(config.clone(), mail_rx).unwrap_or_else(report_err));
|
|
let mut stdout = std::io::stdout();
|
|
tokio::spawn(ui::run_ui(stdout, exit_tx).unwrap_or_else(report_err));
|
|
|
|
exit_rx.await?;
|
|
Ok(())
|
|
}
|
|
|
|
fn report_err(err: anyhow::Error) {
|
|
error!("error: {:?}", err);
|
|
}
|
|
|
|
fn setup_logger(opt: &Opt) -> Result<()> {
|
|
let mut fern = fern::Dispatch::new()
|
|
.format(|out, message, record| {
|
|
out.finish(format_args!(
|
|
"{}[{}][{}] {}",
|
|
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
|
|
record.target(),
|
|
record.level(),
|
|
message
|
|
))
|
|
})
|
|
.level(log::LevelFilter::Debug);
|
|
|
|
if let Some(path) = &opt.log_file {
|
|
fern = fern.chain(fern::log_file(path)?);
|
|
}
|
|
|
|
fern.apply()?;
|
|
Ok(())
|
|
}
|