136 lines
5.7 KiB
Python
136 lines
5.7 KiB
Python
import gi
|
|
gi.require_version('Gtk', '4.0')
|
|
gi.require_version('Adw', '1')
|
|
import gc
|
|
import time
|
|
from gi.repository import Gtk, Adw, GLib, Notify, Gdk
|
|
import goldwarden
|
|
import clipboard
|
|
from threading import Thread
|
|
import sys
|
|
import os
|
|
import totp
|
|
Notify.init("Goldwarden")
|
|
|
|
class MyApp(Adw.Application):
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.connect('activate', self.on_activate)
|
|
|
|
def on_activate(self, app):
|
|
self.autofill_window = MainWindow(application=app)
|
|
self.autofill_window.logins = []
|
|
self.autofill_window.present()
|
|
logins = goldwarden.get_vault_logins()
|
|
if logins == None:
|
|
os._exit(0)
|
|
return
|
|
app.autofill_window.logins = logins
|
|
|
|
class MainWindow(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.box = Gtk.Box()
|
|
self.box.set_orientation(Gtk.Orientation.VERTICAL)
|
|
self.stack.add_named(self.box, "box")
|
|
|
|
self.text_view = Adw.EntryRow()
|
|
self.text_view.set_title("Search")
|
|
# on type func
|
|
def on_type(entry):
|
|
if len(entry.get_text()) > 1:
|
|
self.history_list.show()
|
|
else:
|
|
self.history_list.hide()
|
|
|
|
while self.history_list.get_first_child() != None:
|
|
self.history_list.remove(self.history_list.get_first_child())
|
|
|
|
self.filtered_logins = list(filter(lambda i: entry.get_text().lower() in i["name"].lower(), self.logins))
|
|
if len( self.filtered_logins) > 10:
|
|
self.filtered_logins = self.filtered_logins[0:10]
|
|
self.starts_with_logins = list(filter(lambda i: i["name"].lower().startswith(entry.get_text().lower()), self.logins))
|
|
self.other_logins = list(filter(lambda i: i not in self.starts_with_logins , self.filtered_logins))
|
|
self.filtered_logins = None
|
|
|
|
for i in self.starts_with_logins + self.other_logins :
|
|
action_row = Adw.ActionRow()
|
|
action_row.set_title(i["name"])
|
|
action_row.set_subtitle(i["username"])
|
|
action_row.set_icon_name("dialog-password")
|
|
action_row.set_activatable(True)
|
|
action_row.password = i["password"]
|
|
action_row.username = i["username"]
|
|
action_row.uuid = i["uuid"]
|
|
action_row.uri = i["uri"]
|
|
action_row.totp = i["totp"]
|
|
self.history_list.append(action_row)
|
|
self.starts_with_logins = None
|
|
self.other_logins = None
|
|
self.text_view.connect("changed", lambda entry: on_type(entry))
|
|
self.box.append(self.text_view)
|
|
|
|
self.history_list = Gtk.ListBox()
|
|
# margin'
|
|
self.history_list.set_margin_start(10)
|
|
self.history_list.set_margin_end(10)
|
|
self.history_list.set_margin_top(10)
|
|
self.history_list.set_margin_bottom(10)
|
|
self.history_list.hide()
|
|
|
|
keycont = Gtk.EventControllerKey()
|
|
def handle_keypress(cotroller, keyval, keycode, state, user_data):
|
|
if keycode == 36:
|
|
print("enter")
|
|
self.hide()
|
|
def do_autotype(username, password):
|
|
time.sleep(0.5)
|
|
goldwarden.autotype(username, password)
|
|
os._exit(0)
|
|
autotypeThread = Thread(target=do_autotype, args=(self.history_list.get_selected_row().username, self.history_list.get_selected_row().password,))
|
|
autotypeThread.start()
|
|
print(self.history_list.get_selected_row().get_title())
|
|
if keyval == 112:
|
|
print("copy password")
|
|
clipboard.write(self.history_list.get_selected_row().password)
|
|
Notify.Notification.new("Goldwarden", "Password Copied", "dialog-information").show()
|
|
elif keyval == 117:
|
|
print("copy username")
|
|
clipboard.write(self.history_list.get_selected_row().username)
|
|
notification=Notify.Notification.new("Goldwarden", "Username Copied", "dialog-information")
|
|
notification.set_timeout(5)
|
|
notification.show()
|
|
elif keyval == 118:
|
|
print("open web vault")
|
|
environment = goldwarden.get_environment()
|
|
if environment == None:
|
|
return
|
|
item_uri = environment["vault"] + "#/vault?itemId=" + self.history_list.get_selected_row().uuid
|
|
Gtk.show_uri(None, item_uri, Gdk.CURRENT_TIME)
|
|
elif keyval == 108:
|
|
print("launch")
|
|
print(self.history_list.get_selected_row().uri)
|
|
Gtk.show_uri(None, self.history_list.get_selected_row().uri, Gdk.CURRENT_TIME)
|
|
elif keyval == 116:
|
|
print("copy totp")
|
|
totp_code = totp.totp(self.history_list.get_selected_row().totp)
|
|
clipboard.write(totp_code)
|
|
notification=Notify.Notification.new("Goldwarden", "Totp Copied", "dialog-information")
|
|
notification.set_timeout(5)
|
|
notification.show()
|
|
|
|
keycont.connect('key-pressed', handle_keypress, self)
|
|
self.add_controller(keycont)
|
|
|
|
self.history_list.get_style_context().add_class("boxed-list")
|
|
self.box.append(self.history_list)
|
|
self.set_default_size(700, 700)
|
|
self.set_title("Goldwarden")
|
|
|
|
app = MyApp(application_id="com.quexten.Goldwarden.autofill-menu")
|
|
app.run(sys.argv) |