Merge pull request 'feature/gemini-web-provider' (#1) from feature/gemini-web-provider into master
Reviewed-on: #1
This commit is contained in:
commit
85f695250e
23
Dockerfile
23
Dockerfile
|
|
@ -1,23 +0,0 @@
|
||||||
# ── Stage 1: 編譯 ─────────────────────────────────────────────
|
|
||||||
FROM golang:1.25-alpine AS builder
|
|
||||||
|
|
||||||
WORKDIR /build
|
|
||||||
|
|
||||||
COPY go.mod go.sum ./
|
|
||||||
RUN go mod download
|
|
||||||
|
|
||||||
COPY . .
|
|
||||||
RUN CGO_ENABLED=0 GOOS=linux go build -ldflags="-s -w" -o cursor-api-proxy .
|
|
||||||
|
|
||||||
# ── Stage 2: 執行環境 ──────────────────────────────────────────
|
|
||||||
FROM alpine:3.21
|
|
||||||
|
|
||||||
RUN apk add --no-cache ca-certificates tzdata
|
|
||||||
|
|
||||||
WORKDIR /app
|
|
||||||
|
|
||||||
COPY --from=builder /build/cursor-api-proxy .
|
|
||||||
|
|
||||||
EXPOSE 8766
|
|
||||||
|
|
||||||
ENTRYPOINT ["./cursor-api-proxy"]
|
|
||||||
39
Makefile
39
Makefile
|
|
@ -43,6 +43,12 @@ ANTHROPIC_BASE_HOST ?= $(HOST)
|
||||||
TLS_CERT ?=
|
TLS_CERT ?=
|
||||||
TLS_KEY ?=
|
TLS_KEY ?=
|
||||||
|
|
||||||
|
# ── Gemini Web Provider ───────────────────────
|
||||||
|
PROVIDER ?= cursor
|
||||||
|
GEMINI_ACCOUNT_DIR ?=
|
||||||
|
GEMINI_BROWSER_VISIBLE ?= false
|
||||||
|
GEMINI_MAX_SESSIONS ?= 3
|
||||||
|
|
||||||
# ── 記錄 ──────────────────────────────────────
|
# ── 記錄 ──────────────────────────────────────
|
||||||
SESSIONS_LOG ?=
|
SESSIONS_LOG ?=
|
||||||
|
|
||||||
|
|
@ -84,6 +90,12 @@ env:
|
||||||
@printf 'CURSOR_BRIDGE_TLS_CERT=%s\n' "$(TLS_CERT)" >> $(ENV_FILE)
|
@printf 'CURSOR_BRIDGE_TLS_CERT=%s\n' "$(TLS_CERT)" >> $(ENV_FILE)
|
||||||
@printf 'CURSOR_BRIDGE_TLS_KEY=%s\n' "$(TLS_KEY)" >> $(ENV_FILE)
|
@printf 'CURSOR_BRIDGE_TLS_KEY=%s\n' "$(TLS_KEY)" >> $(ENV_FILE)
|
||||||
@printf 'CURSOR_BRIDGE_SESSIONS_LOG=%s\n' "$(SESSIONS_LOG)" >> $(ENV_FILE)
|
@printf 'CURSOR_BRIDGE_SESSIONS_LOG=%s\n' "$(SESSIONS_LOG)" >> $(ENV_FILE)
|
||||||
|
@printf '# ── Provider 設定 ───────────────────────────\n' >> $(ENV_FILE)
|
||||||
|
@printf 'CURSOR_BRIDGE_PROVIDER=%s\n' "$(PROVIDER)" >> $(ENV_FILE)
|
||||||
|
@printf '# Gemini Web Provider 設定(當 PROVIDER=gemini-web 時使用)\n' >> $(ENV_FILE)
|
||||||
|
@printf 'GEMINI_ACCOUNT_DIR=%s\n' "$(GEMINI_ACCOUNT_DIR)" >> $(ENV_FILE)
|
||||||
|
@printf 'GEMINI_BROWSER_VISIBLE=%s\n' "$(GEMINI_BROWSER_VISIBLE)" >> $(ENV_FILE)
|
||||||
|
@printf 'GEMINI_MAX_SESSIONS=%s\n' "$(GEMINI_MAX_SESSIONS)" >> $(ENV_FILE)
|
||||||
@echo "已產生 $(ENV_FILE)"
|
@echo "已產生 $(ENV_FILE)"
|
||||||
|
|
||||||
## 編譯二進位檔
|
## 編譯二進位檔
|
||||||
|
|
@ -102,18 +114,18 @@ run: build
|
||||||
clean:
|
clean:
|
||||||
rm -f cursor-api-proxy $(ENV_FILE)
|
rm -f cursor-api-proxy $(ENV_FILE)
|
||||||
|
|
||||||
## 設定 OpenCode 使用此代理(更新 opencode.json 的 cursor provider)
|
## 設定 OpenCode 使用此代理(更新 opencode.json 的 cursor 與 gemini-web provider)
|
||||||
opencode: build
|
opencode: build
|
||||||
@if [ ! -f "$(OPENCODE_CONFIG)" ]; then \
|
@if [ ! -f "$(OPENCODE_CONFIG)" ]; then \
|
||||||
echo "找不到 $(OPENCODE_CONFIG),建立新設定檔"; \
|
echo "找不到 $(OPENCODE_CONFIG),建立新設定檔"; \
|
||||||
mkdir -p $$(dirname "$(OPENCODE_CONFIG)"); \
|
mkdir -p $$(dirname "$(OPENCODE_CONFIG)"); \
|
||||||
printf '{\n "model": "$(OPENCODE_MODEL)",\n "small_model": "$(OPENCODE_SMALL_MODEL)",\n "provider": {\n "cursor": {\n "npm": "@ai-sdk/openai-compatible",\n "name": "Cursor Agent",\n "options": {\n "baseURL": "http://$(HOST):$(PORT)/v1",\n "apiKey": "unused"\n },\n "models": { "auto": { "name": "Cursor Auto" } }\n }\n }\n}\n' > "$(OPENCODE_CONFIG)"; \
|
printf '{\n "model": "$(OPENCODE_MODEL)",\n "small_model": "$(OPENCODE_SMALL_MODEL)",\n "provider": {\n "cursor": {\n "npm": "@ai-sdk/openai-compatible",\n "name": "Cursor Agent",\n "options": {\n "baseURL": "http://$(HOST):$(PORT)/v1",\n "apiKey": "unused"\n },\n "models": { "auto": { "name": "Cursor Auto" } }\n },\n "gemini-web": {\n "npm": "@ai-sdk/openai-compatible",\n "name": "Gemini Web",\n "options": {\n "baseURL": "http://$(HOST):$(PORT)/v1",\n "apiKey": "unused"\n },\n "models": {\n "gemini-2.0-flash": { "name": "Gemini 2.0 Flash" },\n "gemini-2.5-pro": { "name": "Gemini 2.5 Pro" },\n "gemini-2.5-pro-thinking": { "name": "Gemini 2.5 Pro Thinking" }\n }\n }\n }\n}\n' > "$(OPENCODE_CONFIG)"; \
|
||||||
echo "已建立 $(OPENCODE_CONFIG)"; \
|
echo "已建立 $(OPENCODE_CONFIG)(包含 cursor 與 gemini-web provider)"; \
|
||||||
elif [ -n "$(API_KEY)" ]; then \
|
elif [ -n "$(API_KEY)" ]; then \
|
||||||
jq --arg model "$(OPENCODE_MODEL)" --arg small "$(OPENCODE_SMALL_MODEL)" --arg base "http://$(HOST):$(PORT)/v1" --arg key "$(API_KEY)" '.model = $$model | .small_model = $$small | .provider.cursor.options.baseURL = $$base | .provider.cursor.options.apiKey = $$key' "$(OPENCODE_CONFIG)" > "$(OPENCODE_CONFIG).tmp" && mv "$(OPENCODE_CONFIG).tmp" "$(OPENCODE_CONFIG)"; \
|
jq --arg model "$(OPENCODE_MODEL)" --arg small "$(OPENCODE_SMALL_MODEL)" --arg base "http://$(HOST):$(PORT)/v1" --arg key "$(API_KEY)" '.model = $$model | .small_model = $$small | .provider.cursor.options.baseURL = $$base | .provider.cursor.options.apiKey = $$key | .provider["gemini-web"].options.baseURL = $$base | .provider["gemini-web"].options.apiKey = $$key' "$(OPENCODE_CONFIG)" > "$(OPENCODE_CONFIG).tmp" && mv "$(OPENCODE_CONFIG).tmp" "$(OPENCODE_CONFIG)"; \
|
||||||
echo "已更新 $(OPENCODE_CONFIG)(model=$(OPENCODE_MODEL), small_model=$(OPENCODE_SMALL_MODEL), baseURL → http://$(HOST):$(PORT)/v1,apiKey 已設定)"; \
|
echo "已更新 $(OPENCODE_CONFIG)(model=$(OPENCODE_MODEL), small_model=$(OPENCODE_SMALL_MODEL), baseURL → http://$(HOST):$(PORT)/v1,apiKey 已設定)"; \
|
||||||
else \
|
else \
|
||||||
jq --arg model "$(OPENCODE_MODEL)" --arg small "$(OPENCODE_SMALL_MODEL)" --arg base "http://$(HOST):$(PORT)/v1" '.model = $$model | .small_model = $$small | .provider.cursor.options.baseURL = $$base' "$(OPENCODE_CONFIG)" > "$(OPENCODE_CONFIG).tmp" && mv "$(OPENCODE_CONFIG).tmp" "$(OPENCODE_CONFIG)"; \
|
jq --arg model "$(OPENCODE_MODEL)" --arg small "$(OPENCODE_SMALL_MODEL)" --arg base "http://$(HOST):$(PORT)/v1" '.model = $$model | .small_model = $$small | .provider.cursor.options.baseURL = $$base | .provider["gemini-web"].options.baseURL = $$base' "$(OPENCODE_CONFIG)" > "$(OPENCODE_CONFIG).tmp" && mv "$(OPENCODE_CONFIG).tmp" "$(OPENCODE_CONFIG)"; \
|
||||||
echo "已更新 $(OPENCODE_CONFIG)(model=$(OPENCODE_MODEL), small_model=$(OPENCODE_SMALL_MODEL), baseURL → http://$(HOST):$(PORT)/v1)"; \
|
echo "已更新 $(OPENCODE_CONFIG)(model=$(OPENCODE_MODEL), small_model=$(OPENCODE_SMALL_MODEL), baseURL → http://$(HOST):$(PORT)/v1)"; \
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
|
@ -126,8 +138,8 @@ opencode-models: opencode
|
||||||
MODELS=$$(curl -s http://$(HOST):$(PORT)/v1/models | jq '[.data[].id]'); \
|
MODELS=$$(curl -s http://$(HOST):$(PORT)/v1/models | jq '[.data[].id]'); \
|
||||||
kill $$PID 2>/dev/null; wait $$PID 2>/dev/null; \
|
kill $$PID 2>/dev/null; wait $$PID 2>/dev/null; \
|
||||||
if [ -n "$$MODELS" ] && [ "$$MODELS" != "null" ]; then \
|
if [ -n "$$MODELS" ] && [ "$$MODELS" != "null" ]; then \
|
||||||
jq --argjson ids "$$MODELS" 'reduce $ids[] as $id (.; .provider.cursor.models[$id] = { name: $id })' "$(OPENCODE_CONFIG)" > "$(OPENCODE_CONFIG).tmp" && mv "$(OPENCODE_CONFIG).tmp" "$(OPENCODE_CONFIG)"; \
|
jq --argjson ids "$$MODELS" 'reduce $ids[] as $id (.; .provider.cursor.models[$id] = { name: $id } | .provider["gemini-web"].models[$id] = { name: $id })' "$(OPENCODE_CONFIG)" > "$(OPENCODE_CONFIG).tmp" && mv "$(OPENCODE_CONFIG).tmp" "$(OPENCODE_CONFIG)"; \
|
||||||
echo "已同步模型列表到 $(OPENCODE_CONFIG)"; \
|
echo "已同步模型列表到 $(OPENCODE_CONFIG)(cursor 與 gemini-web)"; \
|
||||||
else \
|
else \
|
||||||
echo "無法取得模型列表,請確認代理已啟動"; \
|
echo "無法取得模型列表,請確認代理已啟動"; \
|
||||||
fi
|
fi
|
||||||
|
|
@ -287,8 +299,21 @@ help:
|
||||||
@echo " make docker-restart 重新建置並啟動容器"
|
@echo " make docker-restart 重新建置並啟動容器"
|
||||||
@echo " make docker-shell 進入容器 shell(除錯用)"
|
@echo " make docker-shell 進入容器 shell(除錯用)"
|
||||||
@echo ""
|
@echo ""
|
||||||
|
@echo "Provider 設定範例:"
|
||||||
|
@echo " make env PROVIDER=cursor # 使用 Cursor(預設)"
|
||||||
|
@echo " make env PROVIDER=gemini-web # 使用 Gemini Web"
|
||||||
|
@echo " make env PROVIDER=gemini-web GEMINI_ACCOUNT_DIR=/path/to/sessions"
|
||||||
|
@echo " make env PROVIDER=gemini-web GEMINI_BROWSER_VISIBLE=true"
|
||||||
|
@echo " make opencode # 設定 OpenCode(含 cursor 與 gemini-web provider)"
|
||||||
|
@echo ""
|
||||||
@echo "覆寫範例:"
|
@echo "覆寫範例:"
|
||||||
@echo " make env PORT=9000 API_KEY=mysecret TIMEOUT_MS=60000"
|
@echo " make env PORT=9000 API_KEY=mysecret TIMEOUT_MS=60000"
|
||||||
@echo " make pm2-claude-code PORT=8765 API_KEY=mykey"
|
@echo " make pm2-claude-code PORT=8765 API_KEY=mykey"
|
||||||
@echo " make pm2-opencode PORT=8765"
|
@echo " make pm2-opencode PORT=8765"
|
||||||
@echo " make claude-settings PORT=8766 ANTHROPIC_BASE_HOST=localhost ANTHROPIC_DEFAULT_OPUS_MODEL=claude-4.6-opus-high"
|
@echo " make claude-settings PORT=8766 ANTHROPIC_BASE_HOST=localhost ANTHROPIC_DEFAULT_OPUS_MODEL=claude-4.6-opus-high"
|
||||||
|
@echo ""
|
||||||
|
@echo "使用 Gemini Web Provider:"
|
||||||
|
@echo " 1. make env PROVIDER=gemini-web"
|
||||||
|
@echo " 2. gemini-login my-session # 登入並儲存 session"
|
||||||
|
@echo " 3. make run # 啟動代理"
|
||||||
|
@echo " 4. 在 OpenCode 設定 model: gemini/gemini-2.5-pro"
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,61 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cursor-api-proxy/internal/config"
|
||||||
|
"cursor-api-proxy/internal/env"
|
||||||
|
"cursor-api-proxy/internal/providers/geminiweb"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
accountName := ""
|
||||||
|
visible := false
|
||||||
|
|
||||||
|
// 解析命令列參數
|
||||||
|
for i := 1; i < len(os.Args); i++ {
|
||||||
|
arg := os.Args[i]
|
||||||
|
if arg == "--visible" || arg == "-v" {
|
||||||
|
visible = true
|
||||||
|
} else if arg == "--help" || arg == "-h" {
|
||||||
|
printHelp()
|
||||||
|
os.Exit(0)
|
||||||
|
} else if !strings.HasPrefix(arg, "-") {
|
||||||
|
accountName = arg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
e := env.OsEnvToMap()
|
||||||
|
loaded := env.LoadEnvConfig(e, "")
|
||||||
|
cfg := config.LoadBridgeConfig(e, "")
|
||||||
|
|
||||||
|
cfg.GeminiAccountDir = loaded.GeminiAccountDir
|
||||||
|
// 命令列參數優先於環境變數
|
||||||
|
cfg.GeminiBrowserVisible = visible || loaded.GeminiBrowserVisible
|
||||||
|
|
||||||
|
fmt.Printf("Session 儲存位置: %s\n", cfg.GeminiAccountDir)
|
||||||
|
fmt.Printf("瀏覽器可見: %v\n", cfg.GeminiBrowserVisible)
|
||||||
|
|
||||||
|
if err := geminiweb.RunLogin(cfg, accountName); err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func printHelp() {
|
||||||
|
fmt.Println("使用方法: gemini-login [options] [session-name]")
|
||||||
|
fmt.Println("")
|
||||||
|
fmt.Println("選項:")
|
||||||
|
fmt.Println(" --visible, -v 顯示瀏覽器視窗(預設隱藏)")
|
||||||
|
fmt.Println(" --help, -h 顯示此說明")
|
||||||
|
fmt.Println("")
|
||||||
|
fmt.Println("環境變數:")
|
||||||
|
fmt.Println(" GEMINI_ACCOUNT_DIR Session 儲存目錄(預設: ~/.cursor-api-proxy/gemini-accounts)")
|
||||||
|
fmt.Println(" GEMINI_BROWSER_VISIBLE 是否顯示瀏覽器(true/false,預設: false)")
|
||||||
|
fmt.Println("")
|
||||||
|
fmt.Println("範例:")
|
||||||
|
fmt.Println(" gemini-login my-session")
|
||||||
|
fmt.Println(" gemini-login --visible my-session")
|
||||||
|
fmt.Println(" GEMINI_BROWSER_VISIBLE=true gemini-login")
|
||||||
|
}
|
||||||
|
|
@ -1,27 +0,0 @@
|
||||||
services:
|
|
||||||
cursor-api-proxy:
|
|
||||||
build:
|
|
||||||
context: .
|
|
||||||
dockerfile: Dockerfile
|
|
||||||
image: cursor-api-proxy:latest
|
|
||||||
container_name: cursor-api-proxy
|
|
||||||
restart: unless-stopped
|
|
||||||
env_file:
|
|
||||||
- .env
|
|
||||||
ports:
|
|
||||||
- "${CURSOR_BRIDGE_PORT:-8766}:${CURSOR_BRIDGE_PORT:-8766}"
|
|
||||||
environment:
|
|
||||||
- CURSOR_BRIDGE_HOST=0.0.0.0
|
|
||||||
volumes:
|
|
||||||
# Cursor CLI 二進位檔(從宿主機掛載,唯讀)
|
|
||||||
- ${CURSOR_AGENT_HOST_BIN:-/usr/local/bin/agent}:/usr/local/bin/agent:ro
|
|
||||||
# 帳號設定目錄(持久化帳號資料)
|
|
||||||
- ${CURSOR_ACCOUNTS_DIR:-~/.cursor-api-proxy}:/root/.cursor-api-proxy
|
|
||||||
# 工作區(選用,掛載你想讓 agent 存取的專案目錄)
|
|
||||||
- ${WORKSPACE_DIR:-/tmp/workspace}:/workspace
|
|
||||||
healthcheck:
|
|
||||||
test: ["CMD", "wget", "-qO-", "http://localhost:${CURSOR_BRIDGE_PORT:-8766}/health"]
|
|
||||||
interval: 30s
|
|
||||||
timeout: 5s
|
|
||||||
retries: 3
|
|
||||||
start_period: 10s
|
|
||||||
10
go.mod
10
go.mod
|
|
@ -8,10 +8,20 @@ require (
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
github.com/deckarep/golang-set/v2 v2.8.0 // indirect
|
||||||
github.com/dustin/go-humanize v1.0.1 // indirect
|
github.com/dustin/go-humanize v1.0.1 // indirect
|
||||||
|
github.com/go-jose/go-jose/v3 v3.0.5 // indirect
|
||||||
|
github.com/go-rod/rod v0.116.2 // indirect
|
||||||
|
github.com/go-stack/stack v1.8.1 // indirect
|
||||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
github.com/ncruces/go-strftime v1.0.0 // indirect
|
github.com/ncruces/go-strftime v1.0.0 // indirect
|
||||||
|
github.com/playwright-community/playwright-go v0.5700.1 // indirect
|
||||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
||||||
|
github.com/ysmood/fetchup v0.2.3 // indirect
|
||||||
|
github.com/ysmood/goob v0.4.0 // indirect
|
||||||
|
github.com/ysmood/got v0.40.0 // indirect
|
||||||
|
github.com/ysmood/gson v0.7.3 // indirect
|
||||||
|
github.com/ysmood/leakless v0.9.0 // indirect
|
||||||
golang.org/x/sys v0.42.0 // indirect
|
golang.org/x/sys v0.42.0 // indirect
|
||||||
modernc.org/libc v1.70.0 // indirect
|
modernc.org/libc v1.70.0 // indirect
|
||||||
modernc.org/mathutil v1.7.1 // indirect
|
modernc.org/mathutil v1.7.1 // indirect
|
||||||
|
|
|
||||||
66
go.sum
66
go.sum
|
|
@ -1,5 +1,15 @@
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/deckarep/golang-set/v2 v2.8.0 h1:swm0rlPCmdWn9mESxKOjWk8hXSqoxOp+ZlfuyaAdFlQ=
|
||||||
|
github.com/deckarep/golang-set/v2 v2.8.0/go.mod h1:VAky9rY/yGXJOLEDv3OMci+7wtDpOF4IN+y82NBOac4=
|
||||||
github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
|
github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
|
||||||
github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
|
github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
|
||||||
|
github.com/go-jose/go-jose/v3 v3.0.5 h1:BLLJWbC4nMZOfuPVxoZIxeYsn6Nl2r1fITaJ78UQlVQ=
|
||||||
|
github.com/go-jose/go-jose/v3 v3.0.5/go.mod h1:5b+7YgP7ZICgJDBdfjZaIt+H/9L9T/YQrVfLAMboGkQ=
|
||||||
|
github.com/go-rod/rod v0.116.2 h1:A5t2Ky2A+5eD/ZJQr1EfsQSe5rms5Xof/qj296e+ZqA=
|
||||||
|
github.com/go-rod/rod v0.116.2/go.mod h1:H+CMO9SCNc2TJ2WfrG+pKhITz57uGNYU43qYHh438Mg=
|
||||||
|
github.com/go-stack/stack v1.8.1 h1:ntEHSVwIt7PNXNpgPmVfMrNhLtgjlmnZha2kOpuRiDw=
|
||||||
|
github.com/go-stack/stack v1.8.1/go.mod h1:dcoOX6HbPZSZptuspn9bctJ+N/CnF5gGygcUP3XYfe4=
|
||||||
|
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
|
||||||
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e h1:ijClszYn+mADRFY17kjQEVQ1XRhq2/JR1M3sGqeJoxs=
|
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e h1:ijClszYn+mADRFY17kjQEVQ1XRhq2/JR1M3sGqeJoxs=
|
||||||
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e/go.mod h1:boTsfXsheKC2y+lKOCMpSfarhxDeIzfZG1jqGcPl3cA=
|
github.com/google/pprof v0.0.0-20250317173921-a4b03ec1a45e/go.mod h1:boTsfXsheKC2y+lKOCMpSfarhxDeIzfZG1jqGcPl3cA=
|
||||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
|
@ -10,17 +20,73 @@ github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWE
|
||||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
github.com/ncruces/go-strftime v1.0.0 h1:HMFp8mLCTPp341M/ZnA4qaf7ZlsbTc+miZjCLOFAw7w=
|
github.com/ncruces/go-strftime v1.0.0 h1:HMFp8mLCTPp341M/ZnA4qaf7ZlsbTc+miZjCLOFAw7w=
|
||||||
github.com/ncruces/go-strftime v1.0.0/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
|
github.com/ncruces/go-strftime v1.0.0/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
|
||||||
|
github.com/playwright-community/playwright-go v0.5700.1 h1:PNFb1byWqrTT720rEO0JL88C6Ju0EmUnR5deFLvtP/U=
|
||||||
|
github.com/playwright-community/playwright-go v0.5700.1/go.mod h1:MlSn1dZrx8rszbCxY6x3qK89ZesJUYVx21B2JnkoNF0=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
|
||||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/ysmood/fetchup v0.2.3 h1:ulX+SonA0Vma5zUFXtv52Kzip/xe7aj4vqT5AJwQ+ZQ=
|
||||||
|
github.com/ysmood/fetchup v0.2.3/go.mod h1:xhibcRKziSvol0H1/pj33dnKrYyI2ebIvz5cOOkYGns=
|
||||||
|
github.com/ysmood/goob v0.4.0 h1:HsxXhyLBeGzWXnqVKtmT9qM7EuVs/XOgkX7T6r1o1AQ=
|
||||||
|
github.com/ysmood/goob v0.4.0/go.mod h1:u6yx7ZhS4Exf2MwciFr6nIM8knHQIE22lFpWHnfql18=
|
||||||
|
github.com/ysmood/got v0.40.0 h1:ZQk1B55zIvS7zflRrkGfPDrPG3d7+JOza1ZkNxcc74Q=
|
||||||
|
github.com/ysmood/got v0.40.0/go.mod h1:W7DdpuX6skL3NszLmAsC5hT7JAhuLZhByVzHTq874Qg=
|
||||||
|
github.com/ysmood/gotrace v0.6.0/go.mod h1:TzhIG7nHDry5//eYZDYcTzuJLYQIkykJzCRIo4/dzQM=
|
||||||
|
github.com/ysmood/gson v0.7.3 h1:QFkWbTH8MxyUTKPkVWAENJhxqdBa4lYTQWqZCiLG6kE=
|
||||||
|
github.com/ysmood/gson v0.7.3/go.mod h1:3Kzs5zDl21g5F/BlLTNcuAGAYLKt2lV5G8D1zF3RNmg=
|
||||||
|
github.com/ysmood/leakless v0.9.0 h1:qxCG5VirSBvmi3uynXFkcnLMzkphdh3xx5FtrORwDCU=
|
||||||
|
github.com/ysmood/leakless v0.9.0/go.mod h1:R8iAXPRaG97QJwqxs74RdwzcRHT1SWCGTNqY8q0JvMQ=
|
||||||
|
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||||
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
|
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||||
|
golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
|
||||||
|
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||||
|
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
||||||
golang.org/x/mod v0.33.0 h1:tHFzIWbBifEmbwtGz65eaWyGiGZatSrT9prnU8DbVL8=
|
golang.org/x/mod v0.33.0 h1:tHFzIWbBifEmbwtGz65eaWyGiGZatSrT9prnU8DbVL8=
|
||||||
golang.org/x/mod v0.33.0/go.mod h1:swjeQEj+6r7fODbD2cqrnje9PnziFuw4bmLbBZFrQ5w=
|
golang.org/x/mod v0.33.0/go.mod h1:swjeQEj+6r7fODbD2cqrnje9PnziFuw4bmLbBZFrQ5w=
|
||||||
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||||
|
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||||
|
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
||||||
|
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
|
||||||
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
|
||||||
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||||
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
|
golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
|
||||||
golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
|
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||||
|
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
||||||
|
golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
|
||||||
|
golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk=
|
||||||
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
|
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
|
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||||
|
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
|
||||||
|
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||||
|
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
||||||
golang.org/x/tools v0.42.0 h1:uNgphsn75Tdz5Ji2q36v/nsFSfR/9BRFvqhGBaJGd5k=
|
golang.org/x/tools v0.42.0 h1:uNgphsn75Tdz5Ji2q36v/nsFSfR/9BRFvqhGBaJGd5k=
|
||||||
golang.org/x/tools v0.42.0/go.mod h1:Ma6lCIwGZvHK6XtgbswSoWroEkhugApmsXyrUmBhfr0=
|
golang.org/x/tools v0.42.0/go.mod h1:Ma6lCIwGZvHK6XtgbswSoWroEkhugApmsXyrUmBhfr0=
|
||||||
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
modernc.org/cc/v4 v4.27.1 h1:9W30zRlYrefrDV2JE2O8VDtJ1yPGownxciz5rrbQZis=
|
modernc.org/cc/v4 v4.27.1 h1:9W30zRlYrefrDV2JE2O8VDtJ1yPGownxciz5rrbQZis=
|
||||||
modernc.org/cc/v4 v4.27.1/go.mod h1:uVtb5OGqUKpoLWhqwNQo/8LwvoiEBLvZXIQ/SmO6mL0=
|
modernc.org/cc/v4 v4.27.1/go.mod h1:uVtb5OGqUKpoLWhqwNQo/8LwvoiEBLvZXIQ/SmO6mL0=
|
||||||
modernc.org/ccgo/v4 v4.32.0 h1:hjG66bI/kqIPX1b2yT6fr/jt+QedtP2fqojG2VrFuVw=
|
modernc.org/ccgo/v4 v4.32.0 h1:hjG66bI/kqIPX1b2yT6fr/jt+QedtP2fqojG2VrFuVw=
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,40 @@
|
||||||
|
package apitypes
|
||||||
|
|
||||||
|
type Message struct {
|
||||||
|
Role string
|
||||||
|
Content string
|
||||||
|
}
|
||||||
|
|
||||||
|
type Tool struct {
|
||||||
|
Type string
|
||||||
|
Function ToolFunction
|
||||||
|
}
|
||||||
|
|
||||||
|
type ToolFunction struct {
|
||||||
|
Name string
|
||||||
|
Description string
|
||||||
|
Parameters interface{}
|
||||||
|
}
|
||||||
|
|
||||||
|
type ToolCall struct {
|
||||||
|
ID string
|
||||||
|
Name string
|
||||||
|
Arguments string
|
||||||
|
}
|
||||||
|
|
||||||
|
type StreamChunk struct {
|
||||||
|
Type ChunkType
|
||||||
|
Text string
|
||||||
|
Thinking string
|
||||||
|
ToolCall *ToolCall
|
||||||
|
Done bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type ChunkType int
|
||||||
|
|
||||||
|
const (
|
||||||
|
ChunkText ChunkType = iota
|
||||||
|
ChunkThinking
|
||||||
|
ChunkToolCall
|
||||||
|
ChunkDone
|
||||||
|
)
|
||||||
|
|
@ -11,6 +11,7 @@ type BridgeConfig struct {
|
||||||
RequiredKey string
|
RequiredKey string
|
||||||
DefaultModel string
|
DefaultModel string
|
||||||
Mode string
|
Mode string
|
||||||
|
Provider string
|
||||||
Force bool
|
Force bool
|
||||||
ApproveMcps bool
|
ApproveMcps bool
|
||||||
StrictModel bool
|
StrictModel bool
|
||||||
|
|
@ -25,6 +26,9 @@ type BridgeConfig struct {
|
||||||
ConfigDirs []string
|
ConfigDirs []string
|
||||||
MultiPort bool
|
MultiPort bool
|
||||||
WinCmdlineMax int
|
WinCmdlineMax int
|
||||||
|
GeminiAccountDir string
|
||||||
|
GeminiBrowserVisible bool
|
||||||
|
GeminiMaxSessions int
|
||||||
}
|
}
|
||||||
|
|
||||||
func LoadBridgeConfig(e env.EnvSource, cwd string) BridgeConfig {
|
func LoadBridgeConfig(e env.EnvSource, cwd string) BridgeConfig {
|
||||||
|
|
@ -36,6 +40,7 @@ func LoadBridgeConfig(e env.EnvSource, cwd string) BridgeConfig {
|
||||||
RequiredKey: loaded.RequiredKey,
|
RequiredKey: loaded.RequiredKey,
|
||||||
DefaultModel: loaded.DefaultModel,
|
DefaultModel: loaded.DefaultModel,
|
||||||
Mode: "ask",
|
Mode: "ask",
|
||||||
|
Provider: loaded.Provider,
|
||||||
Force: loaded.Force,
|
Force: loaded.Force,
|
||||||
ApproveMcps: loaded.ApproveMcps,
|
ApproveMcps: loaded.ApproveMcps,
|
||||||
StrictModel: loaded.StrictModel,
|
StrictModel: loaded.StrictModel,
|
||||||
|
|
@ -50,5 +55,8 @@ func LoadBridgeConfig(e env.EnvSource, cwd string) BridgeConfig {
|
||||||
ConfigDirs: loaded.ConfigDirs,
|
ConfigDirs: loaded.ConfigDirs,
|
||||||
MultiPort: loaded.MultiPort,
|
MultiPort: loaded.MultiPort,
|
||||||
WinCmdlineMax: loaded.WinCmdlineMax,
|
WinCmdlineMax: loaded.WinCmdlineMax,
|
||||||
|
GeminiAccountDir: loaded.GeminiAccountDir,
|
||||||
|
GeminiBrowserVisible: loaded.GeminiBrowserVisible,
|
||||||
|
GeminiMaxSessions: loaded.GeminiMaxSessions,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -20,6 +20,7 @@ type LoadedEnv struct {
|
||||||
Port int
|
Port int
|
||||||
RequiredKey string
|
RequiredKey string
|
||||||
DefaultModel string
|
DefaultModel string
|
||||||
|
Provider string
|
||||||
Force bool
|
Force bool
|
||||||
ApproveMcps bool
|
ApproveMcps bool
|
||||||
StrictModel bool
|
StrictModel bool
|
||||||
|
|
@ -34,6 +35,9 @@ type LoadedEnv struct {
|
||||||
ConfigDirs []string
|
ConfigDirs []string
|
||||||
MultiPort bool
|
MultiPort bool
|
||||||
WinCmdlineMax int
|
WinCmdlineMax int
|
||||||
|
GeminiAccountDir string
|
||||||
|
GeminiBrowserVisible bool
|
||||||
|
GeminiMaxSessions int
|
||||||
}
|
}
|
||||||
|
|
||||||
type AgentCommand struct {
|
type AgentCommand struct {
|
||||||
|
|
@ -256,6 +260,13 @@ func LoadEnvConfig(e EnvSource, cwd string) LoadedEnv {
|
||||||
workspace = cwd
|
workspace = cwd
|
||||||
}
|
}
|
||||||
|
|
||||||
|
geminiAccountDir := getEnvVal(e, []string{"GEMINI_ACCOUNT_DIR"})
|
||||||
|
if geminiAccountDir == "" {
|
||||||
|
geminiAccountDir = filepath.Join(home, ".cursor-api-proxy", "gemini-accounts")
|
||||||
|
} else {
|
||||||
|
geminiAccountDir = resolveAbs(geminiAccountDir, cwd)
|
||||||
|
}
|
||||||
|
|
||||||
return LoadedEnv{
|
return LoadedEnv{
|
||||||
AgentBin: agentBin,
|
AgentBin: agentBin,
|
||||||
AgentNode: getEnvVal(e, []string{"CURSOR_AGENT_NODE"}),
|
AgentNode: getEnvVal(e, []string{"CURSOR_AGENT_NODE"}),
|
||||||
|
|
@ -265,6 +276,7 @@ func LoadEnvConfig(e EnvSource, cwd string) LoadedEnv {
|
||||||
Port: port,
|
Port: port,
|
||||||
RequiredKey: getEnvVal(e, []string{"CURSOR_BRIDGE_API_KEY"}),
|
RequiredKey: getEnvVal(e, []string{"CURSOR_BRIDGE_API_KEY"}),
|
||||||
DefaultModel: normalizeModelId(getEnvVal(e, []string{"CURSOR_BRIDGE_DEFAULT_MODEL"})),
|
DefaultModel: normalizeModelId(getEnvVal(e, []string{"CURSOR_BRIDGE_DEFAULT_MODEL"})),
|
||||||
|
Provider: getEnvVal(e, []string{"CURSOR_BRIDGE_PROVIDER"}),
|
||||||
Force: envBool(e, []string{"CURSOR_BRIDGE_FORCE"}, false),
|
Force: envBool(e, []string{"CURSOR_BRIDGE_FORCE"}, false),
|
||||||
ApproveMcps: envBool(e, []string{"CURSOR_BRIDGE_APPROVE_MCPS"}, false),
|
ApproveMcps: envBool(e, []string{"CURSOR_BRIDGE_APPROVE_MCPS"}, false),
|
||||||
StrictModel: envBool(e, []string{"CURSOR_BRIDGE_STRICT_MODEL"}, true),
|
StrictModel: envBool(e, []string{"CURSOR_BRIDGE_STRICT_MODEL"}, true),
|
||||||
|
|
@ -279,6 +291,9 @@ func LoadEnvConfig(e EnvSource, cwd string) LoadedEnv {
|
||||||
ConfigDirs: configDirs,
|
ConfigDirs: configDirs,
|
||||||
MultiPort: envBool(e, []string{"CURSOR_BRIDGE_MULTI_PORT"}, false),
|
MultiPort: envBool(e, []string{"CURSOR_BRIDGE_MULTI_PORT"}, false),
|
||||||
WinCmdlineMax: winMax,
|
WinCmdlineMax: winMax,
|
||||||
|
GeminiAccountDir: geminiAccountDir,
|
||||||
|
GeminiBrowserVisible: envBool(e, []string{"GEMINI_BROWSER_VISIBLE"}, false),
|
||||||
|
GeminiMaxSessions: envInt(e, []string{"GEMINI_MAX_SESSIONS"}, 3),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,203 @@
|
||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"cursor-api-proxy/internal/apitypes"
|
||||||
|
"cursor-api-proxy/internal/config"
|
||||||
|
"cursor-api-proxy/internal/httputil"
|
||||||
|
"cursor-api-proxy/internal/logger"
|
||||||
|
"cursor-api-proxy/internal/providers/geminiweb"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func HandleGeminiChatCompletions(w http.ResponseWriter, r *http.Request, cfg config.BridgeConfig, rawBody, method, pathname, remoteAddress string) {
|
||||||
|
_ = context.Background() // 確保 context 被使用
|
||||||
|
var bodyMap map[string]interface{}
|
||||||
|
if err := json.Unmarshal([]byte(rawBody), &bodyMap); err != nil {
|
||||||
|
httputil.WriteJSON(w, 400, map[string]interface{}{
|
||||||
|
"error": map[string]string{"message": "invalid JSON body", "code": "bad_request"},
|
||||||
|
}, nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
rawModel, _ := bodyMap["model"].(string)
|
||||||
|
if rawModel == "" {
|
||||||
|
rawModel = "gemini-2.0-flash"
|
||||||
|
}
|
||||||
|
|
||||||
|
var messages []interface{}
|
||||||
|
if m, ok := bodyMap["messages"].([]interface{}); ok {
|
||||||
|
messages = m
|
||||||
|
}
|
||||||
|
|
||||||
|
isStream := false
|
||||||
|
if s, ok := bodyMap["stream"].(bool); ok {
|
||||||
|
isStream = s
|
||||||
|
}
|
||||||
|
|
||||||
|
// 轉換 messages 為 apitypes.Message
|
||||||
|
var apiMessages []apitypes.Message
|
||||||
|
for _, m := range messages {
|
||||||
|
if msgMap, ok := m.(map[string]interface{}); ok {
|
||||||
|
role, _ := msgMap["role"].(string)
|
||||||
|
content := ""
|
||||||
|
if c, ok := msgMap["content"].(string); ok {
|
||||||
|
content = c
|
||||||
|
}
|
||||||
|
apiMessages = append(apiMessages, apitypes.Message{
|
||||||
|
Role: role,
|
||||||
|
Content: content,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.LogRequestStart(method, pathname, rawModel, cfg.TimeoutMs, isStream)
|
||||||
|
start := time.Now().UnixMilli()
|
||||||
|
|
||||||
|
// 創建 Gemini provider (使用 Playwright)
|
||||||
|
provider, provErr := geminiweb.NewPlaywrightProvider(cfg)
|
||||||
|
if provErr != nil {
|
||||||
|
logger.LogAgentError(cfg.SessionsLogPath, method, pathname, remoteAddress, -1, provErr.Error())
|
||||||
|
httputil.WriteJSON(w, 500, map[string]interface{}{
|
||||||
|
"error": map[string]string{"message": provErr.Error(), "code": "provider_error"},
|
||||||
|
}, nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if isStream {
|
||||||
|
httputil.WriteSSEHeaders(w, nil)
|
||||||
|
flusher, _ := w.(http.Flusher)
|
||||||
|
|
||||||
|
id := "chatcmpl_" + uuid.New().String()
|
||||||
|
created := time.Now().Unix()
|
||||||
|
var accumulated string
|
||||||
|
|
||||||
|
err := provider.Generate(r.Context(), rawModel, apiMessages, nil, func(chunk apitypes.StreamChunk) {
|
||||||
|
if chunk.Type == apitypes.ChunkText {
|
||||||
|
accumulated += chunk.Text
|
||||||
|
respChunk := map[string]interface{}{
|
||||||
|
"id": id,
|
||||||
|
"object": "chat.completion.chunk",
|
||||||
|
"created": created,
|
||||||
|
"model": rawModel,
|
||||||
|
"choices": []map[string]interface{}{
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"delta": map[string]string{"content": chunk.Text},
|
||||||
|
"finish_reason": nil,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
data, _ := json.Marshal(respChunk)
|
||||||
|
fmt.Fprintf(w, "data: %s\n\n", data)
|
||||||
|
if flusher != nil {
|
||||||
|
flusher.Flush()
|
||||||
|
}
|
||||||
|
} else if chunk.Type == apitypes.ChunkThinking {
|
||||||
|
respChunk := map[string]interface{}{
|
||||||
|
"id": id,
|
||||||
|
"object": "chat.completion.chunk",
|
||||||
|
"created": created,
|
||||||
|
"model": rawModel,
|
||||||
|
"choices": []map[string]interface{}{
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"delta": map[string]interface{}{"reasoning_content": chunk.Thinking},
|
||||||
|
"finish_reason": nil,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
data, _ := json.Marshal(respChunk)
|
||||||
|
fmt.Fprintf(w, "data: %s\n\n", data)
|
||||||
|
if flusher != nil {
|
||||||
|
flusher.Flush()
|
||||||
|
}
|
||||||
|
} else if chunk.Type == apitypes.ChunkDone {
|
||||||
|
stopChunk := map[string]interface{}{
|
||||||
|
"id": id,
|
||||||
|
"object": "chat.completion.chunk",
|
||||||
|
"created": created,
|
||||||
|
"model": rawModel,
|
||||||
|
"choices": []map[string]interface{}{
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"delta": map[string]interface{}{},
|
||||||
|
"finish_reason": "stop",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
data, _ := json.Marshal(stopChunk)
|
||||||
|
fmt.Fprintf(w, "data: %s\n\n", data)
|
||||||
|
fmt.Fprintf(w, "data: [DONE]\n\n")
|
||||||
|
if flusher != nil {
|
||||||
|
flusher.Flush()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
latencyMs := time.Now().UnixMilli() - start
|
||||||
|
if err != nil {
|
||||||
|
logger.LogAgentError(cfg.SessionsLogPath, method, pathname, remoteAddress, -1, err.Error())
|
||||||
|
logger.LogRequestDone(method, pathname, rawModel, latencyMs, -1)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.LogTrafficResponse(cfg.Verbose, rawModel, accumulated, true)
|
||||||
|
logger.LogRequestDone(method, pathname, rawModel, latencyMs, 0)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 非串流模式
|
||||||
|
var resultText string
|
||||||
|
var resultThinking string
|
||||||
|
|
||||||
|
err := provider.Generate(r.Context(), rawModel, apiMessages, nil, func(chunk apitypes.StreamChunk) {
|
||||||
|
if chunk.Type == apitypes.ChunkText {
|
||||||
|
resultText += chunk.Text
|
||||||
|
} else if chunk.Type == apitypes.ChunkThinking {
|
||||||
|
resultThinking += chunk.Thinking
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
latencyMs := time.Now().UnixMilli() - start
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
logger.LogAgentError(cfg.SessionsLogPath, method, pathname, remoteAddress, -1, err.Error())
|
||||||
|
logger.LogRequestDone(method, pathname, rawModel, latencyMs, -1)
|
||||||
|
httputil.WriteJSON(w, 500, map[string]interface{}{
|
||||||
|
"error": map[string]string{"message": err.Error(), "code": "gemini_error"},
|
||||||
|
}, nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.LogTrafficResponse(cfg.Verbose, rawModel, resultText, false)
|
||||||
|
logger.LogRequestDone(method, pathname, rawModel, latencyMs, 0)
|
||||||
|
|
||||||
|
id := "chatcmpl_" + uuid.New().String()
|
||||||
|
created := time.Now().Unix()
|
||||||
|
|
||||||
|
resp := map[string]interface{}{
|
||||||
|
"id": id,
|
||||||
|
"object": "chat.completion",
|
||||||
|
"created": created,
|
||||||
|
"model": rawModel,
|
||||||
|
"choices": []map[string]interface{}{
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"message": map[string]interface{}{
|
||||||
|
"role": "assistant",
|
||||||
|
"content": resultText,
|
||||||
|
},
|
||||||
|
"finish_reason": "stop",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"usage": map[string]int{"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0},
|
||||||
|
}
|
||||||
|
|
||||||
|
httputil.WriteJSON(w, 200, resp, nil)
|
||||||
|
}
|
||||||
|
|
@ -71,18 +71,29 @@ func LogDebug(format string, args ...interface{}) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func LogServerStart(version, scheme, host string, port int, cfg config.BridgeConfig) {
|
func LogServerStart(version, scheme, host string, port int, cfg config.BridgeConfig) {
|
||||||
|
provider := cfg.Provider
|
||||||
|
if provider == "" {
|
||||||
|
provider = "cursor"
|
||||||
|
}
|
||||||
fmt.Printf("\n%s%s╔══════════════════════════════════════════╗%s\n", cBold, cBCyan, cReset)
|
fmt.Printf("\n%s%s╔══════════════════════════════════════════╗%s\n", cBold, cBCyan, cReset)
|
||||||
fmt.Printf("%s%s cursor-api-proxy %sv%s%s%s%s ready%s\n",
|
fmt.Printf("%s%s cursor-api-proxy %sv%s%s%s%s ready%s\n",
|
||||||
cBold, cBCyan, cReset, cBold, cWhite, version, cBCyan, cReset)
|
cBold, cBCyan, cReset, cBold, cWhite, version, cBCyan, cReset)
|
||||||
fmt.Printf("%s%s╚══════════════════════════════════════════╝%s\n\n", cBold, cBCyan, cReset)
|
fmt.Printf("%s%s╚══════════════════════════════════════════╝%s\n\n", cBold, cBCyan, cReset)
|
||||||
url := fmt.Sprintf("%s://%s:%d", scheme, host, port)
|
url := fmt.Sprintf("%s://%s:%d", scheme, host, port)
|
||||||
fmt.Printf(" %s●%s listening %s%s%s\n", cBGreen, cReset, cBold, url, cReset)
|
fmt.Printf(" %s●%s listening %s%s%s\n", cBGreen, cReset, cBold, url, cReset)
|
||||||
|
fmt.Printf(" %s▸%s provider %s%s%s\n", cCyan, cReset, cBold, provider, cReset)
|
||||||
fmt.Printf(" %s▸%s agent %s%s%s\n", cCyan, cReset, cDim, cfg.AgentBin, cReset)
|
fmt.Printf(" %s▸%s agent %s%s%s\n", cCyan, cReset, cDim, cfg.AgentBin, cReset)
|
||||||
fmt.Printf(" %s▸%s workspace %s%s%s\n", cCyan, cReset, cDim, cfg.Workspace, cReset)
|
fmt.Printf(" %s▸%s workspace %s%s%s\n", cCyan, cReset, cDim, cfg.Workspace, cReset)
|
||||||
fmt.Printf(" %s▸%s model %s%s%s\n", cCyan, cReset, cDim, cfg.DefaultModel, cReset)
|
fmt.Printf(" %s▸%s model %s%s%s\n", cCyan, cReset, cDim, cfg.DefaultModel, cReset)
|
||||||
fmt.Printf(" %s▸%s mode %s%s%s\n", cCyan, cReset, cDim, cfg.Mode, cReset)
|
fmt.Printf(" %s▸%s mode %s%s%s\n", cCyan, cReset, cDim, cfg.Mode, cReset)
|
||||||
fmt.Printf(" %s▸%s timeout %s%d ms%s\n", cCyan, cReset, cDim, cfg.TimeoutMs, cReset)
|
fmt.Printf(" %s▸%s timeout %s%d ms%s\n", cCyan, cReset, cDim, cfg.TimeoutMs, cReset)
|
||||||
|
|
||||||
|
// 顯示 Gemini Web Provider 相關設定
|
||||||
|
if provider == "gemini-web" {
|
||||||
|
fmt.Printf(" %s▸%s gemini-dir %s%s%s\n", cCyan, cReset, cDim, cfg.GeminiAccountDir, cReset)
|
||||||
|
fmt.Printf(" %s▸%s max-sess %s%d%s\n", cCyan, cReset, cDim, cfg.GeminiMaxSessions, cReset)
|
||||||
|
}
|
||||||
|
|
||||||
flags := []string{}
|
flags := []string{}
|
||||||
if cfg.Force {
|
if cfg.Force {
|
||||||
flags = append(flags, "force")
|
flags = append(flags, "force")
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,27 @@
|
||||||
|
package cursor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"cursor-api-proxy/internal/apitypes"
|
||||||
|
"cursor-api-proxy/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Provider struct {
|
||||||
|
cfg config.BridgeConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewProvider(cfg config.BridgeConfig) *Provider {
|
||||||
|
return &Provider{cfg: cfg}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Provider) Name() string {
|
||||||
|
return "cursor"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Provider) Close() error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Provider) Generate(ctx context.Context, model string, messages []apitypes.Message, tools []apitypes.Tool, cb func(apitypes.StreamChunk)) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,32 @@
|
||||||
|
package providers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"cursor-api-proxy/internal/apitypes"
|
||||||
|
"cursor-api-proxy/internal/config"
|
||||||
|
"cursor-api-proxy/internal/providers/cursor"
|
||||||
|
"cursor-api-proxy/internal/providers/geminiweb"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Provider interface {
|
||||||
|
Name() string
|
||||||
|
Close() error
|
||||||
|
Generate(ctx context.Context, model string, messages []apitypes.Message, tools []apitypes.Tool, cb func(apitypes.StreamChunk)) error
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewProvider(cfg config.BridgeConfig) (Provider, error) {
|
||||||
|
providerType := cfg.Provider
|
||||||
|
if providerType == "" {
|
||||||
|
providerType = "cursor"
|
||||||
|
}
|
||||||
|
|
||||||
|
switch providerType {
|
||||||
|
case "cursor":
|
||||||
|
return cursor.NewProvider(cfg), nil
|
||||||
|
case "gemini-web":
|
||||||
|
return geminiweb.NewPlaywrightProvider(cfg)
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unknown provider: %s", providerType)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,125 @@
|
||||||
|
package geminiweb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/go-rod/rod"
|
||||||
|
"github.com/go-rod/rod/lib/launcher"
|
||||||
|
"github.com/go-rod/rod/lib/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Browser struct {
|
||||||
|
browser *rod.Browser
|
||||||
|
visible bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewBrowser(visible bool) (*Browser, error) {
|
||||||
|
l := launcher.New()
|
||||||
|
if visible {
|
||||||
|
l = l.Headless(false)
|
||||||
|
} else {
|
||||||
|
l = l.Headless(true)
|
||||||
|
}
|
||||||
|
|
||||||
|
url, err := l.Launch()
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to launch browser: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
b := rod.New().ControlURL(url)
|
||||||
|
if err := b.Connect(); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to connect browser: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Browser{browser: b, visible: visible}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Browser) Close() error {
|
||||||
|
if b.browser != nil {
|
||||||
|
return b.browser.Close()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Browser) NewPage() (*rod.Page, error) {
|
||||||
|
return b.browser.Page(proto.TargetCreateTarget{URL: "about:blank"})
|
||||||
|
}
|
||||||
|
|
||||||
|
type Cookie struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Value string `json:"value"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
Expires float64 `json:"expires"`
|
||||||
|
HTTPOnly bool `json:"httpOnly"`
|
||||||
|
Secure bool `json:"secure"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadCookiesFromFile(cookieFile string) ([]Cookie, error) {
|
||||||
|
data, err := os.ReadFile(cookieFile)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to read cookies: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var cookies []Cookie
|
||||||
|
if err := json.Unmarshal(data, &cookies); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to parse cookies: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return cookies, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func SaveCookiesToFile(cookies []Cookie, cookieFile string) error {
|
||||||
|
data, err := json.MarshalIndent(cookies, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to marshal cookies: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
dir := filepath.Dir(cookieFile)
|
||||||
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||||
|
return fmt.Errorf("failed to create cookie dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := os.WriteFile(cookieFile, data, 0644); err != nil {
|
||||||
|
return fmt.Errorf("failed to write cookies: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetCookiesOnPage(page *rod.Page, cookies []Cookie) error {
|
||||||
|
var protoCookies []*proto.NetworkCookieParam
|
||||||
|
for _, c := range cookies {
|
||||||
|
p := &proto.NetworkCookieParam{
|
||||||
|
Name: c.Name,
|
||||||
|
Value: c.Value,
|
||||||
|
Domain: c.Domain,
|
||||||
|
Path: c.Path,
|
||||||
|
HTTPOnly: c.HTTPOnly,
|
||||||
|
Secure: c.Secure,
|
||||||
|
}
|
||||||
|
if c.Expires > 0 {
|
||||||
|
exp := proto.TimeSinceEpoch(c.Expires)
|
||||||
|
p.Expires = exp
|
||||||
|
}
|
||||||
|
protoCookies = append(protoCookies, p)
|
||||||
|
}
|
||||||
|
return page.SetCookies(protoCookies)
|
||||||
|
}
|
||||||
|
|
||||||
|
func WaitForElement(page *rod.Page, selector string, timeout time.Duration) (*rod.Element, error) {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
||||||
|
defer cancel()
|
||||||
|
return page.Context(ctx).Element(selector)
|
||||||
|
}
|
||||||
|
|
||||||
|
func WaitForElements(page *rod.Page, selector string, timeout time.Duration) (rod.Elements, error) {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
||||||
|
defer cancel()
|
||||||
|
return page.Context(ctx).Elements(selector)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,173 @@
|
||||||
|
package geminiweb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/go-rod/rod"
|
||||||
|
"github.com/go-rod/rod/lib/launcher"
|
||||||
|
"github.com/go-rod/rod/lib/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
// BrowserManager 管理瀏覽器實例的生命週期
|
||||||
|
type BrowserManager struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
browser *rod.Browser
|
||||||
|
userDataDir string
|
||||||
|
page *rod.Page
|
||||||
|
visible bool
|
||||||
|
isRunning bool
|
||||||
|
currentModel string
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
globalManager *BrowserManager
|
||||||
|
globalMu sync.Mutex
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetBrowserManager 獲取全域瀏覽器管理器(單例)
|
||||||
|
func GetBrowserManager(userDataDir string, visible bool) (*BrowserManager, error) {
|
||||||
|
globalMu.Lock()
|
||||||
|
defer globalMu.Unlock()
|
||||||
|
|
||||||
|
if globalManager != nil {
|
||||||
|
return globalManager, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
manager, err := NewBrowserManager(userDataDir, visible)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
globalManager = manager
|
||||||
|
return globalManager, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewBrowserManager 建立新的瀏覽器管理器
|
||||||
|
func NewBrowserManager(userDataDir string, visible bool) (*BrowserManager, error) {
|
||||||
|
cleanLockFiles(userDataDir)
|
||||||
|
|
||||||
|
if err := os.MkdirAll(userDataDir, 0755); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to create user data dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &BrowserManager{
|
||||||
|
userDataDir: userDataDir,
|
||||||
|
visible: visible,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// cleanLockFiles 清理 Chrome 的殘留鎖檔案
|
||||||
|
func cleanLockFiles(userDataDir string) {
|
||||||
|
lockFiles := []string{
|
||||||
|
"SingletonLock",
|
||||||
|
"SingletonCookie",
|
||||||
|
"SingletonSocket",
|
||||||
|
"Default/SingletonLock",
|
||||||
|
"Default/SingletonCookie",
|
||||||
|
"Default/SingletonSocket",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, file := range lockFiles {
|
||||||
|
path := filepath.Join(userDataDir, file)
|
||||||
|
os.Remove(path)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Launch 啟動瀏覽器(如果尚未啟動)
|
||||||
|
func (m *BrowserManager) Launch() error {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
|
||||||
|
if m.isRunning && m.browser != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
l := launcher.New()
|
||||||
|
|
||||||
|
if m.visible {
|
||||||
|
l = l.Headless(false)
|
||||||
|
} else {
|
||||||
|
l = l.Headless(true)
|
||||||
|
}
|
||||||
|
|
||||||
|
l = l.UserDataDir(m.userDataDir)
|
||||||
|
|
||||||
|
url, err := l.Launch()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to launch browser: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
b := rod.New().ControlURL(url)
|
||||||
|
if err := b.Connect(); err != nil {
|
||||||
|
return fmt.Errorf("failed to connect browser: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
m.browser = b
|
||||||
|
|
||||||
|
page, err := b.Page(proto.TargetCreateTarget{URL: "about:blank"})
|
||||||
|
if err != nil {
|
||||||
|
_ = b.Close()
|
||||||
|
return fmt.Errorf("failed to create page: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
m.page = page
|
||||||
|
m.isRunning = true
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetPage 獲取頁面
|
||||||
|
func (m *BrowserManager) GetPage() (*rod.Page, error) {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
|
||||||
|
if !m.isRunning || m.browser == nil {
|
||||||
|
return nil, fmt.Errorf("browser not running")
|
||||||
|
}
|
||||||
|
|
||||||
|
return m.page, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close 關閉瀏覽器
|
||||||
|
func (m *BrowserManager) Close() error {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
|
||||||
|
if !m.isRunning {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
if m.browser != nil {
|
||||||
|
err = m.browser.Close()
|
||||||
|
m.browser = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
m.page = nil
|
||||||
|
m.isRunning = false
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsRunning 檢查瀏覽器是否正在運行
|
||||||
|
func (m *BrowserManager) IsRunning() bool {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
return m.isRunning
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetCurrentModel 設定當前模型
|
||||||
|
func (m *BrowserManager) SetCurrentModel(model string) {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
m.currentModel = model
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetCurrentModel 獲取當前模型
|
||||||
|
func (m *BrowserManager) GetCurrentModel() string {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
return m.currentModel
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,250 @@
|
||||||
|
package geminiweb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/go-rod/rod"
|
||||||
|
)
|
||||||
|
|
||||||
|
const geminiURL = "https://gemini.google.com/app"
|
||||||
|
|
||||||
|
// 輸入框選擇器(依優先順序)
|
||||||
|
var inputSelectors = []string{
|
||||||
|
".ProseMirror",
|
||||||
|
"rich-textarea",
|
||||||
|
"div[role='textbox'][contenteditable='true']",
|
||||||
|
"div[contenteditable='true']",
|
||||||
|
"textarea",
|
||||||
|
}
|
||||||
|
|
||||||
|
// NavigateToGemini 導航到 Gemini
|
||||||
|
func NavigateToGemini(page *rod.Page) error {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
if err := page.Context(ctx).Navigate(geminiURL); err != nil {
|
||||||
|
return fmt.Errorf("failed to navigate: %w", err)
|
||||||
|
}
|
||||||
|
return page.Context(ctx).WaitLoad()
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsLoggedIn 檢查是否已登入
|
||||||
|
func IsLoggedIn(page *rod.Page) bool {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
for _, sel := range inputSelectors {
|
||||||
|
if _, err := page.Context(ctx).Element(sel); err == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// SelectModel 選擇模型(可選)
|
||||||
|
func SelectModel(page *rod.Page, model string) error {
|
||||||
|
fmt.Printf("[GeminiWeb] Model selection skipped (using current model)\n")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// TypeInput 在輸入框中輸入文字
|
||||||
|
func TypeInput(page *rod.Page, text string) error {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
fmt.Println("[GeminiWeb] Looking for input field...")
|
||||||
|
|
||||||
|
// 1. 嘗試所有選擇器
|
||||||
|
var inputEl *rod.Element
|
||||||
|
var err error
|
||||||
|
|
||||||
|
for _, sel := range inputSelectors {
|
||||||
|
fmt.Printf(" Trying: %s\n", sel)
|
||||||
|
inputEl, err = page.Context(ctx).Element(sel)
|
||||||
|
if err == nil {
|
||||||
|
fmt.Printf(" ✓ Found with: %s\n", sel)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
// 2. Fallback: 嘗試等待頁面載入完成後重試
|
||||||
|
fmt.Println("[GeminiWeb] Waiting for page to fully load...")
|
||||||
|
time.Sleep(3 * time.Second)
|
||||||
|
|
||||||
|
for _, sel := range inputSelectors {
|
||||||
|
fmt.Printf(" Retrying: %s\n", sel)
|
||||||
|
inputEl, err = page.Context(ctx).Element(sel)
|
||||||
|
if err == nil {
|
||||||
|
fmt.Printf(" ✓ Found with: %s\n", sel)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
// 3. Debug: 印出頁面標題和 URL
|
||||||
|
info, _ := page.Info()
|
||||||
|
fmt.Printf("[GeminiWeb] DEBUG: URL=%s Title=%s\n", info.URL, info.Title)
|
||||||
|
|
||||||
|
// 4. Fallback: 嘗試更通用的選擇器
|
||||||
|
fmt.Println("[GeminiWeb] Trying generic selectors...")
|
||||||
|
genericSelectors := []string{
|
||||||
|
"div[contenteditable]",
|
||||||
|
"[contenteditable]",
|
||||||
|
"textarea",
|
||||||
|
"input[type='text']",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, sel := range genericSelectors {
|
||||||
|
fmt.Printf(" Trying generic: %s\n", sel)
|
||||||
|
inputEl, err = page.Context(ctx).Element(sel)
|
||||||
|
if err == nil {
|
||||||
|
fmt.Printf(" ✓ Found with: %s\n", sel)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
info, _ := page.Info()
|
||||||
|
return fmt.Errorf("input field not found after trying all selectors (URL=%s)", info.URL)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Focus 輸入框
|
||||||
|
fmt.Printf("[GeminiWeb] Focusing input field...\n")
|
||||||
|
if err := inputEl.Focus(); err != nil {
|
||||||
|
return fmt.Errorf("failed to focus input: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
|
||||||
|
// 3. 使用 Input 方法
|
||||||
|
fmt.Printf("[GeminiWeb] Typing %d chars...\n", len(text))
|
||||||
|
if err := inputEl.Input(text); err != nil {
|
||||||
|
return fmt.Errorf("failed to input text: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(200 * time.Millisecond)
|
||||||
|
|
||||||
|
fmt.Println("[GeminiWeb] Input complete")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ClickSend 發送訊息
|
||||||
|
func ClickSend(page *rod.Page) error {
|
||||||
|
// 方法 1: 按 Enter
|
||||||
|
if err := page.Keyboard.Press('\r'); err != nil {
|
||||||
|
return fmt.Errorf("failed to press Enter: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(200 * time.Millisecond)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// WaitForReady 等待頁面空閒
|
||||||
|
func WaitForReady(page *rod.Page) error {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
fmt.Println("[GeminiWeb] Checking if page is ready...")
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
fmt.Println("[GeminiWeb] Page ready check timeout, proceeding anyway")
|
||||||
|
return nil
|
||||||
|
default:
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
|
||||||
|
// 檢查是否有停止按鈕
|
||||||
|
hasStopBtn := false
|
||||||
|
stopBtns, _ := page.Elements("button[aria-label*='Stop'], button[aria-label*='停止']")
|
||||||
|
for _, btn := range stopBtns {
|
||||||
|
visible, _ := btn.Visible()
|
||||||
|
if visible {
|
||||||
|
hasStopBtn = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !hasStopBtn {
|
||||||
|
fmt.Println("[GeminiWeb] Page is ready")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExtractResponse 提取回應文字
|
||||||
|
func ExtractResponse(page *rod.Page) (string, error) {
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
var lastText string
|
||||||
|
lastUpdate := time.Now()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
if lastText != "" {
|
||||||
|
return lastText, nil
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("response timeout")
|
||||||
|
default:
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
|
||||||
|
// 尋找回應文字
|
||||||
|
for _, sel := range responseSelectors {
|
||||||
|
elements, err := page.Elements(sel)
|
||||||
|
if err != nil || len(elements) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// 取得最後一個元素的文字
|
||||||
|
lastEl := elements[len(elements)-1]
|
||||||
|
text, err := lastEl.Text()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
text = strings.TrimSpace(text)
|
||||||
|
if text != "" && text != lastText && len(text) > len(lastText) {
|
||||||
|
lastText = text
|
||||||
|
lastUpdate = time.Now()
|
||||||
|
fmt.Printf("[GeminiWeb] Response length: %d\n", len(text))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 檢查是否已完成(2 秒內沒有新內容)
|
||||||
|
if time.Since(lastUpdate) > 2*time.Second && lastText != "" {
|
||||||
|
// 最後檢查一次是否還有停止按鈕
|
||||||
|
hasStopBtn := false
|
||||||
|
stopBtns, _ := page.Elements("button[aria-label*='Stop'], button[aria-label*='停止']")
|
||||||
|
for _, btn := range stopBtns {
|
||||||
|
visible, _ := btn.Visible()
|
||||||
|
if visible {
|
||||||
|
hasStopBtn = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !hasStopBtn {
|
||||||
|
return lastText, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 默認的回應選擇器
|
||||||
|
var responseSelectors = []string{
|
||||||
|
".model-response-text",
|
||||||
|
".message-content",
|
||||||
|
".markdown",
|
||||||
|
".prose",
|
||||||
|
"model-response",
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,641 @@
|
||||||
|
package geminiweb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"cursor-api-proxy/internal/apitypes"
|
||||||
|
"cursor-api-proxy/internal/config"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/playwright-community/playwright-go"
|
||||||
|
)
|
||||||
|
|
||||||
|
// PlaywrightProvider 使用 Playwright 的 Gemini Provider
|
||||||
|
type PlaywrightProvider struct {
|
||||||
|
cfg config.BridgeConfig
|
||||||
|
pw *playwright.Playwright
|
||||||
|
browser playwright.Browser
|
||||||
|
context playwright.BrowserContext
|
||||||
|
page playwright.Page
|
||||||
|
mu sync.Mutex
|
||||||
|
userDataDir string
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
playwrightInstance *playwright.Playwright
|
||||||
|
playwrightOnce sync.Once
|
||||||
|
playwrightErr error
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewPlaywrightProvider 建立新的 Playwright Provider
|
||||||
|
func NewPlaywrightProvider(cfg config.BridgeConfig) (*PlaywrightProvider, error) {
|
||||||
|
// 確保 Playwright 已初始化(單例)
|
||||||
|
playwrightOnce.Do(func() {
|
||||||
|
playwrightInstance, playwrightErr = playwright.Run()
|
||||||
|
if playwrightErr != nil {
|
||||||
|
playwrightErr = fmt.Errorf("failed to run playwright: %w", playwrightErr)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
if playwrightErr != nil {
|
||||||
|
return nil, playwrightErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// 清理 Chrome 鎖檔案
|
||||||
|
userDataDir := filepath.Join(cfg.GeminiAccountDir, "default-session")
|
||||||
|
cleanLockFiles(userDataDir)
|
||||||
|
|
||||||
|
// 確保目錄存在
|
||||||
|
if err := os.MkdirAll(userDataDir, 0755); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to create user data dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &PlaywrightProvider{
|
||||||
|
cfg: cfg,
|
||||||
|
pw: playwrightInstance,
|
||||||
|
userDataDir: userDataDir,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getName 返回 Provider 名稱
|
||||||
|
func (p *PlaywrightProvider) Name() string {
|
||||||
|
return "gemini-web"
|
||||||
|
}
|
||||||
|
|
||||||
|
// launchIfNeeded 如果需要則啟動瀏覽器
|
||||||
|
func (p *PlaywrightProvider) launchIfNeeded() error {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
|
||||||
|
if p.context != nil && p.page != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("[GeminiWeb] Launching Chromium...")
|
||||||
|
|
||||||
|
// 使用 LaunchPersistentContext(自動保存 session)
|
||||||
|
context, err := p.pw.Chromium.LaunchPersistentContext(p.userDataDir,
|
||||||
|
playwright.BrowserTypeLaunchPersistentContextOptions{
|
||||||
|
Headless: playwright.Bool(!p.cfg.GeminiBrowserVisible),
|
||||||
|
Args: []string{
|
||||||
|
"--no-first-run",
|
||||||
|
"--no-default-browser-check",
|
||||||
|
"--disable-background-networking",
|
||||||
|
"--disable-extensions",
|
||||||
|
"--disable-plugins",
|
||||||
|
"--disable-sync",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to launch persistent context: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
p.context = context
|
||||||
|
|
||||||
|
// 取得或建立頁面
|
||||||
|
pages := context.Pages()
|
||||||
|
if len(pages) > 0 {
|
||||||
|
p.page = pages[0]
|
||||||
|
} else {
|
||||||
|
page, err := context.NewPage()
|
||||||
|
if err != nil {
|
||||||
|
_ = context.Close()
|
||||||
|
return fmt.Errorf("failed to create page: %w", err)
|
||||||
|
}
|
||||||
|
p.page = page
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("[GeminiWeb] Browser launched")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate 生成回應
|
||||||
|
func (p *PlaywrightProvider) Generate(ctx context.Context, model string, messages []apitypes.Message, tools []apitypes.Tool, cb func(apitypes.StreamChunk)) (err error) {
|
||||||
|
// 確保在返回錯誤時保存診斷
|
||||||
|
defer func() {
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("[GeminiWeb] Error occurred, saving diagnostics...")
|
||||||
|
_ = p.saveDiagnostics()
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
fmt.Printf("[GeminiWeb] Starting generation with model: %s\n", model)
|
||||||
|
|
||||||
|
// 1. 確保瀏覽器已啟動
|
||||||
|
if err := p.launchIfNeeded(); err != nil {
|
||||||
|
return fmt.Errorf("failed to launch browser: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. 導航到 Gemini(如果需要)
|
||||||
|
currentURL := p.page.URL()
|
||||||
|
if !strings.Contains(currentURL, "gemini.google.com") {
|
||||||
|
fmt.Println("[GeminiWeb] Navigating to Gemini...")
|
||||||
|
if _, err := p.page.Goto("https://gemini.google.com/app", playwright.PageGotoOptions{
|
||||||
|
WaitUntil: playwright.WaitUntilStateDomcontentloaded,
|
||||||
|
Timeout: playwright.Float(60000),
|
||||||
|
}); err != nil {
|
||||||
|
return fmt.Errorf("failed to navigate: %w", err)
|
||||||
|
}
|
||||||
|
// 額外等待 JavaScript 載入
|
||||||
|
fmt.Println("[GeminiWeb] Waiting for page to initialize...")
|
||||||
|
time.Sleep(3 * time.Second)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. 調試模式:等待用戶確認
|
||||||
|
if p.cfg.GeminiBrowserVisible {
|
||||||
|
fmt.Println("\n" + strings.Repeat("=", 70))
|
||||||
|
fmt.Println("🔍 調試模式:瀏覽器已開啟")
|
||||||
|
fmt.Println("請檢查瀏覽器畫面,然後按 ENTER 繼續...")
|
||||||
|
fmt.Println("如果有問題,請查看: /tmp/gemini-debug.*")
|
||||||
|
fmt.Println(strings.Repeat("=", 70))
|
||||||
|
|
||||||
|
var input string
|
||||||
|
fmt.Scanln(&input)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. 等待頁面完全載入(project-golem 策略)
|
||||||
|
fmt.Println("[GeminiWeb] Waiting for page to be ready...")
|
||||||
|
if err := p.waitForPageReady(); err != nil {
|
||||||
|
fmt.Printf("[GeminiWeb] Warning: %v\n", err)
|
||||||
|
|
||||||
|
// 額外調試:輸出頁面 HTML 結構
|
||||||
|
if p.cfg.GeminiBrowserVisible {
|
||||||
|
html, _ := p.page.Content()
|
||||||
|
debugPath := "/tmp/gemini-debug.html"
|
||||||
|
if err := os.WriteFile(debugPath, []byte(html), 0644); err == nil {
|
||||||
|
fmt.Printf("[GeminiWeb] HTML saved to: %s\n", debugPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. 檢查登入狀態
|
||||||
|
fmt.Println("[GeminiWeb] Checking login status...")
|
||||||
|
loggedIn := p.isLoggedIn()
|
||||||
|
if !loggedIn {
|
||||||
|
fmt.Println("[GeminiWeb] Not logged in, continuing anyway")
|
||||||
|
if p.cfg.GeminiBrowserVisible {
|
||||||
|
fmt.Println("\n========================================")
|
||||||
|
fmt.Println("Browser is open. You can:")
|
||||||
|
fmt.Println("1. Log in to Gemini now")
|
||||||
|
fmt.Println("2. Continue without login")
|
||||||
|
fmt.Println("========================================\n")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
fmt.Println("[GeminiWeb] ✓ Logged in")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 5. 選擇模型(如果支援)
|
||||||
|
if err := p.selectModel(model); err != nil {
|
||||||
|
fmt.Printf("[GeminiWeb] Warning: model selection failed: %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 6. 建構提示詞
|
||||||
|
prompt := buildPromptFromMessagesPlaywright(messages)
|
||||||
|
fmt.Printf("[GeminiWeb] Typing prompt (%d chars)...\n", len(prompt))
|
||||||
|
|
||||||
|
// 7. 輸入文字(使用 Playwright 的 Auto-wait)
|
||||||
|
if err := p.typeInput(prompt); err != nil {
|
||||||
|
return fmt.Errorf("failed to type: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 7. 發送訊息
|
||||||
|
fmt.Println("[GeminiWeb] Sending message...")
|
||||||
|
if err := p.sendMessage(); err != nil {
|
||||||
|
return fmt.Errorf("failed to send: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 8. 提取回應
|
||||||
|
fmt.Println("[GeminiWeb] Waiting for response...")
|
||||||
|
response, err := p.extractResponse()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to extract response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 9. 回調
|
||||||
|
cb(apitypes.StreamChunk{Type: apitypes.ChunkText, Text: response})
|
||||||
|
cb(apitypes.StreamChunk{Type: apitypes.ChunkDone, Done: true})
|
||||||
|
|
||||||
|
fmt.Printf("[GeminiWeb] Response complete (%d chars)\n", len(response))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close 關閉 Provider
|
||||||
|
func (p *PlaywrightProvider) Close() error {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
|
||||||
|
if p.context != nil {
|
||||||
|
if err := p.context.Close(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
p.context = nil
|
||||||
|
p.page = nil
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// saveDiagnostics 保存診斷信息
|
||||||
|
func (p *PlaywrightProvider) saveDiagnostics() error {
|
||||||
|
if p.page == nil {
|
||||||
|
return fmt.Errorf("no page available")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 截圖
|
||||||
|
screenshotPath := "/tmp/gemini-debug.png"
|
||||||
|
if _, err := p.page.Screenshot(playwright.PageScreenshotOptions{
|
||||||
|
Path: playwright.String(screenshotPath),
|
||||||
|
}); err == nil {
|
||||||
|
fmt.Printf("[GeminiWeb] Screenshot saved: %s\n", screenshotPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
// HTML
|
||||||
|
htmlPath := "/tmp/gemini-debug.html"
|
||||||
|
if html, err := p.page.Content(); err == nil {
|
||||||
|
if err := os.WriteFile(htmlPath, []byte(html), 0644); err == nil {
|
||||||
|
fmt.Printf("[GeminiWeb] HTML saved: %s\n", htmlPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 輸出頁面信息
|
||||||
|
url := p.page.URL()
|
||||||
|
title, _ := p.page.Title()
|
||||||
|
fmt.Printf("[GeminiWeb] Diagnostics: URL=%s, Title=%s\n", url, title)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// waitForPageReady 等待頁面完全就緒(project-golem 策略)
|
||||||
|
func (p *PlaywrightProvider) waitForPageReady() error {
|
||||||
|
fmt.Println("[GeminiWeb] Checking for ready state...")
|
||||||
|
|
||||||
|
// 1. 等待停止按鈕消失(如果存在)
|
||||||
|
_, _ = p.page.WaitForSelector("button[aria-label*='Stop'], button[aria-label*='停止']", playwright.PageWaitForSelectorOptions{
|
||||||
|
State: playwright.WaitForSelectorStateDetached,
|
||||||
|
Timeout: playwright.Float(5000),
|
||||||
|
})
|
||||||
|
|
||||||
|
// 2. 嘗試多種等待策略
|
||||||
|
inputSelectors := []string{
|
||||||
|
".ql-editor.ql-blank",
|
||||||
|
".ql-editor",
|
||||||
|
"div[contenteditable='true'][role='textbox']",
|
||||||
|
"div[contenteditable='true']",
|
||||||
|
".ProseMirror",
|
||||||
|
"rich-textarea",
|
||||||
|
"textarea",
|
||||||
|
}
|
||||||
|
|
||||||
|
// 策略 A: 等待任一輸入框出現
|
||||||
|
for i, sel := range inputSelectors {
|
||||||
|
fmt.Printf(" [%d/%d] Waiting for: %s\n", i+1, len(inputSelectors), sel)
|
||||||
|
locator := p.page.Locator(sel)
|
||||||
|
if err := locator.WaitFor(playwright.LocatorWaitForOptions{
|
||||||
|
Timeout: playwright.Float(5000),
|
||||||
|
State: playwright.WaitForSelectorStateVisible,
|
||||||
|
}); err == nil {
|
||||||
|
fmt.Printf(" ✓ Input field found: %s\n", sel)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 策略 B: 等待頁面完全載入
|
||||||
|
fmt.Println("[GeminiWeb] Waiting for page load...")
|
||||||
|
time.Sleep(3 * time.Second)
|
||||||
|
|
||||||
|
// 策略 C: 使用 JavaScript 檢查
|
||||||
|
fmt.Println("[GeminiWeb] Checking with JavaScript...")
|
||||||
|
result, err := p.page.Evaluate(`
|
||||||
|
() => {
|
||||||
|
// 檢查所有可能的輸入元素
|
||||||
|
const selectors = [
|
||||||
|
'.ql-editor.ql-blank',
|
||||||
|
'.ql-editor',
|
||||||
|
'div[contenteditable="true"][role="textbox"]',
|
||||||
|
'div[contenteditable="true"]',
|
||||||
|
'.ProseMirror',
|
||||||
|
'rich-textarea',
|
||||||
|
'textarea'
|
||||||
|
];
|
||||||
|
|
||||||
|
for (const sel of selectors) {
|
||||||
|
const el = document.querySelector(sel);
|
||||||
|
if (el) {
|
||||||
|
return {
|
||||||
|
found: true,
|
||||||
|
selector: sel,
|
||||||
|
tagName: el.tagName,
|
||||||
|
className: el.className,
|
||||||
|
visible: el.offsetParent !== null
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return { found: false };
|
||||||
|
}
|
||||||
|
`)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
if m, ok := result.(map[string]interface{}); ok {
|
||||||
|
if found, _ := m["found"].(bool); found {
|
||||||
|
sel, _ := m["selector"].(string)
|
||||||
|
fmt.Printf(" ✓ JavaScript found: %s\n", sel)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 策略 D: 調試模式 - 輸出頁面結構
|
||||||
|
if p.cfg.GeminiBrowserVisible {
|
||||||
|
fmt.Println("[GeminiWeb].dump: Page structure analysis")
|
||||||
|
_, _ = p.page.Evaluate(`
|
||||||
|
() => {
|
||||||
|
const allElements = document.querySelectorAll('*');
|
||||||
|
const inputLike = [];
|
||||||
|
for (const el of allElements) {
|
||||||
|
if (el.contentEditable === 'true' ||
|
||||||
|
el.role === 'textbox' ||
|
||||||
|
el.tagName === 'TEXTAREA' ||
|
||||||
|
el.tagName === 'INPUT') {
|
||||||
|
inputLike.push({
|
||||||
|
tag: el.tagName,
|
||||||
|
class: el.className,
|
||||||
|
id: el.id,
|
||||||
|
role: el.role,
|
||||||
|
contentEditable: el.contentEditable
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
console.log('Input-like elements:', inputLike);
|
||||||
|
}
|
||||||
|
`)
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Errorf("no input field found after all strategies")
|
||||||
|
}
|
||||||
|
|
||||||
|
// isLoggedIn 檢查是否已登入
|
||||||
|
func (p *PlaywrightProvider) isLoggedIn() bool {
|
||||||
|
// 嘗試找輸入框(登入狀態的主要特徵)
|
||||||
|
selectors := []string{
|
||||||
|
".ProseMirror",
|
||||||
|
"rich-textarea",
|
||||||
|
"div[role='textbox']",
|
||||||
|
"div[contenteditable='true']",
|
||||||
|
"textarea",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, sel := range selectors {
|
||||||
|
locator := p.page.Locator(sel)
|
||||||
|
if count, _ := locator.Count(); count > 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// typeInput 輸入文字(使用 Playwright 的 Auto-wait)
|
||||||
|
func (p *PlaywrightProvider) typeInput(text string) error {
|
||||||
|
fmt.Println("[GeminiWeb] Looking for input field...")
|
||||||
|
|
||||||
|
selectors := []string{
|
||||||
|
".ql-editor.ql-blank",
|
||||||
|
".ql-editor",
|
||||||
|
"div[contenteditable='true'][role='textbox']",
|
||||||
|
"div[contenteditable='true']",
|
||||||
|
".ProseMirror",
|
||||||
|
"rich-textarea",
|
||||||
|
"textarea",
|
||||||
|
}
|
||||||
|
|
||||||
|
var inputLocator playwright.Locator
|
||||||
|
var found bool
|
||||||
|
|
||||||
|
for _, sel := range selectors {
|
||||||
|
fmt.Printf(" Trying: %s\n", sel)
|
||||||
|
locator := p.page.Locator(sel)
|
||||||
|
if err := locator.WaitFor(playwright.LocatorWaitForOptions{
|
||||||
|
Timeout: playwright.Float(3000),
|
||||||
|
}); err == nil {
|
||||||
|
inputLocator = locator
|
||||||
|
found = true
|
||||||
|
fmt.Printf(" ✓ Found with: %s\n", sel)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !found {
|
||||||
|
// 錯誤會被 Generate 的 defer 捕獲並保存診斷
|
||||||
|
url := p.page.URL()
|
||||||
|
title, _ := p.page.Title()
|
||||||
|
return fmt.Errorf("input field not found (URL=%s, Title=%s). Diagnostics will be saved to /tmp/", url, title)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Focus 並填充(Playwright 自動等待)
|
||||||
|
fmt.Printf("[GeminiWeb] Typing %d chars...\n", len(text))
|
||||||
|
if err := inputLocator.Fill(text); err != nil {
|
||||||
|
return fmt.Errorf("failed to fill: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("[GeminiWeb] Input complete")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// sendMessage 發送訊息
|
||||||
|
func (p *PlaywrightProvider) sendMessage() error {
|
||||||
|
// 方法 1: 按 Enter(最可靠)
|
||||||
|
if err := p.page.Keyboard().Press("Enter"); err != nil {
|
||||||
|
return fmt.Errorf("failed to press Enter: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(200 * time.Millisecond)
|
||||||
|
|
||||||
|
// 方法 2: 嘗試點擊發送按鈕(補強)
|
||||||
|
_, _ = p.page.Evaluate(`
|
||||||
|
() => {
|
||||||
|
const keywords = ['發送', 'Send', '傳送'];
|
||||||
|
const buttons = Array.from(document.querySelectorAll('button, [role="button"]'));
|
||||||
|
|
||||||
|
for (const btn of buttons) {
|
||||||
|
const text = (btn.innerText || btn.textContent || '').trim();
|
||||||
|
const label = (btn.getAttribute('aria-label') || '').trim();
|
||||||
|
|
||||||
|
// 跳過停止按鈕
|
||||||
|
if (['停止', 'Stop', '中斷'].includes(text) || label.toLowerCase().includes('stop')) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (keywords.some(kw => text.includes(kw) || label.includes(kw))) {
|
||||||
|
btn.click();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
`)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// extractResponse 提取回應
|
||||||
|
func (p *PlaywrightProvider) extractResponse() (string, error) {
|
||||||
|
var lastText string
|
||||||
|
var stableCount int
|
||||||
|
lastUpdate := time.Now()
|
||||||
|
timeout := 120 * time.Second
|
||||||
|
startTime := time.Now()
|
||||||
|
|
||||||
|
for time.Since(startTime) < timeout {
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
|
||||||
|
// 使用 JavaScript 提取回應文字(更精確)
|
||||||
|
result, err := p.page.Evaluate(`
|
||||||
|
() => {
|
||||||
|
// 尋找所有可能的回應容器
|
||||||
|
const selectors = [
|
||||||
|
'model-response',
|
||||||
|
'.model-response',
|
||||||
|
'message-content',
|
||||||
|
'.message-content'
|
||||||
|
];
|
||||||
|
|
||||||
|
for (const sel of selectors) {
|
||||||
|
const el = document.querySelector(sel);
|
||||||
|
if (el) {
|
||||||
|
// 嘗試找markdown內容
|
||||||
|
const markdown = el.querySelector('.markdown, .prose, [class*="markdown"]');
|
||||||
|
if (markdown && markdown.innerText.trim()) {
|
||||||
|
let text = markdown.innerText.trim();
|
||||||
|
// 移除常見的標籤前綴
|
||||||
|
text = text.replace(/^Gemini said\s*\n*/i, '').replace(/^Gemini\s*[::]\s*\n*/i, '').trim();
|
||||||
|
return { text: text, source: sel + ' .markdown' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 嘗試找純文字內容(排除標籤)
|
||||||
|
let textContent = el.innerText.trim();
|
||||||
|
if (textContent) {
|
||||||
|
// 移除常見的標籤前綴
|
||||||
|
textContent = textContent.replace(/^Gemini said\s*\n*/i, '').replace(/^Gemini\s*[::]\s*\n*/i, '').trim();
|
||||||
|
return { text: textContent, source: sel };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return { text: '', source: 'none' };
|
||||||
|
}
|
||||||
|
`)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
if m, ok := result.(map[string]interface{}); ok {
|
||||||
|
text, _ := m["text"].(string)
|
||||||
|
text = strings.TrimSpace(text)
|
||||||
|
|
||||||
|
if text != "" && len(text) > len(lastText) {
|
||||||
|
lastText = text
|
||||||
|
lastUpdate = time.Now()
|
||||||
|
stableCount = 0
|
||||||
|
fmt.Printf("[GeminiWeb] Response: %d chars\n", len(text))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 檢查是否完成(需要連續 3 次穩定)
|
||||||
|
if time.Since(lastUpdate) > 500*time.Millisecond && lastText != "" {
|
||||||
|
stableCount++
|
||||||
|
if stableCount >= 3 {
|
||||||
|
// 最終檢查:停止按鈕是否還存在
|
||||||
|
stopBtn := p.page.Locator("button[aria-label*='Stop'], button[aria-label*='停止'], button[data-test-id='stop-button']")
|
||||||
|
count, _ := stopBtn.Count()
|
||||||
|
|
||||||
|
if count == 0 {
|
||||||
|
fmt.Println("[GeminiWeb] ✓ Response complete")
|
||||||
|
return lastText, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if lastText != "" {
|
||||||
|
fmt.Println("[GeminiWeb] ✓ Response complete (timeout)")
|
||||||
|
return lastText, nil
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("response timeout")
|
||||||
|
}
|
||||||
|
|
||||||
|
// selectModel 選擇 Gemini 模型
|
||||||
|
// Gemini Web 只有三種模型:fast, thinking, pro
|
||||||
|
func (p *PlaywrightProvider) selectModel(model string) error {
|
||||||
|
// 映射模型名稱到 Gemini Web 的模型選擇器
|
||||||
|
modelMap := map[string]string{
|
||||||
|
"fast": "Fast",
|
||||||
|
"thinking": "Thinking",
|
||||||
|
"pro": "Pro",
|
||||||
|
"gemini-fast": "Fast",
|
||||||
|
"gemini-thinking": "Thinking",
|
||||||
|
"gemini-pro": "Pro",
|
||||||
|
"gemini-2.0-fast": "Fast",
|
||||||
|
"gemini-2.0-flash": "Fast", // 相容舊名稱
|
||||||
|
"gemini-2.5-pro": "Pro",
|
||||||
|
"gemini-2.5-pro-thinking": "Thinking",
|
||||||
|
}
|
||||||
|
|
||||||
|
// 從完整模型名稱中提取類型
|
||||||
|
modelType := ""
|
||||||
|
modelLower := strings.ToLower(model)
|
||||||
|
for key, value := range modelMap {
|
||||||
|
if strings.Contains(modelLower, strings.ToLower(key)) || modelLower == strings.ToLower(key) {
|
||||||
|
modelType = value
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if modelType == "" {
|
||||||
|
// 默認使用 Fast
|
||||||
|
fmt.Printf("[GeminiWeb] Unknown model '%s', defaulting to Fast\n", model)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("[GeminiWeb] Selecting model: %s\n", modelType)
|
||||||
|
|
||||||
|
// 點擊模型選擇器
|
||||||
|
modelSelector := p.page.Locator("button[aria-label*='Model'], button[aria-label*='模型'], [data-test-id='model-selector']")
|
||||||
|
if count, _ := modelSelector.Count(); count > 0 {
|
||||||
|
if err := modelSelector.First().Click(); err != nil {
|
||||||
|
fmt.Printf("[GeminiWeb] Warning: could not click model selector: %v\n", err)
|
||||||
|
} else {
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
|
||||||
|
// 選擇對應的模型選項
|
||||||
|
optionSelector := p.page.Locator(fmt.Sprintf("button:has-text('%s'), [role='menuitem']:has-text('%s')", modelType, modelType))
|
||||||
|
if count, _ := optionSelector.Count(); count > 0 {
|
||||||
|
if err := optionSelector.First().Click(); err != nil {
|
||||||
|
fmt.Printf("[GeminiWeb] Warning: could not select model: %v\n", err)
|
||||||
|
} else {
|
||||||
|
fmt.Printf("[GeminiWeb] ✓ Model selected: %s\n", modelType)
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildPromptFromMessages 從訊息列表建構提示詞
|
||||||
|
func buildPromptFromMessagesPlaywright(messages []apitypes.Message) string {
|
||||||
|
var prompt string
|
||||||
|
for _, m := range messages {
|
||||||
|
switch m.Role {
|
||||||
|
case "system":
|
||||||
|
prompt += "System: " + m.Content + "\n\n"
|
||||||
|
case "user":
|
||||||
|
prompt += m.Content + "\n\n"
|
||||||
|
case "assistant":
|
||||||
|
prompt += "Assistant: " + m.Content + "\n\n"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return prompt
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,169 @@
|
||||||
|
package geminiweb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type GeminiSession struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
CookieFile string `json:"cookie_file"`
|
||||||
|
LastUsed int64 `json:"last_used"`
|
||||||
|
ActiveCount int `json:"active_count"`
|
||||||
|
RateLimitEnd int64 `json:"rate_limit_end"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type SessionPool struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
sessions []*GeminiSession
|
||||||
|
dir string
|
||||||
|
maxCount int
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSessionPool(dir string, maxSessions int) (*SessionPool, error) {
|
||||||
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to create session dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
sessions, err := loadSessions(dir)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to load sessions: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &SessionPool{
|
||||||
|
sessions: sessions,
|
||||||
|
dir: dir,
|
||||||
|
maxCount: maxSessions,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadSessions(dir string) ([]*GeminiSession, error) {
|
||||||
|
entries, err := os.ReadDir(dir)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var sessions []*GeminiSession
|
||||||
|
for _, entry := range entries {
|
||||||
|
if !entry.IsDir() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
name := entry.Name()
|
||||||
|
metaPath := filepath.Join(dir, name, "session.json")
|
||||||
|
data, err := os.ReadFile(metaPath)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
var s GeminiSession
|
||||||
|
if err := json.Unmarshal(data, &s); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
sessions = append(sessions, &s)
|
||||||
|
}
|
||||||
|
|
||||||
|
return sessions, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) Count() int {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
return len(p.sessions)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) GetAvailable() *GeminiSession {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
|
||||||
|
now := time.Now().UnixMilli()
|
||||||
|
|
||||||
|
var available []*GeminiSession
|
||||||
|
for _, s := range p.sessions {
|
||||||
|
if s.RateLimitEnd < now {
|
||||||
|
available = append(available, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(available) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var best *GeminiSession
|
||||||
|
for _, s := range available {
|
||||||
|
if best == nil || s.ActiveCount < best.ActiveCount {
|
||||||
|
best = s
|
||||||
|
} else if s.ActiveCount == best.ActiveCount && s.LastUsed < best.LastUsed {
|
||||||
|
best = s
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return best
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) StartSession(s *GeminiSession) {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
s.ActiveCount++
|
||||||
|
s.LastUsed = time.Now().UnixMilli()
|
||||||
|
p.saveSession(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) EndSession(s *GeminiSession) {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
if s.ActiveCount > 0 {
|
||||||
|
s.ActiveCount--
|
||||||
|
}
|
||||||
|
p.saveSession(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) RateLimitSession(s *GeminiSession, durationMs int64) {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
s.RateLimitEnd = time.Now().UnixMilli() + durationMs
|
||||||
|
p.saveSession(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) saveSession(s *GeminiSession) {
|
||||||
|
metaPath := filepath.Join(p.dir, s.Name, "session.json")
|
||||||
|
data, err := json.MarshalIndent(s, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_ = os.WriteFile(metaPath, data, 0644)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) CreateSession(name string) (*GeminiSession, error) {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
|
||||||
|
sessionDir := filepath.Join(p.dir, name)
|
||||||
|
if err := os.MkdirAll(sessionDir, 0755); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to create session dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
s := &GeminiSession{
|
||||||
|
Name: name,
|
||||||
|
CookieFile: filepath.Join(sessionDir, "cookies.json"),
|
||||||
|
LastUsed: time.Now().UnixMilli(),
|
||||||
|
}
|
||||||
|
|
||||||
|
p.sessions = append(p.sessions, s)
|
||||||
|
p.saveSession(s)
|
||||||
|
|
||||||
|
return s, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *SessionPool) GetSessionNames() []string {
|
||||||
|
p.mu.Lock()
|
||||||
|
defer p.mu.Unlock()
|
||||||
|
names := make([]string, len(p.sessions))
|
||||||
|
for i, s := range p.sessions {
|
||||||
|
names[i] = s.Name
|
||||||
|
}
|
||||||
|
return names
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,196 @@
|
||||||
|
package geminiweb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"cursor-api-proxy/internal/apitypes"
|
||||||
|
"cursor-api-proxy/internal/config"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Provider 使用持久化瀏覽器管理器
|
||||||
|
type Provider struct {
|
||||||
|
cfg config.BridgeConfig
|
||||||
|
managerOnce sync.Once
|
||||||
|
manager *BrowserManager
|
||||||
|
managerErr error
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewProvider 建立新的 Provider
|
||||||
|
func NewProvider(cfg config.BridgeConfig) *Provider {
|
||||||
|
return &Provider{cfg: cfg}
|
||||||
|
}
|
||||||
|
|
||||||
|
// getName 返回 Provider 名稱
|
||||||
|
func (p *Provider) Name() string {
|
||||||
|
return "gemini-web"
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close 關閉瀏覽器
|
||||||
|
func (p *Provider) Close() error {
|
||||||
|
if p.manager != nil {
|
||||||
|
return p.manager.Close()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getManager 獲取或初始化瀏覽器管理器(單例)
|
||||||
|
func (p *Provider) getManager() (*BrowserManager, error) {
|
||||||
|
p.managerOnce.Do(func() {
|
||||||
|
sessionDir := p.getSessionDir()
|
||||||
|
p.manager, p.managerErr = GetBrowserManager(sessionDir, p.cfg.GeminiBrowserVisible)
|
||||||
|
})
|
||||||
|
return p.manager, p.managerErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// getSessionDir 獲取 session 目錄
|
||||||
|
func (p *Provider) getSessionDir() string {
|
||||||
|
// 使用單一 session 目錄(簡化設計)
|
||||||
|
return filepath.Join(p.cfg.GeminiAccountDir, "default-session")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate 生成回應
|
||||||
|
func (p *Provider) Generate(ctx context.Context, model string, messages []apitypes.Message, tools []apitypes.Tool, cb func(apitypes.StreamChunk)) error {
|
||||||
|
fmt.Printf("[GeminiWeb] Starting generation with model: %s\n", model)
|
||||||
|
|
||||||
|
// 1. 獲取瀏覽器管理器
|
||||||
|
manager, err := p.getManager()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get browser manager: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. 啟動瀏覽器(如果尚未啟動)
|
||||||
|
if !manager.IsRunning() {
|
||||||
|
fmt.Printf("[GeminiWeb] Launching browser...\n")
|
||||||
|
if err := manager.Launch(); err != nil {
|
||||||
|
return fmt.Errorf("failed to launch browser: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. 獲取頁面
|
||||||
|
page, err := manager.GetPage()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get page: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. 檢查當前 URL,如果不是 Gemini 則導航
|
||||||
|
currentURL, _ := page.Info()
|
||||||
|
if !strings.Contains(currentURL.URL, "gemini.google.com") {
|
||||||
|
fmt.Printf("[GeminiWeb] Navigating to Gemini...\n")
|
||||||
|
if err := NavigateToGemini(page); err != nil {
|
||||||
|
return fmt.Errorf("failed to navigate: %w", err)
|
||||||
|
}
|
||||||
|
time.Sleep(2 * time.Second)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 5. 檢查登入狀態
|
||||||
|
fmt.Printf("[GeminiWeb] Checking login status...\n")
|
||||||
|
if !IsLoggedIn(page) {
|
||||||
|
fmt.Printf("[GeminiWeb] Not logged in, continuing anyway\n")
|
||||||
|
|
||||||
|
if p.cfg.GeminiBrowserVisible {
|
||||||
|
fmt.Println("\n========================================")
|
||||||
|
fmt.Println("Browser is open. You can:")
|
||||||
|
fmt.Println("1. Log in to Gemini now")
|
||||||
|
fmt.Println("2. Continue without login")
|
||||||
|
fmt.Println("========================================\n")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
fmt.Printf("[GeminiWeb] Logged in\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 6. 等待頁面就緒
|
||||||
|
if err := WaitForReady(page); err != nil {
|
||||||
|
fmt.Printf("[GeminiWeb] Warning: %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 7. 建構提示詞
|
||||||
|
prompt := buildPromptFromMessages(messages)
|
||||||
|
fmt.Printf("[GeminiWeb] Typing prompt (%d chars)...\n", len(prompt))
|
||||||
|
|
||||||
|
// 8. 輸入文字
|
||||||
|
if err := TypeInput(page, prompt); err != nil {
|
||||||
|
return fmt.Errorf("failed to type input: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 9. 發送
|
||||||
|
fmt.Printf("[GeminiWeb] Sending message...\n")
|
||||||
|
if err := ClickSend(page); err != nil {
|
||||||
|
return fmt.Errorf("failed to send: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 10. 提取回應
|
||||||
|
fmt.Printf("[GeminiWeb] Waiting for response...\n")
|
||||||
|
response, err := ExtractResponse(page)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to extract response: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 11. 串流回調
|
||||||
|
cb(apitypes.StreamChunk{Type: apitypes.ChunkText, Text: response})
|
||||||
|
cb(apitypes.StreamChunk{Type: apitypes.ChunkDone, Done: true})
|
||||||
|
|
||||||
|
fmt.Printf("[GeminiWeb] Response complete (%d chars)\n", len(response))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// buildPromptFromMessages 從訊息列表建構提示詞
|
||||||
|
func buildPromptFromMessages(messages []apitypes.Message) string {
|
||||||
|
var prompt string
|
||||||
|
for _, m := range messages {
|
||||||
|
switch m.Role {
|
||||||
|
case "system":
|
||||||
|
prompt += "System: " + m.Content + "\n\n"
|
||||||
|
case "user":
|
||||||
|
prompt += m.Content + "\n\n"
|
||||||
|
case "assistant":
|
||||||
|
prompt += "Assistant: " + m.Content + "\n\n"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return prompt
|
||||||
|
}
|
||||||
|
|
||||||
|
// RunLogin 執行登入流程(供 gemini-login 命令使用)
|
||||||
|
func RunLogin(cfg config.BridgeConfig, sessionName string) error {
|
||||||
|
if sessionName == "" {
|
||||||
|
sessionName = "default-session"
|
||||||
|
}
|
||||||
|
|
||||||
|
sessionDir := filepath.Join(cfg.GeminiAccountDir, sessionName)
|
||||||
|
if err := os.MkdirAll(sessionDir, 0755); err != nil {
|
||||||
|
return fmt.Errorf("failed to create session dir: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("Starting browser for login. Session: %s\n", sessionName)
|
||||||
|
fmt.Printf("Session directory: %s\n", sessionDir)
|
||||||
|
fmt.Println("Please log in to your Gemini account in the browser window.")
|
||||||
|
fmt.Println("Press Ctrl+C when you have completed the login...")
|
||||||
|
|
||||||
|
manager, err := NewBrowserManager(sessionDir, true) // visible=true
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to create browser manager: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := manager.Launch(); err != nil {
|
||||||
|
return fmt.Errorf("failed to launch browser: %w", err)
|
||||||
|
}
|
||||||
|
defer manager.Close()
|
||||||
|
|
||||||
|
page, err := manager.GetPage()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get page: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := NavigateToGemini(page); err != nil {
|
||||||
|
return fmt.Errorf("failed to navigate: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 等待用戶手動登入...
|
||||||
|
// 使用 Ctrl+C 退出,瀏覽器資料會自動保存在 userDataDir
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -61,7 +61,16 @@ func NewRouter(opts RouterOptions) http.HandlerFunc {
|
||||||
}, nil)
|
}, nil)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
// 根據 Provider 選擇處理方式
|
||||||
|
provider := cfg.Provider
|
||||||
|
if provider == "" {
|
||||||
|
provider = "cursor"
|
||||||
|
}
|
||||||
|
if provider == "gemini-web" {
|
||||||
|
handlers.HandleGeminiChatCompletions(w, r, cfg, raw, method, pathname, remoteAddress)
|
||||||
|
} else {
|
||||||
handlers.HandleChatCompletions(w, r, cfg, opts.Pool, opts.LastModel, raw, method, pathname, remoteAddress)
|
handlers.HandleChatCompletions(w, r, cfg, opts.Pool, opts.LastModel, raw, method, pathname, remoteAddress)
|
||||||
|
}
|
||||||
|
|
||||||
case method == "POST" && pathname == "/v1/messages":
|
case method == "POST" && pathname == "/v1/messages":
|
||||||
raw, err := httputil.ReadBody(r)
|
raw, err := httputil.ReadBody(r)
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,159 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cursor-api-proxy/internal/providers/geminiweb"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/go-rod/rod"
|
||||||
|
"github.com/go-rod/rod/lib/launcher"
|
||||||
|
"github.com/go-rod/rod/lib/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
fmt.Println("Starting Gemini DOM detection...")
|
||||||
|
fmt.Println("This will open a browser and analyze the Gemini web interface.")
|
||||||
|
fmt.Println()
|
||||||
|
|
||||||
|
// 啟動可見瀏覽器
|
||||||
|
l := launcher.New().Headless(false)
|
||||||
|
url, err := l.Launch()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Failed to launch browser: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
browser := rod.New().ControlURL(url)
|
||||||
|
if err := browser.Connect(); err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Failed to connect browser: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
defer browser.Close()
|
||||||
|
|
||||||
|
page, err := browser.Page(proto.TargetCreateTarget{URL: "about:blank"})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Failed to create page: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 載入 cookies(如果有)
|
||||||
|
home, _ := os.UserHomeDir()
|
||||||
|
cookieFile := home + "/.cursor-api-proxy/gemini-accounts/session-1/cookies.json"
|
||||||
|
if _, err := os.Stat(cookieFile); err == nil {
|
||||||
|
cookies, err := geminiweb.LoadCookiesFromFile(cookieFile)
|
||||||
|
if err == nil {
|
||||||
|
geminiweb.SetCookiesOnPage(page, cookies)
|
||||||
|
fmt.Println("Loaded existing cookies")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 導航到 Gemini
|
||||||
|
fmt.Println("Navigating to gemini.google.com...")
|
||||||
|
if err := geminiweb.NavigateToGemini(page); err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Failed to navigate: %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("Browser is now open. Please:")
|
||||||
|
fmt.Println("1. Log in if needed")
|
||||||
|
fmt.Println("2. Wait for the chat interface to fully load")
|
||||||
|
fmt.Println("3. Look for the model selector dropdown")
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("Press Enter to analyze the DOM...")
|
||||||
|
fmt.Scanln()
|
||||||
|
|
||||||
|
// 分析 DOM
|
||||||
|
analyzeDOM(page)
|
||||||
|
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("Press Enter to close...")
|
||||||
|
fmt.Scanln()
|
||||||
|
}
|
||||||
|
|
||||||
|
func analyzeDOM(page *rod.Page) {
|
||||||
|
fmt.Println("=== DOM Analysis ===")
|
||||||
|
fmt.Println()
|
||||||
|
|
||||||
|
// 尋找可能的輸入框
|
||||||
|
fmt.Println("Looking for input fields...")
|
||||||
|
selectors := []string{
|
||||||
|
`textarea`,
|
||||||
|
`[contenteditable="true"]`,
|
||||||
|
`[role="textbox"]`,
|
||||||
|
`input[type="text"]`,
|
||||||
|
}
|
||||||
|
for _, sel := range selectors {
|
||||||
|
elements, err := page.Elements(sel)
|
||||||
|
if err == nil && len(elements) > 0 {
|
||||||
|
fmt.Printf(" Found %d elements with: %s\n", len(elements), sel)
|
||||||
|
for i, el := range elements {
|
||||||
|
tag, _ := el.Property("tagName")
|
||||||
|
class, _ := el.Attribute("class")
|
||||||
|
ariaLabel, _ := el.Attribute("aria-label")
|
||||||
|
placeholder, _ := el.Attribute("placeholder")
|
||||||
|
fmt.Printf(" [%d] tag=%s class=%s aria-label=%s placeholder=%s\n",
|
||||||
|
i, tag, class, ariaLabel, placeholder)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尋找可能的發送按鈕
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("Looking for send buttons...")
|
||||||
|
buttonSelectors := []string{
|
||||||
|
`button`,
|
||||||
|
`[role="button"]`,
|
||||||
|
`[type="submit"]`,
|
||||||
|
}
|
||||||
|
for _, sel := range buttonSelectors {
|
||||||
|
elements, err := page.Elements(sel)
|
||||||
|
if err == nil && len(elements) > 0 {
|
||||||
|
fmt.Printf(" Found %d elements with: %s\n", len(elements), sel)
|
||||||
|
for i, el := range elements {
|
||||||
|
if i >= 5 {
|
||||||
|
fmt.Printf(" ... and %d more\n", len(elements)-5)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
tag, _ := el.Property("tagName")
|
||||||
|
class, _ := el.Attribute("class")
|
||||||
|
ariaLabel, _ := el.Attribute("aria-label")
|
||||||
|
text, _ := el.Text()
|
||||||
|
text = truncate(text, 30)
|
||||||
|
fmt.Printf(" [%d] tag=%s class=%s aria-label=%s text=%s\n",
|
||||||
|
i, tag, class, ariaLabel, text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尋找模型選擇器
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("Looking for model selector...")
|
||||||
|
modelSelectors := []string{
|
||||||
|
`[aria-label*="model"]`,
|
||||||
|
`[aria-label*="Model"]`,
|
||||||
|
`button[aria-haspopup]`,
|
||||||
|
`[data-test-id*="model"]`,
|
||||||
|
`[class*="model"]`,
|
||||||
|
}
|
||||||
|
for _, sel := range modelSelectors {
|
||||||
|
elements, err := page.Elements(sel)
|
||||||
|
if err == nil && len(elements) > 0 {
|
||||||
|
fmt.Printf(" Found with: %s\n", sel)
|
||||||
|
for i, el := range elements {
|
||||||
|
tag, _ := el.Property("tagName")
|
||||||
|
class, _ := el.Attribute("class")
|
||||||
|
ariaLabel, _ := el.Attribute("aria-label")
|
||||||
|
text, _ := el.Text()
|
||||||
|
fmt.Printf(" [%d] tag=%s class=%s aria-label=%s text=%s\n",
|
||||||
|
i, tag, class, ariaLabel, truncate(text, 30))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func truncate(s string, max int) string {
|
||||||
|
if len(s) <= max {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
return s[:max] + "..."
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue