summary refs log tree commit diff
path: root/src/system/types.rs
blob: c4da066a26df9aef84c76c02a2ac5aced31d7ac2 (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
use regex::Regex;
use twilight_http::Client;
use twilight_gateway::Shard;
use twilight_mention::ParseMention;
use twilight_model::id::{marker::UserMarker, Id};
use std::sync::Arc;
use tokio::sync::Mutex;

#[derive(Clone)]
pub struct Member {
    pub name: String,
    pub discord_token: String,
    pub user_id: Arc<Mutex<Option<Id<UserMarker>>>>,
    pub message_pattern: Regex,
    pub shard: Arc<Mutex<Shard>>,
    pub client: Arc<Mutex<Client>>,
}

#[derive(Clone)]
pub struct System {
    pub followed_user: Id<UserMarker>,
    pub command_prefix: String,
    pub members: Vec<Member>
}

#[derive(Clone)]
pub enum Response {
    Proxy {member: Member, content: String},
    Noop {delete_source: bool},
}

pub enum SystemThreadCommand {
    Restart,
    ReloadConfig,
    ShutdownSystem,
    ShutdownAll,
}

impl System {
    pub async fn resolve_mention<'system>(&'system self, maybe_mention: Option<&str>) -> Option<&'system Member> {
        if let Some(mention) = maybe_mention {
            if let Ok(mention) = Id::<UserMarker>::parse(mention) {
                for member in &self.members {
                    let is_member = {member.user_id.lock().await.map(|id| id == mention).unwrap_or(false)};

                    if is_member {
                        return Some(&member)
                    }
                }
            }
        }

        None
    }
}