aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMykyta Holubakha <hilobakho@gmail.com>2017-07-25 03:46:09 +0300
committerMykyta Holubakha <hilobakho@gmail.com>2017-07-25 03:46:09 +0300
commit058de401f4ffb1fe4ac562a16a7bdae3583ed4f3 (patch)
tree72d7a5bde8bbeda242e5ce10cdf2d81459619d5b
parentGenerify the repo code to allow merging patches (diff)
downloadpomu-058de401f4ffb1fe4ac562a16a7bdae3583ed4f3.tar.gz
pomu-058de401f4ffb1fe4ac562a16a7bdae3583ed4f3.tar.bz2
pomu-058de401f4ffb1fe4ac562a16a7bdae3583ed4f3.zip
Unsealed url and bugz package source modules
added a module for interactive queries depend on pbraw and curtsies
-rw-r--r--pomu/source/__init__.py5
-rw-r--r--pomu/util/iquery.py126
-rw-r--r--setup.py2
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
diff --git a/setup.py b/setup.py
index 53fcb80..d0fe3f9 100644
--- a/setup.py
+++ b/setup.py
@@ -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_']
}