diff options
Diffstat (limited to 'import/src/lib.rs')
| -rw-r--r-- | import/src/lib.rs | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/import/src/lib.rs b/import/src/lib.rs index eee7a42..0d8a10a 100644 --- a/import/src/lib.rs +++ b/import/src/lib.rs @@ -460,6 +460,7 @@ fn apply_node_flag( "movie" | "" => TraktKind::Movie, "show" => TraktKind::Show, "season" => TraktKind::Season, + "episode" => TraktKind::Episode, _ => bail!("unknown trakt kind"), }; apply_trakt_tmdb(db, rthandle, apis, node, kind, id)?; @@ -476,6 +477,25 @@ fn apply_node_flag( Ok(()) })?; } + if let Some(kind) = flag.strip_prefix("kind-").or(flag.strip_prefix("kind=")) { + let kind = match kind { + "movie" => NodeKind::Movie, + "video" => NodeKind::Video, + "music" => NodeKind::Music, + "short_form_video" => NodeKind::ShortFormVideo, + "collection" => NodeKind::Collection, + "channel" => NodeKind::Channel, + "show" => NodeKind::Show, + "series" => NodeKind::Series, + "season" => NodeKind::Season, + "episode" => NodeKind::Episode, + _ => bail!("unknown node kind"), + }; + db.update_node_init(node, |node| { + node.kind = kind; + Ok(()) + })?; + } Ok(()) } |