goldwarden/ui/settings.py

#!/usr/bin/python
import sys
import gi
gi.require_version('Gtk', '4.0')
gi.require_version('Adw', '1')
import gc

from gi.repository import Gtk, Adw, GLib, Gdk
import goldwarden
from threading import Thread
import subprocess
import components

class SettingsWinvdow(Gtk.ApplicationWindow):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.stack = Gtk.Stack()
        self.stack.set_transition_type(Gtk.StackTransitionType.SLIDE_LEFT_RIGHT)
        self.set_child(self.stack)

        self.preferences_page = Adw.PreferencesPage()
        self.preferences_page.set_title("General")
        self.stack.add_named(self.preferences_page, "preferences_page")

        self.preferences_group = Adw.PreferencesGroup()
        self.preferences_group.set_title("Services")
        self.preferences_page.add(self.preferences_group)

        self.ssh_row = Adw.ActionRow()
        self.ssh_row.set_title("SSH Daemon")
        self.ssh_row.set_subtitle("Listening at ~/.goldwarden-ssh-agent.sock")
        self.preferences_group.add(self.ssh_row)

        self.icon = components.status_icon_ok("emblem-default")
        self.ssh_row.add_prefix(self.icon)

        self.login_with_device = Adw.ActionRow()
        self.login_with_device.set_title("Login with device")
        self.login_with_device.set_subtitle("Waiting for requests...")
        self.preferences_group.add(self.login_with_device)

        self.status_row = Adw.ActionRow()
        self.status_row.set_title("DBUS Service")
        self.status_row.set_subtitle("Listening")
        self.preferences_group.add(self.status_row)

        self.shortcut_preferences_group = Adw.PreferencesGroup()
        self.shortcut_preferences_group.set_title("Shortcuts")
        self.preferences_page.add(self.shortcut_preferences_group)

        self.autofill_row = Adw.ActionRow()
        self.autofill_row.set_title("Autofill Shortcut")
        self.autofill_row.set_subtitle("Unavailable, please set up a shortcut in your desktop environment (README)")
        self.shortcut_preferences_group.add(self.autofill_row)

        self.autofill_icon = components.StatusIcon()
        self.autofill_icon.set_icon("dialog-warning", "warning")
        self.autofill_row.add_prefix(self.autofill_icon)

        self.copy_username_shortcut_row = Adw.ActionRow()
        self.copy_username_shortcut_row.set_title("Copy Username Shortcut")
        self.copy_username_shortcut_row.set_subtitle("U")
        self.shortcut_preferences_group.add(self.copy_username_shortcut_row)

        self.copy_password_shortcut_row = Adw.ActionRow()
        self.copy_password_shortcut_row.set_title("Copy Password Shortcut")
        self.copy_password_shortcut_row.set_subtitle("P")
        self.shortcut_preferences_group.add(self.copy_password_shortcut_row)

        self.vault_status_preferences_group = Adw.PreferencesGroup()
        self.vault_status_preferences_group.set_title("Vault Status")
        self.preferences_page.add(self.vault_status_preferences_group)
        
        self.status_row = Adw.ActionRow()
        self.status_row.set_title("Vault Status")
        self.status_row.set_subtitle("Locked")
        self.vault_status_preferences_group.add(self.status_row)

        self.vault_status_icon = components.StatusIcon()
        self.vault_status_icon.set_icon("dialog-error", "error")
        self.status_row.add_prefix(self.vault_status_icon)

        self.last_sync_row = Adw.ActionRow()
        self.last_sync_row.set_title("Last Sync")
        self.last_sync_row.set_subtitle("Never")
        self.last_sync_row.set_icon_name("emblem-synchronizing-symbolic")
        self.vault_status_preferences_group.add(self.last_sync_row)

        self.websocket_connected_row = Adw.ActionRow()
        self.websocket_connected_row.set_title("Websocket Connected")
        self.websocket_connected_row.set_subtitle("False")
        self.vault_status_preferences_group.add(self.websocket_connected_row)

        self.websocket_connected_status_icon = components.StatusIcon()
        self.websocket_connected_status_icon.set_icon("dialog-error", "error")
        self.websocket_connected_row.add_prefix(self.websocket_connected_status_icon)
        
        self.login_row = Adw.ActionRow()
        self.login_row.set_title("Vault Login Entries")
        self.login_row.set_subtitle("0")
        self.login_row.set_icon_name("dialog-password-symbolic")
        self.vault_status_preferences_group.add(self.login_row)

        self.notes_row = Adw.ActionRow()
        self.notes_row.set_title("Vault Notes")
        self.notes_row.set_subtitle("0")
        self.notes_row.set_icon_name("emblem-documents-symbolic")
        self.vault_status_preferences_group.add(self.notes_row)

        self.action_preferences_group = Adw.PreferencesGroup()
        self.action_preferences_group.set_title("Actions")
        self.preferences_page.add(self.action_preferences_group)
        
        self.autotype_button = Gtk.Button()
        self.autotype_button.set_label("Autotype")
        self.autotype_button.set_margin_top(10)
        self.autotype_button.connect("clicked", lambda button: subprocess.Popen(["python3", "/app/bin/autofill.py"], start_new_session=True))
        self.autotype_button.get_style_context().add_class("suggested-action")
        self.action_preferences_group.add(self.autotype_button)

        self.login_button = Gtk.Button()
        self.login_button.set_label("Login")
        self.login_button.connect("clicked", lambda button: show_login())
        self.login_button.set_sensitive(False)
        self.login_button.set_margin_top(10)
        self.login_button.get_style_context().add_class("suggested-action")
        self.action_preferences_group.add(self.login_button)
    
        self.set_pin_button = Gtk.Button()
        self.set_pin_button.set_label("Set Pin")
        def set_pin():
            set_pin_thread = Thread(target=goldwarden.enable_pin)
            set_pin_thread.start()
        self.set_pin_button.connect("clicked", lambda button: set_pin())
        self.set_pin_button.set_margin_top(10)
        self.set_pin_button.set_sensitive(False)
        self.set_pin_button.get_style_context().add_class("suggested-action")
        self.action_preferences_group.add(self.set_pin_button)

        self.unlock_button = Gtk.Button()
        self.unlock_button.set_label("Unlock")
        self.unlock_button.set_margin_top(10)
        def unlock_button_clicked():
            action = goldwarden.unlock if self.unlock_button.get_label() == "Unlock" else goldwarden.lock
            unlock_thread = Thread(target=action)
            unlock_thread.start()
        self.unlock_button.connect("clicked", lambda button: unlock_button_clicked())
        # set disabled
        self.unlock_button.set_sensitive(False)
        self.action_preferences_group.add(self.unlock_button)

        self.logout_button = Gtk.Button()
        self.logout_button.set_label("Logout")
        self.logout_button.set_margin_top(10)
        self.logout_button.connect("clicked", lambda button: goldwarden.purge())
        self.logout_button.get_style_context().add_class("destructive-action")
        self.action_preferences_group.add(self.logout_button)
        
        def update_labels():
            pin_set = goldwarden.is_pin_enabled()
            status = goldwarden.get_vault_status()
            if status != None:
                if pin_set:
                    self.unlock_button.set_sensitive(True)
                else:
                    self.unlock_button.set_sensitive(False)
                logged_in = status["loggedIn"]
                if logged_in and not status["locked"]:
                    self.preferences_group.set_visible(True)
                    self.shortcut_preferences_group.set_visible(True)
                    self.autotype_button.set_visible(True)
                    self.login_row.set_sensitive(True)
                    self.notes_row.set_sensitive(True)
                    self.websocket_connected_row.set_sensitive(True)
                else:
                    self.preferences_group.set_visible(False)
                    self.shortcut_preferences_group.set_visible(False)
                    self.autotype_button.set_visible(False)
                    self.websocket_connected_row.set_sensitive(False)
                    self.login_row.set_sensitive(False)
                    self.notes_row.set_sensitive(False)

                locked = status["locked"]
                self.login_button.set_sensitive(pin_set and not locked)
                self.set_pin_button.set_sensitive(not pin_set or not locked)
                self.autotype_button.set_sensitive(not locked)
                self.status_row.set_subtitle(str("Logged in" if (logged_in and not locked) else "Logged out") if not locked else "Locked")
                if locked or not logged_in:
                    self.vault_status_icon.set_icon("dialog-warning", "warning")
                else:
                    self.vault_status_icon.set_icon("emblem-default", "ok")
                if not logged_in:
                    self.logout_button.set_sensitive(False)
                else:
                    self.logout_button.set_sensitive(True)
                self.login_row.set_subtitle(str(status["loginEntries"]))
                self.notes_row.set_subtitle(str(status["noteEntries"]))
                self.websocket_connected_row.set_subtitle("Connected" if status["websocketConnected"] else "Disconnected")
                if status["websocketConnected"]:
                    self.websocket_connected_status_icon.set_icon("emblem-default", "ok")
                else:
                    self.websocket_connected_status_icon.set_icon("dialog-error", "error")
                self.last_sync_row.set_subtitle(str(status["lastSynced"]))
                if status["lastSynced"].startswith("1970"):
                    self.last_sync_row.set_subtitle("Never")
                self.unlock_button.set_label("Unlock" if locked else "Lock")
            else:
                is_daemon_running = goldwarden.is_daemon_running()
                if not is_daemon_running:
                    self.status_row.set_subtitle("Daemon not running")
                    self.vault_status_icon.set_icon("dialog-error", "error")
            GLib.timeout_add(1000, update_labels)

        GLib.timeout_add(1000, update_labels)
        self.set_default_size(400, 700)
        self.set_title("Goldwarden")


        #add title buttons
        self.title_bar = Gtk.HeaderBar()
        self.set_titlebar(self.title_bar)

class MyApp(Adw.Application):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.connect('activate', self.on_activate)

    def on_activate(self, app):
        self.settings_win = SettingsWinvdow(application=app)
        self.settings_win.present()

def show_login():
    dialog = Gtk.Dialog(title="Goldwarden")
    preference_group = Adw.PreferencesGroup()
    preference_group.set_title("Config")
    preference_group.set_margin_top(10)
    preference_group.set_margin_bottom(10)
    preference_group.set_margin_start(10)
    preference_group.set_margin_end(10)

    dialog.get_content_area().append(preference_group)

    api_url_entry = Adw.EntryRow()
    api_url_entry.set_title("API Url")
    # set value
    api_url_entry.set_text("https://vault.bitwarden.com/api")
    preference_group.add(api_url_entry)

    identity_url_entry = Adw.EntryRow()
    identity_url_entry.set_title("Identity Url")
    identity_url_entry.set_text("https://vault.bitwarden.com/identity")
    preference_group.add(identity_url_entry)

    notification_url_entry = Adw.EntryRow()
    notification_url_entry.set_title("Notification URL")
    notification_url_entry.set_text("https://notifications.bitwarden.com/")
    preference_group.add(notification_url_entry)

    auth_preference_group = Adw.PreferencesGroup()
    auth_preference_group.set_title("Authentication")
    dialog.get_content_area().append(auth_preference_group)

    email_entry = Adw.EntryRow()
    email_entry.set_title("Email")
    email_entry.set_text("")
    auth_preference_group.add(email_entry)

    dialog.add_button("Login", Gtk.ResponseType.OK)
    def on_save(res):
        if res != Gtk.ResponseType.OK:
            return
        goldwarden.set_api_url(api_url_entry.get_text())
        goldwarden.set_identity_url(identity_url_entry.get_text())
        goldwarden.set_notification_url(notification_url_entry.get_text())
        def login():
            res = goldwarden.login_with_password(email_entry.get_text(), "password")
            def handle_res():
                print("handle res", res)
                if res == "ok":
                    dialog.close()
                    print("ok")
                elif res == "badpass":
                    bad_pass_diag = Gtk.MessageDialog(transient_for=dialog, modal=True, message_type=Gtk.MessageType.ERROR, buttons=Gtk.ButtonsType.OK, text="Bad password")
                    bad_pass_diag.connect("response", lambda dialog, response: bad_pass_diag.close())
                    bad_pass_diag.present()
            GLib.idle_add(handle_res)

        login_thread = Thread(target=login)
        login_thread.start()

    #ok response
    dialog.connect("response", lambda dialog, response: on_save(response))
    dialog.set_default_size(400, 200)
    dialog.set_modal(True)
    dialog.present()

isflatpak = os.path.exists("/.flatpak-info")
pathprefix = "/app/bin/" if isflatpak else "./"
css_provider = Gtk.CssProvider()
css_provider.load_from_path(pathprefix+"style.css")
Gtk.StyleContext.add_provider_for_display(
    Gdk.Display.get_default(),
    css_provider,
    Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION
)


app = MyApp(application_id="com.quexten.Goldwarden.settings")
app.run(sys.argv)