diff options
Diffstat (limited to 'src/settings.rs')
| -rw-r--r-- | src/settings.rs | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/src/settings.rs b/src/settings.rs index f0fef31..5cf0dec 100644 --- a/src/settings.rs +++ b/src/settings.rs @@ -1,9 +1,10 @@ -use crate::utils; +use crate::{consts, utils}; use log::*; -use poise::serenity_prelude::{ChannelId, EmojiId, MessageReaction, ReactionType}; +use poise::serenity_prelude::{ChannelId, EmojiId, GuildId, MessageReaction, ReactionType}; #[derive(Clone)] pub struct Settings { + pub allowed_guilds: Vec<GuildId>, pub pinboard_target: ChannelId, pub pinboard_sources: Option<Vec<ChannelId>>, pub reactboard_target: ChannelId, @@ -14,6 +15,9 @@ pub struct Settings { impl Settings { pub fn new() -> Option<Self> { + let allowed_guilds = utils::parse_snowflakes_from_env("ALLOWED_GUILDS", GuildId) + .unwrap_or_else(|| vec![consts::TEAWIE_GUILD, GuildId(1091969030694375444)]); + let Some(pinboard_target) = utils::parse_snowflake_from_env("PIN_BOARD_TARGET", ChannelId) else { return None; @@ -56,6 +60,7 @@ impl Settings { ); Some(Self { + allowed_guilds, pinboard_target, pinboard_sources, reactboard_target, @@ -77,4 +82,8 @@ impl Settings { _ => false, } } + + pub fn is_guild_allowed(&self, gid: GuildId) -> bool { + self.allowed_guilds.contains(&gid) + } } |