forked from Skynet/discord-bot
72 lines
2 KiB
Rust
72 lines
2 KiB
Rust
|
use serenity::all::{ChunkGuildFilter, GuildId};
|
||
|
|
use serenity::{
|
||
|
|
async_trait,
|
||
|
|
client::{Context, EventHandler},
|
||
|
|
model::gateway::{GatewayIntents, Ready},
|
||
|
|
Client,
|
||
|
|
};
|
||
|
|
use skynet_discord_bot::common::server_icon::{get_config_icons, update_icon};
|
||
|
|
use skynet_discord_bot::{
|
||
|
|
common::database::{db_init, DataBase},
|
||
|
|
get_config, Config,
|
||
|
|
};
|
||
|
|
use std::{process, sync::Arc};
|
||
|
|
use tokio::sync::RwLock;
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn main() {
|
||
|
|
let config = get_config();
|
||
|
|
let db = match db_init(&config).await {
|
||
|
|
Ok(x) => x,
|
||
|
|
Err(_) => return,
|
||
|
|
};
|
||
|
|
|
||
|
|
// Intents are a bitflag, bitwise operations can be used to dictate which intents to use
|
||
|
|
let intents = GatewayIntents::GUILDS | GatewayIntents::GUILD_MESSAGES | GatewayIntents::MESSAGE_CONTENT | GatewayIntents::GUILD_MEMBERS;
|
||
|
|
// Build our client.
|
||
|
|
let mut client = Client::builder(&config.discord_token, intents)
|
||
|
|
.event_handler(Handler {})
|
||
|
|
.cache_settings(serenity::cache::Settings::default())
|
||
|
|
.await
|
||
|
|
.expect("Error creating client");
|
||
|
|
|
||
|
|
{
|
||
|
|
let mut data = client.data.write().await;
|
||
|
|
|
||
|
|
data.insert::<Config>(Arc::new(RwLock::new(config)));
|
||
|
|
data.insert::<DataBase>(Arc::new(RwLock::new(db)));
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Err(why) = client.start().await {
|
||
|
|
println!("Client error: {:?}", why);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
struct Handler;
|
||
|
|
#[async_trait]
|
||
|
|
impl EventHandler for Handler {
|
||
|
|
async fn ready(&self, ctx: Context, ready: Ready) {
|
||
|
|
let ctx = Arc::new(ctx);
|
||
|
|
println!("{} is connected!", ready.user.name);
|
||
|
|
|
||
|
|
let db_lock = {
|
||
|
|
let data_read = ctx.data.read().await;
|
||
|
|
data_read.get::<DataBase>().expect("Expected Config in TypeMap.").clone()
|
||
|
|
};
|
||
|
|
let db = db_lock.read().await;
|
||
|
|
|
||
|
|
let config_lock = {
|
||
|
|
let data_read = ctx.data.read().await;
|
||
|
|
data_read.get::<Config>().expect("Expected Config in TypeMap.").clone()
|
||
|
|
};
|
||
|
|
|
||
|
|
let config_global = config_lock.read().await;
|
||
|
|
let config_toml = get_config_icons::minimal();
|
||
|
|
|
||
|
|
update_icon::update_icon_main(&ctx, &db, &config_global, &config_toml).await;
|
||
|
|
|
||
|
|
// finish up
|
||
|
|
process::exit(0);
|
||
|
|
}
|
||
|
|
}
|