Implement multiple episode numbers

crates/cli/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flix-cli"
-version = "0.0.5"
+version = "0.0.6"
 
 categories = ["command-line-utilities"]
 description = "CLI for interacting with flix media"
@@ -48,5 +48,6 @@ clap = { workspace = true, features = [
 ] }
 home = { workspace = true }
 serde = { workspace = true, features = ["derive"] }
+futures = { workspace = true }
 tokio = { workspace = true, features = ["rt", "fs", "macros"] }
 toml = { workspace = true, features = ["display", "parse"] }

crates/cli/src/cli/mod.rs

@@ -23,8 +23,8 @@ impl Cli {
 		}
 	}
 
-	pub fn command(&self) -> &Command {
-		&self.command
+	pub fn command(self) -> Command {
+		self.command
 	}
 }
 

crates/cli/src/cli/tmdb.rs

@@ -25,6 +25,7 @@ pub enum Command {
 		season: u32,
 	},
 	/// Process a TMDB episode
+	#[command(trailing_var_arg = true)]
 	Episode {
 		#[arg(value_name = "TMDB_ID")]
 		id: u32,
@@ -32,5 +33,7 @@ pub enum Command {
 		season: u32,
 		#[arg(value_name = "EPISODE_NUM")]
 		episode: u32,
+		#[arg(value_name = "...")]
+		episodes: Vec<u32>,
 	},
 }

crates/cli/src/main.rs

@@ -33,14 +33,14 @@ async fn main() -> Result<()> {
 			force,
 			output,
 			command,
-		} => exec_write(client, *force, output, command).await?,
-		Command::Update { output } => exec_update(client, output).await?,
+		} => exec_write(client, force, &output, command).await?,
+		Command::Update { output } => exec_update(client, &output).await?,
 	}
 
 	Ok(())
 }
 
-async fn exec_print(client: Client, command: &BackendCommand) -> Result<()> {
+async fn exec_print(client: Client, command: BackendCommand) -> Result<()> {
 	match command {
 		BackendCommand::Tmdb { command } => {
 			let object = run::tmdb::TmdbObject::fetch(&client, command).await?;
@@ -54,7 +54,7 @@ async fn exec_write(
 	client: Client,
 	force: bool,
 	output: &Path,
-	command: &BackendCommand,
+	command: BackendCommand,
 ) -> Result<()> {
 	match command {
 		BackendCommand::Tmdb { command } => {
@@ -83,8 +83,9 @@ async fn exec_update(client: Client, output: &Path) -> Result<()> {
 	let content = fs::read_to_string(output)
 		.await
 		.with_context(|| format!("failed to read file at path: {}", output.display()))?;
-	let object: FlixObject = toml::from_str(&content).context("failed to deserialize flix file")?;
+	let object: FlixObject = toml::from_str(&content)
+		.with_context(|| format!("failed to deserialize flix file: {}", output.display()))?;
 
 	let command = object.backend_command()?;
-	exec_write(client, true, output, &command).await
+	exec_write(client, true, output, command).await
 }

crates/cli/src/run/flix.rs

@@ -1,6 +1,6 @@
 use flix::model::{Collection, Episode, Movie, Season, Show, Verse};
 
-use anyhow::{Result, bail};
+use anyhow::{Result, anyhow, bail};
 
 use crate::cli::BackendCommand;
 use crate::cli::tmdb;
@@ -56,7 +56,14 @@ impl FlixObject {
 				tmdb::Command::Episode {
 					id: tmdb.show_id.into(),
 					season: episode.episode.season,
-					episode: episode.episode.number,
+					episode: episode
+						.episode
+						.number
+						.primary_episode_number()
+						.ok_or_else(|| {
+							anyhow!("the episode does not have a primary episode number")
+						})?,
+					episodes: episode.episode.number.additional_episode_numbers(),
 				}
 				.into()
 			}

crates/cli/src/run/tmdb.rs

@@ -1,6 +1,7 @@
 use flix::model::{
-	Collection, Episode, GenericCollection, GenericEpisode, GenericMovie, GenericSeason,
-	GenericShow, Movie, Season, Show, TmdbCollection, TmdbEpisode, TmdbMovie, TmdbSeason, TmdbShow,
+	Collection, Episode, EpisodeNumber, GenericCollection, GenericEpisode, GenericMovie,
+	GenericSeason, GenericShow, Movie, Season, Show, TmdbCollection, TmdbEpisode, TmdbMovie,
+	TmdbSeason, TmdbShow,
 };
 use flix_tmdb::Client;
 use flix_tmdb::model::{
@@ -9,6 +10,7 @@ use flix_tmdb::model::{
 };
 
 use anyhow::{Context, Result};
+use futures::{StreamExt, TryStreamExt, stream};
 
 use crate::cli::tmdb::Command;
 
@@ -17,7 +19,7 @@ pub enum TmdbObject {
 	Movie(TMovie),
 	Show(TShow),
 	Season(TSeason, ShowId),
-	Episode(TEpisode, u32, ShowId),
+	Episode(TEpisode, Vec<u32>, u32, ShowId),
 }
 
 impl TmdbObject {
@@ -63,23 +65,34 @@ impl TmdbObject {
 				},
 				tmdb: Some(TmdbSeason { show_id }),
 			})?,
-			TmdbObject::Episode(tmdb, season_number, show_id) => toml::to_string(&Episode {
-				episode: GenericEpisode {
-					number: tmdb.episode_number,
-					season: season_number,
-					title: tmdb.title,
-					overview: tmdb.overview,
-					air_date: tmdb.air_date,
-				},
-				tmdb: Some(TmdbEpisode { show_id }),
-			})?,
+			TmdbObject::Episode(tmdb, mut episode_numbers, season_number, show_id) => {
+				toml::to_string(&Episode {
+					episode: GenericEpisode {
+						number: if episode_numbers.is_empty() {
+							EpisodeNumber::Single {
+								number: tmdb.episode_number,
+							}
+						} else {
+							episode_numbers.insert(0, tmdb.episode_number);
+							EpisodeNumber::Multiple {
+								numbers: episode_numbers,
+							}
+						},
+						season: season_number,
+						title: tmdb.title,
+						overview: tmdb.overview,
+						air_date: tmdb.air_date,
+					},
+					tmdb: Some(TmdbEpisode { show_id }),
+				})?
+			}
 		})
 	}
 }
 
 impl TmdbObject {
-	pub async fn fetch(client: &Client, command: &Command) -> Result<Self> {
-		Ok(match *command {
+	pub async fn fetch(client: &Client, command: Command) -> Result<Self> {
+		Ok(match command {
 			Command::Collection { id } => Self::Collection(
 				client
 					.collections()
@@ -113,17 +126,32 @@ impl TmdbObject {
 				id,
 				season,
 				episode,
-			} => Self::Episode(
-				client
+				episodes,
+			} => {
+				let mut episode = client
 					.episodes()
 					.get_details(id, season, episode, None)
 					.await
 					.with_context(|| {
 						format!("could not get show details for '{id}' S{season}E{episode}")
-					})?,
-				season,
-				id.into(),
-			),
+					})?;
+				let title = stream::once(async { Ok(episode.title) })
+					.chain(stream::iter(episodes.clone()).then(|episode| async move {
+						client
+							.episodes()
+							.get_details(id, season, episode, None)
+							.await
+							.with_context(|| {
+								format!("could not get show details for '{id}' S{season}E{episode}")
+							})
+							.map(|episode| episode.title)
+					}))
+					.try_collect::<Vec<_>>()
+					.await?
+					.join(" + ");
+				episode.title = title;
+				Self::Episode(episode, episodes, season, id.into())
+			}
 		})
 	}
 }

crates/flix/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flix"
-version = "0.0.5"
+version = "0.0.6"
 
 categories = []
 description = "Types for storing persistent data about media"

crates/flix/src/model/episode.rs

@@ -14,11 +14,46 @@ pub struct Episode {
 	pub tmdb: Option<TmdbEpisode>,
 }
 
+/// A wrapper for handling single and multi-episode entries
+#[derive(Debug, serde::Serialize, serde::Deserialize)]
+#[serde(untagged)]
+pub enum EpisodeNumber {
+	/// The entry contains a single episode
+	Single {
+		/// The episode's number
+		number: u32,
+	},
+	/// The entry contains multiple episodes
+	Multiple {
+		/// The list of episode numbers
+		numbers: Vec<u32>,
+	},
+}
+
+impl EpisodeNumber {
+	/// Get the primary episode number of this episode
+	pub fn primary_episode_number(&self) -> Option<u32> {
+		match self {
+			EpisodeNumber::Single { number } => Some(*number),
+			EpisodeNumber::Multiple { numbers } => numbers.first().copied(),
+		}
+	}
+
+	/// Get additional episode numbers of this episode
+	pub fn additional_episode_numbers(&self) -> Vec<u32> {
+		match self {
+			EpisodeNumber::Single { number: _ } => vec![],
+			EpisodeNumber::Multiple { numbers } => numbers.iter().skip(1).copied().collect(),
+		}
+	}
+}
+
 /// The generic episode data
 #[derive(Debug, serde::Serialize, serde::Deserialize)]
 pub struct GenericEpisode {
-	/// The episode's number
-	pub number: u32,
+	/// The episode's number(s)
+	#[serde(flatten)]
+	pub number: EpisodeNumber,
 	/// The episode's season's number
 	pub season: u32,
 	/// The episode's title

crates/tmdb/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flix-tmdb"
-version = "0.0.5"
+version = "0.0.6"
 
 categories = []
 description = "Clients and models for fetching data from TMDB"