diff --git a/crates/service/storyteller_web/src/database/queries/list_ip_bans.rs b/crates/lib/database_queries/src/ip_bans/list_ip_bans.rs similarity index 100% rename from crates/service/storyteller_web/src/database/queries/list_ip_bans.rs rename to crates/lib/database_queries/src/ip_bans/list_ip_bans.rs diff --git a/crates/lib/database_queries/src/ip_bans/mod.rs b/crates/lib/database_queries/src/ip_bans/mod.rs new file mode 100644 index 0000000000..2cebd7b6e9 --- /dev/null +++ b/crates/lib/database_queries/src/ip_bans/mod.rs @@ -0,0 +1 @@ +pub mod list_ip_bans; \ No newline at end of file diff --git a/crates/lib/database_queries/src/lib.rs b/crates/lib/database_queries/src/lib.rs index 891437aa3d..9fc2829f95 100644 --- a/crates/lib/database_queries/src/lib.rs +++ b/crates/lib/database_queries/src/lib.rs @@ -5,6 +5,7 @@ mod builders; pub mod api_tokens; pub mod column_types; pub mod helpers; +pub mod ip_bans; pub mod model_categories; pub mod tokens; pub mod tts; diff --git a/crates/lib/database_queries/src/users/mod.rs b/crates/lib/database_queries/src/users/mod.rs index 2327971b74..a4b95c51c7 100644 --- a/crates/lib/database_queries/src/users/mod.rs +++ b/crates/lib/database_queries/src/users/mod.rs @@ -1,4 +1,5 @@ pub mod get_user_profile_by_username; pub mod list_users_query_builder; pub mod user_badges; -pub mod user_roles; \ No newline at end of file +pub mod user_roles; +pub mod user_sessions; \ No newline at end of file diff --git a/crates/service/storyteller_web/src/database/queries/create_session.rs b/crates/lib/database_queries/src/users/user_sessions/create_session.rs similarity index 91% rename from crates/service/storyteller_web/src/database/queries/create_session.rs rename to crates/lib/database_queries/src/users/user_sessions/create_session.rs index aa9654f5e8..3748195e67 100644 --- a/crates/service/storyteller_web/src/database/queries/create_session.rs +++ b/crates/lib/database_queries/src/users/user_sessions/create_session.rs @@ -1,6 +1,6 @@ use anyhow::anyhow; -use crate::AnyhowResult; -use database_queries::tokens::Tokens; +use container_common::anyhow_result::AnyhowResult; +use crate::tokens::Tokens; use log::{info, warn}; use sqlx::MySqlPool; diff --git a/crates/lib/database_queries/src/users/user_sessions/mod.rs b/crates/lib/database_queries/src/users/user_sessions/mod.rs new file mode 100644 index 0000000000..6a3f6dcb74 --- /dev/null +++ b/crates/lib/database_queries/src/users/user_sessions/mod.rs @@ -0,0 +1 @@ +pub mod create_session; \ No newline at end of file diff --git a/crates/service/storyteller_web/src/database/mod.rs b/crates/service/storyteller_web/src/database/mod.rs index 1c04fcc19f..2fdbb114a4 100644 --- a/crates/service/storyteller_web/src/database/mod.rs +++ b/crates/service/storyteller_web/src/database/mod.rs @@ -1,2 +1 @@ -pub mod mediators; -pub mod queries; \ No newline at end of file +pub mod mediators; \ No newline at end of file diff --git a/crates/service/storyteller_web/src/database/queries/badges.rs b/crates/service/storyteller_web/src/database/queries/badges.rs deleted file mode 100644 index 9425f0069e..0000000000 --- a/crates/service/storyteller_web/src/database/queries/badges.rs +++ /dev/null @@ -1,29 +0,0 @@ -use sqlx::MySqlPool; -use crate::AnyhowResult; - -pub struct NewBadge { - pub slug: String, - pub title: String, - pub description: String, - pub image_url: String, -} - -impl NewBadge { - pub async fn insert(&self, pool: &MySqlPool) -> AnyhowResult { - let record_id = sqlx::query!( - r#" -INSERT INTO badges ( slug, title, description, image_url ) -VALUES ( ?, ?, ?, ? ) - "#, - self.slug, - self.title, - self.description, - self.image_url, - ) - .execute(pool) - .await? - .last_insert_id(); - - Ok(record_id) - } -} diff --git a/crates/service/storyteller_web/src/database/queries/mod.rs b/crates/service/storyteller_web/src/database/queries/mod.rs deleted file mode 100644 index 7b1b03af01..0000000000 --- a/crates/service/storyteller_web/src/database/queries/mod.rs +++ /dev/null @@ -1,3 +0,0 @@ -pub mod badges; -pub mod create_session; -pub mod list_ip_bans; \ No newline at end of file diff --git a/crates/service/storyteller_web/src/http_server/endpoints/users/create_account.rs b/crates/service/storyteller_web/src/http_server/endpoints/users/create_account.rs index a14e9f9cfe..f81565276b 100644 --- a/crates/service/storyteller_web/src/http_server/endpoints/users/create_account.rs +++ b/crates/service/storyteller_web/src/http_server/endpoints/users/create_account.rs @@ -5,7 +5,6 @@ use actix_web::cookie::Cookie; use actix_web::error::ResponseError; use actix_web::http::StatusCode; use actix_web::{Responder, web, HttpResponse, error, HttpRequest}; -use crate::database::queries::create_session::create_session_for_user; use crate::http_server::endpoints::users::create_account::CreateAccountErrorType::{BadInput, ServerError, UsernameTaken, EmailTaken}; use crate::http_server::web_utils::ip_address::get_request_ip; use crate::http_server::web_utils::response_error_helpers::to_simple_json_error; @@ -16,7 +15,7 @@ use crate::validations::passwords::validate_passwords; use crate::validations::username::validate_username; use crate::validations::username_reservations::is_reserved_username; use database_queries::tokens::Tokens; -use derive_more::{Display, Error}; +use database_queries::users::user_sessions::create_session::create_session_for_user; use log::{info, warn, log}; use regex::Regex; use sqlx::error::DatabaseError; diff --git a/crates/service/storyteller_web/src/http_server/endpoints/users/edit_profile.rs b/crates/service/storyteller_web/src/http_server/endpoints/users/edit_profile.rs index 21d64e593c..9a6211d573 100644 --- a/crates/service/storyteller_web/src/http_server/endpoints/users/edit_profile.rs +++ b/crates/service/storyteller_web/src/http_server/endpoints/users/edit_profile.rs @@ -6,7 +6,6 @@ use actix_web::error::ResponseError; use actix_web::http::StatusCode; use actix_web::web::Path; use actix_web::{Responder, web, HttpResponse, error, HttpRequest}; -use crate::database::queries::create_session::create_session_for_user; use crate::http_server::web_utils::ip_address::get_request_ip; use crate::http_server::web_utils::response_error_helpers::to_simple_json_error; use crate::server_state::ServerState; diff --git a/crates/service/storyteller_web/src/http_server/endpoints/users/list_user_tts_inference_results.rs b/crates/service/storyteller_web/src/http_server/endpoints/users/list_user_tts_inference_results.rs index c01e2b2585..69ff5c10f0 100644 --- a/crates/service/storyteller_web/src/http_server/endpoints/users/list_user_tts_inference_results.rs +++ b/crates/service/storyteller_web/src/http_server/endpoints/users/list_user_tts_inference_results.rs @@ -8,7 +8,6 @@ use actix_web::web::{Path, Query}; use actix_web::{Responder, web, HttpResponse, error, HttpRequest, HttpMessage}; use chrono::{DateTime, Utc}; use crate::AnyhowResult; -use crate::database::queries::create_session::create_session_for_user; use crate::http_server::web_utils::ip_address::get_request_ip; use crate::http_server::web_utils::response_error_helpers::to_simple_json_error; use crate::server_state::ServerState; diff --git a/crates/service/storyteller_web/src/http_server/endpoints/users/login.rs b/crates/service/storyteller_web/src/http_server/endpoints/users/login.rs index 6a7dad39c0..9bf940f700 100644 --- a/crates/service/storyteller_web/src/http_server/endpoints/users/login.rs +++ b/crates/service/storyteller_web/src/http_server/endpoints/users/login.rs @@ -6,12 +6,12 @@ use actix_web::error::ResponseError; use actix_web::http::StatusCode; use actix_web::{Responder, web, HttpResponse, error, HttpRequest}; use crate::AnyhowResult; -use crate::database::queries::create_session::create_session_for_user; use crate::http_server::web_utils::ip_address::get_request_ip; use crate::http_server::web_utils::response_error_helpers::to_simple_json_error; use crate::http_server::web_utils::serialize_as_json_error::serialize_as_json_error; use crate::server_state::ServerState; use database_queries::helpers::boolean_converters::i8_to_bool; +use database_queries::users::user_sessions::create_session::create_session_for_user; use log::{info, warn, log}; use regex::Regex; use sqlx::MySqlPool; diff --git a/crates/service/storyteller_web/src/threads/poll_ip_banlist_thread.rs b/crates/service/storyteller_web/src/threads/poll_ip_banlist_thread.rs index 89b59aa32c..7706e84201 100644 --- a/crates/service/storyteller_web/src/threads/poll_ip_banlist_thread.rs +++ b/crates/service/storyteller_web/src/threads/poll_ip_banlist_thread.rs @@ -1,6 +1,6 @@ use anyhow::Error; -use crate::database::queries::list_ip_bans::list_ip_bans; use crate::threads::ip_banlist_set::IpBanlistSet; +use database_queries::ip_bans::list_ip_bans::list_ip_bans; use log::info; use log::warn; use sqlx::MySqlPool;