package api

import (
	"net/http/httptest"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"codespace/internal/model"
	"codespace/internal/process"
	"codespace/internal/service"
	"codespace/internal/shell"
	"codespace/internal/workspace"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
)

func TestAcpStreamHandlerRoutes(t *testing.T) {
	tmpDir := t.TempDir()

	opencodePath := filepath.Join(tmpDir, "opencode")
	script := []byte(`#!/bin/sh
while IFS= read -r line; do
  id=$(printf '%s' "$line" | grep -o '"id":[0-9]*' | cut -d: -f2)
  case "$line" in
    *initialize*)
      printf '{"jsonrpc":"2.0","id":%s,"result":{"protocolVersion":1,"agentCapabilities":{},"agentInfo":{}}}\n' "$id"
      ;;
    *session/new*)
      printf '{"jsonrpc":"2.0","id":%s,"result":{"sessionId":"sess-1"}}\n' "$id"
      ;;
    *session/prompt*)
      printf '{"jsonrpc":"2.0","method":"session/update","params":{"sessionId":"sess-1","update":{"sessionUpdate":"agent_message_chunk","messageId":"m1","content":{"type":"text","text":"hello"}}}}\n'
      printf '{"jsonrpc":"2.0","method":"session/update","params":{"sessionId":"sess-1","update":{"sessionUpdate":"agent_message_chunk","messageId":"m1","content":{"type":"text","text":" world"}}}}\n'
      printf '{"jsonrpc":"2.0","id":%s,"result":{"stopReason":"end_turn"}}\n' "$id"
      ;;
  esac
done
`)
	if err := os.WriteFile(opencodePath, script, 0o755); err != nil {
		t.Fatalf("write fake opencode: %v", err)
	}

	wsRoot := filepath.Join(tmpDir, "workspaces")
	wsMgr := workspace.NewLocalManager(wsRoot)
	procMgr := process.NewManager(opencodePath, nil)
	shellMgr := shell.NewManager("bash", []string{"-i"})
	wsSvc := service.NewWorkspaceService(wsMgr, procMgr, shellMgr, nil)
	fileSvc := service.NewFileService(wsMgr, 1<<20)
	procSvc := service.NewProcessService(wsMgr, procMgr, nil)
	shellSvc := service.NewShellService(wsMgr, shellMgr, nil)
	acpSvc := service.NewAcpService(procMgr, wsMgr, nil)

	r := NewRouter(wsSvc, fileSvc, procSvc, shellSvc, acpSvc, nil, gin.TestMode)
	srv := httptest.NewServer(r)
	defer srv.Close()

	ws, err := wsSvc.Create("streamws")
	if err != nil {
		t.Fatalf("create workspace: %v", err)
	}
	t.Cleanup(func() {
		_ = wsSvc.Delete(ws.ID)
	})

	wsURL := "ws" + strings.TrimPrefix(srv.URL, "http") + "/api/workspaces/" + ws.ID + "/acp/stream"
	conn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
	if err != nil {
		t.Fatalf("dial websocket: %v", err)
	}
	defer conn.Close()

	if err := conn.WriteJSON(model.AcpStreamRequest{Type: "prompt", Content: "hello"}); err != nil {
		t.Fatalf("write prompt request: %v", err)
	}

	if err := conn.SetReadDeadline(time.Now().Add(3 * time.Second)); err != nil {
		t.Fatalf("set read deadline: %v", err)
	}

	var chunks int
	var complete bool
	for {
		var ev model.AcpStreamEvent
		if err := conn.ReadJSON(&ev); err != nil {
			t.Fatalf("read event: %v", err)
		}
		switch ev.Type {
		case "chunk":
			chunks++
		case "complete":
			complete = true
		case "error":
			t.Fatalf("unexpected error event: %s", ev.Error)
		default:
			t.Fatalf("unexpected event type %q", ev.Type)
		}
		if complete {
			break
		}
	}

	if chunks != 2 {
		t.Fatalf("chunks = %d, want 2", chunks)
	}
	if !complete {
		t.Fatal("expected complete event")
	}
}