VIM-like-text-editor/mvc/models.py

from enum import Enum

class ReturnCode(Enum):
    GOOD = -101
    EXIT_CODE = -100
    SET_BASIC_MODE = -99
    SET_COMMAND_MODE = -98
    SET_EDIT_MODE = -97
    CONTINUE = -96

class VimModel:
    def __init__(self):
        self.displayBuffer = []  # буфер для хранения всех строк
        self.currentLine = 0  # текущий индекс строки
        self.currentCol = 0  # текущий индекс колонки
        self.scrollY = 0  # вертикальная прокрутка
        self.scrollX = 0  # горизонтальная прокрутка
        self.file_path = ""  # путь к файлу
        self.mode = ""
        self.commandBuffer = [] # буффер для команды
        self.exchangeBuffer = [] # буффер обмена

    def ModeBar(self) -> str:
        modeBar = f"MODE: {self.mode} | FILE: {self.file_path} | LINE: {self.currentLine+1}/{len(self.displayBuffer)}"
        if self.mode == "COMMAND":
            modeBar += f" | COMMAND BUFFER: {''.join(self.commandBuffer)}"
        return modeBar

    def Scroll(self, displayLines: int, displayCols: int) -> None:
        if self.currentLine < self.scrollY:
            self.scrollY = self.currentLine
        elif self.currentLine >= self.scrollY + displayLines - 1:
            self.scrollY = self.currentLine - displayLines + 2

        if self.currentCol < self.scrollX:
            self.scrollX = self.currentCol
        elif self.currentCol >= self.scrollX + displayCols:
            self.scrollX = self.currentCol - displayCols + 1

    def InsertCommandSymbol(self, symbolCode: int) -> None:
        self.commandBuffer.append(chr(symbolCode))

    def InsertSymbol(self, symbolCode: int) -> None:
        if self.currentCol <= len(self.displayBuffer[self.currentLine]):  # проверяем, не превышает ли индекс колонки длину строки
            self.displayBuffer[self.currentLine].insert(self.currentCol, chr(symbolCode))
            self.currentCol += 1

    def EnterCommand(self):
        """Обработка введенной команды"""
        cmd = ''.join(self.commandBuffer)
        self.commandBuffer.clear()
        match cmd:
            case "$": # Перемещение в конец строки
                self.currentCol = len(self.displayBuffer[self.currentLine])
            case "^" | "0": # Перемещение в начало строки
                self.currentCol = 0
            case "q": # Выход из программы
                return ReturnCode.EXIT_CODE
            case "i": # Вход в режим редактирования
                return ReturnCode.SET_EDIT_MODE
            case "I": # Переходит в начало строки и начинает ввод текста
                self.currentCol = 0
                return ReturnCode.SET_EDIT_MODE
            case "A": # Переходит в конец строки и начинает ввод текста
                self.currentCol = len(self.displayBuffer[self.currentLine])
                return ReturnCode.SET_EDIT_MODE
            case "S": # Удаление строки на которой курсор и вход в режим редактирования
                self.currentCol = 0
                self.displayBuffer[self.currentLine] = []
                return ReturnCode.SET_EDIT_MODE
            case "RIGHT": # Перемещение курсора на 1 позицию вправо
                self.currentCol += 1
            case "LEFT": # Перемещение курсора на 1 позицию влево
                if self.currentCol > 0:
                    self.currentCol -= 1
            case "UP": # Перемещение курсора на 1 пизицию вверх
                if self.currentLine > 0:
                    self.currentLine -= 1
                    if self.currentCol > len(self.displayBuffer[self.currentLine]):
                        self.currentCol = len(self.displayBuffer[self.currentLine])
            case "DOWN": # Перемещение курсора на 1 пизицию вниз
                if self.currentLine < len(self.displayBuffer) - 1:
                    self.currentLine += 1
                    if self.currentCol > len(self.displayBuffer[self.currentLine]):
                        self.currentCol = len(self.displayBuffer[self.currentLine])
            case "w": # Перемещение курсора в конец слова справа от курсора
                line = ''.join(self.displayBuffer[self.currentLine])
                # Находим ближайший непробельный символ
                non_space_index = next((i for i, char in enumerate(line[self.currentCol:]) if char != ' '), None)
                if non_space_index is not None:
                    non_space_index += self.currentCol
                    right_space_index = line.find(' ', non_space_index)
                    if right_space_index != -1:
                        self.currentCol = right_space_index
                    else:
                        self.currentCol = len(self.displayBuffer[self.currentLine])
            case "b": # Перемещает курсор в конец в начало слова слева от курсора
                line = ''.join(self.displayBuffer[self.currentLine])
                non_space_index = next((i for i in range(self.currentCol - 1, -1, -1) if line[i] != ' '), None)
                if non_space_index is not None:
                    left_space_index = line.rfind(' ', 0, non_space_index)
                    if left_space_index != -1:
                        self.currentCol = left_space_index + 1
                    else:
                        self.currentCol = 0
            case "gg": # Переход в начало файла
                self.currentLine = 0
                self.currentCol = 0
            case "G": # Переход в конец файла
                self.currentLine = len(self.displayBuffer) - 1
                self.currentCol = len(self.displayBuffer[self.currentLine])
            case "x": # Удаляет символ после курсора
                if self.currentCol + 1 < len(self.displayBuffer[self.currentLine]):
                    del self.displayBuffer[self.currentLine][self.currentCol + 1]
            case "yy": # копирует текущую строку
                self.exchangeBuffer = self.displayBuffer[self.currentLine].copy()
            case "yw":  # Копирует слово под курсором
                line = ''.join(self.displayBuffer[self.currentLine])
                start_index = line.rfind(' ', 0, self.currentCol)
                start_index = 0 if start_index == -1 else start_index + 1
                end_index = line.find(' ', self.currentCol)
                end_index = len(line) if end_index == -1 else end_index
                self.exchangeBuffer = self.displayBuffer[self.currentLine][start_index:end_index].copy()
            case "dd": # вырезает текущую строку
                self.exchangeBuffer = self.displayBuffer[self.currentLine].copy()
                del self.displayBuffer[self.currentLine]
            case "p": # вставить после курсора
                self.displayBuffer[self.currentLine][self.currentCol+1:self.currentCol+1] = self.exchangeBuffer

        return ReturnCode.GOOD

    def Enter(self) -> None:
        # Разделяем текущую строку на две части
        new_line = self.displayBuffer[self.currentLine][self.currentCol:]
        self.displayBuffer[self.currentLine] = self.displayBuffer[self.currentLine][:self.currentCol]
        self.currentLine += 1  # Переходим на следующую строку
        self.displayBuffer.insert(self.currentLine, new_line)  # Вставляем новую строку
        self.currentCol = 0  # Сбрасываем индекс колонки

    def BackspaceCommand(self) -> None:
        if len(self.commandBuffer) > 0: 
            self.commandBuffer.pop()

    def Backspace(self) -> None:
        if self.currentCol > 0:  # Если символ существует в текущей строке
            self.currentCol -= 1
            del self.displayBuffer[self.currentLine][self.currentCol]  # Удаляем символ
        elif self.currentLine > 0:  # Если текущая строка не первая
            # Объединяем текущую строку с предыдущей
            prev_line_length = len(self.displayBuffer[self.currentLine - 1])
            self.displayBuffer[self.currentLine - 1].extend(self.displayBuffer[self.currentLine])
            del self.displayBuffer[self.currentLine]
            self.currentLine -= 1
            self.currentCol = prev_line_length  # Переходим в конец предыдущей строки

    def MoveLeft(self) -> None:
        if self.currentCol > 0:
            self.currentCol -= 1
        elif self.currentLine > 0:
            self.currentLine -= 1
            self.currentCol = len(self.displayBuffer[self.currentLine])

    def MoveRight(self) -> None:
        if self.currentCol < len(self.displayBuffer[self.currentLine]):
            self.currentCol += 1
        elif self.currentLine < len(self.displayBuffer) - 1:
            self.currentLine += 1
            self.currentCol = 0

    def MoveUp(self) -> None:
        if self.currentLine > 0:
            self.currentLine -= 1
            self.currentCol = min(self.currentCol, len(self.displayBuffer[self.currentLine]))

    def MoveDown(self) -> None:
        if self.currentLine < len(self.displayBuffer) - 1:
            self.currentLine += 1
            self.currentCol = min(self.currentCol, len(self.displayBuffer[self.currentLine]))

    def LoadFile(self, file_path) -> None:
        """Загрузка файла для редактирования"""
        self.file_path = file_path
        try:
            with open(file_path, "r") as file:
                self.displayBuffer = [list(line.rstrip('\n')) for line in file.readlines()]
        except FileNotFoundError:
            print(f"File {file_path} not found. Starting with empty buffer.")
            self.displayBuffer = []

    def SaveFile(self) -> None:
        """Сохранение файла"""
        try:
            with open(self.file_path, "w") as file:
                for line in self.displayBuffer:
                    file.write(''.join(line) + '\n')
            print(f"File {self.file_path} saved successfully.")
        except Exception as e:
            print(f"Error saving file: {str(e)}")