added burger menu, About section and Preferences menu item, which is WIP

This commit is contained in:
2024-11-22 13:42:43 +01:00
parent 9003e1d099
commit 90036da40e
42 changed files with 87 additions and 27 deletions

View File

@@ -4,6 +4,10 @@
Created on Mon Nov 18 12:04:53 2024
@author: angoosh
references:
https://pygobject.gnome.org/tutorials/gtk4/introduction.html
https://docs.gtk.org/gtk3/index.html#classes
"""
import sys
import os
@@ -18,6 +22,8 @@ from gi.repository import Gtk, Adw, Gdk, Gio
APPID = "com.angoosh.RDPConnect"
HOMEDIR = os.path.expanduser('~')
VERSION = "1.0.5rc1"
conn_info = {}
settings = {}
fernet = ""
@@ -34,7 +40,7 @@ def load_keys():
with open(HOMEDIR+"/.config/rdpconnect/.key", "w") as keyfile:
keyfile.write(cryptoKey.decode("utf-8"))
print("Encription key generated")
fernet = Fernet(cryptoKey)
def load_config():
@@ -44,23 +50,23 @@ def load_config():
with open(HOMEDIR+"/.config/rdpconnect/connection.json", "r") as connection_file:
for line in connection_file:
loaded_json += line
conn_info = json.loads(loaded_json)
conn_info["passwd"] = fernet.decrypt(str.encode(conn_info["passwd"])).decode()
except:
print("[WARN] FILE: "+HOMEDIR+"/.config/rdpconnect/connection.json doesn't exist")
loaded_json = ""
try:
with open(HOMEDIR+"/.config/rdpconnect/settings.json", "r") as settings_file:
for line in settings_file:
loaded_json += line
settings = json.loads(loaded_json)
except:
print("[WARN] FILE: "+HOMEDIR+"/.config/rdpconnect/settings.json doesn't exist")
class MyApp(Adw.Application):
def __init__(self, **kwargs):
@@ -70,62 +76,98 @@ class MyApp(Adw.Application):
def on_activate(self, app):
builder = Gtk.Builder()
builder.add_from_file("/app/"+APPID+".ui")
buildermenu = Gtk.Builder()
buildermenu.add_from_file("/app/menu.ui")
menu_model = buildermenu.get_object('app-menu')
css_provider = Gtk.CssProvider()
if Adw.StyleManager().get_default().get_dark():
css_provider.load_from_file(Gio.File.new_for_path("/app/style-dark.css"))
else:
css_provider.load_from_file(Gio.File.new_for_path("/app/style-light.css"))
Gtk.StyleContext.add_provider_for_display(Gdk.Display.get_default(), css_provider, Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION)
connect = builder.get_object("connect")
connect.connect("clicked", self.connectRDP)
self.ip = builder.get_object("ip")
self.user = builder.get_object("user")
self.passwd = builder.get_object("pass")
self.save_conf = builder.get_object("save_conf")
try:
if settings["save_conn"]:
self.ip.get_buffer().set_text(str(conn_info["ip"]), len(conn_info["ip"]))
self.user.get_buffer().set_text(str(conn_info["user"]), len(conn_info["user"]))
self.passwd.get_buffer().set_text(str(conn_info["passwd"]), len(conn_info["passwd"]))
self.save_conf.set_active(True)
except:
pass
#menu config
self.win = builder.get_object("main_window")
header_bar = Gtk.HeaderBar()
self.win.set_titlebar(header_bar)
menubutton = Gtk.MenuButton(menu_model=menu_model, icon_name='open-menu-symbolic')
header_bar.pack_end(menubutton)
self.win.set_application(self)
self.win.present()
def do_startup(self):
Gtk.Application.do_startup(self)
action = Gio.SimpleAction(name='preferences')
action.connect('activate', self.on_preferences)
self.add_action(action)
action = Gio.SimpleAction(name='about')
action.connect('activate', self.on_about)
self.add_action(action)
def on_about(self, action, param):
about_dialog = Gtk.AboutDialog(transient_for=self.win, modal=True)
about_dialog.set_copyright("Antonin Kaplan")
about_dialog.set_program_name("RDP Connect")
about_dialog.set_license_type(Gtk.License(16))
about_dialog.set_logo_icon_name("com.angoosh.RDPConnect")
about_dialog.set_version(VERSION)
about_dialog.set_website("https://gitea.farmdash.org/angoosh/Flatpaks")
about_dialog.present()
def on_preferences(self, action, param):
pref_window = Gtk.Window(title="Preferences")
pref_window.show()
print("Preferences")
def saveConnConf(self):
if settings["save_conn"]:
password = conn_info["passwd"]
conn_info["passwd"] = fernet.encrypt(password.encode()).decode("utf-8")
print("Saving connection config to "+HOMEDIR+"/.config/rdpconnect/connection.json")
with open(HOMEDIR+"/.config/rdpconnect/connection.json", "w") as connection_file:
js = json.dumps(conn_info, sort_keys=True, indent=4, separators=(',', ': '))
connection_file.write(js)
conn_info["passwd"] = password
else:
with open(HOMEDIR+"/.config/rdpconnect/connection.json", "w") as connection_file:
connection_file.write("")
print("Saving settings config to "+HOMEDIR+"/.config/rdpconnect/settings.json")
with open(HOMEDIR+"/.config/rdpconnect/settings.json", "w") as settings_file:
js = json.dumps(settings, sort_keys=True, indent=4, separators=(',', ': '))
settings_file.write(js)
def connectRDP(self, button):
conn_info["ip"] = self.ip.get_buffer().get_text()
conn_info["user"] = self.user.get_buffer().get_text()
conn_info["passwd"] = self.passwd.get_buffer().get_text()
if self.save_conf.get_active():
settings["save_conn"] = True
else:
@@ -134,15 +176,14 @@ class MyApp(Adw.Application):
settings["extra_params"] = []
if not "rdp_bin" in settings:
settings["rdp_bin"] = "sdl-freerdp"
self.saveConnConf()
try:
subprocess.Popen([settings["rdp_bin"], "/v:"+str(conn_info["ip"]), "/u:"+str(conn_info["user"]), "/p:"+str(conn_info["passwd"])]+settings["extra_params"])
except:
subprocess.Popen(["sdl-freerdp", "/v:"+str(conn_info["ip"]), "/u:"+str(conn_info["user"]), "/p:"+str(conn_info["passwd"])]+settings["extra_params"])
#subprocess.run(["xfreerdp", "/v:"+str(ip), "/u:"+str(user), "/p:"+str(passwd)])
if not os.path.isdir(HOMEDIR+"/.config/rdpconnect"):
os.makedirs(HOMEDIR+"/.config/rdpconnect")