feat: LockScreen

lock.py

@@ -0,0 +1,218 @@
+import gi, pam, fabric
+
+gi.require_version("GtkSessionLock", "0.1")
+from gi.repository import Gdk, GtkSessionLock
+
+from fabric.widgets.window import Window
+
+# from widgets.wayland import WaylandWindow as Window
+import os
+from fabric.widgets.entry import Entry
+from fabric.widgets.datetime import DateTime
+from modules.panel.components.indicators import (
+    BatteryIndicator,
+    BluetoothIndicator,
+    NetworkIndicator,
+)
+from fabric.widgets.box import Box
+from fabric.widgets.label import Label
+from fabric.widgets.centerbox import CenterBox
+
+# from fabric.widgets.image import Image
+import getpass
+from fabric import Application
+from gi.repository import GLib
+from widgets.circle_image import CircleImage as Image
+from fabric.utils import get_relative_path
+
+
+class IndicatorBox(Box):
+    def __init__(self, *args, **kwargs):
+        super().__init__(
+            h_align="end",
+            name="indicatorbox",
+            spacing=5,
+            h_expand=True,
+            children=[
+                BatteryIndicator(),
+                BluetoothIndicator(),
+                NetworkIndicator(),
+            ],
+        )
+
+
+class ContentBox(CenterBox):
+    def __init__(self, on_activate, *args, **kwargs):
+        self.password_entry = Entry(
+            placeholder="Enter Password",
+            name="password-entry",
+            h_align="center",
+            v_align="center",
+            visible=False,
+            password=True,
+            # on_activate=on_activate,
+        )
+
+        self.password_entry.set_property("xalign", 0.5)
+        self.username_label = Label(
+            label=f"{getpass.getuser().title()}",
+            name="username",
+            visible=True,
+            h_align="center",
+            v_align="center",
+        )
+        self.unlock_text = Label(
+            label="Touch ID or Enter Password",
+            name="unlock-text",
+        )
+        super().__init__(
+            name="content-box",
+            h_expand=True,
+            orientation="vertical",
+            v_expand=True,
+            start_children=[
+                IndicatorBox(),
+                DateTime(
+                    formatters=["%A,%B %-d"],
+                    interval=10000,
+                    h_expand=False,
+                    v_align="start",
+                    v_expand=False,
+                    name="lock-date",
+                ),
+                DateTime(formatters=["%I:%M"], name="lock-clock"),
+            ],
+            end_children=[
+                Box(
+                    name="profile-box",
+                    h_align="center",
+                    h_expand=True,
+                    v_expand=True,
+                    v_align="end",
+                    children=[
+                        Image(
+                            name="face-icon",
+                            image_file="/home/saumya/.face.icon",
+                            size=64,
+                        ),
+                    ],
+                ),
+                Box(
+                    name="container-box",
+                    orientation="v",
+                    v_align="end",
+                    h_align="center",
+                    h_expand=True,
+                    v_expand=True,
+                    children=[
+                        self.username_label,
+                        self.password_entry,
+                    ],
+                ),
+                Box(
+                    name="unlock-box",
+                    v_align="end",
+                    h_align="center",
+                    h_expand=True,
+                    v_expand=True,
+                    children=[
+                        self.unlock_text,
+                    ],
+                ),
+            ],
+            **kwargs,
+        )
+
+
+class LockScreen(Window):
+    def __init__(self, lock: GtkSessionLock.Lock):
+        self._hide_timeout_id = None  # prevent AttributeError
+        self.lock = lock
+        self.content = ContentBox(self.on_activate)
+        super().__init__(
+            title="lock",
+            visible=True,
+            all_visible=False,
+            name="lockscreen-bg",
+            anchor="center",
+            child=self.content,
+        )
+
+        self.content.password_entry.set_visible(False)
+        self.connect("key-press-event", self._on_keypress)
+
+        bg = os.path.expanduser("~/.current.wall")
+        self.set_style(f"background-image: url('{bg}');")
+
+    def _on_keypress(self, widget, event):
+        keyval = event.keyval
+
+        # ESC pressed → hide entry immediately
+        if keyval == Gdk.KEY_Escape and self.content.password_entry.get_visible():
+            self._hide_entry()
+            return
+
+        # Show entry if hidden
+        if not self.content.password_entry.get_visible():
+            self.content.username_label.set_visible(False)
+            self.content.password_entry.set_visible(True)
+            self.content.password_entry.grab_focus()
+            self._start_hide_timer()
+        else:
+            # Reset timer if already visible
+            self._restart_hide_timer()
+
+    def _start_hide_timer(self):
+        self._stop_hide_timer()  # just in case
+        self._hide_timeout_id = GLib.timeout_add_seconds(500, self._hide_entry)
+        # 10 seconds of inactivity before hiding
+
+    def _restart_hide_timer(self):
+        self._start_hide_timer()
+
+    def _stop_hide_timer(self):
+        if self._hide_timeout_id:
+            GLib.source_remove(self._hide_timeout_id)
+            self._hide_timeout_id = None
+
+    def _hide_entry(self):
+        self._stop_hide_timer()
+        self.content.password_entry.set_visible(False)
+        self.content.username_label.set_visible(True)
+        return False  # stop timeout
+
+    def on_activate(self, entry: Entry, *args):
+        if not pam.authenticate(getpass.getuser(), (entry.get_text() or "").strip()):
+            return
+        self.lock.unlock_and_destroy()
+        self.destroy()
+
+
+def initialize():
+    lock = GtkSessionLock.prepare_lock()
+    lock.lock_lock()
+    lockscreen = LockScreen(lock)
+    lock.new_surface(
+        lockscreen,
+        Gdk.Display.get_default().get_monitor(
+            0
+        ),  # pyright: ignore[reportAttributeAccessIssue, reportOptionalMemberAccess]
+    )
+    lockscreen.show_all()
+
+
+if __name__ == "__main__":
+    # initialize()
+    lockscreen = LockScreen(GtkSessionLock.Lock())
+
+    app = Application("lock", lockscreen)
+
+    def set_css():
+        app.set_stylesheet_from_file(
+            get_relative_path("main.css"),
+        )
+
+    app.set_css = set_css  # pyright: ignore[reportAttributeAccessIssue]
+
+    app.set_css()  # pyright: ignore[reportAttributeAccessIssue]
+    app.run()