aboutsummaryrefslogtreecommitdiff
path: root/src/hooks/mod.rs
blob: 39198c18a4c28ebeec0ec1bce3a9a0457041fe4b (plain)
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
//! The bots hooks, commands and matchers. For explanation of different types see [crate::catinator]
//!
//! # Implementing hooks

use anyhow::Result;
use base64::{alphabet, engine, Engine};
use irc::client::prelude::*;

mod intensify;
mod nitter;
mod pet;
mod shifty_eyes;

pub use intensify::*;
pub use nitter::*;
pub use pet::*;
pub use shifty_eyes::*;

pub mod sed;
pub mod wolfram_alpha;

/// Replies with some information about the bot
pub fn about(bot: &crate::Bot, msg: Message) -> Result<()> {
    bot.send_privmsg(
        msg.response_target().unwrap(),
        &format!(
            "{name} is {name} - https://gitlab.com/cocainefarm/gnulag/catinator",
            name = bot.config.user.nickname
        )
        .to_string(),
    )
    .unwrap();

    Ok(())
}

const ENGINE: engine::GeneralPurpose =
    engine::GeneralPurpose::new(&alphabet::URL_SAFE, engine::general_purpose::NO_PAD);

/// Listen to AUTHENTICATE messages and perform SASL authentication
pub fn sasl(bot: &crate::Bot, msg: Message) -> Result<()> {
    match msg.command {
        Command::AUTHENTICATE(text) => {
            use sasl::client::mechanisms::Plain;
            use sasl::client::Mechanism;
            use sasl::common::Credentials;

            if text == "+" {
                let creds = Credentials::default()
                    .with_username(bot.config.clone().user.username)
                    .with_password(bot.config.clone().user.password.unwrap());

                let mut mechanism = Plain::from_credentials(creds)?;

                let initial_data = mechanism.initial();

                bot.irc_client.send_sasl(ENGINE.encode(initial_data))?;
                bot.irc_client.send(Command::CAP(
                    None,
                    irc_proto::command::CapSubCommand::END,
                    None,
                    None,
                ))?;
            }
        }
        _ => (),
    }

    Ok(())
}