import os import subprocess import ijson from fabric.utils import remove_handler from fabric.utils.helpers import get_relative_path from fabric.widgets.box import Box from fabric.widgets.button import Button from fabric.widgets.entry import Entry from fabric.widgets.label import Label from fabric.widgets.stack import Stack from gi.repository import Gdk import config.data as data import modules.icons as icons vertical_mode = data.PANEL_THEME == "Panel" and (data.BAR_POSITION in ["Left", "Right"] or data.PANEL_POSITION in ["Start", "End"]) emoji_rows = 3 if not vertical_mode else 9 emoji_columns = 9 if not vertical_mode else 5 class EmojiPicker(Box): def __init__(self, **kwargs): super().__init__( name="emoji", visible=False, all_visible=False, **kwargs, ) self.notch = kwargs["notch"] self.selected_index = -1 self.emojis_per_page = emoji_columns * emoji_rows self.current_page_index = 0 self.filtered_emojis = [] self.total_pages = 0 self._arranger_handler: int = 0 self._all_emojis = self._load_emoji_data() self.stack = Stack( name="viewport", spacing=4, orientation="v", transition_type="slide-up-down", transition_duration=200, ) self.search_entry = Entry( name="search-entry", placeholder="Search Emojis...", h_expand=True, notify_text=lambda entry, *_: self.arrange_viewport(entry.get_text()), on_activate=lambda entry, *_: self.on_search_entry_activate(entry.get_text()), on_key_press_event=self.on_search_entry_key_press, ) self.search_entry.props.xalign = 0.5 self.header_box = Box( name="header_box", spacing=10, orientation="h", children=[ self.search_entry, Button( name="close-button", child=Label(name="close-label", markup=icons.cancel), tooltip_text="Exit", on_clicked=lambda *_: self.close_picker() ), ], ) self.picker_box = Box( name="picker-box", spacing=10, h_expand=True, orientation="v", children=[ self.header_box, self.stack, ], ) self.resize_viewport() self.add(self.picker_box) self.show_all() def _load_emoji_data(self): emoji_data = {} emoji_file_path = get_relative_path("../assets/emoji.json") if not os.path.exists(emoji_file_path): print(f"Emoji JSON file not found at: {emoji_file_path}") return {} with open(emoji_file_path, 'r') as f: for emoji_char, emoji_info in ijson.kvitems(f, ''): emoji_data[emoji_char] = emoji_info return emoji_data def close_picker(self): self.stack.children = [] self.selected_index = -1 self.notch.close_notch() def open_picker(self): self.search_entry.set_text("") self.current_page_index = 0 self.arrange_viewport() self.search_entry.grab_focus() def arrange_viewport(self, query: str = ""): remove_handler(self._arranger_handler) if self._arranger_handler else None self.stack.children = [] self.selected_index = -1 self.current_page_index = 0 self.filtered_emojis = [ (emoji_char, emoji_info) for emoji_char, emoji_info in self._all_emojis.items() if query.casefold() in (emoji_info.get("name", "") + " " + emoji_info.get("group", "")).casefold() ] self.total_pages = (len(self.filtered_emojis) + self.emojis_per_page - 1) // self.emojis_per_page if self.filtered_emojis else 0 self.load_page(self.current_page_index) should_resize = not query if should_resize: self.resize_viewport() if query.strip() != "" and self.get_all_emoji_buttons(): self.update_selection(0) def load_page(self, page_index): self.update_selection(-1) page_box = Box(name=f"page-box-{page_index}", orientation="v", spacing=4) start_index = page_index * self.emojis_per_page end_index = min((page_index + 1) * self.emojis_per_page, len(self.filtered_emojis)) page_emojis = self.filtered_emojis[start_index:end_index] grid_box = Box(name="emoji-grid-box", orientation="v", spacing=2) row_box = None for i, (emoji_char, emoji_info) in enumerate(page_emojis): if i % emoji_columns == 0: row_box = Box(name="emoji-row-box", orientation="h", spacing=2) grid_box.add(row_box) if row_box is not None: row_box.add(self.bake_emoji_slot(emoji_char, emoji_info)) page_box.add(grid_box) self.stack.add_named(page_box, f"page-{page_index}") self.stack.set_visible_child_name(f"page-{page_index}") page_box.show_all() buttons = self.get_all_emoji_buttons() if buttons and self.selected_index != -1: page_relative_index = self.selected_index % self.emojis_per_page if page_relative_index < len(buttons): self.update_selection(page_relative_index) else: self.update_selection(len(buttons) - 1) def resize_viewport(self): return False def bake_emoji_slot(self, emoji_char: str, emoji_info: dict, **kwargs) -> Button: button = Button( name="emoji-slot-button", child=Box( name="emoji-slot-box", orientation="horizontal", halign="center", valign="center", children=[ Label( name="emoji-char-label", label=emoji_char, use_markup=True, v_align="center", h_align="center", css_name="emoji-char-label" ), ], ), tooltip_text=emoji_info.get("name", "Unknown"), on_clicked=lambda *_: (self.copy_emoji_to_clipboard(emoji_char), self.close_picker()), **kwargs, ) return button def update_selection(self, new_index: int): buttons = self.get_all_emoji_buttons() if not buttons: self.selected_index = -1 return if self.selected_index != -1 and self.selected_index < len(buttons): current_button = buttons[self.selected_index] current_button.get_style_context().remove_class("selected") if not buttons or current_button not in buttons: self.selected_index = -1 if 0 <= new_index < len(buttons): new_button = buttons[new_index] new_button.get_style_context().add_class("selected") self.selected_index = new_index else: self.selected_index = -1 def get_all_emoji_buttons(self): buttons = [] current_page = self.stack.get_visible_child() if current_page and current_page.get_children(): if current_page.get_children()[0].get_children(): for row_box in current_page.get_children()[0].get_children(): buttons.extend(row_box.get_children()) return buttons def on_search_entry_activate(self, text): buttons = self.get_all_emoji_buttons() if buttons: if self.selected_index != -1: buttons[self.selected_index].clicked() elif buttons and text.strip() != "": buttons[0].clicked() def on_search_entry_key_press(self, widget, event): if event.keyval in (Gdk.KEY_Up, Gdk.KEY_Down, Gdk.KEY_Left, Gdk.KEY_Right): self.move_selection_2d(event.keyval) return True elif event.keyval == Gdk.KEY_Escape: self.close_picker() return True return False def move_selection_2d(self, keyval): buttons = self.get_all_emoji_buttons() total_items_current_page = len(buttons) if total_items_current_page == 0: return rows = emoji_rows columns = emoji_columns if self.selected_index == -1: if keyval in (Gdk.KEY_Down, Gdk.KEY_Right): new_index = 0 elif keyval in (Gdk.KEY_Up, Gdk.KEY_Left): new_index = total_items_current_page - 1 else: return else: current_index_page = self.selected_index row = current_index_page // columns col = current_index_page % columns if keyval == Gdk.KEY_Right: new_col = (col + 1) % columns new_row = row if new_col == 0: new_row = (row + 1) elif keyval == Gdk.KEY_Left: new_col = (col - 1) % columns new_row = row if new_col == (columns - 1): new_row = (row - 1) elif keyval == Gdk.KEY_Down: new_row = row + 1 new_col = col elif keyval == Gdk.KEY_Up: new_row = row - 1 new_col = col else: return if new_row >= rows: if self.current_page_index < self.total_pages - 1: current_col = col # Keep track of current column self.current_page_index += 1 self.load_page(self.current_page_index) new_index = current_col # Try to keep the same column if new_index >= total_items_current_page: # if column is out of bound, select last new_index = total_items_current_page - 1 self.selected_index = -1 self.update_selection(new_index) return else: new_index = total_items_current_page - 1 elif new_row < 0: if self.current_page_index > 0: current_col = col # Keep track of current column self.current_page_index -= 1 self.load_page(self.current_page_index) new_index = (rows - 1) * columns + current_col # Select last row, same column if new_index >= total_items_current_page: # if column is out of bound, select last new_index = total_items_current_page -1 self.selected_index = -1 self.update_selection(new_index) return else: new_index = 0 else: new_index = new_row * columns + new_col if new_index >= total_items_current_page: new_index = total_items_current_page - 1 if new_index < 0: new_index = 0 elif new_index >= total_items_current_page: new_index = total_items_current_page - 1 self.update_selection(new_index) def copy_emoji_to_clipboard(self, emoji_char: str): try: subprocess.run(["wl-copy"], input=emoji_char.encode('utf-8'), check=True) except subprocess.CalledProcessError as e: print(f"Clipboard copy failed: {e}")