84 lines
2.4 KiB
Rust
84 lines
2.4 KiB
Rust
use crate::model_ext::EntityPrefixExt;
|
|
use crate::{
|
|
model_ext::{joined_prefix, joined_prefix_str, SelectColumnsExt},
|
|
AliasSourceExt, CalckeyModel,
|
|
};
|
|
use ck::{drive_file, follow_request, following, user};
|
|
use ext_calckey_model_migration::{Alias, SelectStatement};
|
|
use sea_orm::{DbErr, EntityName, FromQueryResult, JoinType, QueryResult};
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
pub struct UserData {
|
|
pub user: user::Model,
|
|
pub avatar: Option<drive_file::Model>,
|
|
pub banner: Option<drive_file::Model>,
|
|
}
|
|
|
|
pub struct UserResolveOptions {
|
|
pub with_avatar_and_banner: bool,
|
|
}
|
|
|
|
const AVATAR: &str = "avatar.";
|
|
const BANNER: &str = "banner.";
|
|
|
|
impl FromQueryResult for UserData {
|
|
fn from_query_result(res: &QueryResult, prefix: &str) -> Result<Self, DbErr> {
|
|
let fallback = user::Entity.base_prefix_str();
|
|
let prefix = if prefix.is_empty() { &fallback } else { prefix };
|
|
|
|
Ok(UserData {
|
|
user: user::Model::from_query_result(res, prefix)?,
|
|
avatar: drive_file::Model::from_query_result_optional(
|
|
res,
|
|
&joined_prefix_str(prefix, AVATAR),
|
|
)?,
|
|
banner: drive_file::Model::from_query_result_optional(
|
|
res,
|
|
&joined_prefix_str(prefix, BANNER),
|
|
)?,
|
|
})
|
|
}
|
|
}
|
|
|
|
pub struct UserResolver {
|
|
db: CalckeyModel,
|
|
}
|
|
|
|
impl UserResolver {
|
|
pub fn new(db: CalckeyModel) -> Self {
|
|
Self { db }
|
|
}
|
|
|
|
pub fn resolve(
|
|
&self,
|
|
q: &mut SelectStatement,
|
|
prefix: &str,
|
|
UserResolveOptions {
|
|
with_avatar_and_banner,
|
|
}: &UserResolveOptions,
|
|
) {
|
|
q.add_aliased_columns::<user::Entity>(prefix);
|
|
|
|
if *with_avatar_and_banner {
|
|
q.add_aliased_columns::<drive_file::Entity>(&joined_prefix_str(prefix, AVATAR))
|
|
.add_aliased_columns::<drive_file::Entity>(&joined_prefix_str(prefix, BANNER));
|
|
|
|
q.join_columns(
|
|
JoinType::LeftJoin,
|
|
user::Relation::DriveFile2.with_alias(Alias::new(prefix)),
|
|
joined_prefix(prefix, AVATAR),
|
|
)
|
|
.join_columns(
|
|
JoinType::LeftJoin,
|
|
user::Relation::DriveFile1.with_alias(Alias::new(prefix)),
|
|
joined_prefix(prefix, BANNER),
|
|
);
|
|
}
|
|
}
|
|
|
|
pub async fn get_followers(&self, user_id: &str) -> Vec<UserData> {
|
|
todo!()
|
|
}
|
|
}
|