Create bridge.py

bridge.py

@@ -0,0 +1,139 @@
+import os, json, uuid, sqlite3
+from datetime import datetime, timezone
+from flask import Blueprint, request, jsonify
+from prometheus_client import Counter, Histogram, generate_latest, CONTENT_TYPE_LATEST
+from functools import wraps
+
+bridge = Blueprint("bridge", __name__)
+
+DB_PATH = os.environ.get("CI_LEGEND_DB", "/data/legend.db")
+API_KEY = os.environ.get("CI_API_KEY", "dev-key-change-me")
+
+REQS = Counter("ci_bridge_requests_total", "Total requests", ["route","code"])
+LAT  = Histogram("ci_bridge_latency_seconds", "Latency", ["route"])
+
+# --- tiny SQLite ---
+def _db():
+    os.makedirs(os.path.dirname(DB_PATH), exist_ok=True)
+    con = sqlite3.connect(DB_PATH)
+    con.execute("""CREATE TABLE IF NOT EXISTS legend (
+        id TEXT PRIMARY KEY,
+        title TEXT,
+        tags TEXT,
+        body TEXT,
+        created_at TEXT,
+        updated_at TEXT
+    )""")
+    return con
+
+def _now(): return datetime.now(timezone.utc).isoformat()
+def _trace(): return f"ci-{uuid.uuid4().hex[:12]}"
+
+def _auth():
+    hdr = request.headers.get("Authorization","")
+    return hdr.startswith("Bearer ") and hdr.split(" ",1)[1] == API_KEY
+
+def secured(fn):
+    @wraps(fn)
+    def wrap(*a, **k):
+        route = request.path
+        with LAT.labels(route).time():
+            if not _auth():
+                REQS.labels(route,"401").inc()
+                return jsonify({"status":"unauthorized"}), 401
+            resp = fn(*a, **k)
+        try:
+            code = resp[1] if isinstance(resp, tuple) else 200
+        except Exception:
+            code = 200
+        REQS.labels(route,str(code)).inc()
+        return resp
+    return wrap
+
+# --- Legend CRUD ---
+def legend_upsert(item):
+    con = _db()
+    now = _now()
+    item_id = item.get("id") or uuid.uuid4().hex
+    con.execute(
+      "INSERT OR REPLACE INTO legend (id,title,tags,body,created_at,updated_at) "
+      "VALUES (?,?,?,?,COALESCE((SELECT created_at FROM legend WHERE id=?),?),?)",
+      (item_id, item.get("title",""), json.dumps(item.get("tags",[]), ensure_ascii=False),
+       item.get("body",""), item_id, now, now))
+    con.commit(); con.close()
+    return {"id": item_id}
+
+def legend_get(item_id):
+    con = _db()
+    r = con.execute("SELECT id,title,tags,body,created_at,updated_at FROM legend WHERE id=?",(item_id,)).fetchone()
+    con.close()
+    if not r: return None
+    return {"id":r[0],"title":r[1],"tags":json.loads(r[2] or "[]"),"body":r[3],"created_at":r[4],"updated_at":r[5]}
+
+def legend_search(q="", limit=10, tags=None):
+    con = _db()
+    rows = con.execute("SELECT id,title,tags,body,created_at,updated_at FROM legend").fetchall()
+    con.close()
+    q = (q or "").lower()
+    out=[]
+    for r in rows:
+        rec={"id":r[0],"title":r[1],"tags":json.loads(r[2] or "[]"),"body":r[3],"created_at":r[4],"updated_at":r[5]}
+        text=(rec["title"]+" "+rec["body"]).lower()
+        if q and q not in text: continue
+        if tags and not set(tags).issubset(set(rec["tags"])): continue
+        out.append(rec)
+    return out[:max(1,min(100,int(limit or 10)))]
+
+# --- Endpoints ---
+@bridge.route("/metrics")
+def metrics():
+    return generate_latest(), 200, {"Content-Type": CONTENT_TYPE_LATEST}
+
+@bridge.route("/api/command", methods=["POST"])
+@secured
+def api_command():
+    trace = _trace()
+    data = request.get_json(force=True, silent=True) or {}
+    intent = (data.get("intent") or "").strip()
+    payload= data.get("payload") or {}
+
+    if intent == "legend.upsert":
+        res = legend_upsert(payload)
+        return jsonify({"status":"ok","result":res,"trace_id":trace})
+    if intent == "legend.get":
+        res = legend_get(payload.get("id"))
+        return (jsonify({"status":"ok","result":res,"trace_id":trace}),200) if res else (jsonify({"status":"not_found","trace_id":trace}),404)
+    if intent == "legend.search":
+        res = legend_search(payload.get("q",""), payload.get("limit",10), payload.get("tags"))
+        return jsonify({"status":"ok","result":res,"trace_id":trace})
+
+    return jsonify({"status":"no_route","trace_id":trace}), 400
+
+@bridge.route("/api/data/query", methods=["POST"])
+@secured
+def api_data_query():
+    trace=_trace()
+    q = request.get_json(force=True, silent=True) or {}
+    domain = q.get("domain","legend")
+    if domain=="legend":
+        res = legend_search(q.get("q",""), q.get("limit",10), q.get("tags"))
+        return jsonify({"status":"ok","result":res,"trace_id":trace})
+    return jsonify({"status":"unsupported_domain","trace_id":trace}),400
+
+@bridge.route("/api/kazkar/legend", methods=["POST"])
+@secured
+def api_legend():
+    trace=_trace()
+    req = request.get_json(force=True, silent=True) or {}
+    action = (req.get("action") or "search").lower()
+    if action=="upsert":  res=legend_upsert(req.get("item",{}))
+    elif action=="get":   res=legend_get(req.get("id"))
+    elif action=="search":res=legend_search(req.get("q",""), req.get("limit",10), req.get("tags"))
+    else: return jsonify({"status":"bad_action","trace_id":trace}),400
+    return jsonify({"status":"ok","result":res,"trace_id":trace})
+
+@bridge.route("/api/kazkar/legend/<item_id>", methods=["GET"])
+def api_legend_get(item_id):
+    res = legend_get(item_id)
+    return (jsonify(res),200) if res else (jsonify({"status":"not_found"}),404)
+