You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
3.1 KiB
Rust
78 lines
3.1 KiB
Rust
extern crate irc;
|
|
extern crate postgres;
|
|
|
|
use irc::client::prelude::{ClientExt, Command, Config, IrcReactor};
|
|
use postgres::{NoTls, Client};
|
|
|
|
|
|
fn main() {
|
|
// Connect to IRC server
|
|
println!("[DEBUG] Connecting to IRC server...");
|
|
let irc_config = Config::load("config.toml").unwrap();
|
|
let mut reactor = IrcReactor::new().unwrap();
|
|
let irc_client = reactor.prepare_client_and_connect(&irc_config).unwrap();
|
|
irc_client.identify().unwrap();
|
|
|
|
reactor.register_client_with_handler(irc_client, move |c, m| {
|
|
// Connect to database
|
|
let pg_conn = format!(
|
|
"host={} user={} password={} dbname={}",
|
|
irc_config.get_option("db_host").unwrap(),
|
|
irc_config.get_option("db_user").unwrap(),
|
|
irc_config.get_option("db_pass").unwrap(),
|
|
irc_config.get_option("db_name").unwrap(),
|
|
);
|
|
let mut pg_client = Client::connect(&pg_conn, NoTls).unwrap();
|
|
|
|
let src_nick = &m.source_nickname().unwrap_or("none");
|
|
let res_target = &m.response_target().unwrap_or("none");
|
|
|
|
if let Command::NOTICE(channel, message) = &m.command {
|
|
println!("[{:?}][{}]: {}", res_target, &channel, &message);
|
|
}
|
|
|
|
if let Command::PRIVMSG(channel, message) = &m.command {
|
|
|
|
|
|
let help_msg = vec![
|
|
format!("Hey there {}, here's what you can do.", res_target),
|
|
"Say '!create' followed by the title of your listing to create a new one.".to_string(),
|
|
"Example: '!create I am offering virtual guitar lessons/sessions'.".to_string(),
|
|
"Say '!delete' followed by the name or ID of an existing listing.".to_string(),
|
|
"Example: '!delete I am offering virtual guitar lessons/sessions'".to_string()
|
|
];
|
|
let help_msg = help_msg.join(" ");
|
|
|
|
// Print user messages to channel
|
|
println!("[{}][{}]: {}", res_target, src_nick, message);
|
|
|
|
if message.starts_with("!create") {
|
|
let user_msg = message.split(" ");
|
|
let mut user_msg: Vec<&str> = user_msg.collect();
|
|
user_msg.remove(0);
|
|
let user_msg = user_msg.join(" ");
|
|
let query_res = pg_client.query_one(
|
|
"INSERT INTO posts (nick, post_title) VALUES ($1, $2) RETURNING id",
|
|
&[&src_nick, &user_msg]
|
|
);
|
|
match query_res {
|
|
Ok(row) => {
|
|
let post_id: i32 = row.get("id");
|
|
let _ = c.send_privmsg(&channel, format!("Created new post: {}", post_id));
|
|
},
|
|
Err(err) => {
|
|
let _ = c.send_privmsg(&channel, format!("There was an error storing to DB! {}", err));
|
|
}
|
|
};
|
|
} else if message.starts_with("!delete") {
|
|
let _ = c.send_privmsg(&channel, "This feature is still being worked on.");
|
|
} else if message.starts_with("!help") {
|
|
let _ = c.send_privmsg(&channel, &help_msg);
|
|
}
|
|
}
|
|
Ok(())
|
|
});
|
|
|
|
reactor.run().unwrap();
|
|
}
|