1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
use std::{io, path};
use std::collections::HashMap;
use std::convert::Infallible;
use std::net::{SocketAddr, ToSocketAddrs};
use std::str::FromStr;
use clap::Parser;
use teloxide::prelude2::*;
use tracing::{debug, info, Level};
use tracing::instrument;
use tracing_appender::non_blocking::WorkerGuard;
use tracing_subscriber::fmt;
use tracing_subscriber::layer::SubscriberExt;
use warp::Filter;
use config::Config;
use crate::database::DbPool;
mod database;
mod user;
mod web;
mod bot;
mod token;
mod config;
const CONFIG_FILE_NAME: &str = "kimikuri.json";
fn with_object<T: Clone + Send>(obj: T) -> impl Filter<Extract=(T, ), Error=Infallible> + Clone {
warp::any().map(move || obj.clone())
}
#[derive(Parser, Debug)]
#[clap(author, version, about, long_about = None)]
struct CliArgs {
/// The configuration file to use.
#[clap(short, long, default_value_t = String::from(CONFIG_FILE_NAME))]
config: String,
}
#[instrument]
#[tokio::main]
async fn main() {
let args = CliArgs::parse();
eprintln!("Loading configuration file {}...", args.config);
let config = Config::from_file(args.config.as_str());
// configure logger
let log_level = match Level::from_str(&*config.log_level) {
Ok(l) => l,
Err(_) => {
eprintln!("Invalid log level: {}. Use {:?} instead.",
config.log_level, config::DEFAULT_LOG_LEVEL);
Level::from_str(config::DEFAULT_LOG_LEVEL).unwrap()
}
};
eprintln!("Configuration is loaded. Set log level to {:?}.", log_level);
let _guard: WorkerGuard;
let subscriber = fmt::Subscriber::builder()
.with_max_level(log_level)
.with_writer(io::stderr) // log to stderr
.finish();
if !config.log_file.is_empty() {
let log_file_path = path::Path::new(&config.log_file);
let parent = log_file_path.parent()
.expect("Invalid log_file: Cannot extract parent.");
let filename = log_file_path.file_name()
.expect("Invalid log_file: Cannot extract file name.");
let (nb_file_appender, guard) = tracing_appender::non_blocking(
tracing_appender::rolling::never(parent, filename));
_guard = guard;
let subscriber = subscriber.with(
fmt::Layer::default()
.with_writer(nb_file_appender) // log to file
.with_ansi(false) // remove color control characters from log file
);
tracing::subscriber::set_global_default(subscriber)
.expect("Failed to set default subscriber");
} else {
// log file is not specified, do not write logs to file
tracing::subscriber::set_global_default(subscriber)
.expect("Failed to set default subscriber");
}
let db = config.db_file.as_str();
info!(db, "Opening database...");
let db: DbPool = database::open(db, config.sqlite_thread_pool_size)
.await.expect(&*format!("cannot open database {}", db));
info!("Spawning bot coroutine...");
let bot = Bot::new(config.bot_token);
tokio::spawn(bot::repl(bot.clone(), db.clone()));
info!("Initializing HTTP routes...");
let route_post = warp::post()
.and(warp::body::content_length_limit(config.max_body_size))
.and(warp::body::json())
.and(with_object(db.clone()))
.and(with_object(bot.clone()))
.and_then(web::handler);
let route_get = warp::get()
.and(warp::query::<HashMap<String, String>>())
.and(with_object(db.clone()))
.and(with_object(bot.clone()))
.and_then(web::get_handler);
let routes = warp::path("message")
.and(route_post).or(route_get);
info!("Starting HTTP server...");
let endpoint: SocketAddr = config.listen.parse()
.unwrap_or_else(|_| config.listen
.to_socket_addrs()
.expect("Cannot resolve endpoint.")
.next()
.expect("Cannot resolve endpoint."));
info!("Start listening on {}", endpoint);
tokio::spawn(warp::serve(routes).run(endpoint));
debug!("Waiting for Ctrl-C in main coroutine...");
tokio::signal::ctrl_c().await.unwrap();
// gracefully shutdown the database connection
info!("Closing database...");
db.close().await;
}
|
