#!/usr/bin/env python3
from upydevice import W_UPYDEVICE, BASE_WS_DEVICE
from binascii import hexlify
import getpass
import socket
import ssl
import secrets
import hashlib
import string
import upydev
import time
import sys
import os
import signal
import netifaces
import nmap
import ast
import textwrap
import select
import subprocess
import shlex
import webbrowser
import argparse
from upydev import upip_host
from upydev.style_customs import one_darkStyle
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
from pygments import highlight
from pygments.styles import get_style_by_name, get_all_styles
# from pygments.style import Style
from pygments.lexers import Python3Lexer
from pygments.formatters import Terminal256Formatter
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 import print_formatted_text
from prompt_toolkit.shortcuts import (radiolist_dialog, message_dialog,
                                      input_dialog, yes_no_dialog)

# 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 SSL_REPL_server:
    """
    SSL HOST Socket REPL server class
    """

    def __init__(self, dev=None, port=8443, buff=1024, key=None,
                 cert=None, init=True, client_key=None, dev_id = None):
        self.host = self.find_localip()
        self.port = port
        self.serv_soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.serv_soc.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.dev = dev
        self.dev_id = dev_id
        self.addr = None
        self.b_buff = bytearray(buff)
        self.buff = b''
        self.raw_buff = b''
        self.message = b''
        self.prompt = b'>>> '
        self.prompt_seen = True
        self.conn = None
        self.addr_client = None
        self.output = None
        self.context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
        self.key = key
        self.cert = cert
        self.passph = None
        self.version = None
        self.cipher = None
        self.SSL_MODE = True
        self.IO = False
        self.color_style = 'monokai'
        if init is True:
            self.start_SOC()
            # CLIENT AUTH
            # random_token = secrets.token_bytes(32)
            # self.ssl_repl("ssl_repl.client_auth(ssl_repl.load_key(), {})".format(random_token))
            # self.dev.get_output()
            # k_hash_token = self.dev.output
            # k_verify = self.client_auth(client_key, random_token)
            # if args.v:
            #     print('HOST KEY: {}'.format(k_verify))
            #     print('DEVICE KEY: {}'.format(k_hash_token))
            #     print('DEVICE VERIFIED: {}'.format(k_verify == k_hash_token))
            # if k_verify == k_hash_token:
            #     pass
            # else:
            #     print('DEVICE UNVERIFIED, closing connection now')
            #     self.conn.close()
            #     self.serv_soc.close()
            #     sys.exit()

    def client_auth(self, rsa_key, token=None):
        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 = token
        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()
        return hashed_key

    def find_localip(self):
        try:
            ip_soc = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            ip_soc.connect(('8.8.8.8', 1))
            local_ip = ip_soc.getsockname()[0]
            ip_soc.close()
            return local_ip
        except Exception as e:
            return [netifaces.ifaddresses(iface)[netifaces.AF_INET][0]['addr'] for
                        iface in netifaces.interfaces() if netifaces.AF_INET in
                        netifaces.ifaddresses(iface)][-1]

    def start_SOC(self):
        while True:
            try:
                self.serv_soc.bind((self.host, self.port))
                break
            except Exception as e:
                self.port += 1
        self.serv_soc.listen(1)
        self.context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
        # self.context = ssl.create_default_context()
        self.context.set_ciphers('ECDHE-ECDSA-AES128-CCM8')
        # for c in self.context.get_ciphers():
        #     print(c['name'])
        while True:
            try:
                my_p = getpass.getpass(prompt="Enter passphrase for key 'SSL_key{}.pem':".format(self.dev_id), stream=None)
                self.passph = my_p
                self.context.load_cert_chain(keyfile=self.key,
                                             certfile=self.cert,
                                             password=my_p)
                break
            except ssl.SSLError as e:
                # print(e)
                print('Passphrase incorrect, try again...')

        # self.context.options |= ssl.OP_NO_TLSv1 | ssl.OP_NO_TLSv1_1
        # self.context.set_ciphers('EECDH+AESGCM:EDH+AESGCM:AES256+EECDH:AES256+EDH:AES256+SHA256')
        try:
            self.context.verify_mode = ssl.CERT_REQUIRED
            self.context.load_verify_locations(cafile=self.cert)
            # for c in self.context.get_ciphers():
            #     print(c['name'])
            # self.dev.wr_cmd("import ssl_repl;ssl_client=ssl_repl.start('{}', {}, auth=False)".format(self.host, self.port))
            self.dev.wr_cmd("import ssl_repl;ssl_client=ssl_repl.start('{}', {})".format(self.host, self.port))
            if args.wss:
                self.dev.close_wconn()
            self.conn, self.addr_client = self.serv_soc.accept()
            self.conn = self.context.wrap_socket(self.conn, server_side=True)
            self.version = self.conn.version()
            self.cipher = self.conn.cipher()
            self.client_cert = self.conn.getpeercert()
        except ssl.SSLError as e:
            print(e)
            print('\033[91;1mWARNING: CERTIFICATE VERIFY FAILED\033[0m')
            print('Please update the certificate\nwith command "upydev sslgen_key -tfkey"')
            sys.exit()
        if args.v:
            print('SSLWebREPL connection from: {}'.format(self.addr_client))
            print("SSL established. Peer: {}".format(self.client_cert))
        self.conn.settimeout(0)

    def flush(self):
        flushed = 0
        while flushed == 0:
            try:
                self.serv_soc.recv(128)
            except Exception as e:
                flushed = 1
                print('flushed!')

    def flush_conn(self):
        flushed = 0
        while flushed < 2:
            try:
                self.conn.recv()
            except Exception as e:
                flushed += 1

    def send_message(self, message):
        self.conn.sendall(bytes(message, 'utf-8'))

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

    def colored(self, msg):
        if self.color_style == 'one_dark':
            result = highlight(msg, Python3Lexer(), Terminal256Formatter(style=one_darkStyle))
            lines = result.split('\n')
            lines_numbered = ["{}{:4}{} {}".format(DARK_GRAY, i, CEND, lines[i-1]) for i in range(1, len(lines)+1)]
            my_code_numb = '\n'.join(lines_numbered)
            print(my_code_numb)

        else:
            pygm_style = get_style_by_name(self.color_style)
            result = highlight(msg, Python3Lexer(), Terminal256Formatter(style=pygm_style))
            lines = result.split('\n')
            lines_numbered = ["{}{:4}{} {}".format(DARK_GRAY, i, CEND, lines[i-1]) for i in range(1, len(lines)+1)]
            my_code_numb = '\n'.join(lines_numbered)
            print(my_code_numb)

    def sw_repl(self, close_wbrepl=False):
        if self.SSL_MODE:
            self.SSL_MODE = False
            self.dev.close_wconn()
            self.ssl_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_ssl_repl()")
            self.dev.close_wconn()
            self.flush_conn()

    def enable_wrepl_io(self):
        if self.SSL_MODE:
            if not self.IO:
                self.dev.close_wconn()
                self.ssl_repl("ssl_client.switch_wrepl()")
                self.dev.close_wconn()
                self.IO = True
            else:
                self.IO = False
                self.dev.open_wconn(ssl=args.wss, auth=True)
                self.dev.wr_cmd("ssl_client.switch_ssl_repl()")
                self.dev.close_wconn()
                self.flush_conn()
        else:
            if not self.IO:
                self.dev.close_wconn()
                self.IO = True
            else:
                self.IO = False
                self.dev.close_wconn()
                self.dev.open_wconn()

    def ssl_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, 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.1)
        while True:
            try:
                self.message = self.conn.recv()
                if not no_raw_buff:
                    self.raw_buff += self.message
                    # self.raw_buff += b'//:'
                if traceback:
                    self.buff += self.message
                # print('DATA RECEIVED: ', self.message)
                # if '>>> '
                break
            except KeyboardInterrupt:
                if prevent_hang:
                    self.conn.sendall(b'\x03')
                    self.flush_conn()
                break
            except Exception as e:
                if debug:
                    pass
                    # print(e, 1)
                pass
        # CATCH MESSAGE
        n = n
        while True:
            try:
                self.message = self.conn.recv(1024)
                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
                    msg = self.buff.decode().split('\n')[-1]
                    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')") or inp.split(';')[0].endswith(".py')")):
                                    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.message == b'>>> ':
                    break
            # except KeyboardInterrupt:
            #     if prevent_hang:
            #         self.conn.sendall(b'\x03')
            #         self.flush_conn()
            #     break
            except Exception as e:
                if debug:
                    print(e, 2)
                if prevent_hang:
                    time.sleep(0.5)
                    self.conn.sendall(b'\x0d')
                pass
        # WAIT TILL NO DATA AVAILABLE
        while True:
            try:
                self.message = self.conn.recv()
                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 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)
                                    get_list = inp.split(';')[0]
                                    if 'import' in get_list:
                                        get_list = inp.split(';')[1]
                                    true_len = self.ssl_repl("len({})".format(get_list),
                                                             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.ssl_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.ssl_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]+[self.raw_buff.splitlines()[-1].replace(b'>>> ', b'')]).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.conn.sendall(b'\x05')
        lines = long_command.split('\n')
        for line in lines:
            self.flush_conn()
            time.sleep(0.1)
            self.send_message(line+'\n')


class SSL_TOOL:
    def __init__(self, ssl_repl, port=8433, key=None,
                 cert=None):
        self.host = self.find_localip()
        self.port = port
        self.serv_soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.serv_soc.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.ssl_repl = ssl_repl
        self.key = key
        self.cert = cert
        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 find_localip(self):
        try:
            ip_soc = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            ip_soc.connect(('8.8.8.8', 1))
            local_ip = ip_soc.getsockname()[0]
            ip_soc.close()
            return local_ip
        except Exception as e:
            return [netifaces.ifaddresses(iface)[netifaces.AF_INET][0]['addr'] for
                        iface in netifaces.interfaces() if netifaces.AF_INET in
                        netifaces.ifaddresses(iface)][-1]

    def start_SOC(self):
        self.serv_soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.serv_soc.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        while True:
            try:
                self.serv_soc.bind((self.host, self.port))
                break
            except Exception as e:
                self.port += 1
        self.serv_soc.listen(1)
        try:
            self.context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
            self.context.load_cert_chain(keyfile=self.key, certfile=self.cert,
                                         password=self.ssl_repl.passph)
        except ssl.SSLError as e:
            print('Passphrase incorrect, try again...')

        # self.context.options |= ssl.OP_NO_TLSv1 | ssl.OP_NO_TLSv1_1
        # self.context.set_ciphers('EECDH+AESGCM:EDH+AESGCM:AES256+EECDH:AES256+EDH:AES256+SHA256')
        self.context.set_ciphers('ECDHE-ECDSA-AES128-CCM8')
        self.context.verify_mode = ssl.CERT_REQUIRED
        self.context.load_verify_locations(cafile=self.cert)
        self.ssl_repl.ssl_repl("ssl_tool=ssl_repl.SSL_socket_client_tool('{}', {})".format(self.host, self.port))
        self.ssl_repl.ssl_repl("ssl_tool.connect_SOC(ssl_client.key, ssl_client.cert)")
        self.conn, self.addr_client = self.serv_soc.accept()
        self.conn = self.context.wrap_socket(self.conn, server_side=True)
        if args.v:
            print('SSLtool connection from: {}'.format(self.addr_client))
        self.conn.settimeout(0)

    def stop_SOC(self):
        self.conn.close()
        self.serv_soc.close()
        self.ssl_repl.ssl_repl("ssl_tool.cli_soc.close()")

    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 put(self, filetoput, chunk=(1024*8), chunk_rx=(4096*3)):
        self.get_pb()
        sz = os.stat(filetoput)[6]
        cnt = 0
        t_start = time.time()
        put_soc = [self.conn]
        # def_chunk = chunk
        print("{}  [{:.2f} KB]".format(filetoput, sz / 1024))
        self.ssl_repl.flush_conn()
        self.ssl_repl.send_message("ssl_tool.put('{}', {}, chunk={})".format(filetoput, sz, chunk_rx) + '\r')
        # self.ssl_repl.ssl_repl("ssl_tool.put('{}', {}, chunk={})".format(filetoput, sz, chunk_rx))
        with open(filetoput, "rb") as f:
            self.buff = f.read(chunk)
            while True:
                try:
                    readable, writable, exceptional = select.select(put_soc,
                                                                    put_soc,
                                                                    put_soc)
                    # self.buff = f.read(1024)  # 1 KB
                    # print(len(chunk))
                    if len(writable) == 1:
                        if self.buff != b'':
                            # in python use 'i'
                            self.conn.sendall(self.buff)
                            cnt += len(self.buff)
                            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)
                            self.buff = f.read(chunk)
                            # chunk = def_chunk
                            # final_file += chunk
                        else:
                            break

                except Exception as e:
                    # print(e)
                    time.sleep(0.02)
                    pass
            print('\n')
        self.ssl_repl.ssl_repl("print('Done!')")

    def get(self, filetoget, chunk=(4096*2), chunk_tx=(2048)):
        self.get_pb()
        cnt = 0
        t_start = time.time()
        put_soc = [self.conn]
        # def_chunk = chunk
        self.ssl_repl.ssl_repl("ssl_tool.get_size('{}')".format(filetoget))
        self.ssl_repl.dev.get_output()
        sz = self.ssl_repl.dev.output
        print("{}  [{:.2f} KB]".format(filetoget, sz / 1024))
        self.ssl_repl.flush_conn()
        self.ssl_repl.send_message("ssl_tool.get('{}', chunk={})".format(filetoget, chunk_tx) + '\r')
        # self.ssl_repl.ssl_repl("ssl_tool.put('{}', {}, chunk={})".format(filetoget, sz, chunk_rx))
        with open(filetoget, 'wb') as f:
            pass
        with open(filetoget, "ab") as f:
            while True:
                try:
                    readable, writable, exceptional = select.select(put_soc,
                                                                    put_soc,
                                                                    put_soc)
                    # # self.buff = f.read(1024)  # 1 KB
                    # # print(len(chunk))
                    if len(readable) == 1:
                        self.buff = self.conn.recv(chunk)
                        if self.buff != b'':
                            # in python use 'i'
                            f.write(self.buff)
                            if isinstance(self.buff, int):
                                cnt += 0
                            else:
                                cnt += len(self.buff)
                            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)
                            if cnt == sz:
                                break
                            # chunk = def_chunk
                            # final_file += chunk
                        else:
                            break

                except Exception as e:
                    # print(e)
                    # print(len(readable))
                    time.sleep(0.02)
                    pass
            print('\n')
        self.ssl_repl.ssl_repl("print('Done!');gc.collect()")


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


# ARGPARSE HERE TO ACCEPT TARGET AND PASSWORD
parser = argparse.ArgumentParser()
parser.add_argument("-p", help='host password', required=True)
parser.add_argument("-v", help='verbose mode',
                    default=False, action='store_true')
parser.add_argument("-t", help='host direction', required=True)
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')
parser.add_argument("-wss",
                    help='Use WebSocket Secure to do initial handshake, this needs WebSecureREPL enabled "wss_repl.start(ssl=True)"',
                    default=False, action='store_true')

args = parser.parse_args()

# HANDSHAKE
espdev = W_UPYDEVICE(args.t, args.p)
ws_upydevice = BASE_WS_DEVICE(args.t, args.p)
base_ws = True
if args.ping:
    if espdev.is_reachable():
        pass
    else:
        print('DEVICE NOT REACHABLE')
        sys.exit()
if args.v:
    print('Welcome to SSLWebREPL 0.0.1!')
if not args.nem:
    if args.v:
        print('Initiating SECURE HANDSHAKE...')
# espdev.kbi(output=False)
if base_ws:
    ws_upydevice.open_wconn(ssl=args.wss, auth=True)
    ws_upydevice.wr_cmd(ws_upydevice._kbi)
    # espdev.wr_cmd("from machine import unique_id; unique_id()",
    #               silent=True)
    ws_upydevice.wr_cmd("from machine import unique_id; unique_id()", silent=True)
    # unique_id = hexlify(espdev.output).decode()
    try:
        unique_id = hexlify(ws_upydevice.output).decode()
    except TypeError:
        time.sleep(0.5)
        ws_upydevice.wr_cmd("from machine import unique_id; unique_id()", silent=True)
        unique_id = hexlify(ws_upydevice.output).decode()
    # espdev.wr_cmd("import sys; sys.platform",
    #               silent=True)
    ws_upydevice.wr_cmd("import sys; sys.platform", silent=True)
    # dev_platform = espdev.output
    dev_platform = ws_upydevice.output
    espdev.platform = dev_platform
    ws_upydevice.platform = dev_platform
else:
    espdev.open_wconn()
    espdev.wr_cmd('C')
    espdev.wr_cmd("from machine import unique_id; unique_id()",
                  silent=True)
    unique_id = hexlify(espdev.output).decode()
    espdev.wr_cmd("import sys; sys.platform",
                  silent=True)
    dev_platform = espdev.output
    espdev.platform = dev_platform
if args.v:
    print('{}@{}'.format(dev_platform, unique_id))
# rk = load_rsa_key(dir=upydev.__path__[0]+'/', show_key=False,
#                   id=unique_id)
# if rk is None:
#     sys.exit()
ssl_key = upydev.__path__[0]+'/'+'SSL_key{}.pem'.format(unique_id)
ssl_cert = upydev.__path__[0]+'/'+'SSL_certificate{}.pem'.format(unique_id)
if not args.nem:
    if base_ws:
        espdev = ws_upydevice
    ssl_cp = SSL_REPL_server(dev=espdev, key=ssl_key, cert=ssl_cert,
                             client_key=None, dev_id=unique_id)
    ssl_htool = SSL_TOOL(ssl_repl=ssl_cp, key=ssl_key, cert=ssl_cert)
# print('Done!')
# if args.v:
#     print('SSLWebREPL connected')
if args.nem:
    if base_ws:
        ws_upydevice.close_wconn()
        espdev.open_wconn()
    ssl_cp = WBREPL_IO(dev=espdev)
    print('\033[91;1m' + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + '\033[0m')
    print('')
    espdev.wr_cmd('B')
    print('')
# espdev.wr_cmd('B')
if not args.nem:
    print('SSLWebREPL {} connected'.format(ssl_cp.version))
    print('{1} @ {0} - {2} bits Encryption'.format(*ssl_cp.cipher))
    # ssl_cp.ssl_repl("\x0d")
    # time.sleep(0.2)
    ssl_cp.ssl_repl("\x02", banner=True)
    ssl_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}
frozen_modules = {'FM': [], 'SUB': []}
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', 'wrepl',
                    'whoami', 'exit', 'pwdl', 'lsl', 'cdl', 'put', 'get',
                    'catl', 'l_micropython', 'ls', 'python', 'python3', 'vim',
                    'set_localtime', 'tree', 'sync', 'd_sync', 'l_ifconfig',
                    'l_ifconfig_t', 'lsof', 'reload', 'docs', 'flush_soc',
                    'getcert', 'get_rawbuff', 'l_tree', 'view', 'bat', 'rcat',
                    'batl', 'du', 'ldu', 'upipl', 'pkg_info', 'uping', 'lping',
                    'timeit', 'i2c', 'dump_mem', 'git', 'emacs', 'batstyle',
                    'upy-config', 'wss', 'tig', 'jupyterc']
CRED = '\033[91;1m'
CGREEN = '\33[32;1m'
CEND = '\033[0m'
YELLOW = '\u001b[33m'
BCYAN = '\u001b[36;1m'
ABLUE_bold = '\u001b[34;1m'
MAGENTA_bold = '\u001b[35;1m'
WHITE_ = '\u001b[37m'
LIGHT_GRAY = "\033[0;37m"
DARK_GRAY = "\033[1;30m"
PYGM_SYNTAX = list(get_all_styles()) + ['one_dark']
batstyle = {'style': 'monokai'}
AUTHMODE_DICT = {0: 'NONE', 1: 'WEP', 2: 'WPA PSK', 3: 'WPA2 PSK',
                    4: 'WPA/WAP2 PSK'}

git_diff_files = {'diff': [], 'commit': '', 'n_commits': 0}
if args.nem:
    encrypted_flag['sec'] = False
# 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
    if encrypted_flag['sec']:
        # time.sleep(0.2)
        ssl_cp.ssl_repl('import gc;import os;from upysh import *;gc.collect()', silent=True)
        time.sleep(0.2)
        ssl_cp.flush_conn()
        ssl_cp.flush_conn()
        espdev.output = None
        try:
            ssl_cp.ssl_repl("help('modules')", silent=True)
        except KeyboardInterrupt:
            try:
                ssl_cp.flush_conn()
                ssl_cp.ssl_repl("help('modules')", silent=True)
            except Exception as e:
                pass
        frozen_modules['FM'] = espdev.output.split()[:-6]
        ssl_cp.flush_conn()
        ssl_cp.flush_conn()
        espdev.output = None
    else:
        espdev.wr_cmd('import gc;import os;from upysh import *;gc.collect()', silent=True)
        espdev.close_wconn()
        espdev.cmd("help('modules')", silent=True, capture_output=True)
        espdev.open_wconn()
        frozen_modules['FM'] = '\n'.join(espdev.long_output).split()[:-6]
        # print(frozen_modules['FM'])
    espdev.output = None

# KEYBINDINGS INFO
kb_info = """
Custom keybindings:
- CTRL-x : to exit SSLWebREPL Terminal
- CTRL-p : toggle RAM STATUS right aligned message (USED/FREE)
- CTRL-e : paste mode in repl,(in shell mode set cursor position at the end)/(edit mode after 'edit' shell command)
- CTRL-d : ends paste mode in repl, (ends edit mode after 'edit' shell command)
- CTRL-c : KeyboardInterrupt, in normal mode, cancel in paste or edit mode
- CTRL-b : prints MicroPython version and sys platform
- CTRL-r : to flush line buffer
- CTRL-o : to list files in cwd (sz 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 commands
     - 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) (allows "*" wildcard or directories)
        - 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)
        - du   : display disk usage statistics (usage: "du", "du [dir or file]" + '-d' deep level option)
        - meminfo: to see RAM info
        - dump_mem: to do a memory dump
        - 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 WLANs available, (ESSID, MAC ADDRESS, CHANNEL, RSSI, AUTH MODE, HIDDEN)
        - uping : to make the device send ICMP ECHO_REQUEST packets to network hosts (do 'uping host' to ping local machine)
        - 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 (also allows "*" wildcard, 'cwd' or multiple files)
        - put    : to upload a file to the device (also allows "*" wildcard, 'cwd' or multiple files)
        - sync   : to get file (faster) from the device (use with > 10 KB files) (no encrypted mode only)
        - d_sync : to recursively sync a local directory with the device filesystem
        - wrepl  : to enter the original WebREPL terminal (no encryption mode)
        - reload : to delete a module from sys.path so it can be imported again.
        - flush_soc: to flush socket in case of wrong output
        - view   : to preview '.pbm' binary image files (image need to be centered and rows = columns) (encryption mode only)
        - bat    : prints the content of a '.py' file with Python syntax hightlighting
        - batstyle: 'bat' output style; to set: 'batstyle [style]', to see: 'batstyle': current style / 'batstyle -a': all styles
        - rcat   : prints the raw content of a file (encryption mode only)
        - timeit : to measure execution time of a script/command
        - i2c    : config/scan (config must be used first, i2c config -scl [SCL] -sda [SDA])
        - upy-config: interactive dialog to configure Network (connect to a WLAN or set an AP) or Interafaces (I2C)
        - wss: on/off; to enable WebSecureREPL for initial handshake (this sets ssl_flag.SSL to True or False)
        - jupyterc: to run MicroPython upydevice kernel for jupyter console
        - exit   : to exit SSLWebREPL Terminal (in encrypted mode soft-reset by default)
                    to exit without reset do 'exit -nr'
                    to exit and do hard reset 'exit -hr'

* 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
    - batl   : prints the content of a local '.py' file with Python syntax hightlighting
    - 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)
    - emacs  : to edit a local file with emacs (e.g. emacs 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)
    - lsof : to scan TCP ports of the device (TCP ports 1-10000)
    - 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'
    - getcert: to print the client SSL Certificate
    - get_rawbuff: to get the raw output of a command (for debugging purpose)
    - ldu  : display local path disk usage statistics (usage: "ldu", "ldu [dir or file]" + '-d' deep level option)
    - upipl : (usage 'upipl' or 'upipl [module]' display available micropython packages that can be installed with install command
    - pkg_info: to see the PGK-INFO file of a module if available at pypi.org or micropython.org/pi
    - lping : to make local machine send ICMP ECHO_REQUEST packets to network hosts (do 'lping dev' to ping the device)
    - git : to call git commands and integrate the git workflow into a project (needs 'git' available in $PATH)
            Use 'git init dev' to initiate device repo
            Use 'git push dev' after a 'git commit ..' or 'git pull' to push the changes to the device.
            Use 'git log dev' to see the latest commit pushed to the device ('git log dev -a' to see all commits)
            Use 'git log host' to see the latest commit in the local repo
            Use 'git status dev' to see if the local repo is ahead of the device repo and track these changes
            Use 'git clone_dev' to clone the local repo into the device
            Use 'git repo' to open the remote repo in the web browser if remote repo exists
            Any other git command will be echoed directly to git
    - tig: to use the 'Text mode interface for git' tool. Must be available in $PATH


"""

# 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:]))

# PBM PREVIEW


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 sortSecond(val):
    return val[1]


def get_ip():
    # scanoutput = subprocess.check_output(["ipconfig", "getifaddr", "en0"])
    # ip = scanoutput.decode('utf-8').split('\n')[0]
    try:
        ip = [netifaces.ifaddresses(iface)[netifaces.AF_INET][0]['addr'] for
                    iface in netifaces.interfaces() if netifaces.AF_INET in
                    netifaces.ifaddresses(iface)][-1]
        return ip
    except Exception as e:
        try:
            ip_soc = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            ip_soc.connect(('8.8.8.8', 1))
            local_ip = ip_soc.getsockname()[0]
            ip_soc.close()
            return local_ip
        except Exception as e:
            return '0.0.0.0'

class LTREE:

    def __repr__(self):
        self.__call__()
        return ""

    def __call__(self, path=".", level=0, is_last=False, is_root=True,
                 carrier="    ", hidden=False):
        os.chdir(path)
        r_path = path
        path = "."
        if hidden:
            l = os.listdir(path)
        else:
            l = [f for f in os.listdir(path) if not f.startswith('.')]
        nf = len([file for file in l if not os.stat("%s/%s" % (path, file))[0] & 0x4000])
        nd = len(l) - nf
        ns_f, ns_d = 0, 0
        l.sort()
        if len(l) > 0:
            last_file = l[-1]
        else:
            last_file = ''
        for f in l:
            st = os.stat("%s/%s" % (path, f))
            if st[0] & 0x4000:  # stat.S_IFDIR
                print(self._treeindent(level, f, last_file, is_last=is_last, carrier=carrier) + "  %s <dir>" % f)
                os.chdir(f)
                level += 1
                lf = last_file == f
                if level > 1:
                    if lf:
                        carrier += "     "
                    else:
                        carrier += "    │"
                ns_f, ns_d = self.__call__(level=level, is_last=lf,
                                           is_root=False, carrier=carrier,
                                           hidden=hidden)
                if level > 1:
                    carrier = carrier[:-5]
                os.chdir('..')
                level += (-1)
                nf += ns_f
                nd += ns_d
            else:
                print(self._treeindent(level, f, last_file, is_last=is_last, carrier=carrier) + "  %s" % (f))
        if is_root:
            print('{} directories, {} files'.format(nd, nf))
            if r_path != ".":
                os.chdir('..')
        else:
            return (nf, nd)

    def _treeindent(self, lev, f, lastfile, is_last=False, carrier=None):
        if lev == 0:
            return ""
        else:
            if f != lastfile:
                return carrier + "    ├────"
            else:
                return carrier + "    └────"

tree = LTREE()

class DISK_USAGE:

    def __repr__(self):
        self.__call__()
        return ""

    def __call__(self, path=".", dlev=0, max_dlev=0, hidden=False, absp=True):
        if path != ".":
                if not os.stat(path)[0] & 0x4000:
                    print('{:9} {}'.format(self.print_filesys_info(os.stat(path)[6]), path))
                else:
                    if hidden:
                        resp = {path+'/'+dir: os.stat(path+'/'+dir)[6] for dir in os.listdir(path)}
                    else:
                        resp = {path+'/'+dir: os.stat(path+'/'+dir)[6] for dir in os.listdir(path) if not dir.startswith('.')}
                    for dir in resp.keys():

                        if not os.stat(dir)[0] & 0x4000:
                            if absp:
                                print('{:9} {}'.format(self.print_filesys_info(resp[dir]), dir))
                            else:
                                print('{:9} {}'.format(self.print_filesys_info(resp[dir]), dir.split('/')[-1]))

                        else:
                            if dlev < max_dlev:
                                dlev += 1
                                self.__call__(path=dir, dlev=dlev, max_dlev=max_dlev, hidden=hidden)
                                dlev += (-1)
                            else:
                                if absp:
                                    print('{:9} {} {}'.format(self.print_filesys_info(self.get_dir_size_recursive(dir)), dir, '<dir>'))
                                else:
                                    print('{:9} {} {}'.format(self.print_filesys_info(self.get_dir_size_recursive(dir)), dir.split('/')[-1], '<dir>'))


        else:
            if hidden:
                resp = {path+'/'+dir: os.stat(path+'/'+dir)[6] for dir in os.listdir(path)}
            else:
                resp = {path+'/'+dir: os.stat(path+'/'+dir)[6] for dir in os.listdir(path) if not dir.startswith('.')}
            for dir in resp.keys():

                if not os.stat(dir)[0] & 0x4000:
                    print('{:9} {}'.format(self.print_filesys_info(resp[dir]), dir))

                else:
                    if dlev < max_dlev:
                        dlev += 1
                        self.__call__(path=dir, dlev=dlev, max_dlev=max_dlev, hidden=hidden)
                        dlev += (-1)
                    else:
                        print('{:9} {} {}'.format(self.print_filesys_info(self.get_dir_size_recursive(dir)), dir, '<dir>'))



    def print_filesys_info(self, 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 get_dir_size_recursive(self, dir):
        return sum([os.stat(dir+'/'+f)[6] if not os.stat(dir+'/'+f)[0] & 0x4000 else self.get_dir_size_recursive(dir+'/'+f) for f in os.listdir(dir)])


du = DISK_USAGE()


def get_dir_size_recursive(dir):
    return sum([os.stat(dir+'/'+f)[6] if not os.stat(dir+'/'+f)[0] & 0x4000 else get_dir_size_recursive(dir+'/'+f) for f in os.listdir(dir)])


def d_sync_recursive(folder, dev=None, rootdir='./', root_sync_folder=None,
                     show_tree=False):
    t0 = time.time()
    type_file_dict = {True:'<f>', False:'<d>'}
    if folder == root_sync_folder:
        print('DIRECTORY TO SYNC: {}'.format(folder))
        print('DIRECTORY SIZE: {}'.format(print_filesys_info(get_dir_size_recursive(folder))))
        ssl_htool.start_SOC()
    if show_tree:
        print('DIRECTORY TREE STRUCTURE:\n')
        tree(path=folder)
    time.sleep(1)
    print('\n')
    print('***** {} *****'.format(folder))
    print('\n')
    directory = folder
    file_list = None
    file_list_abs_path = []
    dir_list_abs_path = []
    current_dir = directory
    # get directory structure:
    print('ROOT DIRECTORY: {}'.format(rootdir))
    print('DIRECTORY TO SYNC: {}'.format(directory))
    print('\n')
    print('CHECKING IF DIRECTORY {} IN: {}'.format(
        directory.split('/')[-1], rootdir))
    if directory.split('/')[-1] in os.listdir(rootdir):
        print('DIRECTORY {} FOUND'.format(directory))
        print('\n')
        print('FILES/DIRS IN DIRECTORY {}:'.format(directory))
        du(path='./{}'.format(directory), absp=False, hidden=True)
        # for file in os.listdir(directory):
        #     print('- {} {} [{}]'.format(type_file_dict[os.path.isfile(os.path.join(current_dir, file))],
        #                            file, print_filesys_info(os.stat(os.path.join(current_dir, file))[6])))
        file_list = os.listdir(directory)
        print('\n')
        for file in file_list:
            if os.path.isfile(os.path.join(current_dir, file)):
                file_list_abs_path.append(os.path.join(current_dir, file))
            elif os.path.isdir(os.path.join(current_dir, file)):
                dir_list_abs_path.append(os.path.join(current_dir, file))
    print('LIST OF FILES TO UPLOAD:')
    for file in file_list_abs_path:
        print('- {}'.format(file.split('/')[-1]))
    print('\n')
    print('LIST OF SUBDIRS TO CREATE:')
    for subdir in dir_list_abs_path:
        print('- {}'.format(subdir.split('/')[-1]))
    print('\n')
    # Now create the root sync dir:
    if './' == rootdir:
        rootdir = ''
    dev_root_list = send_custom_sh_cmd("os.listdir('{}')".format(rootdir))
    if current_dir.split('/')[-1] not in dev_root_list:
        print('\n')
        print('MAKING DIR: {}'.format(current_dir))
        print('\n')
        dev_mkdir = ssl_cp.ssl_repl("os.mkdir('{}')".format(current_dir))
        print('\n')
    print('UPLOADING FILES TO {}'.format(current_dir))
    if len(file_list_abs_path) > 1:
        for file in file_list_abs_path:
            print('- {}'.format(file))
        # ssl_htool.start_SOC()
        for file in file_list_abs_path:
            ssl_htool.put(file)
            time.sleep(0.2)
        # ssl_cp.ssl_repl("print('Done!')")
        time.sleep(0.2)
        # ssl_htool.stop_SOC()
    elif len(file_list_abs_path) == 1:
        file_to_put = file_list_abs_path[0]
        print('- {}'.format(file_to_put))
        # ssl_htool.start_SOC()
        ssl_htool.put(file_to_put)
        # ssl_cp.ssl_repl("print('Done!')")
        time.sleep(0.2)
        # ssl_htool.stop_SOC()
    else:
        print('NO FILES IN DIR TO UPLOAD')
    # Now create subdirs:
    print('\n')
    print('MAKING SUBDIRS NOW...')
    for dir_ in dir_list_abs_path:
        print('\n')
        current_dir = dir_
        # Now create the root sync dir:
        try:
            dir_to_sync = directory
            dev_directory = send_custom_sh_cmd("os.listdir('{}')".format(dir_to_sync))
        except Exception as e:
            pass
        if not current_dir.split('/')[-1] in dev_directory:
            print('Creating dir: {}'.format(current_dir))
            # print(f'os.mkdir("{current_dir}")')
            dev_mkdir = ssl_cp.ssl_repl("os.mkdir('{}')".format(current_dir))
        else:
            print('DIRECTORY {} ALREADY EXISTS'.format(current_dir.split('/')[-1]))
    if len(dir_list_abs_path) == 0:
        print('NO MORE SUBDIRS')
    root = directory
    for dir_ in dir_list_abs_path:
        d_sync_recursive(dir_, dev, root)

    if directory == root_sync_folder:
        ssl_htool.stop_SOC()
        print('Done in : {:.2f} seconds'.format(time.time()-t0))


def send_custom_sh_cmd(cmd, capture_output=False, sh_silent=True, collect=False):
    if not encrypted_flag['sec']:
        if not capture_output:
            espdev.wr_cmd(cmd, silent=sh_silent)
        else:
            espdev.close_wconn()
            espdev.cmd(cmd, silent=sh_silent, capture_output=True)
            espdev.output = espdev.long_output
            espdev.open_wconn()
    else:
        ssl_cp.ssl_repl(cmd)
        espdev.get_output()
        if not sh_silent:
            print(espdev.output)
        if collect:
            ssl_cp.send_message('gc.collect()'+'\r')
            ssl_cp.flush_conn()
    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 get_rprompt():
    if status_encryp_msg['S']:
        if encrypted_flag['sec']:
            RAM = send_custom_sh_cmd(
                'from micropython import mem_info;mem_info()')
            mem_info = RAM.splitlines()[1]
        else:
            RAM = send_custom_sh_cmd(
                'from micropython import mem_info;mem_info()', True)
            mem_info = RAM[1].strip()
        mem = {elem.strip().split(':')[0]: int(elem.strip().split(':')[
                          1]) for elem in mem_info[4:].split(',')}
        used_mem = mem['used']/1024
        free_mem = mem['free']/1024
        used_mem_s = "{:.3f} KB".format(used_mem)
        free_mem_s = "{:.3f} KB".format(free_mem)
        total_mem = mem['total']/1024
        use_percent = round((used_mem/total_mem)*100, 2)

        if use_percent < 85:
            return HTML('<aaa fg="ansiblack" bg="ansiwhite"> RAM </aaa><b><style fg="ansigreen"> USED </style></b><aaa fg="ansiblack" bg="ansiwhite"> {} </aaa><b><style fg="ansigreen"> FREE </style></b><aaa fg="ansiblack" bg="ansiwhite"> {} </aaa>'.format(used_mem_s, free_mem_s))
        else:
            return HTML('<aaa fg="ansiblack" bg="ansiwhite"> RAM </aaa><b><style fg="ansired"> USED </style></b><aaa fg="ansiblack" bg="ansiwhite"> {} </aaa><b><style fg="ansired"> FREE </style></b><aaa fg="ansiblack" bg="ansiwhite"> {} </aaa>'.format(used_mem_s, free_mem_s))
    else:
        return HTML('<b><style fg="ansiblack"> {} </style></b>'.format(' '*10))


def custom_sh_cmd(cmd_inp):
    # map cmd_inp to command 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']))
        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), ' ', '/'))
        vfs_resp = send_custom_sh_cmd("{dir:os.statvfs(dir) for dir in os.listdir() if os.stat(dir)[0] & 0x4000 and os.statvfs('') != os.statvfs(dir)}")
        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.split()[0] == 'ldu':
        if len(cmd_inp.split()) == 1:
            du()

        elif len(cmd_inp.split()) == 2:
            if cmd_inp.split()[1] != '-o':
                file = cmd_inp.split()[1]
                du(path='./{}'.format(file))

            elif cmd_inp.split()[1] == '-o':
                resp = {dir:os.stat(dir)[6] for dir in os.listdir()}
                is_dir = {}
                for dir in resp.keys():
                    if resp[dir] != 0:
                        pass
                        is_dir[dir] = 0
                        # print('{:9} ./{}'.format(print_filesys_info(resp[dir]), dir))
                    else:
                        # print('{:9} ./{}'.format('<dir>', dir))
                        resp_dir = sum([os.stat(dir+sdir)[6] for sdir in os.listdir(dir)])
                        # print('{:9} ./{} <dir>'.format(print_filesys_info(resp_dir), dir))
                        resp[dir] = resp_dir
                        is_dir[dir] = 1

                du_sorted_list = [(dir, resp[dir], is_dir[dir]) for dir in resp.keys()]
                du_sorted_list.sort(key=sortSecond, reverse=True)

                for val in du_sorted_list:
                    if val[2] == 0:
                        print('{:9} ./{}'.format(print_filesys_info(val[1]), val[0]))
                    else:
                        print('{:9} ./{} <dir>'.format(print_filesys_info(val[1]), val[0]))
        elif len(cmd_inp.split()) == 3:
            if cmd_inp.split()[1] == '-d':
                try:
                    dlevmx = int(cmd_inp.split()[2])
                    du(max_dlev=dlevmx)

                except Exception as e:
                    print(e)

        elif len(cmd_inp.split()) == 4:
            if cmd_inp.split()[2] == '-d':
                try:
                    file = cmd_inp.split()[1]
                    dlevmx = int(cmd_inp.split()[3])
                    du(path='./{}'.format(file), max_dlev=dlevmx)

                except Exception as e:
                    print(e)

    if cmd_inp.split()[0] == 'du':
        if len(cmd_inp.split()) == 1:
            # send_custom_sh_cmd('from upysh2 import tree;tree()', sh_silent=False)
            if encrypted_flag['sec']:
                ssl_cp.ssl_repl('from upysh2 import du;du();gc.collect()', follow=True)
                ssl_cp.flush_conn()
            else:
                send_custom_sh_cmd('from upysh2 import du;du();gc.collect()', sh_silent=False)

        elif len(cmd_inp.split()) == 2:
            if cmd_inp.split()[1] != '-o':
                file = cmd_inp.split()[1]
                if encrypted_flag['sec']:
                    ssl_cp.ssl_repl("from upysh2 import du;du(path='./{}');gc.collect()".format(file), follow=True)
                    ssl_cp.flush_conn()
                else:
                    send_custom_sh_cmd("from upysh2 import du;du(path='./{}');gc.collect()".format(file), sh_silent=False)

            elif cmd_inp.split()[1] == '-o':
                resp = send_custom_sh_cmd("{dir:os.stat(dir)[6] for dir in os.listdir()}")
                is_dir = {}
                for dir in resp.keys():
                    if resp[dir] != 0:
                        pass
                        is_dir[dir] = 0
                        # print('{:9} ./{}'.format(print_filesys_info(resp[dir]), dir))
                    else:
                        # print('{:9} ./{}'.format('<dir>', dir))
                        resp_dir = send_custom_sh_cmd("sum([os.stat('{0}/'+dir)[6] for dir in os.listdir('{0}')])".format(dir))
                        # print('{:9} ./{} <dir>'.format(print_filesys_info(resp_dir), dir))
                        resp[dir] = resp_dir
                        is_dir[dir] = 1

                du_sorted_list = [(dir, resp[dir], is_dir[dir]) for dir in resp.keys()]
                du_sorted_list.sort(key=sortSecond, reverse=True)

                for val in du_sorted_list:
                    if val[2] == 0:
                        print('{:9} ./{}'.format(print_filesys_info(val[1]), val[0]))
                    else:
                        print('{:9} ./{} <dir>'.format(print_filesys_info(val[1]), val[0]))
        elif len(cmd_inp.split()) == 3:
            if cmd_inp.split()[1] == '-d':
                try:
                    dlev = cmd_inp.split()[2]
                    if encrypted_flag['sec']:
                        ssl_cp.ssl_repl("from upysh2 import du;du(max_dlev={});gc.collect()".format(dlev), follow=True)
                        ssl_cp.flush_conn()
                    else:
                        send_custom_sh_cmd("from upysh2 import du;du(max_dlev={});gc.collect()".format(dlev), sh_silent=False)

                except Exception as e:
                    print(e)

        elif len(cmd_inp.split()) == 4:
            if cmd_inp.split()[2] == '-d':
                try:
                    file = cmd_inp.split()[1]
                    dlev = cmd_inp.split()[3]
                    if encrypted_flag['sec']:
                        ssl_cp.ssl_repl("from upysh2 import du;du(path='./{}',max_dlev={});gc.collect()".format(file, dlev), follow=True)
                        ssl_cp.flush_conn()
                    else:
                        send_custom_sh_cmd("from upysh2 import du;du(path='./{}',max_dlev={});gc.collect()".format(file, dlev), sh_silent=False)

                except Exception as e:
                    print(e)




    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(
            "import network;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(
            "import network;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':
        if espdev.platform == 'esp8266':
            enable_sta = send_custom_sh_cmd(
                "import network;network.WLAN(network.STA_IF).active()")
        else:
            enable_sta = send_custom_sh_cmd(
                "import network;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'))

            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(
            "import network;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(
            "import network;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':
        if espdev.platform == 'esp8266':
            enable_sta = send_custom_sh_cmd(
                "import network;network.WLAN(network.AP_IF).active()")
        else:
            enable_sta = send_custom_sh_cmd(
                "import network;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':
        if encrypted_flag['sec']:
            RAM = send_custom_sh_cmd(
                'from micropython import mem_info;mem_info()')
            mem_info = RAM.splitlines()[1]
        else:
            RAM = send_custom_sh_cmd(
                'from micropython import mem_info;mem_info()', True)
            mem_info = RAM[1].strip()
        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 == 'dump_mem':
        if encrypted_flag['sec']:
            RAM_dump = send_custom_sh_cmd(
                'from micropython import mem_info;mem_info(1)')
            print(RAM_dump)
        else:
            RAM_dump = send_custom_sh_cmd(
                'from micropython import mem_info;mem_info(1)', True)
            print('\n'.join([line.replace('\r', '') for line in RAM_dump]))

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

    if cmd_inp.split(' ')[0] == 'pkg_info':
        try:
            lib = cmd_inp.split(' ')[1]
            upip_host.install_pkg(lib, ".", read_pkg_info=True)

        except Exception as e:
            print('Please indicate a library to see PKG-INFO')

    if cmd_inp.split(' ')[0] == 'upipl':
        try:
            if len(cmd_inp.split()) == 1:
                print('Available MicroPython libs (just a random subset):')
                list_upylibs = shlex.split('pip search micropython')
                result = subprocess.call(list_upylibs)
            elif len(cmd_inp.split()) == 2:
                lib = cmd_inp.split()[1]
                print("Available MicroPython libs that match {}:".format(lib))
                list_upylibs = shlex.split("pip search {}".format(lib))
                result = subprocess.call(list_upylibs)
        except Exception as e:
            print(e)

    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] == 'wrepl':
        if not args.wss:
            print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
        print('<-- Device {} MicroPython -->'.format(dev_platform))
        print('Use CTRL-x to exit, Use CTRL-k to see custom wrepl Keybindings')
        try:
            ssl_cp.enable_wrepl_io()
        except Exception as e:
            pass
        if not args.wss:
            web_repl_cmd_str = 'web_repl {} -p {}'.format(args.t, args.p)
        else:
            web_repl_cmd_str = 'web_repl {} -p {} -wss'.format(args.t, args.p)
        web_repl_cmd = shlex.split(web_repl_cmd_str)
        try:
            web_repl = subprocess.call(web_repl_cmd)
            try:
                ssl_cp.enable_wrepl_io()
            except Exception as e:
                pass
        except KeyboardInterrupt:
            try:
                ssl_cp.enable_wrepl_io()
            except Exception as e:
                pass
            pass
            print('')

    if cmd_inp.split(' ')[0] == 'jupyterc':
        if not args.wss:
            print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
        # print('<-- Device {} MicroPython -->'.format(dev_platform))
        print('Use CTRL-D to exit, Use %lsmagic to see custom magic commands')
        try:
            ssl_cp.enable_wrepl_io()
        except Exception as e:
            pass
        if not args.wss:
            jupyter_cmd_str = 'jupyter console --kernel=micropython-upydevice'
        else:
            jupyter_cmd_str = 'jupyter console --kernel=micropython-upydevice'

        jupyter_cmd = shlex.split(jupyter_cmd_str)
        old_action = signal.signal(signal.SIGINT, signal.SIG_IGN)

        def preexec_function(action=old_action):
            signal.signal(signal.SIGINT, action)

        try:
            jupyter_console = subprocess.call(jupyter_cmd, preexec_fn=preexec_function)
            signal.signal(signal.SIGINT, old_action)
            try:
                ssl_cp.enable_wrepl_io()
            except Exception as e:
                pass
        except KeyboardInterrupt:
            try:
                ssl_cp.enable_wrepl_io()
            except Exception as e:
                pass
            pass
            print('')

    if cmd_inp == 'whoami':
        print('DEVICE: {}, IP: {} , ID: {}'.format(host_name, args.t, unique_id))
        sysinfo = send_custom_sh_cmd('os.uname()', capture_output=True)
        if encrypted_flag['sec']:
            dev_info = sysinfo.split("'")
        else:
            dev_info = sysinfo[0].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.split()[0] == 'lsl':
        if len(cmd_inp.split()) == 1:
            list_of_files = os.listdir()
        elif len(cmd_inp.split()) > 1:
            if '*' in cmd_inp.split()[1]: # re to match
                r_e = cmd_inp.split()[1].replace('*', '')
                list_of_files = [f for f in os.listdir() if r_e in f]
            else:
                list_of_files = os.listdir(cmd_inp.split()[1])
        if list_of_files != []:
            print_table(list_of_files, wide=28, format_SH=True)

    if cmd_inp.split(' ')[0] == 'get':
        if not encrypted_flag['sec']:
            print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
            try:
                ssl_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 get -fre {} -t {} -p {} {}'.format(
                        file, args.t, args.p, dev_dir)
                elif '*' in file:
                    get_str = 'upydev get -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 get -fre {} -t {} -p {} {}'.format(
                        file, args.t, args.p, dev_dir)
                else:
                    get_str = 'upydev get -f {} -t {} -p {} {}'.format(
                        file, args.t, args.p, dev_dir)
                get_cmd = shlex.split(get_str)
                try:
                    get_file = subprocess.call(get_cmd)
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                except KeyboardInterrupt:
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                    pass
                    print('Operation aborted')
            except Exception as e:
                print('Indicate a file to get')

        else:
            try:
                print('File/s to get:')
                file = cmd_inp.split(' ')[1]
                if file == 'cwd':
                    ssl_cp.ssl_repl("ssl_repl.get_files_cwd()")
                    espdev.get_output()
                    files_to_put = espdev.output
                    for file in files_to_put:
                        print('- {}'.format(file))
                    ssl_htool.start_SOC()
                    for file in files_to_put:
                        ssl_htool.get(file)
                        time.sleep(0.2)
                    # ssl_cp.ssl_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_htool.stop_SOC()
                elif '*' in file:
                    ssl_cp.ssl_repl("ssl_repl.get_files_re('{}')".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))
                        ssl_htool.start_SOC()
                        for file in files_to_put_re:
                            ssl_htool.get(file)
                            time.sleep(0.2)
                        # ssl_cp.ssl_repl("print('Done!')")
                        time.sleep(0.2)
                        ssl_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]
                    ssl_cp.ssl_repl("ssl_repl.get_files_dir({})".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))
                        ssl_htool.start_SOC()
                        for file in files_to_put_dir:
                            ssl_htool.get(file)
                            time.sleep(0.2)
                        # ssl_cp.ssl_repl("print('Done!')")
                        time.sleep(0.2)
                        ssl_htool.stop_SOC()
                else:
                    print('- {}'.format(file))
                    ssl_htool.start_SOC()
                    ssl_htool.get(file)
                    # ssl_cp.ssl_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_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:
            ssl_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:
                    ssl_cp.enable_wrepl_io()
                except Exception as e:
                    pass
            except KeyboardInterrupt:
                try:
                    ssl_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':
        if not encrypted_flag['sec']:
            print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
            try:
                ssl_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]
                # put_str = 'upydev put -f {} -t {} -p {} -rst f'.format(file, args.t, args.p)
                # put_cmd = shlex.split(put_str)
                file = cmd_inp.split(' ')[1]  #  HERE CALL ONE OR -FRE
                if file == 'cwd':
                    put_str = 'upydev put -fre {} -t {} -p {} {} -rst f'.format(
                        file, args.t, args.p, dev_dir)
                elif '*' in file:
                    put_str = 'upydev put -fre {} -t {} -p {} {} -rst f'.format(
                        file.replace('*', ''), args.t, args.p, dev_dir)
                elif len(cmd_inp.split(' ')) > 2:
                    file = ' '.join(cmd_inp.split(' ')[1:])
                    put_str = 'upydev put -fre {} -t {} -p {} {} -rst f'.format(
                        file, args.t, args.p, dev_dir)
                else:
                    put_str = 'upydev put -f {} -t {} -p {} {} -rst f'.format(
                        file, args.t, args.p, dev_dir)
                put_cmd = shlex.split(put_str)
                try:
                    put_file = subprocess.call(put_cmd)
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                except KeyboardInterrupt:
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                    pass
                    print('Operation aborted')
            except Exception as e:
                # print(e)
                print('Indicate a file to put')
        else:
            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))
                    ssl_htool.start_SOC()
                    for file in files_to_put:
                        ssl_htool.put(file)
                        time.sleep(0.2)
                    # ssl_cp.ssl_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_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))
                    ssl_htool.start_SOC()
                    for file in files_to_put_re:
                        ssl_htool.put(file)
                        time.sleep(0.2)
                    # ssl_cp.ssl_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_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))
                        ssl_htool.start_SOC()
                        for file in files_to_put_dir:
                            ssl_htool.put(file)
                            time.sleep(0.2)
                        # ssl_cp.ssl_repl("print('Done!')")
                        time.sleep(0.2)
                        ssl_htool.stop_SOC()
                else:
                    print('- {}'.format(file))
                    ssl_htool.start_SOC()
                    ssl_htool.put(file)
                    # ssl_cp.ssl_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_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':
        if not encrypted_flag['sec']:
            print(CRED + 'WARNING: ENCRYPTION DISABLED IN THIS MODE' + CEND)
            try:
                ssl_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:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                except KeyboardInterrupt:
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                    pass
                    print('Operation aborted')
            except Exception as e:
                print('Indicate a file to get')
        else:
            if len(cmd_inp.split()) == 1: # root_dir
                dir_to_sync = os.getcwd().split('/')[-1]
                # print('Directory to sync: {}'.format(dir_to_sync))
                os.chdir('..')
                d_sync_recursive(dir_to_sync, show_tree=True, root_sync_folder=dir_to_sync)
                os.chdir(dir_to_sync)
            elif len(cmd_inp.split()) == 2:
                dir_to_sync = cmd_inp.split()[-1]
                # print('Directory to sync: {}'.format(dir_to_sync))
                d_sync_recursive(dir_to_sync, show_tree=True, root_sync_folder=dir_to_sync)

    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.split()[0] == 'ls':
        if len(cmd_inp.split()) == 1:
            dev_files = send_custom_sh_cmd('os.listdir();gc.collect()')
        elif len(cmd_inp.split()) > 1:
            if '*' in cmd_inp.split()[1]: # re to match
                r_e = cmd_inp.split()[1].replace('*', '')
                dev_files = send_custom_sh_cmd("[f for f in os.listdir() if '{}' in f];gc.collect()".format(r_e))
            else:
                dev_files = send_custom_sh_cmd("os.listdir('{}');gc.collect()".format(cmd_inp.split()[1]))
        try:
            print_table(dev_files, wide=28, format_SH=True)
        except Exception as e:
            pass

    if cmd_inp.split(' ')[0].startswith('python'):
        print('<-- Local machine Python -->')
        print('Use Ctrl-D to exit')
        try:
            lpy = cmd_inp.split(' ')[0]
            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(1)
        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 'emacs' in cmd_inp.split(' ')[0]:
        print('<-- Emacs editor -->')
        print('Use Ctrl-x-Ctrl-s to Save,  Ctrl-x-Ctrl-c to exit')
        time.sleep(1)
        try:
            file_to_edit = cmd_inp.split(' ')[1]
            lemacs = 'emacs {} -nw'.format(file_to_edit)
            lemacs_cmd = shlex.split(lemacs)
            old_action = signal.signal(signal.SIGINT, signal.SIG_IGN)

            def preexec_function(action=old_action):
                signal.signal(signal.SIGINT, action)
            try:
                l_emacs = subprocess.call(lemacs_cmd, preexec_fn=preexec_function)
                signal.signal(signal.SIGINT, old_action)
            except Exception as e:
                print(e)
                print('Local Emacs not found')
        except Exception as e:
            print(e)
            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':
                if encrypted_flag['sec']:
                    ssl_cp.ssl_repl('from upysh2 import tree;tree(hidden=True);gc.collect()', follow=True)
                    ssl_cp.flush_conn()
                else:
                    send_custom_sh_cmd('from upysh2 import tree;tree(hidden=True);gc.collect()', sh_silent=False)
            else:
                if len(cmd_inp.split(' ')) == 2:
                    dir = cmd_inp.split(' ')[1]
                    if encrypted_flag['sec']:
                        ssl_cp.ssl_repl("from upysh2 import tree;tree(path='{}');gc.collect()".format(dir), follow=True)
                        ssl_cp.flush_conn()
                    else:
                        send_custom_sh_cmd("from upysh2 import tree;tree(path='{}');gc.collect()".format(dir), sh_silent=False)
                else:
                    if cmd_inp.split(' ')[2] == '-a':
                        dir = cmd_inp.split(' ')[1]
                        if encrypted_flag['sec']:
                            ssl_cp.ssl_repl("from upysh2 import tree;tree(path='{}', hidden=True);gc.collect()".format(dir), follow=True)
                            ssl_cp.flush_conn()
                        else:
                            send_custom_sh_cmd("from upysh2 import tree;tree(path='{}', hidden=True);gc.collect()".format(dir), sh_silent=False)

        else:
            # send_custom_sh_cmd('from upysh2 import tree;tree()', sh_silent=False)
            if encrypted_flag['sec']:
                ssl_cp.ssl_repl('from upysh2 import tree;tree();gc.collect()', follow=True)
                ssl_cp.flush_conn()
            else:
                send_custom_sh_cmd('from upysh2 import tree;tree();gc.collect()', sh_silent=False)

    if cmd_inp == 'l_ifconfig':
        ifconfig_local = [netifaces.ifaddresses(iface)[netifaces.AF_INET][0]
                          for iface in netifaces.interfaces() if
                          netifaces.AF_INET in netifaces.ifaddresses(iface)][-1]
        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(iface)[netifaces.AF_INET][0]
                          for iface in netifaces.interfaces() if
                          netifaces.AF_INET in netifaces.ifaddresses(iface)][-1]
        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)
        if not encrypted_flag['sec']:
            espdev.wr_cmd(reload_cmd)
        else:
            ssl_cp.ssl_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':
        if encrypted_flag['sec']:
            ssl_cp.flush_conn()

    if cmd_inp == 'getcert':
        if encrypted_flag['sec']:
            for key in ssl_cp.client_cert.keys():
                value = ssl_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(), ssl_cp.client_cert[key]))
                except Exception as e:
                    print("{}: {}".format(key.upper(), ssl_cp.client_cert[key]))

    if cmd_inp == 'get_rawbuff':
        if encrypted_flag['sec']:
            print(ssl_cp.raw_buff)

    if cmd_inp.split()[0] == 'l_tree':
        try:
            if len(cmd_inp.split(' ')) > 1:
                if cmd_inp.split(' ')[1] == '-a':
                    tree(hidden=True)
                else:
                    if len(cmd_inp.split(' ')) == 2:
                        dir = cmd_inp.split(' ')[1]
                        tree(path=dir)
                    else:
                        if cmd_inp.split(' ')[2] == '-a':
                            dir = cmd_inp.split(' ')[1]
                            tree(path=dir, hidden=True)
            else:
                tree()
        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'):
                ssl_cp.ssl_repl("cat('{}')".format(img_file), dec=False)
                dimensions, data, raw_bin_img, img_1 = read_pbm(ssl_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('Indicate a ".pbm" file to view')

    if cmd_inp.split(' ')[0] == 'batstyle':
        if len(cmd_inp.split()) == 1:
            if encrypted_flag['sec']:
                print('Current Color Syntax Style: {}'.format(ssl_cp.color_style))
            else:
                print('Current Color Syntax Style: {}'.format(batstyle['style']))
        elif len(cmd_inp.split()) == 2:
            if cmd_inp.split()[1] == '-a':
                print('Styles availables:')
                print_table(PYGM_SYNTAX)
            else:
                style_to_set = cmd_inp.split()[1]
                if style_to_set in PYGM_SYNTAX:
                    if encrypted_flag['sec']:
                        ssl_cp.color_style = style_to_set
                        print('Style {} configured'.format(style_to_set))
                    else:
                        batstyle['style'] = style_to_set
                        print('Style {} configured'.format(style_to_set))
                else:
                    print('Style not available')

    if cmd_inp.split(' ')[0] == 'bat':
        try:
            color_file = cmd_inp.split(' ')[1]
            if color_file.endswith('.py'):
                if encrypted_flag['sec']:
                    ssl_cp.ssl_repl("cat('{}')".format(color_file), silent=False, color=True)
                else:
                    color_f = '\n'.join(send_custom_sh_cmd("cat('{}')".format(color_file), capture_output=True))
                    if batstyle['style'] == 'one_dark':
                        result = highlight(color_f, Python3Lexer(), Terminal256Formatter(style=one_darkStyle))
                        lines = result.split('\n')
                        lines_numbered = ["{}{:4}{} {}".format(DARK_GRAY, i, CEND, lines[i-1]) for i in range(1, len(lines)+1)]
                        my_code_numb = '\n'.join(lines_numbered)
                        print(my_code_numb)

                    else:
                        pygm_style = get_style_by_name(batstyle['style'])
                        result = highlight(color_f, Python3Lexer(), Terminal256Formatter(style=pygm_style))
                        lines = result.split('\n')
                        lines_numbered = ["{}{:4}{} {}".format(DARK_GRAY, i, CEND, lines[i-1]) for i in range(1, len(lines)+1)]
                        my_code_numb = '\n'.join(lines_numbered)
                        print(my_code_numb)

            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] == 'batl':
        try:
            color_file = cmd_inp.split(' ')[1]
            if color_file.endswith('.py'):
                try:
                    with open(color_file, 'r') as catfile:
                        content = catfile.read()
                        if encrypted_flag['sec']:
                            current_style = ssl_cp.color_style
                        else:
                            current_style = batstyle['style']
                        if current_style == 'one_dark':
                            result = highlight(content, Python3Lexer(), Terminal256Formatter(style=one_darkStyle))
                            lines = result.split('\n')
                            lines_numbered = ["{}{:4}{} {}".format(DARK_GRAY, i, CEND, lines[i-1]) for i in range(1, len(lines)+1)]
                            my_code_numb = '\n'.join(lines_numbered)
                            print(my_code_numb)

                        else:
                            pygm_style = get_style_by_name(current_style)
                            result = highlight(content, Python3Lexer(), Terminal256Formatter(style=pygm_style))
                            lines = result.split('\n')
                            lines_numbered = ["{}{:4}{} {}".format(DARK_GRAY, i, CEND, lines[i-1]) for i in range(1, len(lines)+1)]
                            my_code_numb = '\n'.join(lines_numbered)
                            print(my_code_numb)
                except Exception as e:
                    print(e)
                    pass

        except Exception as e:
            print('Indicate a ".py" file to view')

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

    if cmd_inp.split(' ')[0] == 'uping':
        try:
            ping_dir = cmd_inp.split(' ')[1]
            if ping_dir == 'host':
                ping_dir = ifconfig_local = netifaces.ifaddresses('en0')[netifaces.AF_INET][0]['addr']
            inp = "import uping;uping.ping('{}', loop=True, rtn=False)".format(ping_dir)
            if not encrypted_flag['sec']:
                espdev.wr_cmd(inp)
            else:
                try:
                    ssl_cp.ssl_repl(inp, follow=True)
                    # if espdev.output is not None:
                    #     print(espdev.output)
                except KeyboardInterrupt:
                    time.sleep(0.2)
                    ssl_cp.ssl_repl('\x03', silent=True,
                                    traceback=True)  # KBI
                    time.sleep(0.2)
                    for i in range(1):
                        ssl_cp.ssl_repl('\x0d', silent=True)
                        ssl_cp.flush_conn()
        except Exception as e:
            print("Indicate a host to ping, e.g google.com")

    if cmd_inp.split(' ')[0] == 'lping':
        try:
            ping_dir = cmd_inp.split(' ')[1]
            if ping_dir == 'dev':
                ping_dir = args.t
            ping_cmd_str = 'ping {}'.format(ping_dir)
            ping_cmd = shlex.split(ping_cmd_str)
            old_action = signal.signal(signal.SIGINT, signal.SIG_IGN)

            def preexec_function(action=old_action):
                signal.signal(signal.SIGINT, action)
            try:
                do_ping = subprocess.call(ping_cmd, preexec_fn=preexec_function)
                signal.signal(signal.SIGINT, old_action)
            except Exception as e:
                print(e)
        except Exception as e:
            print("Indicate a host to ping, e.g google.com")

    if cmd_inp.split(' ')[0] == 'timeit':
        try:
            script_to_time = cmd_inp.split(' ')[1]
            if script_to_time.endswith('.py') or script_to_time.endswith('.mpy'):
                print('Timing script: {}'.format(script_to_time))
                script_to_time = script_to_time.replace('.py', '').replace('.mpy', '')
                inp = "from time_it import tzero, tdiff, result;t_0 = tzero();import {0};diff=tdiff(t_0);result('{0}',diff);".format(script_to_time)
            else:
                cmd_to_time = ' '.join(cmd_inp.split()[1:])
                inp = "from time_it import tzero, tdiff, result;t_0 = tzero();{0};diff=tdiff(t_0);result('{0}',diff, cmd=True);".format(cmd_to_time)
            if not encrypted_flag['sec']:
                espdev.wr_cmd(inp)
            else:
                try:
                    ssl_cp.ssl_repl(inp, follow=True)
                    # if espdev.output is not None:
                    #     print(espdev.output)
                except KeyboardInterrupt:
                    time.sleep(0.2)
                    ssl_cp.ssl_repl('\x03', silent=True,
                                    traceback=True)  # KBI
                    time.sleep(0.2)
                    for i in range(1):
                        ssl_cp.ssl_repl('\x0d', silent=True)
                        ssl_cp.flush_conn()
        except Exception as e:
            # print(e)
            print("Indicate a script/command to measure execution time")

    if cmd_inp.split()[0] == 'i2c':
        try:
            i2c_mode = cmd_inp.split()[1]
            if i2c_mode == 'config':
                if dev_platform == 'pyboard':
                    scl = cmd_inp.split()[3]
                    sda = cmd_inp.split()[5]
                    if not encrypted_flag['sec']:
                        espdev.wr_cmd("from machine import I2C;i2c = I2C(scl='{}', sda='{}')".format(scl,sda))
                    else:
                        ssl_cp.ssl_repl("from machine import I2C;i2c = I2C(scl='{}', sda='{}')".format(scl,sda))
                else:
                    scl = cmd_inp.split()[3]
                    sda = cmd_inp.split()[5]
                    if not encrypted_flag['sec']:
                        espdev.wr_cmd("from machine import Pin, I2C;i2c = I2C(scl=Pin({}), sda=Pin({}))".format(scl,sda))
                    else:
                        ssl_cp.ssl_repl("from machine import Pin, I2C;i2c = I2C(scl=Pin({}), sda=Pin({}))".format(scl,sda))
            elif i2c_mode == 'scan':
                i2c_devs = send_custom_sh_cmd("i2c.scan()")
                if len(i2c_devs) > 0:
                    print('Found {} devices/addrs:'.format(len(i2c_devs)))
                    print('int:', i2c_devs)
                    print('Hex:', [hex(dev) for dev in i2c_devs])
                else:
                    print('No device found')
        except Exception as e:
            print('Indicate i2c mode (config/scan)')

    if cmd_inp.split()[0] == 'git':
        cmd_git = shlex.split(cmd_inp)
        if 'push' in cmd_git and 'dev' in cmd_git:  # push changes to "dev repo"
            # Parse changed files
            # try:
            #     latest_commit = shlex.split('git show --name-only --oneline HEAD')
            #     commit_sha = subprocess.check_output(latest_commit).decode().split()[0]
            #     files_diff = shlex.split('git ls-tree --name-only -r {}'.format(commit_sha))
            #     files_to_push = subprocess.check_output(files_diff).decode().splitlines()
            # except Exception as e:
            #     print(e)
            files_to_push = []
            for file_tp in git_diff_files['diff']:
                if file_tp not in files_to_push:
                    files_to_push.append(file_tp)
            # files_to_push = git_diff_files['diff']
            if len(files_to_push) == 0:
                print('Everything up-to-date with {}'.format(dev_platform))
            else:
                # Parse changed files
                print('Changes to push:')
                latest_commit = shlex.split('git show -{} --name-only --oneline HEAD'.format(git_diff_files['n_commits']))
                subprocess.call(latest_commit)
                files_to_put_dir = [fl for fl in files_to_push if os.path.isfile(fl)]
                dirs_to_put_dir = ['/'.join(dir.split('/')[:-1]) for dir in files_to_put_dir]
                # dirs_to_put_dir = [fl for fl in os.listdir() if not os.path.isfile(fl) and fl in files_to_push]
                # print(files_to_put_dir)
                # print(dirs_to_put_dir)
                log_cmd = shlex.split('git log -{}'.format(git_diff_files['n_commits']))
                log_to_dev = subprocess.check_output(log_cmd)
                hash_cmd = shlex.split('git log -{} --pretty=format:"%H"'.format(git_diff_files['n_commits']))
                hash_to_dev = subprocess.check_output(hash_cmd)

                cwd_now = send_custom_sh_cmd("os.getcwd()")
                if len(files_to_put_dir) > 0:
                    if dev_platform == 'pyboard':
                        send_custom_sh_cmd("cd('/')")
                        send_custom_sh_cmd("cd('/flash')")
                    else:
                        send_custom_sh_cmd("cd('/')")
                if len(dirs_to_put_dir) > 0:
                    # print(dirs_to_put_dir)
                    print('New directories to make:')
                    order_dirs = [(len(dir.split('/')), dir) for dir in dirs_to_put_dir if dir != '']
                    order_dirs.sort()
                    # check if already in dev:
                    for _dir in order_dirs:
                        if '/' in _dir[1]:
                            dir_in_dev = send_custom_sh_cmd("os.listdir('{}')".format('/'+'/'.join(_dir[1].split('/')[:-1])))
                        else:
                            dir_in_dev = send_custom_sh_cmd("os.listdir()")
                        if _dir[1].split('/')[-1] not in dir_in_dev:
                            if _dir[1] != '':
                                print('- {}'.format(_dir[1]))
                                send_custom_sh_cmd("os.mkdir('{}')".format(_dir[1]))
                    # for _dir in order_dirs:
                    #     send_custom_sh_cmd("os.mkdir('{}')".format(_dir[1]))
                if not encrypted_flag['sec']:
                    if len(files_to_put_dir) > 0:
                        print('Files to push:')
                        for _file in files_to_put_dir:
                            print('- {}'.format(_file))
                        try:
                            ssl_cp.enable_wrepl_io()
                        except Exception as e:
                            pass
                        # if len(files_to_put_dir) > 1:
                        #     file = ' '.join(files_to_put_dir)
                        #     put_str = 'upydev put -fre {} -t {} -p {} {} -rst f'.format(
                        #         file, args.t, args.p, '')
                        for _file in files_to_put_dir:
                            file_dir = '/'.join(_file.split('/')[:-1])
                            if file_dir != '':
                                put_str = 'upydev put -f {} -t {} -p {} -dir {} -rst f'.format(
                                    _file, args.t, args.p, file_dir)
                            else:
                                put_str = 'upydev put -f {} -t {} -p {} {} -rst f'.format(
                                    _file, args.t, args.p, '')

                            try:
                                put_cmd = shlex.split(put_str)
                                put_file = subprocess.call(put_cmd)
                            except KeyboardInterrupt:
                                try:
                                    ssl_cp.enable_wrepl_io()
                                except Exception as e:
                                    pass
                                pass
                                print('Operation aborted')
                        try:
                            ssl_cp.enable_wrepl_io()
                        except Exception as e:
                            pass
                else:
                    if len(files_to_put_dir) > 0:
                        print('Files to push:')
                    for file in files_to_put_dir:
                        print('- {}'.format(file))
                    ssl_htool.start_SOC()
                    for file in files_to_put_dir:
                        ssl_htool.put(file)
                        time.sleep(0.2)
                    # shr_cp.sh_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_htool.stop_SOC()
                # Save commit log in dev
                print('\nUpdating dev .gitlog.', end='')
                send_custom_sh_cmd("gitlog = open('.gitlog', 'ab+');gitlog.seek(0);gl=gitlog.read();gitlog.seek(0)")
                for log_ in log_to_dev.split(b'\n\n'):
                    send_custom_sh_cmd("gitlog.write({})".format(log_+b'\n\n'))
                    # time.sleep(0.1)
                send_custom_sh_cmd("gitlog.write(gl);gitlog.close()")
                print('\rUpdating dev .gitlog..', end='')
                time.sleep(0.1)
                print('\rUpdating dev .gitlog...', end='')
                send_custom_sh_cmd("gitlog = open('.githash', 'ab+');gitlog.seek(0);gl=gitlog.read();gitlog.seek(0)")
                send_custom_sh_cmd("gitlog.write({})".format(hash_to_dev+b'\n'))
                send_custom_sh_cmd("gitlog.write(gl);gitlog.close()")
                send_custom_sh_cmd("cd('{}')".format(cwd_now))
                print('\r\r', end='')
                git_diff_files['diff'] = []
                git_diff_files['n_commits'] = 0
                # print('TESTING')
        elif 'commit' in cmd_git:
            status_cmd = shlex.split('git status -s')
            files_to_push = [file for file in subprocess.check_output(status_cmd).decode().split('\n') if file != '']
            files_to_pushdev = [file.split()[1] for file in files_to_push if file.split()[0] == 'M' or file.split()[0] == 'A']
            if len(git_diff_files['diff']) == 0:
                git_diff_files['diff'] = files_to_pushdev
                if len(files_to_pushdev) > 0:
                    git_diff_files['n_commits'] = 1
            else:
                git_diff_files['diff'] = git_diff_files['diff'] + [file for file in files_to_pushdev if file not in git_diff_files['diff']]
                if len(files_to_pushdev) > 0:
                    git_diff_files['n_commits'] += 1
            # print(git_diff_files)
            subprocess.call(cmd_git)

        elif 'pull' in cmd_git:
            subprocess.call(cmd_git)
            # git_pull_diff_cmd = shlex.split('git show --name-only --oneline FETCH_HEAD')
            # files_to_pushdev = [file for file in subprocess.check_output(git_pull_diff_cmd).decode().split('\n')[1:] if file != '']
            # git_diff_files['diff'] = files_to_pushdev
            cmd_git = shlex.split('git log --pretty=format:"%H"')
            local_gitlog = subprocess.check_output(cmd_git).decode().split()
            if not encrypted_flag['sec']:
                log_dev_head = send_custom_sh_cmd("cat('.githash')"+';gc.collect()', True)
                log_info_list = [commit_hash.replace('\r', '') for commit_hash in log_dev_head]
                # print(log_info_list)
            else:
                log_dev_head = send_custom_sh_cmd("cat('.githash')"+';gc.collect()')
                # print(log_dev_head)
                log_info_list = log_dev_head.split()
            if local_gitlog == log_info_list:
                pass
            else:
                if len(git_diff_files['diff']) > 0:
                    cmd_git_l = shlex.split('git log --name-status  --pretty=format:"%H"')
                    local_gitlog_files = subprocess.check_output(cmd_git_l).decode()
                    commit_dict = {c.split('\n')[0] : [s for s in c.replace('\t', '\n').split()[1:]] for c in local_gitlog_files.split('\n\n')}
                    for commit in commit_dict.keys():
                        if commit not in log_info_list:
                            if any([file not in git_diff_files['diff'] for file in commit_dict[commit][1::2]]):
                                git_diff_files['diff'] += [file for file in commit_dict[commit][1::2] if file not in git_diff_files['diff']]
                    git_diff_files['n_commits'] = int(len(local_gitlog) - len(log_info_list))
                else:
                    cmd_git_l = shlex.split('git log --name-status  --pretty=format:"%H"')
                    local_gitlog = subprocess.check_output(cmd_git_l).decode()
                    commit_dict = {c.split('\n')[0] : [s for s in c.replace('\t', '\n').split()[1:]] for c in local_gitlog.split('\n\n')}
                    for commit in commit_dict.keys():
                        if commit not in log_info_list:
                            git_diff_files['diff'] += commit_dict[commit][1::2]
                            git_diff_files['n_commits'] += 1


        elif 'clone_dev' in cmd_git:
            if not encrypted_flag['sec']:
                try:
                    ssl_cp.enable_wrepl_io()
                except Exception as e:
                    pass
                file =' '.join([fl for fl in os.listdir() if os.path.isfile(fl) and not fl.startswith('.')])
                put_str = 'upydev put -fre {} -t {} -p {} {} -rst f'.format(
                    file, args.t, args.p, '')
                put_cmd = shlex.split(put_str)
                try:
                    put_file = subprocess.call(put_cmd)
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                except KeyboardInterrupt:
                    try:
                        ssl_cp.enable_wrepl_io()
                    except Exception as e:
                        pass
                    pass
                    print('Operation aborted')

                dirs_to_sync = [fl for fl in os.listdir() if os.path.isdir(fl) and not fl.startswith('.')]
                try:
                    ssl_cp.enable_wrepl_io()
                except Exception as e:
                    pass
                print('Dir/s to sync:')
                for _dir in dirs_to_sync:
                    print('- {}'.format(_dir))

                    d_sync_str = 'upydev d_sync -tree -t {} -p {} -dir {} -rst f'.format(args.t,
                                                                               args.p,
                                                                               _dir)

                    d_sync_cmd = shlex.split(d_sync_str)
                    try:
                        dir_sync = subprocess.call(d_sync_cmd)
                    except KeyboardInterrupt:
                        try:
                            ssl_cp.enable_wrepl_io()
                        except Exception as e:
                            pass
                        pass
                        print('Operation aborted')

                try:
                    ssl_cp.enable_wrepl_io()
                except Exception as e:
                    pass
            else:
                print('File/s to put:')
                file = 'cwd'
                if file == 'cwd':
                    files_to_put = [fl for fl in os.listdir() if os.path.isfile(fl) and not fl.startswith('.')]
                    for file in files_to_put:
                        print('- {}'.format(file))
                    ssl_htool.start_SOC()
                    for file in files_to_put:
                        ssl_htool.put(file)
                        time.sleep(0.2)
                    # ssl_cp.ssl_repl("print('Done!')")
                    time.sleep(0.2)
                    ssl_htool.stop_SOC()

                dirs_to_sync = [fl for fl in os.listdir() if os.path.isdir(fl) and not fl.startswith('.')]
                print('Dir/s to sync:')
                for _dir in dirs_to_sync:
                    print('- {}'.format(_dir))
                    # print('Directory to sync: {}'.format(dir_to_sync))
                    d_sync_recursive(_dir, show_tree=True, root_sync_folder=_dir)

        elif 'log' in cmd_git and 'dev' in cmd_git:
            try:
                if '-a' in cmd_git:
                    if not encrypted_flag['sec']:
                        # espdev.close_wconn()
                        # espdev.cmd("cat('.gitlog')"+';gc.collect()')
                        # espdev.open_wconn()
                        log_dev_head = send_custom_sh_cmd("cat('.gitlog')"+';gc.collect()', True)
                        log_info_list = [line.replace('\r','') for line in log_dev_head if line.replace('\r', '') != '']
                        log_dev_head = '\n'.join(log_info_list)
                        log_dev_head = log_dev_head.replace('\n ', '\n\n ').replace('\ncommit', '\n\ncommit')
                        log_dev_head = log_dev_head.replace('\n\ncommit', '\n\n{}commit'.format(YELLOW)).replace('\nAuthor: ', '{}\nAuthor: '.format(CEND))
                        last_commit_header = log_dev_head.split('\n')[0]
                        if 'Traceback (most recent call last)' in last_commit_header:
                            print('Device {} repo not initiated'.format(dev_platform))
                        else:
                            if last_commit_header != '' and last_commit_header != ' ':
                                log_dev_head = log_dev_head.replace(last_commit_header, '')
                                head_dev = ' ({}HEAD -> {}master{} {}'.format(BCYAN, CGREEN, CEND+YELLOW+','+BCYAN+CRED,dev_platform+CEND+YELLOW)
                                commit = "{}{}){}".format(YELLOW, last_commit_header + head_dev, CEND)
                                print(commit, end='')
                                print(log_dev_head)
                            else:
                                print('No commits in dev log to show')

                    else:
                        # ssl_cp.ssl_repl("cat('.gitlog')"+';gc.collect()', follow=True)
                        log_dev_head = send_custom_sh_cmd("cat('.gitlog')"+';gc.collect()')
                        log_dev_head = log_dev_head.replace('\n ', '\n\n ').replace('\ncommit', '\n\ncommit')
                        log_dev_head = log_dev_head.replace('\n\ncommit', '\n\n{}commit'.format(YELLOW)).replace('\nAuthor: ', '{}\nAuthor: '.format(CEND))
                        last_commit_header = log_dev_head.split('\n')[0]
                        if 'Traceback (most recent call last)' in last_commit_header:
                            print('Device {} repo not initiated'.format(dev_platform))
                        else:
                            log_dev_head = log_dev_head.replace(last_commit_header, '')
                            head_dev = ' ({}HEAD -> {}master{} {}'.format(BCYAN, CGREEN, CEND+YELLOW+','+BCYAN+CRED,dev_platform+CEND+YELLOW)
                            commit = "{}{}){}".format(YELLOW, last_commit_header + head_dev, CEND)
                            print(commit, end='')
                            print(log_dev_head)
                else:
                    if not encrypted_flag['sec']:

                        log_dev_head = send_custom_sh_cmd("head('.gitlog')"+';gc.collect()', True)
                        log_info_list = ''.join([line.replace('\r', '\n') for line in log_dev_head if line.replace('\r', '\n') != ''])
                        log_info_list = log_info_list.split('\ncommit ')[0].replace('\n    ', '\n\n    ', 1)
                        commit_body = '\n'.join(log_info_list.split('\n')[1:])
                        commit_head = log_info_list.split('\n')[0]
                    else:
                        log_dev_head = send_custom_sh_cmd("head('.gitlog')"+';gc.collect()')
                        # print(log_dev_head)
                        log_info_list = log_dev_head.split('\ncommit ')[0].replace('\n    ', '\n\n    ', 1)
                        commit_body = '\n'.join(log_info_list.split('\n')[1:])
                        commit_head = log_info_list.split('\n')[0]
                    if 'Traceback (most recent call last)' in commit_head:
                        print('Device {} repo not initiated'.format(dev_platform))
                    else:
                        if commit_head != '' and commit_head != ' ':
                            head_dev = ' ({}HEAD -> {}master{} {}'.format(BCYAN, CGREEN,CEND+YELLOW+','+BCYAN+CRED,dev_platform+CEND+YELLOW)
                            commit = "{}{}){}".format(YELLOW, commit_head + head_dev, CEND)
                            print(commit)
                            print(commit_body)
                        else:
                            print('No commits in dev log to show')
            except Exception as e:
                print('No commits in dev log to show')

        elif 'log' in cmd_git and 'host' in cmd_git:
            cmd_git = shlex.split('git log -1')
            subprocess.call(cmd_git)

        elif 'init' in cmd_git and 'dev' in cmd_git:
            cwd_now = send_custom_sh_cmd("os.getcwd()")
            if dev_platform == 'pyboard':
                send_custom_sh_cmd("cd('/')")
                send_custom_sh_cmd("cd('/flash')")
            else:
                send_custom_sh_cmd("cd('/')")
            is_gitlog = "'.gitlog' in os.listdir()"
            is_init = send_custom_sh_cmd(is_gitlog)
            if is_init:
                print('Device {} repo already initiated'.format(dev_platform))
            else:
                send_custom_sh_cmd("gitlog = open('.gitlog', 'w');gitlog.close()")
                send_custom_sh_cmd("gitlog = open('.githash', 'w');gitlog.close()")
                print('Device {} repo initiated succesfully!'.format(dev_platform))

            send_custom_sh_cmd("cd('{}')".format(cwd_now))

        elif 'status' in cmd_git and 'dev' in cmd_git:
            cmd_git = shlex.split('git log --pretty=format:"%H"')
            local_gitlog = subprocess.check_output(cmd_git).decode().split()
            try:
                if not encrypted_flag['sec']:
                    log_dev_head = send_custom_sh_cmd("cat('.githash')"+';gc.collect()', True)
                    log_info_list = [commit_hash.replace('\r', '') for commit_hash in log_dev_head if commit_hash != '' and commit_hash != ' ']
                    # print(log_info_list)
                else:
                    log_dev_head = send_custom_sh_cmd("cat('.githash')"+';gc.collect()')
                    # print(log_dev_head)

                    log_info_list = log_dev_head.split()
                if 'Traceback' in log_info_list and 'ENOENT' in log_info_list:
                    log_info_list = []
            except Exception as e:
                log_info_list = []
            if local_gitlog == log_info_list:
                status_msg = "On branch master\nYour branch is up to date with '{}'.\n\nnothing to commit, working tree clean".format(dev_platform)
            else:
                if len(git_diff_files['diff']) > 0:
                    cmd_git_l = shlex.split('git log --name-status  --pretty=format:"%H"')
                    local_gitlog_files = subprocess.check_output(cmd_git_l).decode()
                    commit_dict = {c.split('\n')[0] : [s for s in c.replace('\t', '\n').split()[1:]] for c in local_gitlog_files.split('\n\n')}
                    for commit in commit_dict.keys():
                        if commit not in log_info_list:
                            if any([file not in git_diff_files['diff'] for file in commit_dict[commit][1::2]]):
                                git_diff_files['diff'] += [file for file in commit_dict[commit][1::2] if file not in git_diff_files['diff']]
                    git_diff_files['n_commits'] = int(len(local_gitlog) - len(log_info_list))
                    status_msg = "On branch master\nYour branch is ahead of '{0}' by {1} commit{2}.\n (use 'git push dev' to push your local commits to {0})\n\nnothing to commit, working tree clean".format(dev_platform, git_diff_files['n_commits'], ['s' if git_diff_files['n_commits'] > 1 else '' for i in range(1)][0])
                else:
                    cmd_git_l = shlex.split('git log --name-status  --pretty=format:"%H"')
                    local_gitlog = subprocess.check_output(cmd_git_l).decode()
                    commit_dict = {c.split('\n')[0] : [s for s in c.replace('\t', '\n').split()[1:]] for c in local_gitlog.split('\n\n')}
                    for commit in commit_dict.keys():
                        if commit not in log_info_list:
                            git_diff_files['diff'] += commit_dict[commit][1::2]
                            git_diff_files['n_commits'] += 1
                    status_msg = "On branch master\nYour branch is ahead of '{0}' by {1} commit{2} not tracked by upydev.\n (use 'git push dev' to push last commit changes to {0})\n\nnothing to commit, working tree clean".format(dev_platform, git_diff_files['n_commits'], ['s' if git_diff_files['n_commits'] > 1 else '' for i in range(1)][0])
            print(status_msg)

        elif 'repo' in cmd_git:
            remote_repo_cmd = shlex.split('git remote -v')
            git_repo = subprocess.check_output(remote_repo_cmd)
            if git_repo != b'':
                webbrowser.open(git_repo.split()[1].decode())
            else:
                print('git repository does not have a remote')

        else:
            subprocess.call(cmd_git)

    if cmd_inp == 'tig':
        try:
            tigtool = 'tig'
            tig_cmd = shlex.split(tigtool)
            # old_action = signal.signal(signal.SIGINT, signal.SIG_IGN)
            #
            # def preexec_function(action=old_action):
            #     signal.signal(signal.SIGINT, action)
            try:
                if '.git' in os.listdir():
                    tig = subprocess.call(tig_cmd)
                else:
                    print('tig: Not a git repository')
                # signal.signal(signal.SIGINT, old_action)
            except Exception as e:
                print(e)
                print('tig not found')
        except Exception as e:
            print(e)
            pass

    if cmd_inp == 'upy-config':
        results_array = radiolist_dialog(
                title="Upydev {} Configuration Tool".format(dev_platform),
                text="Select an option and press Select (Use tab to change focus)",
                ok_text="Select",
                cancel_text="Finish",
                values=[
                    ("network", "{:25}{}".format("1 Network Options",
                                                 "Configure network settings")),
                    ("interface", "{:25}{}".format("2 Interfacing Options",
                                                   "Configure connections to peripherals")),
                ]
            ).run()
        if results_array is not None:
            # print('Configure dialog: {}'.format(results_array))
            if results_array == 'network':
                if args.t == '192.168.4.1':
                    results_array = radiolist_dialog(
                            title="Upydev {} Configuration Tool".format(
                                dev_platform),
                            text="Select an option and press Select (Use tab to change focus)",
                            ok_text="Select",
                            cancel_text="Finish",
                            values=[
                                ("STA", "{:10}{}".format("1 STA:",
                                                         "Configure network SSID and PASSWORD to connect to a WLAN"))
                            ]
                        ).run()
                else:
                    results_array = radiolist_dialog(
                            title="Upydev {} Configuration Tool".format(
                                dev_platform),
                            text="Select an option and press Select (Use tab to change focus)",
                            ok_text="Select",
                            cancel_text="Finish",
                            values=[
                                ("AP", "{:10}{}".format("1 AP:",
                                                        "Configure SSID and PASSWORD to enable {} Acces Point".format(dev_platform)))
                            ]
                        ).run()
                if results_array == 'STA':
                    print('Scanning WLANs...')
                    if espdev.platform == 'esp8266':
                        enable_sta = send_custom_sh_cmd(
                            "import network;network.WLAN(network.STA_IF).active()")
                    else:
                        enable_sta = send_custom_sh_cmd(
                            "import network;network.WLAN(network.STA_IF).active(1)")
                    if enable_sta:
                        scan = send_custom_sh_cmd(
                            "network.WLAN(network.STA_IF).scan()")
                        if len(scan) == 0:
                            message_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                           text="No WLAN available").run()
                        else:
                            scan_list = []
                            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)])
                                scan_list.append((netscan[0].decode(), '{0:20}[{1}] RSSI:{2} AUTH: {3}'.format(
                                    netscan[0].decode(), bssid, netscan[3],
                                    auth)))

                            results_array = radiolist_dialog(
                                    title="Upydev {} Configuration Tool".format(
                                        dev_platform),
                                    text="Select an option and press Select (Use tab to change focus)",
                                    ok_text="Select",
                                    cancel_text="Finish",
                                    values=scan_list
                                ).run()
                            if results_array is not None:
                                ssid = results_array
                                passwd = input_dialog(
                                        title="Upydev {} Configuration Tool".format(
                                            dev_platform),
                                        text='Please type {} Password:'.format(ssid)).run()
                                connect_to = "network.WLAN(network.STA_IF).connect('{}', '{}')".format(
                                    ssid, passwd)
                                if passwd is not None:
                                    print('Connecting to {}...'.format(ssid))
                                    disconn_cmd = "network.WLAN(network.STA_IF).disconnect()"
                                    conn_cmd = "{};gc.collect()".format(connect_to)
                                    dconn_sta = send_custom_sh_cmd(disconn_cmd)
                                    # send_custom_sh_cmd('\r\r')
                                    time.sleep(1)
                                    send_custom_sh_cmd('\r\r')
                                    is_connected = False
                                    conn_sta = send_custom_sh_cmd(conn_cmd)
                                    time.sleep(1)
                                    send_custom_sh_cmd('\r\r')
                                    while not is_connected:
                                        is_connected = send_custom_sh_cmd(
                                            "network.WLAN(network.STA_IF).isconnected()")
                                        time.sleep(1)

                                    saveconfig = yes_no_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                                               text="{} connected succesfully to {}\n Do you want to save configuration?".format(dev_platform, ssid)).run()

                                    if saveconfig:
                                        saveconfig_cmd = "import json;sta_config=open('wifi_.config', 'w');sta_config.write(json.dumps(dict(ssid='{}', password='{}')));sta_config.close()".format(
                                            ssid, passwd)
                                        send_custom_sh_cmd(saveconfig_cmd)
                                        message_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                                       text="{} Configuration saved succesfully".format(ssid)).run()

                    else:
                        message_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                       text="Can't enable STA interface").run()

                elif results_array == 'AP':
                    ssid = input_dialog(
                            title="Upydev {} Configuration Tool".format(
                                dev_platform),
                            text='Please set {} SSID:'.format(dev_platform)).run()
                    passwd = input_dialog(
                            title="Upydev {} Configuration Tool".format(
                                dev_platform),
                            text='Please set {} Password:'.format(ssid)).run()
                    if ssid is not None and passwd is not None:
                        ap_enbale = send_custom_sh_cmd(
                            "import network;network.WLAN(network.AP_IF).active(0)")
                        time.sleep(0.5)
                        ap_enbale = send_custom_sh_cmd(
                            "import network;network.WLAN(network.AP_IF).active(1)")
                        print('Configuring {} AP...'.format(ssid))
                        time.sleep(0.5)
                        ap_config = "import network;network.WLAN(network.AP_IF).config(essid='{}',authmode=network.AUTH_WPA_WPA2_PSK, password='{}')".format(
                            ssid, passwd)
                        resp = send_custom_sh_cmd(ap_config)
                        time.sleep(1)
                        saveconfig = yes_no_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                                   text="{} AP:{} enabled succesfully\n Do you want to save configuration?".format(dev_platform, ssid)).run()

                        if saveconfig:
                            saveconfig_cmd = "import json;sta_config=open('ap_.config', 'w');sta_config.write(json.dumps(dict(ssid='{}', password='{}')));sta_config.close()".format(
                                ssid, passwd)
                            send_custom_sh_cmd(saveconfig_cmd)
                            message_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                           text="{} Configuration saved succesfully".format(ssid)).run()
            elif results_array == 'interface':
                results_array = radiolist_dialog(
                        title="Upydev {} Configuration Tool".format(
                            dev_platform),
                        text="Select an option and press Select (Use tab to change focus)",
                        ok_text="Select",
                        cancel_text="Finish",
                        values=[
                            ("I2C", "{:10}{}".format("1 I2C:",
                                                     "Configure I2C interface"))
                        ]
                    ).run()

                if results_array == 'I2C':
                    if dev_platform == 'esp32':
                        default = 'I2C: SCL = 22 , SDA = 23, FREQ = 100000'
                        scl = 22
                        sda = 23
                    elif dev_platform == 'esp8266':
                        default = 'I2C: SCL = 5 , SDA = 4, FREQ = 100000'
                        scl = 5
                        sda = 4
                    elif dev_platform == 'pyboard':
                        default = 'I2C: SCL = X1 , SDA = X2, FREQ = 100000'
                        scl = 'X9'
                        sda = 'X10'
                    results_array = radiolist_dialog(
                            title="Upydev {} Configuration Tool".format(
                                dev_platform),
                            text="Select an option and press Select (Use tab to change focus)",
                            ok_text="Select",
                            cancel_text="Finish",
                            values=[
                                ("I2C_DEF", "{:10}{}".format("1 I2C DEFAULT:",
                                                         default)),
                                ("I2C_CUSTOM", "{:10}{}".format("2 I2C CUSTOM:",
                                                         "SET SCL, SDA and FREQ"))
                            ]
                        ).run()
                    if results_array == 'I2C_DEF':
                        if dev_platform == 'pyboard':
                            if not encrypted_flag['sec']:
                                espdev.wr_cmd("from machine import I2C;i2c = I2C(scl='{}', sda='{}')".format(scl,sda))
                            else:
                                ssl_cp.ssl_repl("from machine import I2C;i2c = I2C(scl='{}', sda='{}')".format(scl,sda))
                        else:
                            if not encrypted_flag['sec']:
                                espdev.wr_cmd("from machine import Pin, I2C;i2c = I2C(scl=Pin({}), sda=Pin({}))".format(scl,sda))
                            else:
                                ssl_cp.ssl_repl("from machine import Pin, I2C;i2c = I2C(scl=Pin({}), sda=Pin({}))".format(scl,sda))

                        message_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                       text="I2C Default configuration enabled succesfully as 'i2c'").run()
                    elif results_array == 'I2C_CUSTOM':
                        custom_i2c = input_dialog(
                                title="Upydev {} Configuration Tool".format(
                                    dev_platform),
                                text='Please set SCL SDA and FREQ separated by space, e.g. : 22 23 100000:'.format(dev_platform)).run()
                        if custom_i2c is None or custom_i2c == '':
                            freq = 100000
                            custom_i2c = "{} {} {}".format(scl, sda, freq)
                        else:
                            scl = custom_i2c.split()[0]
                            sda = custom_i2c.split()[1]
                            freq = custom_i2c.split()[2]
                        if dev_platform == 'pyboard':
                            if not encrypted_flag['sec']:
                                espdev.wr_cmd("from machine import I2C;i2c_custom = I2C(scl='{}', sda='{}', freq={})".format(scl, sda, freq))
                            else:
                                ssl_cp.ssl_repl("from machine import I2C;i2c_custom = I2C(scl='{}', sda='{}', freq={})".format(scl, sda, freq))
                        else:
                            if not encrypted_flag['sec']:
                                espdev.wr_cmd("from machine import Pin, I2C;i2c_custom = I2C(scl=Pin({}), sda=Pin({}), freq={})".format(scl, sda, freq))
                            else:
                                ssl_cp.ssl_repl("from machine import Pin, I2C;i2c_custom = I2C(scl=Pin({}), sda=Pin({}), freq={})".format(scl, sda, freq))

                        message_dialog(title="Upydev {} Configuration Tool".format(dev_platform),
                                       text="I2C Custom configuration enabled succesfully as 'i2c_custom'").run()

    if cmd_inp.split(' ')[0] == 'wss':
        try:
            mode = cmd_inp.split(' ')[1]
            if mode == 'on':
                send_custom_sh_cmd("sslconfig = open('ssl_flag.py', 'wb')")
                send_custom_sh_cmd("sslconfig.write({});sslconfig.close()".format(b'SSL = True'))
                print('WebSecureREPL enabled')

            elif mode == 'off':
                send_custom_sh_cmd("sslconfig = open('ssl_flag.py', 'wb')")
                send_custom_sh_cmd("sslconfig.write({});sslconfig.close()".format(b'SSL = False'))
                print('WebSecureREPL disabled')
        except Exception as e:
            print('Select a mode (on/off) to enable or disable WebSecureREPL')
    if cmd_inp.split(' ')[0] == 'exit':
        # print('closing...')
        if encrypted_flag['sec']:
            # print('SHUTDOWN SSL') # make exit or exit -r
            if reset_flag['R']:
                if len(cmd_inp.split(' ')) > 1:
                    if cmd_inp.split(' ')[1] == '-nr':
                        ssl_cp.ssl_repl("ssl_client.switch_wrepl();gc.collect()")
                        time.sleep(0.5)
                    elif cmd_inp.split(' ')[1] == '-hr':
                        print('Device Hard Reset...')
                        ssl_cp.send_message("import machine;machine.reset()\r")
                        print('Done!')
                else:
                    print('Rebooting device...')
                    ssl_cp.send_message("ssl_client.switch_wrepl();gc.collect()\r\x04")
                    print('Done!')
            else:
                ssl_cp.ssl_repl("ssl_client.switch_wrepl();gc.collect()")
                time.sleep(0.5)
            espdev.close_wconn()
        # time.sleep(0.5)
        else:
            espdev.close_wconn()
            if len(cmd_inp.split(' ')) > 1:
                if cmd_inp.split(' ')[1] == '-r':
                    print('Rebooting device...')
                    espdev.reset(output=False)
                    print('Done!')
                elif cmd_inp.split(' ')[1] == '-hr':
                    print('Device Hard Reset...')
                    espdev.cmd_nb("import machine;machine.reset()", block_dev=False)
                    print('Done!')
            if reset_flag['R']:
                print('Rebooting device...')
                espdev.reset(output=False)
                print('Done!')
        exit_flag['exit'] = True


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


# KEYBINDINGS

@kb.add('c-x')
def exitpress(event):
    " Exit SSLWebREPL terminal "
    if shell_mode['S']:
        print('\nclosing...')
    else:
        print('\n>>> closing...')
    if encrypted_flag['sec']:
        if reset_flag['R']:
            print('Rebooting device...')
            ssl_cp.send_message("\x04")
            print('Done!')
        else:
            ssl_cp.ssl_repl("ssl_client.switch_wrepl()")
            time.sleep(0.5)
        espdev.close_wconn()
    # time.sleep(0.5)
    else:
        espdev.close_wconn()
        time.sleep(0.5)
        if reset_flag['R']:
            print('Rebooting device...')
            espdev.reset(output=False)
            print('Done!')
    exit_flag['exit'] = True
    event.app.exit()


@kb.add('c-b')
def bannerpress(event):
    "Prints MicroPython sys version info"
    def upy_sysversion():
        if encrypted_flag['sec']:
            ssl_cp.ssl_repl("\x02", banner=True)
        else:
            espdev.wr_cmd('B')
    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')
        if not encrypted_flag['sec']:
            espdev.wr_cmd('ls;gc.collect()')
        else:
            ssl_cp.ssl_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()')
        if not encrypted_flag['sec']:
            espdev.wr_cmd('from micropython import mem_info; mem_info()')
        else:
            ssl_cp.ssl_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()')
        if not encrypted_flag['sec']:
            espdev.wr_cmd('import gc;gc.collect()')
        else:
            ssl_cp.ssl_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.split()[0] in custom_sh_cmd_kw:
                custom_sh_cmd(last_cmd)
            else:
                inp, frst_cmd = map_upysh(last_cmd)
                if frst_cmd == 'run':
                    if not encrypted_flag['sec']:
                        espdev.wr_cmd(inp)
                    else:
                        try:
                            ssl_cp.ssl_repl(inp, follow=True)
                            # if espdev.output is not None:
                            #     print(espdev.output)
                        except KeyboardInterrupt:
                            time.sleep(0.2)
                            ssl_cp.ssl_repl('\x03', silent=True,
                                            traceback=True)  # KBI
                            time.sleep(1)
                            for i in range(3):
                                ssl_cp.ssl_repl('\x0d', silent=True)
                                ssl_cp.flush_conn()
                            pass
                else:
                    if not encrypted_flag['sec']:
                        espdev.wr_cmd(inp)
                    else:
                        if inp == 'ls()' or frst_cmd == 'tree' or frst_cmd == 'cat':
                            ssl_cp.ssl_repl(inp+';gc.collect()', follow=True)
                            # ssl_cp.send_message('gc.collect()\r')
                        else:
                            ssl_cp.ssl_repl(inp)
                            if espdev.output is not None:
                                print(espdev.output)
        else:
            print('>>> {}'.format(last_cmd))
            if not encrypted_flag['sec']:
                espdev.wr_cmd(last_cmd)
            else:
                try:
                    ssl_cp.ssl_repl(last_cmd, follow=True)
                    # if espdev.output is not None:
                    #     print(espdev.output)
                except KeyboardInterrupt:
                    time.sleep(0.2)
                    ssl_cp.ssl_repl('\x03', silent=True,
                                    traceback=True)  # KBI
                    time.sleep(0.2)
                    for i in range(1):
                        ssl_cp.ssl_repl('\x0d', silent=True)
                        ssl_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))
        if not encrypted_flag['sec']:
            espdev.wr_cmd(test_cmd)
        else:
            try:
                ssl_cp.ssl_repl(test_cmd, follow=True)
                # if espdev.output is not None:
                #     print(espdev.output)
            except KeyboardInterrupt:
                time.sleep(0.2)
                ssl_cp.ssl_repl('\x03', silent=True,
                                traceback=True)  # KBI
                time.sleep(0.2)
                for i in range(1):
                    ssl_cp.ssl_repl('\x0d', silent=True)
                    ssl_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))
        if not encrypted_flag['sec']:
            espdev.wr_cmd(reload_cmd)
        else:
            ssl_cp.ssl_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
        if not encrypted_flag['sec']:
            espdev.wr_cmd('from upysh import *', silent=True)
        else:
            ssl_cp.ssl_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:
            if encrypted_flag['sec']:
                ssl_cp.ssl_repl('\x03') # KBI
            else:
                espdev.wr_cmd('C')
            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 = ''
            if not encrypted_flag['sec']:
                file_content = send_custom_sh_cmd(
                    "cat('{}')".format(edit_mode['File']), True)
                if len(file_content) > 0:
                    file_content = '\n'.join(file_content).replace('\r', '')
            else:
                ssl_cp.ssl_repl("cat('{}')".format(edit_mode['File']))
                # espdev.get_output()
                # print(espdev.output.split('\n'))
                file_content = espdev.output
                # print(len(file_content))
                # file_content = send_custom_sh_cmd("cat('{}')".format(edit_mode['File']))
            # paste_flag['p'] = True
            # event.app.current_buffer.insert_text(file_content)
            # print(file_content)
            try:
                # process_content = ''.join(file_content).replace('\r', '\n')
                process_content = file_content
                # print(len(process_content))
            except Exception as e:
                process_content = ' '
            if isinstance(file_content, str):
                event.app.current_buffer.insert_text(process_content)
        else:
            # "Move cursor to final position"
            buff_text = event.app.current_buffer.document.text
            event.app.current_buffer.reset()
            event.app.current_buffer.insert_text(buff_text, move_cursor=True)


@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 != '':
                # espdev.wr_cmd("cp.gbls = globals()", silent=True)
                if not encrypted_flag['sec']:
                    # block_commad = uparser_dec('\n' + buff_text, end='\r\r\r')
                    # espdev.wr_cmd(block_commad)
                    # print(block_commad.splitlines())
                    espdev._wconn.child.sendline('E')
                    time.sleep(0.2)
                    for line in buff_text.split('\n'):
                        espdev._wconn.child.sendline(line)
                        cmd_echo = espdev._wconn.child.readline()
                        # time.sleep(0.2)
                    # espdev._wconn.child.sendline('D')
                    try:
                        cmd_echo = espdev._wconn.child.read_nonblocking(2**16, 0.2)
                    except Exception as e:
                        pass
                    espdev.wr_cmd('D')

                else:
                    # espdev.wr_cmd("cp.gbls = globals()", silent=True)
                    try:
                        ssl_cp.paste_buff(buff_text)
                        ssl_cp.ssl_repl("\x04", follow=True)
                    except KeyboardInterrupt:
                        time.sleep(0.2)
                        ssl_cp.ssl_repl('\x03', silent=True,
                                        traceback=True)  # KBI
                        time.sleep(0.2)
                        for i in range(1):
                            ssl_cp.ssl_repl('\x0d', silent=True)
                            ssl_cp.flush_conn()
                        pass
                    # ssl_cp.ssl_repl('', just_recv=True, silent=False) blocking
                    # if espdev.output is not None:
                    #     print(espdev.output)
                    espdev.output = None
                    time.sleep(1)
                    if encrypted_flag['sec']:
                        ssl_cp.flush_conn()
                        ssl_cp.flush_conn()
            # print('paste mode; Ctrl-C to cancel, Ctrl-D to finish')
            event.app.current_buffer.reset()
            # print(buff_text)
            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))
                        # if not encrypted_flag['sec']:
                        send_custom_sh_cmd('_file_to_edit.write(nl)')
                    # time.sleep(0.1)
                send_custom_sh_cmd("_file_to_edit.close()")
                send_custom_sh_cmd("gc.collect()")
                time.sleep(0.5)
                if encrypted_flag['sec']:
                    ssl_cp.flush_conn()
                    ssl_cp.flush_conn()
            # print('paste mode; Ctrl-C to cancel, Ctrl-D to finish')
            event.app.current_buffer.reset()
            # print(buff_text)
            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'] = ['']
        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
            if ').' not in event.app.current_buffer.document.text:
                buff_text = event.app.current_buffer.document.text.replace('=', ' ').replace('(', ' ').split(' ')[-1]
            else:
                buff_text = event.app.current_buffer.document.text.replace('=', ' ').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 != '':
                        if encrypted_flag['sec']:
                            ssl_cp.ssl_repl("[val for val in dir({}) if val.startswith('{}')]".format(root_text, rest))
                            ssl_cp.flush_conn()
                        else:
                            espdev.wr_cmd("[val for val in dir({}) if val.startswith('{}')]".format(root_text, rest), silent=True)
                    else:
                        if encrypted_flag['sec']:
                            try:
                                ssl_cp.ssl_repl('dir({});gc.collect()'.format(root_text),
                                                debug=False, prevent_hang=True)
                            except KeyboardInterrupt:
                                time.sleep(0.2)
                                ssl_cp.ssl_repl('\x03', silent=True,
                                                traceback=True)

                            ssl_cp.flush_conn()
                        else:
                            espdev.wr_cmd('dir({})'.format(root_text), silent=True)
                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:
                        fbuff_text = event.app.current_buffer.document.text.split()
                        if 'import' in fbuff_text and 'from' in fbuff_text and len(fbuff_text) >= 3:
                            if fbuff_text[1] not in frozen_modules['FM']:
                                if len(fbuff_text) == 3:
                                    cmd_ls_glb = "import {0};dir({0});del(sys.modules['{0}'])".format(fbuff_text[1])
                                if len(fbuff_text) == 4:
                                    cmd_ls_glb = "import {0};[val for val in dir({0}) if val.startswith('{1}')];del(sys.modules['{0}'])".format(fbuff_text[1], fbuff_text[3])
                            else:
                                if len(fbuff_text) == 3:
                                    cmd_ls_glb = "import {0};dir({0})".format(fbuff_text[1])
                                if len(fbuff_text) == 4:
                                    cmd_ls_glb = "import {0};[val for val in dir({0}) if val.startswith('{1}')]".format(fbuff_text[1], fbuff_text[3])

                        else:
                            cmd_ls_glb = "[scp.split('.')[0] for scp in os.listdir()+os.listdir('/lib') if '.py' in scp]"
                            frozen_modules['SUB'] = frozen_modules['FM']
                            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)
                                frozen_modules['SUB'] = [mod for mod in frozen_modules['FM'] if mod.startswith(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)
                    if encrypted_flag['sec']:
                        try:
                            ssl_cp.ssl_repl(cmd_ls_glb+';gc.collect()',
                                            debug=False, prevent_hang=True)
                        except KeyboardInterrupt:
                            time.sleep(0.2)
                            ssl_cp.ssl_repl('\x03', silent=True,
                                            traceback=True)
                        ssl_cp.flush_conn()
                    else:
                        espdev.wr_cmd(cmd_ls_glb, silent=True)

            else:
                root_text = buff_text.split('.')[0]
                rest = buff_text.split('.')[1]
                if encrypted_flag['sec']:
                    try:
                        ssl_cp.ssl_repl('dir({});gc.collect()'.format(root_text))
                    except KeyboardInterrupt:
                        time.sleep(0.2)
                        ssl_cp.ssl_repl('\x03', silent=True,
                                        traceback=True)
                    ssl_cp.flush_conn()
                else:
                    espdev.wr_cmd('dir({})'.format(root_text), silent=True)
        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 glb:
                        kw_line_buff = event.app.current_buffer.document.text.split()
                        if len(kw_line_buff) > 0 and len(kw_line_buff) <= 2:
                            if 'import' == kw_line_buff[0] or 'from' == kw_line_buff[0]:
                                espdev.output = espdev.output + frozen_modules['SUB']
                    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()
                    # if encrypted_flag['sec']:
                    #     h_cp.crepl(cmd_ls_glb)
                    # else:
                    #     espdev.wr_cmd(cmd_ls_glb, silent=True)

            else:
                pass
                # root_text = buff_text.split('.')[0]
                # rest = buff_text.split('.')[1]
                # if encrypted_flag['sec']:
                #     h_cp.crepl('dir({})'.format(root_text))
                # else:
                #     espdev.wr_cmd('dir({})'.format(root_text), silent=True)
        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
#             ssl_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
#             ssl_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_ram_msg(event):
    "Toggle Right RAM status"
    if status_encryp_msg['Toggle']:
        if status_encryp_msg['S']:
            status_encryp_msg['S'] = False
        else:
            status_encryp_msg['S'] = True


@kb.add('c-g')
def listen_SSL_WREPL(event):
    "Echo SSL_WREPL --> Active listening for timer/hardware interrupts"
    if encrypted_flag['sec']:
        # espdev.wr_cmd("cp.gbls = globals()", silent=True)
        def echo_ouput():
            try:
                buff_text = "import time\nwhile True:\n    time.sleep(1)"
                ssl_cp.paste_buff(buff_text)
                ssl_cp.ssl_repl("\x04", follow=True)
            except KeyboardInterrupt:
                time.sleep(0.2)
                ssl_cp.ssl_repl('\x03', silent=True,
                                traceback=True)  # KBI
                time.sleep(0.2)
                for i in range(1):
                    ssl_cp.ssl_repl('\x0d', silent=True)
                    ssl_cp.flush_conn()
                pass
            espdev.output = None
            time.sleep(1)
            if encrypted_flag['sec']:
                ssl_cp.flush_conn()
                ssl_cp.flush_conn()
            event.app.current_buffer.reset()
        run_in_terminal(echo_ouput)

    if not encrypted_flag['sec']:
        # block_commad = uparser_dec('\n' + buff_text, end='\r\r\r')
        # espdev.wr_cmd(block_commad)
        # print(block_commad.splitlines())
        def echo_ouput_nem():
            buff_text = "import time\nwhile True:\n    time.sleep(1)"
            espdev._wconn.child.sendline('E')
            time.sleep(0.2)
            for line in buff_text.split('\n'):
                espdev._wconn.child.sendline(line)
                cmd_echo = espdev._wconn.child.readline()
                # time.sleep(0.2)
            # espdev._wconn.child.sendline('D')
            try:
                cmd_echo = espdev._wconn.child.read_nonblocking(2**16, 0.2)
            except Exception as e:
                pass
            espdev.wr_cmd('D')
        run_in_terminal(echo_ouput_nem)

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':
                            if not encrypted_flag['sec']:

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

                        else:
                            if not encrypted_flag['sec']:
                                if frst_cmd != 'cat':
                                    espdev.wr_cmd(inp)
                                else:
                                    espdev.close_wconn()
                                    espdev.cmd(inp)
                                    espdev.open_wconn()
                            else:
                                if inp == 'ls()' or frst_cmd == 'cat' or frst_cmd == 'tree' or 'help' in frst_cmd:
                                    ssl_cp.ssl_repl(inp+';gc.collect()', follow=True)
                                    ssl_cp.flush_conn()
                                else:
                                    ssl_cp.ssl_repl(inp)
                                    if espdev.output is not None:
                                        print(espdev.output)
                        if frst_cmd == 'cd':
                            espdev.output = None
                            if not encrypted_flag['sec']:
                                espdev.wr_cmd('pwd', silent=True)
                                espdev.output = espdev.response
                            else:
                                ssl_cp.ssl_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
                    if not encrypted_flag['sec']:
                        espdev.wr_cmd(inp)
                    else:
                        try:
                            ssl_cp.ssl_repl(inp, follow=True)
                            # if espdev.output is not None:
                            #     print(espdev.output)
                        except KeyboardInterrupt:
                            time.sleep(0.2)
                            ssl_cp.ssl_repl('\x03', silent=True,
                                            traceback=True)  # KBI
                            time.sleep(0.2)
                            for i in range(1):
                                ssl_cp.ssl_repl('\x0d', silent=True)
                                ssl_cp.flush_conn()
                            pass
                if inp != '':  # PASTE BUFFER
                    pass
                    # paste_buffer['B'].append(inp)
    except Exception as e:
        print(e)
        continue
    except KeyboardInterrupt:
        try:
            # print('KBI_1!!')
            # send_KBI()
            if encrypted_flag['sec']:
                ssl_cp.flush_conn()
                if hasattr(SSL_TOOL, 'conn'):
                    ssl_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')
if host_name != unique_id:
    print('Connection to {} closed.'.format(host_name))
else:
    print('Connection to {} closed.'.format(args.t))
