""" tray_app.py — System tray icon and balloon/toast notifications for OneSign. Uses: - pystray (cross-platform tray icon with menu) - win10toast (Windows balloon notifications) with ctypes fallback - Pillow (PIL) for tray icon rendering The tray icon shows: - Green/grey status dot on the icon - Right-click menu: Status, Check for Updates, Enroll card, About, Exit - Balloon notification on card tap events """ import logging import os import subprocess import sys import threading import time try: import tkinter as tk from tkinter import messagebox except Exception: # pragma: no cover - depends on host runtime tk = None messagebox = None import pystray from PIL import Image, ImageDraw logger = logging.getLogger(__name__) # ── Icon generation ────────────────────────────────────────────────────────── _ICON_SIZE = 64 _COLOR_GREEN = (10, 200, 140, 255) _COLOR_GREY = (140, 140, 140, 255) _COLOR_ORANGE = (247, 103, 7, 255) _COLOR_BLUE = (26, 111, 196, 255) _COLOR_WHITE = (255, 255, 255, 255) _COLOR_WHITE_80 = (255, 255, 255, 80) def _make_icon_image(status: str = "idle") -> Image.Image: """ Draw a simple badge icon with a status dot. status: 'idle' | 'connected' | 'locked' | 'auth' """ sz = _ICON_SIZE img = Image.new("RGBA", (sz, sz), (0, 0, 0, 0)) draw = ImageDraw.Draw(img) # Badge rectangle draw.rounded_rectangle([6, 8, sz - 6, sz - 8], radius=8, fill=_COLOR_BLUE) # Magnetic stripe draw.rectangle([10, 14, sz - 10, 20], fill=_COLOR_WHITE_80) # Person circle cx, cy = sz // 2, sz // 2 + 4 draw.ellipse([cx - 10, cy - 10, cx + 10, cy + 10], fill=_COLOR_WHITE) draw.ellipse([cx - 5, cy - 5, cx + 5, cy + 5], fill=_COLOR_BLUE) # Status dot (bottom-right) dot_colors = { "idle": _COLOR_GREY, "connected": _COLOR_GREEN, "locked": _COLOR_ORANGE, "auth": _COLOR_GREEN, } dot_color = dot_colors.get(status, _COLOR_GREY) d = 14 draw.ellipse([sz - d - 2, sz - d - 2, sz - 2, sz - 2], fill=dot_color) return img # ── Balloon notifications ──────────────────────────────────────────────────── def _balloon_ctypes(title: str, message: str, duration: int = 5): """ Show a Windows balloon / toast notification using Shell_NotifyIconW. Falls back to win10toast if available. """ try: from win10toast import ToastNotifier _toaster = ToastNotifier() _toaster.show_toast( title, message, icon_path=None, duration=duration, threaded=True, ) return except ImportError: pass # ctypes Shell_NotifyIcon fallback (basic tray balloon) try: import ctypes import ctypes.wintypes as wt NIM_MODIFY = 0x00000001 NIF_INFO = 0x00000010 NIIF_INFO = 0x00000001 class NOTIFYICONDATA(ctypes.Structure): _fields_ = [ ("cbSize", ctypes.c_ulong), ("hWnd", wt.HWND), ("uID", ctypes.c_uint), ("uFlags", ctypes.c_uint), ("uCallbackMessage", ctypes.c_uint), ("hIcon", wt.HICON), ("szTip", ctypes.c_wchar * 128), ("dwState", ctypes.c_ulong), ("dwStateMask", ctypes.c_ulong), ("szInfo", ctypes.c_wchar * 256), ("uTimeout", ctypes.c_uint), ("szInfoTitle", ctypes.c_wchar * 64), ("dwInfoFlags", ctypes.c_ulong), ] nid = NOTIFYICONDATA() nid.cbSize = ctypes.sizeof(NOTIFYICONDATA) nid.uFlags = NIF_INFO nid.szInfo = message[:255] nid.szInfoTitle = title[:63] nid.uTimeout = duration * 1000 nid.dwInfoFlags = NIIF_INFO ctypes.windll.shell32.Shell_NotifyIconW(NIM_MODIFY, ctypes.byref(nid)) except Exception as exc: logger.debug("Balloon notification failed: %s", exc) # ── Tray application ───────────────────────────────────────────────────────── class TrayApp: """ Manages the system tray icon lifecycle. Call start() in a daemon thread; stop() to exit. """ def __init__(self, agent_ref=None): self._agent = agent_ref self._icon = None self._status = "idle" # idle | connected | locked | auth self._status_text = "OneSign Agent — Idle" self._lock = threading.Lock() # ── Public API ─────────────────────────────────────────────────────── def set_status(self, status: str, text: str = None): with self._lock: self._status = status if text: self._status_text = text if self._icon: self._icon.icon = _make_icon_image(status) self._icon.title = self._status_text def notify(self, title: str, message: str, duration: int = 5): """Show a balloon notification (non-blocking).""" t = threading.Thread( target=_balloon_ctypes, args=(title, message, duration), daemon=True, ) t.start() def start(self): """Build and run the tray icon (blocks until stop() is called).""" menu = pystray.Menu( pystray.MenuItem(lambda _: self._status_text, None, enabled=False), pystray.Menu.SEPARATOR, pystray.MenuItem("Sync with Server", self._on_sync), pystray.MenuItem("Check for Updates", self._on_check_updates), pystray.MenuItem("Show Connection Status", self._on_status), pystray.MenuItem("Open Agent Log", self._on_open_log), pystray.MenuItem("Clear Agent Log…", self._on_clear_log), pystray.MenuItem("Open Admin Panel", self._on_open_admin), pystray.MenuItem("Enroll Card…", self._on_enroll), pystray.MenuItem("Reader Control…", self._on_reader_control), pystray.Menu.SEPARATOR, pystray.MenuItem("Exit Agent", self._on_exit), ) self._icon = pystray.Icon( name="OneSign", icon=_make_icon_image("idle"), title="OneSign Agent", menu=menu, ) self._icon.run() def stop(self): if self._icon: self._icon.stop() # ── Menu handlers ──────────────────────────────────────────────────── def _on_exit(self, icon, item): icon.stop() if self._agent: self._agent.stop() os._exit(0) def _on_enroll(self, icon, item): self.notify( "OneSign Enrollment", "Start enrollment in the admin panel, then tap the card on this reader.", duration=8, ) if self._agent: self._agent.start_enrollment_mode() def _on_sync(self, icon, item): if self._agent: self._agent.sync_with_server() def _on_status(self, icon, item): if self._agent: self.notify("OneSign Status", self._agent.get_status_summary(), duration=6) def _on_check_updates(self, icon, item): if not self._agent: self.notify("OneSign Updates", "Agent context is unavailable.", duration=5) return threading.Thread( target=self._agent.check_for_updates, kwargs={"manual": True, "install": True}, daemon=True, ).start() def _on_open_log(self, icon, item): if not self._agent: return log_path = self._agent.get_log_file_path() if not log_path or not os.path.isfile(log_path): self.notify("OneSign Log", "No log file found yet.", duration=5) return try: os.startfile(log_path) # type: ignore[attr-defined] except Exception: try: subprocess.Popen(["notepad.exe", log_path], close_fds=True) except Exception as exc: self.notify("OneSign Log Error", f"Could not open log file: {exc}", duration=6) def _on_clear_log(self, icon, item): if not self._agent: self.notify("OneSign Log", "Agent context is unavailable.", duration=5) return ok, msg = self._agent.clear_log() if ok: self.notify("OneSign Log", "Log file cleared successfully.", duration=4) else: self.notify("OneSign Log — Error", msg, duration=6) def _on_open_admin(self, icon, item): import webbrowser if self._agent: url = self._agent.get_admin_url() else: url = "http://localhost/admin" webbrowser.open(url) def _on_reader_control(self, icon, item): if not self._agent: self.notify("Reader Control", "Agent context is unavailable.", duration=5) return if tk is None or messagebox is None: self.notify("Reader Control", "Tkinter is not available in this build.", duration=6) return threading.Thread(target=self._open_reader_control_popup, daemon=True).start() def _open_reader_control_popup(self): root = tk.Tk() root.title("OneSign Reader Control") root.geometry("560x420") root.resizable(False, False) status_var = tk.StringVar() readers_list = tk.Listbox(root, height=12, width=80) readers_list.pack(padx=10, pady=(10, 6), fill="x") tk.Label(root, textvariable=status_var, anchor="w", justify="left").pack( padx=10, pady=(0, 8), fill="x" ) footer_var = tk.StringVar(value="Select a reader and click Apply Selection.") tk.Label(root, textvariable=footer_var, anchor="w").pack(padx=10, pady=(0, 8), fill="x") readers_cache: list[dict] = [] def refresh(): readers_list.delete(0, tk.END) readers, err = self._agent.enumerate_readers() readers_cache.clear() readers_cache.extend(readers) if err: footer_var.set(f"Reader enumerate failed: {err}") elif not readers: footer_var.set("No readers found. Check USB/cable/drivers and DLL path.") else: footer_var.set(f"Found {len(readers)} reader(s).") selected = self._agent.get_reader_control_status().get("selected_index", -1) for idx, r in enumerate(readers): text = f"[{r.get('index', idx)}] {r.get('part_number') or 'Unknown'} {r.get('vid_pid_vendor') or ''} LUID={r.get('luid', 0)}" readers_list.insert(tk.END, text) if r.get("index") == selected: readers_list.select_set(idx) stat = self._agent.get_reader_control_status() status_var.set( f"Connected: {'Yes' if stat.get('connected') else 'No'} | " f"Selected index: {stat.get('selected_index')} | " f"DLL: {stat.get('dll_path')}" ) def apply_selection(): selection = readers_list.curselection() if not selection: messagebox.showinfo("Reader Control", "Select a reader first.") return row = readers_cache[selection[0]] ok, msg = self._agent.select_reader_index(int(row.get("index", -1))) if not ok: messagebox.showerror("Reader Control", msg) return footer_var.set(msg) refresh() def reconnect(): _, msg = self._agent.reconnect_reader() footer_var.set(msg) root.after(1000, refresh) def test_read(): footer_var.set("Waiting for badge tap (5 seconds)…") root.update_idletasks() card = self._agent.read_card_once(timeout_seconds=5.0) if card: messagebox.showinfo("Reader Test", f"Card detected: {card}") footer_var.set("Card read succeeded.") else: messagebox.showwarning("Reader Test", "No card detected within timeout.") footer_var.set("No card read.") button_bar = tk.Frame(root) button_bar.pack(padx=10, pady=8, fill="x") tk.Button(button_bar, text="Refresh", width=12, command=refresh).pack(side=tk.LEFT, padx=4) tk.Button(button_bar, text="Apply Selection", width=14, command=apply_selection).pack(side=tk.LEFT, padx=4) tk.Button(button_bar, text="Reconnect", width=12, command=reconnect).pack(side=tk.LEFT, padx=4) tk.Button(button_bar, text="Test Read", width=12, command=test_read).pack(side=tk.LEFT, padx=4) tk.Button(button_bar, text="Close", width=12, command=root.destroy).pack(side=tk.RIGHT, padx=4) refresh() root.mainloop()