From a6d82264a5b099ebfe8f3ecf6d1b81b146804231 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Thu, 17 Sep 2020 21:17:35 +0200 Subject: rust: Use new tomsg-rs conveniences --- rust/src/main.rs | 24 ++++++++---------------- 1 file changed, 8 insertions(+), 16 deletions(-) (limited to 'rust') diff --git a/rust/src/main.rs b/rust/src/main.rs index d2377c7..937cc88 100644 --- a/rust/src/main.rs +++ b/rust/src/main.rs @@ -13,15 +13,7 @@ use termion::event::Key; use termion::raw::{IntoRawMode, RawTerminal}; use termion::input::{TermRead, MouseTerminal}; use termion::screen::AlternateScreen; -use tomsg_rs::Connection; -use tomsg_rs::connection; -use tomsg_rs::PushMessage; -use tomsg_rs::Command; -use tomsg_rs::Word; -use tomsg_rs::Line; -use tomsg_rs::Reply; -use tomsg_rs::Message; -use tomsg_rs::Id; +use tomsg_rs::{connection, Command, Connection, Id, Line, Message, PushMessage, Reply, Word}; use unicode_width::UnicodeWidthChar; use crate::editor::Editor; use crate::error::IntoIOError; @@ -111,9 +103,9 @@ impl Default for Layout { impl Layout { fn compute(size: (u16, u16), state: &State) -> Self { - let rlsepx = u16::try_from(state.roomlist.iter().map(|s| s.as_str().len()).max().unwrap_or(5)).unwrap(); + let rlsepx = u16::try_from(state.roomlist.iter().map(|s| s.len()).max().unwrap_or(5)).unwrap(); let room = state.rooms.get(&state.currentroom); - let maxnicklen = room.map(|r| r.members.iter().map(|s| s.as_str().len()).max().unwrap_or(5)).unwrap_or(5); + let maxnicklen = room.map(|r| r.members.iter().map(|s| s.len()).max().unwrap_or(5)).unwrap_or(5); let nlsepx = size.0 - u16::min(size.0, u16::try_from(maxnicklen).unwrap() + 1); Self { rlsepx, nlsepx } } @@ -158,7 +150,7 @@ impl App { layout: Layout::default(), roomlist: Vec::new(), rooms: HashMap::new(), - currentroom: Word::try_from(String::new()).unwrap(), + currentroom: Word::default(), msgs: HashMap::new(), }; App { @@ -202,7 +194,7 @@ impl App { if state.rooms.insert(room.clone(), RoomData::default()).is_none() { state.roomlist.push(room.clone()); - if state.currentroom.as_str().len() == 0 { + if state.currentroom.len() == 0 { state.currentroom = room.clone(); } } @@ -434,7 +426,7 @@ impl State { } } - if self.currentroom.as_str().len() > 0 { + if self.currentroom.len() > 0 { let data = self.rooms.get_mut(&self.currentroom).unwrap(); for (i, user) in data.members.iter().enumerate() { @@ -452,7 +444,7 @@ impl State { let (prefix, text_to_format) = match item { HItem::Message(id) => { let msg = self.msgs.get(&id).unwrap(); - (format!("<{}>", msg.username), msg.message.as_str()) + (format!("<{}>", msg.username), &*msg.message) } HItem::Service(msg) => { @@ -496,7 +488,7 @@ impl State { match &item { HItem::Message(id) => { let msg = self.msgs.get(&id).unwrap(); - data.maxnicklen = data.maxnicklen.max(msg.username.as_str().len() + 2); + data.maxnicklen = data.maxnicklen.max(msg.username.len() + 2); } HItem::Service(_) => { data.maxnicklen = data.maxnicklen.max(2); -- cgit v1.2.3-70-g09d2