package gemini

import (
	"context"
	"io"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
	"github.com/router-for-me/CLIProxyAPI/v7/internal/interfaces"
	"github.com/router-for-me/CLIProxyAPI/v7/internal/registry"
	"github.com/router-for-me/CLIProxyAPI/v7/internal/runtime/executor"
	_ "github.com/router-for-me/CLIProxyAPI/v7/internal/translator"
	"github.com/router-for-me/CLIProxyAPI/v7/sdk/api/handlers"
	coreauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"
	sdkconfig "github.com/router-for-me/CLIProxyAPI/v7/sdk/config"
	"github.com/tidwall/gjson"
)

func TestParseInteractionsRequestTarget(t *testing.T) {
	tests := []struct {
		name      string
		body      string
		wantModel string
		wantAgent string
		wantErr   bool
	}{
		{name: "model", body: `{"model":"gemini-3.5-flash","input":"hi"}`, wantModel: "gemini-3.5-flash"},
		{name: "model resource name", body: `{"model":"models/gemini-3.5-flash","input":"hi"}`, wantModel: "models/gemini-3.5-flash"},
		{name: "agent", body: `{"agent":"agents/test-agent","input":"hi"}`, wantAgent: "agents/test-agent"},
		{name: "missing", body: `{"input":"hi"}`, wantErr: true},
		{name: "both", body: `{"model":"gemini-3.5-flash","agent":"agents/test-agent","input":"hi"}`, wantErr: true},
		{name: "stream string", body: `{"model":"gemini-3.5-flash","stream":"true","input":"hi"}`, wantErr: true},
		{name: "stream true", body: `{"model":"gemini-3.5-flash","stream":true,"input":"hi"}`, wantModel: "gemini-3.5-flash"},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			target, errParse := parseInteractionsRequestTarget([]byte(tt.body))
			if tt.wantErr {
				if errParse == nil {
					t.Fatal("parseInteractionsRequestTarget() error = nil, want error")
				}
				return
			}
			if errParse != nil {
				t.Fatalf("parseInteractionsRequestTarget() error = %v", errParse)
			}
			if target.Model != tt.wantModel || target.Agent != tt.wantAgent {
				t.Fatalf("target = %#v, want model %q agent %q", target, tt.wantModel, tt.wantAgent)
			}
		})
	}
}

func TestPrepareInteractionsExecutionTargetNormalizesModelResourceName(t *testing.T) {
	target, errParse := parseInteractionsRequestTarget([]byte(`{"model":"models/gemini-3.5-flash","input":"hi"}`))
	if errParse != nil {
		t.Fatalf("parseInteractionsRequestTarget() error = %v", errParse)
	}
	model, body := prepareInteractionsExecutionTarget([]byte(`{"model":"models/gemini-3.5-flash","input":"hi"}`), target)
	if model != "gemini-3.5-flash" {
		t.Fatalf("model = %q, want gemini-3.5-flash", model)
	}
	if got := gjson.GetBytes(body, "model").String(); got != "gemini-3.5-flash" {
		t.Fatalf("body model = %q, want gemini-3.5-flash. Body: %s", got, string(body))
	}
}

func TestPrepareInteractionsExecutionTargetPreservesBareModel(t *testing.T) {
	target, errParse := parseInteractionsRequestTarget([]byte(`{"model":"gemini-3.5-flash","input":"hi"}`))
	if errParse != nil {
		t.Fatalf("parseInteractionsRequestTarget() error = %v", errParse)
	}
	model, body := prepareInteractionsExecutionTarget([]byte(`{"model":"gemini-3.5-flash","input":"hi"}`), target)
	if model != "gemini-3.5-flash" {
		t.Fatalf("model = %q, want gemini-3.5-flash", model)
	}
	if got := gjson.GetBytes(body, "model").String(); got != "gemini-3.5-flash" {
		t.Fatalf("body model = %q, want gemini-3.5-flash. Body: %s", got, string(body))
	}
}

func TestBuildInteractionsExecutionRequestUsesAgentAuthSelectionModel(t *testing.T) {
	target, errParse := parseInteractionsRequestTarget([]byte(`{"agent":"agents/test-agent","input":"hi"}`))
	if errParse != nil {
		t.Fatalf("parseInteractionsRequestTarget() error = %v", errParse)
	}
	req := buildInteractionsExecutionRequest(target, "agents/test-agent", []byte(`{"agent":"agents/test-agent","input":"hi"}`), "")
	if req.ForcedProvider != "gemini-interactions" {
		t.Fatalf("ForcedProvider = %q, want gemini-interactions", req.ForcedProvider)
	}
	if req.AuthSelectionModel != interactionsAgentAuthSelectionModel {
		t.Fatalf("AuthSelectionModel = %q, want %q", req.AuthSelectionModel, interactionsAgentAuthSelectionModel)
	}
	if req.Model != "agents/test-agent" {
		t.Fatalf("Model = %q, want agents/test-agent", req.Model)
	}
	if got := gjson.GetBytes(req.Body, "agent").String(); got != "agents/test-agent" {
		t.Fatalf("body agent = %q, want agents/test-agent", got)
	}
}

func TestInteractionsRejectsInvalidJSON(t *testing.T) {
	gin.SetMode(gin.TestMode)
	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{`))
	h := NewGeminiAPIHandler(&handlers.BaseAPIHandler{})

	h.Interactions(ctx)

	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want %d; body=%s", rec.Code, http.StatusBadRequest, rec.Body.String())
	}
	if !strings.Contains(rec.Body.String(), "invalid_request_error") {
		t.Fatalf("body = %s, want invalid_request_error", rec.Body.String())
	}
}

func TestInteractionsRejectsMissingModelAndAgent(t *testing.T) {
	gin.SetMode(gin.TestMode)
	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{"input":"hi"}`))
	h := NewGeminiAPIHandler(&handlers.BaseAPIHandler{})

	h.Interactions(ctx)

	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want %d; body=%s", rec.Code, http.StatusBadRequest, rec.Body.String())
	}
	if !strings.Contains(rec.Body.String(), "exactly one of model or agent") {
		t.Fatalf("body = %s, want model/agent validation error", rec.Body.String())
	}
}

func TestInteractionsRejectsBothModelAndAgent(t *testing.T) {
	gin.SetMode(gin.TestMode)
	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{"model":"gemini-3.5-flash","agent":"agents/test-agent","input":"hi"}`))
	h := NewGeminiAPIHandler(&handlers.BaseAPIHandler{})

	h.Interactions(ctx)

	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want %d; body=%s", rec.Code, http.StatusBadRequest, rec.Body.String())
	}
	if !strings.Contains(rec.Body.String(), "exactly one of model or agent") {
		t.Fatalf("body = %s, want model/agent validation error", rec.Body.String())
	}
}

func TestInteractionsRejectsNonBooleanStream(t *testing.T) {
	gin.SetMode(gin.TestMode)
	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{"model":"gemini-3.5-flash","stream":"true","input":"hi"}`))
	h := NewGeminiAPIHandler(&handlers.BaseAPIHandler{})

	h.Interactions(ctx)

	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want %d; body=%s", rec.Code, http.StatusBadRequest, rec.Body.String())
	}
	if !strings.Contains(rec.Body.String(), "invalid_request_error") {
		t.Fatalf("body = %s, want invalid_request_error", rec.Body.String())
	}
}

func TestInteractionsAgentUsesNativeInteractionsEndpoint(t *testing.T) {
	gin.SetMode(gin.TestMode)
	var gotPath string
	var upstreamBody []byte
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		gotPath = r.URL.Path
		body, errRead := io.ReadAll(r.Body)
		if errRead != nil {
			http.Error(w, errRead.Error(), http.StatusBadRequest)
			return
		}
		upstreamBody = body
		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(`{"id":"interaction_1","object":"interaction","status":"completed","steps":[{"type":"model_output","content":[{"text":"ok"}]}],"usage":{"input_tokens":1,"output_tokens":1,"total_tokens":2}}`))
	}))
	defer server.Close()

	manager := coreauth.NewManager(nil, nil, nil)
	manager.RegisterExecutor(executor.NewGeminiInteractionsExecutor(&config.Config{RequestRetry: 1}))
	auth := &coreauth.Auth{
		ID:       "interactions-agent-native-auth",
		Provider: "gemini-interactions",
		Status:   coreauth.StatusActive,
		Attributes: map[string]string{
			"api_key":  "test-key",
			"base_url": server.URL,
		},
		Metadata: map[string]any{"email": "interactions-agent@example.com"},
	}
	if _, errRegister := manager.Register(context.Background(), auth); errRegister != nil {
		t.Fatalf("manager.Register(): %v", errRegister)
	}
	registry.GetGlobalRegistry().RegisterClient(auth.ID, auth.Provider, []*registry.ModelInfo{{ID: interactionsAgentAuthSelectionModel}})
	t.Cleanup(func() {
		registry.GetGlobalRegistry().UnregisterClient(auth.ID)
	})

	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{"agent":"agents/test-agent","input":"hi"}`))
	h := NewGeminiAPIHandler(handlers.NewBaseAPIHandlers(&sdkconfig.SDKConfig{}, manager))

	h.Interactions(ctx)

	if rec.Code != http.StatusOK {
		t.Fatalf("status = %d, want %d; body=%s", rec.Code, http.StatusOK, rec.Body.String())
	}
	if gotPath != "/v1beta/interactions" {
		t.Fatalf("path = %q, want /v1beta/interactions", gotPath)
	}
	if got := gjson.GetBytes(upstreamBody, "agent").String(); got != "agents/test-agent" {
		t.Fatalf("upstream agent = %q, want agents/test-agent. Body: %s", got, string(upstreamBody))
	}
	if got := gjson.GetBytes(rec.Body.Bytes(), "id").String(); got != "interaction_1" {
		t.Fatalf("response id = %q, want interaction_1. Body: %s", got, rec.Body.String())
	}
}

func TestInteractionsAntigravityModelUsesTranslatorBridge(t *testing.T) {
	gin.SetMode(gin.TestMode)
	model := "interactions-antigravity-bridge-model"
	var upstreamBody []byte
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/v1internal:generateContent" {
			http.Error(w, "unexpected path: "+r.URL.Path, http.StatusNotFound)
			return
		}
		body, errRead := io.ReadAll(r.Body)
		if errRead != nil {
			http.Error(w, errRead.Error(), http.StatusBadRequest)
			return
		}
		upstreamBody = body
		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(`{"response":{"responseId":"resp_1","candidates":[{"content":{"role":"model","parts":[{"text":"translated-ok"}]},"finishReason":"STOP"}],"usageMetadata":{"promptTokenCount":1,"candidatesTokenCount":2,"totalTokenCount":3}}}`))
	}))
	defer server.Close()

	manager := coreauth.NewManager(nil, nil, nil)
	manager.RegisterExecutor(executor.NewAntigravityExecutor(&config.Config{RequestRetry: 1}))
	auth := &coreauth.Auth{
		ID:       "interactions-antigravity-bridge-auth",
		Provider: "antigravity",
		Status:   coreauth.StatusActive,
		Attributes: map[string]string{
			"base_url": server.URL,
		},
		Metadata: map[string]any{
			"access_token": "token",
			"project_id":   "project-1",
			"expired":      time.Now().Add(time.Hour).Format(time.RFC3339),
		},
	}
	if _, errRegister := manager.Register(context.Background(), auth); errRegister != nil {
		t.Fatalf("manager.Register(): %v", errRegister)
	}
	registry.GetGlobalRegistry().RegisterClient(auth.ID, auth.Provider, []*registry.ModelInfo{{ID: model}})
	t.Cleanup(func() {
		registry.GetGlobalRegistry().UnregisterClient(auth.ID)
	})

	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{"model":"`+model+`","input":"hi","generation_config":{"top_p":0.8}}`))
	h := NewGeminiAPIHandler(handlers.NewBaseAPIHandlers(&sdkconfig.SDKConfig{}, manager))

	h.Interactions(ctx)

	if rec.Code != http.StatusOK {
		t.Fatalf("status = %d, want %d; body=%s", rec.Code, http.StatusOK, rec.Body.String())
	}
	if gjson.GetBytes(upstreamBody, "input").Exists() {
		t.Fatalf("upstream body still contains raw interactions input: %s", string(upstreamBody))
	}
	if got := gjson.GetBytes(upstreamBody, "request.contents.0.parts.0.text").String(); got != "hi" {
		t.Fatalf("upstream request text = %q, want hi. Body: %s", got, string(upstreamBody))
	}
	if got := gjson.GetBytes(upstreamBody, "request.generationConfig.topP").Float(); got != 0.8 {
		t.Fatalf("upstream topP = %v, want 0.8. Body: %s", got, string(upstreamBody))
	}
	if got := gjson.GetBytes(rec.Body.Bytes(), "steps.0.content.0.text").String(); got != "translated-ok" {
		t.Fatalf("response text = %q, want translated-ok. Body: %s", got, rec.Body.String())
	}
	if gjson.GetBytes(rec.Body.Bytes(), "response").Exists() {
		t.Fatalf("response still contains raw antigravity response wrapper: %s", rec.Body.String())
	}
}

func TestForwardInteractionsStreamWrapsBareJSONAsSSEData(t *testing.T) {
	gin.SetMode(gin.TestMode)
	rec := httptest.NewRecorder()
	ctx, _ := gin.CreateTestContext(rec)
	ctx.Request = httptest.NewRequest(http.MethodPost, "/v1beta/interactions", strings.NewReader(`{}`))
	data := make(chan []byte, 1)
	errs := make(chan *interfaces.ErrorMessage)
	data <- []byte(`{"type":"interaction.completed"}`)
	close(data)
	close(errs)
	h := NewGeminiAPIHandler(&handlers.BaseAPIHandler{})

	h.forwardInteractionsStream(ctx, rec, func(error) {}, data, errs)

	if got := rec.Body.String(); got != "data: {\"type\":\"interaction.completed\"}\n\n" {
		t.Fatalf("body = %q, want SSE data frame", got)
	}
}