250 lines
10 KiB
Python
250 lines
10 KiB
Python
# Python imports
|
|
import sys, os, signal
|
|
from dataclasses import dataclass
|
|
|
|
# Lib imports
|
|
import gi
|
|
gi.require_version('Gtk', '3.0')
|
|
from gi.repository import Gtk
|
|
from gi.repository import GLib
|
|
|
|
# Application imports
|
|
from shellfm.windows.controller import WindowController
|
|
from plugins.plugins_controller import PluginsController
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class State:
|
|
wid: int = None
|
|
tid: int = None
|
|
tab: type = None
|
|
icon_grid: gi.overrides.Gtk.IconView = None
|
|
store: gi.overrides.Gtk.ListStore = None
|
|
selected_files: [] = None
|
|
to_copy_files: [] = None
|
|
to_cut_files: [] = None
|
|
warning_alert: type = None
|
|
|
|
|
|
class Controller_Data:
|
|
""" Controller_Data contains most of the state of the app at ay given time. It also has some support methods. """
|
|
__slots__ = "settings", "builder", "logger", "keybindings", "trashman", "fm_controller", "window", "window1", "window2", "window3", "window4"
|
|
|
|
def setup_controller_data(self) -> None:
|
|
self.builder = settings.get_builder()
|
|
self.keybindings = settings.get_keybindings()
|
|
|
|
self.fm_controller = WindowController()
|
|
self.plugins = PluginsController()
|
|
self.fm_controller_data = self.fm_controller.get_state_from_file()
|
|
|
|
self.window = settings.get_main_window()
|
|
self.window1 = self.builder.get_object("window_1")
|
|
self.window2 = self.builder.get_object("window_2")
|
|
self.window3 = self.builder.get_object("window_3")
|
|
self.window4 = self.builder.get_object("window_4")
|
|
self.message_popup_widget = self.builder.get_object("message_popup_widget")
|
|
self.message_text_view = self.builder.get_object("message_text_view")
|
|
self.message_buffer = self.builder.get_object("message_buffer")
|
|
self.arc_command_buffer = self.builder.get_object("arc_command_buffer")
|
|
|
|
self.exists_file_rename_bttn = self.builder.get_object("exists_file_rename_bttn")
|
|
self.warning_alert = self.builder.get_object("warning_alert")
|
|
self.new_file_menu = self.builder.get_object("new_file_menu")
|
|
self.edit_file_menu = self.builder.get_object("edit_file_menu")
|
|
self.file_exists_dialog = self.builder.get_object("file_exists_dialog")
|
|
self.exists_file_label = self.builder.get_object("exists_file_label")
|
|
self.exists_file_field = self.builder.get_object("exists_file_field")
|
|
self.path_menu = self.builder.get_object("path_menu")
|
|
self.path_entry = self.builder.get_object("path_entry")
|
|
|
|
self.bottom_size_label = self.builder.get_object("bottom_size_label")
|
|
self.bottom_file_count_label = self.builder.get_object("bottom_file_count_label")
|
|
self.bottom_path_label = self.builder.get_object("bottom_path_label")
|
|
|
|
self.trash_files_path = f"{GLib.get_user_data_dir()}/Trash/files"
|
|
self.trash_info_path = f"{GLib.get_user_data_dir()}/Trash/info"
|
|
self.icon_theme = settings.get_icon_theme()
|
|
|
|
self.notebooks = [self.window1, self.window2, self.window3, self.window4]
|
|
self.selected_files = []
|
|
self.to_copy_files = []
|
|
self.to_cut_files = []
|
|
self.soft_update_lock = {}
|
|
|
|
self.single_click_open = False
|
|
self.is_pane1_hidden = False
|
|
self.is_pane2_hidden = False
|
|
self.is_pane3_hidden = False
|
|
self.is_pane4_hidden = False
|
|
|
|
self.override_drop_dest = None
|
|
self.is_searching = False
|
|
self.search_icon_grid = None
|
|
self.search_tab = None
|
|
|
|
self.cancel_creation = False
|
|
self.skip_edit = False
|
|
self.cancel_edit = False
|
|
self.ctrl_down = False
|
|
self.shift_down = False
|
|
self.alt_down = False
|
|
|
|
self.success_color = settings.get_success_color()
|
|
self.warning_color = settings.get_warning_color()
|
|
self.error_color = settings.get_error_color()
|
|
|
|
# sys.excepthook = self.custom_except_hook
|
|
self.window.connect("delete-event", self.tear_down)
|
|
GLib.unix_signal_add(GLib.PRIORITY_DEFAULT, signal.SIGINT, self.tear_down)
|
|
|
|
self.window.show()
|
|
if settings.is_debug():
|
|
self.window.set_interactive_debugging(True)
|
|
|
|
self.build_context_menu()
|
|
|
|
|
|
def build_context_menu(self) -> None:
|
|
main_menu = Gtk.Menu()
|
|
open_menu = Gtk.Menu()
|
|
filea_menu = Gtk.Menu()
|
|
plugins_menu = Gtk.Menu()
|
|
open_a = Gtk.MenuItem("Open Actions")
|
|
file_a = Gtk.MenuItem("File Actions")
|
|
plugins = Gtk.MenuItem("Plugins")
|
|
|
|
self._appen_menu_items(main_menu, [open_a, file_a, plugins])
|
|
|
|
open = Gtk.ImageMenuItem("Open")
|
|
open.set_name("open")
|
|
open.set_image( Gtk.Image(stock=Gtk.STOCK_OPEN) )
|
|
open.set_always_show_image(True)
|
|
open.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
open_with = Gtk.ImageMenuItem("Open With")
|
|
open_with.set_name("open_with")
|
|
open_with.set_image( Gtk.Image(stock=Gtk.STOCK_OPEN) )
|
|
open_with.set_always_show_image(True)
|
|
open_with.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
execute = Gtk.ImageMenuItem("Execute")
|
|
execute.set_name("execute")
|
|
execute.set_image( Gtk.Image(stock=Gtk.STOCK_EXECUTE) )
|
|
execute.set_always_show_image(True)
|
|
execute.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
execute_term = Gtk.ImageMenuItem("Execute in Terminal")
|
|
execute_term.set_name("execute_in_terminal")
|
|
execute_term.set_image( Gtk.Image(stock=Gtk.STOCK_EXECUTE) )
|
|
execute_term.set_always_show_image(True)
|
|
execute_term.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
self._appen_menu_items(open_menu, [open, open_with, execute, execute_term])
|
|
|
|
new = Gtk.ImageMenuItem("New")
|
|
new.set_name("create")
|
|
new.set_image( Gtk.Image(stock=Gtk.STOCK_ADD) )
|
|
new.set_always_show_image(True)
|
|
new.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
rename = Gtk.ImageMenuItem("Rename")
|
|
rename.set_name("rename")
|
|
rename.set_image( Gtk.Image(stock=Gtk.STOCK_EDIT) )
|
|
rename.set_always_show_image(True)
|
|
rename.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
cut = Gtk.ImageMenuItem("Cut")
|
|
cut.set_name("cut")
|
|
cut.set_image( Gtk.Image(stock=Gtk.STOCK_CUT) )
|
|
cut.set_always_show_image(True)
|
|
cut.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
copy = Gtk.ImageMenuItem("Copy")
|
|
copy.set_name("copy")
|
|
copy.set_image( Gtk.Image(stock=Gtk.STOCK_COPY) )
|
|
copy.set_always_show_image(True)
|
|
copy.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
paste = Gtk.ImageMenuItem("Paste")
|
|
paste.set_name("paste")
|
|
paste.set_image( Gtk.Image(stock=Gtk.STOCK_PASTE) )
|
|
paste.set_always_show_image(True)
|
|
paste.connect("activate", self.do_action_from_menu_controls)
|
|
|
|
self._appen_menu_items(filea_menu, [new, rename, cut, copy, paste])
|
|
open_a.set_submenu(open_menu)
|
|
file_a.set_submenu(filea_menu)
|
|
plugins.set_submenu(plugins_menu)
|
|
|
|
main_menu.attach_to_widget(self.window, None)
|
|
main_menu.show_all()
|
|
self.builder.expose_object("context_menu", main_menu)
|
|
self.builder.expose_object("context_menu_plugins", plugins_menu)
|
|
|
|
def _appen_menu_items(self, menu, items):
|
|
for item in items:
|
|
menu.append(item)
|
|
|
|
def get_current_state(self) -> State:
|
|
'''
|
|
Returns the state info most useful for any given context and action intent.
|
|
|
|
Parameters:
|
|
a (obj): self
|
|
|
|
Returns:
|
|
state (obj): State
|
|
'''
|
|
state = State()
|
|
state.wid, state.tid = self.fm_controller.get_active_wid_and_tid()
|
|
state.tab = self.get_fm_window(state.wid).get_tab_by_id(state.tid)
|
|
state.icon_grid = self.builder.get_object(f"{state.wid}|{state.tid}|icon_grid")
|
|
state.store = state.icon_grid.get_model()
|
|
state.warning_alert = self.warning_alert
|
|
|
|
|
|
selected_files = state.icon_grid.get_selected_items()
|
|
if selected_files:
|
|
state.selected_files = self.format_to_uris(state.store, state.wid, state.tid, selected_files, True)
|
|
|
|
# if self.to_copy_files:
|
|
# state.to_copy_files = self.format_to_uris(state.store, state.wid, state.tid, self.to_copy_files, True)
|
|
#
|
|
# if self.to_cut_files:
|
|
# state.to_cut_files = self.format_to_uris(state.store, state.wid, state.tid, self.to_cut_files, True)
|
|
|
|
event_system.emit("update_state_info_plugins", state)
|
|
return state
|
|
|
|
|
|
def clear_console(self) -> None:
|
|
''' Clears the terminal screen. '''
|
|
os.system('cls' if os.name == 'nt' else 'clear')
|
|
|
|
def call_method(self, _method_name: str, data: type = None) -> type:
|
|
'''
|
|
Calls a method from scope of class.
|
|
|
|
Parameters:
|
|
a (obj): self
|
|
b (str): method name to be called
|
|
c (*): Data (if any) to be passed to the method.
|
|
Note: It must be structured according to the given methods requirements.
|
|
|
|
Returns:
|
|
Return data is that which the calling method gives.
|
|
'''
|
|
method_name = str(_method_name)
|
|
method = getattr(self, method_name, lambda data: f"No valid key passed...\nkey={method_name}\nargs={data}")
|
|
return method(data) if data else method()
|
|
|
|
def has_method(self, obj, name) -> type:
|
|
''' Checks if a given method exists. '''
|
|
return callable(getattr(obj, name, None))
|
|
|
|
def clear_children(self, widget: type) -> None:
|
|
''' Clear children of a gtk widget. '''
|
|
for child in widget.get_children():
|
|
widget.remove(child)
|