lokuma-cli 1.4.9 → 1.4.10

package/assets/design.py

@@ -235,6 +235,129 @@ def _format_output(result: dict, fmt: str) -> str:
     return "\n".join(lines) + IMAGE_REMINDER
 
 
+# ─────────────────────────────────────────────
+# SSE streaming client
+# ─────────────────────────────────────────────
+
+_DOMAIN_LABELS = {
+    "style":        "Visual Style",
+    "color":        "Color Palette",
+    "typography":   "Typography",
+    "product":      "Product Type",
+    "reasoning":    "Design Reasoning",
+    "ux":           "UX Guidelines",
+    "chart":        "Charts",
+    "landing":      "Landing Pattern",
+    "icons":        "Icon Library",
+    "google-fonts": "Google Fonts",
+}
+
+
+def _post_stream(endpoint: str, payload: dict) -> dict:
+    """
+    POST to a streaming SSE endpoint.
+    Prints progress as events arrive, returns the final 'done' payload.
+    Falls back to non-streaming if SSE is not supported.
+    """
+    api_key = _get_api_key()
+    url = f"{_BASE}/{endpoint}"
+    data = json.dumps(payload, ensure_ascii=False).encode("utf-8")
+
+    req = urllib.request.Request(
+        url,
+        data=data,
+        headers={
+            "Content-Type": "application/json",
+            "X-API-Key": api_key,
+            "User-Agent": "lokuma-skill/2.0",
+            "Accept": "text/event-stream",
+        },
+        method="POST",
+    )
+
+    try:
+        with urllib.request.urlopen(req, timeout=120) as resp:
+            content_type = resp.headers.get("Content-Type", "")
+            if "text/event-stream" not in content_type:
+                # Server returned regular JSON — parse and return
+                body = resp.read().decode("utf-8")
+                return json.loads(body)
+
+            # Parse SSE stream
+            result = {}
+            buffer = ""
+            total = 0
+            done_count = 0
+
+            print("", file=sys.stderr)  # blank line before progress
+
+            for raw_line in resp:
+                line = raw_line.decode("utf-8", errors="replace").rstrip("\n\r")
+
+                if not line.startswith("data:"):
+                    continue
+
+                json_str = line[len("data:"):].strip()
+                if not json_str:
+                    continue
+
+                try:
+                    event = json.loads(json_str)
+                except json.JSONDecodeError:
+                    continue
+
+                kind = event.get("event")
+
+                if kind == "start":
+                    total = event.get("total", 10)
+                    name = event.get("project_name", "")
+                    print(f"  ⟳ Analyzing: {name}", file=sys.stderr)
+
+                elif kind == "domain_done":
+                    done_count += 1
+                    domain = event.get("domain", "")
+                    match = event.get("match", "")
+                    label = _DOMAIN_LABELS.get(domain, domain.title())
+                    bar = "█" * done_count + "░" * (total - done_count)
+                    print(f"  [{bar}] {label}: {match}", file=sys.stderr)
+
+                elif kind == "synthesizing":
+                    print(f"\n  ✦ Synthesizing design recommendation...", file=sys.stderr)
+
+                elif kind == "done":
+                    result = event
+                    warning = result.get("warning")
+                    if warning:
+                        print(f"\n⚠️  {warning}", file=sys.stderr)
+                    print("", file=sys.stderr)  # blank line after progress
+
+                elif kind == "error":
+                    print(f"\n  ✗ Error: {event.get('message', 'Unknown error')}", file=sys.stderr)
+
+            return result
+
+    except urllib.error.HTTPError as e:
+        body = e.read().decode("utf-8", errors="replace")
+        try:
+            err = json.loads(body)
+            msg = err.get("error", body)
+        except Exception:
+            msg = body
+        if e.code in (401, 403):
+            print(f"Error: Invalid or expired API key (HTTP {e.code})", file=sys.stderr)
+        elif e.code == 402:
+            print(f"Error: {msg}", file=sys.stderr)
+        else:
+            print(f"Error: API returned HTTP {e.code}: {msg}", file=sys.stderr)
+        sys.exit(1)
+    except urllib.error.URLError as e:
+        print(f"Error: Could not reach Lokuma API: {e.reason}", file=sys.stderr)
+        sys.exit(1)
+    except json.JSONDecodeError as e:
+        print(f"Error: Invalid JSON response: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
 # ─────────────────────────────────────────────
 # CLI
 # ─────────────────────────────────────────────
@@ -257,7 +380,7 @@ def main():
     if args.project_name:
         payload["project_name"] = args.project_name
 
-    result = _post("design", payload)
+    result = _post_stream("design/stream", payload)
     print(_format_output(result, args.format))
 
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lokuma-cli",
-  "version": "1.4.9",
+  "version": "1.4.10",
   "description": "CLI to install Lokuma design intelligence skill for AI coding assistants",
   "type": "module",
   "bin": {