Fix some formatting and improve structure

This commit is contained in:
Lol3rrr
2024-09-16 23:20:55 +02:00
parent a1faa1a815
commit 1e6c3da58b
12 changed files with 423 additions and 249 deletions

View File

@@ -10,17 +10,15 @@ pub struct AnalysisInput {
#[derive(Debug)] #[derive(Debug)]
pub struct BaseInfo { pub struct BaseInfo {
pub map: String, pub map: String,
pub players: Vec<()> pub players: Vec<()>,
} }
#[tracing::instrument(skip(input))] #[tracing::instrument(skip(input))]
pub fn analyse_base(input: AnalysisInput) -> BaseInfo { pub fn analyse_base(input: AnalysisInput) -> BaseInfo {
tracing::info!("Performing Base analysis"); tracing::info!("Performing Base analysis");
let file = std::fs::File::open(&input.path).unwrap(); let file = std::fs::File::open(&input.path).unwrap();
let mmap = unsafe { memmap2::MmapOptions::new().map(&file).unwrap() }; let mmap = unsafe { memmap2::MmapOptions::new().map(&file).unwrap() };
let tmp = csdemo::Container::parse(&mmap).unwrap(); let tmp = csdemo::Container::parse(&mmap).unwrap();
let output = csdemo::parser::parse(csdemo::FrameIterator::parse(tmp.inner)).unwrap(); let output = csdemo::parser::parse(csdemo::FrameIterator::parse(tmp.inner)).unwrap();
@@ -85,6 +83,6 @@ pub fn analyse_base(input: AnalysisInput) -> BaseInfo {
BaseInfo { BaseInfo {
map, map,
players: Vec::new() players: Vec::new(),
} }
} }

View File

@@ -2,14 +2,14 @@ pub mod demos;
pub mod steam { pub mod steam {
use axum::extract::State; use axum::extract::State;
use serde::Deserialize;
use std::{sync::Arc, collections::HashMap};
use diesel::prelude::*; use diesel::prelude::*;
use diesel_async::RunQueryDsl; use diesel_async::RunQueryDsl;
use serde::Deserialize;
use std::{collections::HashMap, sync::Arc};
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct ProfileInfoResponse { struct ProfileInfoResponse {
players: Vec<ProfileInfo> players: Vec<ProfileInfo>,
} }
#[derive(Debug, Deserialize)] #[derive(Debug, Deserialize)]
struct ProfileInfo { struct ProfileInfo {
@@ -23,11 +23,15 @@ pub mod steam {
axum::Router::new() axum::Router::new()
.route("/login", axum::routing::get(steam_login)) .route("/login", axum::routing::get(steam_login))
.route("/callback", axum::routing::get(steam_callback)) .route("/callback", axum::routing::get(steam_callback))
.with_state(Arc::new(steam_openid::SteamOpenId::new(url, callback_path).unwrap())) .with_state(Arc::new(
steam_openid::SteamOpenId::new(url, callback_path).unwrap(),
))
} }
#[tracing::instrument(skip(openid))] #[tracing::instrument(skip(openid))]
async fn steam_login(State(openid): State<Arc<steam_openid::SteamOpenId>>) -> Result<axum::response::Redirect, axum::http::StatusCode> { async fn steam_login(
State(openid): State<Arc<steam_openid::SteamOpenId>>,
) -> Result<axum::response::Redirect, axum::http::StatusCode> {
let url = openid.get_redirect_url(); let url = openid.get_redirect_url();
Ok(axum::response::Redirect::to(url)) Ok(axum::response::Redirect::to(url))
@@ -52,7 +56,13 @@ pub mod steam {
})?; })?;
let steam_client = crate::steam_api::Client::new(std::env::var("STEAM_API_KEY").unwrap()); let steam_client = crate::steam_api::Client::new(std::env::var("STEAM_API_KEY").unwrap());
let profile_response_data: ProfileInfoResponse = match steam_client.get("http://api.steampowered.com/ISteamUser/GetPlayerSummaries/v2/", &[("steamids", &format!("{}", id))]).await { let profile_response_data: ProfileInfoResponse = match steam_client
.get(
"http://api.steampowered.com/ISteamUser/GetPlayerSummaries/v2/",
&[("steamids", &format!("{}", id))],
)
.await
{
Ok(r) => r, Ok(r) => r,
Err(e) => { Err(e) => {
tracing::error!("Getting Steam Profile Info: {:?}", e); tracing::error!("Getting Steam Profile Info: {:?}", e);
@@ -62,10 +72,14 @@ pub mod steam {
let mut db_con = crate::db_connection().await; let mut db_con = crate::db_connection().await;
for player in profile_response_data.players { for player in profile_response_data.players {
let query = diesel::dsl::insert_into(crate::schema::users::dsl::users).values(crate::models::User { let query = diesel::dsl::insert_into(crate::schema::users::dsl::users)
steamid: player.steamid, .values(crate::models::User {
name: player.personaname.clone(), steamid: player.steamid,
}).on_conflict(crate::schema::users::dsl::steamid).do_update().set((crate::schema::users::dsl::name.eq(player.personaname))); name: player.personaname.clone(),
})
.on_conflict(crate::schema::users::dsl::steamid)
.do_update()
.set((crate::schema::users::dsl::name.eq(player.personaname)));
tracing::debug!("Running Query: {:?}", query); tracing::debug!("Running Query: {:?}", query);
if let Err(e) = query.execute(&mut db_con).await { if let Err(e) = query.execute(&mut db_con).await {
@@ -88,12 +102,13 @@ pub mod user {
use diesel_async::RunQueryDsl; use diesel_async::RunQueryDsl;
pub fn router() -> axum::Router { pub fn router() -> axum::Router {
axum::Router::new() axum::Router::new().route("/status", axum::routing::get(status))
.route("/status", axum::routing::get(status))
} }
#[tracing::instrument(skip(session))] #[tracing::instrument(skip(session))]
async fn status(session: crate::UserSession) -> Result<axum::response::Json<common::UserStatus>, reqwest::StatusCode> { async fn status(
session: crate::UserSession,
) -> Result<axum::response::Json<common::UserStatus>, reqwest::StatusCode> {
let steam_id = match session.data().steam_id { let steam_id = match session.data().steam_id {
Some(s) => s, Some(s) => s,
None => { None => {
@@ -105,16 +120,20 @@ pub mod user {
let mut db_con = crate::db_connection().await; let mut db_con = crate::db_connection().await;
let query = crate::schema::users::dsl::users.filter(crate::schema::users::dsl::steamid.eq(format!("{}", steam_id))); let query = crate::schema::users::dsl::users
.filter(crate::schema::users::dsl::steamid.eq(format!("{}", steam_id)));
let mut result = query.load::<crate::models::User>(&mut db_con).await.unwrap(); let mut result = query
.load::<crate::models::User>(&mut db_con)
.await
.unwrap();
if result.len() != 1 { if result.len() != 1 {
tracing::error!("Unexpected query result: {:?}", result); tracing::error!("Unexpected query result: {:?}", result);
return Err(axum::http::StatusCode::INTERNAL_SERVER_ERROR); return Err(axum::http::StatusCode::INTERNAL_SERVER_ERROR);
} }
let user_entry = result.pop().unwrap(); let user_entry = result.pop().unwrap();
Ok(axum::Json(common::UserStatus { Ok(axum::Json(common::UserStatus {
name: user_entry.name, name: user_entry.name,
steamid: user_entry.steamid, steamid: user_entry.steamid,
@@ -122,9 +141,14 @@ pub mod user {
} }
} }
pub fn router(base_analysis: tokio::sync::mpsc::UnboundedSender<crate::analysis::AnalysisInput>) -> axum::Router { pub fn router(
base_analysis: tokio::sync::mpsc::UnboundedSender<crate::analysis::AnalysisInput>,
) -> axum::Router {
axum::Router::new() axum::Router::new()
.nest("/steam/", steam::router("http://192.168.0.156:3000", "/api/steam/callback")) .nest(
"/steam/",
steam::router("http://192.168.0.156:3000", "/api/steam/callback"),
)
.nest("/demos/", demos::router("uploads/", base_analysis)) .nest("/demos/", demos::router("uploads/", base_analysis))
.nest("/user/", user::router()) .nest("/user/", user::router())
} }

View File

@@ -1,94 +1,148 @@
use crate::UserSession; use crate::UserSession;
use diesel_async::RunQueryDsl; use axum::extract::{Path, State};
use diesel::prelude::*; use diesel::prelude::*;
use axum::extract::{State, Path}; use diesel_async::RunQueryDsl;
use std::sync::Arc; use std::sync::Arc;
struct DemoState { struct DemoState {
upload_folder: std::path::PathBuf, upload_folder: std::path::PathBuf,
base_analysis: tokio::sync::mpsc::UnboundedSender<crate::analysis::AnalysisInput> base_analysis: tokio::sync::mpsc::UnboundedSender<crate::analysis::AnalysisInput>,
}
pub fn router<P>(
upload_folder: P,
base_analysis: tokio::sync::mpsc::UnboundedSender<crate::analysis::AnalysisInput>,
) -> axum::Router
where
P: Into<std::path::PathBuf>,
{
axum::Router::new()
.route("/list", axum::routing::get(list))
.route(
"/upload",
axum::routing::post(upload)
.layer(axum::extract::DefaultBodyLimit::max(500 * 1024 * 1024)),
)
.route("/:id/info", axum::routing::get(info))
.route("/:id/reanalyse", axum::routing::get(analyise))
.with_state(Arc::new(DemoState {
upload_folder: upload_folder.into(),
base_analysis,
}))
}
#[tracing::instrument(skip(session))]
async fn list(
session: UserSession,
) -> Result<axum::response::Json<Vec<common::BaseDemoInfo>>, axum::http::StatusCode> {
let steam_id = session
.data()
.steam_id
.ok_or_else(|| axum::http::StatusCode::UNAUTHORIZED)?;
tracing::info!("SteamID: {:?}", steam_id);
let query = crate::schema::demos::dsl::demos
.inner_join(crate::schema::demo_info::dsl::demo_info)
.select((
crate::models::Demo::as_select(),
crate::models::DemoInfo::as_select(),
))
.filter(crate::schema::demos::dsl::steam_id.eq(steam_id.to_string()));
let results: Vec<(crate::models::Demo, crate::models::DemoInfo)> =
query.load(&mut crate::db_connection().await).await.unwrap();
Ok(axum::response::Json(
results
.into_iter()
.map(|(demo, info)| common::BaseDemoInfo {
id: demo.demo_id,
map: info.map,
})
.collect::<Vec<_>>(),
))
}
#[tracing::instrument(skip(state, session))]
async fn upload(
State(state): State<Arc<DemoState>>,
session: crate::UserSession,
form: axum::extract::Multipart,
) -> Result<axum::response::Redirect, (axum::http::StatusCode, &'static str)> {
let steam_id = session
.data()
.steam_id
.ok_or_else(|| (axum::http::StatusCode::UNAUTHORIZED, "Not logged in"))?;
tracing::info!("Upload for Session: {:?}", steam_id);
let file_content = crate::get_demo_from_upload("demo", form).await.unwrap();
let user_folder = std::path::Path::new(&state.upload_folder).join(format!("{}/", steam_id));
if !tokio::fs::try_exists(&user_folder).await.unwrap_or(false) {
tokio::fs::create_dir_all(&user_folder).await.unwrap();
} }
pub fn router<P>(upload_folder: P, base_analysis: tokio::sync::mpsc::UnboundedSender<crate::analysis::AnalysisInput>) -> axum::Router where P: Into<std::path::PathBuf> { let timestamp_secs = std::time::SystemTime::now()
axum::Router::new() .duration_since(std::time::SystemTime::UNIX_EPOCH)
.route("/list", axum::routing::get(list)) .unwrap()
.route("/upload", axum::routing::post(upload).layer(axum::extract::DefaultBodyLimit::max(500*1024*1024))) .as_secs();
.route("/:id/info", axum::routing::get(info)) let demo_id = timestamp_secs as i64;
.route("/:id/reanalyse", axum::routing::get(analyise)) let demo_file_path = user_folder.join(format!("{}.dem", timestamp_secs));
.with_state(Arc::new(DemoState {
upload_folder: upload_folder.into(),
base_analysis,
}))
}
#[tracing::instrument(skip(session))] tokio::fs::write(&demo_file_path, file_content)
async fn list(session: UserSession) -> Result<axum::response::Json<Vec<common::BaseDemoInfo>>, axum::http::StatusCode> { .await
let steam_id = session.data().steam_id.ok_or_else(|| axum::http::StatusCode::UNAUTHORIZED)?; .unwrap();
tracing::info!("SteamID: {:?}", steam_id);
let query = crate::schema::demos::dsl::demos.inner_join(crate::schema::demo_info::dsl::demo_info).select((crate::models::Demo::as_select(), crate::models::DemoInfo::as_select())).filter(crate::schema::demos::dsl::steam_id.eq(steam_id.to_string()));
let results: Vec<(crate::models::Demo, crate::models::DemoInfo)> = query.load(&mut crate::db_connection().await).await.unwrap();
Ok(axum::response::Json(results.into_iter().map(|(demo, info)| common::BaseDemoInfo {
id: demo.demo_id,
map: info.map,
}).collect::<Vec<_>>()))
}
#[tracing::instrument(skip(state, session))]
async fn upload(State(state): State<Arc<DemoState>>, session: crate::UserSession, form: axum::extract::Multipart) -> Result<axum::response::Redirect, (axum::http::StatusCode, &'static str)> {
let steam_id = session.data().steam_id.ok_or_else(|| (axum::http::StatusCode::UNAUTHORIZED, "Not logged in"))?;
tracing::info!("Upload for Session: {:?}", steam_id);
let file_content = crate::get_demo_from_upload("demo", form).await.unwrap();
let user_folder = std::path::Path::new(&state.upload_folder).join(format!("{}/", steam_id));
if !tokio::fs::try_exists(&user_folder).await.unwrap_or(false) {
tokio::fs::create_dir_all(&user_folder).await.unwrap();
}
let timestamp_secs = std::time::SystemTime::now().duration_since(std::time::SystemTime::UNIX_EPOCH).unwrap().as_secs();
let demo_id = timestamp_secs as i64;
let demo_file_path = user_folder.join(format!("{}.dem", timestamp_secs));
tokio::fs::write(&demo_file_path, file_content).await.unwrap();
let mut db_con = crate::db_connection().await;
let query = diesel::dsl::insert_into(crate::schema::demos::dsl::demos).values(crate::models::Demo {
demo_id,
steam_id: steam_id.to_string(),
});
query.execute(&mut db_con).await.unwrap();
state.base_analysis.send(crate::analysis::AnalysisInput {
steamid: steam_id.to_string(),
demoid: demo_id,
path: demo_file_path,
});
let processing_query = diesel::dsl::insert_into(crate::schema::processing_status::dsl::processing_status).values(crate::models::ProcessingStatus {
demo_id,
info: 0,
});
processing_query.execute(&mut db_con).await.unwrap();
Ok(axum::response::Redirect::to("/"))
}
#[tracing::instrument(skip(state, session))]
async fn analyise(State(state): State<Arc<DemoState>>, session: crate::UserSession, Path(demo_id): Path<i64>) -> Result<(), (axum::http::StatusCode, &'static str)> {
let steam_id = session.data().steam_id.ok_or_else(|| (axum::http::StatusCode::UNAUTHORIZED, "Not logged in"))?;
tracing::info!("Upload for Session: {:?}", steam_id);
let mut db_con = crate::db_connection().await; let mut db_con = crate::db_connection().await;
let query = crate::schema::demos::dsl::demos.filter(crate::schema::demos::dsl::steam_id.eq(steam_id.to_string())).filter(crate::schema::demos::dsl::demo_id.eq(demo_id)); let query =
let result: Vec<_> = query.load::<crate::models::Demo>(&mut db_con).await.unwrap(); diesel::dsl::insert_into(crate::schema::demos::dsl::demos).values(crate::models::Demo {
demo_id,
steam_id: steam_id.to_string(),
});
query.execute(&mut db_con).await.unwrap();
state.base_analysis.send(crate::analysis::AnalysisInput {
steamid: steam_id.to_string(),
demoid: demo_id,
path: demo_file_path,
});
let processing_query =
diesel::dsl::insert_into(crate::schema::processing_status::dsl::processing_status)
.values(crate::models::ProcessingStatus { demo_id, info: 0 });
processing_query.execute(&mut db_con).await.unwrap();
Ok(axum::response::Redirect::to("/"))
}
#[tracing::instrument(skip(state, session))]
async fn analyise(
State(state): State<Arc<DemoState>>,
session: crate::UserSession,
Path(demo_id): Path<i64>,
) -> Result<(), (axum::http::StatusCode, &'static str)> {
let steam_id = session
.data()
.steam_id
.ok_or_else(|| (axum::http::StatusCode::UNAUTHORIZED, "Not logged in"))?;
tracing::info!("Upload for Session: {:?}", steam_id);
let mut db_con = crate::db_connection().await;
let query = crate::schema::demos::dsl::demos
.filter(crate::schema::demos::dsl::steam_id.eq(steam_id.to_string()))
.filter(crate::schema::demos::dsl::demo_id.eq(demo_id));
let result: Vec<_> = query
.load::<crate::models::Demo>(&mut db_con)
.await
.unwrap();
if result.len() != 1 { if result.len() != 1 {
return Err((axum::http::StatusCode::BAD_REQUEST, "Expected exactly 1 demo to match")); return Err((
axum::http::StatusCode::BAD_REQUEST,
"Expected exactly 1 demo to match",
));
} }
let user_folder = std::path::Path::new(&state.upload_folder).join(format!("{}/", steam_id)); let user_folder = std::path::Path::new(&state.upload_folder).join(format!("{}/", steam_id));
@@ -99,24 +153,30 @@ use crate::UserSession;
}); });
Ok(()) Ok(())
} }
#[tracing::instrument(skip(session))] #[tracing::instrument(skip(session))]
async fn info(session: UserSession, Path(demo_id): Path<i64>) -> Result<axum::response::Json<common::DemoInfo>, axum::http::StatusCode> { async fn info(
tracing::info!("Get info for Demo: {:?}", demo_id); session: UserSession,
Path(demo_id): Path<i64>,
) -> Result<axum::response::Json<common::DemoInfo>, axum::http::StatusCode> {
tracing::info!("Get info for Demo: {:?}", demo_id);
let query = crate::schema::demo_info::dsl::demo_info.select(crate::models::DemoInfo::as_select()).filter(crate::schema::demo_info::dsl::demo_id.eq(demo_id)); let query = crate::schema::demo_info::dsl::demo_info
let mut results: Vec<crate::models::DemoInfo> = query.load(&mut crate::db_connection().await).await.unwrap(); .select(crate::models::DemoInfo::as_select())
.filter(crate::schema::demo_info::dsl::demo_id.eq(demo_id));
let mut results: Vec<crate::models::DemoInfo> =
query.load(&mut crate::db_connection().await).await.unwrap();
if results.len() != 1 { if results.len() != 1 {
tracing::error!("Expected only 1 match but got {} matches", results.len()); tracing::error!("Expected only 1 match but got {} matches", results.len());
return Err(axum::http::StatusCode::INTERNAL_SERVER_ERROR); return Err(axum::http::StatusCode::INTERNAL_SERVER_ERROR);
} }
let result = results.pop().unwrap(); let result = results.pop().unwrap();
Ok(axum::Json(common::DemoInfo { Ok(axum::Json(common::DemoInfo {
id: result.demo_id, id: result.demo_id,
map: result.map, map: result.map,
})) }))
} }

View File

@@ -6,21 +6,25 @@ use diesel_async::RunQueryDsl;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub struct DieselStore {} pub struct DieselStore {}
static EXPIRY_FORMAT: std::sync::LazyLock<&[time::format_description::BorrowedFormatItem<'static>]> = std::sync::LazyLock::new(|| { static EXPIRY_FORMAT: std::sync::LazyLock<
time::macros::format_description!( &[time::format_description::BorrowedFormatItem<'static>],
> = std::sync::LazyLock::new(|| {
time::macros::format_description!(
"[year]-[month]-[day] [hour]:[minute]:[second] [offset_hour sign:mandatory]:[offset_minute]:[offset_second]" "[year]-[month]-[day] [hour]:[minute]:[second] [offset_hour sign:mandatory]:[offset_minute]:[offset_second]"
) )
}); });
impl DieselStore { impl DieselStore {
pub fn new() -> Self { pub fn new() -> Self {
Self {} Self {}
} }
fn id_to_bytes(&self, val: i128) -> Vec<i64> { fn id_to_bytes(&self, val: i128) -> Vec<i64> {
let id_bytes = val.to_be_bytes(); let id_bytes = val.to_be_bytes();
vec![i64::from_be_bytes((id_bytes[0..8]).try_into().unwrap()), i64::from_be_bytes((id_bytes[8..16]).try_into().unwrap())] vec![
i64::from_be_bytes((id_bytes[0..8]).try_into().unwrap()),
i64::from_be_bytes((id_bytes[8..16]).try_into().unwrap()),
]
} }
fn bytes_to_id(&self, val: Vec<i64>) -> i128 { fn bytes_to_id(&self, val: Vec<i64>) -> i128 {
assert_eq!(2, val.len()); assert_eq!(2, val.len());
@@ -28,7 +32,10 @@ impl DieselStore {
let fb = val[0].to_be_bytes(); let fb = val[0].to_be_bytes();
let sb = val[1].to_be_bytes(); let sb = val[1].to_be_bytes();
i128::from_be_bytes([fb[0], fb[1], fb[2], fb[3], fb[4], fb[5], fb[6], fb[7], sb[0], sb[1], sb[2], sb[3], sb[4], sb[5], sb[6], sb[7]]) i128::from_be_bytes([
fb[0], fb[1], fb[2], fb[3], fb[4], fb[5], fb[6], fb[7], sb[0], sb[1], sb[2], sb[3],
sb[4], sb[5], sb[6], sb[7],
])
} }
fn expiry_to_string(&self, expiry_date: &time::OffsetDateTime) -> String { fn expiry_to_string(&self, expiry_date: &time::OffsetDateTime) -> String {
@@ -41,15 +48,22 @@ impl DieselStore {
#[async_trait::async_trait] #[async_trait::async_trait]
impl tower_sessions::SessionStore for DieselStore { impl tower_sessions::SessionStore for DieselStore {
async fn save(&self,session_record: &tower_sessions::session::Record) -> tower_sessions::session_store::Result<()> { async fn save(
&self,
session_record: &tower_sessions::session::Record,
) -> tower_sessions::session_store::Result<()> {
let db_id = session_record.id.0.to_string(); let db_id = session_record.id.0.to_string();
let data = serde_json::to_value(&session_record.data).unwrap(); let data = serde_json::to_value(&session_record.data).unwrap();
let expiry_date = self.expiry_to_string(&session_record.expiry_date); let expiry_date = self.expiry_to_string(&session_record.expiry_date);
let steamid = session_record.data.get(crate::UserSession::KEY).map(|e| serde_json::from_value::<crate::UserSessionData>(e.clone()).ok()).flatten().map(|d| { let steamid = session_record
d.steam_id.map(|s| s.to_string()) .data
}).flatten(); .get(crate::UserSession::KEY)
.map(|e| serde_json::from_value::<crate::UserSessionData>(e.clone()).ok())
.flatten()
.map(|d| d.steam_id.map(|s| s.to_string()))
.flatten();
let query = diesel::dsl::insert_into(crate::schema::sessions::dsl::sessions) let query = diesel::dsl::insert_into(crate::schema::sessions::dsl::sessions)
.values(crate::models::Session { .values(crate::models::Session {
@@ -59,7 +73,10 @@ impl tower_sessions::SessionStore for DieselStore {
}) })
.on_conflict(crate::schema::sessions::dsl::id) .on_conflict(crate::schema::sessions::dsl::id)
.do_update() .do_update()
.set((crate::schema::sessions::dsl::steamid.eq(steamid), crate::schema::sessions::dsl::expiry_date.eq(expiry_date))); .set((
crate::schema::sessions::dsl::steamid.eq(steamid),
crate::schema::sessions::dsl::expiry_date.eq(expiry_date),
));
let mut connection = crate::db_connection().await; let mut connection = crate::db_connection().await;
@@ -68,10 +85,14 @@ impl tower_sessions::SessionStore for DieselStore {
Ok(()) Ok(())
} }
async fn load(&self,session_id: &tower_sessions::session::Id) -> tower_sessions::session_store::Result<Option<tower_sessions::session::Record>> { async fn load(
&self,
session_id: &tower_sessions::session::Id,
) -> tower_sessions::session_store::Result<Option<tower_sessions::session::Record>> {
let db_id = session_id.0.to_string(); let db_id = session_id.0.to_string();
let query = crate::schema::sessions::dsl::sessions.filter(crate::schema::sessions::dsl::id.eq(db_id)); let query = crate::schema::sessions::dsl::sessions
.filter(crate::schema::sessions::dsl::id.eq(db_id));
let mut connection = crate::db_connection().await; let mut connection = crate::db_connection().await;
@@ -79,7 +100,9 @@ impl tower_sessions::SessionStore for DieselStore {
if result.len() > 1 { if result.len() > 1 {
tracing::error!("Found more than 1 result"); tracing::error!("Found more than 1 result");
return Err(tower_sessions::session_store::Error::Backend("Found more than 1 result".to_string())); return Err(tower_sessions::session_store::Error::Backend(
"Found more than 1 result".to_string(),
));
} }
if result.is_empty() { if result.is_empty() {
@@ -90,9 +113,13 @@ impl tower_sessions::SessionStore for DieselStore {
let data = { let data = {
let mut tmp = HashMap::<String, _>::new(); let mut tmp = HashMap::<String, _>::new();
tmp.insert(crate::UserSession::KEY.to_string(), serde_json::to_value(&crate::UserSessionData { tmp.insert(
steam_id: result.steamid.map(|s| s.parse().ok()).flatten(), crate::UserSession::KEY.to_string(),
}).unwrap()); serde_json::to_value(&crate::UserSessionData {
steam_id: result.steamid.map(|s| s.parse().ok()).flatten(),
})
.unwrap(),
);
tmp tmp
}; };
@@ -103,13 +130,20 @@ impl tower_sessions::SessionStore for DieselStore {
})) }))
} }
async fn delete(&self,session_id: &tower_sessions::session::Id) -> tower_sessions::session_store::Result<()> { async fn delete(
&self,
session_id: &tower_sessions::session::Id,
) -> tower_sessions::session_store::Result<()> {
let db_id = session_id.0.to_string(); let db_id = session_id.0.to_string();
let query = crate::schema::sessions::dsl::sessions.filter(crate::schema::sessions::dsl::id.eq(db_id)); let query = crate::schema::sessions::dsl::sessions
.filter(crate::schema::sessions::dsl::id.eq(db_id));
let mut connection = crate::db_connection().await; let mut connection = crate::db_connection().await;
diesel::dsl::delete(query).execute(&mut connection).await.unwrap(); diesel::dsl::delete(query)
.execute(&mut connection)
.await
.unwrap();
Ok(()) Ok(())
} }

View File

@@ -2,7 +2,7 @@ pub mod models;
pub mod schema; pub mod schema;
mod usersession; mod usersession;
pub use usersession::{UserSessionData, UserSession}; pub use usersession::{UserSession, UserSessionData};
pub mod diesel_sessionstore; pub mod diesel_sessionstore;
@@ -13,10 +13,15 @@ pub async fn db_connection() -> diesel_async::AsyncPgConnection {
let database_url = std::env::var("DATABASE_URL").expect("'DATABASE_URL' must be set"); let database_url = std::env::var("DATABASE_URL").expect("'DATABASE_URL' must be set");
diesel_async::AsyncPgConnection::establish(&database_url).await.unwrap_or_else(|e| panic!("Error connecting to {} - {:?}", database_url, e)) diesel_async::AsyncPgConnection::establish(&database_url)
.await
.unwrap_or_else(|e| panic!("Error connecting to {} - {:?}", database_url, e))
} }
pub async fn get_demo_from_upload(name: &str, mut form: axum::extract::Multipart) -> Option<axum::body::Bytes> { pub async fn get_demo_from_upload(
name: &str,
mut form: axum::extract::Multipart,
) -> Option<axum::body::Bytes> {
while let Ok(field) = form.next_field().await { while let Ok(field) = form.next_field().await {
let field = match field { let field = match field {
Some(f) => f, Some(f) => f,
@@ -36,49 +41,23 @@ pub async fn get_demo_from_upload(name: &str, mut form: axum::extract::Multipart
} }
pub mod api; pub mod api;
pub mod steam_api { pub mod steam_api;
use serde::Deserialize;
pub struct Client {
http: reqwest::Client,
api_key: String,
}
#[derive(Debug, Deserialize)]
struct Response<T> {
response: T,
}
impl Client {
pub fn new<IS>(api_key: IS) -> Self where IS: Into<String> {
Self {
http: reqwest::Client::new(),
api_key: api_key.into(),
}
}
pub async fn get<T>(&self, path: &str, args: &[(&str, &str)]) -> Result<T, ()> where T: serde::de::DeserializeOwned {
let response = self.http.get(path).query(&[("key", &self.api_key)]).query(args).send().await.map_err(|e| ())?;
if !response.status().is_success() {
dbg!(&response);
return Err(());
}
response.json::<Response<T>>().await.map(|r| r.response).map_err(|e| ())
}
}
}
#[tracing::instrument(skip(upload_folder, base_analysis_tx))] #[tracing::instrument(skip(upload_folder, base_analysis_tx))]
pub async fn run_api<UP>(upload_folder: UP, base_analysis_tx: tokio::sync::mpsc::UnboundedSender<analysis::AnalysisInput>) where UP: Into<std::path::PathBuf> { pub async fn run_api<UP>(
upload_folder: UP,
base_analysis_tx: tokio::sync::mpsc::UnboundedSender<analysis::AnalysisInput>,
) where
UP: Into<std::path::PathBuf>,
{
let upload_folder: std::path::PathBuf = upload_folder.into(); let upload_folder: std::path::PathBuf = upload_folder.into();
let session_store = crate::diesel_sessionstore::DieselStore::new(); let session_store = crate::diesel_sessionstore::DieselStore::new();
let session_layer = tower_sessions::SessionManagerLayer::new(session_store) let session_layer = tower_sessions::SessionManagerLayer::new(session_store)
.with_secure(false) .with_secure(false)
.with_expiry(tower_sessions::Expiry::OnInactivity( .with_expiry(tower_sessions::Expiry::OnInactivity(time::Duration::hours(
time::Duration::hours(48), 48,
)); )));
if !tokio::fs::try_exists(&upload_folder).await.unwrap_or(false) { if !tokio::fs::try_exists(&upload_folder).await.unwrap_or(false) {
tokio::fs::create_dir_all(&upload_folder).await.unwrap(); tokio::fs::create_dir_all(&upload_folder).await.unwrap();
@@ -87,42 +66,54 @@ pub async fn run_api<UP>(upload_folder: UP, base_analysis_tx: tokio::sync::mpsc:
let router = axum::Router::new() let router = axum::Router::new()
.nest("/api/", crate::api::router(base_analysis_tx)) .nest("/api/", crate::api::router(base_analysis_tx))
.layer(session_layer) .layer(session_layer)
.nest_service("/", tower_http::services::ServeDir::new("../frontend/dist/")); .nest_service(
"/",
tower_http::services::ServeDir::new("../frontend/dist/"),
);
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap(); let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
axum::serve(listener, router).await.unwrap(); axum::serve(listener, router).await.unwrap();
} }
#[tracing::instrument] #[tracing::instrument]
pub async fn run_analysis(mut base_analysis_rx: tokio::sync::mpsc::UnboundedReceiver<analysis::AnalysisInput>) { pub async fn run_analysis(
mut base_analysis_rx: tokio::sync::mpsc::UnboundedReceiver<analysis::AnalysisInput>,
) {
use diesel::prelude::*; use diesel::prelude::*;
use diesel_async::{RunQueryDsl, AsyncConnection}; use diesel_async::{AsyncConnection, RunQueryDsl};
while let Some(input) = base_analysis_rx.recv().await { while let Some(input) = base_analysis_rx.recv().await {
let demo_id = input.demoid; let demo_id = input.demoid;
let result = tokio::task::spawn_blocking(move || { let result = tokio::task::spawn_blocking(move || crate::analysis::analyse_base(input))
crate::analysis::analyse_base(input) .await
}).await.unwrap(); .unwrap();
dbg!(&result); dbg!(&result);
let mut db_con = crate::db_connection().await; let mut db_con = crate::db_connection().await;
let store_info_query = diesel::dsl::insert_into(crate::schema::demo_info::dsl::demo_info).values(crate::models::DemoInfo {
demo_id,
map: result.map,
});
let update_process_info = diesel::dsl::update(crate::schema::processing_status::dsl::processing_status).set(crate::schema::processing_status::dsl::info.eq(1)).filter(crate::schema::processing_status::dsl::demo_id.eq(demo_id));
tracing::trace!(?store_info_query, "Store demo info query"); let store_info_query = diesel::dsl::insert_into(crate::schema::demo_info::dsl::demo_info)
tracing::trace!(?update_process_info, "Update processing info query"); .values(crate::models::DemoInfo {
demo_id,
map: result.map,
});
let update_process_info =
diesel::dsl::update(crate::schema::processing_status::dsl::processing_status)
.set(crate::schema::processing_status::dsl::info.eq(1))
.filter(crate::schema::processing_status::dsl::demo_id.eq(demo_id));
db_con.transaction::<'_, '_, '_, _, diesel::result::Error, _>(|conn| Box::pin(async move { tracing::trace!(?store_info_query, "Store demo info query");
store_info_query.execute(conn).await.map(|e| ())?; tracing::trace!(?update_process_info, "Update processing info query");
update_process_info.execute(conn).await.map(|e| ())?;
Ok(()) db_con
})).await; .transaction::<'_, '_, '_, _, diesel::result::Error, _>(|conn| {
Box::pin(async move {
} store_info_query.execute(conn).await.map(|e| ())?;
update_process_info.execute(conn).await.map(|e| ())?;
Ok(())
})
})
.await;
}
} }

View File

@@ -2,7 +2,8 @@ use tracing_subscriber::prelude::__tracing_subscriber_SubscriberExt;
static UPLOAD_FOLDER: &str = "uploads/"; static UPLOAD_FOLDER: &str = "uploads/";
const MIGRATIONS: diesel_async_migrations::EmbeddedMigrations = diesel_async_migrations::embed_migrations!("../migrations/"); const MIGRATIONS: diesel_async_migrations::EmbeddedMigrations =
diesel_async_migrations::embed_migrations!("../migrations/");
async fn run_migrations(connection: &mut diesel_async::AsyncPgConnection) { async fn run_migrations(connection: &mut diesel_async::AsyncPgConnection) {
MIGRATIONS.run_pending_migrations(connection).await.unwrap(); MIGRATIONS.run_pending_migrations(connection).await.unwrap();
@@ -23,7 +24,8 @@ async fn main() {
run_migrations(&mut backend::db_connection().await).await; run_migrations(&mut backend::db_connection().await).await;
tracing::info!("Completed Migrations"); tracing::info!("Completed Migrations");
let (base_analysis_tx, base_analysis_rx) = tokio::sync::mpsc::unbounded_channel::<backend::analysis::AnalysisInput>(); let (base_analysis_tx, base_analysis_rx) =
tokio::sync::mpsc::unbounded_channel::<backend::analysis::AnalysisInput>();
let mut component_set = tokio::task::JoinSet::new(); let mut component_set = tokio::task::JoinSet::new();

View File

@@ -39,10 +39,4 @@ diesel::table! {
diesel::joinable!(demo_info -> demos (demo_id)); diesel::joinable!(demo_info -> demos (demo_id));
diesel::joinable!(processing_status -> demos (demo_id)); diesel::joinable!(processing_status -> demos (demo_id));
diesel::allow_tables_to_appear_in_same_query!( diesel::allow_tables_to_appear_in_same_query!(demo_info, demos, processing_status, sessions, users,);
demo_info,
demos,
processing_status,
sessions,
users,
);

47
backend/src/steam_api.rs Normal file
View File

@@ -0,0 +1,47 @@
use serde::Deserialize;
pub struct Client {
http: reqwest::Client,
api_key: String,
}
#[derive(Debug, Deserialize)]
struct Response<T> {
response: T,
}
impl Client {
pub fn new<IS>(api_key: IS) -> Self
where
IS: Into<String>,
{
Self {
http: reqwest::Client::new(),
api_key: api_key.into(),
}
}
pub async fn get<T>(&self, path: &str, args: &[(&str, &str)]) -> Result<T, ()>
where
T: serde::de::DeserializeOwned,
{
let response = self
.http
.get(path)
.query(&[("key", &self.api_key)])
.query(args)
.send()
.await
.map_err(|e| ())?;
if !response.status().is_success() {
dbg!(&response);
return Err(());
}
response
.json::<Response<T>>()
.await
.map(|r| r.response)
.map_err(|e| ())
}
}

View File

@@ -6,16 +6,20 @@ pub fn demo() -> impl leptos::IntoView {
let params = leptos_router::use_params_map(); let params = leptos_router::use_params_map();
let id = move || params.with(|params| params.get("id").cloned().unwrap_or_default()); let id = move || params.with(|params| params.get("id").cloned().unwrap_or_default());
let demo_info = create_resource(|| (), move |_| async move { let demo_info = create_resource(
let res = reqwasm::http::Request::get(&format!("/api/demos/{}/info", id())).send().await.unwrap(); || (),
res.json::<common::DemoInfo>().await.unwrap() move |_| async move {
}); let res = reqwasm::http::Request::get(&format!("/api/demos/{}/info", id()))
.send()
.await
.unwrap();
res.json::<common::DemoInfo>().await.unwrap()
},
);
let map = move || { let map = move || match demo_info.get() {
match demo_info.get() { Some(v) => v.map.clone(),
Some(v) => v.map.clone(), None => String::new(),
None => String::new(),
}
}; };
let selected_tab = move || { let selected_tab = move || {
@@ -24,7 +28,7 @@ pub fn demo() -> impl leptos::IntoView {
let trailing = loc_path.split('/').last(); let trailing = loc_path.split('/').last();
trailing.unwrap_or("/").to_owned() trailing.unwrap_or("/").to_owned()
}; };
let style = stylers::style! { let style = stylers::style! {
"Demo", "Demo",
.analysis_bar { .analysis_bar {

View File

@@ -25,7 +25,10 @@ pub fn demo_list_entry(demo: common::BaseDemoInfo) -> impl leptos::IntoView {
} }
#[leptos::component] #[leptos::component]
pub fn upload_demo(shown: ReadSignal<DemoUploadStatus>, update_shown: WriteSignal<DemoUploadStatus>) -> impl leptos::IntoView { pub fn upload_demo(
shown: ReadSignal<DemoUploadStatus>,
update_shown: WriteSignal<DemoUploadStatus>,
) -> impl leptos::IntoView {
use leptos_router::Form; use leptos_router::Form;
let style = stylers::style! { let style = stylers::style! {
@@ -69,11 +72,17 @@ pub fn upload_demo(shown: ReadSignal<DemoUploadStatus>, update_shown: WriteSigna
#[leptos::component] #[leptos::component]
pub fn homepage() -> impl leptos::IntoView { pub fn homepage() -> impl leptos::IntoView {
let demo_data = create_resource(|| (), |_| async move { let demo_data = create_resource(
let res = reqwasm::http::Request::get("/api/demos/list").send().await.unwrap(); || (),
let demos: Vec<common::BaseDemoInfo> = res.json().await.unwrap(); |_| async move {
demos let res = reqwasm::http::Request::get("/api/demos/list")
}); .send()
.await
.unwrap();
let demos: Vec<common::BaseDemoInfo> = res.json().await.unwrap();
demos
},
);
view! { view! {
<div> <div>

View File

@@ -1,39 +1,42 @@
use leptos::*; use leptos::*;
use leptos_router::*; use leptos_router::*;
use frontend::{UploadDemo, TopBar, Homepage, Demo}; use frontend::{Demo, Homepage, TopBar, UploadDemo};
async fn load_demos() -> Vec<common::BaseDemoInfo> { async fn load_demos() -> Vec<common::BaseDemoInfo> {
let res = reqwasm::http::Request::get("/api/demos/list").send().await.unwrap(); let res = reqwasm::http::Request::get("/api/demos/list")
.send()
.await
.unwrap();
let demos: Vec<common::BaseDemoInfo> = res.json().await.unwrap(); let demos: Vec<common::BaseDemoInfo> = res.json().await.unwrap();
demos demos
} }
fn main() { fn main() {
let async_data = create_resource(|| (), |_| async move { let async_data = create_resource(|| (), |_| async move { load_demos().await });
load_demos().await
});
let (upload_demo_read, upload_demo_write) = create_signal(frontend::DemoUploadStatus::Hidden); let (upload_demo_read, upload_demo_write) = create_signal(frontend::DemoUploadStatus::Hidden);
mount_to_body(move || view! { mount_to_body(move || {
<Router> view! {
<nav> <Router>
<TopBar update_demo_visible=upload_demo_write /> <nav>
</nav> <TopBar update_demo_visible=upload_demo_write />
<main> </nav>
<UploadDemo shown=upload_demo_read update_shown=upload_demo_write /> <main>
<UploadDemo shown=upload_demo_read update_shown=upload_demo_write />
<Routes> <Routes>
<Route path="/" view=Homepage /> <Route path="/" view=Homepage />
<Route path="/demo/:id" view=Demo> <Route path="/demo/:id" view=Demo>
<Route path="perround" view=frontend::demo::PerRound /> <Route path="perround" view=frontend::demo::PerRound />
<Route path="scoreboard" view=frontend::demo::Scoreboard /> <Route path="scoreboard" view=frontend::demo::Scoreboard />
<Route path="" view=frontend::demo::Scoreboard /> <Route path="" view=frontend::demo::Scoreboard />
</Route> </Route>
</Routes> </Routes>
</main> </main>
</Router> </Router>
}
}) })
} }

View File

@@ -5,12 +5,19 @@ use crate::DemoUploadStatus;
#[leptos::component] #[leptos::component]
fn steam_login(height: &'static str, width: &'static str) -> impl leptos::IntoView { fn steam_login(height: &'static str, width: &'static str) -> impl leptos::IntoView {
let user_status = create_resource(|| (), |_| async move { let user_status = create_resource(
let res = reqwasm::http::Request::get("/api/user/status").send().await.unwrap(); || (),
res.json::<common::UserStatus>().await.map_err(|e| ()) |_| async move {
}); let res = reqwasm::http::Request::get("/api/user/status")
.send()
.await
.unwrap();
res.json::<common::UserStatus>().await.map_err(|e| ())
},
);
let tmp = move || match user_status.get() { let tmp = move || {
match user_status.get() {
Some(Ok(user)) => Some(Ok(user)) =>
view! { view! {
<p>{user.name}</p> <p>{user.name}</p>
@@ -21,6 +28,7 @@ fn steam_login(height: &'static str, width: &'static str) -> impl leptos::IntoVi
<img src="https://community.akamai.steamstatic.com/public/images/signinthroughsteam/sits_01.png" alt="Steam Login" style=format!("height: {height}; width: {width}") /> <img src="https://community.akamai.steamstatic.com/public/images/signinthroughsteam/sits_01.png" alt="Steam Login" style=format!("height: {height}; width: {width}") />
</a> </a>
}.into_view(), }.into_view(),
}
}; };
view! { view! {
@@ -66,7 +74,7 @@ pub fn top_bar(update_demo_visible: WriteSignal<DemoUploadStatus>) -> impl lepto
<A href="/"> <A href="/">
<p class="logo">Knifer</p> <p class="logo">Knifer</p>
</A> </A>
<div class="elem" style="grid-column-start: 3"> <div class="elem" style="grid-column-start: 3">
<button on:click=move |_| update_demo_visible(DemoUploadStatus::Shown)> <button on:click=move |_| update_demo_visible(DemoUploadStatus::Shown)>
Upload Demo Upload Demo