diff options
| -rw-r--r-- | pomu/source/__init__.py | 5 | ||||
| -rw-r--r-- | pomu/util/iquery.py | 126 | ||||
| -rw-r--r-- | setup.py | 2 |
3 files changed, 129 insertions, 4 deletions
diff --git a/pomu/source/__init__.py b/pomu/source/__init__.py index 07cc9d6..7d587c0 100644 --- a/pomu/source/__init__.py +++ b/pomu/source/__init__.py @@ -4,6 +4,5 @@ dispatcher = PackageDispatcher() import pomu.source.portage import pomu.source.file -# sealed until pbraw is released -#import pomu.source.url -#import pomu.source.bugz +import pomu.source.url +import pomu.source.bugz 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 @@ -9,7 +9,7 @@ setup( author_email='hilobakho@gmail.com', license='GNU GPLv2', packages=find_packages(exclude=['tests']), - install_requires=['Click', 'GitPython', 'patch'], + install_requires=['Click', 'GitPython', 'patch', 'pbraw', 'curtsies'], entry_points={ 'console_scripts':['pomu = pomu.cli:main_'] } |