editor.py

from prompt_toolkit import Application
from prompt_toolkit.buffer import Buffer
from prompt_toolkit.layout.containers import Window, HSplit, VSplit
from prompt_toolkit.layout.controls import BufferControl, FormattedTextControl
from prompt_toolkit.layout.layout import Layout
from prompt_toolkit.key_binding import KeyBindings
from prompt_toolkit.lexers import PygmentsLexer
from prompt_toolkit.styles import Style
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.layout.containers import WindowAlign
from prompt_toolkit.filters import Condition
from prompt_toolkit.widgets import Frame
from prompt_toolkit.document import Document
from pygments.lexers.python import PythonLexer
from prompt_toolkit.lexers import Lexer
from prompt_toolkit.formatted_text import FormattedText
from lexer import SqlLexer
from kb import create_key_bindings
from prompt_toolkit.enums import EditingMode
from layout import EditorLayout


background_color = "#2C3E50"
text_color = "#ECF0F1"
highlight_color = "#96aeff"
line_numbers = "#dbbfbf"
special = "#FF0000 bold"

style = Style.from_dict(
    {
        "window": f"bg:{background_color} {text_color}",
        "line-numbers": f"bg:{background_color} #000000",
        "current-line": f"bg:{highlight_color}",
        "sql-keyword": f"{special}",  # Red and bold for special words
    }
)


class Editor:
    def __init__(self):
        self.buffer = Buffer()
        self.editor_layout = EditorLayout(self)
        self.key_bindings = create_key_bindings(self)
        self.app = self._create_app()

    async def run(self):
        await self.app.run_async()

    def _create_app(self):
        application = Application(
            editing_mode=EditingMode.EMACS,
            layout=self.editor_layout.layout,
            key_bindings=self.key_bindings,
            style=style,
            full_screen=True,
        )
        return application