aboutsummaryrefslogtreecommitdiffstats
path: root/src/comments
diff options
context:
space:
mode:
authorBenedikt Peetz <benedikt.peetz@b-peetz.de>2024-10-14 14:56:29 +0200
committerBenedikt Peetz <benedikt.peetz@b-peetz.de>2024-10-14 14:56:29 +0200
commit6c9286857ef8b314962b67f4a16a66e8c35531bc (patch)
tree9ced4485ec38b39f82cba258c06321a21c40000a /src/comments
parentbuild(Cargo.toml): Add further lints (diff)
downloadyt-6c9286857ef8b314962b67f4a16a66e8c35531bc.zip
refactor(treewide): Combine the separate crates in one workspace
Diffstat (limited to 'src/comments')
-rw-r--r--src/comments/comment.rs63
-rw-r--r--src/comments/display.rs117
-rw-r--r--src/comments/mod.rs178
3 files changed, 0 insertions, 358 deletions
diff --git a/src/comments/comment.rs b/src/comments/comment.rs
deleted file mode 100644
index 752c510..0000000
--- a/src/comments/comment.rs
+++ /dev/null
@@ -1,63 +0,0 @@
-// yt - A fully featured command line YouTube client
-//
-// Copyright (C) 2024 Benedikt Peetz <benedikt.peetz@b-peetz.de>
-// SPDX-License-Identifier: GPL-3.0-or-later
-//
-// This file is part of Yt.
-//
-// You should have received a copy of the License along with this program.
-// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.
-
-use yt_dlp::wrapper::info_json::Comment;
-
-#[derive(Debug, Clone)]
-pub struct CommentExt {
- pub value: Comment,
- pub replies: Vec<CommentExt>,
-}
-
-#[derive(Debug, Default)]
-pub struct Comments {
- pub(super) vec: Vec<CommentExt>,
-}
-
-impl Comments {
- pub fn new() -> Self {
- Self::default()
- }
- pub fn push(&mut self, value: CommentExt) {
- self.vec.push(value);
- }
- pub fn get_mut(&mut self, key: &str) -> Option<&mut CommentExt> {
- self.vec.iter_mut().filter(|c| c.value.id.id == key).last()
- }
- pub fn insert(&mut self, key: &str, value: CommentExt) {
- let parent = self
- .vec
- .iter_mut()
- .filter(|c| c.value.id.id == key)
- .last()
- .expect("One of these should exist");
- parent.push_reply(value);
- }
-}
-impl CommentExt {
- pub fn push_reply(&mut self, value: CommentExt) {
- self.replies.push(value)
- }
- pub fn get_mut_reply(&mut self, key: &str) -> Option<&mut CommentExt> {
- self.replies
- .iter_mut()
- .filter(|c| c.value.id.id == key)
- .last()
- }
-}
-
-impl From<Comment> for CommentExt {
- fn from(value: Comment) -> Self {
- Self {
- replies: vec![],
- value,
- }
- }
-}
diff --git a/src/comments/display.rs b/src/comments/display.rs
deleted file mode 100644
index 7000063..0000000
--- a/src/comments/display.rs
+++ /dev/null
@@ -1,117 +0,0 @@
-// yt - A fully featured command line YouTube client
-//
-// Copyright (C) 2024 Benedikt Peetz <benedikt.peetz@b-peetz.de>
-// SPDX-License-Identifier: GPL-3.0-or-later
-//
-// This file is part of Yt.
-//
-// You should have received a copy of the License along with this program.
-// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.
-
-use std::fmt::Write;
-
-use chrono::{Local, TimeZone};
-use chrono_humanize::{Accuracy, HumanTime, Tense};
-
-use crate::comments::comment::CommentExt;
-
-use super::comment::Comments;
-
-impl Comments {
- pub fn render(&self, color: bool) -> String {
- self.render_help(color).expect("This should never fail.")
- }
-
- fn render_help(&self, color: bool) -> Result<String, std::fmt::Error> {
- let mut f = String::new();
-
- macro_rules! c {
- ($color_str:expr, $write:ident, $color:expr) => {
- if $color {
- $write.write_str(concat!("\x1b[", $color_str, "m"))?
- }
- };
- }
-
- fn format(
- comment: &CommentExt,
- f: &mut String,
- ident_count: u32,
- color: bool,
- ) -> std::fmt::Result {
- let ident = &(0..ident_count).map(|_| " ").collect::<String>();
- let value = &comment.value;
-
- f.write_str(ident)?;
-
- if value.author_is_uploader {
- c!("91;1", f, color);
- } else {
- c!("35", f, color);
- }
-
- f.write_str(&value.author)?;
- c!("0", f, color);
- if value.edited || value.is_favorited {
- f.write_str("[")?;
- if value.edited {
- f.write_str("")?;
- }
- if value.edited && value.is_favorited {
- f.write_str(" ")?;
- }
- if value.is_favorited {
- f.write_str("")?;
- }
- f.write_str("]")?;
- }
-
- c!("36;1", f, color);
- write!(
- f,
- " {}",
- HumanTime::from(
- Local
- .timestamp_opt(value.timestamp, 0)
- .single()
- .expect("This should be valid")
- )
- .to_text_en(Accuracy::Rough, Tense::Past)
- )?;
- c!("0", f, color);
-
- // c!("31;1", f);
- // f.write_fmt(format_args!(" [{}]", comment.value.like_count))?;
- // c!("0", f);
-
- f.write_str(":\n")?;
- f.write_str(ident)?;
-
- f.write_str(&value.text.replace('\n', &format!("\n{}", ident)))?;
- f.write_str("\n")?;
-
- if !comment.replies.is_empty() {
- let mut children = comment.replies.clone();
- children.sort_by(|a, b| a.value.timestamp.cmp(&b.value.timestamp));
-
- for child in children {
- format(&child, f, ident_count + 4, color)?;
- }
- } else {
- f.write_str("\n")?;
- }
-
- Ok(())
- }
-
- if !&self.vec.is_empty() {
- let mut children = self.vec.clone();
- children.sort_by(|a, b| b.value.like_count.cmp(&a.value.like_count));
-
- for child in children {
- format(&child, &mut f, 0, color)?
- }
- }
- Ok(f)
- }
-}
diff --git a/src/comments/mod.rs b/src/comments/mod.rs
deleted file mode 100644
index 5fbc3fb..0000000
--- a/src/comments/mod.rs
+++ /dev/null
@@ -1,178 +0,0 @@
-// yt - A fully featured command line YouTube client
-//
-// Copyright (C) 2024 Benedikt Peetz <benedikt.peetz@b-peetz.de>
-// SPDX-License-Identifier: GPL-3.0-or-later
-//
-// This file is part of Yt.
-//
-// You should have received a copy of the License along with this program.
-// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.
-
-use std::{
- io::Write,
- mem,
- process::{Command, Stdio},
-};
-
-use anyhow::{bail, Context, Result};
-use comment::{CommentExt, Comments};
-use regex::Regex;
-use yt_dlp::wrapper::info_json::{Comment, InfoJson, Parent};
-
-use crate::{
- app::App,
- storage::video_database::{
- getters::{get_currently_playing_video, get_video_info_json},
- Video,
- },
-};
-
-mod comment;
-mod display;
-
-pub async fn get_comments(app: &App) -> Result<Comments> {
- let currently_playing_video: Video =
- if let Some(video) = get_currently_playing_video(app).await? {
- video
- } else {
- bail!("Could not find a currently playing video!");
- };
-
- let mut info_json: InfoJson = get_video_info_json(&currently_playing_video)
- .await?
- .expect("A currently *playing* must be cached. And thus the info.json should be available");
-
- let base_comments = mem::take(&mut info_json.comments).expect("A video should have comments");
- drop(info_json);
-
- let mut comments = Comments::new();
- base_comments.into_iter().for_each(|c| {
- if let Parent::Id(id) = &c.parent {
- comments.insert(&(id.clone()), CommentExt::from(c));
- } else {
- comments.push(CommentExt::from(c));
- }
- });
-
- comments.vec.iter_mut().for_each(|comment| {
- let replies = mem::take(&mut comment.replies);
- let mut output_replies: Vec<CommentExt> = vec![];
-
- let re = Regex::new(r"\u{200b}?(@[^\t\s]+)\u{200b}?").unwrap();
- for reply in replies {
- if let Some(replyee_match) = re.captures(&reply.value.text){
- let full_match = replyee_match.get(0).expect("This always exists");
- let text = reply.
- value.
- text[0..full_match.start()]
- .to_owned()
- +
- &reply
- .value
- .text[full_match.end()..];
- let text: &str = text.trim().trim_matches('\u{200b}');
-
- let replyee = replyee_match.get(1).expect("This should exist").as_str();
-
-
- if let Some(parent) = output_replies
- .iter_mut()
- // .rev()
- .flat_map(|com| &mut com.replies)
- .flat_map(|com| &mut com.replies)
- .flat_map(|com| &mut com.replies)
- .filter(|com| com.value.author == replyee)
- .last()
- {
- parent.replies.push(CommentExt::from(Comment {
- text: text.to_owned(),
- ..reply.value
- }))
- } else if let Some(parent) = output_replies
- .iter_mut()
- // .rev()
- .flat_map(|com| &mut com.replies)
- .flat_map(|com| &mut com.replies)
- .filter(|com| com.value.author == replyee)
- .last()
- {
- parent.replies.push(CommentExt::from(Comment {
- text: text.to_owned(),
- ..reply.value
- }))
- } else if let Some(parent) = output_replies
- .iter_mut()
- // .rev()
- .flat_map(|com| &mut com.replies)
- .filter(|com| com.value.author == replyee)
- .last()
- {
- parent.replies.push(CommentExt::from(Comment {
- text: text.to_owned(),
- ..reply.value
- }))
- } else if let Some(parent) = output_replies.iter_mut()
- // .rev()
- .filter(|com| com.value.author == replyee)
- .last()
- {
- parent.replies.push(CommentExt::from(Comment {
- text: text.to_owned(),
- ..reply.value
- }))
- } else {
- eprintln!(
- "Failed to find a parent for ('{}') both directly and via replies! The reply text was:\n'{}'\n",
- replyee,
- reply.value.text
- );
- output_replies.push(reply);
- }
- } else {
- output_replies.push(reply);
- }
- }
- comment.replies = output_replies;
- });
-
- Ok(comments)
-}
-
-pub async fn comments(app: &App) -> Result<()> {
- let comments = get_comments(app).await?;
-
- let mut less = Command::new("less")
- .args(["--raw-control-chars"])
- .stdin(Stdio::piped())
- .stderr(Stdio::inherit())
- .spawn()
- .context("Failed to run less")?;
-
- let mut child = Command::new("fmt")
- .args(["--uniform-spacing", "--split-only", "--width=90"])
- .stdin(Stdio::piped())
- .stderr(Stdio::inherit())
- .stdout(less.stdin.take().expect("Should be open"))
- .spawn()
- .context("Failed to run fmt")?;
-
- let mut stdin = child.stdin.take().context("Failed to open stdin")?;
- std::thread::spawn(move || {
- stdin
- .write_all(comments.render(true).as_bytes())
- .expect("Should be able to write to stdin of fmt");
- });
-
- let _ = less.wait().context("Failed to await less")?;
-
- Ok(())
-}
-
-#[cfg(test)]
-mod test {
- #[test]
- fn test_string_replacement() {
- let s = "A \n\nB\n\nC".to_owned();
- assert_eq!("A \n \n B\n \n C", s.replace('\n', "\n "))
- }
-}