Remote-Desktop-App/structures/gui.py

import tkinter as tk
from PIL import Image, ImageTk
from pyperclip import copy as copy_to_clipboard
from .protocol import is_server_listening


class GUI:
	def __init__(self, app):
		self.app = app
		self.screen_dimensions = [0, 0, 0, 0]
		self.focused = False

		self.geometry = "800x600"

		self.root = tk.Tk()
		self.root.title("Remote Desktop App | Muh ALif Basri (1217 280 225)")
		self.root.geometry(self.geometry)

		self.main_frame = tk.Frame(self.root)
		self.main_frame.pack(fill='both', expand=True)

		self.label3=tk.Label(self.main_frame,text="Pilih Server Atau Client",relief="solid",font=("arial",16,"bold"))
		self.label3.pack(pady=20)

		self.btn_server = tk.Button(self.main_frame, text="Server",fg='blue',bg='white',relief="ridge",command=self.show_server_page)
		self.btn_server.pack(pady=20)

		self.btn_client = tk.Button(self.main_frame, text="Client",fg='blue',bg='white',relief="ridge",command=self.show_client_page)
		self.btn_client.pack(pady=20)

		self.client_frame = tk.Frame(self.root)
		self.link_entry = tk.Entry(self.client_frame)
		self.link_entry.pack(pady=20, padx=100, fill='x', expand=True)

		self.connect_button = tk.Button(self.client_frame, text="Connect", command=self.is_server_alive)
		self.connect_button.pack(pady=20)


		# self.root.protocol("WM_DELETE_WINDOW", self.on_window_close)

	def init_server_gui(self):
		self.server_root = tk.Tk()
		self.server_root.title("Remote Desktop App | Server")
		self.server_root.geometry(self.geometry)

		self.server_frame = tk.Frame(self.server_root)

		self.label4=tk.Label(self.server_frame,text="Share Link",fg='blue',bg='white',font=("arial",16,"bold"))
		self.label4.pack(pady=100)

		self.password_entry = tk.Entry(self.server_frame)
		self.password_entry.pack(pady=20, padx=50, fill='x', expand=True)

		self.copy_link_button = tk.Button(self.server_frame, text="Copy Link", command=self.copy_link)
		self.copy_link_button.pack(pady=20)
		self.server_root.protocol("WM_DELETE_WINDOW", self.on_window_close)

	def init_client_gui(self):
		self.client_root = tk.Tk()
		self.client_root.title("Remote Desktop App | Client")
		self.client_root.geometry(self.geometry)

		self.connected_client_page = tk.Frame(self.client_root)
		self.connected_client_page.bind('<Configure>', self.adjust_image)  # Bind resize event

		self.tools_frame = tk.Frame(self.connected_client_page)
		self.tools_frame.pack(fill='x')

		self.disconnect_button = tk.Button(self.tools_frame, text="Disconnect", command=self.disconnect)
		self.disconnect_button.pack(side='left', padx=10)

		self.file_transfer_button = tk.Button(self.tools_frame, text="File Transfer", command=self.app.file_handler.choose_file)
		self.file_transfer_button.pack(side='left', padx=10)

		# self.chat_button = tk.Button(self.tools_frame, text="Chat")
		# self.chat_button.pack(side='left', padx=10)

		self.original_image = Image.open("city.jpg")

		self.screen_area = tk.Label(self.connected_client_page)
		self.screen_area.pack(fill='both', expand=True)
		self.client_root.protocol("WM_DELETE_WINDOW", self.on_window_close)

	def start(self):
		self.root.mainloop()

	def adjust_image(self, event=None):
		# Calculating new size maintaining 16:9 aspect ratio
		new_width = self.connected_client_page.winfo_width()
		if new_width < 5:
			new_width = 800
		new_height = int(new_width * 9 / 16)

		# Resize the image using Pillow
		resized_image = self.original_image.resize((new_width, new_height), Image.Resampling.LANCZOS)
		photo_image = ImageTk.PhotoImage(resized_image)

		# Update the label image
		self.screen_area.config(image=photo_image)
		self.screen_area.image = photo_image  # Keep a reference!
		self.set_screen_dimensions(new_height)
		self.focused = self.client_root == self.client_root.focus_get()

	def set_screen_dimensions(self, img_height):
		geometry_string = self.screen_area.winfo_geometry()
		offset_list = geometry_string.split("+")
		if len(offset_list) != 3:
			return None

		rootx = self.client_root.winfo_rootx()
		rooty = self.client_root.winfo_rooty()

		[width, height] = [int(num) for num in offset_list[0].split("x")]
		xmin = int(offset_list[1]) + rootx
		ymin = int(offset_list[2]) + rooty
		xmax = xmin + width
		ymax = ymin + height

		ymin += (height - img_height) / 2
		ymax -= (height - img_height) / 2
		
		self.screen_dimensions = [xmin, ymin, xmax, ymax]

	def disconnect(self):
		self.client_root.destroy()
		self.on_window_close()

	def show_server_page(self):
		self.app.setMode("server")
		self.main_frame.pack_forget()
		self.root.destroy()
		self.init_server_gui()
		self.app.startGUI(self)
		self.server_frame.pack(fill='both', expand=True)

		self.app.generate_pass_code()
		self.password_entry.delete(0, tk.END)
		self.password_entry.insert(0, self.app.get_link())

		self.app.start()

	def show_client_page(self):
		self.main_frame.pack_forget()
		self.client_frame.pack(fill='both', expand=True)

	def copy_link(self):
		copy_to_clipboard(self.app.get_link())
		self.change_button_text_temp(self.server_root, self.copy_link_button, "Copied to Clipboard")

	def is_server_alive(self):
		link = self.link_entry.get()

		# validate link
		valid = self.app.auth_handler.validate_link(link)

		if valid:
			[ip, pass_code] = link.split("?p=")
			if self.app.debug:
				print(f"DEBUG: Attempting to connect to {link}")
			is_alive = is_server_listening(ip, self.app.port, pass_code)
			if is_alive:
				self.app.host = ip
				self.connect_to_server()
			else:
				self.change_button_text_temp(self.root, self.connect_button, "Server not Active")
		else:
			self.change_button_text_temp(self.root, self.connect_button, "Invalid Link")

	def connect_to_server(self):
		self.app.setMode("client")
		self.client_frame.pack_forget()
		self.root.destroy()
		self.init_client_gui()
		self.app.startGUI(self)
		self.connected_client_page.pack(fill='both', expand=True)
		self.adjust_image()
		self.app.start()

	def on_window_close(self):
		self.app.stop()

	def change_button_text_temp(self, root, button, new_text):
		old_text = button.config()["text"][4]
		button.config(text=new_text)
		root.after(1500, lambda: button.config(text=old_text))