mirror of
https://codeberg.org/demostf/frontend.git
synced 2026-06-03 18:24:12 +02:00
updates
This commit is contained in:
parent
780d5fcf50
commit
de3a2f6009
7 changed files with 571 additions and 353 deletions
|
|
@ -1,7 +1,6 @@
|
|||
use maud::Render;
|
||||
use sea_query::Value;
|
||||
use serde::{de::Error, Deserialize, Deserializer, Serialize};
|
||||
use sqlx::database::HasValueRef;
|
||||
use sqlx::error::BoxDynError;
|
||||
use sqlx::{Database, Decode, Type};
|
||||
use std::borrow::Cow;
|
||||
|
|
@ -83,7 +82,7 @@ where
|
|||
DB: Database,
|
||||
&'r str: Decode<'r, DB>,
|
||||
{
|
||||
fn decode(value: <DB as HasValueRef<'r>>::ValueRef) -> Result<Self, BoxDynError> {
|
||||
fn decode(value: DB::ValueRef<'r>) -> Result<Self, BoxDynError> {
|
||||
let str = <&str as Decode<DB>>::decode(value)?;
|
||||
Ok(str.parse().unwrap())
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
use crate::data::steam_id::SteamId;
|
||||
use crate::{Error, Result};
|
||||
use maud::Render;
|
||||
use rand::distributions::Alphanumeric;
|
||||
use rand::distr::Alphanumeric;
|
||||
use rand::Rng;
|
||||
use reqwest::get;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
|
@ -79,7 +79,7 @@ impl User {
|
|||
})
|
||||
} else {
|
||||
let profile = Self::fetch(&steam_id).await?;
|
||||
let token: String = rand::thread_rng()
|
||||
let token: String = rand::rng()
|
||||
.sample_iter(&Alphanumeric)
|
||||
.take(64)
|
||||
.map(char::from)
|
||||
|
|
|
|||
|
|
@ -2,6 +2,7 @@ use axum::http::StatusCode;
|
|||
use axum::response::{IntoResponse, Response};
|
||||
use config::ConfigError;
|
||||
use opentelemetry::trace::TraceError;
|
||||
use opentelemetry_sdk::logs::LogError;
|
||||
use secretfile::SecretError;
|
||||
use tracing_subscriber::util::TryInitError;
|
||||
|
||||
|
|
@ -34,6 +35,8 @@ pub enum SetupError {
|
|||
#[error(transparent)]
|
||||
Tracing(#[from] TraceError),
|
||||
#[error(transparent)]
|
||||
Log(#[from] LogError),
|
||||
#[error(transparent)]
|
||||
TracingSubscriber(#[from] TryInitError),
|
||||
#[error(transparent)]
|
||||
Config(#[from] ConfigError),
|
||||
|
|
|
|||
63
src/main.rs
63
src/main.rs
|
|
@ -46,8 +46,9 @@ use include_dir::{include_dir, Dir};
|
|||
use maud::{Markup, Render};
|
||||
use opentelemetry::trace::TracerProvider;
|
||||
use opentelemetry::KeyValue;
|
||||
use opentelemetry_otlp::WithExportConfig;
|
||||
use opentelemetry_sdk::{runtime, trace, Resource};
|
||||
use opentelemetry_otlp::{SpanExporter, WithExportConfig, WithTonicConfig};
|
||||
use opentelemetry_sdk::trace::SdkTracerProvider;
|
||||
use opentelemetry_sdk::Resource;
|
||||
use secretfile::load;
|
||||
use sqlx::PgPool;
|
||||
use std::convert::Infallible;
|
||||
|
|
@ -104,8 +105,8 @@ fn setup() -> Result<Config, SetupError> {
|
|||
{
|
||||
let mut tls_config = ClientTlsConfig::new().with_webpki_roots();
|
||||
|
||||
let mut otlp_exporter = opentelemetry_otlp::new_exporter()
|
||||
.tonic()
|
||||
let mut otlp_exporter = SpanExporter::builder()
|
||||
.with_tonic()
|
||||
.with_endpoint(&tracing_cfg.endpoint);
|
||||
|
||||
if let Some(tracing_ident) = tracing_cfg.tls.as_ref().map(|tracing_tls_cfg| {
|
||||
|
|
@ -117,13 +118,14 @@ fn setup() -> Result<Config, SetupError> {
|
|||
}
|
||||
otlp_exporter = otlp_exporter.with_tls_config(tls_config);
|
||||
|
||||
let tracer = opentelemetry_otlp::new_pipeline()
|
||||
.tracing()
|
||||
.with_exporter(otlp_exporter)
|
||||
.with_trace_config(trace::Config::default().with_resource(Resource::new(vec![
|
||||
KeyValue::new("service.name", "demos.tf"),
|
||||
])))
|
||||
.install_batch(runtime::Tokio)?
|
||||
let tracer = SdkTracerProvider::builder()
|
||||
.with_resource(
|
||||
Resource::builder()
|
||||
.with_attribute(KeyValue::new("service.name", "demos.tf"))
|
||||
.build(),
|
||||
)
|
||||
.with_batch_exporter(otlp_exporter.build()?)
|
||||
.build()
|
||||
.tracer("demos.tf");
|
||||
Some(tracing_opentelemetry::layer().with_tracer(tracer))
|
||||
} else {
|
||||
|
|
@ -164,8 +166,8 @@ async fn main() -> Result<()> {
|
|||
|
||||
let app = Router::new()
|
||||
.route("/", get(index))
|
||||
.route("/uploads/:uploader", get(uploads))
|
||||
.route("/profiles/:uploader", get(profiles))
|
||||
.route("/uploads/{uploader}", get(uploads))
|
||||
.route("/profiles/{uploader}", get(profiles))
|
||||
.route(GlobalStyle::route(), get(serve_asset::<GlobalStyle>))
|
||||
.route(
|
||||
ClassIconsStyle::route(),
|
||||
|
|
@ -198,9 +200,9 @@ async fn main() -> Result<()> {
|
|||
.route("/upload", get(upload))
|
||||
.route("/viewer", get(viewer))
|
||||
.route("/edit", get(edit))
|
||||
.route("/viewer/:id", get(viewer))
|
||||
.route("/:id", get(demo))
|
||||
.route("/images/kill_icons/:icon", get(kill_icons))
|
||||
.route("/viewer/{id}", get(viewer))
|
||||
.route("/{id}", get(demo))
|
||||
.route("/images/kill_icons/{icon}", get(kill_icons))
|
||||
.layer(
|
||||
TraceLayer::new_for_http().make_span_with(|request: &Request<_>| {
|
||||
let matched_path = request
|
||||
|
|
@ -298,12 +300,12 @@ async fn main() -> Result<()> {
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn index(
|
||||
State(app): State<Arc<App>>,
|
||||
session: SessionData,
|
||||
filter: Option<Query<Filter>>,
|
||||
Query(filter): Query<Filter>,
|
||||
) -> Result<Markup> {
|
||||
let filter = filter.map(|filter| filter.0).unwrap_or_default();
|
||||
let demos = ListDemo::list(&app.connection, filter).await?;
|
||||
Ok(render(
|
||||
Index {
|
||||
|
|
@ -316,6 +318,7 @@ async fn index(
|
|||
}
|
||||
|
||||
#[instrument(skip(_app))]
|
||||
#[axum::debug_handler]
|
||||
async fn about(State(_app): State<Arc<App>>, session: SessionData) -> Result<Markup> {
|
||||
Ok(render(
|
||||
AboutPage {
|
||||
|
|
@ -326,6 +329,7 @@ async fn about(State(_app): State<Arc<App>>, session: SessionData) -> Result<Mar
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn api(State(app): State<Arc<App>>, session: SessionData) -> Result<Markup> {
|
||||
Ok(render(
|
||||
ApiPage {
|
||||
|
|
@ -337,6 +341,7 @@ async fn api(State(app): State<Arc<App>>, session: SessionData) -> Result<Markup
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn demo(
|
||||
State(app): State<Arc<App>>,
|
||||
Path(id): Path<String>,
|
||||
|
|
@ -349,6 +354,7 @@ async fn demo(
|
|||
Ok(render(DemoPage { demo }, session))
|
||||
}
|
||||
|
||||
#[axum::debug_handler]
|
||||
async fn login_callback(
|
||||
State(app): State<Arc<App>>,
|
||||
RawQuery(query): RawQuery,
|
||||
|
|
@ -387,6 +393,7 @@ async fn login_callback(
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn login(State(app): State<Arc<App>>) -> impl IntoResponse {
|
||||
(
|
||||
StatusCode::FOUND,
|
||||
|
|
@ -398,6 +405,7 @@ async fn login(State(app): State<Arc<App>>) -> impl IntoResponse {
|
|||
}
|
||||
|
||||
#[instrument(skip(app, cookie))]
|
||||
#[axum::debug_handler]
|
||||
async fn logout(
|
||||
State(app): State<Arc<App>>,
|
||||
cookie: Option<TypedHeader<Cookie>>,
|
||||
|
|
@ -424,6 +432,7 @@ async fn logout(
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn upload(State(app): State<Arc<App>>, session: SessionData) -> impl IntoResponse {
|
||||
if let Some(token) = session.token() {
|
||||
render(
|
||||
|
|
@ -444,20 +453,20 @@ async fn upload(State(app): State<Arc<App>>, session: SessionData) -> impl IntoR
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
async fn demo_list(State(app): State<Arc<App>>, filter: Option<Query<Filter>>) -> Result<Markup> {
|
||||
let filter = filter.map(|filter| filter.0).unwrap_or_default();
|
||||
#[axum::debug_handler]
|
||||
async fn demo_list(State(app): State<Arc<App>>, Query(filter): Query<Filter>) -> Result<Markup> {
|
||||
let demos = ListDemo::list(&app.connection, filter).await?;
|
||||
Ok(DemoList { demos: &demos }.render())
|
||||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn uploads(
|
||||
State(app): State<Arc<App>>,
|
||||
session: SessionData,
|
||||
filter: Option<Query<Filter>>,
|
||||
Query(mut filter): Query<Filter>,
|
||||
Path(uploader): Path<SteamId>,
|
||||
) -> Result<Markup> {
|
||||
let mut filter = filter.map(|filter| filter.0).unwrap_or_default();
|
||||
filter.uploader = Some(uploader.clone());
|
||||
|
||||
let demos = ListDemo::list(&app.connection, filter).await?;
|
||||
|
|
@ -476,13 +485,13 @@ async fn uploads(
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn profiles(
|
||||
State(app): State<Arc<App>>,
|
||||
session: SessionData,
|
||||
filter: Option<Query<Filter>>,
|
||||
Query(mut filter): Query<Filter>,
|
||||
Path(profile): Path<SteamId>,
|
||||
) -> Result<Markup> {
|
||||
let mut filter = filter.map(|filter| filter.0).unwrap_or_default();
|
||||
filter.players.push(profile.clone());
|
||||
|
||||
let demos = ListDemo::list(&app.connection, filter).await?;
|
||||
|
|
@ -501,6 +510,7 @@ async fn profiles(
|
|||
}
|
||||
|
||||
#[instrument(skip(app))]
|
||||
#[axum::debug_handler]
|
||||
async fn viewer(
|
||||
State(app): State<Arc<App>>,
|
||||
id: Option<Path<String>>,
|
||||
|
|
@ -526,14 +536,17 @@ async fn viewer(
|
|||
))
|
||||
}
|
||||
|
||||
async fn edit(session: SessionData) -> Result<Markup> {
|
||||
#[axum::debug_handler]
|
||||
async fn edit(State(_app): State<Arc<App>>, session: SessionData) -> Result<Markup> {
|
||||
Ok(render(EditorPage, session))
|
||||
}
|
||||
|
||||
#[axum::debug_handler]
|
||||
async fn handler_404() -> impl IntoResponse {
|
||||
Error::NotFound
|
||||
}
|
||||
|
||||
#[axum::debug_handler]
|
||||
pub async fn kill_icons(path: Path<String>) -> impl IntoResponse {
|
||||
let path = path.as_str();
|
||||
match KILL_ICONS.get_file(path) {
|
||||
|
|
|
|||
|
|
@ -4,11 +4,11 @@ use crate::{App, Result};
|
|||
use async_session::SessionStore as _;
|
||||
use axum::extract::{FromRef, FromRequestParts};
|
||||
use axum::http::request::Parts;
|
||||
use axum::{async_trait, RequestPartsExt};
|
||||
use std::convert::Infallible;
|
||||
use std::sync::Arc;
|
||||
use axum::RequestPartsExt;
|
||||
use axum_extra::headers::Cookie;
|
||||
use axum_extra::TypedHeader;
|
||||
use std::convert::Infallible;
|
||||
use std::sync::Arc;
|
||||
use tracing::debug;
|
||||
|
||||
pub const COOKIE_NAME: &str = "tf_session";
|
||||
|
|
@ -34,7 +34,6 @@ impl SessionData {
|
|||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl<S> FromRequestParts<S> for SessionData
|
||||
where
|
||||
Arc<App>: FromRef<S>,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue