ui-mirror-skill 1.0.0

package/skill/scripts/download_screenshot.py

@@ -0,0 +1,125 @@
+#!/usr/bin/env python3
+"""
+download_screenshot.py — Download a screenshot URL or decode a base64 data URL to a local file.
+
+Usage:
+    python3 download_screenshot.py <url> <output_path>
+    python3 download_screenshot.py --url <url> --output <output_path>
+
+Handles:
+    - HTTP/HTTPS URLs: fetched via urllib.request
+    - data: URLs (data:image/png;base64,...): decoded from base64
+
+Dependencies: Python 3 standard library only.
+"""
+
+import base64
+import os
+import sys
+import urllib.request
+import urllib.error
+
+
+TIMEOUT = 15  # seconds
+MAX_RETRIES = 1
+
+
+def download_url(url: str, output_path: str) -> None:
+    """Download a URL to a local file with retry."""
+    os.makedirs(os.path.dirname(os.path.abspath(output_path)), exist_ok=True)
+
+    last_error = None
+    for attempt in range(1 + MAX_RETRIES):
+        try:
+            req = urllib.request.Request(
+                url,
+                headers={"User-Agent": "ui-mirror-screenshot-downloader/1.0"},
+            )
+            with urllib.request.urlopen(req, timeout=TIMEOUT) as resp:
+                data = resp.read()
+            with open(output_path, "wb") as f:
+                f.write(data)
+            return
+        except (urllib.error.URLError, urllib.error.HTTPError, OSError) as e:
+            last_error = e
+            if attempt < MAX_RETRIES:
+                print(
+                    "Retry {}/{}: {}".format(attempt + 1, MAX_RETRIES, e),
+                    file=sys.stderr,
+                )
+
+    raise last_error  # type: ignore[misc]
+
+
+def decode_data_url(data_url: str, output_path: str) -> None:
+    """Decode a base64 data URL and write to file."""
+    os.makedirs(os.path.dirname(os.path.abspath(output_path)), exist_ok=True)
+
+    # Parse: data:[<mediatype>][;base64],<data>
+    if not data_url.startswith("data:"):
+        raise ValueError("Not a data URL: {}".format(data_url[:50]))
+
+    header_end = data_url.index(",")
+    header = data_url[5:header_end]  # skip "data:"
+    encoded = data_url[header_end + 1:]
+
+    if ";base64" not in header:
+        raise ValueError("Only base64 data URLs are supported, got: {}".format(header))
+
+    raw = base64.b64decode(encoded)
+    with open(output_path, "wb") as f:
+        f.write(raw)
+
+
+def main():
+    # Parse args: positional or --url/--output flags
+    url = None
+    output_path = None
+
+    args = sys.argv[1:]
+    i = 0
+    positional = []
+    while i < len(args):
+        if args[i] == "--url" and i + 1 < len(args):
+            url = args[i + 1]
+            i += 2
+        elif args[i] == "--output" and i + 1 < len(args):
+            output_path = args[i + 1]
+            i += 2
+        elif not args[i].startswith("--"):
+            positional.append(args[i])
+            i += 1
+        else:
+            i += 1
+
+    if url is None and len(positional) >= 1:
+        url = positional[0]
+    if output_path is None and len(positional) >= 2:
+        output_path = positional[1]
+
+    if not url or not output_path:
+        print(
+            "Usage: {} <url> <output_path>".format(sys.argv[0]),
+            file=sys.stderr,
+        )
+        print(
+            "       {} --url <url> --output <output_path>".format(sys.argv[0]),
+            file=sys.stderr,
+        )
+        sys.exit(1)
+
+    try:
+        if url.startswith("data:"):
+            decode_data_url(url, output_path)
+            print("Decoded data URL -> {}".format(output_path))
+        else:
+            download_url(url, output_path)
+            size = os.path.getsize(output_path)
+            print("Downloaded {} -> {} ({} bytes)".format(url, output_path, size))
+    except Exception as e:
+        print("Error: {}".format(e), file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()