Compare commits

..

3 Commits

Author SHA1 Message Date
f52b3a2167 tsk-55: Version bump
Some checks failed
Rust Build / Check (pull_request) Successful in 46s
Rust Build / Rustfmt (pull_request) Successful in 37s
Rust Build / Test Suite (pull_request) Successful in 1m29s
Rust Build / Clippy (pull_request) Failing after 2m1s
Rust Build / build (pull_request) Successful in 3m29s
2025-10-20 11:42:19 -04:00
cf67d37e85 Tidying up changes 2025-10-20 11:40:32 -04:00
f912844aa9 tsk-55: Fixing issue with register endpoint 2025-10-20 11:23:37 -04:00
12 changed files with 111 additions and 150 deletions

View File

@@ -10,4 +10,3 @@ POSTGRES_AUTH_PASSWORD=password
POSTGRES_AUTH_DB=icarus_auth_db POSTGRES_AUTH_DB=icarus_auth_db
POSTGRES_AUTH_HOST=auth_db POSTGRES_AUTH_HOST=auth_db
DATABASE_URL=postgresql://${POSTGRES_AUTH_USER}:${POSTGRES_AUTH_PASSWORD}@${POSTGRES_AUTH_HOST}:5432/${POSTGRES_AUTH_DB} DATABASE_URL=postgresql://${POSTGRES_AUTH_USER}:${POSTGRES_AUTH_PASSWORD}@${POSTGRES_AUTH_HOST}:5432/${POSTGRES_AUTH_DB}
ENABLE_REGISTRATION=TRUE

View File

@@ -10,4 +10,3 @@ POSTGRES_AUTH_PASSWORD=password
POSTGRES_AUTH_DB=icarus_auth_test_db POSTGRES_AUTH_DB=icarus_auth_test_db
POSTGRES_AUTH_HOST=localhost POSTGRES_AUTH_HOST=localhost
DATABASE_URL=postgresql://${POSTGRES_AUTH_USER}:${POSTGRES_AUTH_PASSWORD}@${POSTGRES_AUTH_HOST}:5432/${POSTGRES_AUTH_DB} DATABASE_URL=postgresql://${POSTGRES_AUTH_USER}:${POSTGRES_AUTH_PASSWORD}@${POSTGRES_AUTH_HOST}:5432/${POSTGRES_AUTH_DB}
ENABLE_REGISTRATION=TRUE

View File

@@ -76,7 +76,6 @@ jobs:
SECRET_KEY: ${{ secrets.TOKEN_SECRET_KEY }} SECRET_KEY: ${{ secrets.TOKEN_SECRET_KEY }}
# Make SSH agent available if tests fetch private dependencies # Make SSH agent available if tests fetch private dependencies
SSH_AUTH_SOCK: ${{ env.SSH_AUTH_SOCK }} SSH_AUTH_SOCK: ${{ env.SSH_AUTH_SOCK }}
ENABLE_REGISTRATION: 'TRUE'
run: | run: |
mkdir -p ~/.ssh mkdir -p ~/.ssh
echo "${{ secrets.MYREPO_TOKEN }}" > ~/.ssh/icarus_models_deploy_key echo "${{ secrets.MYREPO_TOKEN }}" > ~/.ssh/icarus_models_deploy_key

2
Cargo.lock generated
View File

@@ -748,7 +748,7 @@ dependencies = [
[[package]] [[package]]
name = "icarus_auth" name = "icarus_auth"
version = "0.6.4" version = "0.6.2"
dependencies = [ dependencies = [
"argon2", "argon2",
"axum", "axum",

View File

@@ -1,6 +1,6 @@
[package] [package]
name = "icarus_auth" name = "icarus_auth"
version = "0.6.4" version = "0.6.2"
edition = "2024" edition = "2024"
rust-version = "1.90" rust-version = "1.90"

View File

@@ -8,26 +8,22 @@ need to be modified. The `SECRET_KEY` variable should be changed since it will b
generation. The `SECRET_PASSPHASE` should also be changed when in production mode, but make sure generation. The `SECRET_PASSPHASE` should also be changed when in production mode, but make sure
the respective `passphrase` database table record exists. the respective `passphrase` database table record exists.
To enable or disable registrations, use `TRUE` or `FALSE` for the `ENABLE_REGISTRATION` variable. Build image
By default it is `TRUE`.
### Build image
``` ```
docker compose build docker compose build
``` ```
### Start images Start images
``` ```
docker compose up -d --force-recreate docker compose up -d --force-recreate
``` ```
### Bring it down Bring it down
``` ```
docker compose down -v docker compose down -v
``` ```
### Pruning Pruning
``` ```
docker system prune -a docker system prune -a
``` ```

View File

@@ -52,108 +52,67 @@ pub async fn register_user(
axum::Extension(pool): axum::Extension<sqlx::PgPool>, axum::Extension(pool): axum::Extension<sqlx::PgPool>,
Json(payload): Json<request::Request>, Json(payload): Json<request::Request>,
) -> (StatusCode, Json<response::Response>) { ) -> (StatusCode, Json<response::Response>) {
let registration_enabled = match is_registration_enabled().await { let mut user = icarus_models::user::User {
Ok(value) => value, username: payload.username.clone(),
Err(err) => { password: payload.password.clone(),
eprintln!("Error: {err:?}"); email: payload.email.clone(),
return ( phone: payload.phone.clone(),
axum::http::StatusCode::INTERNAL_SERVER_ERROR, firstname: payload.firstname.clone(),
Json(response::Response { lastname: payload.lastname.clone(),
message: String::from("Registration check failed"), status: String::from("Active"),
data: Vec::new(), email_verified: true,
}), ..Default::default()
);
}
}; };
if registration_enabled { match repo::user::exists(&pool, &user.username).await {
let mut user = icarus_models::user::User { Ok(res) => {
username: payload.username.clone(), if res {
password: payload.password.clone(), (
email: payload.email.clone(), StatusCode::BAD_REQUEST,
phone: payload.phone.clone(), Json(response::Response {
firstname: payload.firstname.clone(), message: String::from("Error"),
lastname: payload.lastname.clone(), data: Vec::new(),
status: String::from("Active"), }),
email_verified: true, )
..Default::default() } else {
}; let salt_string = hashing::generate_salt().unwrap();
let mut salt = icarus_models::user::salt::Salt::default();
let generated_salt = salt_string;
salt.salt = generated_salt.to_string();
salt.id = repo::salt::insert(&pool, &salt).await.unwrap();
user.salt_id = salt.id;
let hashed_password =
hashing::hash_password(&user.password, &generated_salt).unwrap();
user.password = hashed_password;
match repo::user::exists(&pool, &user.username).await { match repo::user::insert(&pool, &user).await {
Ok(res) => { Ok((id, date_created)) => {
if res { user.id = id;
( user.date_created = date_created;
StatusCode::BAD_REQUEST, (
Json(response::Response { StatusCode::CREATED,
message: String::from("Error"),
data: Vec::new(),
}),
)
} else {
let salt_string = hashing::generate_salt().unwrap();
let mut salt = icarus_models::user::salt::Salt::default();
let generated_salt = salt_string;
salt.salt = generated_salt.to_string();
salt.id = repo::salt::insert(&pool, &salt).await.unwrap();
user.salt_id = salt.id;
let hashed_password =
hashing::hash_password(&user.password, &generated_salt).unwrap();
user.password = hashed_password;
match repo::user::insert(&pool, &user).await {
Ok((id, date_created)) => {
user.id = id;
user.date_created = date_created;
(
StatusCode::CREATED,
Json(response::Response {
message: String::from("User created"),
data: vec![user],
}),
)
}
Err(err) => (
StatusCode::BAD_REQUEST,
Json(response::Response { Json(response::Response {
message: err.to_string(), message: String::from("User created"),
data: vec![user], data: vec![user],
}), }),
), )
} }
Err(err) => (
StatusCode::BAD_REQUEST,
Json(response::Response {
message: err.to_string(),
data: vec![user],
}),
),
} }
} }
Err(err) => (
StatusCode::BAD_REQUEST,
Json(response::Response {
message: err.to_string(),
data: vec![user],
}),
),
} }
} else { Err(err) => (
( StatusCode::BAD_REQUEST,
axum::http::StatusCode::NOT_ACCEPTABLE,
Json(response::Response { Json(response::Response {
message: String::from("Registration is not enabled"), message: err.to_string(),
data: Vec::new(), data: vec![user],
}), }),
) ),
}
}
/// Checks to see if registration is enabled
async fn is_registration_enabled() -> Result<bool, std::io::Error> {
let key = String::from("ENABLE_REGISTRATION");
let var = icarus_envy::environment::get_env(&key).await;
let parsed_value = var.value.to_uppercase();
if parsed_value == "TRUE" {
Ok(true)
} else if parsed_value == "FALSE" {
Ok(false)
} else {
Err(std::io::Error::other(
"Could not determine value of ENABLE_REGISTRATION",
))
} }
} }

View File

@@ -1,20 +0,0 @@
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");
}

View File

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

36
src/lib.rs Normal file
View File

@@ -0,0 +1,36 @@
// TODO: Get rid of this file and place the code in more appropriate places
pub mod callers;
pub mod config;
pub mod hashing;
pub mod repo;
pub mod token_stuff;
mod connection_settings {
pub const MAXCONN: u32 = 5;
}
pub mod db {
use sqlx::postgres::PgPoolOptions;
use crate::connection_settings;
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(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");
}
}

View File

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

View File

@@ -124,10 +124,10 @@ pub mod user {
.map_err(|_e| sqlx::Error::RowNotFound)?, .map_err(|_e| sqlx::Error::RowNotFound)?,
}; };
if result.id.is_nil() && result.date_created.is_none() { if !result.id.is_nil() && !result.date_created.is_none() {
Err(sqlx::Error::RowNotFound)
} else {
Ok((result.id, result.date_created)) Ok((result.id, result.date_created))
} else {
Err(sqlx::Error::RowNotFound)
} }
} }
} }