silver/discord-bot
1
0
Fork 0
forked from Skynet/discord-bot
Code Pull requests Activity

fmt: fmt and clippy

Browse source
This commit is contained in:
silver 2023-09-11 20:03:56 +01:00
parent a745d7631d
commit 974173857c
1 changed files with 15 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
src/bin/update_data.rs
View file

@ -1,4 +1,4 @@
use skynet_discord_bot::{db_init, get_config, Accounts, Config, DataBase, get_server_config_bulk, Servers};
use skynet_discord_bot::{db_init, get_config, get_server_config_bulk, Accounts, Config, DataBase, Servers};
use std::{process, sync::Arc};
use serde::Deserialize;
@ -78,7 +78,6 @@ async fn fetch_accounts(ctx: &Context) {
// get from skynet for the compsoc server only
get_skynet(&db, &config).await;
}
#[derive(Debug, Deserialize)]
@ -86,11 +85,11 @@ pub struct SkynetResult {
discord: String,
wolves_id: String,
}
async fn get_skynet(db: &Pool<Sqlite>, config: &Config){
async fn get_skynet(db: &Pool<Sqlite>, config: &Config) {
let url = format!("{}/ldap/discord?auth={}", &config.ldap_api, &config.auth);
if let Ok(result) = surf::get(url).recv_json::<Vec<SkynetResult>>().await {
for user in result {
add_users_skynet(&db, &config.skynet_server, &user).await;
add_users_skynet(db, &config.skynet_server, &user).await;
}
}
}
@ -122,36 +121,36 @@ struct WolvesResult {
pub email: String,
pub expiry: String,
}
async fn get_wolves(db: &Pool<Sqlite>){
async fn get_wolves(db: &Pool<Sqlite>) {
for server_config in get_server_config_bulk(db).await {
let Servers {
server,
wolves_api,
..
} = server_config;
// get the data here
let mut result: Vec<WolvesResult> = vec![];
let result: Vec<WolvesResult> = vec![];
for user in result {
add_users_wolves(&db, &server, &user).await;
add_users_wolves(db, &server, &user).await;
}
}
}
async fn add_users_wolves(db: &Pool<Sqlite>, server: &GuildId, user: &WolvesResult) {
match sqlx::query_as::<_, Accounts>(
"
"
INSERT OR REPLACE INTO accounts (server, wolves_id, email, expiry)
VALUES (?1, ?2, ?3, ?4)
",
)
.bind(*server.as_u64() as i64)
.bind(&user.id_wolves)
.bind(&user.email)
.bind(&user.expiry)
.fetch_optional(db)
.await
.bind(*server.as_u64() as i64)
.bind(&user.id_wolves)
.bind(&user.email)
.bind(&user.expiry)
.fetch_optional(db)
.await
{
Ok(_) => {}
Err(e) => {
@ -159,4 +158,4 @@ async fn add_users_wolves(db: &Pool<Sqlite>, server: &GuildId, user: &WolvesResu
println!("{:?}", e);
}
}
}
}