#!python
from upydevice import S_UPYDEVICE
import serial
from binascii import hexlify
import getpass
import secrets
import hashlib
import string
import upydev
import time
import sys
import os
import signal
import netifaces
import nmap
import textwrap
import ast
import subprocess
import shlex
import webbrowser
import argparse
from datetime import datetime, date, timedelta
from sys import platform as _platform
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.primitives import serialization
# from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
import pygments
from pygments.lexers.python import PythonLexer
from prompt_toolkit import PromptSession
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
from prompt_toolkit.auto_suggest import ConditionalAutoSuggest
from prompt_toolkit.key_binding import KeyBindings
from prompt_toolkit.filters import Condition
from prompt_toolkit.application import run_in_terminal
from prompt_toolkit.styles import Style
from prompt_toolkit.formatted_text import HTML
from prompt_toolkit.formatted_text import PygmentsTokens
from prompt_toolkit import print_formatted_text


def get_cmd(raw):
    shcmd = shlex.split(raw)
    output = subprocess.check_output(shcmd).decode()
    return output


def get_rprompt():
    if status_encryp_msg['S']:
        if encrypted_flag['sec']:
            return HTML('<b><style fg="ansigreen"> INFO: </style></b><aaa fg="ansiblack" bg="ansiwhite"> ENCRYPTION </aaa><b><style fg="ansigreen"> ENABLED </style></b>')
        else:
            return HTML('<b><style fg="ansired"> WARNING: </style></b><aaa fg="ansiblack" bg="ansiwhite"> ENCRYPTION: </aaa><b><style fg="ansired"> DISABLED </style></b>')
    else:
        return HTML('<b><style fg="ansiblack"> {} </style></b>'.format(' '*10))


# CRYPTO


def rsa_keygen(dir='', store=True, show_key=False, id='00'):
    private_key = rsa.generate_private_key(
        public_exponent=65537, key_size=2048, backend=default_backend())
    my_p = getpass.getpass(prompt='Password: ', stream=None)
    pem = private_key.private_bytes(encoding=serialization.Encoding.PEM, format=serialization.PrivateFormat.PKCS8,
                                    encryption_algorithm=serialization.BestAvailableEncryption(bytes(my_p, 'utf-8')))
    if show_key:
        print(pem)
    if store:
        with open(dir+'upy_pub_rsa{}.key'.format(id), 'wb') as keyfile:
            keyfile.write(pem)
    return pem


def load_rsa_key(dir='', show_key=False, id='00'):
    buff_key = b''
    try:
        with open(dir+'upy_pub_rsa{}.key'.format(id), 'rb') as keyfile:
            while True:
                try:
                    buff = keyfile.read(2000)
                    if buff != b'':
                        buff_key += buff
                    else:
                        break
                except Exception as e:
                    print(e)
        if show_key:
            print(buff_key)
        return buff_key
    except Exception as e:
        print("No RSA key found for this device, generate one first with '$ upydev gen_rsakey -tfkey' ")
        return None


def upy_keygen(rsa_key):
    aZ09 = bytes(string.ascii_letters + string.digits, 'ascii')
    raw_key_list = [line for line in rsa_key.splitlines()[1:-1]]
    raw_key = b''
    for line in raw_key_list:
        raw_key += line
    random_token = secrets.token_bytes(32)  # send this
    for b in random_token:
        raw_key += bytes(chr(raw_key[b]), 'utf-8')
    key_hash = hashlib.sha256()
    key_hash.update(raw_key)
    hashed_key = key_hash.digest()
    index_key = [secrets.randbelow(len(hashed_key))
                 for i in range(8)]  # send this
    password_long = bytes([hashed_key[val] for val in index_key])
    password_short = bytes([aZ09[val % len(aZ09)]
                            for val in password_long]).decode()

    return (password_short, random_token + bytes(index_key))


def upy_session_keygen(rsa_key, save_sessionkey=False, token=None, uid='00'):
    raw_key_list = [line for line in rsa_key.splitlines()[1:-1]]
    raw_key = b''
    for line in raw_key_list:
        raw_key += line
    if token is None:
        random_token = secrets.token_bytes(32)  # send this
    else:
        random_token = token[:32]
    for b in random_token:
        raw_key += bytes(chr(raw_key[b]), 'utf-8')
    key_hash = hashlib.sha256()
    key_hash.update(raw_key)
    hashed_key = key_hash.digest()
    if token is None:
        index_pvkey = [secrets.randbelow(len(hashed_key))
                       for i in range(32)]  # send this
    else:
        index_pvkey = token[32:64]
    pv_key = bytes([hashed_key[val] for val in index_pvkey])
    if token is None:
        index_ivkey = [secrets.randbelow(len(hashed_key))
                       for i in range(16)]  # send this
    else:
        index_ivkey = token[64:]
    iv = bytes([hashed_key[val] for val in index_ivkey])

    if save_sessionkey:
        path = upydev.__path__[0]
        if token is None:
            with open(path+'/.session_host_{}.key'.format(uid), 'w') as sess_config:
                sess_keys = "{}\n{}\n".format(pv_key, iv)
                sess_config.write(sess_keys)
            print('New session host key saved in upydev session directory!')
        else:
            with open(path+'/.session_dev_{}.key'.format(uid), 'w') as sess_config:
                sess_keys = "{}\n{}\n".format(pv_key, iv)
                sess_config.write(sess_keys)
            print('New session dev key saved in upydev session directory!')
    if token is None:
        return (pv_key, iv, [random_token + bytes(index_pvkey) + bytes(index_ivkey)])
    else:
        return (pv_key, iv)


class SERIAL_REPL_server:
    """
    SERIAL REPL server class
    """

    def __init__(self, dev=None, port=None, buff=1024, dev_id = None):
        self.port = port
        self.dev = dev
        self.dev_id = dev_id
        self.buff = b''
        self.raw_buff = b''
        self.message = b''
        self.n_bytes_sent = 0
        self.prompt = b'>>> '
        self.output = None
        self.IO = False

    def flush_conn(self):
        flushed = 0
        while flushed < 2:
            try:
                if self.dev.serial.readable():
                    self.raw_buff = self.dev.serial.read_all()
                    flushed += 1
            except Exception as e:
                flushed += 1

    def send_message(self, message):
        self.n_bytes_sent = self.dev.serial.write(bytes(message, 'utf-8'))

    def recv_message(self):
        try:
            self.buff = b''
            self.buff += self.dev.serial.read_all()
            print(self.buff.decode())
        except Exception as e:
            if self.buff.decode() == '':
                pass
            else:
                print(self.buff.decode())
            pass

    def colored(self, msg):
        tokens = list(pygments.lex(msg, lexer=PythonLexer()))
        print_formatted_text(PygmentsTokens(tokens), end='')

    # def sw_repl(self, close_wbrepl=False):
    #     if self.SSL_MODE:
    #         self.SSL_MODE = False
    #         self.dev.close_wconn()
    #         self.sh_repl("ssl_client.switch_wrepl()")
    #         if not close_wbrepl:
    #             self.dev.open_wconn()
    #     else:
    #         self.SSL_MODE = True
    #         self.dev.wr_cmd("ssl_client.switch_sh_repl()")
    #         self.dev.close_wconn()
    #         self.flush_conn()

    def enable_wrepl_io(self):
        if not self.IO:
            self.dev.serial.close()
            self.IO = True
        else:
            self.IO = False
            self.dev.serial.open()
            # self.dev.open_wconn()

    def sh_repl(self, inp, just_recv=False, silent=True, follow=False, n=0,
                 traceback=False, debug=False, prevent_hang=False,
                 banner=False, check_list=False, list_th=20,
                 no_raw_buff=False, assert_len=True, dec=True, color=False):
        if not check_list:
            self.dev.output = None
        self.buff = b''
        if not no_raw_buff:
            self.raw_buff = b''
        self.flush_conn()
        if not just_recv:
            self.send_message(inp+'\r')
        # WAIT TILL DATA AVAILABLE
        # CATCH PROMPT
        if prevent_hang:
            time.sleep(0.5)
        time.sleep(0.2)
        # CATCH MESSAGE
        n = n
        if follow:
            self.raw_buff += self.dev.serial.read(self.n_bytes_sent)
        while True:
            try:
                self.message = self.dev.serial.read_all()
                self.buff += self.message
                if not no_raw_buff:
                    self.raw_buff += self.message
                    # self.raw_buff += b'*'
                # print(self.buff)
                if follow:
                    # printed = False
                    if self.message == b'':
                        pass
                    else:
                        # msg = [line.replace('\r', '') for line in self.message.decode().split('\n')]
                        if self.message.startswith(b'\n'):
                            self.message = self.message[1:]
                        msg = self.message.replace(b'\r', b'').decode()
                        # for line in msg[:-1]:
                        #     if line != '>>> ':
                        #         print(line)
                        if 'cat' in inp:
                            if msg.endswith('>>> '):
                                msg = msg.replace('>>> ', '')
                                if not msg.endswith('\n'):
                                    msg = msg+'\n'
                        print(msg.replace('>>> ', ''), end='')
                    # if msg == '':
                    #     msg = self.buff.decode().split('\n')[-2]
                    # if msg.endswith('\r'):  # repl command never ends without \r
                    #     if n >= 1:  # filter first \r of command
                    #         print(msg.replace('\r', ''))
                    #     else:
                    #         n += 1
                    # else:
                    #     if n >= 1:
                    #         try:
                    #             if b'>>> ' in self.buff:  # parse last line of cat command
                    #                 last_line = self.buff.decode().split('\n')[-1].replace('\r', '').replace('>>> ', '')
                    #                 var = ast.literal_eval(last_line)
                    #                 if isinstance(var, list) or isinstance(var, dict):
                    #                     print(var)
                    #         except Exception as e:
                    #             if inp.split(';')[0].startswith('cat') and inp.split(';')[0].endswith(".txt')"):
                    #                 if b'>>> ' in self.buff:
                    #                     last_line = self.buff.decode().split('\n')[-1].replace('\r', '').replace('>>> ', '')
                    #                     if last_line != '':
                    #                         print(last_line)
                    #             pass
                if self.buff.endswith(b'>>> '):
                    break
            # except KeyboardInterrupt:
            #     if prevent_hang:
            #         self.conn.sendall(b'\x03')
            #         self.flush_conn()
            #     break
            except Exception as e:
                if True:
                    print(e, 2)
                if prevent_hang:
                    time.sleep(0.5)
                    self.dev.serial.write(b'\x03\x0d')
                pass
        # WAIT TILL NO DATA AVAILABLE
        # while True:
        #     time.sleep(0.1)
        #     try:
        #         self.message = self.dev.serial.readall()
        #         self.buff += self.message
        #         if not no_raw_buff:
        #             self.raw_buff += self.message
        #         if self.message == b'':
        #             break
        #     except Exception as e:
        #         break
        if self.buff != b'':
            msg_b = self.buff.splitlines()[1:]
            try:
                msg_b[-1] = msg_b[-1].replace(b'>>> ', b'')
            except Exception as e:
                pass
            if dec:
                self.output = '\n'.join([m.decode() for m in msg_b if m != b''])
            if self.output != '':
                if not check_list:
                    self.dev.output = None
                    self.dev.output = self.output
                    self.dev.response = self.output
                # check_list
                if check_list:
                    try:
                        return ast.literal_eval(self.output)
                    except Exception as e:
                        # print(self.output)
                        return None
                if not follow:
                    if assert_len:
                        if self.output.endswith('\n>>> '):
                            self.output = self.output.replace('\n>>> ', '')
                            self.dev.output = self.output
                            self.dev.response = self.output

                        if self.output.startswith('[') and self.output.endswith(']') and "'" in self.output:
                            try:
                                recv_list = ast.literal_eval(self.output)

                                if isinstance(recv_list, list):
                                    recv_len = len(recv_list)
                                    if recv_len > list_th:
                                        time.sleep(0.3)
                                        true_len = self.sh_repl("len({})".format(inp.split(';')[0]),
                                                                 silent=True, check_list=True,
                                                                 no_raw_buff=True)
                                        # print(true_len, recv_len)
                                        if true_len == recv_len:
                                            # print(True)
                                            pass
                                        else:
                                            # print('MALFORMED LIST')
                                            time.sleep(0.3)
                                            self.sh_repl(inp=inp, just_recv=just_recv,
                                                          silent=True, follow=follow, n=n,
                                                          traceback=False, debug=False,
                                                          prevent_hang=False,
                                                          banner=False)
                            except Exception as e:
                                # print('MALFORMED LIST')
                                time.sleep(0.2)
                                self.sh_repl(inp=inp, just_recv=just_recv,
                                              silent=True, follow=follow, n=n,
                                              traceback=False, debug=False,
                                              prevent_hang=prevent_hang,
                                              banner=False)

                if not silent:
                    if color:
                        self.colored(b'\n'.join(self.raw_buff.splitlines()[1:-1]).decode())
                    else:
                        print(self.output)
            if traceback:
                print(self.buff.decode())
            if banner:
                print(self.buff.decode().replace('>>> ', ''), end='\r')

    def paste_buff(self, long_command):
        self.dev.serial.write(b'\x05')
        lines = long_command.split('\n')
        for line in lines:
            self.flush_conn()
            time.sleep(0.1)
            self.send_message(line+'\n')


class SERIAL_TOOL:
    def __init__(self, sh_srepl, port=None):
        self.host = None
        self.port = port
        self.sh_srepl = sh_srepl
        self.buff = bytearray(1024*2)
        self.bloc_progress = ["▏", "▎", "▍", "▌", "▋", "▊", "▉"]
        self.columns, self.rows = os.get_terminal_size(0)
        self.cnt_size = 65
        self.bar_size = int((self.columns - self.cnt_size))
        self.pb = False
        self.wheel = ['|', '/', '-', "\\"]

    def start_SOC(self):
        # self.sh_srepl.dev.serial.close()
        # self.serv_soc = Pyboard(self.sh_srepl.dev.serial_port)
        pass

    def stop_SOC(self):
        # self.conn.close()
        # self.serv_soc.close()
        # self.sh_srepl.dev.serial.open()
        # self.sh_srepl.sh_repl("\x03\x0d")
        pass

    def get_pb(self):
        self.columns, self.rows = os.get_terminal_size(0)
        if self.columns > self.cnt_size:
            self.bar_size = int((self.columns - self.cnt_size))
            self.pb = True
        else:
            self.bar_size = 1
            self.pb = False

    def do_pg_bar(self, index, wheel, nb_of_total, speed, time_e, loop_l,
                  percentage, ett):
        l_bloc = self.bloc_progress[loop_l]
        if index == self.bar_size:
            l_bloc = "█"
        sys.stdout.write("\033[K")
        print('▏{}▏{:>2}{:>5} % | {} | {:>5} KB/s | {}/{} s'.format("█" *index + l_bloc  + " "*((self.bar_size+1) - len("█" *index + l_bloc)),
                                                                        wheel[index % 4],
                                                                        int((percentage)*100),
                                                                        nb_of_total, speed,
                                                                        str(timedelta(seconds=time_e)).split('.')[0][2:],
                                                                        str(timedelta(seconds=ett)).split('.')[0][2:]), end='\r')
        sys.stdout.flush()

    def get(self, src, chunk_size=256): # from Pyboard.py
        self.get_pb()
        cnt = 0
        t_start = time.time()
        # def_chunk = chunk
        self.sh_srepl.sh_repl("os.stat('{}')[6]".format(src))
        self.sh_srepl.dev.get_output()
        sz = self.sh_srepl.dev.output
        print("{}  [{:.2f} KB]".format(src, sz / 1024))
        self.sh_srepl.flush_conn()
        # self.start_SOC()
        self.sh_srepl.sh_repl("f=open('%s','rb');r=f.read" % src)
        with open(src, 'wb') as f:
            pass
        with open(src, 'ab') as f:
            while True:
                data = b''
                self.sh_srepl.sh_repl("print(r(%u))" % chunk_size, assert_len=False)
                if self.sh_srepl.buff != b'':
                    msg_b = self.sh_srepl.buff.split(b'\n')[1:]
                    try:
                        msg_b[-1] = msg_b[-1].replace(b'>>> ', b'')
                    except Exception as e:
                        pass
                    self.sh_srepl.output = b'\n'.join([m for m in msg_b if m != b''])
                try:
                    data = eval(self.sh_srepl.output.replace(b'\r', b''))
                except Exception as e:
                    print(e)
                    data = b''
                # print(type(data), data)
                if data == b'':
                    break
                f.write(data)
                cnt += len(data)
                loop_index_f = (cnt/sz)*self.bar_size
                loop_index = int(loop_index_f)
                loop_index_l = int(round(loop_index_f-loop_index, 1)*6)
                nb_of_total = "{:.2f}/{:.2f} KB".format(cnt/(1024), sz/(1024))
                percentage = cnt / sz
                t_elapsed = time.time() - t_start
                t_speed = "{:^2.2f}".format((cnt/(1024))/t_elapsed)
                ett = sz / (cnt / t_elapsed)
                if self.pb:
                    self.do_pg_bar(loop_index, self.wheel,
                                   nb_of_total, t_speed, t_elapsed,
                                   loop_index_l, percentage, ett)
        print('\n')
        self.sh_srepl.sh_repl("f.close()")

    def put(self, src, chunk_size=256): # from Pyboard.py
        self.get_pb()
        sz = os.stat(src)[6]
        cnt = 0
        t_start = time.time()
        print("{}  [{:.2f} KB]".format(src, sz / 1024))
        self.sh_srepl.sh_repl("f=open('%s','wb');w=f.write" % src)
        with open(src, 'rb') as f:
            while True:
                data = f.read(chunk_size)
                if not data:
                    break
                # if sys.version_info < (3,):
                #     self.serv_soc.exec_('w(b' + repr(data) + ')')
                # else:
                #     self.serv_soc.exec_('w(' + repr(data) + ')')
                self.sh_srepl.sh_repl('w(' + repr(data) + ')', assert_len=False)
                cnt += len(data)
                loop_index_f = (cnt/sz)*self.bar_size
                loop_index = int(loop_index_f)
                loop_index_l = int(round(loop_index_f-loop_index, 1)*6)
                nb_of_total = "{:.2f}/{:.2f} KB".format(cnt/(1024), sz/(1024))
                percentage = cnt / sz
                t_elapsed = time.time() - t_start
                t_speed = "{:^2.2f}".format((cnt/(1024))/t_elapsed)
                ett = sz / (cnt / t_elapsed)
                if self.pb:
                    self.do_pg_bar(loop_index, self.wheel,
                                   nb_of_total, t_speed, t_elapsed,
                                   loop_index_l, percentage, ett)
        print('\n')
        self.sh_srepl.sh_repl("f.close()")


class WBREPL_IO:
    def __init__(self, dev):
        self.WBREPL_ON = True
        self.dev = dev

    def enable_wrepl_io(self):
        if self.WBREPL_ON:
            self.dev.close_wconn()
            self.WBREPL_ON = False
        else:
            self.dev.open_wconn()
            self.WBREPL_ON = True

class BASE_SERIAL_DEVICE:
    def __init__(self, serial_port, baudrate):
        self.serial = serial.Serial(serial_port, baudrate)
        self.bytes_sent = 0
        self.buff = b''
    def cmd(self, cmd, decode=True):
        self.bytes_sent = self.serial.write(bytes(cmd+'\r', 'utf-8'))
        time.sleep(0.2)
        self.buff = self.serial.read_all()[self.bytes_sent+1:]
        if self.buff == b'':
            time.sleep(0.2)
            self.buff = self.serial.read_all()
        # print(self.buff)
        if decode:
            try:
                return ast.literal_eval(self.buff.replace(b'\r\n',b'').replace(b'>>> ', b'').decode())
            except Exception as e:
                return self.buff.replace(b'\r\n',b'').replace(b'>>> ', b'').decode()
        else:
            try:
                return ast.literal_eval(self.buff.replace(b'\r\n',b'').replace(b'>>> ', b''))
            except Exception as e:
                return self.buff.replace(b'\r\n',b'').replace(b'>>> ', b'')

# ARGPARSE HERE TO ACCEPT TARGET AND PASSWORD
parser = argparse.ArgumentParser()
parser.add_argument("-p", help='device port', required=True)
parser.add_argument("-v", help='verbose mode',
                    default=False, action='store_true')
parser.add_argument("-t", help='device baudrate', default=115200)
parser.add_argument("-ping", help='Test if device is reachable first',
                    required=False, action='store_true')
parser.add_argument(
    "-dev", help='device name, default unique_id', required=False)
parser.add_argument("-r", help='reset on exit',
                    default=False, action='store_true')
parser.add_argument("-nem",
                    help='No encryption mode, this bypass handshake and does not allow encryption afterwards',
                    default=False, action='store_true')
args = parser.parse_args()

# HANDSHAKE
basedev = BASE_SERIAL_DEVICE(args.p, args.t)
espdev = S_UPYDEVICE(args.p, timeout=1000, baudrate=args.t, reset=False)
if args.ping:
    if basedev.serial.writable():
        pass
    else:
        print('DEVICE NOT REACHABLE')
        sys.exit()
if args.v:
    print('Welcome to SERIAL SHELL-REPL 0.0.1!')

time.sleep(0.2)
kbi = basedev.cmd('\x03\r')
uid = basedev.cmd("from machine import unique_id; unique_id()")
try:
    unique_id = hexlify(uid).decode()
except Exception as e:
    unique_id = uid
devp = basedev.cmd("import sys; sys.platform")
dev_platform = devp
espdev.platform = dev_platform
if args.v:
    print('{}@{}'.format(dev_platform, unique_id))
shr_cp = SERIAL_REPL_server(dev=espdev, dev_id=unique_id)
shr_htool = SERIAL_TOOL(sh_srepl=shr_cp)
print('SERIAL SHELL-REPL connected')
shr_cp.sh_repl("\x02", banner=True)
shr_cp.flush_conn()
espdev.output = None
print('Use CTRL-x to exit, Use CTRL-s,ENTER to toggle shell/repl mode\nUse CTRL-k to see more info')

# PROMT SESSION CONFIGURATION

# Style
style_p = Style.from_dict({
    # User input (default text).
    '':          '#ffffff',

    # Prompt.
    'userpath': 'ansimagenta bold',
    'username': 'ansigreen bold',
    'at':       'ansigreen bold',
    'colon':    '#ffffff',
    'pound':    'ansiblue bold',
    'host':     'ansigreen bold',
    'path':     'ansiblue bold',
})

# SET DEV NAME
host_name = unique_id
if args.dev is not None:
    host_name = args.dev

shell_message = [
    ('class:userpath',    ''),
    ('class:username', dev_platform),
    ('class:at',       '@'),
    ('class:host',     host_name),
    ('class:colon',    ':'),
    ('class:path',     '~ '),
    ('class:pound',    '$ '),
]

d_prompt = '>>> '
# KEYBINDINGS
kb = KeyBindings()

dev_path = {'p': ' '}
local_path = {'p': ''}
show_local_path = {'s': False}
status_encryp_msg = {'S': False, 'Toggle': True}
exit_flag = {'exit': False}
encrypted_flag = {'sec': True}
prompt = {'p': '>>> '}
paste_flag = {'p': False}
paste_buffer = {'B': []}
reset_flag = {'R': args.r}
autosuggest = {'A': False}
shell_mode = {'S': False}
edit_mode = {'E': False, 'File': ''}
shell_mode_run = {'R': False}
script_is_running = {'R': False, 'script': 'test_code'}
shell_prompt = {'s': shell_message}
shell_commands = ['sz', 'cd', 'mkdir', 'cat', 'head', 'rm', 'rmdir', 'pwd',
                  'run']
custom_sh_cmd_kw = ['df', 'datetime', 'ifconfig', 'ifconfig_t', 'netscan',
                    'apconfig', 'apconfig_t', 'meminfo', 'install',
                    'apscan', 'touch', 'edit', 'srepl',
                    'whoami', 'exit', 'pwdl', 'lsl', 'cdl', 'put', 'get',
                    'catl', 'l_micropython', 'ls', 'python', 'vim',
                    'set_localtime', 'tree', 'l_ifconfig',
                    'l_ifconfig_t', 'reload', 'docs', 'flush_soc',
                    'get_rawbuff', 'l_tree', 'view', 'bat', 'rcat']
CRED = '\033[91;1m'
CGREEN = '\33[32;1m'
CEND = '\033[0m'
ABLUE_bold = '\u001b[34;1m'
MAGENTA_bold = '\u001b[35;1m'
AUTHMODE_DICT = {0: 'NONE', 1: 'WEP', 2: 'WPA PSK', 3: 'WPA2 PSK',
                    4: 'WPA/WAP2 PSK'}


# START IN SHELL MODE
if shell_mode['S']:
    prompt['p'] = d_prompt
    shell_mode['S'] = False
else:
    prompt['p'] = shell_prompt['s']
    shell_mode['S'] = True
    shr_cp.sh_repl('import gc;import os;from upysh import *;gc.collect()', silent=True)
    time.sleep(1)
    shr_cp.flush_conn()
    shr_cp.flush_conn()
    espdev.output = None


# KEYBINDINGS INFO
kb_info = """
Custom keybindings:
- CTRL-x : to exit SHELL-REPL Terminal
- CTRL-u : toggle encryption mode (on/off), this prints a right aligned status message
- CTRL-p : toggle encryption right aligned status message
- CTRL-e : paste mode in repl, (edit mode after 'edit' shell command)
- CTRL-d : ends paste mode in repl, (ends edit mode after 'edit' shell command)
          (or soft-reset in repl, CTRL-C to start repl again)
- CTRL-c : KeyboardInterrupt, in normal mode, cancel in paste mode
- CTRL-b : prints MicroPython version and sys platform
- CTRL-r : to flush line buffer
- CTRL-o : to list files in cwd (ls shorcut command)
- CTRL-n : shows mem_info()
- CTRL-y : gc.collect() shortcut command
- CTRL-space : repeats last command
- CTRL-t : runs test_code.py if present
- CTRL-w : flush test_code from sys modules, so it can be run again
- CTRL-a : set cursor position at the beggining
- CTRL-f : toggle autosuggest mode (Fish shell like)(use right arrow to complete)
- CTRL-g : To active listen for device output (Timer or hardware interrupts), CTRL-c to break
- CRTL-s , ENTER : toggle shell mode to navigate filesystem (see shell commands)
- CTRL-k : prints the custom keybindings (this list) (+ shell commands if in shell mode)
>>> """

shell_commands_info = """
* Autocompletion commands:
     - tab to autocomplete device file / dirs names / raw micropython (repl commands)
     - shift-tab to autocomplete shell comands
     - shift-right to autocomplete local file / dirs names
     - shift-left,ENTER to toggle local path in prompt

* Device shell commands:
    * upysh commands:
        - sz   : list files and size in bytes
        - head : print the head of a file
        - cat  : prints the content of a file
        - mkdir: make directory
        - cd   : change directory (cd .. to go back one level)
        - pwd  : print working directory
        - rm   : to remove a file
        - rmdir: to remove a directory

    * custom shell commands:
        - ls  : list device files in colored format (same as pressing tab on empty line)
        - tree : to print a tree version of filesystem (to see also hidden files/dirs use 'tree -a')
        - run  : to run a 'script.py'
        - df   : to see filesystem flash usage (and SD if already mounted)
        - meminfo: to see RAM info
        - whoami : to see user, system and machine info
        - datetime: to see device datetime (if not set, will display uptime)
        - set_localtime : to set the device datetime from the local machine time
        - ifconfig: to see STATION interface configuration (IP, SUBNET, GATEAWAY, DNS)
        - ifconfig_t: to see STATION interface configuration in table format
                      (IP, SUBNET, GATEAWAY, DNS, ESSID, RSSI)
        - netscan: to scan WLAN's available, (ESSID, MAC ADDRESS, CHANNEL, RSSI, AUTH MODE, HIDDEN)
        - apconfig: to see ACCESS POINT (AP) interface configuration (IP, SUBNET, GATEAWAY, DNS)
        - apconfig_t: to see ACCESS POINT (AP) interface configuration in table format
                     (SSID, BSSID, CHANNEL, AUTH, IP, SUBNET, GATEAWAY, DNS)
        - install: to install a library into the device with upip.
        - touch  : to create a new file (e.g. touch test.txt)
        - edit   : to edit a file (e.g. edit my_script.py)
        - get    : to get a file from the device
        - put    : to upload a file to the device
        - srepl  : to enter the Serial Terminal (This needs Picocom)
        - reload : to delete a module from sys.path so it can be imported again.
        - flush_soc: to flush serial in case of wrong output
        - view   : to preview '.pbm' binary image files (image need to be centered and rows = columns)
        - bat    : prints the content of a '.py' file with Python syntax hightlighting
        - rcat   : prints the raw content of a file
        - exit   : to exit SHELL-REPL Terminal

* Local shell commands:
    - pwdl   : to see local path
    - cdl    : to change local directory
    - lsl    : to list local directory
    - catl   : to print the contents of a local file
    - l_micropython: if "micropython" local machine version available in $PATH, runs it.
    - python : switch to local python3 repl
    - vim    : to edit a local file with vim  (e.g. vim script.py)
    - l_ifconfig: to see local machine STATION interface configuration (IP, SUBNET, GATEAWAY, DNS)
    - l_ifconfig_t: to see local machine STATION interface configuration in table format
                  (IP, SUBNET, GATEAWAY, DNS, ESSID, RSSI)
    - docs : to open MicroPython docs site in the default web browser, if a second term
            is passed e.g. 'docs machine' it will open the docs site and search for 'machine'
    - get_rawbuff: to get the raw output of a command (for debugging purpose)


"""

# TAB OPTIONS FORMATTER


# from @The Data Scientician : https://stackoverflow.com/questions/9535954/printing-lists-as-tabular-data
def print_table(data, cols=4, wide=16, format_SH=False, autocols=True,
                autocol_tab=False, sort=True, autowide=False, max_cols=4):
    '''Prints formatted data on columns of given width.'''
    if sort:
        data.sort(key=str.lower)
    if format_SH:
        if autocols:
            wide_data = max([len(namefile) for namefile in data]) + 2
            if wide_data > wide:
                wide = wide_data
            columns, rows = os.get_terminal_size(0)
            cols = int(columns/(wide))
        data = ['{}{}{}{}'.format(ABLUE_bold, val, CEND, ' '*(wide-len(val)))
                if '.' not in val else val for val in data]
        data = ['{}{}{}{}'.format(MAGENTA_bold, val, CEND, ' '*(wide-len(val)))
                if '.py' not in val and '.mpy' not in val and '.' in val else val for val in data]
        data = ['{}{}{}{}'.format(
            CGREEN, val, CEND, ' '*(wide-len(val))) if '.mpy' in val else val for val in data]
    if autocol_tab:
        data = [namefile if len(namefile) < wide else namefile+'\n' for namefile in data]
    if autowide:
        wide_data = max([len(namefile) for namefile in data]) + 2
        if wide_data > wide:
            wide = wide_data
        columns, rows = os.get_terminal_size(0)
        cols = int(columns/(wide))
        if max_cols < cols:
            cols = max_cols
    n, r = divmod(len(data), cols)
    pat = '{{:{}}}'.format(wide)
    line = '\n'.join(pat * cols for _ in range(n))
    last_line = pat * r
    print(line.format(*data))
    print(last_line.format(*data[n*cols:]))


def read_pbm(fname):
    fmt, banner, dims, data = fname.splitlines()[1:]
    data = data.replace(b'>>> ', b'')
    dimensions = [int(dim.decode()) for dim in dims.split(b' ')]
    pixel_blocks = [bin(x).replace('0b', '0'*(8-len(bin(x)[2:]))) for x in data]
    raw_bin_img = ''.join(pixel_blocks)
    img_term = ["█" if int(x) > 0 else " " for x in raw_bin_img]
    term_images = '\n'.join(textwrap.wrap(''.join(img_term[6:]), dimensions[0]*2)[:dimensions[0]])
    img_1 = '\n'.join([line[:dimensions[0]] for line in term_images.split('\n')])
    return dimensions, data, raw_bin_img, img_1


# SHELL COMMANDS HANDLERS

def map_upysh(cmd_inp):
    frst_cmd = cmd_inp.split(' ')[0]
    if len(cmd_inp.split(' ')) > 1:
        scnd_cmd = cmd_inp.split(' ')[1]
        if scnd_cmd != '':
            shell_cmd = "{}('{}')".format(frst_cmd, scnd_cmd)
    else:
        shell_cmd = frst_cmd

    if shell_cmd == 'sz':
        shell_cmd = 'ls()'
    if frst_cmd == 'run':
        shell_cmd = 'import {}'.format(scnd_cmd.split('.')[0])
        # make a run interactive mode that do not escape input
        # conditional ENTER, flush buffer, send run command, then CTRL-C can
        # be catched, print in terminal
    if shell_cmd == 'cd':
        shell_cmd = "cd('/')"
    return shell_cmd, frst_cmd


def _dt_format(number):
        rtc_n = str(number)
        if len(rtc_n) == 1:
            rtc_n = "0{}".format(rtc_n)
            return rtc_n
        else:
            return rtc_n


def _ft_datetime(t_now):
    return([_dt_format(i) for i in t_now])


def send_custom_sh_cmd(cmd, capture_output=False, sh_silent=True):
    shr_cp.sh_repl(cmd)
    espdev.get_output()
    if not sh_silent:
        print(espdev.output)
    return espdev.output


def print_filesys_info(filesize):
    _kB = 1024
    if filesize < _kB:
        sizestr = str(filesize) + " by"
    elif filesize < _kB**2:
        sizestr = "%0.1f KB" % (filesize / _kB)
    elif filesize < _kB**3:
        sizestr = "%0.1f MB" % (filesize / _kB**2)
    else:
        sizestr = "%0.1f GB" % (filesize / _kB**3)
    return sizestr


def custom_sh_cmd(cmd_inp):
    # map cmd_inp to comand string
    if cmd_inp == 'df':
        resp = send_custom_sh_cmd("os.statvfs('')")
        # resp_sd = send_custom_sh_cmd("os.statvfs('sd')")
        size_info = resp
        total_b = size_info[0] * size_info[2]
        used_b = (size_info[0] * size_info[2]) - (size_info[0] * size_info[3])
        total_mem = print_filesys_info(total_b)
        free_mem = print_filesys_info(size_info[0] * size_info[3])
        used_mem = print_filesys_info(used_b)

        print("{0:12}{1:^12}{2:^12}{3:^12}{4:^12}{5:^12}".format(*['Filesystem',
                                                                   'Size', 'Used',
                                                                   'Avail',
                                                                   'Use%', 'Mounted on']))
        try:
            print('{0:12}{1:^12}{2:^12}{3:^12}{4:>8}{5:>5}{6:12}'.format('Flash', total_mem,
                                                                    used_mem, free_mem,
                                                                    "{:.1f} %".format((used_b/total_b)*100), ' ', '/'))
        except Exception as e:
            pass
        vfs_resp = send_custom_sh_cmd("{dir:os.statvfs(dir) for dir in os.listdir() if os.stat(dir)[0] & 0x4000}")
        for vfs in vfs_resp.keys():
            if vfs_resp[vfs] != resp:
                size_info_sd = vfs_resp[vfs]
                total_b_sd = size_info_sd[0] * size_info_sd[2]
                used_b_sd = (size_info_sd[0] * size_info_sd[2]) - \
                    (size_info_sd[0] * size_info_sd[3])
                total_mem_sd = print_filesys_info(total_b_sd)
                free_mem_sd = print_filesys_info(size_info_sd[0] * size_info_sd[3])
                used_mem_sd = print_filesys_info(used_b_sd)
                print('{0:12}{1:^12}{2:^12}{3:^12}{4:>8}{5:>5}{6:12}'.format(vfs, total_mem_sd,
                                                                        used_mem_sd, free_mem_sd,
                                                                        "{:.1f} %".format((used_b_sd/total_b_sd)*100), ' ', '/{}'.format(vfs)))

    if cmd_inp == 'datetime':
        send_custom_sh_cmd("import time;tnow=time.localtime()")
        resp = send_custom_sh_cmd('tnow[:6]')
        print("{}-{}-{}T{}:{}:{}".format(*_ft_datetime(resp)))

    if cmd_inp == 'ifconfig':
        sta_isconnected = send_custom_sh_cmd(
            "network.WLAN(network.STA_IF).isconnected()")
        if sta_isconnected:
            ifconf = send_custom_sh_cmd(
                "network.WLAN(network.STA_IF).ifconfig()")
            print(ifconf)

        else:
            print('STA interface not connected')

    if cmd_inp == 'ifconfig_t':
        sta_isconnected = send_custom_sh_cmd(
            "network.WLAN(network.STA_IF).isconnected()")
        if sta_isconnected:
            ifconf = send_custom_sh_cmd(
                "network.WLAN(network.STA_IF).ifconfig()")
            essid = send_custom_sh_cmd(
                "network.WLAN(network.STA_IF).config('essid')")
            signal_rssi = send_custom_sh_cmd(
                "network.WLAN(network.STA_IF).status('rssi')")
            print('=' * 106)
            print('{0:^15} | {1:^15} | {2:^15} | {3:^15} | {4:^15} | {5:^15} '.format(
                'IP', 'SUBNET', 'GATEAWAY', 'DNS', 'SSID', 'RSSI'))
            print('{0:^15} | {1:^15} | {2:^15} | {3:^15} | {4:^15} | {5:^15} '.format(*ifconf,
                                                                                      essid,
                                                                                      signal_rssi))

        else:
            print('STA interface not connected')

    if cmd_inp == 'netscan':
        enable_sta = send_custom_sh_cmd(
            "network.WLAN(network.STA_IF).active(1)")
        if enable_sta:
            scan = send_custom_sh_cmd("network.WLAN(network.STA_IF).scan()")
            print('=' * 110)
            print('{0:^20} | {1:^25} | {2:^10} | {3:^15} | {4:^15} | {5:^10} '.format(
                'SSID', 'BSSID', 'CHANNEL', 'RSSI', 'AUTHMODE', 'HIDDEN'))
            if 'event' in espdev.response:
                # print(espdev.response)
                try:
                    scan = ast.literal_eval(espdev.response.splitlines()[-1])
                except Exception as e:
                    scan = []
                # print(scan)
            for netscan in scan:
                auth = AUTHMODE_DICT[netscan[4]]
                vals = hexlify(netscan[1]).decode()
                bssid = ':'.join([vals[i:i+2] for i in range(0, len(vals), 2)])
                print('{0:^20} | {1:^25} | {2:^10} | {3:^15} | {4:^15} | {5:^10} '.format(
                    netscan[0].decode(), bssid, netscan[2], netscan[3],
                    auth, str(netscan[5])))

        else:
            print("Can't enable STA interface")

    if cmd_inp == 'apconfig':
        ap_isconnected = send_custom_sh_cmd(
            "network.WLAN(network.AP_IF).active()")
        if ap_isconnected:
            apconf = send_custom_sh_cmd(
                "network.WLAN(network.AP_IF).ifconfig()")
            print(apconf)

        else:
            print('AP interface not connected')

    if cmd_inp == 'apconfig_t':
        ap_isconnected = send_custom_sh_cmd(
            "network.WLAN(network.AP_IF).active()")
        if ap_isconnected:
            apconf = send_custom_sh_cmd(
                "network.WLAN(network.AP_IF).ifconfig()")
            essid = send_custom_sh_cmd(
                "network.WLAN(network.AP_IF).config('essid')")
            channel = send_custom_sh_cmd(
                "network.WLAN(network.AP_IF).config('channel')")
            auth_mode = send_custom_sh_cmd(
                "network.WLAN(network.AP_IF).config('authmode')")
            mac_addrs = ':'.join([unique_id[i:i+2]
                                  for i in range(0, len(unique_id), 2)])
            print('=' * 70)
            print('{0:^15} | {1:^18} | {2:^15} | {3:^15} '.format(
                'SSID', 'BSSID', 'CHANNEL', 'AUTHMODE'))
            print('{0:^15} | {1:^18} | {2:^15} | {3:^15}'.format(essid,
                                                                 mac_addrs,
                                                                 channel,
                                                                 AUTHMODE_DICT[auth_mode]))
            print('=' * 70)
            print('{0:^15} | {1:^15} | {2:^15} | {3:^15} '.format(
                'IP', 'SUBNET', 'GATEAWAY', 'DNS'))
            print('{0:^15} | {1:^15} | {2:^15} | {3:^15}'.format(*apconf))

        else:
            print('AP interface not connected')

    if cmd_inp == 'apscan':
        enable_sta = send_custom_sh_cmd(
            "network.WLAN(network.AP_IF).active(1)")
        if enable_sta:
            scan = send_custom_sh_cmd(
                "network.WLAN(network.AP_IF).status('stations')")
            print('Found {} device/s'.format(len(scan)))
            for dev in scan:
                bytdev = hexlify(dev[0]).decode()
                mac_ad = ':'.join([bytdev[i:i+2] for i in range(0, len(bytdev),
                                                                2)])
                print('MAC: {}'.format(mac_ad))

        else:
            print("Can't enable AP interface")

    if cmd_inp == 'meminfo':
        RAM = send_custom_sh_cmd(
            'from micropython import mem_info;mem_info()')
        mem_info = RAM.splitlines()[1]
        mem = {elem.strip().split(':')[0]: int(elem.strip().split(':')[
                          1]) for elem in mem_info[4:].split(',')}
        print("{0:12}{1:^12}{2:^12}{3:^12}{4:^12}".format(*['Memmory',
                                                            'Size', 'Used',
                                                            'Avail',
                                                            'Use%']))
        total_mem = mem['total']/1024
        used_mem = mem['used']/1024
        free_mem = mem['free']/1024
        total_mem_s = "{:.3f} KB".format(total_mem)
        used_mem_s = "{:.3f} KB".format(used_mem)
        free_mem_s = "{:.3f} KB".format(free_mem)

        print('{0:12}{1:^12}{2:^12}{3:^12}{4:>8}'.format('RAM', total_mem_s,
                                                          used_mem_s, free_mem_s,
                                                          "{:.1f} %".format((used_mem/total_mem)*100)))

    if cmd_inp.split(' ')[0] == 'install':
        try:
            lib = cmd_inp.split(' ')[1]
            install_resp = send_custom_sh_cmd(
                "import upip;upip.install('{}')".format(lib), sh_silent=False)
        except Exception as e:
            print('Please indicate a library to install')


    if cmd_inp.split(' ')[0] == 'touch':
        try:
            n_file = cmd_inp.split(' ')[1]
            send_custom_sh_cmd("new_file = open('{}','w')".format(n_file))
            send_custom_sh_cmd("new_file.close()")
        except Exception as e:
            print('Please write the name of the new file')

    if cmd_inp.split(' ')[0] == 'edit':
        try:
            e_file = cmd_inp.split(' ')[1]
            print('Press CTRL-E to edit {}'.format(e_file))
            edit_mode['E'] = True
            edit_mode['File'] = e_file
            prompt['p'] = ""
        except Exception as e:
            print('Please write the name of the new file')

    if cmd_inp.split(' ')[0] == 'srepl':
        print('<-- Device {} MicroPython -->'.format(dev_platform))
        print('Use CTRL-a,CTRL-x to exit')
        try:
            shr_cp.enable_wrepl_io()
        except Exception as e:
            pass
        serial_repl_cmd_str = 'picocom {} -b115200'.format(args.p)
        serial_repl_cmd = shlex.split(serial_repl_cmd_str)
        try:
            serial_repl = subprocess.call(serial_repl_cmd)
            try:
                shr_cp.enable_wrepl_io()
            except Exception as e:
                pass
        except KeyboardInterrupt:
            try:
                shr_cp.enable_wrepl_io()
            except Exception as e:
                pass
            pass
            print('')

    if cmd_inp == 'whoami':
        print('DEVICE: {}, SERIAL PORT: {} , BAUDRATE: {},  ID: {}'.format(host_name, args.p, args.t, unique_id))
        sysinfo = send_custom_sh_cmd('os.uname()', capture_output=True)
        dev_info = sysinfo.split("'")
        print('SYSTEM NAME: {}'.format(dev_info[1]))
        print('NODE NAME: {}'.format(dev_info[3]))
        print('RELEASE: {}'.format(dev_info[5]))
        print('VERSION: {}'.format(dev_info[7]))
        print('MACHINE: {}'.format(dev_info[9]))
        # print(sysinfo)

    if cmd_inp == 'pwdl':
        print(os.getcwd())

    if cmd_inp.split(' ')[0] == 'cdl':
        if len(cmd_inp.split(' ')) == 1:
            os.chdir(os.environ['HOME'])
        else:
            os.chdir('{}'.format(cmd_inp.split(' ')[1]))
        if os.getcwd() == os.environ['HOME']:
            local_path['p'] = '~:/'
            if not show_local_path['s']:
                local_path['p'] = ''
        elif not show_local_path['s']:
            local_path['p'] = ''
        else:
            local_path['p'] = os.getcwd().split('/')[-1]+':/'

        # SET ROOT USER PATH:
        shell_message = [
            ('class:userpath',    local_path['p']),
            ('class:username', dev_platform),
            ('class:at',       '@'),
            ('class:host',     host_name),
            ('class:colon',    ':'),
            ('class:path',     '~{}'.format(dev_path['p'])),
            ('class:pound',    '$ '),
        ]
        shell_prompt['s'] = shell_message
        prompt['p'] = shell_prompt['s']

    if cmd_inp == 'lsl':
        list_of_files = os.listdir()
        if list_of_files != []:
            print_table(list_of_files, wide=28, format_SH=True)

    if cmd_inp.split(' ')[0] == 'get':
        try:
            print('File/s to get:')
            file = cmd_inp.split(' ')[1]
            if file == 'cwd':
                shr_cp.sh_repl("[file for file in os.listdir() if not os.stat(file)[0] & 0x4000]")
                espdev.get_output()
                files_to_put = espdev.output
                for file in files_to_put:
                    print('- {}'.format(file))
                shr_htool.start_SOC()
                for file in files_to_put:
                    shr_htool.get(file)
                    time.sleep(0.2)
                # shr_cp.sh_repl("print('Done!')")
                time.sleep(0.2)
                shr_htool.stop_SOC()
            elif '*' in file:
                shr_cp.sh_repl("[file for file in os.listdir() if '{}' in file and not os.stat(file)[0] & 0x4000]".format(file.replace('*', '')))
                espdev.get_output()
                files_to_put_re = espdev.output
                if files_to_put_re is None or files_to_put_re == []:
                    print('No files found that match "{}"'.format(file))
                else:
                    for file in files_to_put_re:
                        print('- {}'.format(file))
                    shr_htool.start_SOC()
                    for file in files_to_put_re:
                        shr_htool.get(file)
                        time.sleep(0.2)
                    # shr_cp.sh_repl("print('Done!')")
                    time.sleep(0.2)
                    shr_htool.stop_SOC()
            elif len(cmd_inp.split(' ')) > 2:
                files_to_put = cmd_inp.split(' ')[1:]
                # files_to_put_dir = [fl for fl in os.listdir() if os.path.isfile(fl) and fl in files_to_put]
                shr_cp.sh_repl("[file for file in os.listdir() if file in {} and not os.stat(file)[0] & 0x4000]".format(files_to_put))
                espdev.get_output()
                files_to_put_dir = espdev.output
                if len(files_to_put_dir) != len(files_to_put):
                    print('One or more files not found in cwd')
                else:
                    for file in files_to_put_dir:
                        print('- {}'.format(file))
                    shr_htool.start_SOC()
                    for file in files_to_put_dir:
                        shr_htool.get(file)
                        time.sleep(0.2)
                    # shr_cp.sh_repl("print('Done!')")
                    time.sleep(0.2)
                    shr_htool.stop_SOC()
            else:
                print('- {}'.format(file))
                # shr_htool.start_SOC()
                shr_htool.get(file)
                # shr_cp.sh_repl("print('Done!')")
                time.sleep(0.2)
                shr_htool.stop_SOC()
        except Exception as e:
            print(e)
            print('Indicate a file to put')

    # if cmd_inp.split(' ')[0] == 'sync':
    #     print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
    #     try:
    #         shr_cp.enable_wrepl_io()
    #     except Exception as e:
    #         pass
    #     if dev_path['p'] != ' ':
    #         dev_dir = "-dir {}".format('/'.join(dev_path['p'].split('/')[1:]))
    #     else:
    #         dev_dir = ''
    #     try:
    #         file = cmd_inp.split(' ')[1]  #  HERE CALL ONE OR -FRE
    #         if file == 'cwd':
    #             get_str = 'upydev sync -fre {} -t {} -p {} {}'.format(
    #                 file, args.t, args.p, dev_dir)
    #         elif '*' in file:
    #             get_str = 'upydev sync -fre {} -t {} -p {} {}'.format(
    #                 file.replace('*', ''), args.t, args.p, dev_dir)
    #         elif len(cmd_inp.split(' ')) > 2:
    #             file = ' '.join(cmd_inp.split(' ')[1:])
    #             get_str = 'upydev sync -fre {} -t {} -p {} {}'.format(
    #                 file, args.t, args.p, dev_dir)
    #         else:
    #             get_str = 'upydev sync -f {} -t {} -p {} {}'.format(
    #                 file, args.t, args.p, dev_dir)
    #
    #         sync_cmd = shlex.split(get_str)
    #         try:
    #             get_file = subprocess.call(sync_cmd)
    #             try:
    #                 shr_cp.enable_wrepl_io()
    #             except Exception as e:
    #                 pass
    #         except KeyboardInterrupt:
    #             try:
    #                 shr_cp.enable_wrepl_io()
    #             except Exception as e:
    #                 pass
    #             pass
    #             print('Operation aborted')
    #     except Exception as e:
    #         print('Indicate a file to get')

    if cmd_inp.split(' ')[0] == 'put':
        try:
            print('File/s to put:')
            file = cmd_inp.split(' ')[1]
            if file == 'cwd':
                files_to_put = [fl for fl in os.listdir() if os.path.isfile(fl)]
                for file in files_to_put:
                    print('- {}'.format(file))
                shr_htool.start_SOC()
                for file in files_to_put:
                    shr_htool.put(file)
                    time.sleep(0.2)
                # shr_cp.sh_repl("print('Done!')")
                time.sleep(0.2)
                shr_htool.stop_SOC()
            elif '*' in file:
                files_to_put = [fl for fl in os.listdir() if os.path.isfile(fl)]
                files_to_put_re = [fl for fl in files_to_put if file.replace('*', '') in fl]
                for file in files_to_put_re:
                    print('- {}'.format(file))
                shr_htool.start_SOC()
                for file in files_to_put_re:
                    shr_htool.put(file)
                    time.sleep(0.2)
                # shr_cp.sh_repl("print('Done!')")
                time.sleep(0.2)
                shr_htool.stop_SOC()
            elif len(cmd_inp.split(' ')) > 2:
                files_to_put = cmd_inp.split(' ')[1:]
                files_to_put_dir = [fl for fl in os.listdir() if os.path.isfile(fl) and fl in files_to_put]
                if len(files_to_put_dir) != len(files_to_put):
                    print('One or more files not found in cwd')
                else:
                    for file in files_to_put_dir:
                        print('- {}'.format(file))
                    shr_htool.start_SOC()
                    for file in files_to_put_dir:
                        shr_htool.put(file)
                        time.sleep(0.2)
                    # shr_cp.sh_repl("print('Done!')")
                    time.sleep(0.2)
                    shr_htool.stop_SOC()
            else:
                print('- {}'.format(file))
                # shr_htool.start_SOC()
                shr_htool.put(file)
                # shr_cp.sh_repl("print('Done!')")
                time.sleep(0.2)
                shr_htool.stop_SOC()
        except Exception as e:
            # print(e)
            print('Indicate a file to put')
    if cmd_inp.split(' ')[0] == 'catl':
        try:
            file = cmd_inp.split(' ')[1]
            try:
                with open(file, 'r') as catfile:
                    content = catfile.read()
                    print(content)
            except Exception as e:
                print(e)
                pass

        except Exception as e:
            print('Indicate a file to show')

    # if cmd_inp.split(' ')[0] == 'd_sync':
    #     print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
    #     try:
    #         shr_cp.enable_wrepl_io()
    #     except Exception as e:
    #         pass
    #     if dev_path['p'] != ' ':
    #         dev_dir = "-dir {}".format('/'.join(dev_path['p'].split('/')[1:]))
    #     else:
    #         dev_dir = ''
    #     try:
    #         dir_to_sync = cmd_inp.split(' ')[1]  #  HERE CALL ONE OR -FRE
    #
    #         d_sync_str = 'upydev d_sync -tree -t {} -p {} -dir {} -rst f'.format(args.t,
    #                                                                    args.p,
    #                                                                    dir_to_sync)
    #
    #         d_sync_cmd = shlex.split(d_sync_str)
    #         try:
    #             dir_sync = subprocess.call(d_sync_cmd)
    #             try:
    #                 shr_cp.enable_wrepl_io()
    #             except Exception as e:
    #                 pass
    #         except KeyboardInterrupt:
    #             try:
    #                 shr_cp.enable_wrepl_io()
    #             except Exception as e:
    #                 pass
    #             pass
    #             print('Operation aborted')
    #     except Exception as e:
    #         print('Indicate a file to get')

    if cmd_inp.split(' ')[0] == 'l_micropython':
        print('<-- Local machine MicroPython -->')
        try:
            lmpy = 'micropython'
            lmpy_cmd = shlex.split(lmpy)
            old_action = signal.signal(signal.SIGINT, signal.SIG_IGN)

            def preexec_function(action=old_action):
                signal.signal(signal.SIGINT, action)
            try:
                l_mpy = subprocess.call(lmpy_cmd, preexec_fn=preexec_function)
                signal.signal(signal.SIGINT, old_action)
            except Exception as e:
                print(e)
                print('Local MicroPython not found')
        except Exception as e:
            print(e)
            pass

    if cmd_inp == 'ls':
        dev_files = send_custom_sh_cmd('os.listdir();gc.collect()')
        try:
            print_table(dev_files, wide=28, format_SH=True)
        except Exception as e:
            pass

    if cmd_inp.split(' ')[0] == 'python':
        print('<-- Local machine Python -->')
        print('Use Ctrl-D to exit')
        try:
            lpy = 'python'
            lpy_cmd = shlex.split(lpy)
            old_action = signal.signal(signal.SIGINT, signal.SIG_IGN)

            def preexec_function(action=old_action):
                signal.signal(signal.SIGINT, action)
            try:
                l_py = subprocess.call(lpy_cmd, preexec_fn=preexec_function)
                signal.signal(signal.SIGINT, old_action)
            except Exception as e:
                print(e)
                print('Local Python not found')
        except Exception as e:
            print(e)
            pass

    if cmd_inp.split(' ')[0] == 'vim':
        print('<-- Vim editor -->')
        print('Use I to insert; (ESC , :w) to write; (ESC ,:q) to exit')
        time.sleep(2)
        try:
            file_to_edit = cmd_inp.split(' ')[1]
            lvim = 'vim {}'.format(file_to_edit)
            lvim_cmd = shlex.split(lvim)
            try:
                l_vim = subprocess.call(lvim_cmd)
            except Exception as e:
                print(e)
                print('Vim editor not found')
        except Exception as e:
            print('Indicate a file to edit')
            pass

    if cmd_inp == 'set_localtime':
        print('Setting local time: {}'.format(
            datetime.now().strftime("%Y-%m-%d T %H:%M:%S")))
        rtc_cmd = "from machine import RTC;rtc = RTC();"
        wkoy = date.today().isocalendar()[1]
        datetime_local = [val for val in datetime.now().timetuple()[:-3]]
        datetime_tuple = datetime_local[:3]
        datetime_tuple.append(wkoy)
        datetime_final = datetime_tuple + datetime_local[3:] + [0]
        utc_zone_cmd = "rtc.datetime(({}, {}, {}, {}, {}, {}, {}+3, {}))".format(
            *datetime_final)
        settime_cmd = rtc_cmd + utc_zone_cmd
        send_custom_sh_cmd(settime_cmd)

    if cmd_inp.split(' ')[0] == 'tree':
        if len(cmd_inp.split(' ')) > 1:
            if cmd_inp.split(' ')[1] == '-a':
                shr_cp.sh_repl('from upysh2 import tree;tree(hidden=True);gc.collect()', follow=True)
        else:
        # send_custom_sh_cmd('from upysh2 import tree;tree()', sh_silent=False)
            shr_cp.sh_repl('from upysh2 import tree;tree();gc.collect()', follow=True)
        shr_cp.flush_conn()

    if cmd_inp == 'l_ifconfig':
        ifconfig_local = netifaces.ifaddresses('en0')[netifaces.AF_INET][0]
        gateaway = list(netifaces.gateways()['default'].values())[0][0]
        print("('{}', '{}', '{}', '{}')".format(ifconfig_local['addr'],
                                                ifconfig_local['netmask'],
                                                gateaway,
                                                ifconfig_local['broadcast']))

    if cmd_inp == 'l_ifconfig_t':
        if _platform == "linux" or _platform == "linux2":
            # linux
            return ''
        elif _platform == "darwin":
            # MAC OS X
            scanoutput = subprocess.check_output(["airport", "-I"])
            wifi_info = [data.strip()
                         for data in scanoutput.decode('utf-8').split('\n')]
            wifi_info_dic = {data.split(':')[0]: data.split(
                ':')[1].strip() for data in wifi_info[:-1]}
        ssid = wifi_info_dic['SSID']
        rssi = wifi_info_dic['agrCtlRSSI']
        ifconfig_local = netifaces.ifaddresses('en0')[netifaces.AF_INET][0]
        gateaway = list(netifaces.gateways()['default'].values())[0][0]
        print('=' * 106)
        print('{0:^15} | {1:^15} | {2:^15} | {3:^15} | {4:^15} | {5:^15} '.format(
            'IP', 'SUBNET', 'GATEAWAY', 'DNS', 'SSID', 'RSSI'))
        print('{0:^15} | {1:^15} | {2:^15} | {3:^15} | {4:^15} | {5:^15} '.format(ifconfig_local['addr'],
                                                                                  ifconfig_local['netmask'],
                                                                                  gateaway,
                                                                                  ifconfig_local['broadcast'],
                                                                                  ssid,
                                                                                  rssi))
    # if cmd_inp == 'lsof':
    #     nm_scan = nmap.PortScanner()
    #     print('Scanning TCP ports...')
    #     nm_scan.scan(hosts=args.t, arguments='-p 1-10000')
    #     result = nm_scan[args.t]['tcp']
    #     ports = list(result.keys())
    #     print('Found {} open port/s'.format(len(ports)))
    #     print('{0:8}  {1:^15}  {2:^15} '.format(
    #         'PORT', 'STATE', 'SERVICE'))
    #     for port in ports:
    #         print('{0:<8}  {1:^15}  {2:^15} '.format(port,
    #                                                      result[port]['state'],
    #                                                      result[port]['name']))

    if cmd_inp.split(' ')[0] == 'reload':
        try:
            module = cmd_inp.split(' ')[1].split('.py')[0]
        except Exception as e:
            print('Indicate a module to reload')
        espdev.output = None
        reload_cmd = "import sys;del(sys.modules['{}']);gc.collect()".format(module)
        shr_cp.sh_repl(reload_cmd)
        if espdev.output is not None:
            print(espdev.output)

    if cmd_inp.split(' ')[0] == 'docs':
        if len(cmd_inp.split(' ')) > 1:
            key_word = cmd_inp.split(' ')[1]
            search = 'http://docs.micropython.org/en/latest/search.html?q={}&check_keywords=yes&area=default'.format(key_word)
            webbrowser.open(search)
        else:
            webbrowser.open('http://docs.micropython.org/en/latest/')

    if cmd_inp == 'flush_soc':
        shr_cp.flush_conn()

    # if cmd_inp == 'getcert':
    #     if encrypted_flag['sec']:
    #         for key in shr_cp.client_cert.keys():
    #             value = shr_cp.client_cert[key]
    #             try:
    #                 if not isinstance(value, str):
    #                     if len(value) > 1:
    #                         print('{}:'.format(key.upper()))
    #                         for val in value:
    #                             if len(val) == 1:
    #                                 print('- {} : {}'.format(*val[0]))
    #                             else:
    #                                 print('- {} : {}'.format(*val))
    #                 else:
    #                     print("{}: {}".format(key.upper(), shr_cp.client_cert[key]))
    #             except Exception as e:
    #                 print("{}: {}".format(key.upper(), shr_cp.client_cert[key]))

    if cmd_inp == 'get_rawbuff':
        print(shr_cp.raw_buff)

    if cmd_inp == 'l_tree':
        try:
            l_tree_cmd = 'tree'
            lt_cmd = shlex.split(l_tree_cmd)
            try:
                l_t = subprocess.call(lt_cmd)
            except Exception as e:
                print(e)
                pass
        except Exception as e:
            print(e)
            pass

    if cmd_inp.split(' ')[0] == 'view':
        try:
            img_file = cmd_inp.split(' ')[1]
            if img_file.endswith('.pbm'):
                shr_cp.sh_repl("cat('{}')".format(img_file), dec=False)
                dimensions, data, raw_bin_img, img_1 = read_pbm(shr_cp.raw_buff)
                print("{} dimensions: {}x{}".format(img_file, *dimensions))
                print(img_1)
            else:
                print('Indicate a ".pbm" file to view')
        except Exception as e:
            print(e)
            print('Indicate a ".pbm" file to view')

    if cmd_inp.split(' ')[0] == 'bat':
        try:
            color_file = cmd_inp.split(' ')[1]
            if color_file.endswith('.py'):
                shr_cp.sh_repl("cat('{}')".format(color_file), silent=False, color=True)
            else:
                print('Indicate a ".py" file to view')
        except Exception as e:
            print(e)
            print('Indicate a ".py" file to view')

    if cmd_inp.split(' ')[0] == 'rcat':
        try:
            raw_file = cmd_inp.split(' ')[1]
            shr_cp.sh_repl("cat('{}')".format(raw_file), dec=False)
            print(b''.join(shr_cp.raw_buff.splitlines()[1:-1]).replace(b'>>> ', b''))
        except Exception as e:
            print(e)

    if cmd_inp.split(' ')[0] == 'exit':
        if reset_flag['R']:
            if len(cmd_inp.split(' ')) > 1:
                if cmd_inp.split(' ')[1] == '-nr':
                    # shr_cp.sh_repl("ssl_client.switch_wrepl()")
                    time.sleep(0.5)
                elif cmd_inp.split(' ')[1] == '-hr':
                    print('Device Hard Reset...')
                    shr_cp.send_message("import machine;machine.reset()\r")
                    print('Done!')
            else:
                print('Rebooting device...')
                shr_cp.send_message("\x04")
                print('Done!')
        else:
            # shr_cp.sh_repl("ssl_client.switch_wrepl()")
            time.sleep(0.5)
        espdev.serial.close()
        exit_flag['exit'] = True


@Condition
def autosuggest_is_on():
    return autosuggest['A']


# KEYBINDINGS

@kb.add('c-x')
def exitpress(event):
    " Exit SHELL-REPL terminal "
    if shell_mode['S']:
        print('\nclosing...')
    else:
        print('\n>>> closing...')
    if reset_flag['R']:
        print('Rebooting device...')
        shr_cp.send_message("\x04")
        print('Done!')
    else:
        shr_cp.sh_repl("ssl_client.switch_wrepl()")
        time.sleep(0.5)
    espdev.serial.close()
    exit_flag['exit'] = True
    event.app.exit()


@kb.add('c-b')
def bannerpress(event):
    "Prints MicroPython sys version info"
    def upy_sysversion():
        shr_cp.sh_repl("\x02", banner=True)

    run_in_terminal(upy_sysversion)


@kb.add('c-k')
def see_cmd_info_press(event):
    "CTRL-Commands info"
    # event.app.current_buffer.insert_text('import')
    def cmd_info():
        print(kb_info)
        if shell_mode['S']:
            print(shell_commands_info)
    run_in_terminal(cmd_info)


@kb.add('c-r')
def flush_line(event):
    event.app.current_buffer.reset()


@kb.add('c-o')
def cmd_ls(event):
    def ls_out():
        espdev.output = None
        print('>>> ls')
        shr_cp.sh_repl('ls();gc.collect()')
        if espdev.output is not None:
            print(espdev.output)

    run_in_terminal(ls_out)


@kb.add('c-n')
def cmd_mem_info(event):

    def mem_info_out():
        espdev.output = None
        print('>>> mem_info()')
        shr_cp.sh_repl('from micropython import mem_info; mem_info()')
        if espdev.output is not None:
            print(espdev.output)

    run_in_terminal(mem_info_out)


@kb.add('c-y')
def cmd_gc_collect(event):

    def gc_collect_out():
        espdev.output = None
        print('>>> gc.collect()')
        shr_cp.sh_repl('import gc;gc.collect()')
        if espdev.output is not None:
            print(espdev.output)

    run_in_terminal(gc_collect_out)


@kb.add('c-f')
def toggle_autosgst(event):
    if autosuggest['A']:
        autosuggest['A'] = False
    else:
        autosuggest['A'] = True


@kb.add('c-space')
def autocomppress(event):
    "Send last command"

    def print_last():
        espdev.output = None
        last_cmd = event.app.current_buffer.history.get_strings()[-1]
        if shell_mode['S']:
            if local_path['p'] == '':
                g_p = [val[1] for val in prompt['p'][1:5]]
                b_p = [val[1] for val in prompt['p'][5:]]
                color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                    "".join(g_p[:-1]), "".join(b_p), last_cmd)
                print(color_p)
            else:
                m_p = [prompt['p'][0][1]]
                g_p = [val[1] for val in prompt['p'][1:5]]
                b_p = [val[1] for val in prompt['p'][5:]]
                color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                    "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                print(color_p)
            if last_cmd in custom_sh_cmd_kw:
                custom_sh_cmd(last_cmd)
            else:
                inp, frst_cmd = map_upysh(last_cmd)
                if frst_cmd == 'run':

                    try:
                        shr_cp.sh_repl(inp, follow=True)
                        # if espdev.output is not None:
                        #     print(espdev.output)
                    except KeyboardInterrupt:
                        time.sleep(0.2)
                        shr_cp.sh_repl('\x03', silent=True,
                                        traceback=True)  # KBI
                        time.sleep(1)
                        for i in range(3):
                            shr_cp.sh_repl('\x0d', silent=True)
                            shr_cp.flush_conn()
                        pass
                else:

                    if inp == 'ls()' or frst_cmd == 'tree' or frst_cmd == 'cat':
                        shr_cp.sh_repl(inp+';gc.collect()', follow=True)
                        # shr_cp.send_message('gc.collect()\r')
                    else:
                        shr_cp.sh_repl(inp)
                        if espdev.output is not None:
                            print(espdev.output)
        else:
            print('>>> {}'.format(last_cmd))
            try:
                shr_cp.sh_repl(last_cmd, follow=True)
                # if espdev.output is not None:
                #     print(espdev.output)
            except KeyboardInterrupt:
                time.sleep(0.2)
                shr_cp.sh_repl('\x03', silent=True,
                                traceback=True)  # KBI
                time.sleep(0.2)
                for i in range(1):
                    shr_cp.sh_repl('\x0d', silent=True)
                    shr_cp.flush_conn()
                pass

    run_in_terminal(print_last)


@kb.add('c-t')
def testpress(event):
    "Test code command"
    def test_code():
        espdev.output = None
        test_cmd = 'import test_code'
        print('>>> {}'.format(test_cmd))
        try:
            shr_cp.sh_repl(test_cmd, follow=True)
            # if espdev.output is not None:
            #     print(espdev.output)
        except KeyboardInterrupt:
            time.sleep(0.2)
            shr_cp.sh_repl('\x03', silent=True,
                            traceback=True)  # KBI
            time.sleep(0.2)
            for i in range(1):
                shr_cp.sh_repl('\x0d', silent=True)
                shr_cp.flush_conn()
            pass

    run_in_terminal(test_code)


@kb.add('c-w')
def reloadpress(event):
    "Reload test_code command"
    def reload_code():
        espdev.output = None
        reload_cmd = "import sys;del(sys.modules['test_code']);gc.collect()"
        print('>>> {}'.format(reload_cmd))

        shr_cp.sh_repl(reload_cmd)
        if espdev.output is not None:
            print(espdev.output)
    if not edit_mode['E']:
        run_in_terminal(reload_code)
    else:
        edit_mode['E'] = False


@kb.add('c-a')
def reset_cursor(event):
    "Move cursor to init position"
    buff_text = event.app.current_buffer.document.text
    event.app.current_buffer.reset()
    event.app.current_buffer.insert_text(buff_text, move_cursor=False)


@kb.add('c-s')
def toggle_shell_mode(event):
    if shell_mode['S']:
        prompt['p'] = d_prompt
        shell_mode['S'] = False
    else:
        prompt['p'] = shell_prompt['s']
        shell_mode['S'] = True
        shr_cp.sh_repl('from upysh import *', no_raw_buff=True)
        espdev.output = None


@kb.add('c-c')
def send_KBI(event):
    # def send_KBI():
    try:
        last_cmd = ''
        if edit_mode['E']:
            prompt['p'] = shell_prompt['s']
            event.app.current_buffer.reset()
        else:

            shr_cp.sh_repl('\x03') # KBI

            paste_flag['p'] = False
            if not shell_mode['S']:
                prompt['p'] = d_prompt
                last_cmd = event.app.current_buffer.document.text
            if shell_mode['S']:
                print('^C')
            event.app.current_buffer.reset()
    except Exception as e:
        pass

    def cmd_kbi(command=last_cmd):
        if prompt['p'] == ">>> ":
            print(prompt['p'] + command)
        elif prompt['p'] == shell_prompt['s']:
            if edit_mode['E'] is True:
                edit_mode['E'] = False
                print("<-----Edition Cancelled---->")
                print('Press ESC, ENTER to exit and return to shell')
    run_in_terminal(cmd_kbi)


@kb.add('c-e')
def paste_mode(event):
    "PASTE MODE IN REPL, EDIT MODE IN SHELL MODE"
    if not shell_mode['S']:
        paste_flag['p'] = True
        event.app.current_buffer.reset()
        # event.app.current_buffer.insert_text('import')

        def cmd_paste():
            print(
                'paste mode; Ctrl-C to cancel, Ctrl-D to finish, Then ESC, ENTER to exit paste mode')
            print("===")
            prompt['p'] = ""
        run_in_terminal(cmd_paste)
    else:
        if edit_mode['E']:
            def cmd_paste():
                print('Edit Mode; Ctrl-C to cancel, Ctrl-D to finish, then ESC, ENTER')
                print("#<---- File: {} --->".format(edit_mode['File']))
                prompt['p'] = ""
            run_in_terminal(cmd_paste)
            event.app.current_buffer.reset()
            espdev.long_output = []
            espdev.output = None
            espdev.response = ''
            shr_cp.sh_repl("cat('{}')".format(edit_mode['File']))
            file_content = espdev.output
            try:
                process_content = file_content
            except Exception as e:
                process_content = ' '
            if isinstance(file_content, str):
                event.app.current_buffer.insert_text(process_content)


@kb.add('c-d')
def paste_mode_exit(event):
    "PASTE MODE exit in REPL, EDIT MODE exit in SHELL"
    # event.app.current_buffer.insert_text('import')

    def cmd_paste_exit(buff_P=paste_buffer['B']):
            buff_text = '\n'.join(
                buff_P + [event.app.current_buffer.document.text])
            if buff_text is not None and buff_text != '':
                try:
                    shr_cp.paste_buff(buff_text)
                    shr_cp.sh_repl("\x04", follow=True)
                except KeyboardInterrupt:
                    time.sleep(0.2)
                    shr_cp.sh_repl('\x03', silent=True,
                                    traceback=True)  # KBI
                    time.sleep(0.2)
                    for i in range(1):
                        shr_cp.sh_repl('\x0d', silent=True)
                        shr_cp.flush_conn()
                    pass
                espdev.output = None
                time.sleep(1)
                shr_cp.flush_conn()
                shr_cp.flush_conn()

            event.app.current_buffer.reset()

            print(">>> ")
            prompt['p'] = ">>> "

    def cmd_edit_exit(file_to_edit=edit_mode['File']):
            buff_text = ''.join([event.app.current_buffer.document.text])
            if buff_text is not None and buff_text != '':
                send_custom_sh_cmd(
                    "_file_to_edit = open('{}','w')".format(file_to_edit))
                send_custom_sh_cmd("nl = '\\n'")
                print('Writing to file...')
                lines_to_write = buff_text.split('\n')
                for line in lines_to_write:
                    if line != ' ' and line != "" and line != '':
                        send_custom_sh_cmd(
                            '_file_to_edit.write("{}")'.format(line))

                        send_custom_sh_cmd('_file_to_edit.write(nl)')

                send_custom_sh_cmd("_file_to_edit.close()")
                send_custom_sh_cmd("gc.collect()")
                time.sleep(0.5)
                shr_cp.flush_conn()
                shr_cp.flush_conn()
            event.app.current_buffer.reset()

            print("File {} edited succesfully!".format(file_to_edit))
            print('PRESS ESC, THEN ENTER TO EXIT EDIT MODE')
    if not shell_mode['S']:
        if paste_flag['p']:
            # print(paste_buffer['B'])
            run_in_terminal(cmd_paste_exit)
            paste_buffer['B'] = ['']
        else:
            shr_cp.sh_repl('\x04', follow=True)
            time.sleep(0.2)
            shr_cp.dev.serial.close()
            time.sleep(1)
            shr_cp.dev.serial.open()
            shr_cp.sh_repl('import os;import gc;from upysh import *', silent=True)
            shr_cp.flush_conn()
        paste_flag['p'] = False
    else:
        if edit_mode['E']:
            run_in_terminal(cmd_edit_exit)
            prompt['p'] = shell_prompt['s']
        edit_mode['E'] = False
    # paste_buffer['B'] = ['']  # RESET PASTE BUFFER


@kb.add('tab')
def tab_press(event):
    "Tab autocompletion info"
    # event.app.current_buffer.insert_text('import')
    if paste_flag['p'] or edit_mode['E']:
        event.app.current_buffer.insert_text('    ')
    else:
        glb = False
        import_cmd = False
        try:
            buff_text_frst_cmd = event.app.current_buffer.document.text.split(' ')[
                                                                              0]
            if buff_text_frst_cmd == 'import' or buff_text_frst_cmd == 'from':
                import_cmd = True
            buff_text = event.app.current_buffer.document.text.split(' ')[-1]
            if isinstance(buff_text, str):
                if '.' in buff_text and not shell_mode['S']:

                    root_text = '.'.join(buff_text.split('.')[:-1])
                    rest = buff_text.split('.')[-1]
                    if rest != '':
                        shr_cp.sh_repl("[val for val in dir({}) if val.startswith('{}')]".format(root_text, rest))
                        shr_cp.flush_conn()

                    else:
                        try:
                            shr_cp.sh_repl('dir({});gc.collect()'.format(root_text),
                                            debug=False, prevent_hang=True)
                        except KeyboardInterrupt:
                            time.sleep(0.2)
                            shr_cp.sh_repl('\x03', silent=True,
                                            traceback=True)

                        shr_cp.flush_conn()

                else:
                    rest = ''
                    glb = True
                    cmd_ls_glb = 'dir()'
                    if buff_text != '':
                        cmd_ls_glb = "[val for val in dir() if val.startswith('{}')]".format(buff_text)
                    if import_cmd:
                        cmd_ls_glb = "[scp.split('.')[0] for scp in os.listdir()+os.listdir('./lib') if '.py' in scp]"
                        if buff_text != '':
                            cmd_ls_glb = "[scp.split('.')[0] for scp in os.listdir()+os.listdir('./lib') if '.py' in scp and scp.startswith('{}')]".format(buff_text)

                    if shell_mode['S']:

                        if '/' in buff_text:
                            glb = False
                            dir_to_list = '/'.join(buff_text.split('/')[:-1])
                            cmd_ls_glb = "os.listdir('{}')".format(dir_to_list)
                            if buff_text.split('/')[-1] != '':
                                cmd_ls_glb = "[val for val in os.listdir('{}') if val.startswith('{}')]".format(dir_to_list, buff_text.split('/')[-1])

                        else:
                            cmd_ls_glb = 'os.listdir()'
                            if buff_text != '':
                                cmd_ls_glb = "[val for val in os.listdir() if val.startswith('{}')]".format(buff_text)

                    try:
                        shr_cp.sh_repl(cmd_ls_glb+';gc.collect()',
                                        debug=False, prevent_hang=True)
                    except KeyboardInterrupt:
                        time.sleep(0.2)
                        shr_cp.sh_repl('\x03', silent=True,
                                        traceback=True)
                    shr_cp.flush_conn()

            else:
                root_text = buff_text.split('.')[0]
                rest = buff_text.split('.')[1]
                try:
                    shr_cp.sh_repl('dir({});gc.collect()'.format(root_text))
                except KeyboardInterrupt:
                    time.sleep(0.2)
                    shr_cp.sh_repl('\x03', silent=True,
                                    traceback=True)
                shr_cp.flush_conn()
        except Exception as e:
            pass

        def tab_cmd_info(rest_part=rest, flag=glb, buff=buff_text):
            try:
                if espdev.output is not None:
                    espdev.get_output()
                    if isinstance(espdev.output, str):
                        # print(espdev.output)
                        pass
                    else:
                        if rest != '':  # print attributes
                            result = [
                                val for val in espdev.output if val.startswith(rest)]
                            if len(result) > 1:
                                comm_part = os.path.commonprefix(result)
                                if comm_part == rest:
                                    print('>>> {}'.format(buff_text))
                                    print_table(result, autowide=True, sort=False)
                                else:
                                    event.app.current_buffer.insert_text(
                                        comm_part[len(rest):])
                            else:
                                event.app.current_buffer.insert_text(
                                    result[0][len(rest):])
                        else:
                            if not glb:
                                if shell_mode['S']:  # print dirs/files
                                    result = [val for val in espdev.output if val.startswith(
                                        buff_text.split('/')[-1])]
                                    if len(result) > 1:
                                        comm_part = os.path.commonprefix(result)
                                        if comm_part == buff_text.split('/')[-1]:
                                            if shell_mode['S']:
                                                last_cmd = event.app.current_buffer.document.text
                                                if local_path['p'] == '':
                                                    g_p = [val[1]
                                                           for val in prompt['p'][1:5]]
                                                    b_p = [val[1]
                                                           for val in prompt['p'][5:]]
                                                    color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                        "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                    print(color_p)
                                                else:
                                                    m_p = [prompt['p'][0][1]]
                                                    g_p = [val[1]
                                                           for val in prompt['p'][1:5]]
                                                    b_p = [val[1]
                                                           for val in prompt['p'][5:]]
                                                    color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                        "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                    print(color_p)
                                                # format ouput
                                                print_table(
                                                    result, wide=28, format_SH=True)
                                        else:
                                            event.app.current_buffer.insert_text(
                                                comm_part[len(buff_text.split('/')[-1]):])
                                    else:
                                        event.app.current_buffer.insert_text(
                                            result[0][len(buff_text.split('/')[-1]):])
                                else:
                                    print('>>> {}'.format(buff_text))   # dir var
                                    print_table(espdev.output, wide=16, autocol_tab=True, sort=False)
                            else:
                                result = [
                                    val for val in espdev.output if val.startswith(buff_text)]
                                if len(result) > 1:
                                    comm_part = os.path.commonprefix(result)
                                    if comm_part == buff_text:
                                        if shell_mode['S']:
                                            last_cmd = event.app.current_buffer.document.text
                                            if local_path['p'] == '':
                                                g_p = [val[1]
                                                       for val in prompt['p'][1:5]]
                                                b_p = [val[1]
                                                       for val in prompt['p'][5:]]
                                                color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                    "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                print(color_p)
                                            else:
                                                m_p = [prompt['p'][0][1]]
                                                g_p = [val[1]
                                                       for val in prompt['p'][1:5]]
                                                b_p = [val[1]
                                                       for val in prompt['p'][5:]]
                                                color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                    "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                print(color_p)
                                            # format ouput
                                            print_table(
                                                result, wide=28, format_SH=True)
                                        else:
                                            print('>>> {}'.format(buff_text))  # globals
                                            print_table(result, wide=16, autowide=True)
                                    else:
                                        event.app.current_buffer.insert_text(
                                            comm_part[len(buff_text):])
                                else:
                                    event.app.current_buffer.insert_text(
                                        result[0][len(buff_text):])

            except Exception as e:
                # print(e)
                pass
        run_in_terminal(tab_cmd_info)


@kb.add('s-tab')
def shif_tab(event):
    "Autocomplete shell commands"
    def autocomplete_sh_cmd():
        if shell_mode['S']:
            buff_text = event.app.current_buffer.document.text
            result = [sh_cmd for sh_cmd in shell_commands +
                      custom_sh_cmd_kw if sh_cmd.startswith(buff_text)]
            if len(result) > 1:
                comm_part = os.path.commonprefix(result)
                if comm_part == buff_text:
                    last_cmd = buff_text
                    if local_path['p'] == '':
                        g_p = [val[1] for val in prompt['p'][1:5]]
                        b_p = [val[1] for val in prompt['p'][5:]]
                        color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                            "".join(g_p[:-1]), "".join(b_p), last_cmd)
                        print(color_p)
                    else:
                        m_p = [prompt['p'][0][1]]
                        g_p = [val[1] for val in prompt['p'][1:5]]
                        b_p = [val[1] for val in prompt['p'][5:]]
                        color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                            "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                        print(color_p)

                    print_table(result)
                else:
                    event.app.current_buffer.insert_text(
                        comm_part[len(buff_text):])
            else:
                if len(result) > 0:
                    event.app.current_buffer.insert_text(
                        result[0][len(buff_text):])

    run_in_terminal(autocomplete_sh_cmd)


@kb.add('s-right')
def autocomplete_locals(event):
    glb = False
    if shell_mode['S']:
        try:
            buff_text_frst_cmd = event.app.current_buffer.document.text.split(' ')[
                                                                              0]
            buff_text = event.app.current_buffer.document.text.split(' ')[-1]
            if isinstance(buff_text, str):
                if '/' in buff_text:
                    root_text = '/'.join(buff_text.split('/')[:-1])
                    rest = buff_text.split('/')[-1]
                    if shell_mode['S']:
                        cmd_ls_glb = os.listdir(root_text)
                else:
                    rest = ''
                    glb = True
                    if shell_mode['S']:
                            cmd_ls_glb = os.listdir()
            else:
                pass
        except Exception as e:
            pass

        def s_r_cmd_info(rest_part=rest, flag=glb, buff=buff_text, output=cmd_ls_glb):
            try:
                    if isinstance(cmd_ls_glb, str):
                        # print(espdev.output)
                        pass
                    else:
                        if rest != '':
                            result = [
                                val for val in output if val.startswith(rest)]
                            if len(result) > 1:
                                comm_part = os.path.commonprefix(result)
                                if comm_part == rest:
                                    if shell_mode['S']:
                                        last_cmd = event.app.current_buffer.document.text
                                        if local_path['p'] == '':
                                            g_p = [val[1]
                                                   for val in prompt['p'][1:5]]
                                            b_p = [val[1]
                                                   for val in prompt['p'][5:]]
                                            color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                            print(color_p)
                                        else:
                                            m_p = [prompt['p'][0][1]]
                                            g_p = [val[1]
                                                   for val in prompt['p'][1:5]]
                                            b_p = [val[1]
                                                   for val in prompt['p'][5:]]
                                            color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                            print(color_p)
                                    print_table(result, wide=28, format_SH=True)
                                else:
                                    event.app.current_buffer.insert_text(
                                        comm_part[len(rest):])
                            else:
                                event.app.current_buffer.insert_text(
                                    result[0][len(rest):])
                        else:
                            if not glb:
                                if shell_mode['S']:
                                    result = [val for val in output if val.startswith(
                                        buff_text.split('/')[-1])]
                                    if len(result) > 1:
                                        comm_part = os.path.commonprefix(result)
                                        if comm_part == buff_text.split('/')[-1]:
                                            if shell_mode['S']:
                                                last_cmd = event.app.current_buffer.document.text
                                                if local_path['p'] == '':
                                                    g_p = [val[1]
                                                           for val in prompt['p'][1:5]]
                                                    b_p = [val[1]
                                                           for val in prompt['p'][5:]]
                                                    color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                        "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                    print(color_p)
                                                else:
                                                    m_p = [prompt['p'][0][1]]
                                                    g_p = [val[1]
                                                           for val in prompt['p'][1:5]]
                                                    b_p = [val[1]
                                                           for val in prompt['p'][5:]]
                                                    color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                        "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                    print(color_p)
                                                # format ouput
                                                print_table(
                                                    result, wide=28, format_SH=True)
                                        else:
                                            event.app.current_buffer.insert_text(
                                                comm_part[len(buff_text.split('/')[-1]):])
                                    else:
                                        event.app.current_buffer.insert_text(
                                            result[0][len(buff_text.split('/')[-1]):])
                                else:
                                    print_table(output, wide=28, format_SH=True)
                            else:
                                result = [
                                    val for val in output if val.startswith(buff_text)]
                                if len(result) > 1:
                                    comm_part = os.path.commonprefix(result)
                                    if comm_part == buff_text:
                                        if shell_mode['S']:
                                            last_cmd = event.app.current_buffer.document.text
                                            if local_path['p'] == '':
                                                g_p = [val[1]
                                                       for val in prompt['p'][1:5]]
                                                b_p = [val[1]
                                                       for val in prompt['p'][5:]]
                                                color_p = "\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                    "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                print(color_p)
                                            else:
                                                m_p = [prompt['p'][0][1]]
                                                g_p = [val[1]
                                                       for val in prompt['p'][1:5]]
                                                b_p = [val[1]
                                                       for val in prompt['p'][5:]]
                                                color_p = "\u001b[35;1m{}\033[0m\33[32;1m{}\033[0m:\u001b[34;1m{}\033[0m{}".format(
                                                    "".join(m_p), "".join(g_p[:-1]), "".join(b_p), last_cmd)
                                                print(color_p)

                                            # format ouput
                                            print_table(
                                                result, wide=28, format_SH=True)
                                        else:
                                            print('>>> {}'.format(buff_text))
                                            print_table(
                                                result, wide=28, format_SH=True)
                                    else:
                                        event.app.current_buffer.insert_text(
                                            comm_part[len(buff_text):])
                                else:
                                    event.app.current_buffer.insert_text(
                                        result[0][len(buff_text):])

            except Exception as e:
                pass
        run_in_terminal(s_r_cmd_info)


@kb.add('s-left')
def toggle_local_path(event):
    if shell_mode['S']:
        if show_local_path['s']:
            show_local_path['s'] = False
            local_path['p'] = ''

            # SET ROOT USER PATH:
            shell_message = [
                ('class:userpath',    local_path['p']),
                ('class:username', dev_platform),
                ('class:at',       '@'),
                ('class:host',     host_name),
                ('class:colon',    ':'),
                ('class:path',     '~{}'.format(dev_path['p'])),
                ('class:pound',    '$ '),
            ]
            shell_prompt['s'] = shell_message
            prompt['p'] = shell_prompt['s']
        else:
            show_local_path['s'] = True
            local_path['p'] = os.getcwd().split('/')[-1]+':/'
            if os.getcwd() == os.environ['HOME']:
                local_path['p'] = '~:/'

            # SET ROOT USER PATH:
            shell_message = [
                ('class:userpath',    local_path['p']),
                ('class:username', dev_platform),
                ('class:at',       '@'),
                ('class:host',     host_name),
                ('class:colon',    ':'),
                ('class:path',     '~{}'.format(dev_path['p'])),
                ('class:pound',    '$ '),
            ]
            shell_prompt['s'] = shell_message
            prompt['p'] = shell_prompt['s']


# @kb.add('c-u')
# def unsecure_mode(event):
#     "Toggle send unencrypted commands"
#     if not args.nem:
#         status_encryp_msg['S'] = True
#         status_encryp_msg['Toggle'] = False
#         if encrypted_flag['sec']:
#             encrypted_flag['sec'] = False
#             espdev.output = None
#             shr_cp.sw_repl()
#
#             def warning_us():
#                 print(CRED + 'WARNING: ENCRYPTION DISABLED' + CEND)
#             # run_in_terminal(warning_us)
#         else:
#             encrypted_flag['sec'] = True
#             espdev.output = None
#             shr_cp.sw_repl()
#             espdev.output = None
#
#             def warning_sec():
#                 print(CGREEN + 'INFO: ENCRYPTION ENABLED' + CEND)
#         # run_in_terminal(warning_sec)


@kb.add('c-p')
def toggle_status_encryp_msg(event):
    "Toggle Right Encryption status"
    if status_encryp_msg['Toggle']:
        if status_encryp_msg['S']:
            pass
            status_encryp_msg['S'] = False
        else:
            status_encryp_msg['S'] = True


@kb.add('c-g')
def listen_SHELL_REPL(event):
    "Echo SHELL_REPL --> Active listening for timer/hardware interrupts"
    def echo_ouput():
        try:
            buff_text = "import time\nwhile True:\n    time.sleep(1)"
            shr_cp.paste_buff(buff_text)
            shr_cp.sh_repl("\x04", follow=True)
        except KeyboardInterrupt:
            time.sleep(0.2)
            shr_cp.sh_repl('\x03', silent=True,
                            traceback=True)  # KBI
            time.sleep(0.2)
            for i in range(1):
                shr_cp.sh_repl('\x0d', silent=True)
                shr_cp.flush_conn()
            pass
        espdev.output = None
        time.sleep(1)
        shr_cp.flush_conn()
        shr_cp.flush_conn()
        event.app.current_buffer.reset()
    run_in_terminal(echo_ouput)


session_p = PromptSession()

# REPL/SHELL LOOP
repl = True
while repl:
    try:
        espdev.output = None
        if exit_flag['exit']:
            break
        else:
            inp = session_p.prompt(prompt['p'],
                                   auto_suggest=ConditionalAutoSuggest(AutoSuggestFromHistory(),
                                                                       autosuggest_is_on),
                                   key_bindings=kb, style=style_p,
                                   multiline=edit_mode['E'] or paste_flag['p'],
                                   rprompt=get_rprompt)
            status_encryp_msg['S'] = False
            status_encryp_msg['Toggle'] = True
            if inp is not None and inp != '' and not paste_flag['p']:
                # HERE IN SHELL MODE PROCESS INPUT BEFORE SENDING # shlex.split
                if shell_mode['S']:  # SHELL
                    inp_args = shlex.split(inp)
                    if inp_args[0] in custom_sh_cmd_kw:
                        custom_sh_cmd(inp)
                    else:
                        inp, frst_cmd = map_upysh(inp)
                        if frst_cmd == 'run':
                            try:
                                shr_cp.sh_repl(inp, follow=True)
                                # if espdev.output is not None:
                                #     print(espdev.output)
                            except KeyboardInterrupt:
                                time.sleep(0.2)
                                shr_cp.sh_repl('\x03', silent=True,
                                                traceback=True)  # KBI
                                time.sleep(0.2)
                                for i in range(1):
                                    shr_cp.sh_repl('\x0d', silent=True)
                                    shr_cp.flush_conn()
                                pass

                        else:
                            if inp == 'ls()' or frst_cmd == 'cat' or frst_cmd == 'tree' or 'help' in frst_cmd:
                                shr_cp.sh_repl(inp+';gc.collect()', follow=True)
                                shr_cp.flush_conn()
                            else:
                                shr_cp.sh_repl(inp)
                                if espdev.output is not None:
                                    print(espdev.output)
                        if frst_cmd == 'cd':
                            espdev.output = None
                            shr_cp.sh_repl('pwd')
                            if espdev.output is not None:
                                # print(espdev.output)
                                devpath = espdev.output
                                dev_path['p'] = devpath
                                if devpath == '/':
                                    devpath = ' '
                                    dev_path['p'] = ' '
                                shell_message = [
                                    ('class:userpath',    local_path['p']),
                                    ('class:username', dev_platform),
                                    ('class:at',       '@'),
                                    ('class:host',     host_name),
                                    ('class:colon',    ':'),
                                    ('class:path',     '~{}'.format(devpath)),
                                    ('class:pound',    '$ '),
                                ]
                                shell_prompt['s'] = shell_message
                                prompt['p'] = shell_prompt['s']
                else:  # REPL
                    try:
                        shr_cp.sh_repl(inp, follow=True)
                        # if espdev.output is not None:
                        #     print(espdev.output)
                    except KeyboardInterrupt:
                        time.sleep(0.2)
                        shr_cp.sh_repl('\x03', silent=True,
                                       traceback=True)  # KBI
                        time.sleep(0.2)
                        for i in range(1):
                            shr_cp.sh_repl('\x0d', silent=True)
                            shr_cp.flush_conn()
                        pass
                if inp != '':  # PASTE BUFFER
                    pass
                    # paste_buffer['B'].append(inp)
    except Exception as e:
        print(e)
        continue
    except KeyboardInterrupt:
        try:
            shr_cp.flush_conn()
            if hasattr(SERIAL_TOOL, 'serv_soc'):
                shr_htool.stop_SOC()
            continue
        except Exception as e:
            print(e)
            continue
    except EOFError:
            # print('This is EOF ERROR!')
            continue
        # espdev.reset()
        # sys.exit()

# sys.exit()
# EXIT MESSAGE
print('logout')
print('Connection to {} closed.'.format(args.p))
