NyxErinys/crunchy-cli
1
0
Fork 0
mirror of https://github.com/crunchy-labs/crunchy-cli.git synced 2026-01-21 12:12:00 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up search a bit

Browse source
This commit is contained in:
bytedream 2023-06-19 17:23:52 +02:00
parent d12f2c036b
commit bd7a970f11
1 changed files with 119 additions and 178 deletions
Download patch file Download diff file Expand all files Collapse all files

297
crunchy-cli-core/src/search/format.rs
View file

@ -178,6 +178,29 @@ enum Scope {
Subtitle, Subtitle,
} }
macro_rules! must_match_if_true {
($condition:expr => $media_collection:ident | $field:pat => $expr:expr) => {
if $condition {
match &$media_collection {
$field => Some($expr),
_ => panic!(),
}
} else {
None
}
};
}
macro_rules! self_and_versions {
($var:expr => $audio:expr) => {
{
let mut items = vec![$var.clone()];
items.extend($var.clone().version($audio).await?);
items
}
};
}
pub struct Format { pub struct Format {
pattern: Vec<(Range<usize>, Scope, String)>, pattern: Vec<(Range<usize>, Scope, String)>,
pattern_count: HashMap<Scope, u32>, pattern_count: HashMap<Scope, u32>,
@ -191,44 +214,29 @@ impl Format {
let mut pattern = vec![]; let mut pattern = vec![];
let mut pattern_count = HashMap::new(); let mut pattern_count = HashMap::new();
let field_check = HashMap::from([ macro_rules! generate_field_check {
( ($($scope:expr => $struct_:ident)+) => {
Scope::Series, HashMap::from([
serde_json::to_value(FormatSeries::default()).unwrap(), $(
), (
( $scope,
Scope::Season, serde_json::from_value::<Map<String, Value>>(serde_json::to_value($struct_::default()).unwrap()).unwrap()
serde_json::to_value(FormatSeason::default()).unwrap(), )
), ),+
( ])
Scope::Episode, };
serde_json::to_value(FormatEpisode::default()).unwrap(), }
), let field_check = generate_field_check!(
( Scope::Series => FormatSeries
Scope::MovieListing, Scope::Season => FormatSeason
serde_json::to_value(FormatMovieListing::default()).unwrap(), Scope::Episode => FormatEpisode
), Scope::MovieListing => FormatMovieListing
( Scope::Movie => FormatMovie
Scope::Movie, Scope::MusicVideo => FormatMusicVideo
serde_json::to_value(FormatMovie::default()).unwrap(), Scope::Concert => FormatConcert
), Scope::Stream => FormatStream
( Scope::Subtitle => FormatSubtitle
Scope::MusicVideo, );
serde_json::to_value(FormatMusicVideo::default()).unwrap(),
),
(
Scope::Concert,
serde_json::to_value(FormatConcert::default()).unwrap(),
),
(
Scope::Stream,
serde_json::to_value(FormatStream::default()).unwrap(),
),
(
Scope::Subtitle,
serde_json::to_value(FormatSubtitle::default()).unwrap(),
),
]);
for capture in scope_regex.captures_iter(&input) { for capture in scope_regex.captures_iter(&input) {
let full = capture.get(0).unwrap(); let full = capture.get(0).unwrap();
@ -251,8 +259,6 @@ impl Format {
if field_check if field_check
.get(&format_pattern_scope) .get(&format_pattern_scope)
.unwrap() .unwrap()
.as_object()
.unwrap()
.get(field) .get(field)
.is_none() .is_none()
{ {
@ -323,16 +329,8 @@ impl Format {
_ => panic!(), _ => panic!(),
}; };
let mut seasons = vec![]; let mut seasons = vec![];
for mut season in tmp_seasons { for season in tmp_seasons {
if self seasons.extend(self_and_versions!(season => self.filter_options.audio.clone()))
.filter_options
.audio
.iter()
.any(|a| season.audio_locales.contains(a))
{
seasons.push(season.clone())
}
seasons.extend(season.version(self.filter_options.audio.clone()).await?);
} }
tree.extend( tree.extend(
self.filter_options self.filter_options
@ -346,17 +344,7 @@ impl Format {
if !episode_empty || !stream_empty { if !episode_empty || !stream_empty {
match &media_collection { match &media_collection {
MediaCollection::Episode(episode) => { MediaCollection::Episode(episode) => {
let mut episodes = vec![]; let episodes = self_and_versions!(episode => self.filter_options.audio.clone());
if self.filter_options.audio.contains(&episode.audio_locale) {
episodes.push(episode.clone())
}
episodes.extend(
episode
.clone()
.version(self.filter_options.audio.clone())
.await?,
);
tree.push(( tree.push((
Season::default(), Season::default(),
episodes episodes
@ -402,31 +390,21 @@ impl Format {
let season_map = self.serializable_to_json_map(FormatSeason::from(&season)); let season_map = self.serializable_to_json_map(FormatSeason::from(&season));
for (episode, streams) in episodes { for (episode, streams) in episodes {
let episode_map = self.serializable_to_json_map(FormatEpisode::from(&episode)); let episode_map = self.serializable_to_json_map(FormatEpisode::from(&episode));
for mut stream in streams { for stream in streams {
let stream_map = self.serializable_to_json_map(FormatStream::from(&stream)); let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
if stream.subtitles.is_empty() {
if !self.check_pattern_count_empty(Scope::Subtitle) { output.push(
continue; self.replace_all(
} HashMap::from([
stream (Scope::Series, &series_map),
.subtitles (Scope::Season, &season_map),
.insert(Locale::Custom("".to_string()), Subtitle::default()); (Scope::Episode, &episode_map),
} (Scope::Stream, &stream_map),
for subtitle in self ]),
.filter_options stream,
.filter_subtitles(stream.subtitles.into_values().collect()) )
{ .unwrap_or_default(),
let subtitle_map = )
self.serializable_to_json_map(FormatSubtitle::from(&subtitle));
let replace_map = HashMap::from([
(Scope::Series, &series_map),
(Scope::Season, &season_map),
(Scope::Episode, &episode_map),
(Scope::Stream, &stream_map),
(Scope::Subtitle, &subtitle_map),
]);
output.push(self.replace(replace_map))
}
} }
} }
} }
@ -477,30 +455,20 @@ impl Format {
self.serializable_to_json_map(FormatMovieListing::from(&movie_listing)); self.serializable_to_json_map(FormatMovieListing::from(&movie_listing));
for (movie, streams) in tree { for (movie, streams) in tree {
let movie_map = self.serializable_to_json_map(FormatMovie::from(&movie)); let movie_map = self.serializable_to_json_map(FormatMovie::from(&movie));
for mut stream in streams { for stream in streams {
let stream_map = self.serializable_to_json_map(FormatStream::from(&stream)); let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
if stream.subtitles.is_empty() {
if !self.check_pattern_count_empty(Scope::Subtitle) { output.push(
continue; self.replace_all(
} HashMap::from([
stream (Scope::MovieListing, &movie_listing_map),
.subtitles (Scope::Movie, &movie_map),
.insert(Locale::Custom("".to_string()), Subtitle::default()); (Scope::Stream, &stream_map),
} ]),
for subtitle in self stream,
.filter_options )
.filter_subtitles(stream.subtitles.into_values().collect()) .unwrap_or_default(),
{ )
let subtitle_map =
self.serializable_to_json_map(FormatSubtitle::from(&subtitle));
let replace_map = HashMap::from([
(Scope::MovieListing, &movie_listing_map),
(Scope::Movie, &movie_map),
(Scope::Stream, &stream_map),
(Scope::Subtitle, &subtitle_map),
]);
output.push(self.replace(replace_map))
}
} }
} }
@ -511,100 +479,73 @@ impl Format {
let music_video_empty = self.check_pattern_count_empty(Scope::MusicVideo); let music_video_empty = self.check_pattern_count_empty(Scope::MusicVideo);
let stream_empty = self.check_pattern_count_empty(Scope::Stream); let stream_empty = self.check_pattern_count_empty(Scope::Stream);
let music_video = if !music_video_empty { let music_video = must_match_if_true!(!music_video_empty => media_collection|MediaCollection::MusicVideo(music_video) => music_video.clone()).unwrap_or_default();
match &media_collection { let stream = must_match_if_true!(!stream_empty => media_collection|MediaCollection::MusicVideo(music_video) => music_video.streams().await?).unwrap_or_default();
MediaCollection::MusicVideo(music_video) => music_video.clone(),
_ => panic!(),
}
} else {
MusicVideo::default()
};
let mut stream = if !stream_empty {
match &media_collection {
MediaCollection::MusicVideo(music_video) => music_video.streams().await?,
_ => panic!(),
}
} else {
Stream::default()
};
let mut output = vec![];
let music_video_map = self.serializable_to_json_map(FormatMusicVideo::from(&music_video)); let music_video_map = self.serializable_to_json_map(FormatMusicVideo::from(&music_video));
let stream_map = self.serializable_to_json_map(FormatStream::from(&stream)); let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
if stream.subtitles.is_empty() {
if !self.check_pattern_count_empty(Scope::Subtitle) {
return Ok("".to_string());
}
stream
.subtitles
.insert(Locale::Custom("".to_string()), Subtitle::default());
}
for subtitle in self
.filter_options
.filter_subtitles(stream.subtitles.into_values().collect())
{
let subtitle_map = self.serializable_to_json_map(FormatSubtitle::from(&subtitle));
let replace_map = HashMap::from([
(Scope::MusicVideo, &music_video_map),
(Scope::Stream, &stream_map),
(Scope::Subtitle, &subtitle_map),
]);
output.push(self.replace(replace_map))
}
Ok(output.join("\n")) let output = self
.replace_all(
HashMap::from([
(Scope::MusicVideo, &music_video_map),
(Scope::Stream, &stream_map),
]),
stream,
)
.unwrap_or_default();
Ok(output)
} }
async fn parse_concert(&self, media_collection: MediaCollection) -> Result<String> { async fn parse_concert(&self, media_collection: MediaCollection) -> Result<String> {
let concert_empty = self.check_pattern_count_empty(Scope::Concert); let concert_empty = self.check_pattern_count_empty(Scope::Concert);
let stream_empty = self.check_pattern_count_empty(Scope::Stream); let stream_empty = self.check_pattern_count_empty(Scope::Stream);
let concert = if !concert_empty { let concert = must_match_if_true!(!concert_empty => media_collection|MediaCollection::Concert(concert) => concert.clone()).unwrap_or_default();
match &media_collection { let stream = must_match_if_true!(!stream_empty => media_collection|MediaCollection::Concert(concert) => concert.streams().await?).unwrap_or_default();
MediaCollection::Concert(concert) => concert.clone(),
_ => panic!(),
}
} else {
Concert::default()
};
let mut stream = if !stream_empty {
match &media_collection {
MediaCollection::Concert(concert) => concert.streams().await?,
_ => panic!(),
}
} else {
Stream::default()
};
let mut output = vec![];
let concert_map = self.serializable_to_json_map(FormatConcert::from(&concert)); let concert_map = self.serializable_to_json_map(FormatConcert::from(&concert));
let stream_map = self.serializable_to_json_map(FormatStream::from(&stream)); let stream_map = self.serializable_to_json_map(FormatStream::from(&stream));
let output = self
.replace_all(
HashMap::from([(Scope::Concert, &concert_map), (Scope::Stream, &stream_map)]),
stream,
)
.unwrap_or_default();
Ok(output)
}
fn serializable_to_json_map<S: Serialize>(&self, s: S) -> Map<String, Value> {
serde_json::from_value(serde_json::to_value(s).unwrap()).unwrap()
}
fn replace_all(
&self,
values: HashMap<Scope, &Map<String, Value>>,
mut stream: Stream,
) -> Option<String> {
if stream.subtitles.is_empty() { if stream.subtitles.is_empty() {
if !self.check_pattern_count_empty(Scope::Subtitle) { if !self.check_pattern_count_empty(Scope::Subtitle) {
return Ok("".to_string()); return None;
} }
stream stream
.subtitles .subtitles
.insert(Locale::Custom("".to_string()), Subtitle::default()); .insert(Locale::Custom("".to_string()), Subtitle::default());
} }
for subtitle in self let subtitles = self
.filter_options .filter_options
.filter_subtitles(stream.subtitles.into_values().collect()) .filter_subtitles(stream.subtitles.into_values().collect());
{
let mut output = vec![];
for subtitle in subtitles {
let subtitle_map = self.serializable_to_json_map(FormatSubtitle::from(&subtitle)); let subtitle_map = self.serializable_to_json_map(FormatSubtitle::from(&subtitle));
let replace_map = HashMap::from([ let mut tmp_values = values.clone();
(Scope::MusicVideo, &concert_map), tmp_values.insert(Scope::Subtitle, &subtitle_map);
(Scope::Stream, &stream_map), output.push(self.replace(tmp_values))
(Scope::Subtitle, &subtitle_map),
]);
output.push(self.replace(replace_map))
} }
Ok(output.join("\n")) Some(output.join("\n"))
}
fn serializable_to_json_map<S: Serialize>(&self, s: S) -> Map<String, Value> {
serde_json::from_value(serde_json::to_value(s).unwrap()).unwrap()
} }
fn replace(&self, values: HashMap<Scope, &Map<String, Value>>) -> String { fn replace(&self, values: HashMap<Scope, &Map<String, Value>>) -> String {