support --screenshot=FULL mode

Author: AlexanderVanhee

gradia/main.py

@@ -23,7 +23,7 @@
 from collections.abc import Sequence
 from typing import Optional
 
-from gi.repository import Adw, Gio
+from gi.repository import Adw, Gio, Xdp
 
 from gradia.ui.window import GradientWindow
 from gradia.backend.logger import Logger
@@ -39,7 +39,7 @@ def __init__(self, version: str):
             flags=Gio.ApplicationFlags.HANDLES_COMMAND_LINE | Gio.ApplicationFlags.HANDLES_OPEN
         )
         self.version = version
-        self.screenshot_mode = False
+        self.screenshot_flags: Optional[Xdp.ScreenshotFlags] = None
         self.temp_dirs: list[str] = []
 
         # Connect to shutdown signal for cleanup
@@ -50,7 +50,7 @@ def do_command_line(self, command_line: Gio.ApplicationCommandLine) -> int:
 
         logging.debug(f"Command line arguments: {args}")
 
-        self.screenshot_mode = "--screenshot" in args
+        self.screenshot_flags = self._parse_screenshot_flag(args)
         files_to_open = []
 
         for arg in args:
@@ -74,6 +74,22 @@ def do_command_line(self, command_line: Gio.ApplicationCommandLine) -> int:
 
         return 0
 
+    def _parse_screenshot_flag(self, args: list[str]) -> Optional[Xdp.ScreenshotFlags]:
+        for arg in args:
+            if arg.startswith("--screenshot"):
+                if "=" in arg:
+                    mode = arg.split("=", 1)[1].strip().upper()
+                else:
+                    mode = "INTERACTIVE"
+                if mode == "INTERACTIVE":
+                    return Xdp.ScreenshotFlags.INTERACTIVE
+                elif mode == "FULL":
+                    return Xdp.ScreenshotFlags.NONE
+                else:
+                    logging.warning(f"Unknown screenshot mode: {mode}. Defaulting to INTERACTIVE.")
+                    return Xdp.ScreenshotFlags.INTERACTIVE
+        return None
+
     def do_open(self, files: Sequence[Gio.File], hint: str):
         logging.debug(f"do_open called with files: {[file.get_path() for file in files]} and hint: {hint}")
         for file in files:
@@ -84,11 +100,10 @@ def do_open(self, files: Sequence[Gio.File], hint: str):
 
     def do_activate(self):
         logging.debug("do_activate called")
-        # Fallback if app is run without args and not via do_open/command_line
         self._open_window(None)
 
     def _open_window(self, file_path: Optional[str]):
-        logging.debug(f"Opening window with file_path={file_path}, screenshot_mode={self.screenshot_mode}")
+        logging.debug(f"Opening window with file_path={file_path}, screenshot_flags={self.screenshot_flags}")
         temp_dir = tempfile.mkdtemp()
         logging.debug(f"Created temp directory: {temp_dir}")
         self.temp_dirs.append(temp_dir)
@@ -97,7 +112,7 @@ def _open_window(self, file_path: Optional[str]):
             temp_dir=temp_dir,
             version=self.version,
             application=self,
-            init_with_screenshot=self.screenshot_mode,
+            init_screenshot_mode=self.screenshot_flags,
             file_path=file_path
         )
         window.build_ui()

gradia/ui/image_loaders.py

@@ -186,12 +186,12 @@ def __init__(self, window: Gtk.ApplicationWindow, temp_dir: str) -> None:
         super().__init__(window, temp_dir)
         self.portal = Xdp.Portal()
 
-    def take_screenshot(self) -> None:
-        """Initiate screenshot capture"""
+    def take_screenshot(self, flags: Xdp.ScreenshotFlags = Xdp.ScreenshotFlags.INTERACTIVE) -> None:
+        """Initiate screenshot capture with optional flags"""
         try:
             self.portal.take_screenshot(
                 None,
-                Xdp.ScreenshotFlags.INTERACTIVE,
+                flags,
                 None,
                 self._on_screenshot_taken,
                 None
@@ -289,8 +289,8 @@ def _on_drop_action(self, action: Optional[object], param: object) -> None:
     def load_from_clipboard(self) -> None:
         self.clipboard_loader.load_from_clipboard()
 
-    def take_screenshot(self) -> None:
-        self.screenshot_loader.take_screenshot()
+    def take_screenshot(self, flags: Xdp.ScreenshotFlags = Xdp.ScreenshotFlags.INTERACTIVE) -> None:
+        self.screenshot_loader.take_screenshot(flags)
 
     def load_from_file(self, file_path: str) -> None:
         self.commandline_loader.load_from_file(file_path)

gradia/ui/window.py

@@ -20,7 +20,7 @@
 from collections.abc import Callable
 from typing import Optional, Any
 
-from gi.repository import Gtk, Gio, Adw, Gdk, GLib
+from gi.repository import Gtk, Gio, Adw, Gdk, GLib, Xdp
 from gradia.graphics.image_processor import ImageProcessor
 from gradia.graphics.gradient import GradientSelector, GradientBackground
 from gradia.ui.ui_parts import *
@@ -46,7 +46,7 @@ class GradientWindow(Adw.ApplicationWindow):
     # Temp file names
     TEMP_PROCESSED_FILENAME: str = "processed.png"
 
-    def __init__(self, temp_dir: str, version: str, init_with_screenshot: bool = False, file_path: str = None, **kwargs) -> None:
+    def __init__(self, temp_dir: str, version: str, init_screenshot_mode: Xdp.ScreenshotFlags , file_path: str = None, **kwargs) -> None:
         super().__init__(**kwargs)
 
         self.app: Adw.Application = kwargs['application']
@@ -96,8 +96,8 @@ def __init__(self, temp_dir: str, version: str, init_with_screenshot: bool = Fal
         self.create_action("del-selected", lambda *_: self.drawing_overlay.remove_selected_action(), ["<Primary>x", "Delete"])
         self.file_path = file_path
 
-        if init_with_screenshot:
-            self.import_manager.take_screenshot()
+        if init_screenshot_mode != None:
+            self.import_manager.take_screenshot(init_screenshot_mode)
 
 
     def create_action(self, name: str, callback: Callable[..., None], shortcuts: Optional[list[str]] = None, enabled: bool = True) -> None: