topgg 1.4.2

The official Rust wrapper for the Top.gg API
Documentation 
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

use crate::autoposter::{Handler, SharedStats};
use std::{collections::HashSet, ops::DerefMut};
use tokio::sync::Mutex;
use twilight_model::gateway::event::Event;

/// A built-in [`Handler`] for the [twilight](https://twilight.rs) library.
pub struct Twilight {
  cache: Mutex<HashSet<u64>>,
  stats: SharedStats,
}

impl Twilight {
  #[inline(always)]
  pub(super) fn new() -> Self {
    Self {
      cache: Mutex::const_new(HashSet::new()),
      stats: SharedStats::new(),
    }
  }

  /// Handles an entire [twilight](https://twilight.rs) [`Event`] enum.
  pub async fn handle(&self, event: &Event) {
    match event {
      Event::Ready(ready) => {
        let mut cache = self.cache.lock().await;
        let mut stats = self.stats.write().await;
        let cache_ref = cache.deref_mut();

        *cache_ref = ready.guilds.iter().map(|guild| guild.id.get()).collect();
        stats.set_server_count(cache.len());
      }

      Event::GuildCreate(guild_create) => {
        let mut cache = self.cache.lock().await;

        if cache.insert(guild_create.0.id.get()) {
          let mut stats = self.stats.write().await;

          stats.set_server_count(cache.len());
        }
      }

      Event::GuildDelete(guild_delete) => {
        let mut cache = self.cache.lock().await;

        if cache.remove(&guild_delete.id.get()) {
          let mut stats = self.stats.write().await;

          stats.set_server_count(cache.len());
        }
      }

      _ => {}
    }
  }
}

impl Handler for Twilight {
  #[inline(always)]
  fn stats(&self) -> &SharedStats {
    &self.stats
  }
}