1 files changed, 126 insertions, 0 deletions
diff --git a/pomu/util/iquery.py b/pomu/util/iquery.py
new file mode 100644
index 0000000..032838d
--- /dev/null
+++ b/pomu/util/iquery.py
@@ -0,0 +1,126 @@
+"""A module to interactively query"""
+from pydoc import pager
+
+from curtsies import CursorAwareWindow, Input, FSArray, fsarray, fmtstr
+from curtsies.fmtfuncs import underline
+from pbraw import grab
+
+
+class Position:
+    row = attr.ib()
+    column = attr.ib()
+    def __init__(self, row=0, column=0):
+        self.row = row
+        self.column = column
+
+def clamp(x, m, M):
+    return max(m, min(x, M))
+
+def render_entry(entry, width, active=False):
+    # (name, contents, state, value)
+    char = '*' if entry[2] else ' '
+    w = 3 + fmtstr(entry[0]).width + 2
+    text = fmtstr(entry[3])
+    return fsmtstr(
+            '[' + underline(char) if active else char + '] ' +
+            entry[0] + ' ' +
+            entry[3][:width - w - 2] + '..' if text.width < width - w else entry[3])
+
+def process_entry(entry):
+    if isinstance(entry, str):
+        return (entry[0], None, False, None)
+    return (entry[0], entry[1], False, entry[0] if entry[0].endswith('.ebuild') else 'files/{}'.format(entry[0]))
+
+class Prompt:
+    def __init__(self, entries):
+        self.entries = [process_entry(x) for x in entries]
+        self.idx = 0
+        self.text = ''
+        self.list = True
+        self.cursor_pos = Position()
+
+    def run(self):
+        with open('/dev/tty', 'r') as tty_in, \
+             open('/dev/tty', 'w') as tty_out, \
+             Input(in_stream=tty_in) as input_, \
+             CursorAwareWindow(in_stream=tty_in,
+                               out_stream=tty_out,
+                               hide_cursor=False,
+                               extra_bytes_callback=input_generator.unget_bytes) as window:
+            self.window = window
+            self.render()
+            for event in input_:
+                if self.process_event(event) == -1:
+                    break
+                self.render()
+        return [(x[0], x[1], x[3]) for x in self.entries if x[2]]
+
+    def clamp(self, x):
+        return clamp(x, 0, len(self.entries))
+
+    def preview(self):
+        entry = self.entries[self.idx]
+        if entry[0] is not None:
+            pydoc.pager(entry[1])
+        else:
+            gr = grab(entry)
+            if not gr:
+                del self.entries[self.idx]
+                self.idx = clamp(self.idx - 1)
+                pydoc.pager('Error: could not fetch '.format(entry))
+            self.entries[self.idx:self.idx+1] = [process_entry((x[0], x[1].encode('utf-8'))) for x in gr]
+            pydoc.pager(self.entries[self.idx][1])
+
+    def toggle(self):
+        if self.idx == len(self.entries):
+            return
+        self.entries[self.idx][3] = not self.entries[self.idx][3]
+
+    def process_event(self):
+        if self.list:
+            if event == '<UP>':
+                self.idx = clamp(self.idx - 1)
+            elif event == '<DOWN>':
+                self.idx = clamp(self.idx + 1)
+            elif event == '<SPACE>':
+                self.toggle(self.idx)
+            elif event in {'p', 'P'}:
+                self.preview()
+            elif event in {'<ESC>', '<Ctrl-g>'}:
+                return -1
+            elif event in {'<Ctrl-j>', '<Ctrl-m>'}:
+                self.list = False
+                if self.idx == len(self.entries):
+                    return -1
+                self.cursor_pos.column = fmtstr(self.entries[self.idx][3]).width
+        else:
+            if event in {'<ESC>', '<Ctrl-g>'}:
+                self.list = True
+            if event == '<BACKSPACE>':
+                self.delete_last_char()
+            elif event == '<SPACE>':
+                self.add_char(' ')
+            elif event in {'<Ctrl-j>', '<Ctrl-m>'}:
+                self.set_text(self.text)
+                self.list = True
+            elif isinstance(event, str) and not event.startswith('<'):
+                self.add_char(event)
+
+    def render(self):
+        if self.list:
+            output = fsarray([render_entry(x) for x in self.entries] + [' [ OK ] '],
+                    self.window.width)
+            self.window.render_to_terminal(output)
+            return
+        cur = self.entries[self.idx]
+        output = fsarray(['Please provide value for {}'.format(cur[0]), cur[3]], self.window.width)
+        self.window.render_to_terminal(output, (self.cursor_pos.row, self.cursor_pos.column))
+
+    def add_char(self, char):
+        self.entries[self.idx][3] += char
+        self.cursor_pos.column += 1
+
+    def delete_last_char(self):
+        if self.text:
+            self.entries[self.idx][3] += self.entries[self.idx][3][:-1]
+            self.cursor_pos.column -= 1