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