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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
//! Requires the 'framework' feature flag be enabled in your project's
//! `Cargo.toml`.
//!
//! This can be enabled by specifying the feature in the dependency section:
//!
//! ```toml
//! [dependencies.serenity]
//! git = "https://github.com/serenity-rs/serenity.git"
//! features = ["framework", "standard_framework"]
//! ```
mod commands;
use std::{
collections::HashSet,
env,
sync::Arc,
};
use serenity::{
client::bridge::gateway::ShardManager,
framework::{
StandardFramework,
standard::macros::group,
},
model::{event::ResumedEvent, gateway::Ready},
prelude::*,
};
use log::{error, info};
use commands::{
math::*,
meta::*,
owner::*,
};
struct ShardManagerContainer;
impl TypeMapKey for ShardManagerContainer {
type Value = Arc<Mutex<ShardManager>>;
}
struct Handler;
impl EventHandler for Handler {
fn ready(&self, _: Context, ready: Ready) {
info!("Connected as {}", ready.user.name);
}
fn resume(&self, _: Context, _: ResumedEvent) {
info!("Resumed");
}
}
#[group]
#[commands(multiply, ping, quit)]
struct General;
fn main() {
// This will load the environment variables located at `./.env`, relative to
// the CWD. See `./.env.example` for an example on how to structure this.
kankyo::load().expect("Failed to load .env file");
// Initialize the logger to use environment variables.
//
// In this case, a good default is setting the environment variable
// `RUST_LOG` to debug`.
env_logger::init();
let token = env::var("DISCORD_TOKEN")
.expect("Expected a token in the environment");
let mut client = Client::new(&token, Handler).expect("Err creating client");
{
let mut data = client.data.write();
data.insert::<ShardManagerContainer>(Arc::clone(&client.shard_manager));
}
let owners = match client.cache_and_http.http.get_current_application_info() {
Ok(info) => {
let mut set = HashSet::new();
set.insert(info.owner.id);
set
},
Err(why) => panic!("Couldn't get application info: {:?}", why),
};
client.with_framework(StandardFramework::new()
.configure(|c| c
.owners(owners)
.prefix("~"))
.group(&GENERAL_GROUP));
if let Err(why) = client.start() {
error!("Client error: {:?}", why);
}
}
|
