squads-cli 0.2.0 → 0.4.0

package/docker/telemetry-ping/Dockerfile

@@ -0,0 +1,10 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY main.py .
+
+ENV PORT=8080
+CMD exec gunicorn --bind :$PORT --workers 1 --threads 2 --timeout 30 main:app

package/docker/telemetry-ping/deploy.sh

@@ -0,0 +1,69 @@
+#!/bin/bash
+set -e
+
+PROJECT_ID="inspired-answer-481202-f6"
+REGION="us-central1"
+SERVICE_NAME="squads-telemetry"
+TOPIC_ID="squads-cli-telemetry"
+DATASET_ID="telemetry"
+TABLE_ID="cli_events"
+
+echo "=== Squads Telemetry Deployment ==="
+
+# 1. Enable APIs
+echo "[1/6] Enabling APIs..."
+gcloud services enable \
+    pubsub.googleapis.com \
+    bigquery.googleapis.com \
+    run.googleapis.com \
+    cloudbuild.googleapis.com \
+    --project=$PROJECT_ID
+
+# 2. Create Pub/Sub topic
+echo "[2/6] Creating Pub/Sub topic..."
+gcloud pubsub topics create $TOPIC_ID --project=$PROJECT_ID 2>/dev/null || echo "Topic exists"
+
+# 3. Create BigQuery dataset
+echo "[3/6] Creating BigQuery dataset..."
+bq --project_id=$PROJECT_ID mk --dataset $DATASET_ID 2>/dev/null || echo "Dataset exists"
+
+# 4. Create BigQuery table
+echo "[4/6] Creating BigQuery table..."
+bq --project_id=$PROJECT_ID mk --table $DATASET_ID.$TABLE_ID \
+    event:STRING,timestamp:TIMESTAMP,anonymous_id:STRING,cli_version:STRING,properties:STRING,server_ts:TIMESTAMP,source:STRING \
+    2>/dev/null || echo "Table exists"
+
+# 5. Create Pub/Sub → BigQuery subscription
+echo "[5/6] Creating BigQuery subscription..."
+gcloud pubsub subscriptions create ${TOPIC_ID}-bq \
+    --topic=$TOPIC_ID \
+    --bigquery-table=$PROJECT_ID:$DATASET_ID.$TABLE_ID \
+    --use-topic-schema=false \
+    --write-metadata \
+    --project=$PROJECT_ID 2>/dev/null || echo "Subscription exists"
+
+# 6. Deploy Cloud Run
+echo "[6/6] Deploying to Cloud Run..."
+gcloud run deploy $SERVICE_NAME \
+    --source=. \
+    --region=$REGION \
+    --platform=managed \
+    --allow-unauthenticated \
+    --set-env-vars="GCP_PROJECT=$PROJECT_ID,PUBSUB_TOPIC=$TOPIC_ID" \
+    --memory=256Mi \
+    --min-instances=0 \
+    --max-instances=5 \
+    --project=$PROJECT_ID
+
+# Get URL
+URL=$(gcloud run services describe $SERVICE_NAME --region=$REGION --project=$PROJECT_ID --format='value(status.url)')
+
+echo ""
+echo "=== Deployment Complete ==="
+echo "Endpoint: $URL/ping"
+echo ""
+echo "Test with:"
+echo "  curl -X POST $URL/ping -H 'Content-Type: application/json' -d '{\"event\":\"test\"}'"
+echo ""
+echo "Update squads-cli telemetry URL:"
+echo "  SQUADS_TELEMETRY_URL=$URL/ping"

package/docker/telemetry-ping/main.py

@@ -0,0 +1,136 @@
+"""
+Squads CLI Telemetry Ping - Public endpoint for anonymous CLI telemetry.
+Receives events and publishes to Pub/Sub for BigQuery streaming.
+"""
+import os
+import json
+import hashlib
+import hmac
+from datetime import datetime
+from flask import Flask, request, jsonify
+from google.cloud import pubsub_v1
+
+app = Flask(__name__)
+
+# Config
+PROJECT_ID = os.environ.get("GCP_PROJECT", "inspired-answer-481202-f6")
+TOPIC_ID = os.environ.get("PUBSUB_TOPIC", "squads-cli-telemetry")
+DEBUG = os.environ.get("DEBUG", "0") == "1"
+
+# API Key for validation (prevents spam/contamination)
+# CLI embeds this key (obfuscated) - not true security, but adds friction
+API_KEY = os.environ.get("TELEMETRY_API_KEY", "sq_tel_v1_7f8a9b2c3d4e5f6a")
+
+# Valid event prefixes (reject unknown events)
+VALID_EVENT_PREFIXES = ("cli.", "agent.", "squad.", "error.")
+
+# Pub/Sub publisher (lazy init)
+publisher = None
+
+def get_publisher():
+    global publisher
+    if publisher is None:
+        publisher = pubsub_v1.PublisherClient()
+    return publisher
+
+def validate_request() -> tuple[bool, str]:
+    """Validate API key and basic structure."""
+    # Check API key header
+    api_key = request.headers.get("X-Squads-Key") or request.headers.get("Authorization", "").replace("Bearer ", "")
+    if not api_key or not hmac.compare_digest(api_key, API_KEY):
+        return False, "invalid_key"
+    return True, ""
+
+def validate_event(event: dict) -> bool:
+    """Validate event structure."""
+    if not isinstance(event, dict):
+        return False
+
+    event_name = event.get("event", "")
+    if not event_name or not any(event_name.startswith(p) for p in VALID_EVENT_PREFIXES):
+        return False
+
+    # Must have properties dict (can be empty)
+    if "properties" in event and not isinstance(event.get("properties"), dict):
+        return False
+
+    return True
+
+def publish_event(event: dict) -> bool:
+    """Publish event to Pub/Sub."""
+    try:
+        pub = get_publisher()
+        topic_path = pub.topic_path(PROJECT_ID, TOPIC_ID)
+
+        # Add server metadata
+        event["server_ts"] = datetime.utcnow().isoformat() + "Z"
+        event["source"] = "squads-cli"
+
+        data = json.dumps(event).encode("utf-8")
+        future = pub.publish(topic_path, data)
+        future.result(timeout=5)  # Wait for ack
+        return True
+    except Exception as e:
+        if DEBUG:
+            print(f"Pub/Sub error: {e}")
+        return False
+
+@app.route("/", methods=["GET"])
+def health():
+    """Health check."""
+    return jsonify({"status": "ok", "service": "squads-telemetry"}), 200
+
+@app.route("/ping", methods=["POST"])
+def ping():
+    """
+    Receive telemetry ping from CLI.
+
+    Headers:
+        X-Squads-Key: <api_key>
+
+    POST /ping
+    Content-Type: application/json
+
+    {"event": "cli.run", "properties": {"squad": "marketing", "durationMs": 1234}}
+    """
+    # Validate API key
+    valid, error = validate_request()
+    if not valid:
+        return jsonify({"error": error}), 401
+
+    try:
+        data = request.get_json(silent=True)
+        if not data:
+            return jsonify({"error": "invalid json"}), 400
+
+        # Support single event or batch
+        events = data.get("events", [data])
+
+        # Validate and publish
+        published = 0
+        rejected = 0
+        for event in events:
+            if validate_event(event):
+                if publish_event(event):
+                    published += 1
+            else:
+                rejected += 1
+
+        return jsonify({
+            "status": "ok",
+            "received": len(events),
+            "published": published,
+            "rejected": rejected,
+        }), 200
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route("/ping", methods=["GET"])
+def ping_get():
+    """Simple GET ping for uptime checks."""
+    return jsonify({"pong": True, "ts": datetime.utcnow().isoformat()}), 200
+
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 8080))
+    app.run(host="0.0.0.0", port=port, debug=DEBUG)

package/docker/telemetry-ping/requirements.txt

@@ -0,0 +1,3 @@
+flask==3.0.0
+gunicorn==21.2.0
+google-cloud-pubsub==2.18.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "squads-cli",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "A CLI for humans and agents",
   "type": "module",
   "bin": {
@@ -15,7 +15,17 @@
     }
   },
   "files": [
-    "dist"
+    "dist",
+    "docker/docker-compose.yml",
+    "docker/docker-compose.engram.yml",
+    "docker/.env.example",
+    "docker/init-db.sql",
+    "docker/init-engram-db.sql",
+    "docker/init-langfuse-db.sh",
+    "docker/otel-collector.yaml",
+    "docker/README.md",
+    "docker/squads-bridge/",
+    "docker/telemetry-ping/"
   ],
   "scripts": {
     "build": "tsup",
@@ -56,6 +66,9 @@
   },
   "dependencies": {
     "@supabase/supabase-js": "^2.89.0",
+    "@types/blessed": "^0.1.27",
+    "blessed": "^0.1.81",
+    "blessed-contrib": "^4.11.0",
     "chalk": "^5.3.0",
     "commander": "^12.1.0",
     "dotenv": "^17.2.3",