Fetch upcoming schedule and backlog count

This commit is contained in:
jude 2023-06-21 13:26:28 +01:00
parent 64dd81e941
commit c3c0dbbbae
2 changed files with 53 additions and 2 deletions

View File

@ -168,7 +168,7 @@ pub async fn initialize(
routes::dashboard::export::import_todos, routes::dashboard::export::import_todos,
], ],
) )
.mount("/admin", routes![routes::admin::admin_dashboard_home]) .mount("/admin", routes![routes::admin::admin_dashboard_home, routes::admin::bot_data])
.launch() .launch()
.await?; .await?;

View File

@ -1,10 +1,22 @@
use std::{collections::HashMap, env}; use std::{collections::HashMap, env};
use chrono::{DateTime, Utc};
use rocket::{ use rocket::{
http::{CookieJar, Status}, http::{CookieJar, Status},
serde::json::{json, Json}, serde::json::json,
State,
}; };
use rocket_dyn_templates::Template; use rocket_dyn_templates::Template;
use serde::Serialize;
use sqlx::{MySql, Pool};
use crate::routes::JsonResult;
fn is_admin(cookies: &CookieJar<'_>) -> bool {
cookies
.get_private("userid")
.map_or(false, |cookie| Some(cookie.value().to_string()) == env::var("ADMIN_ID").ok())
}
#[get("/")] #[get("/")]
pub async fn admin_dashboard_home(cookies: &CookieJar<'_>) -> Result<Template, Status> { pub async fn admin_dashboard_home(cookies: &CookieJar<'_>) -> Result<Template, Status> {
@ -19,3 +31,42 @@ pub async fn admin_dashboard_home(cookies: &CookieJar<'_>) -> Result<Template, S
Err(Status::Unauthorized) Err(Status::Unauthorized)
} }
} }
#[derive(Serialize)]
struct TimeFrame {
time_key: DateTime<Utc>,
count: i64,
}
#[get("/data")]
pub async fn bot_data(cookies: &CookieJar<'_>, pool: &State<Pool<MySql>>) -> JsonResult {
if !is_admin(cookies) {
return json_err!("Not authorized");
}
let backlog = sqlx::query!(
"SELECT COUNT(1) AS backlog FROM reminders WHERE `utc_time` < NOW() AND enabled = 1"
)
.fetch_one(pool.inner())
.await
.unwrap();
let schedule = sqlx::query_as_unchecked!(
TimeFrame,
"SELECT
FROM_UNIXTIME(FLOOR(UNIX_TIMESTAMP(`utc_time`) / 60) * 60) AS `time_key`,
COUNT(1) AS `count`
FROM reminders
WHERE
`utc_time` < DATE_ADD(NOW(), INTERVAL 1 DAY) AND
`utc_time` >= NOW() AND
`enabled` = 1
GROUP BY `time_key`
ORDER BY `time_key`"
)
.fetch_all(pool.inner())
.await
.unwrap();
Ok(json!({ "backlog": backlog.backlog, "schedule": schedule }))
}