Thelie/Mention2Mail
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added the beginning of the threaded code.

Browse source
This commit is contained in:
Thelie 2021-05-14 21:16:20 +02:00
parent c47b511e04
commit 8ab294277b
1 changed files with 50 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

85
src/main.rs
View file

@ -23,6 +23,11 @@ use std::{
env::args,
path::PathBuf,
collections::VecDeque,
sync::mpsc::{
channel,
Sender,
},
thread,
};
use irc_proto::command::{
CapSubCommand,
@ -134,41 +139,7 @@ fn handle_message(message: Message) -> Option<Message> {
return None
}
fn main() {
let mut server_conf_flag = false;
let mut config_path = None;
let mut server_conf_path = None;
for arg in args().skip(1) {
match arg.as_str() {
"-s" => server_conf_flag = true,
"-h" => {
println!("Usage: mention2mail [config file] [-s server config directory]");
},
_ => if server_conf_flag {
server_conf_path = Some(arg);
} else {
config_path = Some(arg);
}
}
}
let config;
match config_path {
Some(p) => config = config::get_main_config(PathBuf::from(p))
.expect("Could not get config"),
None => config = config::get_main_config(PathBuf::from("/etc/Mention2Mail/config.toml"))
.expect("Could not get default config in /etc/Mention2Mail/default.toml"),
}
let mut server_configs;
match server_conf_path {
Some(p) => server_configs = config::get_server_configs(config, PathBuf::from(p))
.expect("Could not get server config."),
None => (),
}
fn handle_server(config: Value, tx: Sender<[String; 4]>) {
let mut stream = connect::connect_irc(&config).unwrap();
let mut message_queue = get_irc_identify_messages(&config).unwrap();
@ -225,3 +196,47 @@ fn main() {
}
}
fn main() {
let mut server_conf_flag = false;
let mut config_path = None;
let mut server_conf_path = None;
for arg in args().skip(1) {
match arg.as_str() {
"-s" => server_conf_flag = true,
"-h" => {
println!("Usage: mention2mail [config file] [-s server config directory]");
},
_ => if server_conf_flag {
server_conf_path = Some(arg);
} else {
config_path = Some(arg);
}
}
}
let config;
match config_path {
Some(p) => config = config::get_main_config(PathBuf::from(p))
.expect("Could not get config"),
None => config = config::get_main_config(PathBuf::from("/etc/Mention2Mail/config.toml"))
.expect("Could not get default config in /etc/Mention2Mail/default.toml"),
}
let mut server_configs;
match server_conf_path {
Some(p) => server_configs = config::get_server_configs(config, PathBuf::from(p))
.expect("Could not get server config."),
None => (),
}
let (tx,rx) = channel();
for s_conf in server_configs {
// TODO: create channel and spawn server threads
handle_server(s_conf, tx.clone());
}
}