#![warn(clippy::too_many_arguments)] use rocket::{ request::{Form, Request}, response::{self, Responder}, }; use rocket_contrib::json::Json; use plume_common::utils::random_hex; use plume_models::{api_tokens::*, apps::App, db_conn::DbConn, users::User, Error}; type Api = Result, ApiError>; #[derive(Debug)] pub struct ApiError(Error); impl From for ApiError { fn from(err: Error) -> ApiError { ApiError(err) } } impl<'r> Responder<'r> for ApiError { fn respond_to(self, req: &Request<'_>) -> response::Result<'r> { match self.0 { Error::NotFound => Json(json!({ "error": "Not found" })) .respond_to(req), Error::Unauthorized => Json(json!({ "error": "You are not authorized to access this resource" })) .respond_to(req), _ => Json(json!({ "error": "Server error" })) .respond_to(req), } } } #[derive(FromForm)] pub struct OAuthRequest { client_id: String, client_secret: String, password: String, username: String, scopes: String, } #[get("/oauth2?")] pub fn oauth(query: Form, conn: DbConn) -> Result, ApiError> { let app = App::find_by_client_id(&conn, &query.client_id)?; if app.client_secret == query.client_secret { if let Ok(user) = User::login(&conn, &query.username, &query.password) { let token = ApiToken::insert( &conn, NewApiToken { app_id: app.id, user_id: user.id, value: random_hex(), scopes: query.scopes.clone(), }, )?; Ok(Json(json!({ "token": token.value }))) } else { Ok(Json(json!({ "error": "Invalid credentials" }))) } } else { Ok(Json(json!({ "error": "Invalid client_secret" }))) } } pub mod apps; pub mod authorization; pub mod posts;