add camoufox server and updated dockerfile

Dockerfile

@@ -1,29 +1,28 @@
 FROM golang:1.23-alpine AS builder
-
 WORKDIR /app
-
-# 安装 git 和 ca-certificates，某些依赖可能需要
 RUN apk add --no-cache git ca-certificates
-
 COPY go.mod go.sum ./
 RUN go mod download -x || (cat go.mod && cat go.sum && exit 1)
-
 COPY . .
 RUN CGO_ENABLED=0 GOOS=linux go build -o zai-proxy .
 
-FROM alpine:latest
-
+FROM python:3.11-slim
 WORKDIR /app
 
-# 安装 ca-certificates 以支持 HTTPS
-RUN apk add --no-cache ca-certificates
+# Instalar dependencias del sistema para Firefox/Camoufox
+RUN apt-get update && apt-get install -y     wget curl ca-certificates     libgtk-3-0 libdbus-glib-1-2     libxt6 libx11-xcb1     xvfb     --no-install-recommends && rm -rf /var/lib/apt/lists/*
 
-COPY --from=builder /app/zai-proxy .
+# Instalar Python deps
+RUN pip install camoufox[geoip] flask --no-cache-dir
 
-# proxies.txt 可通过 docker run -v ./proxies.txt:/app/proxies.txt 挂载
-VOLUME ["/app/proxies.txt"]
+# Descargar Camoufox browser
+RUN python3 -m camoufox fetch
 
-EXPOSE 7860
+# Copiar binario Go
+COPY --from=builder /app/zai-proxy .
+COPY . .
 
+EXPOSE 7860
 ENV PORT=7860
-CMD ["./zai-proxy"]
+
+CMD ["sh", "-c", "python3 camoufox_server.py & ./zai-proxy"]

camoufox_server.py

@@ -0,0 +1,63 @@
+import json, uuid, threading
+from flask import Flask, request, jsonify
+from camoufox.sync_api import Camoufox
+
+app = Flask(__name__)
+lock = threading.Lock()
+
+with Camoufox(headless=True) as browser:
+    ctx = browser.new_context()
+    page = ctx.new_page()
+    page.goto("https://chat.z.ai")
+    page.wait_for_load_state("networkidle")
+    print("Camoufox ready", flush=True)
+
+    def do_chat(messages, model):
+        body = json.dumps({
+            "stream": True,
+            "model": model,
+            "messages": messages,
+            "params": {},
+            "extra": {},
+            "features": {"image_generation": False, "web_search": False, "preview_mode": True, "flags": [], "enable_thinking": False},
+            "chat_id": str(uuid.uuid4()),
+            "id": str(uuid.uuid4()),
+        })
+        js = """(bodyStr) => {
+            return new Promise((resolve) => {
+                fetch("/api/v2/chat/completions", {
+                    method: "POST",
+                    headers: {"Content-Type": "application/json"},
+                    body: bodyStr
+                }).then(r => r.text()).then(t => resolve(t)).catch(e => resolve("ERROR:" + e));
+                setTimeout(() => resolve("TIMEOUT"), 30000);
+            });
+        }"""
+        result = page.evaluate(js, body)
+        content = ""
+        for line in str(result).splitlines():
+            if line.startswith("data: "):
+                try:
+                    d = json.loads(line[6:])
+                    delta = d.get("data", {}).get("delta_content", "")
+                    phase = d.get("data", {}).get("phase", "")
+                    if phase == "answer":
+                        content += delta
+                except:
+                    pass
+        return content
+
+    @app.route("/camoufox/chat", methods=["POST"])
+    def completions():
+        data = request.json
+        messages = data.get("messages", [])
+        model = data.get("model", "glm-4.7")
+        with lock:
+            content = do_chat(messages, model)
+        return jsonify({"content": content})
+
+    @app.route("/camoufox/health", methods=["GET"])
+    def health():
+        return jsonify({"status": "ok"})
+
+    app.run(port=8002, threaded=False)