feat: add basic graphql support

apps/recorder/src/models/auth.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use sea_orm::entity::prelude::*;
 use serde::{Deserialize, Serialize};
 
@@ -50,5 +51,5 @@ pub enum RelatedEntity {
     Subscriber,
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}

apps/recorder/src/models/bangumi.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use loco_rs::app::AppContext;
 use sea_orm::{entity::prelude::*, sea_query::OnConflict, ActiveValue, FromJsonQueryResult};
 use serde::{Deserialize, Serialize};
@@ -59,6 +60,8 @@ pub enum Relation {
     Subscriber,
     #[sea_orm(has_many = "super::episodes::Entity")]
     Episode,
+    #[sea_orm(has_many = "super::subscription_bangumi::Entity")]
+    SubscriptionBangumi,
 }
 
 impl Related<super::episodes::Entity> for Entity {
@@ -67,6 +70,12 @@ impl Related<super::episodes::Entity> for Entity {
     }
 }
 
+impl Related<super::subscription_bangumi::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::SubscriptionBangumi.def()
+    }
+}
+
 impl Related<super::subscriptions::Entity> for Entity {
     fn to() -> RelationDef {
         super::subscription_bangumi::Relation::Subscription.def()
@@ -83,6 +92,18 @@ impl Related<super::subscribers::Entity> for Entity {
     }
 }
 
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscriptions::Entity")]
+    Subscription,
+    #[sea_orm(entity = "super::subscribers::Entity")]
+    Subscriber,
+    #[sea_orm(entity = "super::episodes::Entity")]
+    Episode,
+    #[sea_orm(entity = "super::subscription_bangumi::Entity")]
+    SubscriptionBangumi,
+}
+
 impl Model {
     pub async fn get_or_insert_from_mikan<F>(
         ctx: &AppContext,
@@ -146,5 +167,5 @@ impl Model {
     }
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}

apps/recorder/src/models/downloaders.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use sea_orm::entity::prelude::*;
 use serde::{Deserialize, Serialize};
 use url::Url;
@@ -5,11 +6,17 @@ use url::Url;
 #[derive(
     Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, DeriveDisplay, Serialize, Deserialize,
 )]
-#[sea_orm(rs_type = "String", db_type = "Enum", enum_name = "downloader_type")]
+#[sea_orm(
+    rs_type = "String",
+    db_type = "Enum",
+    enum_name = "downloader_category"
+)]
 #[serde(rename_all = "snake_case")]
 pub enum DownloaderCategory {
     #[sea_orm(string_value = "qbittorrent")]
     QBittorrent,
+    #[sea_orm(string_value = "dandanplay")]
+    Dandanplay,
 }
 
 #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
@@ -55,7 +62,15 @@ impl Related<super::downloads::Entity> for Entity {
     }
 }
 
-#[async_trait::async_trait]
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscribers::Entity")]
+    Subscriber,
+    #[sea_orm(entity = "super::downloads::Entity")]
+    Download,
+}
+
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}
 
 impl Model {

apps/recorder/src/models/downloads.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use sea_orm::entity::prelude::*;
 use serde::{Deserialize, Serialize};
 
@@ -101,7 +102,17 @@ impl Related<super::episodes::Entity> for Entity {
     }
 }
 
-#[async_trait::async_trait]
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscribers::Entity")]
+    Subscriber,
+    #[sea_orm(entity = "super::downloaders::Entity")]
+    Downloader,
+    #[sea_orm(entity = "super::episodes::Entity")]
+    Episode,
+}
+
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}
 
 impl ActiveModel {}

apps/recorder/src/models/entities/auth.rs

@@ -1,43 +0,0 @@
-use sea_orm::entity::prelude::*;
-use serde::{Deserialize, Serialize};
-
-#[derive(
-    Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, DeriveDisplay, Serialize, Deserialize,
-)]
-#[sea_orm(rs_type = "String", db_type = "Enum", enum_name = "auth_type")]
-#[serde(rename_all = "snake_case")]
-pub enum AuthType {
-    #[sea_orm(string_value = "basic")]
-    Basic,
-    #[sea_orm(string_value = "oidc")]
-    Oidc,
-}
-
-#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, DeriveEntityModel)]
-#[sea_orm(table_name = "auth")]
-pub struct Model {
-    pub created_at: DateTime,
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    pub pid: String,
-    pub subscriber_id: i32,
-    pub auth_type: AuthType,
-    pub avatar_url: Option<String>,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(
-        belongs_to = "super::subscribers::Entity",
-        from = "Column::SubscriberId",
-        to = "super::subscribers::Column::Id"
-    )]
-    SubscriberId,
-}
-
-impl Related<super::subscribers::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::SubscriberId.def()
-    }
-}

apps/recorder/src/models/entities/bangumi.rs

@@ -1,80 +0,0 @@
-use sea_orm::{entity::prelude::*, FromJsonQueryResult};
-use serde::{Deserialize, Serialize};
-
-#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, FromJsonQueryResult)]
-pub struct BangumiFilter {
-    pub name: Option<Vec<String>>,
-    pub group: Option<Vec<String>>,
-}
-
-#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, FromJsonQueryResult)]
-pub struct BangumiExtra {
-    pub name_zh: Option<String>,
-    pub s_name_zh: Option<String>,
-    pub name_en: Option<String>,
-    pub s_name_en: Option<String>,
-    pub name_jp: Option<String>,
-    pub s_name_jp: Option<String>,
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
-#[sea_orm(table_name = "bangumi")]
-pub struct Model {
-    pub created_at: DateTime,
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    pub mikan_bangumi_id: Option<String>,
-    pub subscription_id: i32,
-    pub subscriber_id: i32,
-    pub display_name: String,
-    pub raw_name: String,
-    pub season: i32,
-    pub season_raw: Option<String>,
-    pub fansub: Option<String>,
-    pub mikan_fansub_id: Option<String>,
-    pub filter: Option<BangumiFilter>,
-    pub rss_link: Option<String>,
-    pub poster_link: Option<String>,
-    pub save_path: Option<String>,
-    #[sea_orm(default = "false")]
-    pub deleted: bool,
-    pub homepage: Option<String>,
-    pub extra: Option<BangumiExtra>,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(
-        belongs_to = "super::subscriptions::Entity",
-        from = "Column::SubscriptionId",
-        to = "super::subscriptions::Column::Id"
-    )]
-    Subscription,
-    #[sea_orm(
-        belongs_to = "super::subscribers::Entity",
-        from = "Column::SubscriberId",
-        to = "super::subscribers::Column::Id"
-    )]
-    Subscriber,
-    #[sea_orm(has_many = "super::episodes::Entity")]
-    Episode,
-}
-
-impl Related<super::episodes::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Episode.def()
-    }
-}
-
-impl Related<super::subscriptions::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscription.def()
-    }
-}
-
-impl Related<super::subscribers::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscriber.def()
-    }
-}

apps/recorder/src/models/entities/downloaders.rs

@@ -1,45 +0,0 @@
-use sea_orm::entity::prelude::*;
-use serde::{Deserialize, Serialize};
-
-#[derive(
-    Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, DeriveDisplay, Serialize, Deserialize,
-)]
-#[sea_orm(rs_type = "String", db_type = "Enum", enum_name = "downloader_type")]
-#[serde(rename_all = "snake_case")]
-pub enum DownloaderCategory {
-    #[sea_orm(string_value = "qbittorrent")]
-    QBittorrent,
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
-#[sea_orm(table_name = "downloaders")]
-pub struct Model {
-    #[sea_orm(column_type = "Timestamp")]
-    pub created_at: DateTime,
-    #[sea_orm(column_type = "Timestamp")]
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    pub category: DownloaderCategory,
-    pub endpoint: String,
-    pub password: String,
-    pub username: String,
-    pub subscriber_id: i32,
-    pub save_path: String,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(
-        belongs_to = "super::subscribers::Entity",
-        from = "Column::SubscriberId",
-        to = "super::subscribers::Column::Id"
-    )]
-    Subscriber,
-}
-
-impl Related<super::subscribers::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscriber.def()
-    }
-}

apps/recorder/src/models/entities/downloads.rs

@@ -1,78 +0,0 @@
-use sea_orm::entity::prelude::*;
-use serde::{Deserialize, Serialize};
-
-#[derive(
-    Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, DeriveDisplay, Serialize, Deserialize,
-)]
-#[sea_orm(rs_type = "String", db_type = "Enum", enum_name = "download_status")]
-#[serde(rename_all = "snake_case")]
-pub enum DownloadStatus {
-    #[sea_orm(string_value = "pending")]
-    Pending,
-    #[sea_orm(string_value = "downloading")]
-    Downloading,
-    #[sea_orm(string_value = "paused")]
-    Paused,
-    #[sea_orm(string_value = "completed")]
-    Completed,
-    #[sea_orm(string_value = "failed")]
-    Failed,
-    #[sea_orm(string_value = "deleted")]
-    Deleted,
-}
-
-#[derive(
-    Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, DeriveDisplay, Serialize, Deserialize,
-)]
-#[sea_orm(rs_type = "String", db_type = "Enum", enum_name = "download_mime")]
-pub enum DownloadMime {
-    #[sea_orm(string_value = "application/octet-stream")]
-    #[serde(rename = "application/octet-stream")]
-    OctetStream,
-    #[sea_orm(string_value = "application/x-bittorrent")]
-    #[serde(rename = "application/x-bittorrent")]
-    BitTorrent,
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
-#[sea_orm(table_name = "downloads")]
-pub struct Model {
-    pub created_at: DateTime,
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    pub origin_name: String,
-    pub display_name: String,
-    pub subscription_id: i32,
-    pub status: DownloadStatus,
-    pub mime: DownloadMime,
-    pub url: String,
-    pub all_size: Option<u64>,
-    pub curr_size: Option<u64>,
-    pub homepage: Option<String>,
-    pub save_path: Option<String>,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(
-        belongs_to = "super::subscriptions::Entity",
-        from = "Column::SubscriptionId",
-        to = "super::subscriptions::Column::Id"
-    )]
-    Subscription,
-    #[sea_orm(has_many = "super::episodes::Entity")]
-    Episode,
-}
-
-impl Related<super::subscriptions::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscription.def()
-    }
-}
-
-impl Related<super::episodes::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Episode.def()
-    }
-}

apps/recorder/src/models/entities/episodes.rs

@@ -1,95 +0,0 @@
-//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.2
-use sea_orm::{entity::prelude::*, FromJsonQueryResult};
-use serde::{Deserialize, Serialize};
-
-#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, FromJsonQueryResult, Default)]
-pub struct EpisodeExtra {
-    pub name_zh: Option<String>,
-    pub s_name_zh: Option<String>,
-    pub name_en: Option<String>,
-    pub s_name_en: Option<String>,
-    pub name_jp: Option<String>,
-    pub s_name_jp: Option<String>,
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
-#[sea_orm(table_name = "episodes")]
-pub struct Model {
-    pub created_at: DateTime,
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    #[sea_orm(indexed)]
-    pub mikan_episode_id: Option<String>,
-    pub raw_name: String,
-    pub display_name: String,
-    pub bangumi_id: i32,
-    pub subscription_id: i32,
-    pub subscriber_id: i32,
-    pub download_id: Option<i32>,
-    pub save_path: Option<String>,
-    pub resolution: Option<String>,
-    pub season: i32,
-    pub season_raw: Option<String>,
-    pub fansub: Option<String>,
-    pub poster_link: Option<String>,
-    pub episode_index: i32,
-    pub homepage: Option<String>,
-    pub subtitle: Option<Vec<String>>,
-    #[sea_orm(default = "false")]
-    pub deleted: bool,
-    pub source: Option<String>,
-    pub extra: EpisodeExtra,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(
-        belongs_to = "super::bangumi::Entity",
-        from = "Column::BangumiId",
-        to = "super::bangumi::Column::Id"
-    )]
-    Bangumi,
-    #[sea_orm(
-        belongs_to = "super::downloads::Entity",
-        from = "Column::DownloadId",
-        to = "super::downloads::Column::Id"
-    )]
-    Downloads,
-    #[sea_orm(
-        belongs_to = "super::subscriptions::Entity",
-        from = "Column::SubscriptionId",
-        to = "super::subscriptions::Column::Id"
-    )]
-    Subscriptions,
-    #[sea_orm(
-        belongs_to = "super::subscribers::Entity",
-        from = "Column::SubscriberId",
-        to = "super::subscribers::Column::Id"
-    )]
-    Subscriber,
-}
-
-impl Related<super::bangumi::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Bangumi.def()
-    }
-}
-
-impl Related<super::downloads::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Downloads.def()
-    }
-}
-
-impl Related<super::subscriptions::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscriptions.def()
-    }
-}
-
-impl Related<super::subscribers::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscriber.def()
-    }
-}

apps/recorder/src/models/entities/mod.rs

@@ -1,8 +0,0 @@
-//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.4
-pub mod auth;
-pub mod bangumi;
-pub mod downloaders;
-pub mod downloads;
-pub mod episodes;
-pub mod subscribers;
-pub mod subscriptions;

apps/recorder/src/models/entities/subscribers.rs

@@ -1,71 +0,0 @@
-//! `SeaORM` Entity. Generated by sea-orm-codegen 0.12.2
-
-use sea_orm::{entity::prelude::*, FromJsonQueryResult};
-use serde::{Deserialize, Serialize};
-
-#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize, FromJsonQueryResult)]
-pub struct SubscriberBangumiConfig {
-    pub leading_group_tag: Option<bool>,
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
-#[sea_orm(table_name = "subscribers")]
-pub struct Model {
-    pub created_at: DateTime,
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    #[sea_orm(unique)]
-    pub pid: String,
-    pub display_name: String,
-    pub downloader_id: Option<i32>,
-    pub bangumi_conf: Option<SubscriberBangumiConfig>,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(has_many = "super::subscriptions::Entity")]
-    Subscription,
-    #[sea_orm(
-        belongs_to = "super::downloaders::Entity",
-        from = "Column::DownloaderId",
-        to = "super::downloaders::Column::Id"
-    )]
-    Downloader,
-    #[sea_orm(has_many = "super::bangumi::Entity")]
-    Bangumi,
-    #[sea_orm(has_many = "super::episodes::Entity")]
-    Episode,
-    #[sea_orm(has_many = "super::auth::Entity")]
-    Auth,
-}
-
-impl Related<super::subscriptions::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscription.def()
-    }
-}
-
-impl Related<super::downloaders::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Downloader.def()
-    }
-}
-
-impl Related<super::bangumi::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Bangumi.def()
-    }
-}
-
-impl Related<super::episodes::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Episode.def()
-    }
-}
-
-impl Related<super::auth::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Auth.def()
-    }
-}

apps/recorder/src/models/entities/subscriptions.rs

@@ -1,66 +0,0 @@
-use sea_orm::entity::prelude::*;
-use serde::{Deserialize, Serialize};
-
-#[derive(
-    Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize, DeriveDisplay,
-)]
-#[sea_orm(
-    rs_type = "String",
-    db_type = "Enum",
-    enum_name = "subscription_category"
-)]
-#[serde(rename_all = "snake_case")]
-pub enum SubscriptionCategory {
-    #[sea_orm(string_value = "mikan")]
-    Mikan,
-    #[sea_orm(string_value = "manual")]
-    Manual,
-}
-
-#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
-#[sea_orm(table_name = "subscriptions")]
-pub struct Model {
-    #[sea_orm(column_type = "Timestamp")]
-    pub created_at: DateTime,
-    #[sea_orm(column_type = "Timestamp")]
-    pub updated_at: DateTime,
-    #[sea_orm(primary_key)]
-    pub id: i32,
-    pub display_name: String,
-    pub subscriber_id: i32,
-    pub category: SubscriptionCategory,
-    pub source_url: String,
-    pub enabled: bool,
-}
-
-#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {
-    #[sea_orm(
-        belongs_to = "super::subscribers::Entity",
-        from = "Column::SubscriberId",
-        to = "super::subscribers::Column::Id"
-    )]
-    Subscriber,
-    #[sea_orm(has_many = "super::bangumi::Entity")]
-    Bangumi,
-    #[sea_orm(has_many = "super::episodes::Entity")]
-    Episodes,
-}
-
-impl Related<super::subscribers::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscriber.def()
-    }
-}
-
-impl Related<super::bangumi::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Bangumi.def()
-    }
-}
-
-impl Related<super::episodes::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Episodes.def()
-    }
-}

apps/recorder/src/models/episodes.rs

@@ -1,5 +1,6 @@
 use std::sync::Arc;
 
+use async_trait::async_trait;
 use loco_rs::app::AppContext;
 use sea_orm::{entity::prelude::*, sea_query::OnConflict, ActiveValue, FromJsonQueryResult};
 use serde::{Deserialize, Serialize};
@@ -70,9 +71,11 @@ pub enum Relation {
     )]
     Bangumi,
     #[sea_orm(has_many = "super::subscriptions::Entity")]
-    Subscriptions,
+    Subscription,
     #[sea_orm(has_one = "super::downloads::Entity")]
-    Downloads,
+    Download,
+    #[sea_orm(has_many = "super::subscription_episode::Entity")]
+    SubscriptionEpisode,
 }
 
 impl Related<super::bangumi::Entity> for Entity {
@@ -83,13 +86,7 @@ impl Related<super::bangumi::Entity> for Entity {
 
 impl Related<super::downloads::Entity> for Entity {
     fn to() -> RelationDef {
-        Relation::Downloads.def()
-    }
-}
-
-impl Related<super::subscriptions::Entity> for Entity {
-    fn to() -> RelationDef {
-        Relation::Subscriptions.def()
+        Relation::Download.def()
     }
 }
 
@@ -99,6 +96,36 @@ impl Related<super::subscribers::Entity> for Entity {
     }
 }
 
+impl Related<super::subscription_episode::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::SubscriptionEpisode.def()
+    }
+}
+
+impl Related<super::subscriptions::Entity> for Entity {
+    fn to() -> RelationDef {
+        super::subscription_episode::Relation::Subscription.def()
+    }
+
+    fn via() -> Option<RelationDef> {
+        Some(Relation::Subscription.def())
+    }
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscribers::Entity")]
+    Subscriber,
+    #[sea_orm(entity = "super::downloads::Entity")]
+    Subscription,
+    #[sea_orm(entity = "super::bangumi::Entity")]
+    Bangumi,
+    #[sea_orm(entity = "super::subscriptions::Entity")]
+    Download,
+    #[sea_orm(entity = "super::subscription_episode::Entity")]
+    SubscriptionEpisode,
+}
+
 #[derive(Clone, Debug, PartialEq)]
 pub struct MikanEpsiodeCreation {
     pub episode: MikanEpisodeMeta,
@@ -206,5 +233,5 @@ impl ActiveModel {
     }
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}

apps/recorder/src/models/mod.rs

@@ -3,7 +3,6 @@ pub mod bangumi;
 pub mod downloaders;
 pub mod downloads;
 pub mod episodes;
-pub mod prelude;
 pub mod query;
 pub mod subscribers;
 pub mod subscription_bangumi;

apps/recorder/src/models/notifications.rs

@@ -1,8 +0,0 @@
-use serde::{Deserialize, Serialize};
-
-#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
-pub struct Notification {
-    season: i32,
-    episode_size: u32,
-    poster_url: Option<String>,
-}

apps/recorder/src/models/prelude.rs

@@ -1,8 +0,0 @@
-pub use super::{
-    bangumi::{self, Entity as Bangumi},
-    downloaders::{self, DownloaderCategory, Entity as Downloader},
-    downloads::{self, DownloadMime, DownloadStatus, Entity as Download},
-    episodes::{self, Entity as Episode},
-    subscribers::{self, Entity as Subscriber},
-    subscriptions::{self, Entity as Subscription, SubscriptionCategory},
-};

apps/recorder/src/models/query/mod.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use sea_orm::{
     prelude::Expr,
     sea_query::{Alias, IntoColumnRef, IntoTableRef, Query, SelectStatement},
@@ -26,7 +27,7 @@ pub fn filter_values_in<
         .to_owned()
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 pub trait InsertManyReturningExt<A>: Sized
 where
     <A::Entity as EntityTrait>::Model: IntoActiveModel<A>,
@@ -49,7 +50,7 @@ where
         I: IntoIterator<Item = <A::Entity as EntityTrait>::Column> + Send;
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 impl<A> InsertManyReturningExt<A> for Insert<A>
 where
     <A::Entity as EntityTrait>::Model: IntoActiveModel<A>,

apps/recorder/src/models/subscribers.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use loco_rs::{
     app::AppContext,
     model::{ModelError, ModelResult},
@@ -69,12 +70,26 @@ impl Related<super::auth::Entity> for Entity {
     }
 }
 
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscriptions::Entity")]
+    Subscription,
+    #[sea_orm(entity = "super::downloaders::Entity")]
+    Downloader,
+    #[sea_orm(entity = "super::bangumi::Entity")]
+    Bangumi,
+    #[sea_orm(entity = "super::episodes::Entity")]
+    Episode,
+    #[sea_orm(entity = "super::auth::Entity")]
+    Auth,
+}
+
 #[derive(Debug, Deserialize, Serialize)]
 pub struct SubscriberIdParams {
     pub id: String,
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {
     async fn before_save<C>(self, _db: &C, insert: bool) -> Result<Self, DbErr>
     where

apps/recorder/src/models/subscription_bangumi.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use sea_orm::{entity::prelude::*, ActiveValue};
 use serde::{Deserialize, Serialize};
 
@@ -42,7 +43,15 @@ impl Related<super::bangumi::Entity> for Entity {
     }
 }
 
-#[async_trait::async_trait]
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscriptions::Entity")]
+    Subscription,
+    #[sea_orm(entity = "super::bangumi::Entity")]
+    Bangumi,
+}
+
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}
 
 impl ActiveModel {

apps/recorder/src/models/subscription_episode.rs

@@ -1,3 +1,4 @@
+use async_trait::async_trait;
 use sea_orm::{entity::prelude::*, ActiveValue};
 use serde::{Deserialize, Serialize};
 
@@ -42,7 +43,15 @@ impl Related<super::episodes::Entity> for Entity {
     }
 }
 
-#[async_trait::async_trait]
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscriptions::Entity")]
+    Subscription,
+    #[sea_orm(entity = "super::episodes::Entity")]
+    Episode,
+}
+
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}
 
 impl ActiveModel {

apps/recorder/src/models/subscriptions.rs

@@ -1,5 +1,6 @@
 use std::{collections::HashSet, sync::Arc};
 
+use async_trait::async_trait;
 use itertools::Itertools;
 use loco_rs::app::AppContext;
 use sea_orm::{entity::prelude::*, ActiveValue};
@@ -81,6 +82,18 @@ impl Related<super::subscribers::Entity> for Entity {
     }
 }
 
+impl Related<super::subscription_bangumi::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::SubscriptionBangumi.def()
+    }
+}
+
+impl Related<super::subscription_episode::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::SubscriptionEpisode.def()
+    }
+}
+
 impl Related<super::bangumi::Entity> for Entity {
     fn to() -> RelationDef {
         super::subscription_bangumi::Relation::Bangumi.def()
@@ -109,6 +122,20 @@ impl Related<super::episodes::Entity> for Entity {
     }
 }
 
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelatedEntity)]
+pub enum RelatedEntity {
+    #[sea_orm(entity = "super::subscribers::Entity")]
+    Subscriber,
+    #[sea_orm(entity = "super::bangumi::Entity")]
+    Bangumi,
+    #[sea_orm(entity = "super::episodes::Entity")]
+    Episode,
+    #[sea_orm(entity = "super::subscription_episode::Entity")]
+    SubscriptionEpisode,
+    #[sea_orm(entity = "super::subscription_bangumi::Entity")]
+    SubscriptionBangumi,
+}
+
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
 pub struct SubscriptionCreateFromRssDto {
     pub rss_link: String,
@@ -122,7 +149,7 @@ pub enum SubscriptionCreateDto {
     Mikan(SubscriptionCreateFromRssDto),
 }
 
-#[async_trait::async_trait]
+#[async_trait]
 impl ActiveModelBehavior for ActiveModel {}
 
 impl ActiveModel {