pausing torrents

src/disk.rs

@@ -161,18 +161,14 @@ impl Disk {
         let mut infos: VecDeque<BlockInfo> = VecDeque::new();
         let mut idxs = VecDeque::new();
 
-        let available = self
-            .block_infos
-            .get(&info_hash)
-            .ok_or(Error::TorrentDoesNotExist)?
-            .len();
-
         let block_infos = self
             .block_infos
             .get_mut(&info_hash)
             .ok_or(Error::TorrentDoesNotExist)?;
 
-        info!("available blocks {}", block_infos.len());
+        let available = block_infos.len();
+
+        info!("disk: available blocks {available}");
 
         // prevent from requesting more blocks than what is available
         if available < qnt {

src/error.rs

@@ -1,5 +1,6 @@
 use std::io;
 
+use futures::Sink;
 use thiserror::Error;
 use tokio::sync::{mpsc, oneshot};
 

src/frontend/mod.rs

@@ -28,6 +28,7 @@ use crate::{
     cli::Args,
     config::Config,
     disk::DiskMsg,
+    error::Error,
     torrent::{Stats, Torrent, TorrentMsg, TorrentStatus},
 };
 
@@ -38,6 +39,7 @@ pub struct AppStyle {
     pub highlight_fg: Style,
     pub success: Style,
     pub error: Style,
+    pub warning: Style,
 }
 
 impl Default for AppStyle {
@@ -54,15 +56,17 @@ impl AppStyle {
             highlight_fg: Style::default().fg(Color::LightBlue),
             success: Style::default().fg(Color::LightGreen),
             error: Style::default().fg(Color::Red),
+            warning: Style::default().fg(Color::Yellow),
         }
     }
 }
 
 #[derive(Debug, Clone)]
 pub enum FrMsg {
-    Quit,
-    Draw([u8; 20], TorrentInfo),
     NewTorrent(String),
+    Draw([u8; 20], TorrentInfo),
+    TogglePause([u8; 20]),
+    Quit,
 }
 
 #[derive(Debug, Clone, Default)]
@@ -74,6 +78,7 @@ pub struct TorrentInfo {
     pub download_rate: u64,
     pub uploaded: u64,
     pub size: u64,
+    pub info_hash: [u8; 20],
 }
 
 pub struct Frontend<'a> {
@@ -112,7 +117,7 @@ impl<'a> Frontend<'a> {
     }
 
     /// Run the UI event loop
-    pub async fn run(&mut self, mut fr_rx: mpsc::Receiver<FrMsg>) -> Result<(), std::io::Error> {
+    pub async fn run(&mut self, mut fr_rx: mpsc::Receiver<FrMsg>) -> Result<(), Error> {
         let mut reader = EventStream::new();
 
         // setup terminal
@@ -150,8 +155,6 @@ impl<'a> Frontend<'a> {
                             return Ok(());
                         },
                         FrMsg::Draw(info_hash, torrent_info) => {
-                            info!("draw {torrent_info:#?}");
-
                             self.torrent_list
                                 .torrent_infos
                                 .insert(info_hash, torrent_info);
@@ -161,6 +164,10 @@ impl<'a> Frontend<'a> {
                         FrMsg::NewTorrent(magnet) => {
                             self.new_torrent(&magnet).await;
                         }
+                        FrMsg::TogglePause(id) => {
+                            let tx = self.torrent_txs.get(&id).ok_or(Error::TorrentDoesNotExist)?;
+                            tx.send(TorrentMsg::TogglePause).await?;
+                        }
                     }
                 }
             }
@@ -204,9 +211,9 @@ impl<'a> Frontend<'a> {
                 .torrent_infos
                 .insert(info_hash, torrent_info_l);
 
-            if self.torrent_list.state.selected().is_none() {
-                self.torrent_list.state.select(Some(0));
-            }
+            // if self.torrent_list.state.selected().is_none() {
+            //     self.torrent_list.state.select(Some(0));
+            // }
 
             let args = Args::parse();
             let mut listen = self.config.listen;

src/frontend/torrent_list.rs

@@ -10,6 +10,7 @@ use ratatui::{
     widgets::{Block, Borders, Clear, List, ListItem, ListState, Paragraph},
     Terminal,
 };
+use tracing::info;
 
 use crate::{to_human_readable, torrent::TorrentStatus};
 
@@ -20,6 +21,7 @@ pub struct TorrentList<'a> {
     pub style: AppStyle,
     pub state: ListState,
     pub torrent_infos: HashMap<[u8; 20], TorrentInfo>,
+    active_torrent: Option<[u8; 20]>,
     ctx: Arc<FrontendCtx>,
     show_popup: bool,
     input: String,
@@ -39,6 +41,8 @@ impl<'a> TorrentList<'a> {
             " move down ".into(),
             Span::styled("t".to_string(), style.highlight_fg),
             " add torrent ".into(),
+            Span::styled("p".to_string(), style.highlight_fg),
+            " toggle pause/resume ".into(),
             Span::styled("q".to_string(), style.highlight_fg),
             " quit".into(),
         ]
@@ -51,6 +55,7 @@ impl<'a> TorrentList<'a> {
             .block(Block::default().borders(Borders::ALL).title("Keybindings"));
 
         Self {
+            active_torrent: None,
             style,
             footer,
             cursor_position: 0,
@@ -108,6 +113,17 @@ impl<'a> TorrentList<'a> {
                     self.show_popup = true;
                     self.draw(terminal).await;
                 }
+                KeyCode::Char('p') => {
+                    info!("sending pause with id {:?}", self.active_torrent);
+                    if let Some(active_torrent) = self.active_torrent {
+                        let _ = self
+                            .ctx
+                            .fr_tx
+                            .send(FrMsg::TogglePause(active_torrent))
+                            .await;
+                        self.draw(terminal).await;
+                    }
+                }
                 _ => {}
             },
         }
@@ -126,6 +142,7 @@ impl<'a> TorrentList<'a> {
             let status_style = match ctx.status {
                 TorrentStatus::Seeding => self.style.success,
                 TorrentStatus::Error => self.style.error,
+                TorrentStatus::Paused => self.style.warning,
                 _ => self.style.highlight_fg,
             };
 
@@ -162,12 +179,13 @@ impl<'a> TorrentList<'a> {
                 line_bottom,
             ];
 
+            if Some(i) == selected {
+                self.active_torrent = Some(ctx.info_hash);
+            }
+
             if Some(i) != selected && selected > Some(0) {
                 items.remove(0);
             }
-            if Some(i) != selected && selected == Some(0) {
-                items.pop();
-            }
 
             rows.push(ListItem::new(items));
         }