From ca7741287b86fe9942878b33955d4928aca813ff Mon Sep 17 00:00:00 2001
From: MrS0m30n3 <ytubedlg@gmail.com>
Date: Mon, 26 Sep 2016 17:50:45 +0300
Subject: [PATCH] Add context menu on wxListCtrl

---
 TODO                        |  1 -
 youtube_dl_gui/mainframe.py | 55 +++++++++++++++++++++++++++++--------
 2 files changed, 44 insertions(+), 12 deletions(-)

diff --git a/TODO b/TODO
index 4984ce3..7c27cbd 100644
--- a/TODO
+++ b/TODO
@@ -1,6 +1,5 @@
 NEW_LAYOUT
 ==========
-* Add context menu on wx.ListCtrl
 * Change download msg: "Total Progress: 0% | Completed (1) Downloading (2) Queued (3)...etc"
 * Create new button widget (groups, auto width)
 * Create new button with two stages (like toggle)
diff --git a/youtube_dl_gui/mainframe.py b/youtube_dl_gui/mainframe.py
index 644fd43..dfb16af 100644
--- a/youtube_dl_gui/mainframe.py
+++ b/youtube_dl_gui/mainframe.py
@@ -218,13 +218,17 @@ class MainFrame(wx.Frame):
 
         # Set the data for the settings menu item
         # label, event_handler
-        menu_data = (
+        settings_menu_data = (
             (self.OPTIONS_LABEL, self._on_options),
             (self.UPDATE_LABEL, self._on_update),
             (self.VIEWLOG_LABEL, self._on_viewlog),
             (self.ABOUT_LABEL, self._on_about)
         )
 
+        statuslist_menu_data = (
+            (_("Get url"), self._on_geturl),
+        )
+
         # Create options frame
         self._options_frame = OptionsFrame(self)
 
@@ -265,18 +269,11 @@ class MainFrame(wx.Frame):
         self._status_bar = self.CreateStatusBar()
 
         # Create extra components
-        self._settings_menu = wx.Menu()
-
-        for item in menu_data:
-            label, evt_handler = item
-            menu_item = self._settings_menu.Append(-1, label)
-
-            self.Bind(wx.EVT_MENU, evt_handler, menu_item)
-
-        # Overwrite the hover event to avoid changing the statusbar
-        self._settings_menu.Bind(wx.EVT_MENU_HIGHLIGHT, lambda event: None)
+        self._settings_menu = self._create_menu_item(settings_menu_data)
+        self._statuslist_menu = self._create_menu_item(statuslist_menu_data)
 
         # Bind extra events
+        self.Bind(wx.EVT_LIST_ITEM_RIGHT_CLICK, self._on_statuslist_right_click, self._status_list)
         self.Bind(wx.EVT_TEXT, self._update_videoformat, self._videoformat_combobox)
         self.Bind(wx.EVT_TEXT, self._update_savepath, self._path_combobox)
         self.Bind(wx.EVT_LIST_ITEM_SELECTED, self._update_pause_button, self._status_list)
@@ -304,6 +301,42 @@ class MainFrame(wx.Frame):
 
         self._set_layout()
 
+    def _create_menu_item(self, items):
+        menu = wx.Menu()
+
+        for item in items:
+            label, evt_handler = item
+            menu_item = menu.Append(-1, label)
+
+            menu.Bind(wx.EVT_MENU, evt_handler, menu_item)
+
+        # Overwrite the hover event to avoid changing the statusbar
+        menu.Bind(wx.EVT_MENU_HIGHLIGHT, lambda event: None)
+
+        return menu
+
+    def _on_statuslist_right_click(self, event):
+        selected = self._status_list.get_selected()
+
+        if selected != -1:
+            self.PopupMenu(self._statuslist_menu)
+
+    def _on_geturl(self, event):
+        selected = self._status_list.get_selected()
+
+        if selected != -1:
+            object_id = self._status_list.GetItemData(selected)
+            download_item = self._download_list.get_item(object_id)
+
+            url = download_item.url
+
+            if not wx.TheClipboard.IsOpened():
+                clipdata = wx.TextDataObject()
+                clipdata.SetText(url)
+                wx.TheClipboard.Open()
+                wx.TheClipboard.SetData(clipdata)
+                wx.TheClipboard.Close()
+
     def _on_timer(self, event):
         msg = self.URL_REPORT_MSG.format(self.download_manager.active())
         self._status_bar_write(msg)