turned models into module
This commit is contained in:
parent
ea2b0f4b0a
commit
1556318d07
@ -1,18 +1,22 @@
|
|||||||
use regex_command_attr::command;
|
use regex_command_attr::command;
|
||||||
|
|
||||||
use serenity::{client::Context, model::channel::Message};
|
use serenity::{builder::CreateEmbedFooter, client::Context, model::channel::Message};
|
||||||
|
|
||||||
use chrono::offset::Utc;
|
use chrono::offset::Utc;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
command_help, consts::DEFAULT_PREFIX, get_ctx_data, language_manager::LanguageManager,
|
command_help,
|
||||||
models::UserData, FrameworkCtx, THEME_COLOR,
|
consts::DEFAULT_PREFIX,
|
||||||
|
get_ctx_data,
|
||||||
|
language_manager::LanguageManager,
|
||||||
|
models::{user_data::UserData, CtxGuildData},
|
||||||
|
FrameworkCtx, THEME_COLOR,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::models::CtxGuildData;
|
use std::{
|
||||||
use serenity::builder::CreateEmbedFooter;
|
sync::Arc,
|
||||||
use std::sync::Arc;
|
time::{SystemTime, UNIX_EPOCH},
|
||||||
use std::time::{SystemTime, UNIX_EPOCH};
|
};
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
#[can_blacklist(false)]
|
#[can_blacklist(false)]
|
||||||
@ -202,7 +206,6 @@ async fn clock(ctx: &Context, msg: &Message, _args: String) {
|
|||||||
|
|
||||||
let language = UserData::language_of(&msg.author, &pool).await;
|
let language = UserData::language_of(&msg.author, &pool).await;
|
||||||
let timezone = UserData::timezone_of(&msg.author, &pool).await;
|
let timezone = UserData::timezone_of(&msg.author, &pool).await;
|
||||||
let meridian = UserData::meridian_of(&msg.author, &pool).await;
|
|
||||||
|
|
||||||
let now = Utc::now().with_timezone(&timezone);
|
let now = Utc::now().with_timezone(&timezone);
|
||||||
|
|
||||||
@ -212,7 +215,7 @@ async fn clock(ctx: &Context, msg: &Message, _args: String) {
|
|||||||
.channel_id
|
.channel_id
|
||||||
.say(
|
.say(
|
||||||
&ctx,
|
&ctx,
|
||||||
clock_display.replacen("{}", &now.format(meridian.fmt_str()).to_string(), 1),
|
clock_display.replacen("{}", &now.format("%H:%M").to_string(), 1),
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
|
@ -24,11 +24,10 @@ use crate::{
|
|||||||
consts::{REGEX_ALIAS, REGEX_CHANNEL, REGEX_COMMANDS, REGEX_ROLE, THEME_COLOR},
|
consts::{REGEX_ALIAS, REGEX_CHANNEL, REGEX_COMMANDS, REGEX_ROLE, THEME_COLOR},
|
||||||
framework::SendIterator,
|
framework::SendIterator,
|
||||||
get_ctx_data,
|
get_ctx_data,
|
||||||
models::{ChannelData, GuildData, UserData},
|
models::{channel_data::ChannelData, guild_data::GuildData, user_data::UserData, CtxGuildData},
|
||||||
FrameworkCtx, PopularTimezones,
|
FrameworkCtx, PopularTimezones,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::models::CtxGuildData;
|
|
||||||
use std::{collections::HashMap, iter};
|
use std::{collections::HashMap, iter};
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
@ -113,11 +112,7 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) {
|
|||||||
let content = lm
|
let content = lm
|
||||||
.get(&user_data.language, "timezone/set_p")
|
.get(&user_data.language, "timezone/set_p")
|
||||||
.replacen("{timezone}", &user_data.timezone, 1)
|
.replacen("{timezone}", &user_data.timezone, 1)
|
||||||
.replacen(
|
.replacen("{time}", &now.format("%H:%M").to_string(), 1);
|
||||||
"{time}",
|
|
||||||
&now.format(user_data.meridian().fmt_str_short()).to_string(),
|
|
||||||
1,
|
|
||||||
);
|
|
||||||
|
|
||||||
let _ =
|
let _ =
|
||||||
msg.channel_id
|
msg.channel_id
|
||||||
@ -154,10 +149,7 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) {
|
|||||||
tz.to_string(),
|
tz.to_string(),
|
||||||
format!(
|
format!(
|
||||||
"🕗 `{}`",
|
"🕗 `{}`",
|
||||||
Utc::now()
|
Utc::now().with_timezone(tz).format("%H:%M").to_string()
|
||||||
.with_timezone(tz)
|
|
||||||
.format(user_data.meridian().fmt_str_short())
|
|
||||||
.to_string()
|
|
||||||
),
|
),
|
||||||
true,
|
true,
|
||||||
)
|
)
|
||||||
@ -211,10 +203,7 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) {
|
|||||||
t.to_string(),
|
t.to_string(),
|
||||||
format!(
|
format!(
|
||||||
"🕗 `{}`",
|
"🕗 `{}`",
|
||||||
Utc::now()
|
Utc::now().with_timezone(t).format("%H:%M").to_string()
|
||||||
.with_timezone(t)
|
|
||||||
.format(user_data.meridian().fmt_str_short())
|
|
||||||
.to_string()
|
|
||||||
),
|
),
|
||||||
true,
|
true,
|
||||||
)
|
)
|
||||||
@ -252,49 +241,6 @@ async fn timezone(ctx: &Context, msg: &Message, args: String) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[command("meridian")]
|
|
||||||
async fn change_meridian(ctx: &Context, msg: &Message, args: String) {
|
|
||||||
let (pool, lm) = get_ctx_data(&ctx).await;
|
|
||||||
|
|
||||||
let mut user_data = UserData::from_user(&msg.author, &ctx, &pool).await.unwrap();
|
|
||||||
|
|
||||||
if &args == "12" {
|
|
||||||
user_data.meridian_time = true;
|
|
||||||
|
|
||||||
user_data.commit_changes(&pool).await;
|
|
||||||
|
|
||||||
let _ = msg
|
|
||||||
.channel_id
|
|
||||||
.send_message(&ctx, |m| {
|
|
||||||
m.embed(|e| {
|
|
||||||
e.title(lm.get(&user_data.language, "meridian/title"))
|
|
||||||
.color(*THEME_COLOR)
|
|
||||||
.description(lm.get(&user_data.language, "meridian/12"))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
.await;
|
|
||||||
} else if &args == "24" {
|
|
||||||
user_data.meridian_time = false;
|
|
||||||
|
|
||||||
user_data.commit_changes(&pool).await;
|
|
||||||
|
|
||||||
let _ = msg
|
|
||||||
.channel_id
|
|
||||||
.send_message(&ctx, |m| {
|
|
||||||
m.embed(|e| {
|
|
||||||
e.title(lm.get(&user_data.language, "meridian/title"))
|
|
||||||
.color(*THEME_COLOR)
|
|
||||||
.description(lm.get(&user_data.language, "meridian/24"))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
.await;
|
|
||||||
} else {
|
|
||||||
let prefix = ctx.prefix(msg.guild_id).await;
|
|
||||||
|
|
||||||
command_help(ctx, msg, lm, &prefix, &user_data.language, "meridian").await;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command("lang")]
|
#[command("lang")]
|
||||||
async fn language(ctx: &Context, msg: &Message, args: String) {
|
async fn language(ctx: &Context, msg: &Message, args: String) {
|
||||||
let (pool, lm) = get_ctx_data(&ctx).await;
|
let (pool, lm) = get_ctx_data(&ctx).await;
|
||||||
|
@ -24,11 +24,14 @@ use crate::{
|
|||||||
},
|
},
|
||||||
framework::SendIterator,
|
framework::SendIterator,
|
||||||
get_ctx_data,
|
get_ctx_data,
|
||||||
models::{ChannelData, CtxGuildData, GuildData, Timer, UserData},
|
models::{
|
||||||
|
channel_data::ChannelData, guild_data::GuildData, timer::Timer, user_data::UserData,
|
||||||
|
CtxGuildData,
|
||||||
|
},
|
||||||
time_parser::{natural_parser, TimeParser},
|
time_parser::{natural_parser, TimeParser},
|
||||||
};
|
};
|
||||||
|
|
||||||
use chrono::{offset::TimeZone, NaiveDateTime};
|
use chrono::NaiveDateTime;
|
||||||
|
|
||||||
use rand::{rngs::OsRng, seq::IteratorRandom};
|
use rand::{rngs::OsRng, seq::IteratorRandom};
|
||||||
|
|
||||||
@ -136,7 +139,6 @@ async fn pause(ctx: &Context, msg: &Message, args: String) {
|
|||||||
|
|
||||||
let language = UserData::language_of(&msg.author, &pool).await;
|
let language = UserData::language_of(&msg.author, &pool).await;
|
||||||
let timezone = UserData::timezone_of(&msg.author, &pool).await;
|
let timezone = UserData::timezone_of(&msg.author, &pool).await;
|
||||||
let meridian = UserData::meridian_of(&msg.author, &pool).await;
|
|
||||||
|
|
||||||
let mut channel = ChannelData::from_channel(msg.channel(&ctx).await.unwrap(), &pool)
|
let mut channel = ChannelData::from_channel(msg.channel(&ctx).await.unwrap(), &pool)
|
||||||
.await
|
.await
|
||||||
@ -172,13 +174,9 @@ async fn pause(ctx: &Context, msg: &Message, args: String) {
|
|||||||
|
|
||||||
channel.commit_changes(&pool).await;
|
channel.commit_changes(&pool).await;
|
||||||
|
|
||||||
let content = lm.get(&language, "pause/paused_until").replace(
|
let content = lm
|
||||||
"{}",
|
.get(&language, "pause/paused_until")
|
||||||
&timezone
|
.replace("{}", &format!("<t:{}:D>", timestamp));
|
||||||
.timestamp(timestamp, 0)
|
|
||||||
.format(meridian.fmt_str())
|
|
||||||
.to_string(),
|
|
||||||
);
|
|
||||||
|
|
||||||
let _ = msg.channel_id.say(&ctx, content).await;
|
let _ = msg.channel_id.say(&ctx, content).await;
|
||||||
}
|
}
|
||||||
@ -864,7 +862,6 @@ impl ReminderScope {
|
|||||||
|
|
||||||
#[derive(PartialEq, Eq, Hash, Debug)]
|
#[derive(PartialEq, Eq, Hash, Debug)]
|
||||||
enum ReminderError {
|
enum ReminderError {
|
||||||
LongTime,
|
|
||||||
LongInterval,
|
LongInterval,
|
||||||
PastTime,
|
PastTime,
|
||||||
ShortInterval,
|
ShortInterval,
|
||||||
@ -891,7 +888,6 @@ trait ToResponse {
|
|||||||
impl ToResponse for ReminderError {
|
impl ToResponse for ReminderError {
|
||||||
fn to_response(&self) -> &'static str {
|
fn to_response(&self) -> &'static str {
|
||||||
match self {
|
match self {
|
||||||
Self::LongTime => "remind/long_time",
|
|
||||||
Self::LongInterval => "interval/long_interval",
|
Self::LongInterval => "interval/long_interval",
|
||||||
Self::PastTime => "remind/past_time",
|
Self::PastTime => "remind/past_time",
|
||||||
Self::ShortInterval => "interval/short_interval",
|
Self::ShortInterval => "interval/short_interval",
|
||||||
@ -904,7 +900,6 @@ impl ToResponse for ReminderError {
|
|||||||
|
|
||||||
fn to_response_natural(&self) -> &'static str {
|
fn to_response_natural(&self) -> &'static str {
|
||||||
match self {
|
match self {
|
||||||
Self::LongTime => "natural/long_time",
|
|
||||||
Self::InvalidTime => "natural/invalid_time",
|
Self::InvalidTime => "natural/invalid_time",
|
||||||
_ => self.to_response(),
|
_ => self.to_response(),
|
||||||
}
|
}
|
||||||
@ -1601,7 +1596,7 @@ async fn create_reminder<'a, U: Into<u64>, T: TryInto<i64>>(
|
|||||||
expires_parser: Option<T>,
|
expires_parser: Option<T>,
|
||||||
interval: Option<i64>,
|
interval: Option<i64>,
|
||||||
content: &mut Content,
|
content: &mut Content,
|
||||||
) -> Result<(), ReminderError> {
|
) -> Result<String, ReminderError> {
|
||||||
let user_id = user_id.into();
|
let user_id = user_id.into();
|
||||||
|
|
||||||
if let Some(g_id) = guild_id {
|
if let Some(g_id) = guild_id {
|
||||||
@ -1681,9 +1676,8 @@ async fn create_reminder<'a, U: Into<u64>, T: TryInto<i64>>(
|
|||||||
.as_secs() as i64;
|
.as_secs() as i64;
|
||||||
|
|
||||||
if time >= unix_time - 10 {
|
if time >= unix_time - 10 {
|
||||||
if time > unix_time + *MAX_TIME {
|
let uid = generate_uid();
|
||||||
Err(ReminderError::LongTime)
|
|
||||||
} else {
|
|
||||||
sqlx::query!(
|
sqlx::query!(
|
||||||
"
|
"
|
||||||
INSERT INTO reminders (
|
INSERT INTO reminders (
|
||||||
@ -1710,13 +1704,13 @@ INSERT INTO reminders (
|
|||||||
(SELECT id FROM users WHERE user = ? LIMIT 1)
|
(SELECT id FROM users WHERE user = ? LIMIT 1)
|
||||||
)
|
)
|
||||||
",
|
",
|
||||||
generate_uid(),
|
uid,
|
||||||
content.content,
|
content.content,
|
||||||
content.tts,
|
content.tts,
|
||||||
content.attachment,
|
content.attachment,
|
||||||
content.attachment_name,
|
content.attachment_name,
|
||||||
db_channel_id,
|
db_channel_id,
|
||||||
time as u32,
|
time,
|
||||||
expires,
|
expires,
|
||||||
interval,
|
interval,
|
||||||
user_id
|
user_id
|
||||||
@ -1725,8 +1719,7 @@ INSERT INTO reminders (
|
|||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
Ok(())
|
Ok(uid)
|
||||||
}
|
|
||||||
} else if time < 0 {
|
} else if time < 0 {
|
||||||
// case required for if python returns -1
|
// case required for if python returns -1
|
||||||
Err(ReminderError::InvalidTime)
|
Err(ReminderError::InvalidTime)
|
||||||
|
@ -12,8 +12,10 @@ use serenity::{
|
|||||||
|
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
|
|
||||||
use crate::models::CtxGuildData;
|
use crate::{
|
||||||
use crate::{command_help, get_ctx_data, models::UserData};
|
command_help, get_ctx_data,
|
||||||
|
models::{user_data::UserData, CtxGuildData},
|
||||||
|
};
|
||||||
use sqlx::MySqlPool;
|
use sqlx::MySqlPool;
|
||||||
use std::convert::TryFrom;
|
use std::convert::TryFrom;
|
||||||
|
|
||||||
|
@ -8,7 +8,7 @@ use serenity::{
|
|||||||
model::{
|
model::{
|
||||||
channel::{Channel, GuildChannel, Message},
|
channel::{Channel, GuildChannel, Message},
|
||||||
guild::{Guild, Member},
|
guild::{Guild, Member},
|
||||||
id::ChannelId,
|
id::{ChannelId, MessageId},
|
||||||
},
|
},
|
||||||
Result as SerenityResult,
|
Result as SerenityResult,
|
||||||
};
|
};
|
||||||
@ -19,10 +19,11 @@ use regex::{Match, Regex, RegexBuilder};
|
|||||||
|
|
||||||
use std::{collections::HashMap, fmt};
|
use std::{collections::HashMap, fmt};
|
||||||
|
|
||||||
use crate::language_manager::LanguageManager;
|
use crate::{
|
||||||
use crate::models::{CtxGuildData, GuildData, UserData};
|
language_manager::LanguageManager,
|
||||||
use crate::{models::ChannelData, LimitExecutors, SQLPool};
|
models::{channel_data::ChannelData, guild_data::GuildData, user_data::UserData, CtxGuildData},
|
||||||
use serenity::model::id::MessageId;
|
LimitExecutors, SQLPool,
|
||||||
|
};
|
||||||
|
|
||||||
type CommandFn = for<'fut> fn(&'fut Context, &'fut Message, String) -> BoxFuture<'fut, ()>;
|
type CommandFn = for<'fut> fn(&'fut Context, &'fut Message, String) -> BoxFuture<'fut, ()>;
|
||||||
|
|
||||||
|
@ -36,7 +36,7 @@ use crate::{
|
|||||||
consts::{CNC_GUILD, DEFAULT_PREFIX, SUBSCRIPTION_ROLES, THEME_COLOR},
|
consts::{CNC_GUILD, DEFAULT_PREFIX, SUBSCRIPTION_ROLES, THEME_COLOR},
|
||||||
framework::RegexFramework,
|
framework::RegexFramework,
|
||||||
language_manager::LanguageManager,
|
language_manager::LanguageManager,
|
||||||
models::GuildData,
|
models::{guild_data::GuildData, user_data::UserData},
|
||||||
};
|
};
|
||||||
|
|
||||||
use inflector::Inflector;
|
use inflector::Inflector;
|
||||||
@ -46,7 +46,6 @@ use dashmap::DashMap;
|
|||||||
|
|
||||||
use tokio::sync::RwLock;
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
use crate::models::UserData;
|
|
||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
use chrono_tz::Tz;
|
use chrono_tz::Tz;
|
||||||
use serenity::model::prelude::{
|
use serenity::model::prelude::{
|
||||||
@ -294,7 +293,7 @@ DELETE FROM guilds WHERE guild = ?
|
|||||||
.replacen("{timezone}", &user_data.timezone, 1)
|
.replacen("{timezone}", &user_data.timezone, 1)
|
||||||
.replacen(
|
.replacen(
|
||||||
"{time}",
|
"{time}",
|
||||||
&now.format(user_data.meridian().fmt_str_short()).to_string(),
|
&now.format("%H:%M").to_string(),
|
||||||
1,
|
1,
|
||||||
);
|
);
|
||||||
|
|
||||||
@ -399,7 +398,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|||||||
.add_command("blacklist", &moderation_cmds::BLACKLIST_COMMAND)
|
.add_command("blacklist", &moderation_cmds::BLACKLIST_COMMAND)
|
||||||
.add_command("restrict", &moderation_cmds::RESTRICT_COMMAND)
|
.add_command("restrict", &moderation_cmds::RESTRICT_COMMAND)
|
||||||
.add_command("timezone", &moderation_cmds::TIMEZONE_COMMAND)
|
.add_command("timezone", &moderation_cmds::TIMEZONE_COMMAND)
|
||||||
.add_command("meridian", &moderation_cmds::CHANGE_MERIDIAN_COMMAND)
|
|
||||||
.add_command("prefix", &moderation_cmds::PREFIX_COMMAND)
|
.add_command("prefix", &moderation_cmds::PREFIX_COMMAND)
|
||||||
.add_command("lang", &moderation_cmds::LANGUAGE_COMMAND)
|
.add_command("lang", &moderation_cmds::LANGUAGE_COMMAND)
|
||||||
.add_command("pause", &reminder_cmds::PAUSE_COMMAND)
|
.add_command("pause", &reminder_cmds::PAUSE_COMMAND)
|
||||||
|
452
src/models.rs
452
src/models.rs
@ -1,452 +0,0 @@
|
|||||||
use serenity::{
|
|
||||||
async_trait,
|
|
||||||
http::CacheHttp,
|
|
||||||
model::{
|
|
||||||
channel::Channel,
|
|
||||||
guild::Guild,
|
|
||||||
id::{GuildId, UserId},
|
|
||||||
user::User,
|
|
||||||
},
|
|
||||||
prelude::Context,
|
|
||||||
};
|
|
||||||
|
|
||||||
use sqlx::MySqlPool;
|
|
||||||
|
|
||||||
use chrono::NaiveDateTime;
|
|
||||||
use chrono_tz::Tz;
|
|
||||||
|
|
||||||
use log::error;
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
consts::{DEFAULT_PREFIX, LOCAL_LANGUAGE, LOCAL_TIMEZONE},
|
|
||||||
GuildDataCache, SQLPool,
|
|
||||||
};
|
|
||||||
|
|
||||||
use std::sync::Arc;
|
|
||||||
use tokio::sync::RwLock;
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
pub trait CtxGuildData {
|
|
||||||
async fn guild_data<G: Into<GuildId> + Send + Sync>(
|
|
||||||
&self,
|
|
||||||
guild_id: G,
|
|
||||||
) -> Result<Arc<RwLock<GuildData>>, sqlx::Error>;
|
|
||||||
|
|
||||||
async fn prefix<G: Into<GuildId> + Send + Sync>(&self, guild_id: Option<G>) -> String;
|
|
||||||
}
|
|
||||||
|
|
||||||
#[async_trait]
|
|
||||||
impl CtxGuildData for Context {
|
|
||||||
async fn guild_data<G: Into<GuildId> + Send + Sync>(
|
|
||||||
&self,
|
|
||||||
guild_id: G,
|
|
||||||
) -> Result<Arc<RwLock<GuildData>>, sqlx::Error> {
|
|
||||||
let guild_id = guild_id.into();
|
|
||||||
|
|
||||||
let guild = guild_id.to_guild_cached(&self.cache).await.unwrap();
|
|
||||||
|
|
||||||
let guild_cache = self
|
|
||||||
.data
|
|
||||||
.read()
|
|
||||||
.await
|
|
||||||
.get::<GuildDataCache>()
|
|
||||||
.cloned()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let x = if let Some(guild_data) = guild_cache.get(&guild_id) {
|
|
||||||
Ok(guild_data.clone())
|
|
||||||
} else {
|
|
||||||
let pool = self.data.read().await.get::<SQLPool>().cloned().unwrap();
|
|
||||||
|
|
||||||
match GuildData::from_guild(guild, &pool).await {
|
|
||||||
Ok(d) => {
|
|
||||||
let lock = Arc::new(RwLock::new(d));
|
|
||||||
|
|
||||||
guild_cache.insert(guild_id, lock.clone());
|
|
||||||
|
|
||||||
Ok(lock)
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(e) => Err(e),
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
x
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn prefix<G: Into<GuildId> + Send + Sync>(&self, guild_id: Option<G>) -> String {
|
|
||||||
if let Some(guild_id) = guild_id {
|
|
||||||
self.guild_data(guild_id)
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
.read()
|
|
||||||
.await
|
|
||||||
.prefix
|
|
||||||
.clone()
|
|
||||||
} else {
|
|
||||||
DEFAULT_PREFIX.clone()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct GuildData {
|
|
||||||
pub id: u32,
|
|
||||||
pub name: Option<String>,
|
|
||||||
pub prefix: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl GuildData {
|
|
||||||
pub async fn from_guild(guild: Guild, pool: &MySqlPool) -> Result<Self, sqlx::Error> {
|
|
||||||
let guild_id = guild.id.as_u64().to_owned();
|
|
||||||
|
|
||||||
match sqlx::query_as!(
|
|
||||||
Self,
|
|
||||||
"
|
|
||||||
SELECT id, name, prefix FROM guilds WHERE guild = ?
|
|
||||||
",
|
|
||||||
guild_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(mut g) => {
|
|
||||||
g.name = Some(guild.name);
|
|
||||||
|
|
||||||
Ok(g)
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(sqlx::Error::RowNotFound) => {
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT INTO guilds (guild, name, prefix) VALUES (?, ?, ?)
|
|
||||||
",
|
|
||||||
guild_id,
|
|
||||||
guild.name,
|
|
||||||
*DEFAULT_PREFIX
|
|
||||||
)
|
|
||||||
.execute(&pool.clone())
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(sqlx::query_as!(
|
|
||||||
Self,
|
|
||||||
"
|
|
||||||
SELECT id, name, prefix FROM guilds WHERE guild = ?
|
|
||||||
",
|
|
||||||
guild_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await?)
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(e) => {
|
|
||||||
error!("Unexpected error in guild query: {:?}", e);
|
|
||||||
|
|
||||||
Err(e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn commit_changes(&self, pool: &MySqlPool) {
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
UPDATE guilds SET name = ?, prefix = ? WHERE id = ?
|
|
||||||
",
|
|
||||||
self.name,
|
|
||||||
self.prefix,
|
|
||||||
self.id
|
|
||||||
)
|
|
||||||
.execute(pool)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct ChannelData {
|
|
||||||
pub id: u32,
|
|
||||||
pub name: Option<String>,
|
|
||||||
pub nudge: i16,
|
|
||||||
pub blacklisted: bool,
|
|
||||||
pub webhook_id: Option<u64>,
|
|
||||||
pub webhook_token: Option<String>,
|
|
||||||
pub paused: bool,
|
|
||||||
pub paused_until: Option<NaiveDateTime>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ChannelData {
|
|
||||||
pub async fn from_channel(
|
|
||||||
channel: Channel,
|
|
||||||
pool: &MySqlPool,
|
|
||||||
) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
|
|
||||||
let channel_id = channel.id().as_u64().to_owned();
|
|
||||||
|
|
||||||
if let Ok(c) = sqlx::query_as_unchecked!(Self,
|
|
||||||
"
|
|
||||||
SELECT id, name, nudge, blacklisted, webhook_id, webhook_token, paused, paused_until FROM channels WHERE channel = ?
|
|
||||||
", channel_id)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await {
|
|
||||||
|
|
||||||
Ok(c)
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
let props = channel.guild().map(|g| (g.guild_id.as_u64().to_owned(), g.name));
|
|
||||||
|
|
||||||
let (guild_id, channel_name) = if let Some((a, b)) = props {
|
|
||||||
(Some(a), Some(b))
|
|
||||||
} else {
|
|
||||||
(None, None)
|
|
||||||
};
|
|
||||||
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT IGNORE INTO channels (channel, name, guild_id) VALUES (?, ?, (SELECT id FROM guilds WHERE guild = ?))
|
|
||||||
", channel_id, channel_name, guild_id)
|
|
||||||
.execute(&pool.clone())
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(sqlx::query_as_unchecked!(Self,
|
|
||||||
"
|
|
||||||
SELECT id, name, nudge, blacklisted, webhook_id, webhook_token, paused, paused_until FROM channels WHERE channel = ?
|
|
||||||
", channel_id)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await?)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn commit_changes(&self, pool: &MySqlPool) {
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
UPDATE channels SET name = ?, nudge = ?, blacklisted = ?, webhook_id = ?, webhook_token = ?, paused = ?, paused_until = ? WHERE id = ?
|
|
||||||
", self.name, self.nudge, self.blacklisted, self.webhook_id, self.webhook_token, self.paused, self.paused_until, self.id)
|
|
||||||
.execute(pool)
|
|
||||||
.await.unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct UserData {
|
|
||||||
pub id: u32,
|
|
||||||
pub user: u64,
|
|
||||||
pub name: String,
|
|
||||||
pub dm_channel: u32,
|
|
||||||
pub language: String,
|
|
||||||
pub timezone: String,
|
|
||||||
pub meridian_time: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct MeridianType(bool);
|
|
||||||
|
|
||||||
impl MeridianType {
|
|
||||||
pub fn fmt_str(&self) -> &str {
|
|
||||||
if self.0 {
|
|
||||||
"%Y-%m-%d %I:%M:%S %p"
|
|
||||||
} else {
|
|
||||||
"%Y-%m-%d %H:%M:%S"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn fmt_str_short(&self) -> &str {
|
|
||||||
if self.0 {
|
|
||||||
"%I:%M %p"
|
|
||||||
} else {
|
|
||||||
"%H:%M"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl UserData {
|
|
||||||
pub async fn language_of<U>(user: U, pool: &MySqlPool) -> String
|
|
||||||
where
|
|
||||||
U: Into<UserId>,
|
|
||||||
{
|
|
||||||
let user_id = user.into().as_u64().to_owned();
|
|
||||||
|
|
||||||
match sqlx::query!(
|
|
||||||
"
|
|
||||||
SELECT language FROM users WHERE user = ?
|
|
||||||
",
|
|
||||||
user_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(r) => r.language,
|
|
||||||
|
|
||||||
Err(_) => LOCAL_LANGUAGE.clone(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn timezone_of<U>(user: U, pool: &MySqlPool) -> Tz
|
|
||||||
where
|
|
||||||
U: Into<UserId>,
|
|
||||||
{
|
|
||||||
let user_id = user.into().as_u64().to_owned();
|
|
||||||
|
|
||||||
match sqlx::query!(
|
|
||||||
"
|
|
||||||
SELECT timezone FROM users WHERE user = ?
|
|
||||||
",
|
|
||||||
user_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(r) => r.timezone,
|
|
||||||
|
|
||||||
Err(_) => LOCAL_TIMEZONE.clone(),
|
|
||||||
}
|
|
||||||
.parse()
|
|
||||||
.unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn meridian_of<U>(user: U, pool: &MySqlPool) -> MeridianType
|
|
||||||
where
|
|
||||||
U: Into<UserId>,
|
|
||||||
{
|
|
||||||
let user_id = user.into().as_u64().to_owned();
|
|
||||||
|
|
||||||
match sqlx::query!(
|
|
||||||
"
|
|
||||||
SELECT meridian_time FROM users WHERE user = ?
|
|
||||||
",
|
|
||||||
user_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(r) => MeridianType(r.meridian_time != 0),
|
|
||||||
|
|
||||||
Err(_) => MeridianType(false),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn from_user(
|
|
||||||
user: &User,
|
|
||||||
ctx: impl CacheHttp,
|
|
||||||
pool: &MySqlPool,
|
|
||||||
) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
|
|
||||||
let user_id = user.id.as_u64().to_owned();
|
|
||||||
|
|
||||||
match sqlx::query_as_unchecked!(
|
|
||||||
Self,
|
|
||||||
"
|
|
||||||
SELECT id, user, name, dm_channel, IF(language IS NULL, ?, language) AS language, IF(timezone IS NULL, ?, timezone) AS timezone, meridian_time FROM users WHERE user = ?
|
|
||||||
",
|
|
||||||
*LOCAL_LANGUAGE, *LOCAL_TIMEZONE, user_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(c) => Ok(c),
|
|
||||||
|
|
||||||
Err(sqlx::Error::RowNotFound) => {
|
|
||||||
let dm_channel = user.create_dm_channel(ctx).await?;
|
|
||||||
let dm_id = dm_channel.id.as_u64().to_owned();
|
|
||||||
|
|
||||||
let pool_c = pool.clone();
|
|
||||||
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT IGNORE INTO channels (channel) VALUES (?)
|
|
||||||
",
|
|
||||||
dm_id
|
|
||||||
)
|
|
||||||
.execute(&pool_c)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT INTO users (user, name, dm_channel, language, timezone) VALUES (?, ?, (SELECT id FROM channels WHERE channel = ?), ?, ?)
|
|
||||||
", user_id, user.name, dm_id, *LOCAL_LANGUAGE, *LOCAL_TIMEZONE)
|
|
||||||
.execute(&pool_c)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(sqlx::query_as_unchecked!(
|
|
||||||
Self,
|
|
||||||
"
|
|
||||||
SELECT id, user, name, dm_channel, language, timezone, meridian_time FROM users WHERE user = ?
|
|
||||||
",
|
|
||||||
user_id
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await?)
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(e) => {
|
|
||||||
error!("Error querying for user: {:?}", e);
|
|
||||||
|
|
||||||
Err(Box::new(e))
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn commit_changes(&self, pool: &MySqlPool) {
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
UPDATE users SET name = ?, language = ?, timezone = ?, meridian_time = ? WHERE id = ?
|
|
||||||
",
|
|
||||||
self.name,
|
|
||||||
self.language,
|
|
||||||
self.timezone,
|
|
||||||
self.meridian_time,
|
|
||||||
self.id
|
|
||||||
)
|
|
||||||
.execute(pool)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn timezone(&self) -> Tz {
|
|
||||||
self.timezone.parse().unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn meridian(&self) -> MeridianType {
|
|
||||||
MeridianType(self.meridian_time)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct Timer {
|
|
||||||
pub name: String,
|
|
||||||
pub start_time: NaiveDateTime,
|
|
||||||
pub owner: u64,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Timer {
|
|
||||||
pub async fn from_owner(owner: u64, pool: &MySqlPool) -> Vec<Self> {
|
|
||||||
sqlx::query_as_unchecked!(
|
|
||||||
Timer,
|
|
||||||
"
|
|
||||||
SELECT name, start_time, owner FROM timers WHERE owner = ?
|
|
||||||
",
|
|
||||||
owner
|
|
||||||
)
|
|
||||||
.fetch_all(pool)
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn count_from_owner(owner: u64, pool: &MySqlPool) -> u32 {
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
SELECT COUNT(1) as count FROM timers WHERE owner = ?
|
|
||||||
",
|
|
||||||
owner
|
|
||||||
)
|
|
||||||
.fetch_one(pool)
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
.count as u32
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn create(name: &str, owner: u64, pool: &MySqlPool) {
|
|
||||||
sqlx::query!(
|
|
||||||
"
|
|
||||||
INSERT INTO timers (name, owner) VALUES (?, ?)
|
|
||||||
",
|
|
||||||
name,
|
|
||||||
owner
|
|
||||||
)
|
|
||||||
.execute(pool)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
}
|
|
||||||
}
|
|
67
src/models/channel_data.rs
Normal file
67
src/models/channel_data.rs
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
use serenity::model::channel::Channel;
|
||||||
|
|
||||||
|
use sqlx::MySqlPool;
|
||||||
|
|
||||||
|
use chrono::NaiveDateTime;
|
||||||
|
|
||||||
|
pub struct ChannelData {
|
||||||
|
pub id: u32,
|
||||||
|
pub name: Option<String>,
|
||||||
|
pub nudge: i16,
|
||||||
|
pub blacklisted: bool,
|
||||||
|
pub webhook_id: Option<u64>,
|
||||||
|
pub webhook_token: Option<String>,
|
||||||
|
pub paused: bool,
|
||||||
|
pub paused_until: Option<NaiveDateTime>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChannelData {
|
||||||
|
pub async fn from_channel(
|
||||||
|
channel: Channel,
|
||||||
|
pool: &MySqlPool,
|
||||||
|
) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
|
||||||
|
let channel_id = channel.id().as_u64().to_owned();
|
||||||
|
|
||||||
|
if let Ok(c) = sqlx::query_as_unchecked!(Self,
|
||||||
|
"
|
||||||
|
SELECT id, name, nudge, blacklisted, webhook_id, webhook_token, paused, paused_until FROM channels WHERE channel = ?
|
||||||
|
", channel_id)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await {
|
||||||
|
|
||||||
|
Ok(c)
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
let props = channel.guild().map(|g| (g.guild_id.as_u64().to_owned(), g.name));
|
||||||
|
|
||||||
|
let (guild_id, channel_name) = if let Some((a, b)) = props {
|
||||||
|
(Some(a), Some(b))
|
||||||
|
} else {
|
||||||
|
(None, None)
|
||||||
|
};
|
||||||
|
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
INSERT IGNORE INTO channels (channel, name, guild_id) VALUES (?, ?, (SELECT id FROM guilds WHERE guild = ?))
|
||||||
|
", channel_id, channel_name, guild_id)
|
||||||
|
.execute(&pool.clone())
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(sqlx::query_as_unchecked!(Self,
|
||||||
|
"
|
||||||
|
SELECT id, name, nudge, blacklisted, webhook_id, webhook_token, paused, paused_until FROM channels WHERE channel = ?
|
||||||
|
", channel_id)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await?)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn commit_changes(&self, pool: &MySqlPool) {
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
UPDATE channels SET name = ?, nudge = ?, blacklisted = ?, webhook_id = ?, webhook_token = ?, paused = ?, paused_until = ? WHERE id = ?
|
||||||
|
", self.name, self.nudge, self.blacklisted, self.webhook_id, self.webhook_token, self.paused, self.paused_until, self.id)
|
||||||
|
.execute(pool)
|
||||||
|
.await.unwrap();
|
||||||
|
}
|
||||||
|
}
|
79
src/models/guild_data.rs
Normal file
79
src/models/guild_data.rs
Normal file
@ -0,0 +1,79 @@
|
|||||||
|
use serenity::model::guild::Guild;
|
||||||
|
|
||||||
|
use sqlx::MySqlPool;
|
||||||
|
|
||||||
|
use log::error;
|
||||||
|
|
||||||
|
use crate::consts::DEFAULT_PREFIX;
|
||||||
|
|
||||||
|
pub struct GuildData {
|
||||||
|
pub id: u32,
|
||||||
|
pub name: Option<String>,
|
||||||
|
pub prefix: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl GuildData {
|
||||||
|
pub async fn from_guild(guild: Guild, pool: &MySqlPool) -> Result<Self, sqlx::Error> {
|
||||||
|
let guild_id = guild.id.as_u64().to_owned();
|
||||||
|
|
||||||
|
match sqlx::query_as!(
|
||||||
|
Self,
|
||||||
|
"
|
||||||
|
SELECT id, name, prefix FROM guilds WHERE guild = ?
|
||||||
|
",
|
||||||
|
guild_id
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(mut g) => {
|
||||||
|
g.name = Some(guild.name);
|
||||||
|
|
||||||
|
Ok(g)
|
||||||
|
}
|
||||||
|
|
||||||
|
Err(sqlx::Error::RowNotFound) => {
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
INSERT INTO guilds (guild, name, prefix) VALUES (?, ?, ?)
|
||||||
|
",
|
||||||
|
guild_id,
|
||||||
|
guild.name,
|
||||||
|
*DEFAULT_PREFIX
|
||||||
|
)
|
||||||
|
.execute(&pool.clone())
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(sqlx::query_as!(
|
||||||
|
Self,
|
||||||
|
"
|
||||||
|
SELECT id, name, prefix FROM guilds WHERE guild = ?
|
||||||
|
",
|
||||||
|
guild_id
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await?)
|
||||||
|
}
|
||||||
|
|
||||||
|
Err(e) => {
|
||||||
|
error!("Unexpected error in guild query: {:?}", e);
|
||||||
|
|
||||||
|
Err(e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn commit_changes(&self, pool: &MySqlPool) {
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
UPDATE guilds SET name = ?, prefix = ? WHERE id = ?
|
||||||
|
",
|
||||||
|
self.name,
|
||||||
|
self.prefix,
|
||||||
|
self.id
|
||||||
|
)
|
||||||
|
.execute(pool)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
77
src/models/mod.rs
Normal file
77
src/models/mod.rs
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
pub mod channel_data;
|
||||||
|
pub mod guild_data;
|
||||||
|
pub mod timer;
|
||||||
|
pub mod user_data;
|
||||||
|
|
||||||
|
use serenity::{async_trait, model::id::GuildId, prelude::Context};
|
||||||
|
|
||||||
|
use crate::{consts::DEFAULT_PREFIX, GuildDataCache, SQLPool};
|
||||||
|
|
||||||
|
use guild_data::GuildData;
|
||||||
|
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
pub trait CtxGuildData {
|
||||||
|
async fn guild_data<G: Into<GuildId> + Send + Sync>(
|
||||||
|
&self,
|
||||||
|
guild_id: G,
|
||||||
|
) -> Result<Arc<RwLock<GuildData>>, sqlx::Error>;
|
||||||
|
|
||||||
|
async fn prefix<G: Into<GuildId> + Send + Sync>(&self, guild_id: Option<G>) -> String;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl CtxGuildData for Context {
|
||||||
|
async fn guild_data<G: Into<GuildId> + Send + Sync>(
|
||||||
|
&self,
|
||||||
|
guild_id: G,
|
||||||
|
) -> Result<Arc<RwLock<GuildData>>, sqlx::Error> {
|
||||||
|
let guild_id = guild_id.into();
|
||||||
|
|
||||||
|
let guild = guild_id.to_guild_cached(&self.cache).await.unwrap();
|
||||||
|
|
||||||
|
let guild_cache = self
|
||||||
|
.data
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.get::<GuildDataCache>()
|
||||||
|
.cloned()
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let x = if let Some(guild_data) = guild_cache.get(&guild_id) {
|
||||||
|
Ok(guild_data.clone())
|
||||||
|
} else {
|
||||||
|
let pool = self.data.read().await.get::<SQLPool>().cloned().unwrap();
|
||||||
|
|
||||||
|
match GuildData::from_guild(guild, &pool).await {
|
||||||
|
Ok(d) => {
|
||||||
|
let lock = Arc::new(RwLock::new(d));
|
||||||
|
|
||||||
|
guild_cache.insert(guild_id, lock.clone());
|
||||||
|
|
||||||
|
Ok(lock)
|
||||||
|
}
|
||||||
|
|
||||||
|
Err(e) => Err(e),
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
x
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn prefix<G: Into<GuildId> + Send + Sync>(&self, guild_id: Option<G>) -> String {
|
||||||
|
if let Some(guild_id) = guild_id {
|
||||||
|
self.guild_data(guild_id)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.prefix
|
||||||
|
.clone()
|
||||||
|
} else {
|
||||||
|
DEFAULT_PREFIX.clone()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
50
src/models/timer.rs
Normal file
50
src/models/timer.rs
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
use sqlx::MySqlPool;
|
||||||
|
|
||||||
|
use chrono::NaiveDateTime;
|
||||||
|
|
||||||
|
pub struct Timer {
|
||||||
|
pub name: String,
|
||||||
|
pub start_time: NaiveDateTime,
|
||||||
|
pub owner: u64,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Timer {
|
||||||
|
pub async fn from_owner(owner: u64, pool: &MySqlPool) -> Vec<Self> {
|
||||||
|
sqlx::query_as_unchecked!(
|
||||||
|
Timer,
|
||||||
|
"
|
||||||
|
SELECT name, start_time, owner FROM timers WHERE owner = ?
|
||||||
|
",
|
||||||
|
owner
|
||||||
|
)
|
||||||
|
.fetch_all(pool)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn count_from_owner(owner: u64, pool: &MySqlPool) -> u32 {
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
SELECT COUNT(1) as count FROM timers WHERE owner = ?
|
||||||
|
",
|
||||||
|
owner
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.count as u32
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn create(name: &str, owner: u64, pool: &MySqlPool) {
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
INSERT INTO timers (name, owner) VALUES (?, ?)
|
||||||
|
",
|
||||||
|
name,
|
||||||
|
owner
|
||||||
|
)
|
||||||
|
.execute(pool)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
146
src/models/user_data.rs
Normal file
146
src/models/user_data.rs
Normal file
@ -0,0 +1,146 @@
|
|||||||
|
use serenity::{
|
||||||
|
http::CacheHttp,
|
||||||
|
model::{id::UserId, user::User},
|
||||||
|
};
|
||||||
|
|
||||||
|
use sqlx::MySqlPool;
|
||||||
|
|
||||||
|
use chrono_tz::Tz;
|
||||||
|
|
||||||
|
use log::error;
|
||||||
|
|
||||||
|
use crate::consts::{LOCAL_LANGUAGE, LOCAL_TIMEZONE};
|
||||||
|
|
||||||
|
pub struct UserData {
|
||||||
|
pub id: u32,
|
||||||
|
pub user: u64,
|
||||||
|
pub name: String,
|
||||||
|
pub dm_channel: u32,
|
||||||
|
pub language: String,
|
||||||
|
pub timezone: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl UserData {
|
||||||
|
pub async fn language_of<U>(user: U, pool: &MySqlPool) -> String
|
||||||
|
where
|
||||||
|
U: Into<UserId>,
|
||||||
|
{
|
||||||
|
let user_id = user.into().as_u64().to_owned();
|
||||||
|
|
||||||
|
match sqlx::query!(
|
||||||
|
"
|
||||||
|
SELECT language FROM users WHERE user = ?
|
||||||
|
",
|
||||||
|
user_id
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(r) => r.language,
|
||||||
|
|
||||||
|
Err(_) => LOCAL_LANGUAGE.clone(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn timezone_of<U>(user: U, pool: &MySqlPool) -> Tz
|
||||||
|
where
|
||||||
|
U: Into<UserId>,
|
||||||
|
{
|
||||||
|
let user_id = user.into().as_u64().to_owned();
|
||||||
|
|
||||||
|
match sqlx::query!(
|
||||||
|
"
|
||||||
|
SELECT timezone FROM users WHERE user = ?
|
||||||
|
",
|
||||||
|
user_id
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(r) => r.timezone,
|
||||||
|
|
||||||
|
Err(_) => LOCAL_TIMEZONE.clone(),
|
||||||
|
}
|
||||||
|
.parse()
|
||||||
|
.unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn from_user(
|
||||||
|
user: &User,
|
||||||
|
ctx: impl CacheHttp,
|
||||||
|
pool: &MySqlPool,
|
||||||
|
) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
|
||||||
|
let user_id = user.id.as_u64().to_owned();
|
||||||
|
|
||||||
|
match sqlx::query_as_unchecked!(
|
||||||
|
Self,
|
||||||
|
"
|
||||||
|
SELECT id, user, name, dm_channel, IF(language IS NULL, ?, language) AS language, IF(timezone IS NULL, ?, timezone) AS timezone FROM users WHERE user = ?
|
||||||
|
",
|
||||||
|
*LOCAL_LANGUAGE, *LOCAL_TIMEZONE, user_id
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(c) => Ok(c),
|
||||||
|
|
||||||
|
Err(sqlx::Error::RowNotFound) => {
|
||||||
|
let dm_channel = user.create_dm_channel(ctx).await?;
|
||||||
|
let dm_id = dm_channel.id.as_u64().to_owned();
|
||||||
|
|
||||||
|
let pool_c = pool.clone();
|
||||||
|
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
INSERT IGNORE INTO channels (channel) VALUES (?)
|
||||||
|
",
|
||||||
|
dm_id
|
||||||
|
)
|
||||||
|
.execute(&pool_c)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
INSERT INTO users (user, name, dm_channel, language, timezone) VALUES (?, ?, (SELECT id FROM channels WHERE channel = ?), ?, ?)
|
||||||
|
", user_id, user.name, dm_id, *LOCAL_LANGUAGE, *LOCAL_TIMEZONE)
|
||||||
|
.execute(&pool_c)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(sqlx::query_as_unchecked!(
|
||||||
|
Self,
|
||||||
|
"
|
||||||
|
SELECT id, user, name, dm_channel, language, timezone FROM users WHERE user = ?
|
||||||
|
",
|
||||||
|
user_id
|
||||||
|
)
|
||||||
|
.fetch_one(pool)
|
||||||
|
.await?)
|
||||||
|
}
|
||||||
|
|
||||||
|
Err(e) => {
|
||||||
|
error!("Error querying for user: {:?}", e);
|
||||||
|
|
||||||
|
Err(Box::new(e))
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn commit_changes(&self, pool: &MySqlPool) {
|
||||||
|
sqlx::query!(
|
||||||
|
"
|
||||||
|
UPDATE users SET name = ?, language = ?, timezone = ? WHERE id = ?
|
||||||
|
",
|
||||||
|
self.name,
|
||||||
|
self.language,
|
||||||
|
self.timezone,
|
||||||
|
self.id
|
||||||
|
)
|
||||||
|
.execute(pool)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn timezone(&self) -> Tz {
|
||||||
|
self.timezone.parse().unwrap()
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user