Commonize and distinguish a lot of struct names in labrinth::database::models (#3691)

This commit is contained in:
Josiah Glosson
2025-05-24 04:38:43 -05:00
committed by GitHub
parent 9c1bdf16e4
commit 4e4a7be7ef
78 changed files with 1075 additions and 1009 deletions

View File

@@ -130,7 +130,7 @@ pub async fn page_view_ingest(
];
if PROJECT_TYPES.contains(&segments_vec[0]) {
let project = crate::database::models::Project::get(
let project = crate::database::models::DBProject::get(
segments_vec[1],
&**pool,
&redis,
@@ -189,7 +189,7 @@ pub async fn playtime_ingest(
));
}
let versions = crate::database::models::Version::get_many(
let versions = crate::database::models::DBVersion::get_many(
&playtimes.iter().map(|x| (*x.0).into()).collect::<Vec<_>>(),
&**pool,
&redis,

View File

@@ -204,7 +204,7 @@ pub async fn delphi_result_ingest(
let webhook_url = dotenvy::var("DELPHI_SLACK_WEBHOOK")?;
let project = crate::database::models::Project::get_id(
let project = crate::database::models::DBProject::get_id(
body.project_id.into(),
&**pool,
&redis,

View File

@@ -1,5 +1,5 @@
use crate::auth::{get_user_from_headers, send_email};
use crate::database::models::charge_item::ChargeItem;
use crate::database::models::charge_item::DBCharge;
use crate::database::models::{
generate_charge_id, generate_user_subscription_id, product_item,
user_subscription_item,
@@ -59,7 +59,8 @@ pub async fn products(
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
) -> Result<HttpResponse, ApiError> {
let products = product_item::QueryProduct::list(&**pool, &redis).await?;
let products =
product_item::QueryProductWithPrices::list(&**pool, &redis).await?;
let products = products
.into_iter()
@@ -107,7 +108,7 @@ pub async fn subscriptions(
.1;
let subscriptions =
user_subscription_item::UserSubscriptionItem::get_all_user(
user_subscription_item::DBUserSubscription::get_all_user(
if let Some(user_id) = query.user_id {
if user.role.is_admin() {
user_id.into()
@@ -173,8 +174,8 @@ pub async fn refund_charge(
));
}
if let Some(charge) = ChargeItem::get(id.into(), &**pool).await? {
let refunds = ChargeItem::get_children(id.into(), &**pool).await?;
if let Some(charge) = DBCharge::get(id.into(), &**pool).await? {
let refunds = DBCharge::get_children(id.into(), &**pool).await?;
let refunds = -refunds
.into_iter()
.filter_map(|x| match x.status {
@@ -261,7 +262,7 @@ pub async fn refund_charge(
let mut transaction = pool.begin().await?;
let charge_id = generate_charge_id(&mut transaction).await?;
ChargeItem {
DBCharge {
id: charge_id,
user_id: charge.user_id,
price_id: charge.price_id,
@@ -284,7 +285,7 @@ pub async fn refund_charge(
if body.0.unprovision.unwrap_or(false) {
if let Some(subscription_id) = charge.subscription_id {
let open_charge =
ChargeItem::get_open_subscription(subscription_id, &**pool)
DBCharge::get_open_subscription(subscription_id, &**pool)
.await?;
if let Some(mut open_charge) = open_charge {
open_charge.status = ChargeStatus::Cancelled;
@@ -332,7 +333,7 @@ pub async fn edit_subscription(
let (id,) = info.into_inner();
if let Some(subscription) =
user_subscription_item::UserSubscriptionItem::get(id.into(), &**pool)
user_subscription_item::DBUserSubscription::get(id.into(), &**pool)
.await?
{
if subscription.user_id != user.id.into() && !user.role.is_admin() {
@@ -342,7 +343,7 @@ pub async fn edit_subscription(
let mut transaction = pool.begin().await?;
let mut open_charge =
crate::database::models::charge_item::ChargeItem::get_open_subscription(
crate::database::models::charge_item::DBCharge::get_open_subscription(
subscription.id,
&mut *transaction,
)
@@ -353,7 +354,7 @@ pub async fn edit_subscription(
)
})?;
let current_price = product_item::ProductPriceItem::get(
let current_price = product_item::DBProductPrice::get(
subscription.price_id,
&mut *transaction,
)
@@ -397,7 +398,7 @@ pub async fn edit_subscription(
let intent = if let Some(product_id) = &edit_subscription.product {
let product_price =
product_item::ProductPriceItem::get_all_product_prices(
product_item::DBProductPrice::get_all_product_prices(
(*product_id).into(),
&mut *transaction,
)
@@ -622,7 +623,7 @@ pub async fn charges(
.1;
let charges =
crate::database::models::charge_item::ChargeItem::get_from_user(
crate::database::models::charge_item::DBCharge::get_from_user(
if let Some(user_id) = query.user_id {
if user.role.is_admin() {
user_id.into()
@@ -829,7 +830,7 @@ pub async fn remove_payment_method(
.await?;
let user_subscriptions =
user_subscription_item::UserSubscriptionItem::get_all_user(
user_subscription_item::DBUserSubscription::get_all_user(
user.id.into(),
&**pool,
)
@@ -935,12 +936,11 @@ pub async fn active_servers(
));
}
let servers =
user_subscription_item::UserSubscriptionItem::get_all_servers(
query.subscription_status,
&**pool,
)
.await?;
let servers = user_subscription_item::DBUserSubscription::get_all_servers(
query.subscription_status,
&**pool,
)
.await?;
#[derive(Serialize)]
struct ActiveServer {
@@ -1153,7 +1153,7 @@ pub async fn initiate_payment(
match payment_request.charge {
ChargeRequestType::Existing { id } => {
let charge =
crate::database::models::charge_item::ChargeItem::get(
crate::database::models::charge_item::DBCharge::get(
id.into(),
&**pool,
)
@@ -1178,7 +1178,7 @@ pub async fn initiate_payment(
interval,
} => {
let product =
product_item::ProductItem::get(product_id.into(), &**pool)
product_item::DBProduct::get(product_id.into(), &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -1188,7 +1188,7 @@ pub async fn initiate_payment(
})?;
let mut product_prices =
product_item::ProductPriceItem::get_all_product_prices(
product_item::DBProductPrice::get_all_product_prices(
product.id, &**pool,
)
.await?;
@@ -1229,14 +1229,14 @@ pub async fn initiate_payment(
if let Price::Recurring { .. } = price_item.prices {
if product.unitary {
let user_subscriptions =
user_subscription_item::UserSubscriptionItem::get_all_user(
user_subscription_item::DBUserSubscription::get_all_user(
user.id.into(),
&**pool,
)
.await?;
let user_products =
product_item::ProductPriceItem::get_many(
product_item::DBProductPrice::get_many(
&user_subscriptions
.iter()
.filter(|x| {
@@ -1413,12 +1413,12 @@ pub async fn stripe_webhook(
&dotenvy::var("STRIPE_WEBHOOK_SECRET")?,
) {
struct PaymentIntentMetadata {
pub user_item: crate::database::models::user_item::User,
pub product_price_item: product_item::ProductPriceItem,
pub product_item: product_item::ProductItem,
pub charge_item: crate::database::models::charge_item::ChargeItem,
pub user_item: crate::database::models::user_item::DBUser,
pub product_price_item: product_item::DBProductPrice,
pub product_item: product_item::DBProduct,
pub charge_item: crate::database::models::charge_item::DBCharge,
pub user_subscription_item:
Option<user_subscription_item::UserSubscriptionItem>,
Option<user_subscription_item::DBUserSubscription>,
pub payment_metadata: Option<PaymentRequestMetadata>,
#[allow(dead_code)]
pub charge_type: ChargeType,
@@ -1447,7 +1447,7 @@ pub async fn stripe_webhook(
};
let user = if let Some(user) =
crate::database::models::user_item::User::get_id(
crate::database::models::user_item::DBUser::get_id(
user_id, pool, redis,
)
.await?
@@ -1483,17 +1483,14 @@ pub async fn stripe_webhook(
let (charge, price, product, subscription) = if let Some(
mut charge,
) =
crate::database::models::charge_item::ChargeItem::get(
crate::database::models::charge_item::DBCharge::get(
charge_id, pool,
)
.await?
{
let price = if let Some(price) =
product_item::ProductPriceItem::get(
charge.price_id,
pool,
)
.await?
product_item::DBProductPrice::get(charge.price_id, pool)
.await?
{
price
} else {
@@ -1501,7 +1498,7 @@ pub async fn stripe_webhook(
};
let product = if let Some(product) =
product_item::ProductItem::get(price.product_id, pool)
product_item::DBProduct::get(price.product_id, pool)
.await?
{
product
@@ -1517,7 +1514,7 @@ pub async fn stripe_webhook(
if let Some(subscription_id) = charge.subscription_id {
let mut subscription = if let Some(subscription) =
user_subscription_item::UserSubscriptionItem::get(
user_subscription_item::DBUserSubscription::get(
subscription_id,
pool,
)
@@ -1567,7 +1564,7 @@ pub async fn stripe_webhook(
};
let price = if let Some(price) =
product_item::ProductPriceItem::get(price_id, pool)
product_item::DBProductPrice::get(price_id, pool)
.await?
{
price
@@ -1576,7 +1573,7 @@ pub async fn stripe_webhook(
};
let product = if let Some(product) =
product_item::ProductItem::get(price.product_id, pool)
product_item::DBProduct::get(price.product_id, pool)
.await?
{
product
@@ -1608,14 +1605,14 @@ pub async fn stripe_webhook(
break 'metadata;
};
let subscription = if let Some(mut subscription) = user_subscription_item::UserSubscriptionItem::get(subscription_id, pool).await? {
let subscription = if let Some(mut subscription) = user_subscription_item::DBUserSubscription::get(subscription_id, pool).await? {
subscription.status = SubscriptionStatus::Unprovisioned;
subscription.price_id = price_id;
subscription.interval = interval;
subscription
} else {
user_subscription_item::UserSubscriptionItem {
user_subscription_item::DBUserSubscription {
id: subscription_id,
user_id,
price_id,
@@ -1637,7 +1634,7 @@ pub async fn stripe_webhook(
}
};
let charge = ChargeItem {
let charge = DBCharge {
id: charge_id,
user_id,
price_id,
@@ -1870,7 +1867,7 @@ pub async fn stripe_webhook(
if let Some(mut subscription) =
metadata.user_subscription_item
{
let open_charge = ChargeItem::get_open_subscription(
let open_charge = DBCharge::get_open_subscription(
subscription.id,
&mut *transaction,
)
@@ -1898,7 +1895,7 @@ pub async fn stripe_webhook(
{
let charge_id =
generate_charge_id(&mut transaction).await?;
ChargeItem {
DBCharge {
id: charge_id,
user_id: metadata.user_item.id,
price_id: metadata.product_price_item.id,
@@ -1936,7 +1933,7 @@ pub async fn stripe_webhook(
}
transaction.commit().await?;
crate::database::models::user_item::User::clear_caches(
crate::database::models::user_item::DBUser::clear_caches(
&[(metadata.user_item.id, None)],
&redis,
)
@@ -2098,7 +2095,7 @@ async fn get_or_create_customer(
.execute(pool)
.await?;
crate::database::models::user_item::User::clear_caches(
crate::database::models::user_item::DBUser::clear_caches(
&[(user_id.into(), None)],
redis,
)
@@ -2116,10 +2113,10 @@ pub async fn index_subscriptions(pool: PgPool, redis: RedisPool) {
let mut clear_cache_users = Vec::new();
// If an active subscription has a canceled charge OR a failed charge more than two days ago, it should be cancelled
let all_charges = ChargeItem::get_unprovision(&pool).await?;
let all_charges = DBCharge::get_unprovision(&pool).await?;
let mut all_subscriptions =
user_subscription_item::UserSubscriptionItem::get_many(
user_subscription_item::DBUserSubscription::get_many(
&all_charges
.iter()
.filter_map(|x| x.subscription_id)
@@ -2129,7 +2126,7 @@ pub async fn index_subscriptions(pool: PgPool, redis: RedisPool) {
&pool,
)
.await?;
let subscription_prices = product_item::ProductPriceItem::get_many(
let subscription_prices = product_item::DBProductPrice::get_many(
&all_subscriptions
.iter()
.map(|x| x.price_id)
@@ -2139,7 +2136,7 @@ pub async fn index_subscriptions(pool: PgPool, redis: RedisPool) {
&pool,
)
.await?;
let subscription_products = product_item::ProductItem::get_many(
let subscription_products = product_item::DBProduct::get_many(
&subscription_prices
.iter()
.map(|x| x.product_id)
@@ -2149,7 +2146,7 @@ pub async fn index_subscriptions(pool: PgPool, redis: RedisPool) {
&pool,
)
.await?;
let users = crate::database::models::User::get_many_ids(
let users = crate::database::models::DBUser::get_many_ids(
&all_subscriptions
.iter()
.map(|x| x.user_id)
@@ -2260,7 +2257,7 @@ pub async fn index_subscriptions(pool: PgPool, redis: RedisPool) {
clear_cache_users.push(user.id);
}
crate::database::models::User::clear_caches(
crate::database::models::DBUser::clear_caches(
&clear_cache_users
.into_iter()
.map(|x| (x, None))
@@ -2289,12 +2286,12 @@ pub async fn index_billing(
let res = async {
// If a charge is open and due or has been attempted more than two days ago, it should be processed
let charges_to_do =
crate::database::models::charge_item::ChargeItem::get_chargeable(
crate::database::models::charge_item::DBCharge::get_chargeable(
&pool,
)
.await?;
let prices = product_item::ProductPriceItem::get_many(
let prices = product_item::DBProductPrice::get_many(
&charges_to_do
.iter()
.map(|x| x.price_id)
@@ -2305,7 +2302,7 @@ pub async fn index_billing(
)
.await?;
let users = crate::database::models::User::get_many_ids(
let users = crate::database::models::DBUser::get_many_ids(
&charges_to_do
.iter()
.map(|x| x.user_id)

View File

@@ -1,7 +1,7 @@
use crate::auth::email::send_email;
use crate::auth::validate::get_user_record_from_bearer_token;
use crate::auth::{AuthProvider, AuthenticationError, get_user_from_headers};
use crate::database::models::flow_item::Flow;
use crate::database::models::flow_item::DBFlow;
use crate::database::redis::RedisPool;
use crate::file_hosting::FileHost;
use crate::models::pats::Scopes;
@@ -76,7 +76,7 @@ impl TempUser {
redis: &RedisPool,
) -> Result<crate::database::models::DBUserId, AuthenticationError> {
if let Some(email) = &self.email {
if crate::database::models::User::get_email(email, client)
if crate::database::models::DBUser::get_email(email, client)
.await?
.is_some()
{
@@ -101,7 +101,7 @@ impl TempUser {
}
);
let new_id = crate::database::models::User::get(
let new_id = crate::database::models::DBUser::get(
&test_username,
client,
redis,
@@ -156,7 +156,7 @@ impl TempUser {
};
if let Some(username) = username {
crate::database::models::User {
crate::database::models::DBUser {
id: user_id,
github_id: if provider == AuthProvider::GitHub {
Some(
@@ -1083,7 +1083,7 @@ pub async fn init(
None
};
let state = Flow::OAuth {
let state = DBFlow::OAuth {
user_id,
url: info.url,
provider: info.provider,
@@ -1112,16 +1112,16 @@ pub async fn auth_callback(
let state = state_string.clone();
let res: Result<HttpResponse, AuthenticationError> = async move {
let flow = Flow::get(&state, &redis).await?;
let flow = DBFlow::get(&state, &redis).await?;
// Extract cookie header from request
if let Some(Flow::OAuth {
if let Some(DBFlow::OAuth {
user_id,
provider,
url,
}) = flow
{
Flow::remove(&state, &redis).await?;
DBFlow::remove(&state, &redis).await?;
let token = provider.get_token(query).await?;
let oauth_user = provider.get_user(&token).await?;
@@ -1138,7 +1138,7 @@ pub async fn auth_callback(
.update_user_id(id, Some(&oauth_user.id), &mut transaction)
.await?;
let user = crate::database::models::User::get_id(id, &**client, &redis).await?;
let user = crate::database::models::DBUser::get_id(id, &**client, &redis).await?;
if provider == AuthProvider::PayPal {
sqlx::query!(
@@ -1165,19 +1165,19 @@ pub async fn auth_callback(
}
transaction.commit().await?;
crate::database::models::User::clear_caches(&[(id, None)], &redis).await?;
crate::database::models::DBUser::clear_caches(&[(id, None)], &redis).await?;
Ok(HttpResponse::TemporaryRedirect()
.append_header(("Location", &*url))
.json(serde_json::json!({ "url": url })))
} else {
let user_id = if let Some(user_id) = user_id_opt {
let user = crate::database::models::User::get_id(user_id, &**client, &redis)
let user = crate::database::models::DBUser::get_id(user_id, &**client, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
if user.totp_secret.is_some() {
let flow = Flow::Login2FA { user_id: user.id }
let flow = DBFlow::Login2FA { user_id: user.id }
.insert(Duration::minutes(30), &redis)
.await?;
@@ -1279,7 +1279,7 @@ pub async fn delete_auth_provider(
}
transaction.commit().await?;
crate::database::models::User::clear_caches(
crate::database::models::DBUser::clear_caches(
&[(user.id.into(), None)],
&redis,
)
@@ -1346,7 +1346,7 @@ pub async fn create_account_with_password(
return Err(ApiError::Turnstile);
}
if crate::database::models::User::get(
if crate::database::models::DBUser::get(
&new_account.username,
&**pool,
&redis,
@@ -1385,7 +1385,7 @@ pub async fn create_account_with_password(
.hash_password(new_account.password.as_bytes(), &salt)?
.to_string();
if crate::database::models::User::get_email(&new_account.email, &**pool)
if crate::database::models::DBUser::get_email(&new_account.email, &**pool)
.await?
.is_some()
{
@@ -1394,7 +1394,7 @@ pub async fn create_account_with_password(
));
}
crate::database::models::User {
crate::database::models::DBUser {
id: user_id,
github_id: None,
discord_id: None,
@@ -1426,7 +1426,7 @@ pub async fn create_account_with_password(
let session = issue_session(req, user_id, &mut transaction, &redis).await?;
let res = crate::models::sessions::Session::from(session, true, None);
let flow = Flow::ConfirmEmail {
let flow = DBFlow::ConfirmEmail {
user_id,
confirm_email: new_account.email.clone(),
}
@@ -1467,17 +1467,19 @@ pub async fn login_password(
}
let user = if let Some(user) =
crate::database::models::User::get(&login.username, &**pool, &redis)
crate::database::models::DBUser::get(&login.username, &**pool, &redis)
.await?
{
user
} else {
let user =
crate::database::models::User::get_email(&login.username, &**pool)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
let user = crate::database::models::DBUser::get_email(
&login.username,
&**pool,
)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
crate::database::models::User::get_id(user, &**pool, &redis)
crate::database::models::DBUser::get_id(user, &**pool, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?
};
@@ -1495,7 +1497,7 @@ pub async fn login_password(
.map_err(|_| AuthenticationError::InvalidCredentials)?;
if user.totp_secret.is_some() {
let flow = Flow::Login2FA { user_id: user.id }
let flow = DBFlow::Login2FA { user_id: user.id }
.insert(Duration::minutes(30), &redis)
.await?;
@@ -1568,7 +1570,7 @@ async fn validate_2fa_code(
Ok(true)
} else if allow_backup {
let backup_codes =
crate::database::models::User::get_backup_codes(user_id, pool)
crate::database::models::DBUser::get_backup_codes(user_id, pool)
.await?;
if !backup_codes.contains(&input) {
@@ -1587,7 +1589,7 @@ async fn validate_2fa_code(
.execute(&mut **transaction)
.await?;
crate::database::models::User::clear_caches(
crate::database::models::DBUser::clear_caches(
&[(user_id, None)],
redis,
)
@@ -1607,13 +1609,13 @@ pub async fn login_2fa(
redis: Data<RedisPool>,
login: web::Json<Login2FA>,
) -> Result<HttpResponse, ApiError> {
let flow = Flow::get(&login.flow, &redis)
let flow = DBFlow::get(&login.flow, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
if let Flow::Login2FA { user_id } = flow {
if let DBFlow::Login2FA { user_id } = flow {
let user =
crate::database::models::User::get_id(user_id, &**pool, &redis)
crate::database::models::DBUser::get_id(user_id, &**pool, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
@@ -1634,7 +1636,7 @@ pub async fn login_2fa(
AuthenticationError::InvalidCredentials,
));
}
Flow::remove(&login.flow, &redis).await?;
DBFlow::remove(&login.flow, &redis).await?;
let session =
issue_session(req, user_id, &mut transaction, &redis).await?;
@@ -1670,7 +1672,7 @@ pub async fn begin_2fa_flow(
let string = totp_rs::Secret::generate_secret();
let encoded = string.to_encoded();
let flow = Flow::Initialize2FA {
let flow = DBFlow::Initialize2FA {
user_id: user.id.into(),
secret: encoded.to_string(),
}
@@ -1696,11 +1698,11 @@ pub async fn finish_2fa_flow(
login: web::Json<Login2FA>,
session_queue: Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
let flow = Flow::get(&login.flow, &redis)
let flow = DBFlow::get(&login.flow, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
if let Flow::Initialize2FA { user_id, secret } = flow {
if let DBFlow::Initialize2FA { user_id, secret } = flow {
let user = get_user_from_headers(
&req,
&**pool,
@@ -1735,7 +1737,7 @@ pub async fn finish_2fa_flow(
));
}
Flow::remove(&login.flow, &redis).await?;
DBFlow::remove(&login.flow, &redis).await?;
sqlx::query!(
"
@@ -1794,7 +1796,7 @@ pub async fn finish_2fa_flow(
}
transaction.commit().await?;
crate::database::models::User::clear_caches(
crate::database::models::DBUser::clear_caches(
&[(user.id.into(), None)],
&redis,
)
@@ -1893,7 +1895,7 @@ pub async fn remove_2fa(
}
transaction.commit().await?;
crate::database::models::User::clear_caches(&[(user.id, None)], &redis)
crate::database::models::DBUser::clear_caches(&[(user.id, None)], &redis)
.await?;
Ok(HttpResponse::NoContent().finish())
@@ -1916,15 +1918,17 @@ pub async fn reset_password_begin(
return Err(ApiError::Turnstile);
}
let user = if let Some(user_id) = crate::database::models::User::get_email(
&reset_password.username,
&**pool,
)
.await?
let user = if let Some(user_id) =
crate::database::models::DBUser::get_email(
&reset_password.username,
&**pool,
)
.await?
{
crate::database::models::User::get_id(user_id, &**pool, &redis).await?
crate::database::models::DBUser::get_id(user_id, &**pool, &redis)
.await?
} else {
crate::database::models::User::get(
crate::database::models::DBUser::get(
&reset_password.username,
&**pool,
&redis,
@@ -1933,7 +1937,7 @@ pub async fn reset_password_begin(
};
if let Some(user) = user {
let flow = Flow::ForgotPassword { user_id: user.id }
let flow = DBFlow::ForgotPassword { user_id: user.id }
.insert(Duration::hours(24), &redis)
.await?;
@@ -1975,13 +1979,14 @@ pub async fn change_password(
session_queue: Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
let user = if let Some(flow) = &change_password.flow {
let flow = Flow::get(flow, &redis).await?;
let flow = DBFlow::get(flow, &redis).await?;
if let Some(Flow::ForgotPassword { user_id }) = flow {
let user =
crate::database::models::User::get_id(user_id, &**pool, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
if let Some(DBFlow::ForgotPassword { user_id }) = flow {
let user = crate::database::models::DBUser::get_id(
user_id, &**pool, &redis,
)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
Some(user)
} else {
@@ -2085,7 +2090,7 @@ pub async fn change_password(
.await?;
if let Some(flow) = &change_password.flow {
Flow::remove(flow, &redis).await?;
DBFlow::remove(flow, &redis).await?;
}
if let Some(email) = user.email {
@@ -2105,7 +2110,7 @@ pub async fn change_password(
}
transaction.commit().await?;
crate::database::models::User::clear_caches(&[(user.id, None)], &redis)
crate::database::models::DBUser::clear_caches(&[(user.id, None)], &redis)
.await?;
Ok(HttpResponse::Ok().finish())
@@ -2183,7 +2188,7 @@ pub async fn set_email(
.await?;
}
let flow = Flow::ConfirmEmail {
let flow = DBFlow::ConfirmEmail {
user_id: user.id.into(),
confirm_email: email.email.clone(),
}
@@ -2197,7 +2202,7 @@ pub async fn set_email(
)?;
transaction.commit().await?;
crate::database::models::User::clear_caches(
crate::database::models::DBUser::clear_caches(
&[(user.id.into(), None)],
&redis,
)
@@ -2230,7 +2235,7 @@ pub async fn resend_verify_email(
));
}
let flow = Flow::ConfirmEmail {
let flow = DBFlow::ConfirmEmail {
user_id: user.id.into(),
confirm_email: email.clone(),
}
@@ -2262,15 +2267,15 @@ pub async fn verify_email(
redis: Data<RedisPool>,
email: web::Json<VerifyEmail>,
) -> Result<HttpResponse, ApiError> {
let flow = Flow::get(&email.flow, &redis).await?;
let flow = DBFlow::get(&email.flow, &redis).await?;
if let Some(Flow::ConfirmEmail {
if let Some(DBFlow::ConfirmEmail {
user_id,
confirm_email,
}) = flow
{
let user =
crate::database::models::User::get_id(user_id, &**pool, &redis)
crate::database::models::DBUser::get_id(user_id, &**pool, &redis)
.await?
.ok_or_else(|| AuthenticationError::InvalidCredentials)?;
@@ -2294,10 +2299,13 @@ pub async fn verify_email(
.execute(&mut *transaction)
.await?;
Flow::remove(&email.flow, &redis).await?;
DBFlow::remove(&email.flow, &redis).await?;
transaction.commit().await?;
crate::database::models::User::clear_caches(&[(user.id, None)], &redis)
.await?;
crate::database::models::DBUser::clear_caches(
&[(user.id, None)],
&redis,
)
.await?;
Ok(HttpResponse::NoContent().finish())
} else {

View File

@@ -30,9 +30,9 @@ pub async fn export(
let user_id = user.id.into();
let collection_ids =
crate::database::models::User::get_collections(user_id, &**pool)
crate::database::models::DBUser::get_collections(user_id, &**pool)
.await?;
let collections = crate::database::models::Collection::get_many(
let collections = crate::database::models::DBCollection::get_many(
&collection_ids,
&**pool,
&redis,
@@ -42,24 +42,25 @@ pub async fn export(
.map(crate::models::collections::Collection::from)
.collect::<Vec<_>>();
let follows = crate::database::models::User::get_follows(user_id, &**pool)
.await?
.into_iter()
.map(crate::models::ids::ProjectId::from)
.collect::<Vec<_>>();
let follows =
crate::database::models::DBUser::get_follows(user_id, &**pool)
.await?
.into_iter()
.map(crate::models::ids::ProjectId::from)
.collect::<Vec<_>>();
let projects =
crate::database::models::User::get_projects(user_id, &**pool, &redis)
crate::database::models::DBUser::get_projects(user_id, &**pool, &redis)
.await?
.into_iter()
.map(crate::models::ids::ProjectId::from)
.collect::<Vec<_>>();
let org_ids =
crate::database::models::User::get_organizations(user_id, &**pool)
crate::database::models::DBUser::get_organizations(user_id, &**pool)
.await?;
let orgs =
crate::database::models::organization_item::Organization::get_many_ids(
crate::database::models::organization_item::DBOrganization::get_many_ids(
&org_ids, &**pool, &redis,
)
.await?
@@ -68,7 +69,7 @@ pub async fn export(
.map(|x| crate::models::organizations::Organization::from(x, vec![]))
.collect::<Vec<_>>();
let notifs = crate::database::models::notification_item::Notification::get_many_user(
let notifs = crate::database::models::notification_item::DBNotification::get_many_user(
user_id, &**pool, &redis,
)
.await?
@@ -77,7 +78,7 @@ pub async fn export(
.collect::<Vec<_>>();
let oauth_clients =
crate::database::models::oauth_client_item::OAuthClient::get_all_user_clients(
crate::database::models::oauth_client_item::DBOAuthClient::get_all_user_clients(
user_id, &**pool,
)
.await?
@@ -85,7 +86,7 @@ pub async fn export(
.map(crate::models::oauth_clients::OAuthClient::from)
.collect::<Vec<_>>();
let oauth_authorizations = crate::database::models::oauth_client_authorization_item::OAuthClientAuthorization::get_all_for_user(
let oauth_authorizations = crate::database::models::oauth_client_authorization_item::DBOAuthClientAuthorization::get_all_for_user(
user_id, &**pool,
)
.await?
@@ -94,12 +95,12 @@ pub async fn export(
.collect::<Vec<_>>();
let pat_ids =
crate::database::models::pat_item::PersonalAccessToken::get_user_pats(
crate::database::models::pat_item::DBPersonalAccessToken::get_user_pats(
user_id, &**pool, &redis,
)
.await?;
let pats =
crate::database::models::pat_item::PersonalAccessToken::get_many_ids(
crate::database::models::pat_item::DBPersonalAccessToken::get_many_ids(
&pat_ids, &**pool, &redis,
)
.await?
@@ -108,12 +109,12 @@ pub async fn export(
.collect::<Vec<_>>();
let payout_ids =
crate::database::models::payout_item::Payout::get_all_for_user(
crate::database::models::payout_item::DBPayout::get_all_for_user(
user_id, &**pool,
)
.await?;
let payouts = crate::database::models::payout_item::Payout::get_many(
let payouts = crate::database::models::payout_item::DBPayout::get_many(
&payout_ids,
&**pool,
)
@@ -122,10 +123,11 @@ pub async fn export(
.map(crate::models::payouts::Payout::from)
.collect::<Vec<_>>();
let report_ids =
crate::database::models::user_item::User::get_reports(user_id, &**pool)
.await?;
let reports = crate::database::models::report_item::Report::get_many(
let report_ids = crate::database::models::user_item::DBUser::get_reports(
user_id, &**pool,
)
.await?;
let reports = crate::database::models::report_item::DBReport::get_many(
&report_ids,
&**pool,
)
@@ -147,7 +149,7 @@ pub async fn export(
.collect::<Vec<_>>();
let messages =
crate::database::models::thread_item::ThreadMessage::get_many(
crate::database::models::thread_item::DBThreadMessage::get_many(
&message_ids,
&**pool,
)
@@ -166,18 +168,19 @@ pub async fn export(
.map(|x| crate::database::models::ids::DBImageId(x.id))
.collect::<Vec<_>>();
let uploaded_images = crate::database::models::image_item::Image::get_many(
&uploaded_images_ids,
&**pool,
&redis,
)
.await?
.into_iter()
.map(crate::models::images::Image::from)
.collect::<Vec<_>>();
let uploaded_images =
crate::database::models::image_item::DBImage::get_many(
&uploaded_images_ids,
&**pool,
&redis,
)
.await?
.into_iter()
.map(crate::models::images::Image::from)
.collect::<Vec<_>>();
let subscriptions =
crate::database::models::user_subscription_item::UserSubscriptionItem::get_all_user(
crate::database::models::user_subscription_item::DBUserSubscription::get_all_user(
user_id, &**pool,
)
.await?

View File

@@ -61,7 +61,7 @@ pub async fn get_projects(
.await?;
let projects: Vec<_> =
database::Project::get_many_ids(&project_ids, &**pool, &redis)
database::DBProject::get_many_ids(&project_ids, &**pool, &redis)
.await?
.into_iter()
.map(crate::models::projects::Project::from)
@@ -88,7 +88,7 @@ pub async fn get_project_meta(
let project_id = info.into_inner().0;
let project =
database::models::Project::get(&project_id, &**pool, &redis).await?;
database::models::DBProject::get(&project_id, &**pool, &redis).await?;
if let Some(project) = project {
let rows = sqlx::query!(

View File

@@ -45,13 +45,13 @@ pub async fn get_pats(
.1;
let pat_ids =
database::models::pat_item::PersonalAccessToken::get_user_pats(
database::models::pat_item::DBPersonalAccessToken::get_user_pats(
user.id.into(),
&**pool,
&redis,
)
.await?;
let pats = database::models::pat_item::PersonalAccessToken::get_many_ids(
let pats = database::models::pat_item::DBPersonalAccessToken::get_many_ids(
&pat_ids, &**pool, &redis,
)
.await?;
@@ -116,7 +116,7 @@ pub async fn create_pat(
let token = format!("mrp_{token}");
let name = info.name.clone();
database::models::pat_item::PersonalAccessToken {
database::models::pat_item::DBPersonalAccessToken {
id,
name: name.clone(),
access_token: token.clone(),
@@ -130,7 +130,7 @@ pub async fn create_pat(
.await?;
transaction.commit().await?;
database::models::pat_item::PersonalAccessToken::clear_cache(
database::models::pat_item::DBPersonalAccessToken::clear_cache(
vec![(None, None, Some(user.id.into()))],
&redis,
)
@@ -180,7 +180,7 @@ pub async fn edit_pat(
.1;
let id = id.into_inner().0;
let pat = database::models::pat_item::PersonalAccessToken::get(
let pat = database::models::pat_item::DBPersonalAccessToken::get(
&id, &**pool, &redis,
)
.await?;
@@ -242,7 +242,7 @@ pub async fn edit_pat(
}
transaction.commit().await?;
database::models::pat_item::PersonalAccessToken::clear_cache(
database::models::pat_item::DBPersonalAccessToken::clear_cache(
vec![(Some(pat.id), Some(pat.access_token), Some(pat.user_id))],
&redis,
)
@@ -271,7 +271,7 @@ pub async fn delete_pat(
.await?
.1;
let id = id.into_inner().0;
let pat = database::models::pat_item::PersonalAccessToken::get(
let pat = database::models::pat_item::DBPersonalAccessToken::get(
&id, &**pool, &redis,
)
.await?;
@@ -279,13 +279,13 @@ pub async fn delete_pat(
if let Some(pat) = pat {
if pat.user_id == user.id.into() {
let mut transaction = pool.begin().await?;
database::models::pat_item::PersonalAccessToken::remove(
database::models::pat_item::DBPersonalAccessToken::remove(
pat.id,
&mut transaction,
)
.await?;
transaction.commit().await?;
database::models::pat_item::PersonalAccessToken::clear_cache(
database::models::pat_item::DBPersonalAccessToken::clear_cache(
vec![(Some(pat.id), Some(pat.access_token), Some(pat.user_id))],
&redis,
)

View File

@@ -1,6 +1,6 @@
use crate::auth::{AuthenticationError, get_user_from_headers};
use crate::database::models::DBUserId;
use crate::database::models::session_item::Session as DBSession;
use crate::database::models::session_item::DBSession;
use crate::database::models::session_item::SessionBuilder;
use crate::database::redis::RedisPool;
use crate::models::pats::Scopes;

View File

@@ -1,6 +1,6 @@
use crate::auth::AuthenticationError;
use crate::auth::validate::get_user_record_from_bearer_token;
use crate::database::models::friend_item::FriendItem;
use crate::database::models::friend_item::DBFriend;
use crate::database::redis::RedisPool;
use crate::models::pats::Scopes;
use crate::models::users::User;
@@ -85,8 +85,7 @@ pub async fn ws_init(
};
let friends =
FriendItem::get_user_friends(user.id.into(), Some(true), &**pool)
.await?;
DBFriend::get_user_friends(user.id.into(), Some(true), &**pool).await?;
let friend_statuses = if !friends.is_empty() {
let db = db.clone();
@@ -383,7 +382,7 @@ pub async fn broadcast_to_local_friends(
user_id,
message,
sockets,
FriendItem::get_user_friends(user_id.into(), Some(true), pool).await?,
DBFriend::get_user_friends(user_id.into(), Some(true), pool).await?,
)
.await
}
@@ -392,7 +391,7 @@ async fn broadcast_to_known_local_friends(
user_id: UserId,
message: ServerToClientMessage,
sockets: &ActiveSockets,
friends: Vec<FriendItem>,
friends: Vec<DBFriend>,
) -> Result<(), crate::database::models::DatabaseError> {
// FIXME Probably shouldn't be using database errors for this. Maybe ApiError?

View File

@@ -1,8 +1,10 @@
use crate::auth::checks::{is_visible_project, is_visible_version};
use crate::database::models::legacy_loader_fields::MinecraftGameVersion;
use crate::database::models::loader_fields::Loader;
use crate::database::models::project_item::QueryProject;
use crate::database::models::version_item::{QueryFile, QueryVersion};
use crate::database::models::project_item::ProjectQueryResult;
use crate::database::models::version_item::{
FileQueryResult, VersionQueryResult,
};
use crate::database::redis::RedisPool;
use crate::models::ids::{ProjectId, VersionId};
use crate::models::pats::Scopes;
@@ -76,7 +78,7 @@ pub async fn maven_metadata(
) -> Result<HttpResponse, ApiError> {
let project_id = params.into_inner().0;
let Some(project) =
database::models::Project::get(&project_id, &**pool, &redis).await?
database::models::DBProject::get(&project_id, &**pool, &redis).await?
else {
return Err(ApiError::NotFound);
};
@@ -159,17 +161,17 @@ pub async fn maven_metadata(
}
async fn find_version(
project: &QueryProject,
project: &ProjectQueryResult,
vcoords: &String,
pool: &PgPool,
redis: &RedisPool,
) -> Result<Option<QueryVersion>, ApiError> {
) -> Result<Option<VersionQueryResult>, ApiError> {
let id_option = ariadne::ids::base62_impl::parse_base62(vcoords)
.ok()
.map(|x| x as i64);
let all_versions =
database::models::Version::get_many(&project.versions, pool, redis)
database::models::DBVersion::get_many(&project.versions, pool, redis)
.await?;
let exact_matches = all_versions
@@ -236,9 +238,9 @@ async fn find_version(
fn find_file<'a>(
project_id: &str,
vcoords: &str,
version: &'a QueryVersion,
version: &'a VersionQueryResult,
file: &str,
) -> Option<&'a QueryFile> {
) -> Option<&'a FileQueryResult> {
if let Some(selected_file) =
version.files.iter().find(|x| x.filename == file)
{
@@ -282,7 +284,7 @@ pub async fn version_file(
) -> Result<HttpResponse, ApiError> {
let (project_id, vnum, file) = params.into_inner();
let Some(project) =
database::models::Project::get(&project_id, &**pool, &redis).await?
database::models::DBProject::get(&project_id, &**pool, &redis).await?
else {
return Err(ApiError::NotFound);
};
@@ -348,7 +350,7 @@ pub async fn version_file_sha1(
) -> Result<HttpResponse, ApiError> {
let (project_id, vnum, file) = params.into_inner();
let Some(project) =
database::models::Project::get(&project_id, &**pool, &redis).await?
database::models::DBProject::get(&project_id, &**pool, &redis).await?
else {
return Err(ApiError::NotFound);
};
@@ -393,7 +395,7 @@ pub async fn version_file_sha512(
) -> Result<HttpResponse, ApiError> {
let (project_id, vnum, file) = params.into_inner();
let Some(project) =
database::models::Project::get(&project_id, &**pool, &redis).await?
database::models::DBProject::get(&project_id, &**pool, &redis).await?
else {
return Err(ApiError::NotFound);
};

View File

@@ -42,7 +42,7 @@ pub async fn forge_updates(
let (id,) = info.into_inner();
let project = database::models::Project::get(&id, &**pool, &redis)
let project = database::models::DBProject::get(&id, &**pool, &redis)
.await?
.ok_or_else(|| ApiError::InvalidInput(ERROR.to_string()))?;
@@ -61,9 +61,12 @@ pub async fn forge_updates(
return Err(ApiError::InvalidInput(ERROR.to_string()));
}
let versions =
database::models::Version::get_many(&project.versions, &**pool, &redis)
.await?;
let versions = database::models::DBVersion::get_many(
&project.versions,
&**pool,
&redis,
)
.await?;
let loaders = match &*neo.neoforge {
"only" => |x: &String| *x == "neoforge",

View File

@@ -258,8 +258,12 @@ pub async fn project_create(
Ok(project) => {
let version_item = match project.versions.first() {
Some(vid) => {
version_item::Version::get((*vid).into(), &**client, &redis)
.await?
version_item::DBVersion::get(
(*vid).into(),
&**client,
&redis,
)
.await?
}
None => None,
};

View File

@@ -229,7 +229,7 @@ pub async fn project_get(
Ok(project) => {
let version_item = match project.versions.first() {
Some(vid) => {
version_item::Version::get((*vid).into(), &**pool, &redis)
version_item::DBVersion::get((*vid).into(), &**pool, &redis)
.await?
}
None => None,
@@ -469,7 +469,7 @@ pub async fn project_edit(
if let Some(donation_urls) = v2_new_project.donation_urls {
// Fetch current donation links from project so we know what to delete
let fetched_example_project =
project_item::Project::get(&info.0, &**pool, &redis).await?;
project_item::DBProject::get(&info.0, &**pool, &redis).await?;
let donation_links = fetched_example_project
.map(|x| {
x.urls
@@ -533,7 +533,7 @@ pub async fn project_edit(
if response.status().is_success()
&& (client_side.is_some() || server_side.is_some())
{
let project_item = project_item::Project::get(
let project_item = project_item::DBProject::get(
&new_slug.unwrap_or(project_id),
&**pool,
&redis,
@@ -541,7 +541,7 @@ pub async fn project_edit(
.await?;
let version_ids = project_item.map(|x| x.versions).unwrap_or_default();
let versions =
version_item::Version::get_many(&version_ids, &**pool, &redis)
version_item::DBVersion::get_many(&version_ids, &**pool, &redis)
.await?;
for version in versions {
let version = Version::from(version);

View File

@@ -288,17 +288,20 @@ async fn get_example_version_fields(
None => return Ok(None),
};
let vid =
match project_item::Project::get_id(project_id.into(), &**pool, redis)
.await?
.and_then(|p| p.versions.first().cloned())
{
Some(vid) => vid,
None => return Ok(None),
};
let vid = match project_item::DBProject::get_id(
project_id.into(),
&**pool,
redis,
)
.await?
.and_then(|p| p.versions.first().cloned())
{
Some(vid) => vid,
None => return Ok(None),
};
let example_version =
match version_item::Version::get(vid, &**pool, redis).await? {
match version_item::DBVersion::get(vid, &**pool, redis).await? {
Some(version) => version,
None => return Ok(None),
};

View File

@@ -578,7 +578,7 @@ async fn filter_allowed_ids(
// If no project_ids or version_ids are provided, we default to all projects the user has *public* access to
if project_ids.is_none() && !remove_defaults.unwrap_or(false) {
project_ids = Some(
user_item::User::get_projects(user.id.into(), &***pool, redis)
user_item::DBUser::get_projects(user.id.into(), &***pool, redis)
.await?
.into_iter()
.map(|x| ProjectId::from(x).to_string())
@@ -589,7 +589,7 @@ async fn filter_allowed_ids(
// Convert String list to list of ProjectIds or VersionIds
// - Filter out unauthorized projects/versions
let project_ids = if let Some(project_strings) = project_ids {
let projects_data = database::models::Project::get_many(
let projects_data = database::models::DBProject::get_many(
&project_strings,
&***pool,
redis,
@@ -601,7 +601,7 @@ async fn filter_allowed_ids(
.map(|x| x.inner.team_id)
.collect::<Vec<database::models::DBTeamId>>();
let team_members =
database::models::TeamMember::get_from_team_full_many(
database::models::DBTeamMember::get_from_team_full_many(
&team_ids, &***pool, redis,
)
.await?;
@@ -610,7 +610,7 @@ async fn filter_allowed_ids(
.iter()
.filter_map(|x| x.inner.organization_id)
.collect::<Vec<database::models::DBOrganizationId>>();
let organizations = database::models::Organization::get_many_ids(
let organizations = database::models::DBOrganization::get_many_ids(
&organization_ids,
&***pool,
redis,
@@ -622,7 +622,7 @@ async fn filter_allowed_ids(
.map(|x| x.team_id)
.collect::<Vec<database::models::DBTeamId>>();
let organization_team_members =
database::models::TeamMember::get_from_team_full_many(
database::models::DBTeamMember::get_from_team_full_many(
&organization_team_ids,
&***pool,
redis,

View File

@@ -85,7 +85,7 @@ pub async fn collection_create(
let collection_id: CollectionId =
generate_collection_id(&mut transaction).await?.into();
let initial_project_ids = project_item::Project::get_many(
let initial_project_ids = project_item::DBProject::get_many(
&collection_create_data.projects,
&mut *transaction,
&redis,
@@ -149,7 +149,7 @@ pub async fn collections_get(
.collect::<Result<Vec<_>, _>>()?;
let collections_data =
database::models::Collection::get_many(&ids, &**pool, &redis).await?;
database::models::DBCollection::get_many(&ids, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
@@ -179,7 +179,7 @@ pub async fn collection_get(
let id = database::models::DBCollectionId(parse_base62(&string)? as i64);
let collection_data =
database::models::Collection::get(id, &**pool, &redis).await?;
database::models::DBCollection::get(id, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
&**pool,
@@ -242,7 +242,8 @@ pub async fn collection_edit(
let string = info.into_inner().0;
let id = database::models::DBCollectionId(parse_base62(&string)? as i64);
let result = database::models::Collection::get(id, &**pool, &redis).await?;
let result =
database::models::DBCollection::get(id, &**pool, &redis).await?;
if let Some(collection_item) = result {
if !can_modify_collection(&collection_item, &user) {
@@ -322,7 +323,7 @@ pub async fn collection_edit(
.collect_vec();
let mut validated_project_ids = Vec::new();
for project_id in new_project_ids {
let project = database::models::Project::get(
let project = database::models::DBProject::get(
project_id, &**pool, &redis,
)
.await?
@@ -359,7 +360,7 @@ pub async fn collection_edit(
}
transaction.commit().await?;
database::models::Collection::clear_cache(collection_item.id, &redis)
database::models::DBCollection::clear_cache(collection_item.id, &redis)
.await?;
Ok(HttpResponse::NoContent().body(""))
@@ -397,7 +398,7 @@ pub async fn collection_icon_edit(
let string = info.into_inner().0;
let id = database::models::DBCollectionId(parse_base62(&string)? as i64);
let collection_item =
database::models::Collection::get(id, &**pool, &redis)
database::models::DBCollection::get(id, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -451,7 +452,7 @@ pub async fn collection_icon_edit(
.await?;
transaction.commit().await?;
database::models::Collection::clear_cache(collection_item.id, &redis)
database::models::DBCollection::clear_cache(collection_item.id, &redis)
.await?;
Ok(HttpResponse::NoContent().body(""))
@@ -478,7 +479,7 @@ pub async fn delete_collection_icon(
let string = info.into_inner().0;
let id = database::models::DBCollectionId(parse_base62(&string)? as i64);
let collection_item =
database::models::Collection::get(id, &**pool, &redis)
database::models::DBCollection::get(id, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -509,7 +510,7 @@ pub async fn delete_collection_icon(
.await?;
transaction.commit().await?;
database::models::Collection::clear_cache(collection_item.id, &redis)
database::models::DBCollection::clear_cache(collection_item.id, &redis)
.await?;
Ok(HttpResponse::NoContent().body(""))
@@ -534,7 +535,7 @@ pub async fn collection_delete(
let string = info.into_inner().0;
let id = database::models::DBCollectionId(parse_base62(&string)? as i64);
let collection = database::models::Collection::get(id, &**pool, &redis)
let collection = database::models::DBCollection::get(id, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -546,7 +547,7 @@ pub async fn collection_delete(
}
let mut transaction = pool.begin().await?;
let result = database::models::Collection::remove(
let result = database::models::DBCollection::remove(
collection.id,
&mut transaction,
&redis,
@@ -554,7 +555,7 @@ pub async fn collection_delete(
.await?;
transaction.commit().await?;
database::models::Collection::clear_cache(collection.id, &redis).await?;
database::models::DBCollection::clear_cache(collection.id, &redis).await?;
if result.is_some() {
Ok(HttpResponse::NoContent().body(""))
@@ -564,7 +565,7 @@ pub async fn collection_delete(
}
fn can_modify_collection(
collection: &database::models::Collection,
collection: &database::models::DBCollection,
user: &models::users::User,
) -> bool {
collection.user_id == user.id.into() || user.role.is_mod()

View File

@@ -43,13 +43,13 @@ pub async fn add_friend(
let string = info.into_inner().0;
let friend =
crate::database::models::User::get(&string, &**pool, &redis).await?;
crate::database::models::DBUser::get(&string, &**pool, &redis).await?;
if let Some(friend) = friend {
let mut transaction = pool.begin().await?;
if let Some(friend) =
crate::database::models::friend_item::FriendItem::get_friend(
crate::database::models::friend_item::DBFriend::get_friend(
user.id.into(),
friend.id,
&**pool,
@@ -68,7 +68,7 @@ pub async fn add_friend(
));
}
crate::database::models::friend_item::FriendItem::update_friend(
crate::database::models::friend_item::DBFriend::update_friend(
friend.user_id,
friend.friend_id,
true,
@@ -115,7 +115,7 @@ pub async fn add_friend(
));
}
crate::database::models::friend_item::FriendItem {
crate::database::models::friend_item::DBFriend {
user_id: user.id.into(),
friend_id: friend.id,
created: Utc::now(),
@@ -161,12 +161,12 @@ pub async fn remove_friend(
let string = info.into_inner().0;
let friend =
crate::database::models::User::get(&string, &**pool, &redis).await?;
crate::database::models::DBUser::get(&string, &**pool, &redis).await?;
if let Some(friend) = friend {
let mut transaction = pool.begin().await?;
crate::database::models::friend_item::FriendItem::remove(
crate::database::models::friend_item::DBFriend::remove(
user.id.into(),
friend.id,
&mut transaction,
@@ -206,7 +206,7 @@ pub async fn friends(
.1;
let friends =
crate::database::models::friend_item::FriendItem::get_user_friends(
crate::database::models::friend_item::DBFriend::get_user_friends(
user.id.into(),
None,
&**pool,

View File

@@ -67,7 +67,7 @@ pub async fn images_add(
ImageContext::Project { project_id } => {
if let Some(id) = data.project_id {
let project =
project_item::Project::get(&id, &**pool, &redis).await?;
project_item::DBProject::get(&id, &**pool, &redis).await?;
if let Some(project) = project {
if is_team_member_project(
&project.inner,
@@ -92,7 +92,7 @@ pub async fn images_add(
ImageContext::Version { version_id } => {
if let Some(id) = data.version_id {
let version =
version_item::Version::get(id.into(), &**pool, &redis)
version_item::DBVersion::get(id.into(), &**pool, &redis)
.await?;
if let Some(version) = version {
if is_team_member_version(
@@ -119,7 +119,7 @@ pub async fn images_add(
ImageContext::ThreadMessage { thread_message_id } => {
if let Some(id) = data.thread_message_id {
let thread_message =
thread_item::ThreadMessage::get(id.into(), &**pool)
thread_item::DBThreadMessage::get(id.into(), &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -127,7 +127,7 @@ pub async fn images_add(
.to_string(),
)
})?;
let thread = thread_item::Thread::get(thread_message.thread_id, &**pool)
let thread = thread_item::DBThread::get(thread_message.thread_id, &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -147,14 +147,14 @@ pub async fn images_add(
}
ImageContext::Report { report_id } => {
if let Some(id) = data.report_id {
let report = report_item::Report::get(id.into(), &**pool)
let report = report_item::DBReport::get(id.into(), &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
"The report could not be found.".to_string(),
)
})?;
let thread = thread_item::Thread::get(report.thread_id, &**pool)
let thread = thread_item::DBThread::get(report.thread_id, &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -198,7 +198,7 @@ pub async fn images_add(
let mut transaction = pool.begin().await?;
let db_image: database::models::Image = database::models::Image {
let db_image: database::models::DBImage = database::models::DBImage {
id: database::models::generate_image_id(&mut transaction).await?,
url: upload_result.url,
raw_url: upload_result.raw_url,

View File

@@ -46,7 +46,7 @@ pub async fn notifications_get(
.1;
use database::models::DBNotificationId;
use database::models::notification_item::Notification as DBNotification;
use database::models::notification_item::DBNotification;
let notification_ids: Vec<DBNotificationId> =
serde_json::from_str::<Vec<NotificationId>>(ids.ids.as_str())?
@@ -55,7 +55,7 @@ pub async fn notifications_get(
.collect();
let notifications_data: Vec<DBNotification> =
database::models::notification_item::Notification::get_many(
database::models::notification_item::DBNotification::get_many(
&notification_ids,
&**pool,
)
@@ -90,7 +90,7 @@ pub async fn notification_get(
let id = info.into_inner().0;
let notification_data =
database::models::notification_item::Notification::get(
database::models::notification_item::DBNotification::get(
id.into(),
&**pool,
)
@@ -127,7 +127,7 @@ pub async fn notification_read(
let id = info.into_inner().0;
let notification_data =
database::models::notification_item::Notification::get(
database::models::notification_item::DBNotification::get(
id.into(),
&**pool,
)
@@ -137,7 +137,7 @@ pub async fn notification_read(
if data.user_id == user.id.into() || user.role.is_admin() {
let mut transaction = pool.begin().await?;
database::models::notification_item::Notification::read(
database::models::notification_item::DBNotification::read(
id.into(),
&mut transaction,
&redis,
@@ -177,7 +177,7 @@ pub async fn notification_delete(
let id = info.into_inner().0;
let notification_data =
database::models::notification_item::Notification::get(
database::models::notification_item::DBNotification::get(
id.into(),
&**pool,
)
@@ -187,7 +187,7 @@ pub async fn notification_delete(
if data.user_id == user.id.into() || user.role.is_admin() {
let mut transaction = pool.begin().await?;
database::models::notification_item::Notification::remove(
database::models::notification_item::DBNotification::remove(
id.into(),
&mut transaction,
&redis,
@@ -234,7 +234,7 @@ pub async fn notifications_read(
let mut transaction = pool.begin().await?;
let notifications_data =
database::models::notification_item::Notification::get_many(
database::models::notification_item::DBNotification::get_many(
&notification_ids,
&**pool,
)
@@ -249,7 +249,7 @@ pub async fn notifications_read(
}
}
database::models::notification_item::Notification::read_many(
database::models::notification_item::DBNotification::read_many(
&notifications,
&mut transaction,
&redis,
@@ -287,7 +287,7 @@ pub async fn notifications_delete(
let mut transaction = pool.begin().await?;
let notifications_data =
database::models::notification_item::Notification::get_many(
database::models::notification_item::DBNotification::get_many(
&notification_ids,
&**pool,
)
@@ -302,7 +302,7 @@ pub async fn notifications_delete(
}
}
database::models::notification_item::Notification::remove_many(
database::models::notification_item::DBNotification::remove_many(
&notifications,
&mut transaction,
&redis,

View File

@@ -5,10 +5,10 @@ use crate::{
auth::{checks::ValidateAuthorized, get_user_from_headers},
database::{
models::{
DBOAuthClientId, DatabaseError, User, generate_oauth_client_id,
DBOAuthClientId, DBUser, DatabaseError, generate_oauth_client_id,
generate_oauth_redirect_id,
oauth_client_authorization_item::OAuthClientAuthorization,
oauth_client_item::{OAuthClient, OAuthRedirectUri},
oauth_client_authorization_item::DBOAuthClientAuthorization,
oauth_client_item::{DBOAuthClient, DBOAuthRedirectUri},
},
redis::RedisPool,
},
@@ -38,7 +38,6 @@ use serde::{Deserialize, Serialize};
use sqlx::PgPool;
use validator::Validate;
use crate::database::models::oauth_client_item::OAuthClient as DBOAuthClient;
use crate::models::ids::OAuthClientId;
use crate::util::img::{delete_old_images, upload_image_optimized};
@@ -75,7 +74,7 @@ pub async fn get_user_clients(
.await?
.1;
let target_user = User::get(&info.into_inner(), &**pool, &redis).await?;
let target_user = DBUser::get(&info.into_inner(), &**pool, &redis).await?;
if let Some(target_user) = target_user {
if target_user.id != current_user.id.into()
@@ -87,7 +86,8 @@ pub async fn get_user_clients(
}
let clients =
OAuthClient::get_all_user_clients(target_user.id, &**pool).await?;
DBOAuthClient::get_all_user_clients(target_user.id, &**pool)
.await?;
let response = clients
.into_iter()
@@ -190,7 +190,7 @@ pub async fn oauth_client_create(
)
.await?;
let client = OAuthClient {
let client = DBOAuthClient {
id: client_id,
icon_url: None,
raw_icon_url: None,
@@ -234,10 +234,10 @@ pub async fn oauth_client_delete(
.1;
let client =
OAuthClient::get(client_id.into_inner().into(), &**pool).await?;
DBOAuthClient::get(client_id.into_inner().into(), &**pool).await?;
if let Some(client) = client {
client.validate_authorized(Some(&current_user))?;
OAuthClient::remove(client.id, &**pool).await?;
DBOAuthClient::remove(client.id, &**pool).await?;
Ok(HttpResponse::NoContent().body(""))
} else {
@@ -295,7 +295,7 @@ pub async fn oauth_client_edit(
})?;
if let Some(existing_client) =
OAuthClient::get(client_id.into_inner().into(), &**pool).await?
DBOAuthClient::get(client_id.into_inner().into(), &**pool).await?
{
existing_client.validate_authorized(Some(&current_user))?;
@@ -368,7 +368,7 @@ pub async fn oauth_client_icon_edit(
.await?
.1;
let client = OAuthClient::get((*client_id).into(), &**pool)
let client = DBOAuthClient::get((*client_id).into(), &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -435,7 +435,7 @@ pub async fn oauth_client_icon_delete(
.await?
.1;
let client = OAuthClient::get((*client_id).into(), &**pool)
let client = DBOAuthClient::get((*client_id).into(), &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -482,7 +482,7 @@ pub async fn get_user_oauth_authorizations(
.await?
.1;
let authorizations = OAuthClientAuthorization::get_all_for_user(
let authorizations = DBOAuthClientAuthorization::get_all_for_user(
current_user.id.into(),
&**pool,
)
@@ -512,7 +512,7 @@ pub async fn revoke_oauth_authorization(
.await?
.1;
OAuthClientAuthorization::remove(
DBOAuthClientAuthorization::remove(
info.client_id.into(),
current_user.id.into(),
&**pool,
@@ -534,11 +534,11 @@ async fn create_redirect_uris(
uri_strings: impl IntoIterator<Item = impl Display>,
client_id: DBOAuthClientId,
transaction: &mut sqlx::Transaction<'_, sqlx::Postgres>,
) -> Result<Vec<OAuthRedirectUri>, DatabaseError> {
) -> Result<Vec<DBOAuthRedirectUri>, DatabaseError> {
let mut redirect_uris = vec![];
for uri in uri_strings.into_iter() {
let id = generate_oauth_redirect_id(transaction).await?;
redirect_uris.push(OAuthRedirectUri {
redirect_uris.push(DBOAuthRedirectUri {
id,
client_id,
uri: uri.to_string(),
@@ -550,7 +550,7 @@ async fn create_redirect_uris(
async fn edit_redirects(
redirects: Vec<String>,
existing_client: &OAuthClient,
existing_client: &DBOAuthClient,
transaction: &mut sqlx::Transaction<'_, sqlx::Postgres>,
) -> Result<(), DatabaseError> {
let updated_redirects: HashSet<String> = redirects.into_iter().collect();
@@ -566,12 +566,12 @@ async fn edit_redirects(
&mut *transaction,
)
.await?;
OAuthClient::insert_redirect_uris(&redirects_to_add, &mut **transaction)
DBOAuthClient::insert_redirect_uris(&redirects_to_add, &mut **transaction)
.await?;
let mut redirects_to_remove = existing_client.redirect_uris.clone();
redirects_to_remove.retain(|r| !updated_redirects.contains(&r.uri));
OAuthClient::remove_redirect_uris(
DBOAuthClient::remove_redirect_uris(
redirects_to_remove.iter().map(|r| r.id),
&mut **transaction,
)
@@ -585,7 +585,7 @@ pub async fn get_clients_inner(
pool: web::Data<PgPool>,
) -> Result<Vec<models::oauth_clients::OAuthClient>, ApiError> {
let ids: Vec<DBOAuthClientId> = ids.iter().map(|i| (*i).into()).collect();
let clients = OAuthClient::get_many(&ids, &**pool).await?;
let clients = DBOAuthClient::get_many(&ids, &**pool).await?;
Ok(clients.into_iter().map(|c| c.into()).collect_vec())
}

View File

@@ -3,9 +3,9 @@ use std::sync::Arc;
use super::ApiError;
use crate::auth::{filter_visible_projects, get_user_from_headers};
use crate::database::models::team_item::TeamMember;
use crate::database::models::team_item::DBTeamMember;
use crate::database::models::{
Organization, generate_organization_id, team_item,
DBOrganization, generate_organization_id, team_item,
};
use crate::database::redis::RedisPool;
use crate::file_hosting::FileHost;
@@ -69,7 +69,7 @@ pub async fn organization_projects_get(
.map(|x| x.1)
.ok();
let organization_data = Organization::get(&id, &**pool, &redis).await?;
let organization_data = DBOrganization::get(&id, &**pool, &redis).await?;
if let Some(organization) = organization_data {
let project_ids = sqlx::query!(
"
@@ -84,7 +84,7 @@ pub async fn organization_projects_get(
.try_collect::<Vec<_>>()
.await?;
let projects_data = crate::database::models::Project::get_many_ids(
let projects_data = crate::database::models::DBProject::get_many_ids(
&project_ids,
&**pool,
&redis,
@@ -146,7 +146,7 @@ pub async fn organization_create(
organization_strings.push(name_organization_id.to_string());
}
organization_strings.push(new_organization.slug.clone());
let results = Organization::get_many(
let results = DBOrganization::get_many(
&organization_strings,
&mut *transaction,
&redis,
@@ -174,7 +174,7 @@ pub async fn organization_create(
let team_id = team.insert(&mut transaction).await?;
// Create organization
let organization = Organization {
let organization = DBOrganization {
id: organization_id,
slug: new_organization.slug.clone(),
name: new_organization.name.clone(),
@@ -188,10 +188,11 @@ pub async fn organization_create(
transaction.commit().await?;
// Only member is the owner, the logged in one
let member_data = TeamMember::get_from_team_full(team_id, &**pool, &redis)
.await?
.into_iter()
.next();
let member_data =
DBTeamMember::get_from_team_full(team_id, &**pool, &redis)
.await?
.into_iter()
.next();
let members_data = if let Some(member_data) = member_data {
vec![crate::models::teams::TeamMember::from_model(
member_data,
@@ -230,13 +231,13 @@ pub async fn organization_get(
.ok();
let user_id = current_user.as_ref().map(|x| x.id.into());
let organization_data = Organization::get(&id, &**pool, &redis).await?;
let organization_data = DBOrganization::get(&id, &**pool, &redis).await?;
if let Some(data) = organization_data {
let members_data =
TeamMember::get_from_team_full(data.team_id, &**pool, &redis)
DBTeamMember::get_from_team_full(data.team_id, &**pool, &redis)
.await?;
let users = crate::database::models::User::get_many_ids(
let users = crate::database::models::DBUser::get_many_ids(
&members_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -293,15 +294,16 @@ pub async fn organizations_get(
) -> Result<HttpResponse, ApiError> {
let ids = serde_json::from_str::<Vec<&str>>(&ids.ids)?;
let organizations_data =
Organization::get_many(&ids, &**pool, &redis).await?;
DBOrganization::get_many(&ids, &**pool, &redis).await?;
let team_ids = organizations_data
.iter()
.map(|x| x.team_id)
.collect::<Vec<_>>();
let teams_data =
TeamMember::get_from_team_full_many(&team_ids, &**pool, &redis).await?;
let users = crate::database::models::User::get_many_ids(
DBTeamMember::get_from_team_full_many(&team_ids, &**pool, &redis)
.await?;
let users = crate::database::models::DBUser::get_many_ids(
&teams_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -405,11 +407,11 @@ pub async fn organizations_edit(
let string = info.into_inner().0;
let result =
database::models::Organization::get(&string, &**pool, &redis).await?;
database::models::DBOrganization::get(&string, &**pool, &redis).await?;
if let Some(organization_item) = result {
let id = organization_item.id;
let team_member = database::models::TeamMember::get_from_user_id(
let team_member = database::models::DBTeamMember::get_from_user_id(
organization_item.team_id,
user.id.into(),
&**pool,
@@ -526,7 +528,7 @@ pub async fn organizations_edit(
}
transaction.commit().await?;
database::models::Organization::clear_cache(
database::models::DBOrganization::clear_cache(
organization_item.id,
Some(organization_item.slug),
&redis,
@@ -564,7 +566,7 @@ pub async fn organization_delete(
let string = info.into_inner().0;
let organization =
database::models::Organization::get(&string, &**pool, &redis)
database::models::DBOrganization::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -574,7 +576,7 @@ pub async fn organization_delete(
if !user.role.is_admin() {
let team_member =
database::models::TeamMember::get_from_user_id_organization(
database::models::DBTeamMember::get_from_user_id_organization(
organization.id,
user.id.into(),
false,
@@ -638,7 +640,7 @@ pub async fn organization_delete(
&mut transaction,
)
.await?;
let member = TeamMember {
let member = DBTeamMember {
id: new_id,
team_id: *organization_project_team,
user_id: owner_id,
@@ -653,7 +655,7 @@ pub async fn organization_delete(
member.insert(&mut transaction).await?;
}
// Safely remove the organization
let result = database::models::Organization::remove(
let result = database::models::DBOrganization::remove(
organization.id,
&mut transaction,
&redis,
@@ -662,7 +664,7 @@ pub async fn organization_delete(
transaction.commit().await?;
database::models::Organization::clear_cache(
database::models::DBOrganization::clear_cache(
organization.id,
Some(organization.slug),
&redis,
@@ -670,7 +672,7 @@ pub async fn organization_delete(
.await?;
for team_id in organization_project_teams {
database::models::TeamMember::clear_cache(team_id, &redis).await?;
database::models::DBTeamMember::clear_cache(team_id, &redis).await?;
}
if result.is_some() {
@@ -704,7 +706,7 @@ pub async fn organization_projects_add(
.1;
let organization =
database::models::Organization::get(&info, &**pool, &redis)
database::models::DBOrganization::get(&info, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -712,7 +714,7 @@ pub async fn organization_projects_add(
)
})?;
let project_item = database::models::Project::get(
let project_item = database::models::DBProject::get(
&project_info.project_id,
&**pool,
&redis,
@@ -731,7 +733,7 @@ pub async fn organization_projects_add(
}
let project_team_member =
database::models::TeamMember::get_from_user_id_project(
database::models::DBTeamMember::get_from_user_id_project(
project_item.inner.id,
current_user.id.into(),
false,
@@ -744,7 +746,7 @@ pub async fn organization_projects_add(
)
})?;
let organization_team_member =
database::models::TeamMember::get_from_user_id_organization(
database::models::DBTeamMember::get_from_user_id_organization(
organization.id,
current_user.id.into(),
false,
@@ -814,17 +816,17 @@ pub async fn organization_projects_add(
transaction.commit().await?;
database::models::User::clear_project_cache(
database::models::DBUser::clear_project_cache(
&[current_user.id.into()],
&redis,
)
.await?;
database::models::TeamMember::clear_cache(
database::models::DBTeamMember::clear_cache(
project_item.inner.team_id,
&redis,
)
.await?;
database::models::Project::clear_cache(
database::models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -866,17 +868,20 @@ pub async fn organization_projects_remove(
.await?
.1;
let organization =
database::models::Organization::get(&organization_id, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
"The specified organization does not exist!".to_string(),
)
})?;
let organization = database::models::DBOrganization::get(
&organization_id,
&**pool,
&redis,
)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
"The specified organization does not exist!".to_string(),
)
})?;
let project_item =
database::models::Project::get(&project_id, &**pool, &redis)
database::models::DBProject::get(&project_id, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -896,7 +901,7 @@ pub async fn organization_projects_remove(
}
let organization_team_member =
database::models::TeamMember::get_from_user_id_organization(
database::models::DBTeamMember::get_from_user_id_organization(
organization.id,
current_user.id.into(),
false,
@@ -916,7 +921,7 @@ pub async fn organization_projects_remove(
.unwrap_or_default();
if permissions.contains(OrganizationPermissions::REMOVE_PROJECT) {
// Now that permissions are confirmed, we confirm the veracity of the new user as an org member
database::models::TeamMember::get_from_user_id_organization(
database::models::DBTeamMember::get_from_user_id_organization(
organization.id,
data.new_owner.into(),
false,
@@ -932,13 +937,14 @@ pub async fn organization_projects_remove(
// Then, we get the team member of the project and that user (if it exists)
// We use the team member get directly
let new_owner = database::models::TeamMember::get_from_user_id_project(
project_item.inner.id,
data.new_owner.into(),
true,
&**pool,
)
.await?;
let new_owner =
database::models::DBTeamMember::get_from_user_id_project(
project_item.inner.id,
data.new_owner.into(),
true,
&**pool,
)
.await?;
let mut transaction = pool.begin().await?;
@@ -951,7 +957,7 @@ pub async fn organization_projects_remove(
&mut transaction,
)
.await?;
let member = TeamMember {
let member = DBTeamMember {
id: new_id,
team_id: project_item.inner.team_id,
user_id: data.new_owner.into(),
@@ -998,17 +1004,17 @@ pub async fn organization_projects_remove(
.await?;
transaction.commit().await?;
database::models::User::clear_project_cache(
database::models::DBUser::clear_project_cache(
&[current_user.id.into()],
&redis,
)
.await?;
database::models::TeamMember::clear_cache(
database::models::DBTeamMember::clear_cache(
project_item.inner.team_id,
&redis,
)
.await?;
database::models::Project::clear_cache(
database::models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -1052,7 +1058,7 @@ pub async fn organization_icon_edit(
let string = info.into_inner().0;
let organization_item =
database::models::Organization::get(&string, &**pool, &redis)
database::models::DBOrganization::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -1061,7 +1067,7 @@ pub async fn organization_icon_edit(
})?;
if !user.role.is_mod() {
let team_member = database::models::TeamMember::get_from_user_id(
let team_member = database::models::DBTeamMember::get_from_user_id(
organization_item.team_id,
user.id.into(),
&**pool,
@@ -1125,7 +1131,7 @@ pub async fn organization_icon_edit(
.await?;
transaction.commit().await?;
database::models::Organization::clear_cache(
database::models::DBOrganization::clear_cache(
organization_item.id,
Some(organization_item.slug),
&redis,
@@ -1155,7 +1161,7 @@ pub async fn delete_organization_icon(
let string = info.into_inner().0;
let organization_item =
database::models::Organization::get(&string, &**pool, &redis)
database::models::DBOrganization::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -1164,7 +1170,7 @@ pub async fn delete_organization_icon(
})?;
if !user.role.is_mod() {
let team_member = database::models::TeamMember::get_from_user_id(
let team_member = database::models::DBTeamMember::get_from_user_id(
organization_item.team_id,
user.id.into(),
&**pool,
@@ -1208,7 +1214,7 @@ pub async fn delete_organization_icon(
transaction.commit().await?;
database::models::Organization::clear_cache(
database::models::DBOrganization::clear_cache(
organization_item.id,
Some(organization_item.slug),
&redis,

View File

@@ -160,7 +160,7 @@ pub async fn paypal_webhook(
transaction.commit().await?;
crate::database::models::user_item::User::clear_caches(
crate::database::models::user_item::DBUser::clear_caches(
&[(
crate::database::models::DBUserId(result.user_id),
None,
@@ -270,7 +270,7 @@ pub async fn tremendous_webhook(
transaction.commit().await?;
crate::database::models::user_item::User::clear_caches(
crate::database::models::user_item::DBUser::clear_caches(
&[(
crate::database::models::DBUserId(result.user_id),
None,
@@ -319,12 +319,12 @@ pub async fn user_payouts(
.1;
let payout_ids =
crate::database::models::payout_item::Payout::get_all_for_user(
crate::database::models::payout_item::DBPayout::get_all_for_user(
user.id.into(),
&**pool,
)
.await?;
let payouts = crate::database::models::payout_item::Payout::get_many(
let payouts = crate::database::models::payout_item::DBPayout::get_many(
&payout_ids,
&**pool,
)
@@ -477,7 +477,7 @@ pub async fn create_payout(
}
let mut payout_item =
crate::database::models::payout_item::Payout {
crate::database::models::payout_item::DBPayout {
id: payout_id,
user_id: user.id,
created: Utc::now(),
@@ -550,7 +550,7 @@ pub async fn create_payout(
if let Some(email) = user.email {
if user.email_verified {
let mut payout_item =
crate::database::models::payout_item::Payout {
crate::database::models::payout_item::DBPayout {
id: payout_id,
user_id: user.id,
created: Utc::now(),
@@ -633,7 +633,7 @@ pub async fn create_payout(
payout_item.insert(&mut transaction).await?;
transaction.commit().await?;
crate::database::models::User::clear_caches(&[(user.id, None)], &redis)
crate::database::models::DBUser::clear_caches(&[(user.id, None)], &redis)
.await?;
Ok(HttpResponse::NoContent().finish())
@@ -660,7 +660,7 @@ pub async fn cancel_payout(
let id = info.into_inner().0;
let payout =
crate::database::models::payout_item::Payout::get(id.into(), &**pool)
crate::database::models::payout_item::DBPayout::get(id.into(), &**pool)
.await?;
if let Some(payout) = payout {

View File

@@ -4,7 +4,7 @@ use crate::database::models::loader_fields::{
Loader, LoaderField, LoaderFieldEnumValue,
};
use crate::database::models::thread_item::ThreadBuilder;
use crate::database::models::{self, User, image_item};
use crate::database::models::{self, DBUser, image_item};
use crate::database::redis::RedisPool;
use crate::file_hosting::{FileHost, FileHostingError};
use crate::models::error::ApiError;
@@ -646,7 +646,7 @@ async fn project_create_inner(
let mut members = vec![];
if let Some(organization_id) = project_create_data.organization_id {
let org = models::Organization::get_id(
let org = models::DBOrganization::get_id(
organization_id.into(),
pool,
redis,
@@ -658,7 +658,7 @@ async fn project_create_inner(
)
})?;
let team_member = models::TeamMember::get_from_user_id(
let team_member = models::DBTeamMember::get_from_user_id(
org.team_id,
current_user.id.into(),
pool,
@@ -773,7 +773,7 @@ async fn project_create_inner(
link_urls,
gallery_items: gallery_urls
.iter()
.map(|x| models::project_item::GalleryItem {
.map(|x| models::project_item::DBGalleryItem {
image_url: x.url.clone(),
raw_image_url: x.raw_url.clone(),
featured: x.featured,
@@ -791,10 +791,10 @@ async fn project_create_inner(
let now = Utc::now();
let id = project_builder_actual.insert(&mut *transaction).await?;
User::clear_project_cache(&[current_user.id.into()], redis).await?;
DBUser::clear_project_cache(&[current_user.id.into()], redis).await?;
for image_id in project_create_data.uploaded_images {
if let Some(db_image) = image_item::Image::get(
if let Some(db_image) = image_item::DBImage::get(
image_id.into(),
&mut **transaction,
redis,
@@ -822,7 +822,8 @@ async fn project_create_inner(
.execute(&mut **transaction)
.await?;
image_item::Image::clear_cache(image.id.into(), redis).await?;
image_item::DBImage::clear_cache(image.id.into(), redis)
.await?;
} else {
return Err(CreateError::InvalidInput(format!(
"Image {image_id} does not exist"

View File

@@ -4,9 +4,9 @@ use std::sync::Arc;
use crate::auth::checks::{filter_visible_versions, is_visible_project};
use crate::auth::{filter_visible_projects, get_user_from_headers};
use crate::database::models::notification_item::NotificationBuilder;
use crate::database::models::project_item::{GalleryItem, ModCategory};
use crate::database::models::project_item::{DBGalleryItem, DBModCategory};
use crate::database::models::thread_item::ThreadMessageBuilder;
use crate::database::models::{TeamMember, ids as db_ids, image_item};
use crate::database::models::{DBTeamMember, ids as db_ids, image_item};
use crate::database::redis::RedisPool;
use crate::database::{self, models as db_models};
use crate::file_hosting::FileHost;
@@ -109,7 +109,7 @@ pub async fn random_projects_get(
.await?;
let projects_data =
db_models::Project::get_many_ids(&project_ids, &**pool, &redis)
db_models::DBProject::get_many_ids(&project_ids, &**pool, &redis)
.await?
.into_iter()
.map(Project::from)
@@ -132,7 +132,7 @@ pub async fn projects_get(
) -> Result<HttpResponse, ApiError> {
let ids = serde_json::from_str::<Vec<&str>>(&ids.ids)?;
let projects_data =
db_models::Project::get_many(&ids, &**pool, &redis).await?;
db_models::DBProject::get_many(&ids, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
@@ -162,7 +162,7 @@ pub async fn project_get(
let string = info.into_inner().0;
let project_data =
db_models::Project::get(&string, &**pool, &redis).await?;
db_models::DBProject::get(&string, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
&**pool,
@@ -268,12 +268,12 @@ pub async fn project_edit(
})?;
let string = info.into_inner().0;
let result = db_models::Project::get(&string, &**pool, &redis).await?;
let result = db_models::DBProject::get(&string, &**pool, &redis).await?;
if let Some(project_item) = result {
let id = project_item.inner.id;
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project_item.inner,
user.id.into(),
&**pool,
@@ -884,7 +884,7 @@ pub async fn project_edit(
.await?;
transaction.commit().await?;
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -929,7 +929,7 @@ pub async fn edit_project_categories(
let mcategories = category_ids
.values()
.map(|&category_id| ModCategory {
.map(|&category_id| DBModCategory {
project_id,
category_id,
is_additional,
@@ -937,7 +937,7 @@ pub async fn edit_project_categories(
.collect::<Vec<_>>();
mod_categories.extend(mcategories);
}
ModCategory::insert_many(mod_categories, &mut *transaction).await?;
DBModCategory::insert_many(mod_categories, &mut *transaction).await?;
Ok(())
}
@@ -980,7 +980,8 @@ pub async fn project_get_check(
) -> Result<HttpResponse, ApiError> {
let slug = info.into_inner().0;
let project_data = db_models::Project::get(&slug, &**pool, &redis).await?;
let project_data =
db_models::DBProject::get(&slug, &**pool, &redis).await?;
if let Some(project) = project_data {
Ok(HttpResponse::Ok().json(json! ({
@@ -1006,7 +1007,7 @@ pub async fn dependency_list(
) -> Result<HttpResponse, ApiError> {
let string = info.into_inner().0;
let result = db_models::Project::get(&string, &**pool, &redis).await?;
let result = db_models::DBProject::get(&string, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
@@ -1026,7 +1027,7 @@ pub async fn dependency_list(
return Err(ApiError::NotFound);
}
let dependencies = database::Project::get_dependencies(
let dependencies = database::DBProject::get_dependencies(
project.inner.id,
&**pool,
&redis,
@@ -1054,8 +1055,8 @@ pub async fn dependency_list(
.unique()
.collect::<Vec<db_models::DBVersionId>>();
let (projects_result, versions_result) = futures::future::try_join(
database::Project::get_many_ids(&project_ids, &**pool, &redis),
database::Version::get_many(&dep_version_ids, &**pool, &redis),
database::DBProject::get_many_ids(&project_ids, &**pool, &redis),
database::DBVersion::get_many(&dep_version_ids, &**pool, &redis),
)
.await?;
@@ -1141,7 +1142,8 @@ pub async fn projects_edit(
.collect();
let projects_data =
db_models::Project::get_many_ids(&project_ids, &**pool, &redis).await?;
db_models::DBProject::get_many_ids(&project_ids, &**pool, &redis)
.await?;
if let Some(id) = project_ids
.iter()
@@ -1157,7 +1159,7 @@ pub async fn projects_edit(
.iter()
.map(|x| x.inner.team_id)
.collect::<Vec<db_models::DBTeamId>>();
let team_members = db_models::TeamMember::get_from_team_full_many(
let team_members = db_models::DBTeamMember::get_from_team_full_many(
&team_ids, &**pool, &redis,
)
.await?;
@@ -1166,7 +1168,7 @@ pub async fn projects_edit(
.iter()
.filter_map(|x| x.inner.organization_id)
.collect::<Vec<db_models::DBOrganizationId>>();
let organizations = db_models::Organization::get_many_ids(
let organizations = db_models::DBOrganization::get_many_ids(
&organization_ids,
&**pool,
&redis,
@@ -1178,7 +1180,7 @@ pub async fn projects_edit(
.map(|x| x.team_id)
.collect::<Vec<db_models::DBTeamId>>();
let organization_team_members =
db_models::TeamMember::get_from_team_full_many(
db_models::DBTeamMember::get_from_team_full_many(
&organization_team_ids,
&**pool,
&redis,
@@ -1315,7 +1317,7 @@ pub async fn projects_edit(
}
}
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project.inner.id,
project.inner.slug,
None,
@@ -1388,13 +1390,13 @@ pub async fn bulk_edit_project_categories(
))
})?
.id;
mod_categories.push(ModCategory {
mod_categories.push(DBModCategory {
project_id,
category_id,
is_additional,
});
}
ModCategory::insert_many(mod_categories, &mut *transaction).await?;
DBModCategory::insert_many(mod_categories, &mut *transaction).await?;
}
Ok(())
@@ -1427,7 +1429,7 @@ pub async fn project_icon_edit(
.1;
let string = info.into_inner().0;
let project_item = db_models::Project::get(&string, &**pool, &redis)
let project_item = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -1437,7 +1439,7 @@ pub async fn project_icon_edit(
if !user.role.is_mod() {
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project_item.inner,
user.id.into(),
&**pool,
@@ -1508,7 +1510,7 @@ pub async fn project_icon_edit(
.await?;
transaction.commit().await?;
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -1538,7 +1540,7 @@ pub async fn delete_project_icon(
.1;
let string = info.into_inner().0;
let project_item = db_models::Project::get(&string, &**pool, &redis)
let project_item = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -1548,7 +1550,7 @@ pub async fn delete_project_icon(
if !user.role.is_mod() {
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project_item.inner,
user.id.into(),
&**pool,
@@ -1597,7 +1599,7 @@ pub async fn delete_project_icon(
.await?;
transaction.commit().await?;
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -1645,7 +1647,7 @@ pub async fn add_gallery_item(
.1;
let string = info.into_inner().0;
let project_item = db_models::Project::get(&string, &**pool, &redis)
let project_item = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -1662,7 +1664,7 @@ pub async fn add_gallery_item(
if !user.role.is_admin() {
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project_item.inner,
user.id.into(),
&**pool,
@@ -1735,7 +1737,7 @@ pub async fn add_gallery_item(
.await?;
}
let gallery_item = vec![db_models::project_item::GalleryItem {
let gallery_item = vec![db_models::project_item::DBGalleryItem {
image_url: upload_result.url,
raw_image_url: upload_result.raw_url,
featured: item.featured,
@@ -1744,7 +1746,7 @@ pub async fn add_gallery_item(
created: Utc::now(),
ordering: item.ordering.unwrap_or(0),
}];
GalleryItem::insert_many(
DBGalleryItem::insert_many(
gallery_item,
project_item.inner.id,
&mut transaction,
@@ -1752,7 +1754,7 @@ pub async fn add_gallery_item(
.await?;
transaction.commit().await?;
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -1822,7 +1824,7 @@ pub async fn edit_gallery_item(
))
})?;
let project_item = db_models::Project::get_id(
let project_item = db_models::DBProject::get_id(
database::models::DBProjectId(result.mod_id),
&**pool,
&redis,
@@ -1836,7 +1838,7 @@ pub async fn edit_gallery_item(
if !user.role.is_mod() {
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project_item.inner,
user.id.into(),
&**pool,
@@ -1935,7 +1937,7 @@ pub async fn edit_gallery_item(
transaction.commit().await?;
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -1985,7 +1987,7 @@ pub async fn delete_gallery_item(
))
})?;
let project_item = db_models::Project::get_id(
let project_item = db_models::DBProject::get_id(
database::models::DBProjectId(item.mod_id),
&**pool,
&redis,
@@ -1999,7 +2001,7 @@ pub async fn delete_gallery_item(
if !user.role.is_mod() {
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project_item.inner,
user.id.into(),
&**pool,
@@ -2049,7 +2051,7 @@ pub async fn delete_gallery_item(
transaction.commit().await?;
db_models::Project::clear_cache(
db_models::DBProject::clear_cache(
project_item.inner.id,
project_item.inner.slug,
None,
@@ -2079,7 +2081,7 @@ pub async fn project_delete(
.1;
let string = info.into_inner().0;
let project = db_models::Project::get(&string, &**pool, &redis)
let project = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -2089,7 +2091,7 @@ pub async fn project_delete(
if !user.role.is_admin() {
let (team_member, organization_team_member) =
db_models::TeamMember::get_for_project_permissions(
db_models::DBTeamMember::get_for_project_permissions(
&project.inner,
user.id.into(),
&**pool,
@@ -2122,9 +2124,10 @@ pub async fn project_delete(
project_id: Some(project.inner.id.into()),
};
let uploaded_images =
db_models::Image::get_many_contexted(context, &mut transaction).await?;
db_models::DBImage::get_many_contexted(context, &mut transaction)
.await?;
for image in uploaded_images {
image_item::Image::remove(image.id, &mut transaction, &redis).await?;
image_item::DBImage::remove(image.id, &mut transaction, &redis).await?;
}
sqlx::query!(
@@ -2137,9 +2140,12 @@ pub async fn project_delete(
.execute(&mut *transaction)
.await?;
let result =
db_models::Project::remove(project.inner.id, &mut transaction, &redis)
.await?;
let result = db_models::DBProject::remove(
project.inner.id,
&mut transaction,
&redis,
)
.await?;
transaction.commit().await?;
@@ -2178,7 +2184,7 @@ pub async fn project_follow(
.1;
let string = info.into_inner().0;
let result = db_models::Project::get(&string, &**pool, &redis)
let result = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -2258,7 +2264,7 @@ pub async fn project_unfollow(
.1;
let string = info.into_inner().0;
let result = db_models::Project::get(&string, &**pool, &redis)
let result = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -2336,7 +2342,7 @@ pub async fn project_get_organization(
let user_id = current_user.as_ref().map(|x| x.id.into());
let string = info.into_inner().0;
let result = db_models::Project::get(&string, &**pool, &redis)
let result = db_models::DBProject::get(&string, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -2350,7 +2356,7 @@ pub async fn project_get_organization(
))
} else if let Some(organization_id) = result.inner.organization_id {
let organization =
db_models::Organization::get_id(organization_id, &**pool, &redis)
db_models::DBOrganization::get_id(organization_id, &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -2358,14 +2364,14 @@ pub async fn project_get_organization(
)
})?;
let members_data = TeamMember::get_from_team_full(
let members_data = DBTeamMember::get_from_team_full(
organization.team_id,
&**pool,
&redis,
)
.await?;
let users = crate::database::models::User::get_many_ids(
let users = crate::database::models::DBUser::get_many_ids(
&members_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,

View File

@@ -87,7 +87,7 @@ pub async fn report_create(
))
})?;
let mut report = crate::database::models::report_item::Report {
let mut report = crate::database::models::report_item::DBReport {
id,
report_type_id: report_type,
project_id: None,
@@ -171,7 +171,7 @@ pub async fn report_create(
for image_id in new_report.uploaded_images {
if let Some(db_image) =
image_item::Image::get(image_id.into(), &mut *transaction, &redis)
image_item::DBImage::get(image_id.into(), &mut *transaction, &redis)
.await?
{
let image: Image = db_image.into();
@@ -195,7 +195,7 @@ pub async fn report_create(
.execute(&mut *transaction)
.await?;
image_item::Image::clear_cache(image.id.into(), &redis).await?;
image_item::DBImage::clear_cache(image.id.into(), &redis).await?;
} else {
return Err(ApiError::InvalidInput(format!(
"Image {image_id} could not be found"
@@ -292,11 +292,12 @@ pub async fn reports(
.await?
};
let query_reports = crate::database::models::report_item::Report::get_many(
&report_ids,
&**pool,
)
.await?;
let query_reports =
crate::database::models::report_item::DBReport::get_many(
&report_ids,
&**pool,
)
.await?;
let mut reports: Vec<Report> = Vec::new();
@@ -325,11 +326,12 @@ pub async fn reports_get(
.map(|x| x.into())
.collect();
let reports_data = crate::database::models::report_item::Report::get_many(
&report_ids,
&**pool,
)
.await?;
let reports_data =
crate::database::models::report_item::DBReport::get_many(
&report_ids,
&**pool,
)
.await?;
let user = get_user_from_headers(
&req,
@@ -369,7 +371,8 @@ pub async fn report_get(
let id = info.into_inner().0.into();
let report =
crate::database::models::report_item::Report::get(id, &**pool).await?;
crate::database::models::report_item::DBReport::get(id, &**pool)
.await?;
if let Some(report) = report {
if !user.role.is_mod() && report.reporter != user.id.into() {
@@ -410,7 +413,8 @@ pub async fn report_edit(
let id = info.into_inner().0.into();
let report =
crate::database::models::report_item::Report::get(id, &**pool).await?;
crate::database::models::report_item::DBReport::get(id, &**pool)
.await?;
if let Some(report) = report {
if !user.role.is_mod() && report.reporter != user.id.into() {
@@ -512,14 +516,16 @@ pub async fn report_delete(
let context = ImageContext::Report {
report_id: Some(id),
};
let uploaded_images =
database::models::Image::get_many_contexted(context, &mut transaction)
.await?;
let uploaded_images = database::models::DBImage::get_many_contexted(
context,
&mut transaction,
)
.await?;
for image in uploaded_images {
image_item::Image::remove(image.id, &mut transaction, &redis).await?;
image_item::DBImage::remove(image.id, &mut transaction, &redis).await?;
}
let result = crate::database::models::report_item::Report::remove_full(
let result = crate::database::models::report_item::DBReport::remove_full(
id.into(),
&mut transaction,
)

View File

@@ -1,9 +1,9 @@
use crate::auth::checks::is_visible_project;
use crate::auth::get_user_from_headers;
use crate::database::Project;
use crate::database::DBProject;
use crate::database::models::notification_item::NotificationBuilder;
use crate::database::models::team_item::TeamAssociationId;
use crate::database::models::{Organization, Team, TeamMember, User};
use crate::database::models::{DBOrganization, DBTeam, DBTeamMember, DBUser};
use crate::database::redis::RedisPool;
use crate::models::ids::TeamId;
use crate::models::notifications::NotificationBody;
@@ -48,7 +48,8 @@ pub async fn team_members_get_project(
) -> Result<HttpResponse, ApiError> {
let string = info.into_inner().0;
let project_data =
crate::database::models::Project::get(&string, &**pool, &redis).await?;
crate::database::models::DBProject::get(&string, &**pool, &redis)
.await?;
if let Some(project) = project_data {
let current_user = get_user_from_headers(
@@ -67,13 +68,13 @@ pub async fn team_members_get_project(
{
return Err(ApiError::NotFound);
}
let members_data = TeamMember::get_from_team_full(
let members_data = DBTeamMember::get_from_team_full(
project.inner.team_id,
&**pool,
&redis,
)
.await?;
let users = User::get_many_ids(
let users = DBUser::get_many_ids(
&members_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -83,7 +84,7 @@ pub async fn team_members_get_project(
let user_id = current_user.as_ref().map(|x| x.id.into());
let logged_in = if let Some(user_id) = user_id {
let (team_member, organization_team_member) =
TeamMember::get_for_project_permissions(
DBTeamMember::get_for_project_permissions(
&project.inner,
user_id,
&**pool,
@@ -132,7 +133,7 @@ pub async fn team_members_get_organization(
) -> Result<HttpResponse, ApiError> {
let string = info.into_inner().0;
let organization_data =
crate::database::models::Organization::get(&string, &**pool, &redis)
crate::database::models::DBOrganization::get(&string, &**pool, &redis)
.await?;
if let Some(organization) = organization_data {
@@ -147,13 +148,13 @@ pub async fn team_members_get_organization(
.map(|x| x.1)
.ok();
let members_data = TeamMember::get_from_team_full(
let members_data = DBTeamMember::get_from_team_full(
organization.team_id,
&**pool,
&redis,
)
.await?;
let users = crate::database::models::User::get_many_ids(
let users = crate::database::models::DBUser::get_many_ids(
&members_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -208,8 +209,8 @@ pub async fn team_members_get(
) -> Result<HttpResponse, ApiError> {
let id = info.into_inner().0;
let members_data =
TeamMember::get_from_team_full(id.into(), &**pool, &redis).await?;
let users = crate::database::models::User::get_many_ids(
DBTeamMember::get_from_team_full(id.into(), &**pool, &redis).await?;
let users = crate::database::models::DBUser::get_many_ids(
&members_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -279,8 +280,9 @@ pub async fn teams_get(
.collect::<Vec<crate::database::models::ids::DBTeamId>>();
let teams_data =
TeamMember::get_from_team_full_many(&team_ids, &**pool, &redis).await?;
let users = crate::database::models::User::get_many_ids(
DBTeamMember::get_from_team_full_many(&team_ids, &**pool, &redis)
.await?;
let users = crate::database::models::DBUser::get_many_ids(
&teams_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -351,7 +353,7 @@ pub async fn join_team(
.await?
.1;
let member = TeamMember::get_from_user_id_pending(
let member = DBTeamMember::get_from_user_id_pending(
team_id,
current_user.id.into(),
&**pool,
@@ -367,7 +369,7 @@ pub async fn join_team(
let mut transaction = pool.begin().await?;
// Edit Team Member to set Accepted to True
TeamMember::edit_team_member(
DBTeamMember::edit_team_member(
team_id,
current_user.id.into(),
None,
@@ -383,8 +385,8 @@ pub async fn join_team(
transaction.commit().await?;
User::clear_project_cache(&[current_user.id.into()], &redis).await?;
TeamMember::clear_cache(team_id, &redis).await?;
DBUser::clear_project_cache(&[current_user.id.into()], &redis).await?;
DBTeamMember::clear_cache(team_id, &redis).await?;
} else {
return Err(ApiError::InvalidInput(
"There is no pending request from this team".to_string(),
@@ -439,27 +441,30 @@ pub async fn add_team_member(
)
.await?
.1;
let team_association = Team::get_association(team_id, &**pool)
let team_association = DBTeam::get_association(team_id, &**pool)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
"The team specified does not exist".to_string(),
)
})?;
let member =
TeamMember::get_from_user_id(team_id, current_user.id.into(), &**pool)
.await?;
let member = DBTeamMember::get_from_user_id(
team_id,
current_user.id.into(),
&**pool,
)
.await?;
match team_association {
// If team is associated with a project, check if they have permissions to invite users to that project
TeamAssociationId::Project(pid) => {
let organization =
Organization::get_associated_organization_project_id(
DBOrganization::get_associated_organization_project_id(
pid, &**pool,
)
.await?;
let organization_team_member =
if let Some(organization) = &organization {
TeamMember::get_from_user_id(
DBTeamMember::get_from_user_id(
organization.team_id,
current_user.id.into(),
&**pool,
@@ -537,7 +542,7 @@ pub async fn add_team_member(
));
}
let request = TeamMember::get_from_user_id_pending(
let request = DBTeamMember::get_from_user_id_pending(
team_id,
new_member.user_id.into(),
&**pool,
@@ -556,7 +561,7 @@ pub async fn add_team_member(
));
}
}
let new_user = crate::database::models::User::get_id(
let new_user = crate::database::models::DBUser::get_id(
new_member.user_id.into(),
&**pool,
&redis,
@@ -570,11 +575,13 @@ pub async fn add_team_member(
if let TeamAssociationId::Project(pid) = team_association {
// We cannot add the owner to a project team in their own org
let organization =
Organization::get_associated_organization_project_id(pid, &**pool)
.await?;
DBOrganization::get_associated_organization_project_id(
pid, &**pool,
)
.await?;
let new_user_organization_team_member =
if let Some(organization) = &organization {
TeamMember::get_from_user_id(
DBTeamMember::get_from_user_id(
organization.team_id,
new_user.id,
&**pool,
@@ -607,7 +614,7 @@ pub async fn add_team_member(
let new_id =
crate::database::models::ids::generate_team_member_id(&mut transaction)
.await?;
TeamMember {
DBTeamMember {
id: new_id,
team_id,
user_id: new_member.user_id.into(),
@@ -653,8 +660,8 @@ pub async fn add_team_member(
}
transaction.commit().await?;
TeamMember::clear_cache(team_id, &redis).await?;
User::clear_project_cache(&[new_member.user_id.into()], &redis).await?;
DBTeamMember::clear_cache(team_id, &redis).await?;
DBUser::clear_project_cache(&[new_member.user_id.into()], &redis).await?;
Ok(HttpResponse::NoContent().body(""))
}
@@ -691,16 +698,16 @@ pub async fn edit_team_member(
.1;
let team_association =
Team::get_association(id, &**pool).await?.ok_or_else(|| {
DBTeam::get_association(id, &**pool).await?.ok_or_else(|| {
ApiError::InvalidInput(
"The team specified does not exist".to_string(),
)
})?;
let member =
TeamMember::get_from_user_id(id, current_user.id.into(), &**pool)
DBTeamMember::get_from_user_id(id, current_user.id.into(), &**pool)
.await?;
let edit_member_db =
TeamMember::get_from_user_id_pending(id, user_id, &**pool)
DBTeamMember::get_from_user_id_pending(id, user_id, &**pool)
.await?
.ok_or_else(|| {
ApiError::CustomAuthentication(
@@ -723,13 +730,13 @@ pub async fn edit_team_member(
match team_association {
TeamAssociationId::Project(project_id) => {
let organization =
Organization::get_associated_organization_project_id(
DBOrganization::get_associated_organization_project_id(
project_id, &**pool,
)
.await?;
let organization_team_member =
if let Some(organization) = &organization {
TeamMember::get_from_user_id(
DBTeamMember::get_from_user_id(
organization.team_id,
current_user.id.into(),
&**pool,
@@ -831,7 +838,7 @@ pub async fn edit_team_member(
}
}
TeamMember::edit_team_member(
DBTeamMember::edit_team_member(
id,
user_id,
edit_member.permissions,
@@ -846,7 +853,7 @@ pub async fn edit_team_member(
.await?;
transaction.commit().await?;
TeamMember::clear_cache(id, &redis).await?;
DBTeamMember::clear_cache(id, &redis).await?;
Ok(HttpResponse::NoContent().body(""))
}
@@ -879,9 +886,10 @@ pub async fn transfer_ownership(
// Forbid transferring ownership of a project team that is owned by an organization
// These are owned by the organization owner, and must be removed from the organization first
// There shouldnt be an ownr on these projects in these cases, but just in case.
let team_association_id = Team::get_association(id.into(), &**pool).await?;
let team_association_id =
DBTeam::get_association(id.into(), &**pool).await?;
if let Some(TeamAssociationId::Project(pid)) = team_association_id {
let result = Project::get_id(pid, &**pool, &redis).await?;
let result = DBProject::get_id(pid, &**pool, &redis).await?;
if let Some(project_item) = result {
if project_item.inner.organization_id.is_some() {
return Err(ApiError::InvalidInput(
@@ -893,7 +901,7 @@ pub async fn transfer_ownership(
}
if !current_user.role.is_admin() {
let member = TeamMember::get_from_user_id(
let member = DBTeamMember::get_from_user_id(
id.into(),
current_user.id.into(),
&**pool,
@@ -914,7 +922,7 @@ pub async fn transfer_ownership(
}
}
let new_member = TeamMember::get_from_user_id(
let new_member = DBTeamMember::get_from_user_id(
id.into(),
new_owner.user_id.into(),
&**pool,
@@ -936,12 +944,12 @@ pub async fn transfer_ownership(
// The following are the only places new_is_owner is modified.
if let Some(former_owner) =
TeamMember::get_from_team_full(id.into(), &**pool, &redis)
DBTeamMember::get_from_team_full(id.into(), &**pool, &redis)
.await?
.into_iter()
.find(|x| x.is_owner)
{
TeamMember::edit_team_member(
DBTeamMember::edit_team_member(
id.into(),
former_owner.user_id,
None,
@@ -956,7 +964,7 @@ pub async fn transfer_ownership(
.await?;
}
TeamMember::edit_team_member(
DBTeamMember::edit_team_member(
id.into(),
new_owner.user_id.into(),
Some(ProjectPermissions::all()),
@@ -1004,7 +1012,7 @@ pub async fn transfer_ownership(
// If the owner of the organization is a member of the project, remove them
for team_id in team_ids.iter() {
TeamMember::delete(
DBTeamMember::delete(
*team_id,
new_owner.user_id.into(),
&mut transaction,
@@ -1018,9 +1026,9 @@ pub async fn transfer_ownership(
};
transaction.commit().await?;
TeamMember::clear_cache(id.into(), &redis).await?;
DBTeamMember::clear_cache(id.into(), &redis).await?;
for team_id in project_teams_edited {
TeamMember::clear_cache(team_id, &redis).await?;
DBTeamMember::clear_cache(team_id, &redis).await?;
}
Ok(HttpResponse::NoContent().body(""))
@@ -1048,17 +1056,17 @@ pub async fn remove_team_member(
.1;
let team_association =
Team::get_association(id, &**pool).await?.ok_or_else(|| {
DBTeam::get_association(id, &**pool).await?.ok_or_else(|| {
ApiError::InvalidInput(
"The team specified does not exist".to_string(),
)
})?;
let member =
TeamMember::get_from_user_id(id, current_user.id.into(), &**pool)
DBTeamMember::get_from_user_id(id, current_user.id.into(), &**pool)
.await?;
let delete_member =
TeamMember::get_from_user_id_pending(id, user_id, &**pool).await?;
DBTeamMember::get_from_user_id_pending(id, user_id, &**pool).await?;
if let Some(delete_member) = delete_member {
if delete_member.is_owner {
@@ -1074,13 +1082,13 @@ pub async fn remove_team_member(
match team_association {
TeamAssociationId::Project(pid) => {
let organization =
Organization::get_associated_organization_project_id(
DBOrganization::get_associated_organization_project_id(
pid, &**pool,
)
.await?;
let organization_team_member =
if let Some(organization) = &organization {
TeamMember::get_from_user_id(
DBTeamMember::get_from_user_id(
organization.team_id,
current_user.id.into(),
&**pool,
@@ -1105,7 +1113,7 @@ pub async fn remove_team_member(
.contains(ProjectPermissions::REMOVE_MEMBER)
// true as if the permission exists, but the member does not, they are part of an org
{
TeamMember::delete(id, user_id, &mut transaction)
DBTeamMember::delete(id, user_id, &mut transaction)
.await?;
} else {
return Err(ApiError::CustomAuthentication(
@@ -1121,7 +1129,7 @@ pub async fn remove_team_member(
// This is a pending invite rather than a member, so the
// user being invited or team members with the MANAGE_INVITES
// permission can remove it.
TeamMember::delete(id, user_id, &mut transaction).await?;
DBTeamMember::delete(id, user_id, &mut transaction).await?;
} else {
return Err(ApiError::CustomAuthentication(
"You do not have permission to cancel a team invite"
@@ -1144,7 +1152,7 @@ pub async fn remove_team_member(
|| organization_permissions
.contains(OrganizationPermissions::REMOVE_MEMBER)
{
TeamMember::delete(id, user_id, &mut transaction)
DBTeamMember::delete(id, user_id, &mut transaction)
.await?;
} else {
return Err(ApiError::CustomAuthentication(
@@ -1160,7 +1168,7 @@ pub async fn remove_team_member(
// This is a pending invite rather than a member, so the
// user being invited or team members with the MANAGE_INVITES
// permission can remove it.
TeamMember::delete(id, user_id, &mut transaction).await?;
DBTeamMember::delete(id, user_id, &mut transaction).await?;
} else {
return Err(ApiError::CustomAuthentication(
"You do not have permission to cancel an organization invite".to_string(),
@@ -1171,8 +1179,8 @@ pub async fn remove_team_member(
transaction.commit().await?;
TeamMember::clear_cache(id, &redis).await?;
User::clear_project_cache(&[delete_member.user_id], &redis).await?;
DBTeamMember::clear_cache(id, &redis).await?;
DBUser::clear_project_cache(&[delete_member.user_id], &redis).await?;
Ok(HttpResponse::NoContent().body(""))
} else {

View File

@@ -34,7 +34,7 @@ pub fn config(cfg: &mut web::ServiceConfig) {
}
pub async fn is_authorized_thread(
thread: &database::models::Thread,
thread: &database::models::DBThread,
user: &User,
pool: &PgPool,
) -> Result<bool, ApiError> {
@@ -94,7 +94,7 @@ pub async fn is_authorized_thread(
}
pub async fn filter_authorized_threads(
threads: Vec<database::models::Thread>,
threads: Vec<database::models::DBThread>,
user: &User,
pool: &web::Data<PgPool>,
redis: &RedisPool,
@@ -230,7 +230,7 @@ pub async fn filter_authorized_threads(
);
let users: Vec<User> =
database::models::User::get_many_ids(&user_ids, &***pool, redis)
database::models::DBUser::get_many_ids(&user_ids, &***pool, redis)
.await?
.into_iter()
.map(From::from)
@@ -278,7 +278,7 @@ pub async fn thread_get(
) -> Result<HttpResponse, ApiError> {
let string = info.into_inner().0.into();
let thread_data = database::models::Thread::get(string, &**pool).await?;
let thread_data = database::models::DBThread::get(string, &**pool).await?;
let user = get_user_from_headers(
&req,
@@ -308,12 +308,13 @@ pub async fn thread_get(
.collect::<Vec<_>>(),
);
let users: Vec<User> =
database::models::User::get_many_ids(authors, &**pool, &redis)
.await?
.into_iter()
.map(From::from)
.collect();
let users: Vec<User> = database::models::DBUser::get_many_ids(
authors, &**pool, &redis,
)
.await?
.into_iter()
.map(From::from)
.collect();
return Ok(
HttpResponse::Ok().json(Thread::from(data, users, &user))
@@ -352,7 +353,7 @@ pub async fn threads_get(
.collect();
let threads_data =
database::models::Thread::get_many(&thread_ids, &**pool).await?;
database::models::DBThread::get_many(&thread_ids, &**pool).await?;
let threads =
filter_authorized_threads(threads_data, &user, &pool, &redis).await?;
@@ -405,7 +406,7 @@ pub async fn thread_send_message(
}
if let Some(replying_to) = replying_to {
let thread_message = database::models::ThreadMessage::get(
let thread_message = database::models::DBThreadMessage::get(
(*replying_to).into(),
&**pool,
)
@@ -430,7 +431,7 @@ pub async fn thread_send_message(
));
}
let result = database::models::Thread::get(string, &**pool).await?;
let result = database::models::DBThread::get(string, &**pool).await?;
if let Some(thread) = result {
if !is_authorized_thread(&thread, &user, &pool).await? {
@@ -449,16 +450,17 @@ pub async fn thread_send_message(
.await?;
if let Some(project_id) = thread.project_id {
let project =
database::models::Project::get_id(project_id, &**pool, &redis)
.await?;
let project = database::models::DBProject::get_id(
project_id, &**pool, &redis,
)
.await?;
if let Some(project) = project {
if project.inner.status != ProjectStatus::Processing
&& user.role.is_mod()
{
let members =
database::models::TeamMember::get_from_team_full(
database::models::DBTeamMember::get_from_team_full(
project.inner.team_id,
&**pool,
&redis,
@@ -482,9 +484,10 @@ pub async fn thread_send_message(
}
}
} else if let Some(report_id) = thread.report_id {
let report =
database::models::report_item::Report::get(report_id, &**pool)
.await?;
let report = database::models::report_item::DBReport::get(
report_id, &**pool,
)
.await?;
if let Some(report) = report {
if report.closed && !user.role.is_mod() {
@@ -513,7 +516,7 @@ pub async fn thread_send_message(
} = &new_message.body
{
for image_id in associated_images {
if let Some(db_image) = image_item::Image::get(
if let Some(db_image) = image_item::DBImage::get(
(*image_id).into(),
&mut *transaction,
&redis,
@@ -543,7 +546,7 @@ pub async fn thread_send_message(
.execute(&mut *transaction)
.await?;
image_item::Image::clear_cache(image.id.into(), &redis)
image_item::DBImage::clear_cache(image.id.into(), &redis)
.await?;
} else {
return Err(ApiError::InvalidInput(format!(
@@ -579,7 +582,7 @@ pub async fn message_delete(
.await?
.1;
let result = database::models::ThreadMessage::get(
let result = database::models::DBThreadMessage::get(
info.into_inner().0.into(),
&**pool,
)
@@ -598,7 +601,7 @@ pub async fn message_delete(
thread_message_id: Some(thread.id.into()),
};
let images =
database::Image::get_many_contexted(context, &mut transaction)
database::DBImage::get_many_contexted(context, &mut transaction)
.await?;
let cdn_url = dotenvy::var("CDN_URL")?;
for image in images {
@@ -606,7 +609,8 @@ pub async fn message_delete(
if let Some(icon_path) = name {
file_host.delete_file_version("", icon_path).await?;
}
database::Image::remove(image.id, &mut transaction, &redis).await?;
database::DBImage::remove(image.id, &mut transaction, &redis)
.await?;
}
let private = if let MessageBody::Text { private, .. } = thread.body {
@@ -617,7 +621,7 @@ pub async fn message_delete(
false
};
database::models::ThreadMessage::remove_full(
database::models::DBThreadMessage::remove_full(
thread.id,
private,
&mut transaction,

View File

@@ -4,7 +4,7 @@ use super::{ApiError, oauth_clients::get_user_clients};
use crate::util::img::delete_old_images;
use crate::{
auth::{filter_visible_projects, get_user_from_headers},
database::{models::User, redis::RedisPool},
database::{models::DBUser, redis::RedisPool},
file_hosting::FileHost,
models::{
collections::{Collection, CollectionStatus},
@@ -88,7 +88,7 @@ pub async fn admin_user_email(
)
})?;
let user = User::get_id(
let user = DBUser::get_id(
crate::database::models::DBUserId(user_id),
&**pool,
&redis,
@@ -120,12 +120,12 @@ pub async fn projects_list(
.map(|x| x.1)
.ok();
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(id) = id_option.map(|x| x.id) {
let project_data = User::get_projects(id, &**pool, &redis).await?;
let project_data = DBUser::get_projects(id, &**pool, &redis).await?;
let projects: Vec<_> = crate::database::Project::get_many_ids(
let projects: Vec<_> = crate::database::DBProject::get_many_ids(
&project_data,
&**pool,
&redis,
@@ -177,7 +177,7 @@ pub async fn users_get(
) -> Result<HttpResponse, ApiError> {
let user_ids = serde_json::from_str::<Vec<String>>(&ids.ids)?;
let users_data = User::get_many(&user_ids, &**pool, &redis).await?;
let users_data = DBUser::get_many(&user_ids, &**pool, &redis).await?;
let users: Vec<crate::models::users::User> =
users_data.into_iter().map(From::from).collect();
@@ -192,7 +192,7 @@ pub async fn user_get(
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
let user_data = User::get(&info.into_inner().0, &**pool, &redis).await?;
let user_data = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(data) = user_data {
let auth_user = get_user_from_headers(
@@ -237,7 +237,7 @@ pub async fn collections_list(
.map(|x| x.1)
.ok();
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(id) = id_option.map(|x| x.id) {
let user_id: UserId = id.into();
@@ -246,9 +246,9 @@ pub async fn collections_list(
.map(|y| y.role.is_mod() || y.id == user_id)
.unwrap_or(false);
let project_data = User::get_collections(id, &**pool).await?;
let project_data = DBUser::get_collections(id, &**pool).await?;
let response: Vec<_> = crate::database::models::Collection::get_many(
let response: Vec<_> = crate::database::models::DBCollection::get_many(
&project_data,
&**pool,
&redis,
@@ -285,13 +285,13 @@ pub async fn orgs_list(
.map(|x| x.1)
.ok();
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(id) = id_option.map(|x| x.id) {
let org_data = User::get_organizations(id, &**pool).await?;
let org_data = DBUser::get_organizations(id, &**pool).await?;
let organizations_data =
crate::database::models::organization_item::Organization::get_many_ids(
crate::database::models::organization_item::DBOrganization::get_many_ids(
&org_data, &**pool, &redis,
)
.await?;
@@ -302,11 +302,11 @@ pub async fn orgs_list(
.collect::<Vec<_>>();
let teams_data =
crate::database::models::TeamMember::get_from_team_full_many(
crate::database::models::DBTeamMember::get_from_team_full_many(
&team_ids, &**pool, &redis,
)
.await?;
let users = User::get_many_ids(
let users = DBUser::get_many_ids(
&teams_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
&**pool,
&redis,
@@ -397,7 +397,7 @@ pub async fn user_edit(
ApiError::Validation(validation_errors_to_string(err, None))
})?;
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(actual_user) = id_option {
let id = actual_user.id;
@@ -408,7 +408,7 @@ pub async fn user_edit(
if let Some(username) = &new_user.username {
let existing_user_id_option =
User::get(username, &**pool, &redis).await?;
DBUser::get(username, &**pool, &redis).await?;
if existing_user_id_option
.map(|x| UserId::from(x.id))
@@ -527,7 +527,7 @@ pub async fn user_edit(
}
transaction.commit().await?;
User::clear_caches(&[(id, Some(actual_user.username))], &redis)
DBUser::clear_caches(&[(id, Some(actual_user.username))], &redis)
.await?;
Ok(HttpResponse::NoContent().body(""))
} else {
@@ -565,7 +565,7 @@ pub async fn user_icon_edit(
)
.await?
.1;
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(actual_user) = id_option {
if user.id != actual_user.id.into() && !user.role.is_mod() {
@@ -612,7 +612,7 @@ pub async fn user_icon_edit(
)
.execute(&**pool)
.await?;
User::clear_caches(&[(actual_user.id, None)], &redis).await?;
DBUser::clear_caches(&[(actual_user.id, None)], &redis).await?;
Ok(HttpResponse::NoContent().body(""))
} else {
@@ -637,7 +637,7 @@ pub async fn user_icon_delete(
)
.await?
.1;
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(actual_user) = id_option {
if user.id != actual_user.id.into() && !user.role.is_mod() {
@@ -665,7 +665,7 @@ pub async fn user_icon_delete(
.execute(&**pool)
.await?;
User::clear_caches(&[(actual_user.id, None)], &redis).await?;
DBUser::clear_caches(&[(actual_user.id, None)], &redis).await?;
Ok(HttpResponse::NoContent().body(""))
} else {
@@ -689,7 +689,7 @@ pub async fn user_delete(
)
.await?
.1;
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(id) = id_option.map(|x| x.id) {
if !user.role.is_admin() && user.id != id.into() {
@@ -700,7 +700,7 @@ pub async fn user_delete(
let mut transaction = pool.begin().await?;
let result = User::remove(id, &mut transaction, &redis).await?;
let result = DBUser::remove(id, &mut transaction, &redis).await?;
transaction.commit().await?;
@@ -730,7 +730,7 @@ pub async fn user_follows(
)
.await?
.1;
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(id) = id_option.map(|x| x.id) {
if !user.role.is_admin() && user.id != id.into() {
@@ -739,8 +739,8 @@ pub async fn user_follows(
));
}
let project_ids = User::get_follows(id, &**pool).await?;
let projects: Vec<_> = crate::database::Project::get_many_ids(
let project_ids = DBUser::get_follows(id, &**pool).await?;
let projects: Vec<_> = crate::database::DBProject::get_many_ids(
&project_ids,
&**pool,
&redis,
@@ -772,7 +772,7 @@ pub async fn user_notifications(
)
.await?
.1;
let id_option = User::get(&info.into_inner().0, &**pool, &redis).await?;
let id_option = DBUser::get(&info.into_inner().0, &**pool, &redis).await?;
if let Some(id) = id_option.map(|x| x.id) {
if !user.role.is_admin() && user.id != id.into() {
@@ -782,7 +782,7 @@ pub async fn user_notifications(
}
let mut notifications: Vec<Notification> =
crate::database::models::notification_item::Notification::get_many_user(
crate::database::models::notification_item::DBNotification::get_many_user(
id, &**pool, &redis,
)
.await?

View File

@@ -7,7 +7,7 @@ use crate::database::models::notification_item::NotificationBuilder;
use crate::database::models::version_item::{
DependencyBuilder, VersionBuilder, VersionFileBuilder,
};
use crate::database::models::{self, Organization, image_item};
use crate::database::models::{self, DBOrganization, image_item};
use crate::database::redis::RedisPool;
use crate::file_hosting::FileHost;
use crate::models::ids::{ImageId, ProjectId, VersionId};
@@ -216,7 +216,7 @@ async fn version_create_inner(
let project_id: models::DBProjectId = version_create_data.project_id.unwrap().into();
// Ensure that the project this version is being added to exists
if models::Project::get_id(project_id, &mut **transaction, redis)
if models::DBProject::get_id(project_id, &mut **transaction, redis)
.await?
.is_none()
{
@@ -227,7 +227,7 @@ async fn version_create_inner(
// Check that the user creating this version is a team member
// of the project the version is being added to.
let team_member = models::TeamMember::get_from_user_id_project(
let team_member = models::DBTeamMember::get_from_user_id_project(
project_id,
user.id.into(),
false,
@@ -236,14 +236,14 @@ async fn version_create_inner(
.await?;
// Get organization attached, if exists, and the member project permissions
let organization = models::Organization::get_associated_organization_project_id(
let organization = models::DBOrganization::get_associated_organization_project_id(
project_id,
&mut **transaction,
)
.await?;
let organization_team_member = if let Some(organization) = &organization {
models::TeamMember::get_from_user_id(
models::DBTeamMember::get_from_user_id(
organization.team_id,
user.id.into(),
&mut **transaction,
@@ -481,7 +481,7 @@ async fn version_create_inner(
for image_id in version_data.uploaded_images {
if let Some(db_image) =
image_item::Image::get(image_id.into(), &mut **transaction, redis)
image_item::DBImage::get(image_id.into(), &mut **transaction, redis)
.await?
{
let image: Image = db_image.into();
@@ -505,7 +505,7 @@ async fn version_create_inner(
.execute(&mut **transaction)
.await?;
image_item::Image::clear_cache(image.id.into(), redis).await?;
image_item::DBImage::clear_cache(image.id.into(), redis).await?;
} else {
return Err(CreateError::InvalidInput(format!(
"Image {image_id} does not exist"
@@ -513,7 +513,7 @@ async fn version_create_inner(
}
}
models::Project::clear_cache(project_id, None, Some(true), redis).await?;
models::DBProject::clear_cache(project_id, None, Some(true), redis).await?;
let project_status = sqlx::query!(
"SELECT status FROM mods WHERE id = $1",
@@ -604,7 +604,7 @@ async fn upload_file_to_version_inner(
.await?
.1;
let result = models::Version::get(version_id, &**client, &redis).await?;
let result = models::DBVersion::get(version_id, &**client, &redis).await?;
let version = match result {
Some(v) => v,
@@ -629,7 +629,7 @@ async fn upload_file_to_version_inner(
})
.collect::<Result<Vec<_>, _>>()?;
if models::Project::get_id(
if models::DBProject::get_id(
version.inner.project_id,
&mut **transaction,
&redis,
@@ -643,7 +643,7 @@ async fn upload_file_to_version_inner(
}
if !user.role.is_admin() {
let team_member = models::TeamMember::get_from_user_id_project(
let team_member = models::DBTeamMember::get_from_user_id_project(
version.inner.project_id,
user.id.into(),
false,
@@ -652,7 +652,7 @@ async fn upload_file_to_version_inner(
.await?;
let organization =
Organization::get_associated_organization_project_id(
DBOrganization::get_associated_organization_project_id(
version.inner.project_id,
&**client,
)
@@ -660,7 +660,7 @@ async fn upload_file_to_version_inner(
let organization_team_member = if let Some(organization) = &organization
{
models::TeamMember::get_from_user_id(
models::DBTeamMember::get_from_user_id(
organization.team_id,
user.id.into(),
&mut **transaction,
@@ -782,7 +782,7 @@ async fn upload_file_to_version_inner(
}
// Clear version cache
models::Version::clear_cache(&version, &redis).await?;
models::DBVersion::clear_cache(&version, &redis).await?;
Ok(HttpResponse::NoContent().body(""))
}

View File

@@ -56,7 +56,7 @@ pub async fn get_version_from_hash(
.algorithm
.clone()
.unwrap_or_else(|| default_algorithm_from_hashes(&[hash.clone()]));
let file = database::models::Version::get_file_from_hash(
let file = database::models::DBVersion::get_file_from_hash(
algorithm,
hash,
hash_query.version_id.map(|x| x.into()),
@@ -66,7 +66,7 @@ pub async fn get_version_from_hash(
.await?;
if let Some(file) = file {
let version =
database::models::Version::get(file.version_id, &**pool, &redis)
database::models::DBVersion::get(file.version_id, &**pool, &redis)
.await?;
if let Some(version) = version {
if !is_visible_version(&version.inner, &user_option, &pool, &redis)
@@ -139,7 +139,7 @@ pub async fn get_update_from_hash(
.map(|x| x.1)
.ok();
let hash = info.into_inner().0.to_lowercase();
if let Some(file) = database::models::Version::get_file_from_hash(
if let Some(file) = database::models::DBVersion::get_file_from_hash(
hash_query
.algorithm
.clone()
@@ -151,11 +151,14 @@ pub async fn get_update_from_hash(
)
.await?
{
if let Some(project) =
database::models::Project::get_id(file.project_id, &**pool, &redis)
.await?
if let Some(project) = database::models::DBProject::get_id(
file.project_id,
&**pool,
&redis,
)
.await?
{
let mut versions = database::models::Version::get_many(
let mut versions = database::models::DBVersion::get_many(
&project.versions,
&**pool,
&redis,
@@ -241,7 +244,7 @@ pub async fn get_versions_from_hashes(
.clone()
.unwrap_or_else(|| default_algorithm_from_hashes(&file_data.hashes));
let files = database::models::Version::get_files_from_hash(
let files = database::models::DBVersion::get_files_from_hash(
algorithm.clone(),
&file_data.hashes,
&**pool,
@@ -251,7 +254,7 @@ pub async fn get_versions_from_hashes(
let version_ids = files.iter().map(|x| x.version_id).collect::<Vec<_>>();
let versions_data = filter_visible_versions(
database::models::Version::get_many(&version_ids, &**pool, &redis)
database::models::DBVersion::get_many(&version_ids, &**pool, &redis)
.await?,
&user_option,
&pool,
@@ -294,7 +297,7 @@ pub async fn get_projects_from_hashes(
.algorithm
.clone()
.unwrap_or_else(|| default_algorithm_from_hashes(&file_data.hashes));
let files = database::models::Version::get_files_from_hash(
let files = database::models::DBVersion::get_files_from_hash(
algorithm.clone(),
&file_data.hashes,
&**pool,
@@ -305,8 +308,12 @@ pub async fn get_projects_from_hashes(
let project_ids = files.iter().map(|x| x.project_id).collect::<Vec<_>>();
let projects_data = filter_visible_projects(
database::models::Project::get_many_ids(&project_ids, &**pool, &redis)
.await?,
database::models::DBProject::get_many_ids(
&project_ids,
&**pool,
&redis,
)
.await?,
&user_option,
&pool,
false,
@@ -343,7 +350,7 @@ pub async fn update_files(
.algorithm
.clone()
.unwrap_or_else(|| default_algorithm_from_hashes(&update_data.hashes));
let files = database::models::Version::get_files_from_hash(
let files = database::models::DBVersion::get_files_from_hash(
algorithm.clone(),
&update_data.hashes,
&**pool,
@@ -378,7 +385,7 @@ pub async fn update_files(
})
.await?;
let versions = database::models::Version::get_many(
let versions = database::models::DBVersion::get_many(
&update_version_ids
.into_iter()
.filter_map(|x| x.1.last().copied())
@@ -447,7 +454,7 @@ pub async fn update_individual_files(
.collect::<Vec<_>>(),
)
});
let files = database::models::Version::get_files_from_hash(
let files = database::models::DBVersion::get_files_from_hash(
algorithm.clone(),
&update_data
.hashes
@@ -459,13 +466,13 @@ pub async fn update_individual_files(
)
.await?;
let projects = database::models::Project::get_many_ids(
let projects = database::models::DBProject::get_many_ids(
&files.iter().map(|x| x.project_id).collect::<Vec<_>>(),
&**pool,
&redis,
)
.await?;
let all_versions = database::models::Version::get_many(
let all_versions = database::models::DBVersion::get_many(
&projects
.iter()
.flat_map(|x| x.versions.clone())
@@ -574,7 +581,7 @@ pub async fn delete_file(
.algorithm
.clone()
.unwrap_or_else(|| default_algorithm_from_hashes(&[hash.clone()]));
let file = database::models::Version::get_file_from_hash(
let file = database::models::DBVersion::get_file_from_hash(
algorithm.clone(),
hash,
hash_query.version_id.map(|x| x.into()),
@@ -586,7 +593,7 @@ pub async fn delete_file(
if let Some(row) = file {
if !user.role.is_admin() {
let team_member =
database::models::TeamMember::get_from_user_id_version(
database::models::DBTeamMember::get_from_user_id_version(
row.version_id,
user.id.into(),
&**pool,
@@ -595,26 +602,27 @@ pub async fn delete_file(
.map_err(ApiError::Database)?;
let organization =
database::models::Organization::get_associated_organization_project_id(
database::models::DBOrganization::get_associated_organization_project_id(
row.project_id,
&**pool,
)
.await
.map_err(ApiError::Database)?;
let organization_team_member =
if let Some(organization) = &organization {
database::models::TeamMember::get_from_user_id_organization(
organization.id,
user.id.into(),
false,
&**pool,
)
.await
.map_err(ApiError::Database)?
} else {
None
};
let organization_team_member = if let Some(organization) =
&organization
{
database::models::DBTeamMember::get_from_user_id_organization(
organization.id,
user.id.into(),
false,
&**pool,
)
.await
.map_err(ApiError::Database)?
} else {
None
};
let permissions = ProjectPermissions::get_permissions_by_role(
&user.role,
@@ -632,7 +640,7 @@ pub async fn delete_file(
}
let version =
database::models::Version::get(row.version_id, &**pool, &redis)
database::models::DBVersion::get(row.version_id, &**pool, &redis)
.await?;
if let Some(version) = version {
if version.files.len() < 2 {
@@ -642,7 +650,7 @@ pub async fn delete_file(
));
}
database::models::Version::clear_cache(&version, &redis).await?;
database::models::DBVersion::clear_cache(&version, &redis).await?;
}
let mut transaction = pool.begin().await?;
@@ -705,7 +713,7 @@ pub async fn download_version(
.algorithm
.clone()
.unwrap_or_else(|| default_algorithm_from_hashes(&[hash.clone()]));
let file = database::models::Version::get_file_from_hash(
let file = database::models::DBVersion::get_file_from_hash(
algorithm.clone(),
hash,
hash_query.version_id.map(|x| x.into()),
@@ -716,7 +724,7 @@ pub async fn download_version(
if let Some(file) = file {
let version =
database::models::Version::get(file.version_id, &**pool, &redis)
database::models::DBVersion::get(file.version_id, &**pool, &redis)
.await?;
if let Some(version) = version {

View File

@@ -9,8 +9,10 @@ use crate::database;
use crate::database::models::loader_fields::{
self, LoaderField, LoaderFieldEnumValue, VersionField,
};
use crate::database::models::version_item::{DependencyBuilder, LoaderVersion};
use crate::database::models::{Organization, image_item};
use crate::database::models::version_item::{
DBLoaderVersion, DependencyBuilder,
};
use crate::database::models::{DBOrganization, image_item};
use crate::database::redis::RedisPool;
use crate::models;
use crate::models::ids::VersionId;
@@ -70,7 +72,8 @@ pub async fn version_project_get_helper(
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
let result = database::models::Project::get(&id.0, &**pool, &redis).await?;
let result =
database::models::DBProject::get(&id.0, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
@@ -90,7 +93,7 @@ pub async fn version_project_get_helper(
return Err(ApiError::NotFound);
}
let versions = database::models::Version::get_many(
let versions = database::models::DBVersion::get_many(
&project.versions,
&**pool,
&redis,
@@ -134,7 +137,7 @@ pub async fn versions_get(
.map(|x| x.into())
.collect::<Vec<database::models::DBVersionId>>();
let versions_data =
database::models::Version::get_many(&version_ids, &**pool, &redis)
database::models::DBVersion::get_many(&version_ids, &**pool, &redis)
.await?;
let user_option = get_user_from_headers(
@@ -174,7 +177,7 @@ pub async fn version_get_helper(
session_queue: web::Data<AuthQueue>,
) -> Result<HttpResponse, ApiError> {
let version_data =
database::models::Version::get(id.into(), &**pool, &redis).await?;
database::models::DBVersion::get(id.into(), &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
@@ -290,11 +293,11 @@ pub async fn version_edit_helper(
let version_id = info.0.into();
let result =
database::models::Version::get(version_id, &**pool, &redis).await?;
database::models::DBVersion::get(version_id, &**pool, &redis).await?;
if let Some(version_item) = result {
let team_member =
database::models::TeamMember::get_from_user_id_project(
database::models::DBTeamMember::get_from_user_id_project(
version_item.inner.project_id,
user.id.into(),
false,
@@ -303,7 +306,7 @@ pub async fn version_edit_helper(
.await?;
let organization =
Organization::get_associated_organization_project_id(
DBOrganization::get_associated_organization_project_id(
version_item.inner.project_id,
&**pool,
)
@@ -311,7 +314,7 @@ pub async fn version_edit_helper(
let organization_team_member = if let Some(organization) = &organization
{
database::models::TeamMember::get_from_user_id(
database::models::DBTeamMember::get_from_user_id(
organization.team_id,
user.id.into(),
&**pool,
@@ -513,15 +516,15 @@ pub async fn version_edit_helper(
.to_string(),
)
})?;
loader_versions.push(LoaderVersion {
loader_versions.push(DBLoaderVersion {
loader_id,
version_id,
});
}
LoaderVersion::insert_many(loader_versions, &mut transaction)
DBLoaderVersion::insert_many(loader_versions, &mut transaction)
.await?;
crate::database::models::Project::clear_cache(
crate::database::models::DBProject::clear_cache(
version_item.inner.project_id,
None,
None,
@@ -679,9 +682,9 @@ pub async fn version_edit_helper(
.await?;
transaction.commit().await?;
database::models::Version::clear_cache(&version_item, &redis)
database::models::DBVersion::clear_cache(&version_item, &redis)
.await?;
database::models::Project::clear_cache(
database::models::DBProject::clear_cache(
version_item.inner.project_id,
None,
Some(true),
@@ -726,7 +729,7 @@ pub async fn version_list(
let string = info.into_inner().0;
let result =
database::models::Project::get(&string, &**pool, &redis).await?;
database::models::DBProject::get(&string, &**pool, &redis).await?;
let user_option = get_user_from_headers(
&req,
@@ -753,7 +756,7 @@ pub async fn version_list(
let loader_filters = filters.loaders.as_ref().map(|x| {
serde_json::from_str::<Vec<String>>(x).unwrap_or_default()
});
let mut versions = database::models::Version::get_many(
let mut versions = database::models::DBVersion::get_many(
&project.versions,
&**pool,
&redis,
@@ -886,7 +889,7 @@ pub async fn version_delete(
.1;
let id = info.into_inner().0;
let version = database::models::Version::get(id.into(), &**pool, &redis)
let version = database::models::DBVersion::get(id.into(), &**pool, &redis)
.await?
.ok_or_else(|| {
ApiError::InvalidInput(
@@ -896,7 +899,7 @@ pub async fn version_delete(
if !user.role.is_admin() {
let team_member =
database::models::TeamMember::get_from_user_id_project(
database::models::DBTeamMember::get_from_user_id_project(
version.inner.project_id,
user.id.into(),
false,
@@ -906,7 +909,7 @@ pub async fn version_delete(
.map_err(ApiError::Database)?;
let organization =
Organization::get_associated_organization_project_id(
DBOrganization::get_associated_organization_project_id(
version.inner.project_id,
&**pool,
)
@@ -914,7 +917,7 @@ pub async fn version_delete(
let organization_team_member = if let Some(organization) = &organization
{
database::models::TeamMember::get_from_user_id(
database::models::DBTeamMember::get_from_user_id(
organization.team_id,
user.id.into(),
&**pool,
@@ -942,14 +945,16 @@ pub async fn version_delete(
let context = ImageContext::Version {
version_id: Some(version.inner.id.into()),
};
let uploaded_images =
database::models::Image::get_many_contexted(context, &mut transaction)
.await?;
let uploaded_images = database::models::DBImage::get_many_contexted(
context,
&mut transaction,
)
.await?;
for image in uploaded_images {
image_item::Image::remove(image.id, &mut transaction, &redis).await?;
image_item::DBImage::remove(image.id, &mut transaction, &redis).await?;
}
let result = database::models::Version::remove_full(
let result = database::models::DBVersion::remove_full(
version.inner.id,
&redis,
&mut transaction,
@@ -957,7 +962,7 @@ pub async fn version_delete(
.await?;
transaction.commit().await?;
remove_documents(&[version.inner.id.into()], &search_config).await?;
database::models::Project::clear_cache(
database::models::DBProject::clear_cache(
version.inner.project_id,
None,
Some(true),