# ba_meta require api 8 # Made by Vishal / Vishuuu / Vishal338 # Made for everyone who uses it. # Let's you edit Plugins which are in your device. # Doesn't work for workspaces. # Safefy Point: Don't mess with the code # or you might lose all your plugins. from __future__ import annotations import os from typing import TYPE_CHECKING, assert_never import babase as ba import bascenev1 as bs import bauiv1 as bui from bauiv1lib import popup from bauiv1lib.confirm import ConfirmWindow from bauiv1lib.settings.plugins import PluginWindow, Category import bauiv1lib.settings.plugins as plugs if TYPE_CHECKING: pass class EditModWindow(bui.Window): """Window for viewing/editing plugins.""" def __init__( self, transition: str = 'in_right', origin_widget: bui.Widget | None = None, plugin: str | None = None, view_type: str | None = None, ): # pylint: disable=too-many-statements app = bui.app # If they provided an origin-widget, scale up from that. scale_origin: tuple[float, float] | None if origin_widget is not None: self._transition_out = 'out_scale' scale_origin = origin_widget.get_screen_space_center() transition = 'in_scale' else: self._transition_out = 'out_right' scale_origin = None assert bui.app.classic is not None uiscale = bui.app.ui_v1.uiscale self._width = 870.0 if uiscale is bui.UIScale.SMALL else 670.0 self._x_inset = 100 if uiscale is bui.UIScale.SMALL else 0 self._height = ( 390.0 if uiscale is bui.UIScale.SMALL else 450.0 if uiscale is bui.UIScale.MEDIUM else 520.0 ) top_extra = 10 if uiscale is bui.UIScale.SMALL else 0 self._theme = ba.app.config.get('Plugin Editor Theme', 'default') super().__init__( root_widget=bui.containerwidget( size=(self._width, self._height + top_extra), transition=transition, toolbar_visibility='menu_minimal', scale_origin_stack_offset=scale_origin, color=(0.13, 0.13, 0.13) if self._theme == 'dark' else None, scale=( 2.06 if uiscale is bui.UIScale.SMALL else 1.4 if uiscale is bui.UIScale.MEDIUM else 1.0 ), stack_offset=( (0, -25) if uiscale is bui.UIScale.SMALL else (0, 0) ), ) ) self._scroll_width = self._width - (100 + 2 * self._x_inset) self._scroll_height = self._height - 115.0 self._sub_width = self._scroll_width * 0.95 self._sub_height = 724.0 Plugin_Name = 'plugin_editor' self._plugin = plugin self._view_type = view_type if view_type is not None else 'view' self._lines = [] self._original_lines = [] self._widgets = [] self._selected_line_widget: ba.Widget | None = None self._saving = False self._shown = False self._plug_path = ba.env()['python_directory_user'] self._same_plug = __file__ in (self._plug_path + '\\' + self._plugin + '.py', self._plug_path + '/' + self._plugin + '.py') self._plug_exists = os.path.exists(self._plug_path + os.sep + self._plugin + '.py') if self._theme == 'dark': self._tint = bs.get_foreground_host_activity().globalsnode.tint bs.get_foreground_host_activity().globalsnode.tint = (0.4, 0.4, 0.4) assert app.classic is not None if app.ui_v1.use_toolbars and uiscale is bui.UIScale.SMALL: bui.containerwidget( edit=self._root_widget, on_cancel_call=self._do_back ) self._back_button = None else: self._back_button = bui.buttonwidget( parent=self._root_widget, position=(53 + self._x_inset, self._height - 60), size=(140, 60), scale=0.8, autoselect=True, label=bui.Lstr(resource='backText'), button_type='back', on_select_call=bui.Call(self._update_line_num), on_activate_call=bui.Call(self._back_button_clicked) ) bui.containerwidget( edit=self._root_widget, cancel_button=self._back_button ) if not self._same_plug and self._plug_exists: self._line_num = bui.textwidget( parent=self._root_widget, position=( self._width * (0.25 if bui.app.ui_v1.uiscale is bui.UIScale.SMALL else 0.175), self._height - 60), size=(120, 30), text='1', color=(0.75, 0.7, 0.8), h_align='center', v_align='center', selectable=True, always_highlight=True, editable=True, ) self._title_text = bui.textwidget( parent=self._root_widget, position=(self._width * 0.5, self._height - 41), size=(0, 0), text=bui.Lstr(value=(self._plugin + '.py')), color=app.ui_v1.title_color, maxwidth=170, h_align='center', v_align='center', ) if self._same_plug: bui.textwidget( parent=self._root_widget, position=(self._width * 0.5, self._height * 0.6), size=(0, 0), text=bui.Lstr(value=f'You can\'t {self._view_type} the Editor, you PSYCHO!'), color=(0.7, 0.8, 0.7), h_align='center', v_align='center', ) if Plugin_Name != self._plugin: bui.textwidget( parent=self._root_widget, position=(self._width * 0.5, self._height * 0.5), size=(0, 0), text=bui.Lstr(value='You tryna change the file name? HAHAHA!'), color=(0.7, 0.8, 0.7), h_align='center', v_align='center', ) if self._back_button is not None: bui.buttonwidget( edit=self._back_button, button_type='backSmall', size=(60, 60), label=bui.charstr(bui.SpecialChar.BACK), ) if not self._same_plug: if self._plug_exists: plug_code = open(self._plug_path + os.sep + self._plugin + '.py', 'r') self._plug_lines = plug_code.readlines() plug_code.close() else: bui.textwidget( parent=self._root_widget, position=(self._width * 0.5, self._height * 0.6), size=(0, 0), text=bui.Lstr(value='Either this is a Workspace Plugin or I can\'t find this.'), color=(0.7, 0.8, 0.7), h_align='center', v_align='center', ) return self._shown = True if self._view_type == 'edit': self._add_line_button = bui.buttonwidget( parent=self._root_widget, position=(self._width * 0.5 + 100, self._height - 60), button_type='square', size=(30, 30), label='+', on_activate_call=bui.Call(self._add_line), ) self._remove_line_button = bui.buttonwidget( parent=self._root_widget, position=(self._width * 0.5 + 145, self._height - 60), button_type='square', size=(30, 30), label='-', color=(1, 0, 0), on_activate_call=bui.Call(self._remove_line), ) self._save_button = bui.buttonwidget( parent=self._root_widget, position=(self._width * 0.5 + 190, self._height - 60), size=(80, 30), label='Save', on_activate_call=lambda: ConfirmWindow( action=self._save_button_clicked ), ) bui.widget( edit=self._save_button, right_widget=self._save_button, up_widget=self._save_button ) self._scrollwidget = bui.scrollwidget( parent=self._root_widget, position=(50 + self._x_inset, 50), simple_culling_v=20.0, highlight=False, size=(self._scroll_width, self._scroll_height), selection_loops_to_parent=True, claims_left_right=True, ) bui.widget(edit=self._scrollwidget, right_widget=self._scrollwidget) self._longest_len = 0 for line in self._plug_lines: line = line[0:len(line)-(1 if line[len(line)-1] == '\n' else 0)] self._lines.append(line) if self._longest_len < len(line): self._longest_len = len(line) self._original_lines = self._lines.copy() plug_line_height = 18 sub_width = self._scroll_width sub_height = (len(self._lines) + 1) * plug_line_height self._subcontainer = bui.containerwidget( parent=self._scrollwidget, size=(sub_width, sub_height), background=False, ) bui.widget( edit=self._back_button, right_widget=self._line_num, down_widget=( self._subcontainer if self._selected_line_widget is None else self._selected_line_widget ) ) bui.widget( edit=self._line_num, up_widget=self._line_num, down_widget=( self._subcontainer if self._selected_line_widget is None else self._selected_line_widget ) ) if self._view_type == 'edit': bui.widget( edit=self._add_line_button, up_widget=self._add_line_button, down_widget=( self._subcontainer if self._selected_line_widget is None else self._selected_line_widget ) ) bui.widget( edit=self._remove_line_button, up_widget=self._remove_line_button, down_widget=( self._subcontainer if self._selected_line_widget is None else self._selected_line_widget ) ) self._show_lines() self._timer = bui.AppTimer(0.1, self._update_line_num, repeat=True) def _show_lines(self) -> None: sub_height = (len(self._lines) + 1) * 18 bui.containerwidget(edit=self._subcontainer, size=(self._scroll_width, sub_height)) self._widgets = [] y_num = 18 * (len(self._lines) - 1) + 10 for num, line in enumerate(self._lines): bui.textwidget( parent=self._subcontainer, position=(23, y_num + 9), size=(0, 0), scale=0.5, text=str(num+1), h_align='right', v_align='center', ) abc = bui.textwidget( parent=self._subcontainer, position=(-151, y_num), size=(880, 18), color=(1.2, 1.2, 1.2), scale=0.6, text=line, selectable=True, maxwidth=527, h_align='left', v_align='center', always_highlight=True, ) bui.widget( edit=abc, left_widget=self._back_button, right_widget=self._add_line_button if self._view_type == 'edit' else abc, ) def edit_them(obj): if self._view_type == 'edit': self._un_editable_all() try: bui.textwidget(edit=obj, editable=True) except: pass self._save_code() self._selected_line_widget = obj try: bui.textwidget(edit=self._line_num, text=str(self._widgets.index(obj)+1) ) except: pass bui.textwidget(edit=abc, on_select_call=bui.Call(edit_them, abc), on_activate_call=bui.Call(edit_them, abc)) y_num -= 18 self._widgets.append(abc) def _un_editable_all(self) -> None: for line in self._widgets: bui.textwidget(edit=line, editable=False) def _remove_line(self) -> None: if self._selected_line_widget is None: bui.screenmessage('Line is not selected.') return index = self._widgets.index(self._selected_line_widget) self._lines.pop(index) if len(self._lines) == 0: self._lines.append('') self._clear_scroll_widget() self._show_lines() bui.containerwidget( edit=self._subcontainer, selected_child=self._widgets[index-1 if index > 0 else 0] ) self._selected_line_widget = self._widgets[index-1 if index > 0 else 0] def _add_line(self) -> None: if self._selected_line_widget is None: bui.screenmessage('Line is not selected.') return index = self._widgets.index(self._selected_line_widget) spaces = '' for ch in self._lines[index]: if ch != ' ': break spaces += ch if self._lines[index][-1:] == ':': spaces += ' ' self._lines.insert(index+1, spaces) self._clear_scroll_widget() self._show_lines() bui.containerwidget( edit=self._subcontainer, selected_child=self._widgets[index+1] ) self._selected_line_widget = self._widgets[index+1] def _save_code(self) -> None: for num, widget in enumerate(self._widgets): self._lines[num] = bui.textwidget(parent=self._subcontainer, query=widget) def _save_button_clicked(self) -> None: if self._saving: return self._saving = True self._save_code() file = open(self._plug_path + os.sep + self._plugin + '.py', 'w') file.writelines([line + '\n' for line in self._lines]) file.close() bui.screenmessage(self._plugin + '.py Saved') self._saving = False self._do_back() def _update_line_num(self) -> None: try: index = int(bui.textwidget(query=self._line_num)) if self._widgets[index-1] != self._selected_line_widget: bui.containerwidget( edit=self._subcontainer, selected_child=self._widgets[index-1], visible_child=self._widgets[index-1] ) self._selected_line_widget = self._widgets[index-1] except (IndexError, ValueError, AttributeError): pass def _clear_scroll_widget(self) -> None: existing_widgets = self._subcontainer.get_children() if existing_widgets: for i in existing_widgets: i.delete() def _back_button_clicked(self) -> None: if self._view_type == 'edit' and not self._same_plug and self._shown: ConfirmWindow(action=self._do_back) else: self._do_back() def _do_back(self) -> None: # pylint: disable=cyclic-import from bauiv1lib.settings.plugins import PluginWindow # no-op if our underlying widget is dead or on its way out. if not self._root_widget or self._root_widget.transitioning_out: return self._timer = None if self._theme == 'dark': bs.get_foreground_host_activity().globalsnode.tint = self._tint bui.containerwidget( edit=self._root_widget, transition=self._transition_out ) assert bui.app.classic is not None bui.app.ui_v1.set_main_menu_window( PluginWindow(transition='in_left').get_root_widget(), from_window=self._root_widget, ) class Plugin_Window(PluginWindow): def __init__( self, transition: str = 'in_right', origin_widget: bui.Widget | None = None, ): self._info = [] self._popup_type = None PluginWindow.__init__(self, transition, origin_widget) def _show_category_options(self) -> None: self._popup_type = 'category' PluginWindow._show_category_options(self) def _show_plugins(self) -> None: # pylint: disable=too-many-locals # pylint: disable=too-many-branches # pylint: disable=too-many-statements plugspecs = bui.app.plugins.plugin_specs plugstates: dict[str, dict] = bui.app.config.setdefault('Plugins', {}) assert isinstance(plugstates, dict) plug_line_height = 50 sub_width = self._scroll_width num_enabled = 0 num_disabled = 0 plugspecs_sorted = sorted(plugspecs.items()) bui.textwidget( edit=self._no_plugins_installed_text, text='', ) for _classpath, plugspec in plugspecs_sorted: # counting number of enabled and disabled plugins # plugstate = plugstates.setdefault(plugspec[0], {}) if plugspec.enabled: num_enabled += 1 else: num_disabled += 1 if self._category is Category.ALL: sub_height = len(plugspecs) * plug_line_height bui.containerwidget( edit=self._subcontainer, size=(self._scroll_width, sub_height) ) elif self._category is Category.ENABLED: sub_height = num_enabled * plug_line_height bui.containerwidget( edit=self._subcontainer, size=(self._scroll_width, sub_height) ) elif self._category is Category.DISABLED: sub_height = num_disabled * plug_line_height bui.containerwidget( edit=self._subcontainer, size=(self._scroll_width, sub_height) ) else: # Make sure we handle all cases. assert_never(self._category) num_shown = 0 for classpath, plugspec in plugspecs_sorted: plugin = plugspec.plugin enabled = plugspec.enabled if self._category is Category.ALL: show = True elif self._category is Category.ENABLED: show = enabled elif self._category is Category.DISABLED: show = not enabled else: assert_never(self._category) if not show: continue item_y = sub_height - (num_shown + 1) * plug_line_height check = bui.checkboxwidget( parent=self._subcontainer, text=bui.Lstr(value=classpath), autoselect=True, value=enabled, maxwidth=self._scroll_width - 200, position=(10, item_y), size=(self._scroll_width - 40, 50), on_value_change_call=bui.Call( self._check_value_changed, plugspec ), textcolor=( (0.8, 0.3, 0.3) if (plugspec.attempted_load and plugspec.plugin is None) else ( (0.6, 0.6, 0.6) if plugspec.plugin is None else (0, 1, 0) ) ), ) # noinspection PyUnresolvedReferences button = bui.buttonwidget( parent=self._subcontainer, label=bui.Lstr(resource='mainMenu.settingsText'), autoselect=True, size=(100, 40), position=(sub_width - 130, item_y + 6), ) # noinspection PyUnresolvedReferences bui.buttonwidget( edit=button, on_activate_call=bui.Call( self._show_menu, button, (plugin.show_settings_ui if plugin is not None and plugin.has_settings_ui() else None), plugin, classpath ), ) # Allow getting back to back button. if num_shown == 0: bui.widget( edit=check, up_widget=self._back_button, left_widget=self._back_button, right_widget=button, ) bui.widget(edit=button, up_widget=self._back_button) # Make sure we scroll all the way to the end when using # keyboard/button nav. bui.widget(edit=check, show_buffer_top=40, show_buffer_bottom=40) num_shown += 1 bui.textwidget( edit=self._num_plugins_text, text=str(num_shown), ) if num_shown == 0: bui.textwidget( edit=self._no_plugins_installed_text, text=bui.Lstr(resource='noPluginsInstalledText'), ) def _show_menu(self, button, func, plugin, classpath) -> None: choices = ['view', 'edit', 'settings' ] if plugin is not None and plugin.has_settings_ui() else ['view', 'edit'] choices_display = [ bui.Lstr(value='View Code'), bui.Lstr(value='Edit Code'), bui.Lstr(value='Settings'), ] if plugin is not None and plugin.has_settings_ui() else [ bui.Lstr(value='View Code'), bui.Lstr(value='Edit Code') ] popup.PopupMenuWindow( position=button.get_screen_space_center(), choices=choices, choices_display=choices_display, current_choice='settings', width=100, delegate=self, scale=( 2.3 if bui.app.ui_v1.uiscale is bui.UIScale.SMALL else 1.65 if bui.app.ui_v1.uiscale is bui.UIScale.MEDIUM else 1.23 ), ) self._info = [button, func, classpath] self._popup_type = 'options' def popup_menu_selected_choice( self, popup_window: popup.PopupMenuWindow, choice: str ) -> None: """Called when a choice is selected in the popup.""" del popup_window # unused if self._popup_type == 'category': self._category = Category(choice) self._clear_scroll_widget() self._show_plugins() bui.buttonwidget( edit=self._category_button, label=bui.Lstr(resource=self._category.resource), ) elif self._popup_type == 'options': self._abc(self._info[0], self._info[1], self._info[2], choice) def _abc(self, button, func, plugin, view_mode: str) -> None: # pylint: disable=cyclic-import if view_mode == 'view' or view_mode == 'edit': # no-op if our underlying widget is dead or on its way out. if not self._root_widget or self._root_widget.transitioning_out: return self._save_state() bui.containerwidget(edit=self._root_widget, transition='out_left') assert bui.app.classic is not None bui.app.ui_v1.set_main_menu_window( EditModWindow( transition='in_right', plugin=plugin.split('.')[0], view_type=view_mode ).get_root_widget(), from_window=self._root_widget, ) elif view_mode == 'settings': func(button) class PluginEditorSettingsWindow(popup.PopupWindow): def __init__(self, origin_widget): self.scale_origin = origin_widget.get_screen_space_center() bui.getsound('swish').play() _uiscale = bui.app.ui_v1.uiscale s = 1.65 if _uiscale is ba.UIScale.SMALL else 1.39 if _uiscale is ba.UIScale.MEDIUM else 1.67 width = 400 * s height = width * 0.5 color = (1, 1, 1) text_scale = 0.7 * s self._transition_out = 'out_scale' transition = 'in_scale' self._root_widget = bui.containerwidget(size=(width, height), on_outside_click_call=self._back, transition=transition, scale=(1.5 if _uiscale is ba.UIScale.SMALL else 1.5 if _uiscale is ba.UIScale.MEDIUM else 1.0), scale_origin_stack_offset=self.scale_origin) bui.textwidget(parent=self._root_widget, position=(width * 0.49, height * 0.87), size=(0, 0), h_align='center', v_align='center', text='Plugin Editor Settings', scale=text_scale * 1.25, color=bui.app.ui_v1.title_color, maxwidth=width * 0.9) back_button = bui.buttonwidget( parent=self._root_widget, position=(width * 0.1, height * 0.8), size=(60, 60), scale=0.8, label=ba.charstr(ba.SpecialChar.BACK), button_type='backSmall', on_activate_call=self._back) bui.containerwidget(edit=self._root_widget, cancel_button=back_button) bui.textwidget(parent=self._root_widget, position=(width * 0.32, height * 0.5), size=(0, 0), h_align='center', v_align='center', text='Theme :', scale=text_scale * 1.15, color=bui.app.ui_v1.title_color ) popup.PopupMenu( parent=self._root_widget, position=(width * 0.42, height * 0.5 - 30), button_size=(200.0, 60.0), width=100.0, choices=[ 'default', 'dark', ], choices_display=[ bui.Lstr(value='Default'), bui.Lstr(value='Dark'), ], current_choice=ba.app.config.get('Plugin Editor Theme', 'default'), on_value_change_call=self._set_theme, ) def _set_theme(self, val: str) -> None: if ba.app.config.get('Plugin Editor Theme', 'default') != val: cfg = bui.app.config cfg['Plugin Editor Theme'] = val cfg.apply_and_commit() bui.screenmessage( bui.Lstr(value=f'Editor Theme changed to {val.title()}'), color=(0.0, 1.0, 0.0), ) def _back(self) -> None: bui.getsound('swish').play() bui.containerwidget(edit=self._root_widget, transition='out_scale') # ba_meta export plugin class ByVishuuu(ba.Plugin): """A plugin to edit plugins.""" def __init__(self) -> None: plugs.PluginWindow = Plugin_Window def has_settings_ui(self): return True def show_settings_ui(self, source_widget): PluginEditorSettingsWindow(source_widget)