Fetch upcoming schedule and backlog count
This commit is contained in:
parent
64dd81e941
commit
c3c0dbbbae
@ -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?;
|
||||||
|
|
||||||
|
@ -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 }))
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user