Delete 'ßqa'
This commit is contained in:
parent
20e5d27974
commit
456830fc3a
1 changed files with 0 additions and 263 deletions
263
ßqa
263
ßqa
|
|
@ -1,263 +0,0 @@
|
|||
/* Copyright 2021 Daniel Mowitz
|
||||
* This file is part of Mention2Mail.
|
||||
*
|
||||
* Mention2Mail is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* Mention2Mail is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with Mention2Mail. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
use std::{
|
||||
collections::VecDeque,
|
||||
collections::HashMap,
|
||||
sync::mpsc::Sender,
|
||||
};
|
||||
use irc_proto::command::{
|
||||
CapSubCommand,
|
||||
Command,
|
||||
};
|
||||
use toml::map::{
|
||||
Map,
|
||||
Keys
|
||||
};
|
||||
use irc_proto::message::Message;
|
||||
use toml::value::Value;
|
||||
|
||||
struct MessageRingBuf<const SIZE: usize> {
|
||||
buf: [Option<[String; 3]>; SIZE],
|
||||
ptr: usize,
|
||||
}
|
||||
|
||||
impl <const SIZE: usize> MessageRingBuf<SIZE> {
|
||||
pub fn new() -> Self {
|
||||
Self{
|
||||
buf: [None; SIZE],
|
||||
ptr: 0
|
||||
}
|
||||
}
|
||||
|
||||
pub fn push(&mut self, val: [String; 3]) {
|
||||
self.ptr = (self.ptr + 1) % SIZE;
|
||||
self.buf[self.ptr] = Some(val);
|
||||
}
|
||||
|
||||
pub fn pop(&mut self) -> Option<[String; 3]> {
|
||||
self.buf[self.ptr].take()
|
||||
}
|
||||
|
||||
pub fn get_queue(& self) -> [Option<String>; SIZE] {
|
||||
let mut queue = self.buf.clone();
|
||||
|
||||
for i in self.ptr .. SIZE {
|
||||
queue[i - self.ptr] = self.buf[i];
|
||||
}
|
||||
|
||||
for i in 0 .. self.ptr {
|
||||
queue[SIZE - self.ptr + i] = self.buf[i];
|
||||
}
|
||||
|
||||
return queue
|
||||
}
|
||||
}
|
||||
|
||||
/// Constructs a VecDeque with the messages
|
||||
/// required to identify with an IRC server.
|
||||
/// Uses the credentials set in -`config.toml`.
|
||||
fn get_irc_identify_messages (config: &Value)
|
||||
-> Result<VecDeque<Message>, Box<dyn std::error::Error>> {
|
||||
let mut queue = VecDeque::new();
|
||||
|
||||
queue.push_back(Message::from(
|
||||
Command::CAP(
|
||||
None,
|
||||
CapSubCommand::END,
|
||||
None,
|
||||
None
|
||||
)));
|
||||
|
||||
match config.get("password") {
|
||||
Some(p) => queue.push_back(Message::from(
|
||||
Command::PASS(
|
||||
String::from(p.as_str().ok_or("Could not parse password.")?)
|
||||
)
|
||||
)),
|
||||
None => ()
|
||||
}
|
||||
|
||||
let nick: String;
|
||||
match config.get("nickname") {
|
||||
Some(n) => {
|
||||
nick = String::from(n.as_str().ok_or("Could not parse nickname.")?);
|
||||
queue.push_back(Message::from(Command::NICK(nick.clone())));
|
||||
},
|
||||
None => return Err("No nickname supplied!".into()),
|
||||
}
|
||||
|
||||
match config.get("username") {
|
||||
Some(u) => queue.push_back(Message::from(
|
||||
Command::USER(
|
||||
String::from(u.as_str().ok_or("Could not parse username.")?),
|
||||
"0".to_owned(),
|
||||
String::from(u.as_str().ok_or("Could not parse username.")?)
|
||||
)
|
||||
)),
|
||||
None => queue.push_back(Message::from(
|
||||
// nick.clone() is only used once because the value
|
||||
// can be moved the second time
|
||||
Command::USER(nick.clone(), "0".to_owned(), nick)
|
||||
))
|
||||
}
|
||||
Ok(queue)
|
||||
}
|
||||
|
||||
/// Appends a given VecDeque to include the messages
|
||||
/// used to join the channels defined in `config.toml`.
|
||||
fn get_irc_join_messages(config: &Value, queue: VecDeque<Message>)
|
||||
-> Result<VecDeque<Message>, Box<dyn std::error::Error>> {
|
||||
let mut queue = queue;
|
||||
|
||||
match config.get("channels") {
|
||||
Some(c) => {
|
||||
for channel in c.as_array().ok_or("Could not parse channels.")? {
|
||||
queue.push_back(Message::from(
|
||||
Command::JOIN(
|
||||
String::from(channel.as_str().ok_or("Could not parse one of the channels")?),
|
||||
None,
|
||||
None
|
||||
)
|
||||
))
|
||||
}
|
||||
},
|
||||
None => ()
|
||||
}
|
||||
|
||||
Ok(queue)
|
||||
}
|
||||
|
||||
fn handle_message(message: Message, nick: &String,
|
||||
client_names: Keys, contexts: HashMap<String, RingBuffer<String>>)
|
||||
-> (Option<Message>, Option<[String; 4]>) {
|
||||
let sender = match message.clone().source_nickname() {
|
||||
Some(s) => String::from(s),
|
||||
None => String::from("anonymous")
|
||||
};
|
||||
match message.clone().command {
|
||||
Command::PING(ref data, _) => {
|
||||
return (
|
||||
Some(Message::from(
|
||||
Command::PONG(data.to_owned(), None)
|
||||
)),
|
||||
None
|
||||
);
|
||||
},
|
||||
Command::PRIVMSG(ref rec, ref msg) => {
|
||||
// Send bot info on receiving a private message.
|
||||
if rec.as_str() == nick.as_str() {
|
||||
return (
|
||||
Some(Message::from(
|
||||
Command::PRIVMSG(
|
||||
sender.clone(),
|
||||
super::texts::get_bot_info(),
|
||||
)
|
||||
)
|
||||
),
|
||||
None
|
||||
)
|
||||
} else {
|
||||
match contexts.get_mut(sender.clone()) {
|
||||
Some(ref buf) => buf.push(
|
||||
for c_name in client_names {
|
||||
if msg.contains(c_name) {
|
||||
return (None, Some([c_name.clone(),
|
||||
rec.clone(),
|
||||
sender.clone(),
|
||||
msg.clone()
|
||||
]))
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
_ => println!("{}", message.clone().to_string())
|
||||
}
|
||||
return (None, None)
|
||||
}
|
||||
|
||||
pub fn handle_server(config: Value, tx: Sender<[String; 5]>)
|
||||
-> Result<(), Box<dyn std::error::Error>> {
|
||||
let server_name = config.get("server").unwrap().as_str()
|
||||
.ok_or("Could not get server adress from config")?;
|
||||
let nick = config.get("nickname").unwrap().as_str()
|
||||
.ok_or("Could not get nickname from config")?.to_owned();
|
||||
|
||||
let clients;
|
||||
match config.get("clients").unwrap().as_table() {
|
||||
Some(t) => clients = t.clone(),
|
||||
None => clients = Map::new(),
|
||||
}
|
||||
|
||||
let mut chat_context = HashMap::new();
|
||||
|
||||
let mut stream = super::connect::connect_irc(&config).unwrap();
|
||||
let mut message_queue = get_irc_identify_messages(&config).unwrap();
|
||||
|
||||
while let Some(message) = message_queue.pop_front() {
|
||||
stream.write(message).unwrap();
|
||||
}
|
||||
|
||||
message_queue = get_irc_join_messages(&config, message_queue).unwrap();
|
||||
|
||||
// Wait for first ping and join channels after sending pong.
|
||||
loop {
|
||||
let message = stream.read()?;
|
||||
match message.command {
|
||||
Command::PING(ref data, _) => {
|
||||
stream.write(Message::from(
|
||||
Command::PONG(data.to_owned(), None)
|
||||
)).unwrap();
|
||||
while let Some(message) = message_queue.pop_front() {
|
||||
stream.write(message).unwrap();
|
||||
}
|
||||
break;
|
||||
}
|
||||
_ => ()
|
||||
}
|
||||
}
|
||||
|
||||
// Handle all incoming messages after joining the channels.
|
||||
loop {
|
||||
let message = stream.read()?;
|
||||
let (answer, data) = handle_message(message, &nick, clients.keys());
|
||||
match answer {
|
||||
Some(a) => stream.write(a).unwrap(),
|
||||
None => ()
|
||||
}
|
||||
match data {
|
||||
Some(d) => match clients.get(&d[0]) {
|
||||
Some(addr) => {
|
||||
// There must be a better way to do this…
|
||||
let out = [
|
||||
addr.as_str()
|
||||
.ok_or("Could not parse email address.")?
|
||||
.to_owned(),
|
||||
server_name.to_owned(),
|
||||
d[1].clone(),
|
||||
d[2].clone(),
|
||||
d[3].clone(),
|
||||
];
|
||||
tx.send(out)?
|
||||
},
|
||||
None => (),
|
||||
},
|
||||
None => (),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in a new issue