wip - content copy is working, header representation is not

1 files changed, 16 insertions, 0 deletions

diff --git a/libmproxy/console/flowlist.py b/libmproxy/console/flowlist.py
index c896ba33..89dc56a9 100644
--- a/libmproxy/console/flowlist.py
+++ b/libmproxy/console/flowlist.py
@@ -1,5 +1,6 @@
 from __future__ import absolute_import
 import urwid
+import pyperclip
 from . import common
 
 def _mkhelp():
@@ -136,6 +137,11 @@ class ConnectionItem(common.WWrap):
                 self.state.last_saveload,
                 self.master.server_playback_path
             )
+    def server_copy_response(self, k):
+        if k == "c":
+            pyperclip.copy(self.flow.response_content())
+        elif k == "t":
+            pyperclip.copy(self.flow.response_headers())
 
     def keypress(self, (maxcol,), key):
         key = common.shortcuts(key)
@@ -202,6 +208,16 @@ class ConnectionItem(common.WWrap):
                 self.master.run_script_once,
                 self.flow
             )
+        elif key == "g":
+            #copy flow part
+            self.master.prompt_onekey(
+                "Copy Response",
+                (
+                    ("content", "c"),
+                    ("headers", "h"),
+                ),
+                self.server_copy_response,
+            )
         else:
             return key