Fix some formatting and improve structure
This commit is contained in:
@@ -10,18 +10,16 @@ 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(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,9 +120,13 @@ 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);
|
||||||
@@ -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())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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 {
|
let store_info_query = diesel::dsl::insert_into(crate::schema::demo_info::dsl::demo_info)
|
||||||
demo_id,
|
.values(crate::models::DemoInfo {
|
||||||
map: result.map,
|
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));
|
});
|
||||||
|
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");
|
tracing::trace!(?store_info_query, "Store demo info query");
|
||||||
tracing::trace!(?update_process_info, "Update processing info query");
|
tracing::trace!(?update_process_info, "Update processing info query");
|
||||||
|
|
||||||
db_con.transaction::<'_, '_, '_, _, diesel::result::Error, _>(|conn| Box::pin(async move {
|
db_con
|
||||||
store_info_query.execute(conn).await.map(|e| ())?;
|
.transaction::<'_, '_, '_, _, diesel::result::Error, _>(|conn| {
|
||||||
update_process_info.execute(conn).await.map(|e| ())?;
|
Box::pin(async move {
|
||||||
Ok(())
|
store_info_query.execute(conn).await.map(|e| ())?;
|
||||||
})).await;
|
update_process_info.execute(conn).await.map(|e| ())?;
|
||||||
|
Ok(())
|
||||||
}
|
})
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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();
|
||||||
|
|
||||||
|
|||||||
@@ -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
47
backend/src/steam_api.rs
Normal 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| ())
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -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 || {
|
||||||
|
|||||||
@@ -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>
|
||||||
|
|||||||
@@ -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>
|
||||||
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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! {
|
||||||
|
|||||||
Reference in New Issue
Block a user