Sulkta-Coop/rustypipe
Archived
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add CLI commands to fetch user library and YTM releases/charts

Browse source
This commit is contained in:
ThetaDev 2025-01-05 04:42:30 +01:00
parent 27f64fc412
commit a1b43ad70a
No known key found for this signature in database
GPG key ID: E319D3C5148D65B6
1 changed files with 215 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

215
cli/src/main.rs
View file

@ -219,6 +219,86 @@ enum Commands {
#[clap(long)]
search: Option<String>,
},
/// Get the channels you subscribed to
Subscriptions {
/// Output format
#[clap(short, long, value_parser)]
format: Option<Format>,
/// Pretty-print output
#[clap(long)]
pretty: bool,
/// Limit the number of items to fetch
#[clap(short, long, default_value_t = 20)]
limit: usize,
/// Use YouTube Music
#[clap(short, long)]
music: bool,
/// Output YouTube subscription feed
#[clap(long)]
feed: bool,
},
/// Get the playlists from your library
Playlists {
/// Output format
#[clap(short, long, value_parser)]
format: Option<Format>,
/// Pretty-print output
#[clap(long)]
pretty: bool,
/// Limit the number of items to fetch
#[clap(short, long, default_value_t = 20)]
limit: usize,
/// Use YouTube Music
#[clap(short, long)]
music: bool,
},
/// Get the albums from your library
Albums {
/// Output format
#[clap(short, long, value_parser)]
format: Option<Format>,
/// Pretty-print output
#[clap(long)]
pretty: bool,
/// Limit the number of items to fetch
#[clap(short, long, default_value_t = 20)]
limit: usize,
},
/// Get the tracks from your library
Tracks {
/// Output format
#[clap(short, long, value_parser)]
format: Option<Format>,
/// Pretty-print output
#[clap(long)]
pretty: bool,
/// Limit the number of items to fetch
#[clap(short, long, default_value_t = 20)]
limit: usize,
},
/// Get the latest music releases
Releases {
/// Get latest music videos
#[clap(long)]
videos: bool,
/// Output format
#[clap(short, long, value_parser)]
format: Option<Format>,
/// Pretty-print output
#[clap(long)]
pretty: bool,
},
/// Get YouTube music charts
Charts {
/// Chart country
country: Option<String>,
/// Output format
#[clap(short, long, value_parser)]
format: Option<Format>,
/// Pretty-print output
#[clap(long)]
pretty: bool,
},
/// Get a YouTube visitor data cookie
Vdata,
/// Log in using your Google account
@ -409,6 +489,31 @@ fn print_entity(e: &impl YtEntity, with_type: bool) {
println!();
}
fn fmt_print_entities<T: YtEntity + Serialize>(
items: &[T],
format: Option<Format>,
pretty: bool,
title: &str,
) {
match format {
Some(format) => print_data(&items, format, pretty),
None => {
print_h1(title);
print_entities(items, false);
}
}
}
fn fmt_print_tracks(tracks: &[TrackItem], format: Option<Format>, pretty: bool, title: &str) {
match format {
Some(format) => print_data(&tracks, format, pretty),
None => {
print_h1(title);
print_tracks(tracks);
}
}
}
fn print_tracks(tracks: &[TrackItem]) {
for t in tracks {
if let Some(n) = t.track_nr {
@ -455,6 +560,7 @@ fn print_music_search<T: Serialize + YtEntity>(
match format {
Some(format) => print_data(data, format, pretty),
None => {
print_h1("Music search");
if let Some(corr) = &data.corrected_query {
anstream::println!("Did you mean `{}`?", corr.magenta());
}
@ -472,6 +578,10 @@ fn print_description(desc: Option<String>) {
}
}
fn print_h1(title: &str) {
anstream::println!("{}", format!("{title}:").on_green().black());
}
fn print_h2(title: &str) {
anstream::println!("\n{}", format!("{title}:").green().underline());
}
@ -1276,6 +1386,7 @@ async fn run() -> anyhow::Result<()> {
match format {
Some(format) => print_data(&res, format, pretty),
None => {
print_h1("Search");
if let Some(corr) = res.corrected_query {
anstream::println!("Did you mean `{}`?", corr.magenta());
}
@ -1386,6 +1497,110 @@ async fn run() -> anyhow::Result<()> {
}
}
}
Commands::Subscriptions {
format,
pretty,
limit,
music,
feed,
} => {
if music {
let mut subscriptions = rp.query().music_saved_artists().await?;
subscriptions.extend_limit(rp.query(), limit).await?;
fmt_print_entities(&subscriptions.items, format, pretty, "Music artists");
} else if feed {
let mut feed = rp.query().subscription_feed().await?;
feed.extend_limit(rp.query(), limit).await?;
fmt_print_entities(&feed.items, format, pretty, "Feed");
} else {
let mut subscriptions = rp.query().subscriptions().await?;
subscriptions.extend_limit(rp.query(), limit).await?;
fmt_print_entities(&subscriptions.items, format, pretty, "Subscriptions");
}
}
Commands::Playlists {
format,
pretty,
limit,
music,
} => {
if music {
let mut playlists = rp.query().music_saved_playlists().await?;
playlists.extend_limit(rp.query(), limit).await?;
fmt_print_entities(&playlists.items, format, pretty, "Music playlists");
} else {
let mut playlists = rp.query().saved_playlists().await?;
playlists.extend_limit(rp.query(), limit).await?;
fmt_print_entities(&playlists.items, format, pretty, "Saved playlists");
}
}
Commands::Albums {
format,
pretty,
limit,
} => {
let mut albums = rp.query().music_saved_albums().await?;
albums.extend_limit(rp.query(), limit).await?;
fmt_print_entities(&albums.items, format, pretty, "Saved albums");
}
Commands::Tracks {
format,
pretty,
limit,
} => {
let mut tracks = rp.query().music_saved_tracks().await?;
tracks.extend_limit(rp.query(), limit).await?;
fmt_print_tracks(&tracks.items, format, pretty, "Saved tracks");
}
Commands::Releases {
videos,
format,
pretty,
} => {
if videos {
let releases = rp.query().music_new_videos().await?;
fmt_print_tracks(&releases, format, pretty, "New music videos");
} else {
let releases = rp.query().music_new_albums().await?;
fmt_print_entities(&releases, format, pretty, "New albums");
}
}
Commands::Charts {
country,
format,
pretty,
} => {
let country = match country {
Some(c) => Some(Country::from_str(&c)?),
None => None,
};
let charts = rp.query().music_charts(country).await?;
match format {
Some(format) => print_data(&charts, format, pretty),
None => {
print_h1("Music charts");
if let Some(plid) = &charts.top_playlist_id {
print_h2(&format!("Top tracks [{plid}]"));
} else {
print_h2("Top tracks");
}
print_tracks(&charts.top_tracks);
if let Some(plid) = &charts.trending_playlist_id {
print_h2(&format!("Trending [{plid}]"));
} else {
print_h2("Trending");
}
print_tracks(&charts.trending_tracks);
print_h2("Artists");
print_entities(&charts.artists, false);
if !charts.playlists.is_empty() {
print_h2("Playlists");
print_entities(&charts.playlists, false);
}
}
}
}
Commands::Vdata => {
let vd = rp.query().get_visitor_data().await?;
println!("{vd}");