from textual.app import App, ComposeResult, SystemCommand from textual.widgets import Header, Footer, ContentSwitcher, DirectoryTree, Static, Button, TextArea, Tabs, Tab, RichLog, Input from textual.containers import HorizontalGroup, Vertical from textual.binding import Binding from textual_window import Window from textual import on from textual_fspicker import FileOpen, FileSave from pathlib import Path from assets.theme_mappings import theme_mappings from plugin_loader import PluginLoader from settings import SettingsScreen from settings_store import ConfigHandler from directory_tree_custom import CustomDirectoryTree from watchdog.observers import Observer from watchdog.events import FileSystemEventHandler import os, sys class Watcher(FileSystemEventHandler): def __init__(self, app: App): super().__init__() self.app = app async def on_any_event(self, event): await self.app.query_one(DirectoryTree).reload() class Berry(App): CSS_PATH = "assets/style.tcss" SUB_TITLE = "New File" BINDINGS = [ Binding("ctrl+o", "open", "Open File"), Binding("ctrl+n", "new", "New File"), Binding("ctrl+s", "save", "Save"), Binding("ctrl+shift+s", "save_as", "Save As...", priority=True), Binding("ctrl+f", "find", "Find", priority=True), Binding("ctrl+f1", "settings", "Settings") ] def __init__(self, path: str): super().__init__() self.path = path self.config_handler = ConfigHandler(self) def compose(self) -> ComposeResult: yield Header() with Vertical(id="sidebar"): with HorizontalGroup(id="sidebar-buttons"): yield Button("📂") yield Button("🔍") with ContentSwitcher(initial="files", id="sidebar-switcher"): with Vertical(id="files"): yield Static("EXPLORER") yield CustomDirectoryTree(self.path, id="directory") with Vertical(id="editor"): first_tab = Tab("New File") first_tab.file_path = None yield Tabs( first_tab, id="file-tabs" ) yield TextArea.code_editor(placeholder="This file is empty.", theme="css", id="code-editor", disabled=True, soft_wrap=True, show_line_numbers=bool(int(self.config_handler.get("editor", "line_numbers")))) #if os.name == "nt": #with Vertical(id="console-container"): # yield RichLog(id="console") # yield Input(placeholder="> ", id="console-input") #else: # yield Terminal(command="bash", id="terminal") yield Footer() if bool(int(self.config_handler.get("plugins", "enabled"))) == True: yield PluginLoader() def action_settings(self): self.push_screen(SettingsScreen()) def get_system_commands(self, screen): yield SystemCommand( "Quit the application", "Quit the application as soon as possible", self.action_quit, ) if screen.query("HelpPanel"): yield SystemCommand( "Hide keys and help panel", "Hide the keys and widget help panel", self.action_hide_help_panel, ) else: yield SystemCommand( "Show keys and help panel", "Show help for the focused widget and a summary of available keys", self.action_show_help_panel, ) yield SystemCommand("Settings", "Open the settings menu", self.action_settings) async def chose_file_to_open(self, result): if result == None: return result = str(result) if self.open_file == result: return def is_within_directory(file_path: str, directory: str) -> bool: file_path = Path(file_path).resolve() directory = Path(directory).resolve() return directory in file_path.parents self.switching = True tabs: Tabs = self.query_one("#file-tabs") if self.open_file not in self.unsaved_files: if self.open_file: self.file_tabs.pop(self.open_file) tabs.remove_tab(tabs.active_tab) self.open_file = result inside_dir = is_within_directory(result, self.path) self.sub_title = os.path.basename(self.open_file) if inside_dir else self.open_file if result not in self.file_tabs: new_tab = Tab(os.path.basename(result) if inside_dir else result) new_tab.tooltip = str(new_tab.label) setattr(new_tab, "file_path", result) await tabs.add_tab(new_tab) self.file_tabs[result] = new_tab tabs.active = new_tab.id else: tabs.active = self.file_tabs[result].id def action_open(self): self.app.push_screen(FileOpen(), self.chose_file_to_open) def action_find(self): try: self.query_one("#find-window") return except: find_window = Window( Vertical( HorizontalGroup( Input(placeholder="Find"), Static("0 of 0", id="num-matches"), Button("↑", flat=True), Button("↓", flat=True), ), HorizontalGroup( Input(placeholder="Replace"), ), ), icon="🔍", start_open=True, allow_resize=False, allow_maximize=False, id="find-window", mode="temporary", name="Find & Replace" ) self.mount(find_window) async def on_directory_tree_file_selected(self, event: DirectoryTree.FileSelected): if self.open_file == str(event.path): return self.file_clicked = True self.switching = True tabs: Tabs = self.query_one("#file-tabs") if self.open_file not in self.unsaved_files: if self.open_file: self.file_tabs.pop(self.open_file) tabs.remove_tab(tabs.active_tab) self.open_file = str(event.path) self.sub_title = os.path.basename(self.open_file) if str(event.path) not in self.file_tabs: new_tab = Tab(os.path.basename(str(event.path))) new_tab.tooltip = str(new_tab.label) setattr(new_tab, "file_path", str(event.path)) await tabs.add_tab(new_tab) self.file_tabs[str(event.path)] = new_tab tabs.active = new_tab.id else: tabs.active = self.file_tabs[str(event.path)].id @on(Tabs.TabActivated) def on_tab_shown(self, event: Tabs.TabActivated): if self.file_clicked: self.file_clicked = False else: self.open_file = getattr(event.tab, "file_path") self.switching = True code_editor: TextArea = self.query_one("#code-editor") if self.open_file: try: f = open(self.open_file, "r", encoding="utf-8") except Exception as e: self.notify(f"Failed to open the file: {e}") return try: if self.open_file in self.unsaved_files: code_editor.text = self.unsaved_files[self.open_file]["current"] else: code_editor.text = f.read() file_extension = self.open_file dot_count = file_extension.count(".") if dot_count == 1: if file_extension.startswith("."): file_extension = file_extension.removeprefix(".") else: file_extension = file_extension.rsplit(".", 1)[1] elif dot_count > 1: file_extension = file_extension.rsplit(".", 1)[1] code_editor.language = theme_mappings.get(file_extension, None) code_editor.disabled = False except UnicodeDecodeError: code_editor.text = "This file is in binary, it can't be opened. Sorry." code_editor.language = None code_editor.disabled = True f.close() else: code_editor.text = "" code_editor.language = None code_editor.disabled = False code_editor.focus() @on(Window.Minimized) def window_minimized(self, event: Window.Minimized): event.window.remove_window() def on_text_area_changed(self, event: TextArea.Changed): if event.text_area.id != "code-editor": return if self.switching: self.switching = False return tabs: Tabs = self.query_one("#file-tabs") if self.open_file: if not self.open_file in self.unsaved_files: with open(self.open_file, "r", encoding="utf-8") as f: if f.read() == event.text_area.text: # TODO: figure out why im guetting what seems like a race conidition which is making this if statement needed return self.unsaved_files[self.open_file] = {"current": event.text_area.text, "original": f.read()} tabs.active_tab.tooltip = f"Unsaved changes in {tabs.active_tab.label}" tabs.active_tab.label = "[d orange]●[/] " + str(tabs.active_tab.label) else: self.unsaved_files[self.open_file]["current"] = event.text_area.text if self.unsaved_files[self.open_file]["original"] == self.unsaved_files[self.open_file]["current"]: tabs.active_tab.label = os.path.basename(self.open_file) tabs.active_tab.tooltip = str(tabs.active_tab.label) self.unsaved_files.pop(self.open_file) def action_new(self): tabs: Tabs = self.query_one("#file-tabs") new_tab = Tab("New File") setattr(new_tab, "file_path", None) tabs.add_tab(new_tab) tabs.active = new_tab.id self.open_file = None self.switching = True code_editor: TextArea = self.query_one("#code-editor") code_editor.disabled = False code_editor.text = "" def done_saving(self, result): if result is None: return with open(result, "w", encoding="utf-8") as f: f.write(self.query_one("#code-editor").text) tabs: Tabs = self.query_one("#file-tabs") tabs.active_tab.label = os.path.basename(result) self.notify(f"Saved to {result} successfully.", title="Done!", markup=False) self.query_one(DirectoryTree).reload() def action_save_as(self): self.push_screen(FileSave(), callback=self.done_saving) def action_save(self): if self.open_file == None and self.query_one("#code-editor").disabled == False: self.action_save_as() # dont bother saving if there are no new changes if not self.open_file in self.unsaved_files: return with open(self.open_file, "w") as f: f.write(self.unsaved_files[self.open_file]["current"]) tabs: Tabs = self.query_one("#file-tabs") tabs.active_tab.label = os.path.basename(self.open_file) tabs.active_tab.tooltip = str(tabs.active_tab.label) self.unsaved_files.pop(self.open_file) self.notify("Saved.") def action_quit(self): self.observer.stop() self.observer.join() return super().action_quit() def on_ready(self): # src/main.py: Tab<> self.file_tabs = {} self.open_file = None self.unsaved_files = {} # list of paths self.switching = False self.file_clicked = False self.observer = Observer() self.observer.schedule(Watcher(self), path=self.path) self.observer.start() self.config_handler.apply_settings() if __name__ == "__main__": working_path = os.getcwd() if len(sys.argv) == 1 else sys.argv[1] app = Berry(working_path) app.run()