magnetar/ext_calckey_model/src/user_model.rs

use crate::model_ext::{AliasSourceExt, AliasSuffixExt, EntityPrefixExt, MagIden};
use crate::{model_ext::SelectColumnsExt, CalckeyModel};
use ck::{drive_file, user};
use ext_calckey_model_migration::{IntoIden, SelectStatement};
use sea_orm::sea_query::Alias;
use sea_orm::{DbErr, FromQueryResult, Iden, 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 prefix = if prefix.is_empty() {
            user::Entity.base_prefix()
        } else {
            MagIden::alias(prefix)
        };

        Ok(UserData {
            user: user::Model::from_query_result(res, &prefix.to_string())?,
            avatar: drive_file::Model::from_query_result_optional(
                res,
                &prefix.join_str_as_str(AVATAR),
            )?,
            banner: drive_file::Model::from_query_result_optional(
                res,
                &prefix.join_str_as_str(BANNER),
            )?,
        })
    }
}

pub struct UserResolver {
    db: CalckeyModel,
}

impl UserResolver {
    pub fn new(db: CalckeyModel) -> Self {
        Self { db }
    }

    pub fn resolve(
        &self,
        q: &mut SelectStatement,
        user_tbl: &MagIden,
        UserResolveOptions {
            with_avatar_and_banner,
        }: &UserResolveOptions,
    ) {
        q.add_aliased_columns::<user::Entity>(user_tbl);

        if *with_avatar_and_banner {
            let avatar_tbl = user_tbl.join_str(AVATAR);
            let banner_tbl = user_tbl.join_str(BANNER);

            q.add_aliased_columns::<drive_file::Entity>(&avatar_tbl)
                .add_aliased_columns::<drive_file::Entity>(&banner_tbl);

            q.join_columns(
                JoinType::LeftJoin,
                user::Relation::DriveFile2.with_from_alias(user_tbl),
                &avatar_tbl,
            )
            .join_columns(
                JoinType::LeftJoin,
                user::Relation::DriveFile1.with_from_alias(user_tbl),
                &banner_tbl,
            );
        }
    }

    pub async fn get_followers(&self, user_id: &str) -> Vec<UserData> {
        todo!()
    }
}