From be71bf80d290c98c0f6a66d3dcab28515b4ea371 Mon Sep 17 00:00:00 2001
From: Lucas Fryzek <lucas.fryzek@fryzekconcepts.com>
Date: Fri, 4 Oct 2024 20:53:17 +0100
Subject: Initial commit
---
src/__init__.py | 0
src/chat.py | 120 ++++++++++++++
src/config.py | 140 ++++++++++++++++
src/gtk/chat.ui | 67 ++++++++
src/gtk/help-overlay.ui | 29 ++++
src/gtk/message.ui | 63 ++++++++
src/gtk/preferences.ui | 83 ++++++++++
src/main.py | 338 +++++++++++++++++++++++++++++++++++++++
src/meson.build | 42 +++++
src/message.py | 48 ++++++
src/preferences.py | 90 +++++++++++
src/relay/color.py | 198 +++++++++++++++++++++++
src/relay/network.py | 403 +++++++++++++++++++++++++++++++++++++++++++++++
src/relay/protocol.py | 362 ++++++++++++++++++++++++++++++++++++++++++
src/style.css | 9 ++
src/weegtk.gresource.xml | 11 ++
src/weegtk.in | 46 ++++++
src/window.py | 70 ++++++++
src/window.ui | 87 ++++++++++
19 files changed, 2206 insertions(+)
create mode 100644 src/__init__.py
create mode 100644 src/chat.py
create mode 100644 src/config.py
create mode 100644 src/gtk/chat.ui
create mode 100644 src/gtk/help-overlay.ui
create mode 100644 src/gtk/message.ui
create mode 100644 src/gtk/preferences.ui
create mode 100644 src/main.py
create mode 100644 src/meson.build
create mode 100644 src/message.py
create mode 100644 src/preferences.py
create mode 100644 src/relay/color.py
create mode 100644 src/relay/network.py
create mode 100644 src/relay/protocol.py
create mode 100644 src/style.css
create mode 100644 src/weegtk.gresource.xml
create mode 100755 src/weegtk.in
create mode 100644 src/window.py
create mode 100644 src/window.ui
(limited to 'src')
diff --git a/src/__init__.py b/src/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/chat.py b/src/chat.py
new file mode 100644
index 0000000..f876854
--- /dev/null
+++ b/src/chat.py
@@ -0,0 +1,120 @@
+# window.py
+#
+# Copyright 2024 Lucas Fryzek
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+from gi.repository import Adw
+from gi.repository import Gtk
+from gi.repository import Gio
+from gi.repository import GObject
+
+from .message import WeegtkMessage
+from .color import Color
+from weegtk import config
+
+import json
+
+@Gtk.Template(resource_path='/com/fryzekconcepts/weegtk/gtk/chat.ui')
+class WeegtkChat(Adw.Bin):
+ __gtype_name__ = 'WeegtkChat'
+
+ __gsignals__ = {
+ "buffer_input" : (
+ GObject.SignalFlags.RUN_FIRST,
+ None,
+ (str, str)
+ )
+ }
+
+ window = Gtk.Template.Child()
+ messages = Gtk.Template.Child()
+ text_entry = Gtk.Template.Child()
+
+ def __init__(self, data=None, **kwargs):
+ super().__init__(**kwargs)
+
+ self.data = data or {}
+
+ self.model = Gtk.StringList()
+ self.select = Gtk.NoSelection(model=self.model)
+
+ self.factory = Gtk.SignalListItemFactory()
+ self.factory.connect("setup", self.setup_list_item)
+ self.factory.connect("bind", self.bind_list_item)
+
+ self.messages.set_model(self.select)
+ self.messages.set_factory(self.factory)
+
+ self.color = Color(config.color_options(), False)
+
+ def scroll_bottom(self, *args):
+ n_items = self.model.get_n_items()
+ self.messages.scroll_to(n_items - 1, Gtk.ListScrollFlags.FOCUS)
+
+ def setup_list_item(self, factory, list_item, *user_data):
+ message = WeegtkMessage()
+ list_item.set_child(message)
+ pass
+
+ def bind_list_item(self, factory, list_item, *user_data):
+ text = list_item.get_item().get_string()
+ data = json.loads(text)
+ message = list_item.get_child()
+ message.set_contents(data["username"], data["text"])
+ pass
+
+ def is_chat(self):
+ buf_type = self.data['local_variables']['type'] if 'type' in self.data['local_variables'] else None
+ return buf_type is not None and (buf_type == "private" or buf_type == "channel")
+
+ def pointer(self):
+ """Return pointer on buffer."""
+ return self.data.get('__path', [''])[0]
+
+ def parse_out_colors(self, text):
+ clean_text = self.color.convert(text)
+ return clean_text
+
+ def display(self, time, prefix, text, forcecolor=None):
+ # TODO using JSON here seems a bit ineffcient
+ # See if its possible to make a model that uses
+ # our own custom struct
+ user = self.parse_out_colors(prefix)
+ msg = self.parse_out_colors(text)
+ count = self.model.get_n_items()
+
+ if count != 0:
+ last = self.model.get_string(count - 1)
+ last_data = json.loads(last)
+ if last_data["username"] == user:
+ last_data["text"].append(msg)
+ self.model.splice(count - 1, 1, [json.dumps(last_data)])
+ return
+
+ data = {
+ "username": user,
+ "text": [msg]
+ }
+ self.model.append(json.dumps(data))
+ self.scroll_bottom()
+
+ @Gtk.Template.Callback()
+ def entry_activate(self, *args):
+ entry_buffer = self.text_entry.get_buffer()
+ text = entry_buffer.get_text()
+ entry_buffer.set_text("", 0)
+ self.emit("buffer_input", self.data['full_name'], text)
+
diff --git a/src/config.py b/src/config.py
new file mode 100644
index 0000000..ee9d611
--- /dev/null
+++ b/src/config.py
@@ -0,0 +1,140 @@
+# main.py
+#
+# Copyright 2024 Lucas Fryzek
+# Based on code from QWeeChat
+# Copyright (C) 2011-2022 Sébastien Helleu <flashcode@flashtux.org>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+"""Configuration for Weegtk."""
+
+import configparser
+import os
+
+from gi.repository import GLib
+
+from pathlib import Path
+
+CONFIG_DIR = f"{GLib.get_user_config_dir()}/weegtk"
+CONFIG_FILENAME = f"{CONFIG_DIR}/weegtk.conf"
+
+CONFIG_DEFAULT_RELAY_LINES = 50
+
+CONFIG_DEFAULT_SECTIONS = ('relay', 'look', 'color', 'ssh')
+CONFIG_DEFAULT_OPTIONS = (('relay.hostname', ''),
+ ('relay.port', ''),
+ ('relay.ssl', 'off'),
+ ('relay.password', ''),
+ ('relay.autoconnect', 'off'),
+ ('relay.lines', str(CONFIG_DEFAULT_RELAY_LINES)),
+ ('ssh.host', ''),
+ ('ssh.port', ''),
+ ('ssh.username', ''),
+ ('ssh.key', ''),
+ ('look.debug', 'off'),
+ ('look.statusbar', 'off'))
+
+# Default colors for WeeChat color options (option name, #rgb value)
+CONFIG_DEFAULT_COLOR_OPTIONS = (
+ ('separator', '#000066'), # 0
+ ('chat', '#000000'), # 1
+ ('chat_time', '#999999'), # 2
+ ('chat_time_delimiters', '#000000'), # 3
+ ('chat_prefix_error', '#FF6633'), # 4
+ ('chat_prefix_network', '#990099'), # 5
+ ('chat_prefix_action', '#000000'), # 6
+ ('chat_prefix_join', '#00CC00'), # 7
+ ('chat_prefix_quit', '#CC0000'), # 8
+ ('chat_prefix_more', '#CC00FF'), # 9
+ ('chat_prefix_suffix', '#330099'), # 10
+ ('chat_buffer', '#000000'), # 11
+ ('chat_server', '#000000'), # 12
+ ('chat_channel', '#000000'), # 13
+ ('chat_nick', '#000000'), # 14
+ ('chat_nick_self', '*#000000'), # 15
+ ('chat_nick_other', '#000000'), # 16
+ ('', '#000000'), # 17 (nick1 -- obsolete)
+ ('', '#000000'), # 18 (nick2 -- obsolete)
+ ('', '#000000'), # 19 (nick3 -- obsolete)
+ ('', '#000000'), # 20 (nick4 -- obsolete)
+ ('', '#000000'), # 21 (nick5 -- obsolete)
+ ('', '#000000'), # 22 (nick6 -- obsolete)
+ ('', '#000000'), # 23 (nick7 -- obsolete)
+ ('', '#000000'), # 24 (nick8 -- obsolete)
+ ('', '#000000'), # 25 (nick9 -- obsolete)
+ ('', '#000000'), # 26 (nick10 -- obsolete)
+ ('chat_host', '#666666'), # 27
+ ('chat_delimiters', '#9999FF'), # 28
+ ('chat_highlight', '#3399CC'), # 29
+ ('chat_read_marker', '#000000'), # 30
+ ('chat_text_found', '#000000'), # 31
+ ('chat_value', '#000000'), # 32
+ ('chat_prefix_buffer', '#000000'), # 33
+ ('chat_tags', '#000000'), # 34
+ ('chat_inactive_window', '#000000'), # 35
+ ('chat_inactive_buffer', '#000000'), # 36
+ ('chat_prefix_buffer_inactive_buffer', '#000000'), # 37
+ ('chat_nick_offline', '#000000'), # 38
+ ('chat_nick_offline_highlight', '#000000'), # 39
+ ('chat_nick_prefix', '#000000'), # 40
+ ('chat_nick_suffix', '#000000'), # 41
+ ('emphasis', '#000000'), # 42
+ ('chat_day_change', '#000000'), # 43
+)
+config_options = {"color": []}
+
+
+def read():
+ """Read config file."""
+ global config_options
+ config = configparser.RawConfigParser()
+ if os.path.isfile(CONFIG_FILENAME):
+ config.read(CONFIG_FILENAME)
+
+ # add missing sections/options
+ for section in CONFIG_DEFAULT_SECTIONS:
+ if not config.has_section(section):
+ config.add_section(section)
+ for option in reversed(CONFIG_DEFAULT_OPTIONS):
+ section, name = option[0].split('.', 1)
+ if not config.has_option(section, name):
+ config.set(section, name, option[1])
+ section = 'color'
+ for option in reversed(CONFIG_DEFAULT_COLOR_OPTIONS):
+ if option[0] and not config.has_option(section, option[0]):
+ config.set(section, option[0], option[1])
+
+ # build list of color options
+ config_options = {"color": []}
+ for option in CONFIG_DEFAULT_COLOR_OPTIONS:
+ if option[0]:
+ config_options["color"].append(config.get('color', option[0]))
+ else:
+ config_options["color"].append('#000000')
+
+ return config
+
+
+def write(config):
+ """Write config file."""
+ Path(CONFIG_DIR).mkdir(mode=0o0700, parents=True, exist_ok=True)
+ with open(CONFIG_FILENAME, 'w') as cfg:
+ config.write(cfg)
+
+
+def color_options():
+ """Return color options."""
+ global config_options
+ return config_options["color"]
diff --git a/src/gtk/chat.ui b/src/gtk/chat.ui
new file mode 100644
index 0000000..23477ad
--- /dev/null
+++ b/src/gtk/chat.ui
@@ -0,0 +1,67 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<interface>
+ <requires lib="gtk" version="4.0"/>
+ <requires lib="libadwaita" version="1.0"/>
+ <template class="WeegtkChat" parent="AdwBin">
+ <style>
+ <class name="view"/>
+ </style>
+ <property name="child">
+ <object class="AdwToolbarView">
+ <child type="top">
+ <object class="AdwHeaderBar">
+ <property name="show-title">True</property>
+ </object>
+ </child>
+ <property name="content">
+ <object class="GtkBox">
+ <property name="orientation">vertical</property>
+ <child>
+ <object class="GtkScrolledWindow" id="window">
+ <property name="vexpand">True</property>
+ <property name="hscrollbar_policy">3</property>
+ <child>
+ <object class="AdwClampScrollable">
+ <property name="maximum-size">750</property>
+ <property name="tightening-threshold">550</property>
+ <property name="child">
+ <object class="GtkListView" id="messages">
+ <property name="orientation">1</property>
+ <style>
+ <class name="navigation-sidebar"/>
+ </style>
+ </object>
+ </property>
+ </object>
+ </child>
+ </object>
+ </child>
+ <child>
+ <object class="AdwClamp">
+ <property name="maximum-size">750</property>
+ <property name="tightening-threshold">550</property>
+ <property name="child">
+ <object class="GtkBox">
+ <property name="margin-top">5</property>
+ <property name="margin-bottom">5</property>
+ <child>
+ <object class="GtkEntry" id="text_entry">
+ <property name="halign">fill</property>
+ <property name="hexpand">True</property>
+ <property name="hexpand-set">True</property>
+ <property name="width-request">50</property>
+ <property name="margin-start">5</property>
+ <property name="margin-end">5</property>
+ <signal name="activate" handler="entry_activate"/>
+ </object>
+ </child>
+ </object>
+ </property>
+ </object>
+ </child>
+ </object>
+ </property>
+ </object>
+ </property>
+ </template>
+</interface>
diff --git a/src/gtk/help-overlay.ui b/src/gtk/help-overlay.ui
new file mode 100644
index 0000000..ef12f02
--- /dev/null
+++ b/src/gtk/help-overlay.ui
@@ -0,0 +1,29 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<interface>
+ <object class="GtkShortcutsWindow" id="help_overlay">
+ <property name="modal">True</property>
+ <child>
+ <object class="GtkShortcutsSection">
+ <property name="section-name">shortcuts</property>
+ <property name="max-height">10</property>
+ <child>
+ <object class="GtkShortcutsGroup">
+ <property name="title" translatable="yes" context="shortcut window">General</property>
+ <child>
+ <object class="GtkShortcutsShortcut">
+ <property name="title" translatable="yes" context="shortcut window">Show Shortcuts</property>
+ <property name="action-name">win.show-help-overlay</property>
+ </object>
+ </child>
+ <child>
+ <object class="GtkShortcutsShortcut">
+ <property name="title" translatable="yes" context="shortcut window">Quit</property>
+ <property name="action-name">app.quit</property>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+</interface>
diff --git a/src/gtk/message.ui b/src/gtk/message.ui
new file mode 100644
index 0000000..dca0237
--- /dev/null
+++ b/src/gtk/message.ui
@@ -0,0 +1,63 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<interface>
+ <requires lib="gtk" version="4.0"/>
+ <template class="WeegtkMessage" parent="GtkBox">
+ <property name="orientation">horizontal</property>
+ <property name="valign">start</property>
+ <property name="halign">start</property>
+ <!-- <property name="margin-top">2</property> -->
+ <!-- <property name="margin-bottom">2</property> -->
+ <style>
+ <class name="message"/>
+ </style>
+ <child>
+ <object class="AdwAvatar" id="avatar">
+ <property name="size">32</property>
+ <property name="text" translatable="true">Username</property>
+ <property name="show-initials">true</property>
+ <property name="margin-start">10</property>
+ <property name="margin-end">10</property>
+ <property name="valign">start</property>
+ </object>
+ </child>
+ <child>
+ <object class="GtkBox" id="MessageText">
+ <property name="orientation">1</property>
+ <property name="valign">1</property>
+ <property name="halign">1</property>
+ <child>
+ <object class="GtkLabel" id="username">
+ <property name="label" translatable="true">Username</property>
+ <property name="halign">1</property>
+ <property name="selectable">true</property>
+ <style>
+ <class name="username"/>
+ </style>
+ </object>
+ </child>
+ <child>
+ <object class="GtkBox" id="message_list">
+ <property name="orientation">1</property>
+ <property name="valign">1</property>
+ <property name="halign">1</property>
+ <!-- <child> -->
+ <!-- <object class="GtkLabel" id="message"> -->
+ <!-- <property name="label" translatable="true">This is the the message that some user sent</property> -->
+ <!-- <property name="halign">1</property> -->
+ <!-- <property name="selectable">true</property> -->
+ <!-- <property name="wrap">true</property> -->
+ <!-- <property name="halign">0</property> -->
+ <!-- <property name="hexpand">true</property> -->
+ <!-- <property name="hexpand-set">true</property> -->
+ <!-- <property name="xalign">0</property> -->
+ <!-- <style> -->
+ <!-- <class name="message"/> -->
+ <!-- </style> -->
+ <!-- </object> -->
+ <!-- </child> -->
+ </object>
+ </child>
+ </object>
+ </child>
+ </template>
+</interface>
diff --git a/src/gtk/preferences.ui b/src/gtk/preferences.ui
new file mode 100644
index 0000000..5823d2f
--- /dev/null
+++ b/src/gtk/preferences.ui
@@ -0,0 +1,83 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<interface>
+ <requires lib="gtk" version="4.0"/>
+ <requires lib="libadwaita" version="1.0"/>
+ <template class="WeegtkPreferences" parent="AdwPreferencesDialog">
+ <property name="content-height">1000</property>
+ <property name="search-enabled">True</property>
+ <signal name="closed" handler="preferences_closed"/>
+ <child>
+ <object class="AdwPreferencesPage">
+ <property name="icon_name">preferences-window-layout-symbolic</property>
+ <property name="title">Settings</property>
+ <property name="use-underline">True</property>
+ <child>
+ <object class="AdwPreferencesGroup">
+ <property name="title" translatable="yes">Connection</property>
+ <child>
+ <object class="AdwEntryRow" id="hostname">
+ <property name="title" translatable="yes">Hostname</property>
+ <property name="use_underline">True</property>
+ </object>
+ </child>
+ <child>
+ <object class="AdwEntryRow" id="port">
+ <property name="title" translatable="yes">Port</property>
+ <property name="use_underline">True</property>
+ </object>
+ </child>
+ <child>
+ <object class="AdwPasswordEntryRow" id="password">
+ <property name="title" translatable="yes">Password</property>
+ </object>
+ </child>
+ <child>
+ <object class="AdwEntryRow" id="ssh_host">
+ <property name="title" translatable="yes">SSH host</property>
+ <property name="use_underline">True</property>
+ </object>
+ </child>
+ <child>
+ <object class="AdwEntryRow" id="ssh_port">
+ <property name="title" translatable="yes">SSH port</property>
+ <property name="use_underline">True</property>
+ </object>
+ </child>
+ <child>
+ <object class="AdwEntryRow" id="ssh_username">
+ <property name="title" translatable="yes">SSH username</property>
+ <property name="use_underline">True</property>
+ </object>
+ </child>
+ <child>
+ <object class="AdwActionRow">
+ <property name="title" translatable="yes">SSH key</property>
+ <child>
+ <object class="GtkButton">
+ <property name="valign">center</property>
+ <property name="action-name">ssh_key.open</property>
+ <child>
+ <object class="GtkBox">
+ <property name="spacing">6</property>
+ <child>
+ <object class="GtkImage">
+ <property name="icon-name">document-open-symbolic</property>
+ </object>
+ </child>
+ <child>
+ <object class="GtkLabel" id="ssh_key_label">
+ <property name="ellipsize">middle</property>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </object>
+ </child>
+ </template>
+</interface>
diff --git a/src/main.py b/src/main.py
new file mode 100644
index 0000000..8997ab5
--- /dev/null
+++ b/src/main.py
@@ -0,0 +1,338 @@
+# main.py
+#
+# Copyright 2024 Lucas Fryzek
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+import sys
+import os
+import gi
+
+gi.require_version('Gtk', '4.0')
+gi.require_version('Adw', '1')
+
+from gi.repository import Gtk, Gio, Adw
+from .window import WeegtkWindow
+
+from .network import Network
+from weegtk import protocol
+from weegtk import config
+from .chat import WeegtkChat
+from .preferences import WeegtkPreferences
+
+class WeegtkApplication(Adw.Application):
+ """The main application singleton class."""
+
+ def __init__(self):
+ super().__init__(application_id='com.fryzekconcepts.weegtk',
+ flags=Gio.ApplicationFlags.DEFAULT_FLAGS)
+ self.create_action('quit', lambda *_: self.quit(), ['<primary>q'])
+ self.create_action('about', self.on_about_action)
+ self.create_action('preferences', self.on_preferences_action)
+
+ self.network = Network()
+ self.network.connect("status_changed", self._network_status_changed)
+ self.network.connect("message_from_weechat", self._network_weechat_msg)
+
+ conf = config.read()
+ #self.network.connect_weechat("localhost", "9001", False, relay_pw, None, "")
+ self.network.connect_weechat_ssh(conf["ssh"]["host"],
+ conf["ssh"]["port"],
+ conf["ssh"]["username"],
+ conf["ssh"]["key"],
+ conf["relay"]["hostname"],
+ conf["relay"]["port"],
+ conf["relay"]["password"])
+
+ self.buffers = []
+
+ def _network_status_changed(self, source_object, status, extra):
+ """Called when the network status has changed."""
+ # TODO handle this
+ #if self.config.getboolean('look', 'statusbar'):
+ # self.statusBar().showMessage(status)
+ #self.network.debug_print(0, '', status, forcecolor='#0000AA')
+ #self.network_status_set(status)
+ print(f"Status is {status}")
+
+ def _network_weechat_msg(self, source_object, message):
+ """Called when a message is received from WeeChat."""
+ #print(f"Message {message}")
+ try:
+ proto = protocol.Protocol()
+ message = proto.decode(message.get_data())
+ # TODO figure this out
+ #if message.uncompressed:
+ # self.network.debug_print(
+ # 0, '==>',
+ # 'message uncompressed (%d bytes):\n%s'
+ # % (message.size_uncompressed,
+ # protocol.hex_and_ascii(message.uncompressed, 20)),
+ # forcecolor='#008800')
+ #self.network.debug_print(0, '', 'Message: %s' % message)
+ #print(f"parsed message is f{message}")
+ self.parse_message(message)
+ except:
+ print('Error while decoding message from WeeChat:\n%s'
+ % traceback.format_exc())
+ self.net.disconnect_weechat()
+
+ def do_activate(self):
+ """Called when the application is activated.
+
+ We raise the application's main window, creating it if
+ necessary.
+ """
+ win = self.props.active_window
+ if not win:
+ win = WeegtkWindow(application=self)
+ win.present()
+
+ def on_about_action(self, *args):
+ """Callback for the app.about action."""
+ about = Adw.AboutDialog(application_name='weegtk',
+ application_icon='com.fryzekconcepts.weegtk',
+ developer_name='Lucas Fryzek',
+ version='0.1.0',
+ developers=['Lucas Fryzek <lucas.fryzek@fryzekconcepts.com>'],
+ copyright='© 2024 Lucas Fryzek')
+ # Translators: Replace "translator-credits" with your name/username, and optionally an email or URL.
+ about.set_translator_credits(_('translator-credits'))
+ about.add_credit_section("QWeechat", ["Sébastien Helleu <flashcode@flashtux.org>"])
+ about.present(self.props.active_window)
+
+ def on_preferences_action(self, widget, _):
+ """Callback for the app.preferences action."""
+ pref = WeegtkPreferences()
+ pref.present(self.props.active_window)
+
+ def create_action(self, name, callback, shortcuts=None):
+ """Add an application action.
+
+ Args:
+ name: the name of the action
+ callback: the function to be called when the action is
+ activated
+ shortcuts: an optional list of accelerators
+ """
+ action = Gio.SimpleAction.new(name, None)
+ action.connect("activate", callback)
+ self.add_action(action)
+ if shortcuts:
+ self.set_accels_for_action(f"app.{name}", shortcuts)
+
+ def _parse_handshake(self, message):
+ """Parse a WeeChat message with handshake response."""
+ for obj in message.objects:
+ if obj.objtype != 'htb':
+ continue
+ self.network.init_with_handshake(obj.value)
+ break
+
+ def _parse_listbuffers(self, message):
+ """Parse a WeeChat message with list of buffers."""
+ # TODO handle gui for this
+ for obj in message.objects:
+ if obj.objtype != 'hda' or obj.value['path'][-1] != 'buffer':
+ continue
+ #self.list_buffers.clear()
+ #while self.stacked_buffers.count() > 0:
+ # buf = self.stacked_buffers.widget(0)
+ # self.stacked_buffers.removeWidget(buf)
+ self.buffers = []
+ for item in obj.value['items']:
+ buf = self.create_buffer(item)
+ self.insert_buffer(len(self.buffers), buf)
+ #self.list_buffers.setCurrentRow(0)
+ #self.buffers[0].widget.input.setFocus()
+
+ def _parse_line(self, message):
+ """Parse a WeeChat message with a buffer line."""
+ # TODO implement text for chats
+ for obj in message.objects:
+ lines = []
+ if obj.objtype != 'hda' or obj.value['path'][-1] != 'line_data':
+ continue
+ for item in obj.value['items']:
+ if message.msgid == 'listlines':
+ ptrbuf = item['__path'][0]
+ else:
+ ptrbuf = item['buffer']
+ index = [i for i, b in enumerate(self.buffers)
+ if b.pointer() == ptrbuf]
+ if index:
+ lines.append(
+ (index[0],
+ (item['date'], item['prefix'],
+ item['message']))
+ )
+ if message.msgid == 'listlines':
+ lines.reverse()
+ for line in lines:
+ #print(line)
+ self.buffers[line[0]].display(*line[1])
+
+ def _parse_nicklist(self, message):
+ """Parse a WeeChat message with a buffer nicklist."""
+ # TODO implement nicklist for chat
+ # buffer_refresh = {}
+ # for obj in message.objects:
+ # if obj.objtype != 'hda' or \
+ # obj.value['path'][-1] != 'nicklist_item':
+ # continue
+ # group = '__root'
+ # for item in obj.value['items']:
+ # index = [i for i, b in enumerate(self.buffers)
+ # if b.pointer() == item['__path'][0]]
+ # if index:
+ # if not index[0] in buffer_refresh:
+ # self.buffers[index[0]].nicklist = {}
+ # buffer_refresh[index[0]] = True
+ # if item['group']:
+ # group = item['name']
+ # self.buffers[index[0]].nicklist_add_item(
+ # group, item['group'], item['prefix'], item['name'],
+ # item['visible'])
+ # for index in buffer_refresh:
+ # self.buffers[index].nicklist_refresh()
+
+ def _parse_nicklist_diff(self, message):
+ """Parse a WeeChat message with a buffer nicklist diff."""
+ buffer_refresh = {}
+ for obj in message.objects:
+ if obj.objtype != 'hda' or \
+ obj.value['path'][-1] != 'nicklist_item':
+ continue
+ group = '__root'
+ for item in obj.value['items']:
+ index = [i for i, b in enumerate(self.buffers)
+ if b.pointer() == item['__path'][0]]
+ if not index:
+ continue
+ buffer_refresh[index[0]] = True
+ if item['_diff'] == ord('^'):
+ group = item['name']
+ elif item['_diff'] == ord('+'):
+ self.buffers[index[0]].nicklist_add_item(
+ group, item['group'], item['prefix'], item['name'],
+ item['visible'])
+ elif item['_diff'] == ord('-'):
+ self.buffers[index[0]].nicklist_remove_item(
+ group, item['group'], item['name'])
+ elif item['_diff'] == ord('*'):
+ self.buffers[index[0]].nicklist_update_item(
+ group, item['group'], item['prefix'], item['name'],
+ item['visible'])
+ for index in buffer_refresh:
+ self.buffers[index].nicklist_refresh()
+
+ def _parse_buffer_opened(self, message):
+ """Parse a WeeChat message with a new buffer (opened)."""
+ for obj in message.objects:
+ if obj.objtype != 'hda' or obj.value['path'][-1] != 'buffer':
+ continue
+ for item in obj.value['items']:
+ buf = self.create_buffer(item)
+ index = self.find_buffer_index_for_insert(item['next_buffer'])
+ self.insert_buffer(index, buf)
+
+ def _parse_buffer(self, message):
+ """Parse a WeeChat message with a buffer event
+ (anything except a new buffer).
+ """
+ for obj in message.objects:
+ if obj.objtype != 'hda' or obj.value['path'][-1] != 'buffer':
+ continue
+ for item in obj.value['items']:
+ index = [i for i, b in enumerate(self.buffers)
+ if b.pointer() == item['__path'][0]]
+ if not index:
+ continue
+ index = index[0]
+ if message.msgid == '_buffer_type_changed':
+ self.buffers[index].data['type'] = item['type']
+ elif message.msgid in ('_buffer_moved', '_buffer_merged',
+ '_buffer_unmerged'):
+ buf = self.buffers[index]
+ buf.data['number'] = item['number']
+ self.remove_buffer(index)
+ index2 = self.find_buffer_index_for_insert(
+ item['next_buffer'])
+ self.insert_buffer(index2, buf)
+ elif message.msgid == '_buffer_renamed':
+ self.buffers[index].data['full_name'] = item['full_name']
+ self.buffers[index].data['short_name'] = item['short_name']
+ elif message.msgid == '_buffer_title_changed':
+ self.buffers[index].data['title'] = item['title']
+ self.buffers[index].update_title()
+ elif message.msgid == '_buffer_cleared':
+ self.buffers[index].widget.chat.clear()
+ elif message.msgid.startswith('_buffer_localvar_'):
+ self.buffers[index].data['local_variables'] = \
+ item['local_variables']
+ self.buffers[index].update_prompt()
+ elif message.msgid == '_buffer_closing':
+ self.remove_buffer(index)
+
+ def parse_message(self, message):
+ """Parse a WeeChat message."""
+ if message.msgid.startswith('debug'):
+ self.network.debug_print(0, '', '(debug message, ignored)')
+ elif message.msgid == 'handshake':
+ self._parse_handshake(message)
+ elif message.msgid == 'listbuffers':
+ self._parse_listbuffers(message)
+ elif message.msgid in ('listlines', '_buffer_line_added'):
+ self._parse_line(message)
+ elif message.msgid in ('_nicklist', 'nicklist'):
+ self._parse_nicklist(message)
+ elif message.msgid == '_nicklist_diff':
+ self._parse_nicklist_diff(message)
+ elif message.msgid == '_buffer_opened':
+ self._parse_buffer_opened(message)
+ elif message.msgid.startswith('_buffer_'):
+ self._parse_buffer(message)
+ elif message.msgid == '_upgrade':
+ self.network.desync_weechat()
+ elif message.msgid == '_upgrade_ended':
+ self.network.sync_weechat()
+ else:
+ print(f"Unknown message with id {message.msgid}")
+
+ def create_buffer(self, item):
+ """Create a new buffer."""
+ buf = WeegtkChat(data=item)
+ buf.connect("buffer_input", self.buffer_input)
+ return buf
+
+ def insert_buffer(self, index, buf):
+ """Insert a buffer in list."""
+ self.buffers.insert(index, buf)
+ buf_name = buf.data['short_name']
+
+ # Only make chats visible on main screen
+ if buf.is_chat():
+ self.props.active_window.stack.add_titled(buf, name=buf_name, title=buf_name)
+
+ def buffer_input(self, source_object, full_name, text):
+ if self.network.is_connected():
+ message = f"input {full_name} {text}\n"
+ self.network.send_to_weechat(message)
+
+def main(version):
+ """The application's entry point."""
+ app = WeegtkApplication()
+ return app.run(sys.argv)
diff --git a/src/meson.build b/src/meson.build
new file mode 100644
index 0000000..3225adc
--- /dev/null
+++ b/src/meson.build
@@ -0,0 +1,42 @@
+pkgdatadir = get_option('prefix') / get_option('datadir') / meson.project_name()
+moduledir = pkgdatadir / 'weegtk'
+gnome = import('gnome')
+
+gnome.compile_resources('weegtk',
+ 'weegtk.gresource.xml',
+ gresource_bundle: true,
+ install: true,
+ install_dir: pkgdatadir,
+)
+
+python = import('python')
+
+conf = configuration_data()
+conf.set('PYTHON', python.find_installation('python3').full_path())
+conf.set('VERSION', meson.project_version())
+conf.set('localedir', get_option('prefix') / get_option('localedir'))
+conf.set('pkgdatadir', pkgdatadir)
+
+configure_file(
+ input: 'weegtk.in',
+ output: 'weegtk',
+ configuration: conf,
+ install: true,
+ install_dir: get_option('bindir'),
+ install_mode: 'r-xr-xr-x'
+)
+
+weegtk_sources = [
+ '__init__.py',
+ 'main.py',
+ 'window.py',
+ 'config.py',
+ 'chat.py',
+ 'message.py',
+ 'preferences.py',
+ 'relay/network.py',
+ 'relay/protocol.py',
+ 'relay/color.py'
+]
+
+install_data(weegtk_sources, install_dir: moduledir)
diff --git a/src/message.py b/src/message.py
new file mode 100644
index 0000000..a6e69a4
--- /dev/null
+++ b/src/message.py
@@ -0,0 +1,48 @@
+# window.py
+#
+# Copyright 2024 Lucas Fryzek
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+from gi.repository import Adw
+from gi.repository import Gtk
+
+@Gtk.Template(resource_path='/com/fryzekconcepts/weegtk/gtk/message.ui')
+class WeegtkMessage(Gtk.Box):
+ __gtype_name__ = 'WeegtkMessage'
+
+ username = Gtk.Template.Child()
+ avatar = Gtk.Template.Child()
+ message_list = Gtk.Template.Child()
+
+ def __init__(self, username="username", messages=[], **kwargs):
+ super().__init__(**kwargs)
+
+ self.set_contents(username, messages)
+
+ def set_contents(self, username, messages):
+ self.username.set_label(username)
+ self.avatar.set_text(username)
+
+ first = True
+ for message in messages:
+ margin = 5 if not first else 0
+ first = False
+ msg = Gtk.Label(label=message, selectable=True,
+ wrap=True, halign=0, hexpand=True, hexpand_set=True,
+ xalign=0, margin_top=margin)
+ self.message_list.append(msg)
+
+
diff --git a/src/preferences.py b/src/preferences.py
new file mode 100644
index 0000000..d864012
--- /dev/null
+++ b/src/preferences.py
@@ -0,0 +1,90 @@
+# window.py
+#
+# Copyright 2024 Lucas Fryzek
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+from gi.repository import Adw
+from gi.repository import Gtk
+from gi.repository import GLib
+
+from weegtk import config
+
+import os
+
+@Gtk.Template(resource_path='/com/fryzekconcepts/weegtk/gtk/preferences.ui')
+class WeegtkPreferences(Adw.PreferencesDialog):
+ __gtype_name__ = 'WeegtkPreferences'
+
+ hostname = Gtk.Template.Child()
+ port = Gtk.Template.Child()
+ password = Gtk.Template.Child()
+
+ ssh_host = Gtk.Template.Child()
+ ssh_port = Gtk.Template.Child()
+ ssh_username = Gtk.Template.Child()
+ ssh_key_label = Gtk.Template.Child()
+
+ ssh_key_path = ""
+
+ def __init__(self, **kwargs):
+ super().__init__(**kwargs)
+
+ self.config = config.read()
+
+ self.hostname.set_text(self.config["relay"]["hostname"])
+ self.port.set_text(self.config["relay"]["port"])
+ self.password.set_text(self.config["relay"]["password"])
+
+ self.ssh_host.set_text(self.config["ssh"]["host"])
+ self.ssh_port.set_text(self.config["ssh"]["port"])
+ self.ssh_username.set_text(self.config["ssh"]["username"])
+ self.set_ssh_key(self.config["ssh"]["key"])
+
+ self.install_action("ssh_key.open", None, self.ssh_key_open)
+
+ def read_preferences(self):
+ self.config["relay"]["hostname"] = self.hostname.get_text()
+ self.config["relay"]["port"] = self.port.get_text()
+ self.config["relay"]["password"] = self.password.get_text()
+
+ self.config["ssh"]["host"] = self.ssh_host.get_text()
+ self.config["ssh"]["port"] = self.ssh_port.get_text()
+ self.config["ssh"]["username"] = self.ssh_username.get_text()
+ self.config["ssh"]["key"] = self.ssh_key_path
+
+ def set_ssh_key(self, key_path):
+ self.ssh_key_path = key_path
+ short_name = os.path.basename(key_path)
+ self.ssh_key_label.set_label(short_name)
+
+ def ssh_key_open_dialog(self, dialog, result, caller):
+ try:
+ file = dialog.open_finish(result)
+ except GLib.GError:
+ # gtk-dialog-error-quark: Dismissed by user
+ pass
+ else:
+ self.set_ssh_key(file.get_path())
+
+ def ssh_key_open(self, *args):
+ dialog = Gtk.FileDialog()
+ dialog.set_title("Select SSH private key")
+ dialog.open(self.get_root(), None, self.ssh_key_open_dialog, self)
+
+ @Gtk.Template.Callback()
+ def preferences_closed(self, *args):
+ self.read_preferences()
+ config.write(self.config)
diff --git a/src/relay/color.py b/src/relay/color.py
new file mode 100644
index 0000000..61b70ff
--- /dev/null
+++ b/src/relay/color.py
@@ -0,0 +1,198 @@
+# -*- coding: utf-8 -*-
+#
+# color.py - remove/replace colors in WeeChat strings
+#
+# Copyright (C) 2011-2022 Sébastien Helleu <flashcode@flashtux.org>
+#
+# This file is part of QWeeChat, a Qt remote GUI for WeeChat.
+#
+# QWeeChat is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# QWeeChat is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with QWeeChat. If not, see <http://www.gnu.org/licenses/>.
+#
+
+"""Remove/replace colors in WeeChat strings."""
+
+import re
+import logging
+
+RE_COLOR_ATTRS = r'[*!/_|]*'
+RE_COLOR_STD = r'(?:%s\d{2})' % RE_COLOR_ATTRS
+RE_COLOR_EXT = r'(?:@%s\d{5})' % RE_COLOR_ATTRS
+RE_COLOR_ANY = r'(?:%s|%s)' % (RE_COLOR_STD, RE_COLOR_EXT)
+# \x19: color code, \x1A: set attribute, \x1B: remove attribute, \x1C: reset
+RE_COLOR = re.compile(
+ r'(\x19(?:\d{2}|F%s|B\d{2}|B@\d{5}|E|\\*%s(~%s)?|@\d{5}|b.|\x1C))|\x1A.|'
+ r'\x1B.|\x1C'
+ % (RE_COLOR_ANY, RE_COLOR_ANY, RE_COLOR_ANY))
+
+TERMINAL_COLORS = \
+ '000000cd000000cd00cdcd000000cdcd00cd00cdcde5e5e5' \
+ '4d4d4dff000000ff00ffff000000ffff00ff00ffffffffff' \
+ '00000000002a0000550000800000aa0000d4002a00002a2a' \
+ '002a55002a80002aaa002ad400550000552a005555005580' \
+ '0055aa0055d400800000802a0080550080800080aa0080d4' \
+ '00aa0000aa2a00aa5500aa8000aaaa00aad400d40000d42a' \
+ '00d45500d48000d4aa00d4d42a00002a002a2a00552a0080' \
+ '2a00aa2a00d42a2a002a2a2a2a2a552a2a802a2aaa2a2ad4' \
+ '2a55002a552a2a55552a55802a55aa2a55d42a80002a802a' \
+ '2a80552a80802a80aa2a80d42aaa002aaa2a2aaa552aaa80' \
+ '2aaaaa2aaad42ad4002ad42a2ad4552ad4802ad4aa2ad4d4' \
+ '55000055002a5500555500805500aa5500d4552a00552a2a' \
+ '552a55552a80552aaa552ad455550055552a555555555580' \
+ '5555aa5555d455800055802a5580555580805580aa5580d4' \
+ '55aa0055aa2a55aa5555aa8055aaaa55aad455d40055d42a' \
+ '55d45555d48055d4aa55d4d480000080002a800055800080' \
+ '8000aa8000d4802a00802a2a802a55802a80802aaa802ad4' \
+ '80550080552a8055558055808055aa8055d480800080802a' \
+ '8080558080808080aa8080d480aa0080aa2a80aa5580aa80' \
+ '80aaaa80aad480d40080d42a80d45580d48080d4aa80d4d4' \
+ 'aa0000aa002aaa0055aa0080aa00aaaa00d4aa2a00aa2a2a' \
+ 'aa2a55aa2a80aa2aaaaa2ad4aa5500aa552aaa5555aa5580' \
+ 'aa55aaaa55d4aa8000aa802aaa8055aa8080aa80aaaa80d4' \
+ 'aaaa00aaaa2aaaaa55aaaa80aaaaaaaaaad4aad400aad42a' \
+ 'aad455aad480aad4aaaad4d4d40000d4002ad40055d40080' \
+ 'd400aad400d4d42a00d42a2ad42a55d42a80d42aaad42ad4' \
+ 'd45500d4552ad45555d45580d455aad455d4d48000d4802a' \
+ 'd48055d48080d480aad480d4d4aa00d4aa2ad4aa55d4aa80' \
+ 'd4aaaad4aad4d4d400d4d42ad4d455d4d480d4d4aad4d4d4' \
+ '0808081212121c1c1c2626263030303a3a3a4444444e4e4e' \
+ '5858586262626c6c6c7676768080808a8a8a9494949e9e9e' \
+ 'a8a8a8b2b2b2bcbcbcc6c6c6d0d0d0dadadae4e4e4eeeeee'
+
+# WeeChat basic colors (color name, index in terminal colors)
+WEECHAT_BASIC_COLORS = (
+ ('default', 0), ('black', 0), ('darkgray', 8), ('red', 1),
+ ('lightred', 9), ('green', 2), ('lightgreen', 10), ('brown', 3),
+ ('yellow', 11), ('blue', 4), ('lightblue', 12), ('magenta', 5),
+ ('lightmagenta', 13), ('cyan', 6), ('lightcyan', 14), ('gray', 7),
+ ('white', 0))
+
+
+log = logging.getLogger(__name__)
+
+
+class Color():
+ def __init__(self, color_options, debug=False):
+ self.color_options = color_options
+ self.debug = debug
+
+ def _rgb_color(self, index):
+ color = TERMINAL_COLORS[index*6:(index*6)+6]
+ col_r = int(color[0:2], 16) * 0.85
+ col_g = int(color[2:4], 16) * 0.85
+ col_b = int(color[4:6], 16) * 0.85
+ return '%02x%02x%02x' % (col_r, col_g, col_b)
+
+ def _convert_weechat_color(self, color):
+ # TODO figure out if I want to do something here
+ return ''
+
+ def _convert_terminal_color(self, fg_bg, attrs, color):
+ try:
+ index = int(color)
+ return '\x01(%s%s#%s)' % (fg_bg, attrs, self._rgb_color(index))
+ except Exception: # noqa: E722
+ log.debug('Error decoding terminal color "%s"', color)
+ return ''
+
+ def _convert_color_attr(self, fg_bg, color):
+ extended = False
+ if color[0].startswith('@'):
+ extended = True
+ color = color[1:]
+ attrs = ''
+ # keep_attrs = False
+ while color.startswith(('*', '!', '/', '_', '|')):
+ # TODO: manage the "keep attributes" flag
+ # if color[0] == '|':
+ # keep_attrs = True
+ attrs += color[0]
+ color = color[1:]
+ if extended:
+ return self._convert_terminal_color(fg_bg, attrs, color)
+ try:
+ index = int(color)
+ return self._convert_terminal_color(fg_bg, attrs,
+ WEECHAT_BASIC_COLORS[index][1])
+ except Exception: # noqa: E722
+ log.debug('Error decoding color "%s"', color)
+ return ''
+
+ def _attrcode_to_char(self, code):
+ codes = {
+ '\x01': '*',
+ '\x02': '!',
+ '\x03': '/',
+ '\x04': '_',
+ }
+ return codes.get(code, '')
+
+ def _convert_color(self, match):
+ # TODO figure out how to deal with color
+ color = match.group(0)
+ if color[0] == '\x19':
+ if color[1] == 'b':
+ # bar code, ignored
+ return ''
+ if color[1] == '\x1C':
+ # reset
+ return ''
+ if color[1] in ('F', 'B'):
+ # foreground or background
+ return self._convert_color_attr(color[1], color[2:])
+ if color[1] == '*':
+ # foreground with optional background
+ items = color[2:].split(',')
+ str_col = self._convert_color_attr('F', items[0])
+ if len(items) > 1:
+ str_col += self._convert_color_attr('B', items[1])
+ return str_col
+ if color[1] == '@':
+ # direct ncurses pair number, ignored
+ return ''
+ if color[1] == 'E':
+ # text emphasis, ignored
+ return ''
+ if color[1:].isdigit():
+ return self._convert_weechat_color(int(color[1:]))
+ elif color[0] == '\x1A':
+ # set attribute
+ return '\x01(+%s)' % self._attrcode_to_char(color[1])
+ elif color[0] == '\x1B':
+ # remove attribute
+ return '\x01(-%s)' % self._attrcode_to_char(color[1])
+ elif color[0] == '\x1C':
+ # reset
+ return ''
+ # should never be executed!
+ return match.group(0)
+
+ def _convert_color_debug(self, match):
+ group = match.group(0)
+ for code in (0x01, 0x02, 0x03, 0x04, 0x19, 0x1A, 0x1B):
+ group = group.replace(chr(code), '<x%02X>' % code)
+ return group
+
+ def convert(self, text):
+ if not text:
+ return ''
+ if self.debug:
+ return RE_COLOR.sub(self._convert_color_debug, text)
+ return RE_COLOR.sub(self._convert_color, text)
+
+
+def remove(text):
+ """Remove colors in a WeeChat string."""
+ if not text:
+ return ''
+ return re.sub(RE_COLOR, '', text)
diff --git a/src/relay/network.py b/src/relay/network.py
new file mode 100644
index 0000000..1071f1a
--- /dev/null
+++ b/src/relay/network.py
@@ -0,0 +1,403 @@
+# -*- coding: utf-8 -*-
+#
+# network.py - I/O with WeeChat/relay
+#
+# Copyright (C) 2011-2022 Sébastien Helleu <flashcode@flashtux.org>
+#
+# This file is part of QWeeChat, a Qt remote GUI for WeeChat.
+#
+# QWeeChat is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# QWeeChat is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with QWeeChat. If not, see <http://www.gnu.org/licenses/>.
+#
+
+"""I/O with WeeChat/relay."""
+
+import hashlib
+import secrets
+import struct
+
+from gi.repository import GObject, GLib, Gio
+
+from weegtk import config
+
+from sshtunnel import SSHTunnelForwarder
+
+# list of supported hash algorithms on our side
+# (the hash algorithm will be negotiated with the remote WeeChat)
+_HASH_ALGOS_LIST = [
+ 'plain',
+ 'sha256',
+ 'sha512',
+ 'pbkdf2+sha256',
+ 'pbkdf2+sha512',
+]
+_HASH_ALGOS = ':'.join(_HASH_ALGOS_LIST)
+
+# handshake with remote WeeChat (before init)
+_PROTO_HANDSHAKE = f'(handshake) handshake password_hash_algo={_HASH_ALGOS}\n'
+
+# initialize with the password (plain text)
+_PROTO_INIT_PWD = 'init password=%(password)s%(totp)s\n' # nosec
+
+# initialize with the hashed password
+_PROTO_INIT_HASH = ('init password_hash='
+ '%(algo)s:%(salt)s%(iter)s:%(hash)s%(totp)s\n')
+
+_PROTO_SYNC_CMDS = [
+ # get buffers
+ '(listbuffers) hdata buffer:gui_buffers(*) number,full_name,short_name,'
+ 'type,nicklist,title,local_variables',
+ # get lines
+ '(listlines) hdata buffer:gui_buffers(*)/own_lines/last_line(-%(lines)d)/'
+ 'data date,displayed,prefix,message',
+ # get nicklist for all buffers
+ '(nicklist) nicklist',
+ # enable synchronization
+ 'sync',
+]
+
+STATUS_DISCONNECTED = 'disconnected'
+STATUS_CONNECTING = 'connecting'
+STATUS_AUTHENTICATING = 'authenticating'
+STATUS_CONNECTED = 'connected'
+
+NETWORK_STATUS = {
+ STATUS_DISCONNECTED: {
+ 'label': 'Disconnected',
+ 'color': '#aa0000',
+ 'icon': 'dialog-close.png',
+ },
+ STATUS_CONNECTING: {
+ 'label': 'Connecting…',
+ 'color': '#dd5f00',
+ 'icon': 'dialog-warning.png',
+ },
+ STATUS_AUTHENTICATING: {
+ 'label': 'Authenticating…',
+ 'color': '#007fff',
+ 'icon': 'dialog-password.png',
+ },
+ STATUS_CONNECTED: {
+ 'label': 'Connected',
+ 'color': 'green',
+ 'icon': 'dialog-ok-apply.png',
+ },
+}
+
+
+class Network(GObject.GObject):
+ """I/O with WeeChat/relay."""
+ __gsignals__ = {
+ "status_changed" : (
+ GObject.SignalFlags.RUN_FIRST,
+ None,
+ (str, str)
+ ),
+ "message_from_weechat" : (
+ GObject.SignalFlags.RUN_FIRST,
+ None,
+ (GLib.Bytes,)
+ )
+ }
+
+ def __init__(self, *args):
+ super().__init__(*args)
+ self._init_connection()
+ self.debug_lines = []
+ self.debug_dialog = None
+ self._lines = config.CONFIG_DEFAULT_RELAY_LINES
+ self.cancel_network_reads = Gio.Cancellable()
+ self._buffer = bytearray()
+ self._socketclient = Gio.SocketClient.new()
+ self._socket = None
+
+ # TODO figure out how to deal with these signals
+ #self._socket.connected.self._socket_connected)
+ #self._socket.readyRead.connect(self._socket_read)
+ #self._socket.disconnected.connect(self._socket_disconnected)
+
+ def _init_connection(self):
+ self.status = STATUS_DISCONNECTED
+ self._hostname = None
+ self._port = None
+ self._ssl = None
+ self._password = None
+ self._totp = None
+ self._handshake_received = False
+ self._handshake_timer = None
+ self._pwd_hash_algo = None
+ self._pwd_hash_iter = 0
+ self._server_nonce = None
+
+ def set_status(self, status):
+ """Set current status."""
+ self.status = status
+ self.emit("status_changed", status, None)
+
+ def pbkdf2(self, hash_name, salt):
+ """Return hashed password with PBKDF2-HMAC."""
+ return hashlib.pbkdf2_hmac(
+ hash_name,
+ password=self._password.encode('utf-8'),
+ salt=salt,
+ iterations=self._pwd_hash_iter,
+ ).hex()
+
+ def _build_init_command(self):
+ """Build the init command to send to WeeChat."""
+ totp = f',totp={self._totp}' if self._totp else ''
+ if self._pwd_hash_algo == 'plain': # nosec
+ cmd = _PROTO_INIT_PWD % {
+ 'password': self._password,
+ 'totp': totp,
+ }
+ else:
+ client_nonce = secrets.token_bytes(16)
+ salt = self._server_nonce + client_nonce
+ pwd_hash = None
+ iterations = ''
+ if self._pwd_hash_algo == 'pbkdf2+sha512': # nosec
+ pwd_hash = self.pbkdf2('sha512', salt)
+ iterations = f':{self._pwd_hash_iter}'
+ elif self._pwd_hash_algo == 'pbkdf2+sha256': # nosec
+ pwd_hash = self.pbkdf2('sha256', salt)
+ iterations = f':{self._pwd_hash_iter}'
+ elif self._pwd_hash_algo == 'sha512': # nosec
+ pwd = salt + self._password.encode('utf-8')
+ pwd_hash = hashlib.sha512(pwd).hexdigest()
+ elif self._pwd_hash_algo == 'sha256': # nosec
+ pwd = salt + self._password.encode('utf-8')
+ pwd_hash = hashlib.sha256(pwd).hexdigest()
+ if not pwd_hash:
+ return None
+ cmd = _PROTO_INIT_HASH % {
+ 'algo': self._pwd_hash_algo,
+ 'salt': bytearray(salt).hex(),
+ 'iter': iterations,
+ 'hash': pwd_hash,
+ 'totp': totp,
+ }
+ return cmd
+
+ def _build_sync_command(self):
+ """Build the sync commands to send to WeeChat."""
+ cmd = '\n'.join(_PROTO_SYNC_CMDS) + '\n'
+ return cmd % {'lines': self._lines}
+
+ def handshake_timer_expired(self):
+ if self.status == STATUS_AUTHENTICATING:
+ self._pwd_hash_algo = 'plain' # nosec
+ self.send_to_weechat(self._build_init_command())
+ self.sync_weechat()
+ self.set_status(STATUS_CONNECTED)
+ return False
+
+ def _socket_connected(self):
+ """Slot: socket connected."""
+ self.set_status(STATUS_AUTHENTICATING)
+ self.send_to_weechat(_PROTO_HANDSHAKE)
+ self._handshake_timer = GLib.timeout_add(2000, self.handshake_timer_expired)
+
+ def _socket_read(self, source_object, res, *user_data):
+ """Slot: data available on socket."""
+ try:
+ gbytes = self.input.read_bytes_finish(res)
+ except GLib.Error as err:
+ self.handle_network_error(err)
+ return
+
+ self._buffer.extend(gbytes.get_data())
+ while len(self._buffer) >= 4:
+ remainder = None
+ length = struct.unpack('>i', self._buffer[0:4])[0]
+ if len(self._buffer) < length:
+ # partial message, just wait for end of message
+ break
+ # more than one message?
+ if length < len(self._buffer):
+ # save beginning of another message
+ remainder = self._buffer[length:]
+ self._buffer = self._buffer[0:length]
+ self.emit("message_from_weechat", GLib.Bytes(self._buffer))
+ if not self.is_connected():
+ return
+ self._buffer.clear()
+ if remainder:
+ self._buffer.extend(remainder)
+
+ self.input.read_bytes_async(
+ 4096, 0, self.cancel_network_reads, self._socket_read)
+
+ def _socket_disconnected(self):
+ """Slot: socket disconnected."""
+ if self._handshake_timer:
+ self._handshake_timer.stop()
+ self._init_connection()
+ self.set_status(STATUS_DISCONNECTED)
+
+ def is_connected(self):
+ """Return True if the socket is connected, False otherwise."""
+ return self._socket.is_connected()
+
+ def is_ssl(self):
+ """Return True if SSL is used, False otherwise."""
+ return self._ssl
+
+ def connect_weechat_ssh(self, ssh_host, ssh_port, ssh_username, ssh_key,
+ relay_host, relay_port, relay_pw):
+ self.ssh_tunnel = SSHTunnelForwarder(
+ ssh_host,
+ ssh_username=ssh_username,
+ ssh_pkey=ssh_key,
+ remote_bind_address=(relay_host, int(relay_port)))
+
+ self.ssh_tunnel.start()
+
+ self.connect_weechat("localhost", self.ssh_tunnel.local_bind_port,
+ False, relay_pw, None, "")
+
+ def connect_weechat(self, hostname, port, ssl, password, totp, lines):
+ """Connect to WeeChat."""
+ self._hostname = hostname
+ try:
+ self._port = int(port)
+ except ValueError:
+ self._port = 0
+ self._ssl = ssl
+ self._password = password
+ self._totp = totp
+ try:
+ self._lines = int(lines)
+ except ValueError:
+ self._lines = config.CONFIG_DEFAULT_RELAY_LINES
+
+ # TODO handle SSL
+ self._socketclient.connect_async(
+ Gio.NetworkAddress.new(self._hostname, self._port),
+ None,
+ self._connected_func, None)
+ self.set_status(STATUS_CONNECTING)
+
+ def _connected_func(self, source_object, res, *user_data):
+ """Callback function called after connection attempt."""
+ try:
+ self._socket = self._socketclient.connect_finish(res)
+ except GLib.Error as err:
+ print("Connection failed:\n{}".format(err.message))
+ self.set_status(STATUS_NOT_CONNECTED)
+ return
+ else:
+ print("Connected")
+ self.set_status(STATUS_CONNECTED)
+ self._socket_connected()
+ self.input = self._socket.get_input_stream()
+ self.input.read_bytes_async(
+ 4096, 0, self.cancel_network_reads, self._socket_read)
+
+ def disconnect_weechat(self):
+ """Disconnect from WeeChat."""
+ if self._socket.state() == QtNetwork.QAbstractSocket.UnconnectedState:
+ self.set_status(STATUS_DISCONNECTED)
+ return
+ if self._socket.state() == QtNetwork.QAbstractSocket.ConnectedState:
+ self.send_to_weechat('quit\n')
+ self._socket.waitForBytesWritten(1000)
+ else:
+ self.set_status(STATUS_DISCONNECTED)
+ self._socket.abort()
+
+ def send_to_weechat(self, message):
+ """Send a message to WeeChat."""
+ output = self._socket.get_output_stream()
+ try:
+ output.write(message.encode("utf-8"))
+ except GLib.Error as err:
+ self.handle_network_error(err)
+
+ def init_with_handshake(self, response):
+ """Initialize with WeeChat using the handshake response."""
+ self._pwd_hash_algo = response['password_hash_algo']
+ self._pwd_hash_iter = int(response['password_hash_iterations'])
+ self._server_nonce = bytearray.fromhex(response['nonce'])
+ if self._pwd_hash_algo:
+ cmd = self._build_init_command()
+ if cmd:
+ self.send_to_weechat(cmd)
+ self.sync_weechat()
+ self.set_status(STATUS_CONNECTED)
+ return
+ # failed to initialize: disconnect
+ self.disconnect_weechat()
+
+ def desync_weechat(self):
+ """Desynchronize from WeeChat."""
+ self.send_to_weechat('desync\n')
+
+ def sync_weechat(self):
+ """Synchronize with WeeChat."""
+ self.send_to_weechat(self._build_sync_command())
+
+ def status_label(self, status):
+ """Return the label for a given status."""
+ return NETWORK_STATUS.get(status, {}).get('label', '')
+
+ def status_color(self, status):
+ """Return the color for a given status."""
+ return NETWORK_STATUS.get(status, {}).get('color', 'black')
+
+ def status_icon(self, status):
+ """Return the name of icon for a given status."""
+ return NETWORK_STATUS.get(status, {}).get('icon', '')
+
+ def get_options(self):
+ """Get connection options."""
+ return {
+ 'hostname': self._hostname,
+ 'port': self._port,
+ 'ssl': 'on' if self._ssl else 'off',
+ 'password': self._password,
+ 'lines': str(self._lines),
+ }
+
+ def debug_print(self, *args, **kwargs):
+ """Display a debug message."""
+ self.debug_lines.append((args, kwargs))
+ if self.debug_dialog:
+ self.debug_dialog.chat.display(*args, **kwargs)
+
+ def _debug_dialog_closed(self, result):
+ """Called when debug dialog is closed."""
+ self.debug_dialog = None
+
+ def debug_input_text_sent(self, text):
+ """Send debug buffer input to WeeChat."""
+ if self.network.is_connected():
+ text = str(text)
+ pos = text.find(')')
+ if text.startswith('(') and pos >= 0:
+ text = '(debug_%s)%s' % (text[1:pos], text[pos+1:])
+ else:
+ text = '(debug) %s' % text
+ self.network.debug_print(0, '<==', text, forcecolor='#AA0000')
+ self.network.send_to_weechat(text + '\n')
+
+ def open_debug_dialog(self):
+ """Open a dialog with debug messages."""
+ if not self.debug_dialog:
+ self.debug_dialog = DebugDialog()
+ self.debug_dialog.input.textSent.connect(
+ self.debug_input_text_sent)
+ self.debug_dialog.finished.connect(self._debug_dialog_closed)
+ self.debug_dialog.display_lines(self.debug_lines)
+ self.debug_dialog.chat.scroll_bottom()
+
diff --git a/src/relay/protocol.py b/src/relay/protocol.py
new file mode 100644
index 0000000..6f70544
--- /dev/null
+++ b/src/relay/protocol.py
@@ -0,0 +1,362 @@
+# -*- coding: utf-8 -*-
+#
+# protocol.py - decode binary messages received from WeeChat/relay
+#
+# Copyright (C) 2011-2022 Sébastien Helleu <flashcode@flashtux.org>
+#
+# This file is part of QWeeChat, a Qt remote GUI for WeeChat.
+#
+# QWeeChat is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# QWeeChat is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with QWeeChat. If not, see <http://www.gnu.org/licenses/>.
+#
+
+#
+# For info about protocol and format of messages, please read document
+# "WeeChat Relay Protocol", available at: https://weechat.org/doc/
+#
+# History:
+#
+# 2011-11-23, Sébastien Helleu <flashcode@flashtux.org>:
+# start dev
+#
+
+"""Decode binary messages received from WeeChat/relay."""
+
+import collections
+import struct
+import zlib
+
+
+class WeechatDict(collections.OrderedDict):
+ def __str__(self):
+ return '{%s}' % ', '.join(
+ ['%s: %s' % (repr(key), repr(self[key])) for key in self])
+
+
+class WeechatObject:
+ def __init__(self, objtype, value, separator='\n'):
+ self.objtype = objtype
+ self.value = value
+ self.separator = separator
+ self.indent = ' ' if separator == '\n' else ''
+ self.separator1 = '\n%s' % self.indent if separator == '\n' else ''
+
+ def _str_value(self, val):
+ if isinstance(val, str) and val is not None:
+ return '\'%s\'' % val
+ return str(val)
+
+ def _str_value_hdata(self):
+ lines = ['%skeys: %s%s%spath: %s' % (self.separator1,
+ str(self.value['keys']),
+ self.separator,
+ self.indent,
+ str(self.value['path']))]
+ for i, item in enumerate(self.value['items']):
+ lines.append(' item %d:%s%s' % (
+ (i + 1), self.separator,
+ self.separator.join(
+ ['%s%s: %s' % (self.indent * 2, key,
+ self._str_value(value))
+ for key, value in item.items()])))
+ return '\n'.join(lines)
+
+ def _str_value_infolist(self):
+ lines = ['%sname: %s' % (self.separator1, self.value['name'])]
+ for i, item in enumerate(self.value['items']):
+ lines.append(' item %d:%s%s' % (
+ (i + 1), self.separator,
+ self.separator.join(
+ ['%s%s: %s' % (self.indent * 2, key,
+ self._str_value(value))
+ for key, value in item.items()])))
+ return '\n'.join(lines)
+
+ def _str_value_other(self):
+ return self._str_value(self.value)
+
+ def __str__(self):
+ obj_cb = {
+ 'hda': self._str_value_hdata,
+ 'inl': self._str_value_infolist,
+ }
+ return '%s: %s' % (self.objtype,
+ obj_cb.get(self.objtype, self._str_value_other)())
+
+
+class WeechatObjects(list):
+ def __init__(self, separator='\n'):
+ super().__init__()
+ self.separator = separator
+
+ def __str__(self):
+ return self.separator.join([str(obj) for obj in self])
+
+
+class WeechatMessage:
+ def __init__(self, size, size_uncompressed, compression, uncompressed,
+ msgid, objects):
+ self.size = size
+ self.size_uncompressed = size_uncompressed
+ self.compression = compression
+ self.uncompressed = uncompressed
+ self.msgid = msgid
+ self.objects = objects
+
+ def __str__(self):
+ if self.compression != 0:
+ return 'size: %d/%d (%d%%), id=\'%s\', objects:\n%s' % (
+ self.size, self.size_uncompressed,
+ 100 - ((self.size * 100) // self.size_uncompressed),
+ self.msgid, self.objects)
+ return 'size: %d, id=\'%s\', objects:\n%s' % (self.size,
+ self.msgid,
+ self.objects)
+
+
+class Protocol:
+ """Decode binary message received from WeeChat/relay."""
+
+ def __init__(self):
+ self.data = ''
+ self._obj_cb = {
+ 'chr': self._obj_char,
+ 'int': self._obj_int,
+ 'lon': self._obj_long,
+ 'str': self._obj_str,
+ 'buf': self._obj_buffer,
+ 'ptr': self._obj_ptr,
+ 'tim': self._obj_time,
+ 'htb': self._obj_hashtable,
+ 'hda': self._obj_hdata,
+ 'inf': self._obj_info,
+ 'inl': self._obj_infolist,
+ 'arr': self._obj_array,
+ }
+
+ def _obj_type(self):
+ """Read type in data (3 chars)."""
+ if len(self.data) < 3:
+ self.data = ''
+ return ''
+ objtype = self.data[0:3].decode()
+ self.data = self.data[3:]
+ return objtype
+
+ def _obj_len_data(self, length_size):
+ """Read length (1 or 4 bytes), then value with this length."""
+ if len(self.data) < length_size:
+ self.data = ''
+ return None
+ if length_size == 1:
+ length = struct.unpack('B', self.data[0:1])[0]
+ self.data = self.data[1:]
+ else:
+ length = self._obj_int()
+ if length < 0:
+ return None
+ if length > 0:
+ value = self.data[0:length]
+ self.data = self.data[length:]
+ else:
+ value = ''
+ return value
+
+ def _obj_char(self):
+ """Read a char in data."""
+ if len(self.data) < 1:
+ return 0
+ value = struct.unpack('b', self.data[0:1])[0]
+ self.data = self.data[1:]
+ return value
+
+ def _obj_int(self):
+ """Read an integer in data (4 bytes)."""
+ if len(self.data) < 4:
+ self.data = ''
+ return 0
+ value = struct.unpack('>i', self.data[0:4])[0]
+ self.data = self.data[4:]
+ return value
+
+ def _obj_long(self):
+ """Read a long integer in data (length on 1 byte + value as string)."""
+ value = self._obj_len_data(1)
+ if value is None:
+ return None
+ return int(value)
+
+ def _obj_str(self):
+ """Read a string in data (length on 4 bytes + content)."""
+ value = self._obj_len_data(4)
+ if value in ("", None):
+ return ""
+ return value.decode()
+
+ def _obj_buffer(self):
+ """Read a buffer in data (length on 4 bytes + data)."""
+ return self._obj_len_data(4)
+
+ def _obj_ptr(self):
+ """Read a pointer in data (length on 1 byte + value as string)."""
+ value = self._obj_len_data(1)
+ if value is None:
+ return None
+ return '0x%s' % value
+
+ def _obj_time(self):
+ """Read a time in data (length on 1 byte + value as string)."""
+ value = self._obj_len_data(1)
+ if value is None:
+ return None
+ return int(value)
+
+ def _obj_hashtable(self):
+ """
+ Read a hashtable in data
+ (type for keys + type for values + count + items).
+ """
+ type_keys = self._obj_type()
+ type_values = self._obj_type()
+ count = self._obj_int()
+ hashtable = WeechatDict()
+ for _ in range(count):
+ key = self._obj_cb[type_keys]()
+ value = self._obj_cb[type_values]()
+ hashtable[key] = value
+ return hashtable
+
+ def _obj_hdata(self):
+ """Read a hdata in data."""
+ path = self._obj_str()
+ keys = self._obj_str()
+ count = self._obj_int()
+ list_path = path.split('/') if path else []
+ list_keys = keys.split(',') if keys else []
+ keys_types = []
+ dict_keys = WeechatDict()
+ for key in list_keys:
+ items = key.split(':')
+ keys_types.append(items)
+ dict_keys[items[0]] = items[1]
+ items = []
+ for _ in range(count):
+ item = WeechatDict()
+ item['__path'] = []
+ pointers = []
+ for _ in enumerate(list_path):
+ pointers.append(self._obj_ptr())
+ for key, objtype in keys_types:
+ item[key] = self._obj_cb[objtype]()
+ item['__path'] = pointers
+ items.append(item)
+ return {
+ 'path': list_path,
+ 'keys': dict_keys,
+ 'count': count,
+ 'items': items,
+ }
+
+ def _obj_info(self):
+ """Read an info in data."""
+ name = self._obj_str()
+ value = self._obj_str()
+ return (name, value)
+
+ def _obj_infolist(self):
+ """Read an infolist in data."""
+ name = self._obj_str()
+ count_items = self._obj_int()
+ items = []
+ for _ in range(count_items):
+ count_vars = self._obj_int()
+ variables = WeechatDict()
+ for _ in range(count_vars):
+ var_name = self._obj_str()
+ var_type = self._obj_type()
+ var_value = self._obj_cb[var_type]()
+ variables[var_name] = var_value
+ items.append(variables)
+ return {
+ 'name': name,
+ 'items': items
+ }
+
+ def _obj_array(self):
+ """Read an array of values in data."""
+ type_values = self._obj_type()
+ count_values = self._obj_int()
+ values = []
+ for _ in range(count_values):
+ values.append(self._obj_cb[type_values]())
+ return values
+
+ def decode(self, data, separator='\n'):
+ """Decode binary data and return list of objects."""
+ self.data = data
+ size = len(self.data)
+ size_uncompressed = size
+ uncompressed = None
+ # uncompress data (if it is compressed)
+ compression = struct.unpack('b', self.data[4:5])[0]
+ if compression:
+ uncompressed = zlib.decompress(self.data[5:])
+ size_uncompressed = len(uncompressed) + 5
+ uncompressed = b'%s%s%s' % (struct.pack('>i', size_uncompressed),
+ struct.pack('b', 0), uncompressed)
+ self.data = uncompressed
+ else:
+ uncompressed = self.data[:]
+ # skip length and compression flag
+ self.data = self.data[5:]
+ # read id
+ msgid = self._obj_str()
+ if msgid is None:
+ msgid = ''
+ # read objects
+ objects = WeechatObjects(separator=separator)
+ while len(self.data) > 0:
+ objtype = self._obj_type()
+ value = self._obj_cb[objtype]()
+ objects.append(WeechatObject(objtype, value, separator=separator))
+ return WeechatMessage(size, size_uncompressed, compression,
+ uncompressed, msgid, objects)
+
+
+def hex_and_ascii(data, bytes_per_line=10):
+ """Convert a QByteArray to hex + ascii output."""
+ num_lines = ((len(data) - 1) // bytes_per_line) + 1
+ if num_lines == 0:
+ return ''
+ lines = []
+ for i in range(num_lines):
+ str_hex = []
+ str_ascii = []
+ for j in range(bytes_per_line):
+ # We can't easily iterate over individual bytes, so we are going to
+ # do it this way.
+ index = (i*bytes_per_line) + j
+ char = data[index:index+1]
+ if not char:
+ char = b'x'
+ byte = struct.unpack('B', char)[0]
+ str_hex.append(b'%02X' % int(byte))
+ if 32 <= byte <= 127:
+ str_ascii.append(char)
+ else:
+ str_ascii.append(b'.')
+ fmt = b'%%-%ds %%s' % ((bytes_per_line * 3) - 1)
+ lines.append(fmt % (b' '.join(str_hex),
+ b''.join(str_ascii)))
+ return b'\n'.join(lines)
+
diff --git a/src/style.css b/src/style.css
new file mode 100644
index 0000000..ae7eab8
--- /dev/null
+++ b/src/style.css
@@ -0,0 +1,9 @@
+label.username {
+ color: rgb(4, 97, 190);
+ font-weight: 700;
+}
+
+box.message {
+ padding-top: 2px;
+ padding-bottom: 2px;
+}
diff --git a/src/weegtk.gresource.xml b/src/weegtk.gresource.xml
new file mode 100644
index 0000000..c54ffe8
--- /dev/null
+++ b/src/weegtk.gresource.xml
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<gresources>
+ <gresource prefix="/com/fryzekconcepts/weegtk">
+ <file preprocess="xml-stripblanks">window.ui</file>
+ <file preprocess="xml-stripblanks">gtk/help-overlay.ui</file>
+ <file preprocess="xml-stripblanks">gtk/chat.ui</file>
+ <file preprocess="xml-stripblanks">gtk/message.ui</file>
+ <file preprocess="xml-stripblanks">gtk/preferences.ui</file>
+ <file compressed="true">style.css</file>
+ </gresource>
+</gresources>
diff --git a/src/weegtk.in b/src/weegtk.in
new file mode 100755
index 0000000..123a529
--- /dev/null
+++ b/src/weegtk.in
@@ -0,0 +1,46 @@
+#!@PYTHON@
+
+# weegtk.in
+#
+# Copyright 2024 Lucas Fryzek
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+import os
+import sys
+import signal
+import locale
+import gettext
+
+VERSION = '@VERSION@'
+pkgdatadir = '@pkgdatadir@'
+localedir = '@localedir@'
+
+sys.path.insert(1, pkgdatadir)
+signal.signal(signal.SIGINT, signal.SIG_DFL)
+locale.bindtextdomain('weegtk', localedir)
+locale.textdomain('weegtk')
+gettext.install('weegtk', localedir)
+
+if __name__ == '__main__':
+ import gi
+
+ from gi.repository import Gio
+ resource = Gio.Resource.load(os.path.join(pkgdatadir, 'weegtk.gresource'))
+ resource._register()
+
+ from weegtk import main
+ sys.exit(main.main(VERSION))
diff --git a/src/window.py b/src/window.py
new file mode 100644
index 0000000..df63e36
--- /dev/null
+++ b/src/window.py
@@ -0,0 +1,70 @@
+# window.py
+#
+# Copyright 2024 Lucas Fryzek
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <https://www.gnu.org/licenses/>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from gi.repository import Adw
+from gi.repository import Gtk
+
+@Gtk.Template(resource_path='/com/fryzekconcepts/weegtk/window.ui')
+class WeegtkWindow(Adw.ApplicationWindow):
+ __gtype_name__ = 'WeegtkWindow'
+
+ stack = Gtk.Template.Child()
+ content_page = Gtk.Template.Child()
+ color_scheme_button = Gtk.Template.Child()
+ split_view = Gtk.Template.Child()
+
+ def __init__(self, **kwargs):
+ super().__init__(**kwargs)
+ manager = Adw.StyleManager.get_default()
+
+ manager.connect("notify::system-supports-color-schemes",
+ self.notify_system_supports_color_schemes_cb,
+ self,
+ False)
+
+ self.notify_system_supports_color_schemes_cb()
+ self.notify_visible_child_cb()
+
+ def notify_system_supports_color_schemes_cb(self):
+ manager = Adw.StyleManager.get_default()
+ supports = manager.get_system_supports_color_schemes()
+
+ self.color_scheme_button.set_visible(not supports)
+
+ if supports:
+ manager.set_color_scheme(Adw.ColorScheme.DEFAULT)
+
+ @Gtk.Template.Callback()
+ def notify_visible_child_cb(self, *args):
+ child = self.stack.get_visible_child()
+ # When application starts up there are no buffers
+ if child is None:
+ return
+ page = self.stack.get_page(child)
+ self.content_page.set_title(page.get_title())
+ self.split_view.set_show_content(True)
+
+ @Gtk.Template.Callback()
+ def get_color_scheme_icon_name(self, user_data, dark):
+ return "light-mode-symbolic" if dark else "dark-mode-symbolic"
+
+ @Gtk.Template.Callback()
+ def color_scheme_button_clicked_cb(self, *args):
+ raise NotImplementedError
+
diff --git a/src/window.ui b/src/window.ui
new file mode 100644
index 0000000..3072280
--- /dev/null
+++ b/src/window.ui
@@ -0,0 +1,87 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<interface>
+ <requires lib="gtk" version="4.0"/>
+ <requires lib="libadwaita" version="1.0"/>
+ <menu id="primary_menu">
+ <section>
+ <item>
+ <attribute name="label" translatable="yes">_Preferences</attribute>
+ <attribute name="action">app.preferences</attribute>
+ </item>
+ <item>
+ <attribute name="label" translatable="yes">_About</attribute>
+ <attribute name="action">app.about</attribute>
+ </item>
+ </section>
+ </menu>
+ <template class="WeegtkWindow" parent="AdwApplicationWindow">
+ <property name="title" translatable="yes">Weegtk</property>
+ <property name="default_width">800</property>
+ <property name="default_height">576</property>
+ <child>
+ <object class="AdwBreakpoint">
+ <condition>max-width: 500sp</condition>
+ <setter object="split_view" property="collapsed">True</setter>
+ </object>
+ </child>
+ <property name="content">
+ <object class="AdwToastOverlay" id="toast_overlay">
+ <property name="child">
+ <object class="AdwNavigationSplitView" id="split_view">
+ <property name="min-sidebar-width">200</property>
+ <property name="sidebar">
+ <object class="AdwNavigationPage">
+ <property name="title" bind-source="WeegtkWindow" bind-property="title" bind-flags="sync-create"/>
+ <property name="child">
+ <object class="AdwToolbarView">
+ <child type="top">
+ <object class="AdwHeaderBar">
+ <child type="start">
+ <object class="GtkButton" id="color_scheme_button">
+ <binding name="icon-name">
+ <closure type="gchararray" function="get_color_scheme_icon_name">
+ <lookup name="dark" type="AdwStyleManager">
+ <lookup name="style-manager" type="AdwApplication">
+ <lookup name="application">WeegtkWindow</lookup>
+ </lookup>
+ </lookup>
+ </closure>
+ </binding>
+ <signal name="clicked" handler="color_scheme_button_clicked_cb" swapped="no"/>
+ </object>
+ </child>
+ <child type="end">
+ <object class="GtkMenuButton">
+ <property name="tooltip-text" translatable="yes">Main Menu</property>
+ <property name="menu-model">primary_menu</property>
+ <property name="icon-name">open-menu-symbolic</property>
+ <property name="primary">True</property>
+ </object>
+ </child>
+ </object>
+ </child>
+ <property name="content">
+ <object class="GtkStackSidebar">
+ <property name="stack">stack</property>
+ </object>
+ </property>
+ </object>
+ </property>
+ </object>
+ </property>
+ <property name="content">
+ <object class="AdwNavigationPage" id="content_page">
+ <property name="child">
+ <object class="GtkStack" id="stack">
+ <property name="vhomogeneous">False</property>
+ <signal name="notify::visible-child" handler="notify_visible_child_cb" after="yes" swapped="no"/>
+ </object>
+ </property>
+ </object>
+ </property>
+ </object>
+ </property>
+ </object>
+ </property>
+ </template>
+</interface>
--
cgit v1.2.3