Velocity/core/buffers.py

import os


class Buffer:
    def __init__(self, path: str, name: str = None, data: list = None):
        self.path = path
        self.name = name or "[No Name]"
        self.data = data or [""]

    def render(self, instance):
        for y, line in enumerate(self.data[instance.offset[0]:]):
            if y <= instance.safe_height:
                for x, character in enumerate(line[instance.offset[1]:]):
                    if x <= instance.safe_width:
                        instance.screen.addstr(y, x + instance.components.get_component_width(
                            instance.components.components["left"]), character)

                # Write blank spaces for the rest of the line
                if instance.safe_width - len(line) > 0:
                    instance.screen.addstr(y, instance.components.get_component_width(
                        instance.components.components["left"]) + len(line), " " * (instance.safe_width - len(line)))

    @staticmethod
    def remove_char(instance):
        # Remove a character from a string at a given index
        instance.buffer.data[instance.cursor[0]] = instance.buffer.data[instance.cursor[0]][:(instance.cursor[1] - 1)] \
                                                   + instance.buffer.data[instance.cursor[0]][(instance.cursor[1] - 1) + 1:]

    @staticmethod
    def insert_char(instance, char: (str, chr)):
        # Insert a character into a string at a given index
        instance.buffer.data[instance.cursor[0]] = instance.buffer.data[instance.cursor[0]][:instance.cursor[1]] + \
                                                   char + \
                                                   instance.buffer.data[instance.cursor[0]][instance.cursor[1]:]


def open_file(file_name):
    # Open the file
    with open(file_name) as f:
        # Convert it into a list of lines
        lines = f.readlines()

    # Add a line if the file is empty or if the last line is not empty
    if lines[-1].endswith("\n") or not len(lines):
        lines.append("")

    # Remove the newlines
    lines = [line.rstrip("\n") for line in lines]

    # Return the list of lines
    return lines


def load_file(file_path=None):
    # Default settings for a file
    file_name = "[No Name]"
    file_data = [""]

    if file_path:
        # Set the file's name
        file_name = os.path.basename(file_path)

        # Only if the file actually exists
        if os.path.exists(file_path):
            # Open the file as a list of lines
            file_data = open_file(file_name)

    # Return a dictionary which will become all the data about the buffer
    return Buffer(file_path, file_name, file_data)