parent
5fde1eb5b9
commit
c2a71e7240
|
@ -1,114 +0,0 @@
|
||||||
from __future__ import print_function, unicode_literals
|
|
||||||
import os, six
|
|
||||||
from .wordlist import (byte_to_even_word, byte_to_odd_word,
|
|
||||||
even_words_lowercase, odd_words_lowercase)
|
|
||||||
|
|
||||||
def make_code(channel_id, code_length):
|
|
||||||
assert isinstance(channel_id, type("")), type(channel_id)
|
|
||||||
words = []
|
|
||||||
for i in range(code_length):
|
|
||||||
# we start with an "odd word"
|
|
||||||
if i % 2 == 0:
|
|
||||||
words.append(byte_to_odd_word[os.urandom(1)].lower())
|
|
||||||
else:
|
|
||||||
words.append(byte_to_even_word[os.urandom(1)].lower())
|
|
||||||
return "%s-%s" % (channel_id, "-".join(words))
|
|
||||||
|
|
||||||
def extract_channel_id(code):
|
|
||||||
channel_id = int(code.split("-")[0])
|
|
||||||
return channel_id
|
|
||||||
|
|
||||||
class CodeInputter:
|
|
||||||
def __init__(self, initial_channelids, get_channel_ids, code_length,
|
|
||||||
used_completion_f):
|
|
||||||
self._initial_channelids = initial_channelids
|
|
||||||
self._get_channel_ids = get_channel_ids
|
|
||||||
self.code_length = code_length
|
|
||||||
self.last_text = None # memoize for a speedup
|
|
||||||
self.last_matches = None
|
|
||||||
self._used_completion_f = used_completion_f
|
|
||||||
|
|
||||||
def get_current_channel_ids(self):
|
|
||||||
if self._initial_channelids is not None:
|
|
||||||
channelids = self._initial_channelids
|
|
||||||
self._initial_channelids = None
|
|
||||||
return channelids
|
|
||||||
return self._get_channel_ids()
|
|
||||||
|
|
||||||
def wrap_completer(self, text, state):
|
|
||||||
try:
|
|
||||||
return self.completer(text, state)
|
|
||||||
except Exception as e:
|
|
||||||
# completer exceptions are normally silently discarded, which
|
|
||||||
# makes debugging challenging
|
|
||||||
print("completer exception: %s" % e)
|
|
||||||
raise e
|
|
||||||
|
|
||||||
def completer(self, text, state):
|
|
||||||
self._used_completion_f()
|
|
||||||
#if state == 0:
|
|
||||||
# print("", file=sys.stderr)
|
|
||||||
#print("completer: '%s' %d '%d'" % (text, state,
|
|
||||||
# readline.get_completion_type()),
|
|
||||||
# file=sys.stderr)
|
|
||||||
#sys.stderr.flush()
|
|
||||||
pieces = text.split("-")
|
|
||||||
last = pieces[-1].lower()
|
|
||||||
if text == self.last_text and len(pieces) >= 2:
|
|
||||||
# if len(pieces) == 1, skip the cache, so we can re-fetch the
|
|
||||||
# channel_id list
|
|
||||||
matches = self.last_matches
|
|
||||||
#print(" old matches", len(matches), file=sys.stderr)
|
|
||||||
else:
|
|
||||||
if len(pieces) <= 1:
|
|
||||||
channel_ids = self.get_current_channel_ids()
|
|
||||||
matches = [str(channel_id) for channel_id in channel_ids
|
|
||||||
if str(channel_id).startswith(last)]
|
|
||||||
else:
|
|
||||||
if len(pieces) % 2 == 0:
|
|
||||||
words = odd_words_lowercase
|
|
||||||
else:
|
|
||||||
words = even_words_lowercase
|
|
||||||
so_far = "-".join(pieces[:-1]) + "-"
|
|
||||||
matches = sorted([so_far+word for word in words
|
|
||||||
if word.startswith(last)])
|
|
||||||
self.last_text = text
|
|
||||||
self.last_matches = matches
|
|
||||||
#print(" new matches:", matches, file=sys.stderr)
|
|
||||||
if state >= len(matches):
|
|
||||||
return None
|
|
||||||
match = matches[state]
|
|
||||||
if len(pieces) < 1+self.code_length:
|
|
||||||
match += "-"
|
|
||||||
#print(" match: '%s'" % match, file=sys.stderr)
|
|
||||||
#sys.stderr.flush()
|
|
||||||
return match
|
|
||||||
|
|
||||||
|
|
||||||
def input_code_with_completion(prompt, initial_channelids, get_channel_ids,
|
|
||||||
code_length):
|
|
||||||
used_completion = []
|
|
||||||
def used_completion_f():
|
|
||||||
used_completion.append(True)
|
|
||||||
try:
|
|
||||||
import readline
|
|
||||||
c = CodeInputter(initial_channelids, get_channel_ids, code_length,
|
|
||||||
used_completion_f)
|
|
||||||
if readline.__doc__ and "libedit" in readline.__doc__:
|
|
||||||
readline.parse_and_bind("bind ^I rl_complete")
|
|
||||||
else:
|
|
||||||
readline.parse_and_bind("tab: complete")
|
|
||||||
readline.set_completer(c.wrap_completer)
|
|
||||||
readline.set_completer_delims("")
|
|
||||||
except ImportError:
|
|
||||||
pass
|
|
||||||
code = six.moves.input(prompt)
|
|
||||||
# Code is str(bytes) on py2, and str(unicode) on py3. We want unicode.
|
|
||||||
if isinstance(code, bytes):
|
|
||||||
code = code.decode("utf-8")
|
|
||||||
return (code, bool(used_completion))
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
code = input_code_with_completion("Enter wormhole code: ",
|
|
||||||
[], lambda: [], 2)
|
|
||||||
print("code is:", code)
|
|
Loading…
Reference in New Issue
Block a user