From 56d2f9fbdbbd5992a8a1af9383936a49ef0fa8e3 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Tue, 7 Feb 2012 12:07:18 +1300
Subject: [PATCH] Restore header edit functionality.

---
 libmproxy/console.py | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/libmproxy/console.py b/libmproxy/console.py
index 8be74352d..9f24c3ab1 100644
--- a/libmproxy/console.py
+++ b/libmproxy/console.py
@@ -283,7 +283,7 @@ class KVEditor(WWrap):
                     [
                         (
                             "fixed",
-                            maxk + 2, 
+                            maxk + 2,
                             urwid.AttrWrap(urwid.Edit(edit_text=k, wrap="any"), "editfield"),
                         ),
                         urwid.AttrWrap(urwid.Edit(edit_text=v, wrap="any"), "editfield"),
@@ -520,12 +520,11 @@ class ConnectionView(WWrap):
             c = self._spawn_editor(conn.content or "")
             conn.content = c.rstrip("\n")
         elif part == "h":
-            self.master.view_kveditor("Editing headers", conn.headers.lst, None)
-            #headertext = self._spawn_editor(repr(conn.headers))
-            #headers = flow.Headers()
-            #fp = cStringIO.StringIO(headertext)
-            #headers.read(fp)
-            #conn.headers = headers
+            headertext = self._spawn_editor(repr(conn.headers))
+            headers = flow.Headers()
+            fp = cStringIO.StringIO(headertext)
+            headers.read(fp)
+            conn.headers = headers
         elif part == "u" and self.state.view_flow_mode == VIEW_FLOW_REQUEST:
             self.master.prompt_edit("URL", conn.get_url(), self.set_url)
         elif part == "m" and self.state.view_flow_mode == VIEW_FLOW_REQUEST: