tsk-70: Moved contents of src/lib.rs elsewhere

This commit is contained in:
2025-10-20 13:02:48 -04:00
parent e86ca4b2c8
commit bc5c3e5e8d
3 changed files with 44 additions and 14 deletions

20
src/db/init.rs Normal file
View File

@@ -0,0 +1,20 @@
use sqlx::postgres::PgPoolOptions;
pub async fn create_pool() -> Result<sqlx::PgPool, sqlx::Error> {
let database_url = icarus_envy::environment::get_db_url().await.value;
println!("Database url: {database_url}");
PgPoolOptions::new()
.max_connections(super::connection_settings::MAXCONN)
.connect(&database_url)
.await
}
pub async fn migrations(pool: &sqlx::PgPool) {
// Run migrations using the sqlx::migrate! macro
// Assumes your migrations are in a ./migrations folder relative to Cargo.toml
sqlx::migrate!("./migrations")
.run(pool)
.await
.expect("Failed to run migrations");
}

5
src/db/mod.rs Normal file
View File

@@ -0,0 +1,5 @@
pub mod init;
mod connection_settings {
pub const MAXCONN: u32 = 5;
}

View File

@@ -1,5 +1,10 @@
use icarus_auth::callers;
use icarus_auth::config;
pub mod callers;
pub mod config;
pub mod db;
pub mod hashing;
pub mod repo;
pub mod token_stuff;
#[tokio::main]
async fn main() {
@@ -21,7 +26,7 @@ mod init {
};
use utoipa::OpenApi;
use crate::callers;
use super::callers;
use callers::common as common_callers;
use callers::login as login_caller;
use callers::register as register_caller;
@@ -124,11 +129,11 @@ mod init {
}
pub async fn app() -> Router {
let pool = icarus_auth::db::create_pool()
let pool = super::db::init::create_pool()
.await
.expect("Failed to create pool");
icarus_auth::db::migrations(&pool).await;
super::db::init::migrations(&pool).await;
routes()
.await
@@ -216,8 +221,8 @@ mod tests {
}
}
fn get_test_register_request() -> icarus_auth::callers::register::request::Request {
icarus_auth::callers::register::request::Request {
fn get_test_register_request() -> callers::register::request::Request {
callers::register::request::Request {
username: String::from("somethingsss"),
password: String::from("Raindown!"),
email: String::from("dev@null.com"),
@@ -228,7 +233,7 @@ mod tests {
}
fn get_test_register_payload(
usr: &icarus_auth::callers::register::request::Request,
usr: &callers::register::request::Request,
) -> serde_json::Value {
json!({
"username": &usr.username,
@@ -245,7 +250,7 @@ mod tests {
pub async fn register(
app: &axum::Router,
usr: &icarus_auth::callers::register::request::Request,
usr: &super::callers::register::request::Request,
) -> Result<axum::response::Response, std::convert::Infallible> {
let payload = super::get_test_register_payload(&usr);
let req = axum::http::Request::builder()
@@ -298,7 +303,7 @@ mod tests {
let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
icarus_auth::db::migrations(&pool).await;
db::init::migrations(&pool).await;
let app = init::routes().await.layer(axum::Extension(pool));
@@ -355,7 +360,7 @@ mod tests {
let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
icarus_auth::db::migrations(&pool).await;
db::init::migrations(&pool).await;
let app = init::routes().await.layer(axum::Extension(pool));
@@ -443,7 +448,7 @@ mod tests {
let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
icarus_auth::db::migrations(&pool).await;
db::init::migrations(&pool).await;
let app = init::routes().await.layer(axum::Extension(pool));
let passphrase =
@@ -497,13 +502,13 @@ mod tests {
let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
icarus_auth::db::migrations(&pool).await;
db::init::migrations(&pool).await;
let app = init::routes().await.layer(axum::Extension(pool));
let id = uuid::Uuid::parse_str("22f9c775-cce9-457a-a147-9dafbb801f61").unwrap();
let key = icarus_envy::environment::get_secret_key().await.value;
match icarus_auth::token_stuff::create_service_token(&key, &id) {
match token_stuff::create_service_token(&key, &id) {
Ok((token, _expire)) => {
let payload = serde_json::json!({
"access_token": token