From 64360f5996ba1a9fa5dd596599d511257da8f6a4 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Wed, 8 Feb 2012 14:58:48 +1300
Subject: [PATCH] Editing now works.

---
 libmproxy/console/kveditor.py | 65 +++++++++++++++++++++--------------
 1 file changed, 40 insertions(+), 25 deletions(-)

diff --git a/libmproxy/console/kveditor.py b/libmproxy/console/kveditor.py
index d9aaadce6..6d708de0a 100644
--- a/libmproxy/console/kveditor.py
+++ b/libmproxy/console/kveditor.py
@@ -9,6 +9,9 @@ class SText(common.WWrap):
             w = urwid.AttrWrap(w, "editfield")
         common.WWrap.__init__(self, w)
 
+    def get_text(self):
+        return self.w.get_text()[0]
+
     def keypress(self, size, key):
         return key
 
@@ -18,9 +21,12 @@ class SText(common.WWrap):
 
 class SEdit(common.WWrap):
     def __init__(self, txt):
-        w = urwid.Edit(txt, wrap="any")
+        w = urwid.Edit(txt, wrap="any", multiline=True)
         common.WWrap.__init__(self, w)
 
+    def get_text(self):
+        return self.w.get_text()[0]
+
     def selectable(self):
         return True
 
@@ -29,12 +35,12 @@ class KVItem(common.WWrap):
     def __init__(self, focused, editing, maxk, k, v):
         self.focused, self.editing = focused, editing
         if focused == 0 and editing:
-            self.kf = SEdit(k)
+            self.editing = self.kf = SEdit(k)
         else:
             self.kf = SText(k, True if focused == 0 else False)
 
         if focused == 1 and editing:
-            self.vf = SEdit(v)
+            self.editing = self.vf = SEdit(v)
         else:
             self.vf = SText(v, True if focused == 1 else False)
 
@@ -49,12 +55,12 @@ class KVItem(common.WWrap):
             w.set_focus_column(focused)
         common.WWrap.__init__(self, w)
 
+    def get_kv(self):
+        return (self.kf.get_text(), self.vf.get_text())
+
     def keypress(self, s, k):
         if self.editing:
-            if self.focused == 0:
-                return self.kf.keypress(s, k)
-            else:
-                return self.vf.keypress(s, k)
+            k = self.editing.keypress(s, k)
         return k
 
     def selectable(self):
@@ -69,10 +75,15 @@ class KVWalker(urwid.ListWalker):
         self.focus_col = 0
         self.editing = False
 
-    def edit(self):
+    def start_edit(self):
         self.editing = KVItem(self.focus_col, True, self.maxk, *self.lst[self.focus])
         self._modified()
 
+    def stop_edit(self):
+        self.lst[self.focus] = self.editing.get_kv()
+        self.editing = False
+        self._modified()
+
     def left(self):
         self.focus_col = 0
         self._modified()
@@ -127,21 +138,25 @@ class KVEditor(common.WWrap):
 
     def keypress(self, size, key):
         if self.walker.editing:
-            self.w.keypress(size, key)
-            return None
-        else:
-            key = common.shortcuts(key)
-            if key == "q":
-                self.master.pop_view()
-            elif key == "h":
-                self.walker.left()
-            elif key == "l":
-                self.walker.right()
-            elif key == "tab":
-                self.walker.tab_next()
-            elif key == "enter":
-                self.walker.edit()
-            elif key == "esc":
-                self.master.view_connlist()
+            if key in ["esc"]:
+                self.walker.stop_edit()
+                return None
+            elif key in ["left", "right", "up", "down"]:
+                self.walker.stop_edit()
             else:
-                return self.w.keypress(size, key)
+                self.w.keypress(size, key)
+                return None
+            
+        key = common.shortcuts(key)
+        if key == ["q", "esc"]:
+            self.master.pop_view()
+        elif key in ["h", "left"]:
+            self.walker.left()
+        elif key in ["l", "right"]:
+            self.walker.right()
+        elif key == "tab":
+            self.walker.tab_next()
+        elif key == "enter":
+            self.walker.start_edit()
+        else:
+            return self.w.keypress(size, key)