diff options
Diffstat (limited to 'src/display')
| -rw-r--r-- | src/display/mod.rs | 50 |
1 files changed, 47 insertions, 3 deletions
diff --git a/src/display/mod.rs b/src/display/mod.rs index 0cc1050..3e9d725 100644 --- a/src/display/mod.rs +++ b/src/display/mod.rs @@ -686,12 +686,56 @@ pub fn render_twete_no_recurse(twete_id: &TweetId, tweeter: &tw::TwitterCache, d result.push(id_string); result.push(author_string); - let mut original_lines: Vec<String> = twete.text -// .replace("\r", "\\r") + let raw_lines: Vec<String> = twete.text .split("\n").map(|line| line.replace("\r", "\\r")).collect(); + // now colorize @'s: + let mut colorized_lines: Vec<String> = vec![]; + + for line in raw_lines { + let mut name: Option<String> = None; + let mut new_line = String::new(); + for c in line.chars() { + name = match name { + Some(mut handle) => { + match c { + 'a'...'z' | 'A'...'Z' | '0'...'9' | '_' => { + // so if we have a handle WIP, append to that string. + handle.push(c); + Some(handle) + }, + c => { + // we HAD a handle, this just terminated it. + // if it was empty string, it's not really a mention, we can + // discard it. + if handle.len() > 0 { + new_line.push_str(&format!("{}@{}{}{}", color_for(&handle), &handle, termion::style::Reset, c)); + } else { + new_line.push('@'); + new_line.push(c); + } + None + } + } + }, + None => { + if c == '@' { + Some(String::new()) + } else { + new_line.push(c); + None + } + } + } + } + if let Some(mut handle) = name { + new_line.push_str(&format!("{}@{}{}", color_for(&handle), &handle, termion::style::Reset)); + } + colorized_lines.push(new_line); + } + let mut urls_to_include: Vec<&str> = vec![]; - let urls_replaced = original_lines + let urls_replaced = colorized_lines .into_iter() .map(|line| { let mut result: String = line.to_owned(); |