Velocity/main.py

from core import colors, cursors, buffers, modes, utils
from core.buffers import Buffer
from core.components import Components
import traceback
import argparse
import curses
import sys
import os


class Lambda:
    def __init__(self, buffer: Buffer):
        self.screen = curses.initscr()
        self.buffer = buffer
        self.cursor = [0, 0]
        self.mode = "normal"
        self.components = Components()
        self.height = self.screen.getmaxyx()[0]
        self.width = self.screen.getmaxyx()[1]
        self.safe_height = self.height - len(self.components.components["bottom"])
        self.config = {"icon": "λ"}

    def update_dimensions(self):
        # Calculate the entire height and width of the terminal
        self.height, self.width = self.screen.getmaxyx()

        # Calculate the safe area for the buffer by removing heights & widths of components
        self.safe_height = self.height - len(self.components.components["bottom"])

    def update(self):
        # Update the dimensions
        self.update_dimensions()

        # Refresh the on-screen components
        self.components.render(self)

        # Move the cursor
        cursors.cursor_move(self.screen, self.cursor)

    def start(self):
        # Initialise colors
        colors.init_colors()

        # Change the cursor shape
        cursors.cursor_mode("block")

        # Don't echo any key-presses
        curses.noecho()

        # Show a welcome message if lambda opens with no file
        if not self.buffer.path:
            utils.welcome(self.screen)

        # Main loop
        self.run()

    def run(self):
        # The main loop, which runs until the user quits
        while True:
            # Write the buffer to the screen
            # buffers.write_buffer(screen, buffer)

            # Update the screen
            self.update()

            # Wait for a keypress
            key = self.screen.getch()

            # Handle the key
            modes.handle_key(self, key)

            # Refresh and clear the screen
            self.screen.refresh()
            self.screen.clear()


def main():
    # Shell arguments
    parser = argparse.ArgumentParser(description="Next generation hackable text editor for nerds.")
    parser.add_argument("file", metavar="file", type=str, nargs="?",
                        help="The name of a file for lambda to open")

    # Collect the arguments passed into lambda at the shell
    args = parser.parse_args()

    # Load the file into a Buffer object
    buffer = buffers.load_file(args.file)

    # Change the escape delay to 25ms
    # Fixes an issue where esc takes way too long to press
    os.environ.setdefault("ESCDELAY", "25")

    # Load lambda with the buffer object
    screen = Lambda(buffer)

    # Start the screen, this will loop until exit
    try:
        screen.start()

    # KeyboardInterrupt is thrown when <C-c> is pressed (exit)
    except KeyboardInterrupt:
        # Clean up the screen
        curses.endwin()

        # Then, just exit
        sys.exit()

    # Excepts *any* errors that occur
    except Exception as exception:
        # Clean up the screen
        curses.endwin()

        # Print the error message and traceback
        print(f"{colors.Codes.red}FATAL ERROR:{colors.Codes.end} "
              f"{colors.Codes.yellow}{exception}{colors.Codes.end}\n")
        print(traceback.format_exc())

        # Exit, with an error code
        sys.exit(0)


if __name__ == "__main__":
    main()