Add option to force recheck torrents (#522)

picotorrent · Feb 25, 2018 · 6272c2f · 6272c2f
1 parent b92da0b
commit 6272c2f
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 2 deletions.
diff --git a/lang/1033.json b/lang/1033.json
@@ -203,6 +203,7 @@
         "invalid_regex": "Invalid regular expression",
         "info_hash": "Info hash",
         "fails": "Fails",
-        "verified": "Verified"
+        "verified": "Verified",
+        "force_recheck": "Force recheck"
     }
 }
diff --git a/src/picotorrent/mainframe.cpp b/src/picotorrent/mainframe.cpp
@@ -370,6 +370,23 @@ void MainFrame::OnSessionAlert()
 
             break;
         }
+        case lt::torrent_checked_alert::alert_type:
+        {
+            lt::torrent_checked_alert* tca = lt::alert_cast<lt::torrent_checked_alert>(alert);
+
+            auto it = std::find(
+                m_state->pause_after_checking.begin(),
+                m_state->pause_after_checking.end(),
+                tca->handle.info_hash());
+
+            if (it != m_state->pause_after_checking.end())
+            {
+                tca->handle.pause();
+                m_state->pause_after_checking.erase(it);
+            }
+
+            break;
+        }
         case lt::torrent_removed_alert::alert_type:
         {
             lt::torrent_removed_alert* tra = lt::alert_cast<lt::torrent_removed_alert>(alert);

diff --git a/src/picotorrent/sessionstate.hpp b/src/picotorrent/sessionstate.hpp
@@ -19,6 +19,7 @@ namespace pt
         bool IsSelected(libtorrent::sha1_hash const& hash);
 
         std::vector<libtorrent::sha1_hash> loaded_torrents;
+        std::vector<libtorrent::sha1_hash> pause_after_checking;
         std::vector<libtorrent::torrent_handle> selected_torrents;
         std::unique_ptr<libtorrent::session> session;
         std::map<libtorrent::sha1_hash, libtorrent::torrent_handle> torrents;

diff --git a/src/picotorrent/torrentcontextmenu.cpp b/src/picotorrent/torrentcontextmenu.cpp
@@ -28,6 +28,7 @@ BEGIN_EVENT_TABLE(TorrentContextMenu, wxMenu)
     EVT_MENU(ptID_REMOVE, TorrentContextMenu::Remove)
     EVT_MENU(ptID_COPY_INFO_HASH, TorrentContextMenu::CopyInfoHash)
     EVT_MENU(ptID_OPEN_IN_EXPLORER, TorrentContextMenu::OpenInExplorer)
+    EVT_MENU(ptID_FORCE_RECHECK, TorrentContextMenu::ForceRecheck)
 END_EVENT_TABLE()
 
 TorrentContextMenu::TorrentContextMenu(
@@ -79,6 +80,7 @@ TorrentContextMenu::TorrentContextMenu(
     }
 
     AppendSeparator();
+    Append(ptID_FORCE_RECHECK, i18n(tr, "force_recheck"));
     Append(ptID_MOVE, i18n(tr, "move"));
     Append(ptID_REMOVE, i18n(tr, "remove"));
     AppendSeparator();
@@ -104,6 +106,25 @@ void TorrentContextMenu::CopyInfoHash(wxCommandEvent& WXUNUSED(event))
     }
 }
 
+void TorrentContextMenu::ForceRecheck(wxCommandEvent& WXUNUSED(event))
+{
+    for (lt::torrent_handle& th : m_state->selected_torrents)
+    {
+        th.force_recheck();
+
+        lt::torrent_status ts = th.status();
+
+        bool paused = ((ts.flags & lt::torrent_flags::paused)
+            && !(ts.flags & lt::torrent_flags::auto_managed));
+
+        if (paused)
+        {
+            th.resume();
+            m_state->pause_after_checking.push_back(ts.info_hash);
+        }
+    }
+}
+
 void TorrentContextMenu::Move(wxCommandEvent& WXUNUSED(event))
 {
     wxDirDialog dlg(

diff --git a/src/picotorrent/torrentcontextmenu.hpp b/src/picotorrent/torrentcontextmenu.hpp
@@ -31,12 +31,14 @@ namespace pt
             ptID_QUEUE_TOP,
             ptID_QUEUE_BOTTOM,
             ptID_COPY_INFO_HASH,
-            ptID_OPEN_IN_EXPLORER
+            ptID_OPEN_IN_EXPLORER,
+            ptID_FORCE_RECHECK
         };
 
         wxDECLARE_EVENT_TABLE();
 
         void CopyInfoHash(wxCommandEvent&);
+        void ForceRecheck(wxCommandEvent&);
         void Move(wxCommandEvent&);
         void OpenInExplorer(wxCommandEvent&);
         void Pause(wxCommandEvent&);