main.py

#!/usr/bin/env python3
import os,sys,argparse,termios,tty,time,signal
import file_list
import text_pager
import string_util as su
import colors
import view_layout
import select
import debug
import text_input
import table

stdin_fd = sys.stdin.fileno()


try:
    old_term_settings = termios.tcgetattr(stdin_fd)
except:
    old_term_settings = None

def reset_term():
    if old_term_settings == None:
        return
    termios.tcsetattr(stdin_fd, termios.TCSADRAIN, old_term_settings)

def getchar(timeout:float=0.01):
    ch = ""
    #Returns a single character from standard input
    try:
        tty.setraw(stdin_fd,when=termios.TCSANOW)

        i, o, e = select.select( [sys.stdin], [], [], timeout)

        if i:
            ch = (sys.stdin.buffer.raw.read(1)).decode('utf-8')
        else:
            return None
    finally:
        reset_term()
    return ch

def getstr():
    s = ""
    ret = ""
    timeout=1 # the first getchar has a big timeout
    while ret != None:
        s += ret
        ret = getchar(timeout)
        timeout=0.01 # subsequent getchars have a small timeout
    return s

parser = argparse.ArgumentParser(
    prog='Explorer',
    description='Explores folders',
    epilog='Schpaloingus!')

parser.add_argument('dir', nargs='?', default=os.getcwd())
parser.add_argument('-o', '--once', action='store_true')
parser.add_argument('-b', '--batch', action='store_true')
parser.add_argument('-n', '--iterations', type=int, default=-1)
parser.add_argument('-p', '--profile', action='store_true')

args = parser.parse_args()

path = args.dir

layout = view_layout.ViewLayout(None,[],10,10)

files = file_list.FileList(layout, path)
pager = text_pager.TextPager(layout, height=50)
search_bar = text_input.TextInput(layout,100,1,"Find: ",placeholder=" ",name="Search bar",use_border=False)
search_bar.hide()

@search_bar.set_ontype
def ontype(self: text_input.TextInput):
    @files.table.set_filter
    def filter(row: table.TableRow):
        name=su.strip_ansi(row.data["name"])
        return self.text in name

@search_bar.set_onsubmit
def onsubmit(self: text_input.TextInput):
    layout.set_focus_by_name("FileList")
    self.hide()

layout.structure = [
    [files,pager],
    [search_bar]
]

pager.visible = False

if args.profile:
    import functiontrace
    functiontrace.trace()

debug_pager = text_pager.TextPager(None,100,15,"Debug Log","DebugPager")
debug_pager.onfocus()


@files.set_onsubmit
def submit(self):
    global path,files,pager
    debug.log(f"Changing directory!")
    file = files.get_selected_file()
    if file != None:
        if os.path.isdir(os.path.realpath(file.path)): # use os.path.realpath to follow symlinks
            path = os.path.abspath( os.path.join( path, file.get_name() ) )
            update_files()
            if file.get_name() == "..": # if going up, then select the previous folder
                new_files=files.table.get_field("name")
                try:
                    index = new_files.index( os.path.basename(os.path.dirname(file.path)) )
                except ValueError:
                    index = 1
                files.table.selected = index
        elif os.path.isfile(os.path.realpath(file.path)):
            pager.visible=True
            pager.load_from_file(file)

running = True

if args.once:
    running = False

def update_files():
    files.set_dir(path)

def update():
    debug_pager.text=debug.debug_text
    try:
        term_size = os.get_terminal_size()
        layout.width = term_size.columns-2
        layout.height = term_size.lines-1
        files.height = layout.height - 10
        search_bar.width=layout.width
        if pager.visible:
            files.width = int(term_size.columns/2 - 1)
            pager.height = files.height
        else:
            files.width = term_size.columns
        pager.width = int(term_size.columns/2 - 2)
    except OSError:
        # term_size = os.terminal_size([],{})
        pass
    pager.update()
    files.update_table()
    debug_pager.scroll_y=debug_pager.get_line_count()
    debug_pager.update()

def view():
    s = "\x1b[2J\x1b[H\x1b[0m"
    if iteration%2==0:
        s += "#"
    else:
        s += " "
    s += f"==== {path} ====\n"
    s += f"{len(files.get_files())} files, "
    s += f"{len(files.get_folders())} folders, "
    s += f"{len(files.get_links())} links"
    s += "\n"
    s += f"{files.table.width}x{files.table.height}"
    s += "\n\n"
    
    s += layout.view()

    if debug.debug_mode:
        x=layout.width-debug_pager.width-1
        s += "\x1b[s" # save cursor position
        s += su.goto(x,4)
        s += su.float_text(debug_pager.view())
        s += "\x1b[u" # restore cursor position

    return s

def handle_input(char):
    global running,path,show_pager,pager_focused
    
    debug.log(f"Input char {repr(char)}")
    if char=="\x03": #Ctrl-C
        running=False


    elif char=="d":
        debug.debug_mode = not debug.debug_mode
    elif char=="q":
        running=False
    else:
        layout.input(char)

if __name__ == "__main__":
    iteration = 0
    def sigwinch_handle(signum,frame):
        reset_term() # reset terminal to not mess up printing
        update()
        print(view())
        tty.setraw(stdin_fd,when=termios.TCSANOW) # set tty back to raw mode for input
    signal.signal(signal.SIGWINCH,sigwinch_handle)
    update()
    print(view())
    while running:
        if args.iterations > 0 and iteration >= args.iterations:
            running=False
        if not args.batch:
            text = getstr()
            if text != "":
                handle_input(text)
        update()

        # print("Loading...")
        print(view())
        iteration += 1