Compare commits

...

2 commits

Author SHA1 Message Date
Joe Ardent
95fd86e851 dry out the input handling 2025-08-05 16:17:56 -07:00
Joe Ardent
95e24d14b3 add keybinds for sending text 2025-08-05 15:18:42 -07:00
2 changed files with 131 additions and 113 deletions

View file

@ -102,51 +102,66 @@ impl App {
Ok(())
}
pub async fn handle_key_event(&mut self, key_event: KeyEvent, event: crossterm::event::Event) {
match self.screen.last_mut().unwrap() {
CurrentScreen::Logging => match key_event.code {
async fn handle_key_event(&mut self, key_event: KeyEvent, event: crossterm::event::Event) {
let code = key_event.code;
let mode = self.screen.last_mut().unwrap();
match mode {
CurrentScreen::Main
| CurrentScreen::Logging
| CurrentScreen::Receiving
| CurrentScreen::Sending(SendingScreen::Files)
| CurrentScreen::Sending(SendingScreen::Peers) => match code {
KeyCode::Esc => self.pop(),
KeyCode::Char('q') => self.exit(),
KeyCode::Char('s') => self.send(),
KeyCode::Char('r') => self.recv(),
KeyCode::Char('l') => self.logs(),
_ => match mode {
CurrentScreen::Logging => match code {
KeyCode::Left => change_log_level(-1),
KeyCode::Right => change_log_level(1),
KeyCode::Char('q') => self.exit(),
_ => {}
},
CurrentScreen::Receiving => match key_event.code {
CurrentScreen::Receiving => match code {
KeyCode::Up => self.receiving_state.select_previous(),
KeyCode::Down => self.receiving_state.select_next(),
KeyCode::Char('a') => self.accept(),
KeyCode::Char('d') => self.deny(),
KeyCode::Esc => self.pop(),
KeyCode::Char('q') => self.exit(),
_ => {}
},
CurrentScreen::Sending(s) => match s {
SendingScreen::Files => match key_event.code {
KeyCode::Esc => self.pop(),
KeyCode::Char('q') => self.exit(),
KeyCode::Tab => *s = SendingScreen::Peers,
KeyCode::Enter => self.send_content().await,
CurrentScreen::Sending(sending_screen) => match sending_screen {
SendingScreen::Files => match code {
KeyCode::Char('t') => *sending_screen = SendingScreen::Text,
KeyCode::Tab => *sending_screen = SendingScreen::Peers,
KeyCode::Enter => self.chdir_or_send_file().await,
_ => self.file_picker.handle(&event).unwrap_or_default(),
},
SendingScreen::Peers => match key_event.code {
KeyCode::Esc => self.pop(),
KeyCode::Char('q') => self.exit(),
KeyCode::Tab => *s = SendingScreen::Files,
SendingScreen::Peers => match code {
KeyCode::Tab => *sending_screen = SendingScreen::Files,
KeyCode::Enter => self.send_content().await,
KeyCode::Up => self.peer_state.select_previous(),
KeyCode::Down => self.peer_state.select_next(),
_ => {}
},
SendingScreen::Text => {}
_ => unreachable!(),
},
_ => match key_event.code {
KeyCode::Char('q') => self.exit(),
KeyCode::Char('s') => self.send(),
KeyCode::Char('r') => self.recv(),
KeyCode::Char('l') => self.logs(),
KeyCode::Esc => self.pop(),
_ => {}
},
},
CurrentScreen::Sending(sending_screen) => match sending_screen {
SendingScreen::Text => match code {
KeyCode::Tab => *sending_screen = SendingScreen::Peers,
KeyCode::Enter => self.send_text().await,
KeyCode::Esc => {
self.text = None;
*sending_screen = SendingScreen::Files;
}
_ => { /* todo: add input widget to handle key input here */ }
},
_ => unreachable!(),
},
CurrentScreen::Stopping => {}
}
}
@ -230,9 +245,14 @@ impl App {
self.receive_requests.remove(key);
}
// send content to selected peer, or change directories in the file explorer
async fn send_content(&mut self) {
debug!("sending content");
async fn chdir_or_send_file(&mut self) {
let file = self.file_picker.current().path().clone();
if file.is_dir()
&& let Err(e) = self.file_picker.set_cwd(&file)
{
error!("could not list directory {file:?}: {e}");
return;
}
let Some(peer_idx) = self.peer_state.selected() else {
warn!("no peer selected to send to");
@ -243,30 +263,43 @@ impl App {
return;
};
if let Some(ref text) = self.text {
if let Err(e) = self.service.send_text(&peer.fingerprint, text).await {
error!("got error sending \"{text}\" to {}: {e:?}", peer.alias);
}
} else {
let file = self.file_picker.current().path().clone();
if file.is_dir()
&& let Err(e) = self.file_picker.set_cwd(&file)
{
error!("could not list directory {file:?}: {e}");
return;
}
if file.is_file() {
debug!("sending {file:?}");
if let Err(e) = self
.service
.send_file(&peer.fingerprint, file.clone())
.await
{
if let Err(e) = self.service.send_file(&peer.fingerprint, file).await {
error!("got error sending content: {e:?}");
}
}
}
// send content to selected peer, or change directories in the file explorer
async fn send_text(&mut self) {
debug!("sending text");
let Some(peer_idx) = self.peer_state.selected() else {
debug!("no peer selected to send to");
return;
};
let Some(peer) = self.peers.get(peer_idx) else {
warn!("invalid peer index {peer_idx}");
return;
};
let Some(text) = &self.text else {
debug!("no text to send");
return;
};
if let Err(e) = self.service.send_text(&peer.fingerprint, text).await {
error!("got error sending \"{text}\" to {}: {e:?}", peer.alias);
}
}
async fn send_content(&mut self) {
if self.text.is_some() {
self.send_text().await;
} else {
self.chdir_or_send_file().await;
}
}
}

View file

@ -128,12 +128,11 @@ impl Widget for &mut App {
let rx_reqs: Vec<_> = self.receive_requests.values().collect();
outer_frame(*current_screen, &MAIN_MENU, area, buf);
logger(header_right.inner(header_margin), buf);
let peers = PeersWidget { peers: &self.peers };
ratatui::widgets::StatefulWidget::render(
peers,
peers(
&self.peers,
&mut self.peer_state,
footer_right.inner(footer_margin),
buf,
&mut self.peer_state,
);
NetworkInfoWidget.render(footer_left.inner(footer_margin), buf);
receive_requests(
@ -169,18 +168,16 @@ impl Widget for &mut App {
SendingScreen::Text => {}
}
let peers = PeersWidget { peers: &self.peers };
self.file_picker
.widget()
.render(header_left.inner(header_margin), buf);
logger(header_right.inner(header_margin), buf);
ratatui::widgets::StatefulWidget::render(
peers,
peers(
&self.peers,
&mut self.peer_state,
bottom.inner(subscreen_margin),
buf,
&mut self.peer_state,
);
}
_ => {
@ -272,31 +269,20 @@ fn receive_requests(
ratatui::widgets::StatefulWidget::render(table, area, buf, state);
}
#[derive(Debug, Clone)]
pub struct PeersWidget<'p> {
pub peers: &'p [Peer],
}
impl<'p> ratatui::widgets::StatefulWidget for PeersWidget<'p> {
type State = ListState;
fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State)
where
Self: Sized,
{
if self.peers.is_empty() {
fn peers(peers: &[Peer], state: &mut ListState, area: Rect, buf: &mut Buffer) {
if peers.is_empty() {
state.select(None);
}
if state.selected().is_none() {
state.select(Some(0));
}
let mut items = Vec::with_capacity(self.peers.len());
let mut items = Vec::with_capacity(peers.len());
for Peer {
addr,
alias,
fingerprint,
} in self.peers.iter()
} in peers
{
let item = format!("{:?}: {} ({})", addr, alias, fingerprint);
let s = Line::from(item.yellow());
@ -312,7 +298,6 @@ impl<'p> ratatui::widgets::StatefulWidget for PeersWidget<'p> {
.block(block)
.highlight_style(Style::new().bg(Color::Rgb(99, 99, 99)));
ratatui::widgets::StatefulWidget::render(list, area, buf, state);
}
}
#[derive(Debug, Clone)]
@ -364,7 +349,7 @@ impl Widget for NetworkInfoWidget {
}
// helpers
fn centered_rect(area: Rect, width_pct: u16, height_pct: u16) -> Rect {
fn _centered_rect(area: Rect, width_pct: u16, height_pct: u16) -> Rect {
let horizontal = Layout::horizontal([Constraint::Percentage(width_pct)]).flex(Flex::Center);
let vertical = Layout::vertical([Constraint::Percentage(height_pct)]).flex(Flex::Center);
let [area] = vertical.areas(area);