diff --git a/src/callers/login.rs b/src/callers/login.rs index eadd977..fcf80e6 100644 --- a/src/callers/login.rs +++ b/src/callers/login.rs @@ -96,8 +96,7 @@ pub mod response { pub async fn extract( response: axum::response::Response, ) -> Result { - let body = match axum::body::to_bytes(response.into_body(), usize::MAX) - .await { + let body = match axum::body::to_bytes(response.into_body(), usize::MAX).await { Ok(body) => body, Err(err) => { return Err(std::io::Error::other(err)); @@ -201,15 +200,18 @@ pub async fn user_login( Ok(matches) => { if matches { // Create token - let key = textsender_models::envy::environment::get_secret_key() - .value; + let key = + textsender_models::envy::environment::get_secret_key().value; let cst = match token_stuff::create_token(&key, &user.id) { Ok(token) => token, Err(_err) => { - return (axum::http::StatusCode::INTERNAL_SERVER_ERROR, axum::Json(response::LoginResponse { - message: String::from("Error creating token"), - data: Vec::new() - })); + return ( + axum::http::StatusCode::INTERNAL_SERVER_ERROR, + axum::Json(response::LoginResponse { + message: String::from("Error creating token"), + data: Vec::new(), + }), + ); } }; @@ -348,8 +350,8 @@ pub async fn service_user_login( if matches { // Create token println!("Creating token"); - let key = textsender_models::envy::environment::get_secret_key() - .value; + let key = + textsender_models::envy::environment::get_secret_key().value; let cst = token_stuff::create_token(&key, &service_user.id).unwrap(); @@ -454,8 +456,7 @@ pub async fn refresh_token( }), ) } else { - let key = textsender_models::envy::environment::get_secret_key() - .value; + let key = textsender_models::envy::environment::get_secret_key().value; if token_stuff::verify_token(&key, &payload.access_token) { match token_stuff::extract_id_from_token(&key, &payload.access_token) { Ok(id) => { diff --git a/src/db/init.rs b/src/db/init.rs index 15d10b5..9d65582 100644 --- a/src/db/init.rs +++ b/src/db/init.rs @@ -1,8 +1,7 @@ use sqlx::postgres::PgPoolOptions; pub async fn create_pool() -> Result { - let database_url = textsender_models::envy::environment::get_db_url() - .value; + let database_url = textsender_models::envy::environment::get_db_url().value; println!("Database url: {database_url}"); PgPoolOptions::new() diff --git a/src/token_stuff/mod.rs b/src/token_stuff/mod.rs index ef6ee37..21eeb1e 100644 --- a/src/token_stuff/mod.rs +++ b/src/token_stuff/mod.rs @@ -119,9 +119,7 @@ mod tests { #[test] fn test_tokenize() { - let special_key = - textsender_models::envy::environment::get_secret_key() - .value; + let special_key = textsender_models::envy::environment::get_secret_key().value; let id = uuid::Uuid::new_v4(); match create_token(&special_key, &id) { Ok(cst) => { diff --git a/tests/tests.rs b/tests/tests.rs index ba88e04..118013e 100644 --- a/tests/tests.rs +++ b/tests/tests.rs @@ -9,9 +9,7 @@ mod db_mgr { pub const LIMIT: usize = 6; pub async fn get_pool() -> Result { - let tm_db_url = textsender_models::envy::environment::get_db_url() - .await - .value; + let tm_db_url = textsender_models::envy::environment::get_db_url().value; let tm_options = sqlx::postgres::PgConnectOptions::from_str(&tm_db_url).unwrap(); sqlx::PgPool::connect_with(tm_options).await } @@ -23,9 +21,7 @@ mod db_mgr { } pub async fn connect_to_db(db_name: &str) -> Result { - let db_url = textsender_models::envy::environment::get_db_url() - .await - .value; + let db_url = textsender_models::envy::environment::get_db_url().value; let options = sqlx::postgres::PgConnectOptions::from_str(&db_url)?.database(db_name); sqlx::PgPool::connect_with(options).await } @@ -35,7 +31,10 @@ mod db_mgr { db_name: &str, ) -> Result<(), sqlx::Error> { let create_query = format!("CREATE DATABASE {}", db_name); - match sqlx::query(&create_query).execute(template_pool).await { + match sqlx::query(sqlx::AssertSqlSafe(create_query)) + .execute(template_pool) + .await + { Ok(_) => Ok(()), Err(e) => Err(e), } @@ -47,14 +46,14 @@ mod db_mgr { db_name: &str, ) -> Result<(), sqlx::Error> { let drop_query = format!("DROP DATABASE IF EXISTS {} WITH (FORCE)", db_name); - sqlx::query(&drop_query).execute(template_pool).await?; + sqlx::query(sqlx::AssertSqlSafe(drop_query)) + .execute(template_pool) + .await?; Ok(()) } pub async fn get_database_name() -> Result> { - let database_url = textsender_models::envy::environment::get_db_url() - .await - .value; + let database_url = textsender_models::envy::environment::get_db_url().value; let parsed_url = url::Url::parse(&database_url)?; if parsed_url.scheme() == "postgres" || parsed_url.scheme() == "postgresql" {