From 580f44092358cc260db00ec9871caecac62de361 Mon Sep 17 00:00:00 2001 From: Andy Wortman Date: Tue, 24 Oct 2017 23:27:32 -0700 Subject: add display --- src/display/mod.rs | 186 ++++++++++++++++++++++++++++++++++------------------- 1 file changed, 121 insertions(+), 65 deletions(-) (limited to 'src/display/mod.rs') diff --git a/src/display/mod.rs b/src/display/mod.rs index fdb8448..c011294 100644 --- a/src/display/mod.rs +++ b/src/display/mod.rs @@ -1,6 +1,10 @@ extern crate termion; +use std::io::Write; +use std::io::stdout; + use self::termion::color; +use self::termion::{clear, cursor}; use ::tw; @@ -8,6 +12,7 @@ use ::tw::TweetId; use std; +#[derive(Clone)] pub enum Infos { Tweet(TweetId), Thread(Vec), @@ -16,27 +21,73 @@ pub enum Infos { } pub fn paint(tweeter: &mut ::tw::TwitterCache) { - println!("Painting, totally."); - println!("Statuses:"); - { - let to_show = tweeter.display_info.log.iter().rev().take(4).collect::>().into_iter().rev(); - for line in to_show { - println!("{}", line); - } - } + match termion::terminal_size() { + Ok((width, height)) => { + // draw input prompt + println!("{}{}", cursor::Goto(1, height - 6), clear::CurrentLine); + println!("{}{}>", cursor::Goto(1, height - 5), clear::CurrentLine); + println!("{}{}", cursor::Goto(1, height - 4), clear::CurrentLine); + let mut i = 0; + let log_size = 4; + let last_elem = tweeter.display_info.log.len().saturating_sub(log_size); + { + let to_show = tweeter.display_info.log.drain(last_elem..); + for line in to_show { + println!("{}{}{}", cursor::Goto(1, height - i), clear::CurrentLine, line); + i = i + 1; + } + } + while i < log_size as u16 { + println!("{}{}", cursor::Goto(1, height - i), clear::CurrentLine); + i = i + 1; + } + // draw status lines + // draw tweets + let last_twevent = tweeter.display_info.infos.len().saturating_sub(height as usize - 4); + let last_few_twevent: Vec = tweeter.display_info.infos[last_twevent..].iter().map(|x| x.clone()).rev().collect::>(); - if let Some(elem) = tweeter.display_info.infos.pop() { - match elem { - Infos::Tweet(id) => render_twete(&id, tweeter), - Infos::Thread(ids) => { - println!("I'd show a thread if I knew how"); - }, - Infos::Event(e) => { - e.render(tweeter); - }, - Infos::DM(msg) => { - println!("DM: {}", msg); + let mut h = 7; + for info in last_few_twevent { + let mut to_draw = match info { + Infos::Tweet(id) => { + render_twete(&id, tweeter).iter().map(|x| x.to_owned()).rev().collect() + } + Infos::Thread(ids) => { + vec![format!("{}{}I'd show a thread if I knew how", cursor::Goto(1, height - h), clear::CurrentLine)] + }, + Infos::Event(e) => { + e.clone().render(tweeter).into_iter().rev().collect() + }, + Infos::DM(msg) => { + vec![format!("{}{}DM: {}", cursor::Goto(1, height - h), clear::CurrentLine, msg)] + } + }; + for line in to_draw { + println!("{}{}{}", cursor::Goto(1, height - h), clear::CurrentLine, line); + h = h + 1; + if h >= height { + print!("{}", cursor::Goto(2, height - 6)); + stdout().flush(); + return; + } + } + println!("{}{}", cursor::Goto(1, height - h), clear::CurrentLine); + h = h + 1; + if h >= height { + print!("{}", cursor::Goto(2, height - 6)); + stdout().flush(); + return; + } } + while h < height { + println!("{}{}", cursor::Goto(1, height - h), clear::CurrentLine); + h = h + 1; + } + print!("{}", cursor::Goto(2, height - 6)); + stdout().flush(); + }, + Err(e) => { + println!("Can't get term dimensions: {}", e); } } } @@ -63,14 +114,15 @@ fn color_for(handle: &String) -> termion::color::Fg<&color::Color> { } pub trait Render { - fn render(self, tweeter: &mut ::tw::TwitterCache); + fn render(self, tweeter: &mut ::tw::TwitterCache) -> Vec; } impl Render for tw::events::Event { - fn render(self, tweeter: &mut ::tw::TwitterCache) { + fn render(self, tweeter: &mut ::tw::TwitterCache) -> Vec { + let mut result = Vec::new(); match self { tw::events::Event::Quoted { user_id, twete_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); { let user = tweeter.retrieve_user(&user_id).unwrap(); println!(" quoted_tweet : {} (@{})", user.name, user.handle); @@ -80,75 +132,75 @@ impl Render for tw::events::Event { tw::events::Event::Deleted { user_id, twete_id } => { if let Some(handle) = tweeter.retrieve_user(&user_id).map(|x| &x.handle).map(|x| x.clone()) { if let Some(_tweet) = tweeter.retrieve_tweet(&TweetId::Twitter(twete_id.to_owned())).map(|x| x.clone()) { - println!("-------------DELETED------------------"); - render_twete(&TweetId::Twitter(twete_id), tweeter); - println!("-------------DELETED------------------"); + result.push(format!("-------------DELETED------------------")); + result.extend(render_twete(&TweetId::Twitter(twete_id), tweeter)); + result.push(format!("-------------DELETED------------------")); } else { - println!("dunno what, but do know who: {} - {}", user_id, handle); + result.push(format!("dunno what, but do know who: {} - {}", user_id, handle)); } } else { - println!("delete..."); - println!("dunno who..."); + result.push("delete... dunno who".to_string()); } }, tw::events::Event::RT_RT { user_id, twete_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); { let user = tweeter.retrieve_user(&user_id).unwrap(); - println!(" +rt_rt : {} (@{})", user.name, user.handle); + result.push(format!(" +rt_rt : {} (@{})", user.name, user.handle)); } - render_twete(&TweetId::Twitter(twete_id), tweeter); + result.extend(render_twete(&TweetId::Twitter(twete_id), tweeter)); }, tw::events::Event::Fav_RT { user_id, twete_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); if let Some(user) = tweeter.retrieve_user(&user_id) { - println!(" +rt_fav : {} (@{})", user.name, user.handle); + result.push(format!(" +rt_fav : {} (@{})", user.name, user.handle)); } else { - println!(" +rt_fav but don't know who {} is", user_id); + result.push(format!(" +rt_fav but don't know who {} is", user_id)); } - render_twete(&TweetId::Twitter(twete_id), tweeter); + result.extend(render_twete(&TweetId::Twitter(twete_id), tweeter)); }, tw::events::Event::Fav { user_id, twete_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); { let user = tweeter.retrieve_user(&user_id).unwrap(); - println!("{} +fav : {} (@{}){}", color::Fg(color::Yellow), user.name, user.handle, color::Fg(color::Reset)); + result.push(format!("{} +fav : {} (@{}){}", color::Fg(color::Yellow), user.name, user.handle, color::Fg(color::Reset))); } - render_twete(&TweetId::Twitter(twete_id), tweeter); + result.extend(render_twete(&TweetId::Twitter(twete_id), tweeter)); }, tw::events::Event::Unfav { user_id, twete_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); { let user = tweeter.retrieve_user(&user_id).unwrap(); - println!("{} -fav : {} (@{}){}", color::Fg(color::Yellow), user.name, user.handle, color::Fg(color::Reset)); + result.push(format!("{} -fav : {} (@{}){}", color::Fg(color::Yellow), user.name, user.handle, color::Fg(color::Reset))); } - render_twete(&TweetId::Twitter(twete_id), tweeter); + result.extend(render_twete(&TweetId::Twitter(twete_id), tweeter)); }, tw::events::Event::Followed { user_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); let user = tweeter.retrieve_user(&user_id).unwrap(); - println!(" +fl : {} (@{})", user.name, user.handle); + result.push(format!(" +fl : {} (@{})", user.name, user.handle)); }, tw::events::Event::Unfollowed { user_id } => { - println!("---------------------------------"); + result.push("---------------------------------".to_string()); let user = tweeter.retrieve_user(&user_id).unwrap(); - println!(" -fl : {} (@{})", user.name, user.handle); + result.push(format!(" -fl : {} (@{})", user.name, user.handle)); } /* Blocked(user_id) => { }, */ } - println!(""); + result } } -pub fn render_twete(twete_id: &TweetId, tweeter: &mut tw::TwitterCache) { +pub fn render_twete(twete_id: &TweetId, tweeter: &mut tw::TwitterCache) -> Vec { + let mut result = Vec::new(); let id_color = color::Fg(color::Rgb(180, 80, 40)); let maybe_twete = tweeter.retrieve_tweet(twete_id).map(|x| x.clone()); if maybe_twete.is_none() { - println!("No such tweet: {:?}", twete_id); - return; + result.push(format!("No such tweet: {:?}", twete_id)); + return result; } let twete = maybe_twete.unwrap(); // if we got the tweet, the API gave us the user too @@ -159,7 +211,7 @@ pub fn render_twete(twete_id: &TweetId, tweeter: &mut tw::TwitterCache) { let rt = tweeter.retrieve_tweet(&TweetId::Twitter(rt_id.to_owned())).unwrap().clone(); // and its author let rt_author = tweeter.retrieve_user(&rt.author_id).unwrap().clone(); - println!("{} id:{} (rt_id:{}){}{}", + result.push(format!("{} id:{} (rt_id:{}){}{}", id_color, rt.internal_id, twete.internal_id, rt.reply_to_tweet.clone() .map(|id| tweeter.retrieve_tweet(&TweetId::Twitter(id.to_owned())) @@ -168,16 +220,16 @@ pub fn render_twete(twete_id: &TweetId, tweeter: &mut tw::TwitterCache) { ) .unwrap_or("".to_string()), color::Fg(color::Reset) - ); - println!(" {}{}{} ({}@{}{}) via {}{}{} ({}@{}{}) RT:", + )); + result.push(format!(" {}{}{} ({}@{}{}) via {}{}{} ({}@{}{}) RT:", color_for(&rt_author.handle), rt_author.name, color::Fg(color::Reset), color_for(&rt_author.handle), rt_author.handle, color::Fg(color::Reset), color_for(&user.handle), user.name, color::Fg(color::Reset), color_for(&user.handle), user.handle, color::Fg(color::Reset) - ); + )); } None => { - println!("{} id:{}{}{}", + result.push(format!("{} id:{}{}{}", id_color, twete.internal_id, twete.reply_to_tweet.clone() .map(|id| tweeter.retrieve_tweet(&TweetId::Twitter(id.to_owned())) @@ -186,21 +238,23 @@ pub fn render_twete(twete_id: &TweetId, tweeter: &mut tw::TwitterCache) { ) .unwrap_or("".to_string()), color::Fg(color::Reset) - ); - println!(" {}{}{} ({}@{}{})", + )); + result.push(format!(" {}{}{} ({}@{}{})", color_for(&user.handle), user.name, color::Fg(color::Reset), color_for(&user.handle), user.handle, color::Fg(color::Reset) - ); + )); } } - println!(" {}", twete.text.replace("\r", "\\r").split("\n").collect::>().join("\n ")); + result.extend( + format!(" {}", twete.text.replace("\r", "\\r").split("\n").collect::>().join("\n ")).split("\n").map(|x| x.to_owned()) + ); if let Some(ref qt_id) = twete.quoted_tweet_id { let maybe_qt = tweeter.retrieve_tweet(&TweetId::Twitter(qt_id.to_owned())).map(|x| x.to_owned()); if let Some(qt) = maybe_qt { let qt_author = tweeter.retrieve_user(&qt.author_id).unwrap().clone(); - println!("{} id:{}{}{}", + result.push(format!("{} id:{}{}{}", id_color, qt.internal_id, qt.reply_to_tweet.clone() .map(|id| tweeter.retrieve_tweet(&TweetId::Twitter(id.to_owned())) @@ -209,18 +263,20 @@ pub fn render_twete(twete_id: &TweetId, tweeter: &mut tw::TwitterCache) { ) .unwrap_or("".to_string()), color::Fg(color::Reset) - ); - println!( + )); + result.push(format!( " {}{}{} ({}@{}{})", color_for(&qt_author.handle), qt_author.name, color::Fg(color::Reset), color_for(&qt_author.handle), qt_author.handle, color::Fg(color::Reset) - ); - println!( + )); + result.push(format!( " {}", qt.text.replace("\r", "\\r").split("\n").collect::>().join("\n ") - ); + )); } else { - println!(" << don't have quoted tweet! >>"); + result.push(format!(" << don't have quoted tweet! >>")); } } + + result } -- cgit v1.1