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, SelectDirectory from pathlib import Path from home_page import HomePage 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" def __init__(self, path: str): super().__init__() self.config_handler = ConfigHandler(self) self.bindings = eval(self.config_handler.get("editor", "bindings")) self.path = path self.BINDINGS = ( Binding(self.bindings["new"], "new", "New File"), Binding(self.bindings["open"], "open", "Open File"), Binding(self.bindings["open-folder"], "open_folder", "Open Folder"), Binding(self.bindings["save"], "save", "Save File"), Binding(self.bindings["save-as"], "save_as", "Save File As..."), Binding(self.bindings["find"], "find", "Find and Replace", priority=True), # this is priority because otherwise TextArea overrides it Binding(self.bindings["settings"], "settings", "Settings"), ) for binding in self.BINDINGS: self._bindings._add_binding(binding) self.refresh_bindings() self.current_editor = None self.file_tabs = {} 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("Home") first_tab.file_path = "berry://home" self.file_tabs["berry://home"] = first_tab yield Tabs( first_tab, id="file-tabs" ) with Vertical() as editor: yield HomePage() self.current_editor = editor #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) async def choose_folder_to_open(self, result): if result == None: return self.path = result tabs: Tabs = self.query_one("#file-tabs") tabs.clear() first_tab = Tab("Home") first_tab.file_path = "berry://home" self.file_tabs["berry://home"] = first_tab tabs.add_tab(first_tab) self.current_editor.remove_children() self.current_editor.mount(HomePage()) file_tree: CustomDirectoryTree = self.query_one("#directory") file_tree.path = result def action_open_folder(self): self.push_screen(SelectDirectory(), self.choose_folder_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 def open_text_editor(self, file_content: bytes, file_extension: str): code_editor = TextArea.code_editor(placeholder="This file is empty.", theme="css", disabled=True, soft_wrap=bool(int(self.config_handler.get("editor", "word_wrap"))), show_line_numbers=bool(int(self.config_handler.get("editor", "line_numbers")))) self.current_editor.mount(code_editor) code_editor.language = theme_mappings.get(file_extension, None) try: code_editor.text = file_content.decode("utf-8") 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 code_editor.focus() def open_editor(self, file_extension: str, file_content: bytes): default_editors: dict = eval(self.config_handler.get("editor", "default_editors")) editor_for_file = default_editors.get(file_extension, None) self.current_editor.remove_children() if not editor_for_file: self.open_text_editor(file_content, file_extension) @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") if self.open_file and self.open_file.startswith(r"berry://"): return self.switching = True content = b"" file_extension = "" if self.open_file: try: f = open(self.open_file, "rb") except Exception as e: self.notify(f"Failed to open the file: {e}") return # if the file is unsaved, use the content we wrote into it # otherwise, read the CURRENT state of the file if self.open_file in self.unsaved_files: content = self.unsaved_files[self.open_file]["current"] else: content = f.read() # get the file extension file_extension = self.open_file.split(".")[-1] if "." in self.open_file else "" # close the file f.close() self.open_editor(file_extension, content) @on(Window.Minimized) def window_minimized(self, event: Window.Minimized): event.window.remove_window() def on_text_area_changed(self, event: TextArea.Changed): 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 # open empty editor self.open_editor("", b"") 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: 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): 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()