Piotr OsiewiczRust 1.78 (#11314)1a9b0532 years agoHistory
240 lines · 6.9 KB · Rust
RawDownload
pub mod events;pub mod extensions;pub mod ips_file;pub mod slack; use crate::{    auth,    db::{ContributorSelector, User, UserId},    rpc, AppState, Error, Result,};use anyhow::anyhow;use axum::{    body::Body,    extract::{self, Path, Query},    http::{self, Request, StatusCode},    middleware::{self, Next},    response::IntoResponse,    routing::{get, post},    Extension, Json, Router,};use axum_extra::response::ErasedJson;use chrono::SecondsFormat;use serde::{Deserialize, Serialize};use std::sync::Arc;use tower::ServiceBuilder; pub use extensions::fetch_extensions_from_blob_store_periodically; pub fn routes(rpc_server: Option<Arc<rpc::Server>>, state: Arc<AppState>) -> Router<(), Body> {    Router::new()        .route("/user", get(get_authenticated_user))        .route("/users/:id/access_tokens", post(create_access_token))        .route("/rpc_server_snapshot", get(get_rpc_server_snapshot))        .route("/contributors", get(get_contributors).post(add_contributor))        .route("/contributor", get(check_is_contributor))        .layer(            ServiceBuilder::new()                .layer(Extension(state))                .layer(Extension(rpc_server))                .layer(middleware::from_fn(validate_api_token)),        )} pub async fn validate_api_token<B>(req: Request<B>, next: Next<B>) -> impl IntoResponse {    let token = req        .headers()        .get(http::header::AUTHORIZATION)        .and_then(|header| header.to_str().ok())        .ok_or_else(|| {            Error::Http(                StatusCode::BAD_REQUEST,                "missing authorization header".to_string(),            )        })?        .strip_prefix("token ")        .ok_or_else(|| {            Error::Http(                StatusCode::BAD_REQUEST,                "invalid authorization header".to_string(),            )        })?;     let state = req.extensions().get::<Arc<AppState>>().unwrap();     if token != state.config.api_token {        Err(Error::Http(            StatusCode::UNAUTHORIZED,            "invalid authorization token".to_string(),        ))?    }     Ok::<_, Error>(next.run(req).await)} #[derive(Debug, Deserialize)]struct AuthenticatedUserParams {    github_user_id: Option<i32>,    github_login: String,    github_email: Option<String>,} #[derive(Debug, Serialize)]struct AuthenticatedUserResponse {    user: User,    metrics_id: String,} async fn get_authenticated_user(    Query(params): Query<AuthenticatedUserParams>,    Extension(app): Extension<Arc<AppState>>,) -> Result<Json<AuthenticatedUserResponse>> {    let initial_channel_id = app.config.auto_join_channel_id;     let user = app        .db        .get_or_create_user_by_github_account(            &params.github_login,            params.github_user_id,            params.github_email.as_deref(),            initial_channel_id,        )        .await?;    let metrics_id = app.db.get_user_metrics_id(user.id).await?;    return Ok(Json(AuthenticatedUserResponse { user, metrics_id }));} #[derive(Deserialize, Debug)]struct CreateUserParams {    github_user_id: i32,    github_login: String,    email_address: String,    email_confirmation_code: Option<String>,    #[serde(default)]    admin: bool,    #[serde(default)]    invite_count: i32,} async fn get_rpc_server_snapshot(    Extension(rpc_server): Extension<Option<Arc<rpc::Server>>>,) -> Result<ErasedJson> {    let Some(rpc_server) = rpc_server else {        return Err(Error::Internal(anyhow!("rpc server is not available")));    };     Ok(ErasedJson::pretty(rpc_server.snapshot().await))} async fn get_contributors(Extension(app): Extension<Arc<AppState>>) -> Result<Json<Vec<String>>> {    Ok(Json(app.db.get_contributors().await?))} #[derive(Debug, Deserialize)]struct CheckIsContributorParams {    github_user_id: Option<i32>,    github_login: Option<String>,} impl CheckIsContributorParams {    fn as_contributor_selector(self) -> Result<ContributorSelector> {        if let Some(github_user_id) = self.github_user_id {            return Ok(ContributorSelector::GitHubUserId { github_user_id });        }         if let Some(github_login) = self.github_login {            return Ok(ContributorSelector::GitHubLogin { github_login });        }         Err(anyhow!(            "must be one of `github_user_id` or `github_login`."        ))?    }} #[derive(Debug, Serialize)]struct CheckIsContributorResponse {    signed_at: Option<String>,} async fn check_is_contributor(    Extension(app): Extension<Arc<AppState>>,    Query(params): Query<CheckIsContributorParams>,) -> Result<Json<CheckIsContributorResponse>> {    let params = params.as_contributor_selector()?;    Ok(Json(CheckIsContributorResponse {        signed_at: app            .db            .get_contributor_sign_timestamp(&params)            .await?            .map(|ts| ts.and_utc().to_rfc3339_opts(SecondsFormat::Millis, true)),    }))} async fn add_contributor(    Extension(app): Extension<Arc<AppState>>,    extract::Json(params): extract::Json<AuthenticatedUserParams>,) -> Result<()> {    let initial_channel_id = app.config.auto_join_channel_id;    app.db        .add_contributor(            &params.github_login,            params.github_user_id,            params.github_email.as_deref(),            initial_channel_id,        )        .await} #[derive(Deserialize)]struct CreateAccessTokenQueryParams {    public_key: String,    impersonate: Option<String>,} #[derive(Serialize)]struct CreateAccessTokenResponse {    user_id: UserId,    encrypted_access_token: String,} async fn create_access_token(    Path(user_id): Path<UserId>,    Query(params): Query<CreateAccessTokenQueryParams>,    Extension(app): Extension<Arc<AppState>>,) -> Result<Json<CreateAccessTokenResponse>> {    let user = app        .db        .get_user_by_id(user_id)        .await?        .ok_or_else(|| anyhow!("user not found"))?;     let mut impersonated_user_id = None;    if let Some(impersonate) = params.impersonate {        if user.admin {            if let Some(impersonated_user) = app.db.get_user_by_github_login(&impersonate).await? {                impersonated_user_id = Some(impersonated_user.id);            } else {                return Err(Error::Http(                    StatusCode::UNPROCESSABLE_ENTITY,                    format!("user {impersonate} does not exist"),                ));            }        } else {            return Err(Error::Http(                StatusCode::UNAUTHORIZED,                "you do not have permission to impersonate other users".to_string(),            ));        }    }     let access_token =        auth::create_access_token(app.db.as_ref(), user_id, impersonated_user_id).await?;    let encrypted_access_token =        auth::encrypt_access_token(&access_token, params.public_key.clone())?;     Ok(Json(CreateAccessTokenResponse {        user_id: impersonated_user_id.unwrap_or(user_id),        encrypted_access_token,    }))}