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

feat: get comment replies, rich text formatting

Browse source
This commit is contained in:
ThetaDev 2024-12-18 19:35:31 +01:00
parent 258f18a99d
commit dceba442fe
No known key found for this signature in database
GPG key ID: E319D3C5148D65B6
1 changed files with 77 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

107
cli/src/main.rs
View file

@ -16,8 +16,10 @@ use owo_colors::OwoColorize;
use rustypipe::{ use rustypipe::{
client::{ClientType, RustyPipe}, client::{ClientType, RustyPipe},
model::{ model::{
richtext::ToPlaintext, traits::YtEntity, ArtistId, MusicSearchResult, TrackItem, TrackType, richtext::{RichText, ToPlaintext},
UrlTarget, Verification, YouTubeItem, traits::YtEntity,
ArtistId, Comment, MusicSearchResult, TrackItem, TrackType, UrlTarget, Verification,
YouTubeItem,
}, },
param::{search_filter, ChannelVideoTab, Country, Language, StreamFilter}, param::{search_filter, ChannelVideoTab, Country, Language, StreamFilter},
report::FileReporter, report::FileReporter,
@ -439,6 +441,67 @@ fn print_verification(verification: Verification) {
} }
} }
fn print_comments(comments: &[Comment]) {
print_h2("Comments");
for c in comments {
if let Some(author) = &c.author {
anstream::print!("{} [{}]", author.name.cyan(), author.id);
print_verification(author.verification);
} else {
anstream::print!("{}", "Unknown author".magenta());
}
if c.by_owner {
print!(" (Owner)");
}
println!();
print_richtext(&c.text);
anstream::print!("{} {}", "Likes:".blue(), c.like_count.unwrap_or_default());
if c.hearted {
anstream::print!(" {}", "♥".red());
}
println!();
if let Some(ctoken) = &c.replies.ctoken {
println!("replies:{ctoken}");
}
println!();
}
}
fn print_richtext(text: &RichText) {
for c in &text.0 {
match c {
rustypipe::model::richtext::TextComponent::Text { text, style } => {
if !text.is_empty() {
let mut tstyle = owo_colors::Style::new();
if style.bold {
tstyle = tstyle.bold();
}
if style.italic {
tstyle = tstyle.italic();
}
if style.strikethrough {
tstyle = tstyle.strikethrough();
}
anstream::print!("{}", text.style(tstyle));
}
}
rustypipe::model::richtext::TextComponent::Web { url, .. } => {
anstream::print!("{}", url.underline());
}
rustypipe::model::richtext::TextComponent::YouTube { text, target } => {
if matches!(target, UrlTarget::Channel { .. }) {
anstream::print!("{}", text.cyan());
} else {
anstream::print!("{}", target.to_url().underline());
}
}
_ => {}
}
}
println!();
}
async fn download_video( async fn download_video(
dl: &Downloader, dl: &Downloader,
id: &str, id: &str,
@ -679,6 +742,16 @@ async fn run() -> anyhow::Result<()> {
player, player,
client_type, client_type,
} => { } => {
if let Some(ctoken) = id.strip_prefix("replies:") {
let mut replies = rp.query().video_comments(ctoken, None).await?;
replies.extend_limit(&rp.query(), limit).await?;
match format {
Some(format) => print_data(&replies.items, format, pretty),
None => print_comments(&replies.items),
}
return Ok(());
}
let target = rp.query().resolve_string(&id, false).await?; let target = rp.query().resolve_string(&id, false).await?;
match target { match target {
@ -793,7 +866,7 @@ async fn run() -> anyhow::Result<()> {
if details.is_live { if details.is_live {
anstream::println!("{}", "Livestream".red()); anstream::println!("{}", "Livestream".red());
} }
print_description(Some(details.description.to_plaintext())); print_richtext(&details.description);
if !details.recommended.is_empty() { if !details.recommended.is_empty() {
print_h2("Recommended"); print_h2("Recommended");
print_entities(&details.recommended.items, false); print_entities(&details.recommended.items, false);
@ -803,33 +876,7 @@ async fn run() -> anyhow::Result<()> {
CommentsOrder::Latest => &details.latest_comments.items, CommentsOrder::Latest => &details.latest_comments.items,
}); });
if let Some(comment_list) = comment_list { if let Some(comment_list) = comment_list {
print_h2("Comments"); print_comments(comment_list)
for c in comment_list {
if let Some(author) = &c.author {
anstream::print!(
"{} [{}]",
author.name.cyan(),
author.id
);
print_verification(author.verification);
} else {
anstream::print!("{}", "Unknown author".magenta());
}
if c.by_owner {
print!(" (Owner)");
}
println!();
println!("{}", c.text.to_plaintext());
anstream::print!(
"{} {}",
"Likes:".blue(),
c.like_count.unwrap_or_default()
);
if c.hearted {
anstream::print!(" {}", "♥".red());
}
println!("\n");
}
} }
} }
} }