// use std::io::Write; // use common_multipart_rfc7578::client::multipart::{Body as MultipartBody, Form as MultipartForm}; // use tower::ServiceExt; use textsender_api::db; mod db_mgr { use std::str::FromStr; 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_options = sqlx::postgres::PgConnectOptions::from_str(&tm_db_url).unwrap(); sqlx::PgPool::connect_with(tm_options).await } pub async fn generate_db_name() -> String { let db_name = get_database_name().await.unwrap() + &"_" + &uuid::Uuid::new_v4().to_string()[..LIMIT]; db_name } pub async fn connect_to_db(db_name: &str) -> Result { let db_url = textsender_models::envy::environment::get_db_url() .await .value; let options = sqlx::postgres::PgConnectOptions::from_str(&db_url)?.database(db_name); sqlx::PgPool::connect_with(options).await } pub async fn create_database( template_pool: &sqlx::PgPool, db_name: &str, ) -> Result<(), sqlx::Error> { let create_query = format!("CREATE DATABASE {}", db_name); match sqlx::query(&create_query).execute(template_pool).await { Ok(_) => Ok(()), Err(e) => Err(e), } } pub async fn get_database_ready() -> ( sqlx::Pool, String, sqlx::Pool, ) { let tm_pool = get_pool().await.unwrap(); let db_name = generate_db_name().await; match create_database(&tm_pool, &db_name).await { Ok(_) => { println!("Success"); } Err(err) => { assert!(false, "Error: {:?}", err); } } let pool = connect_to_db(&db_name).await.unwrap(); migrations(&pool).await; // db::migrations(&pool).await; (tm_pool, db_name, pool) } // Function to drop a database pub async fn drop_database( template_pool: &sqlx::PgPool, 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?; Ok(()) } pub async fn get_database_name() -> Result> { let database_url = textsender_models::envy::environment::get_db_url() .await .value; let parsed_url = url::Url::parse(&database_url)?; if parsed_url.scheme() == "postgres" || parsed_url.scheme() == "postgresql" { match parsed_url .path_segments() .and_then(|segments| segments.last().map(|s| s.to_string())) { Some(sss) => Ok(sss), None => Err("Error parsing".into()), } } else { // Handle other database types if needed Err("Error parsing".into()) } } pub async fn migrations(pool: &sqlx::PgPool) { super::db::migrations(pool).await; } } mod init { pub async fn app(pool: sqlx::PgPool) -> axum::Router { textsender_api::config::init::routes() .await .layer(axum::Extension(pool)) .layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024)) .layer(tower_http::timeout::TimeoutLayer::with_status_code( axum::http::StatusCode::OK, std::time::Duration::from_secs(300), )) } } mod util { pub async fn resp_to_bytes( response: axum::response::Response, ) -> Result { axum::body::to_bytes(response.into_body(), std::usize::MAX).await } pub async fn get_resp_data(response: axum::response::Response) -> Data where Data: for<'a> serde::Deserialize<'a>, { let body = resp_to_bytes(response).await.unwrap(); serde_json::from_slice(&body).unwrap() } /* pub async fn format_url_with_value(endpoint: &str, value: &uuid::Uuid) -> String { let last = endpoint.len() - 5; format!("{}/{value}", &endpoint[0..last]) } */ } pub fn token_fields() -> (String, String, String) { ( String::from("What a twist!"), String::from("textsender_test"), String::from("textsender"), ) } /// Test User Id pub const TEST_USER_ID: uuid::Uuid = uuid::uuid!("cc938368-615a-4694-b2ca-6e122fa31c52"); /// Test contact phone number pub const TEST_CONTACT_PHONE_NUMBER: &str = "+10123456789"; /// Test contact updated firstname pub const TEST_CONTACT_UPDATED_FIRSTNAME: &str = "Johnny"; /// Test contact updated lastname pub const TEST_CONTACT_UPDATED_LASTNAME: &str = "CASH"; /// Test contact updated nickname pub const TEST_CONTACT_UPDATED_NICKNAME: &str = "The Man in Black"; /// Test message content pub const TEST_MESSAGE_CONTENT: &str = "The wind cries mary"; pub async fn test_token() -> Result { let key: String = textsender_models::envy::environment::get_secret_main_key() .await .value; let (message, issuer, audience) = token_fields(); let token_resource = textsender_models::token::TokenResource { message: message, issuer: issuer, audiences: vec![audience], user_id: TEST_USER_ID, }; match textsender_models::token::create_token(&key, &token_resource, time::Duration::hours(1)) { Ok((access_token, _some_time)) => Ok(access_token), Err(err) => Err(err), } } pub async fn bearer_auth() -> String { let token = match test_token().await { Ok(access_token) => access_token, Err(err) => { assert!(false, "Error: {err:?}"); String::new() } }; format!("Bearer {token}") } mod request { // use common_multipart_rfc7578::client::multipart::{ // Body as MultipartBody, Form as MultipartForm, // }; use tower::ServiceExt; pub async fn create_contact( app: &axum::Router, ) -> Result { let payload = serde_json::json!({ "phone_number": super::TEST_CONTACT_PHONE_NUMBER, "user_id": super::TEST_USER_ID, }); match run_post( Some(&payload), textsender_api::caller::endpoints::ADD_CONTACT, axum::http::Method::POST, true, ) .await { Ok(req) => match app.clone().oneshot(req).await { Ok(response) => Ok(response), Err(err) => Err(axum::http::Error::from(err)), }, Err(err) => Err(err), } } pub async fn get_contact( app: &axum::Router, id: &uuid::Uuid, ) -> Result { let uri = format!( "{}?id={}", textsender_api::caller::endpoints::GET_CONTACT, id ); match run_post(None, &uri, axum::http::Method::GET, false).await { Ok(req) => match app.clone().oneshot(req).await { Ok(response) => Ok(response), Err(err) => Err(axum::http::Error::from(err)), }, Err(err) => Err(err), } } pub async fn update_contact_names( app: &axum::Router, id: &uuid::Uuid, firstname: Option<&str>, lastname: Option<&str>, nickname: Option<&str>, ) -> Result { let payload = serde_json::json!({ "firstname": firstname, "lastname": lastname, "nickname": nickname, "phone_number": super::TEST_CONTACT_PHONE_NUMBER, "id": id }); match run_post( Some(&payload), textsender_api::caller::endpoints::UPDATE_CONTACT_NAME, axum::http::Method::PATCH, true, ) .await { Ok(req) => match app.clone().oneshot(req).await { Ok(response) => Ok(response), Err(err) => Err(axum::http::Error::from(err)), }, Err(err) => Err(err), } } pub mod message { use tower::ServiceExt; pub async fn create_message( app: &axum::Router, ) -> Result { let payload = serde_json::json!({ "content": super::super::TEST_MESSAGE_CONTENT, "user_id": super::super::TEST_USER_ID, }); match super::run_post( Some(&payload), textsender_api::caller::endpoints::ADD_MESSAGE, axum::http::Method::POST, true, ) .await { Ok(req) => match app.clone().oneshot(req).await { Ok(response) => Ok(response), Err(err) => Err(axum::http::Error::from(err)), }, Err(err) => Err(err), } } pub async fn get_message( app: &axum::Router, id: &uuid::Uuid, ) -> Result { let uri = format!( "{}?id={}", textsender_api::caller::endpoints::GET_MESSAGE, id ); match super::run_post(None, &uri, axum::http::Method::GET, false).await { Ok(req) => match app.clone().oneshot(req).await { Ok(response) => Ok(response), Err(err) => Err(axum::http::Error::from(err)), }, Err(err) => Err(err), } } pub mod scheduling { use tower::ServiceExt; pub async fn create_scheduled_message( app: &axum::Router, ) -> Result { /* let twenty_minutes_from_now = time::OffsetDateTime::now_utc().checked_add_signed( time::Duration::