refactor tmdb api, cast&crew, node ext

1 files changed, 73 insertions, 1 deletions

diff --git a/common/src/lib.rs b/common/src/lib.rs
index 57b210b..f16338b 100644
--- a/common/src/lib.rs
+++ b/common/src/lib.rs
@@ -18,7 +18,7 @@ use bincode::{Decode, Encode};
 #[cfg(feature = "rocket")]
 use rocket::{FromFormField, UriDisplayQuery};
 use serde::{Deserialize, Serialize};
-use std::{collections::BTreeMap, path::PathBuf};
+use std::{collections::BTreeMap, fmt::Display, path::PathBuf};
 
 #[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
 pub struct Node {
@@ -55,6 +55,57 @@ pub struct NodePublic {
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
+pub struct ExtendedNode {
+    pub ids: ObjectIds,
+    pub people: BTreeMap<PeopleGroup, Vec<Appearance>>,
+}
+
+#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
+pub struct Appearance {
+    #[serde(default)]
+    pub jobs: Vec<String>,
+    #[serde(default)]
+    pub characters: Vec<String>,
+    pub person: Person,
+}
+
+#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
+pub struct Person {
+    pub name: String,
+    pub asset: Option<AssetLocation>,
+    pub ids: ObjectIds,
+}
+
+#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
+pub struct ObjectIds {
+    pub trakt: Option<u64>,
+    pub slug: Option<String>,
+    pub imdb: Option<String>,
+    pub tmdb: Option<u64>,
+    pub omdb: Option<u64>,
+    pub tvdb: Option<u64>,
+}
+
+#[rustfmt::skip]
+#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, PartialOrd, Ord, Encode, Decode)]
+#[cfg_attr(feature = "rocket", derive(FromFormField, UriDisplayQuery))]
+#[serde(rename_all = "snake_case")]
+pub enum PeopleGroup {
+    #[cfg_attr(feature = "rocket", field(value = "cast"))] Cast,
+    #[cfg_attr(feature = "rocket", field(value = "writing"))] Writing,
+    #[cfg_attr(feature = "rocket", field(value = "directing"))] Directing,
+    #[cfg_attr(feature = "rocket", field(value = "art"))] Art,
+    #[cfg_attr(feature = "rocket", field(value = "sound"))] Sound,
+    #[cfg_attr(feature = "rocket", field(value = "camera"))] Camera,
+    #[cfg_attr(feature = "rocket", field(value = "lighting"))] Lighting,
+    #[cfg_attr(feature = "rocket", field(value = "crew"))] Crew,
+    #[cfg_attr(feature = "rocket", field(value = "editing"))] Editing,
+    #[cfg_attr(feature = "rocket", field(value = "production"))] Production,
+    #[cfg_attr(feature = "rocket", field(value = "vfx"))] Vfx,
+    #[cfg_attr(feature = "rocket", field(value = "costume"))] CostumeMakeup,
+}
+
+#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
 pub struct ImportOptions {
     pub id: String,
     pub sources: Vec<ImportSource>,
@@ -240,3 +291,24 @@ impl ToString for TraktKind {
         .to_string()
     }
 }
+impl Display for ObjectIds {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        f.write_str("trakt")?;
+        if self.slug.is_some() {
+            f.write_str(",slug")?;
+        }
+        if self.tmdb.is_some() {
+            f.write_str(",tmdb")?;
+        }
+        if self.imdb.is_some() {
+            f.write_str(",imdb")?;
+        }
+        if self.tvdb.is_some() {
+            f.write_str(",tvdb")?;
+        }
+        if self.omdb.is_some() {
+            f.write_str(",omdb")?;
+        }
+        Ok(())
+    }
+}