package instance

import (
	"context"
	"net/http"
	"testing"

	"github.com/kandev/kandev/internal/agentctl/server/config"
	"github.com/kandev/kandev/internal/agentctl/server/process"
	"github.com/kandev/kandev/internal/common/logger"
	"github.com/kandev/kandev/pkg/agent"
)

// TestCreateInstance_RecordsCreateReadyMillis pins the diagnostic
// agentctl_create_ready_ms metric's write side: CreateInstance must stamp a
// positive elapsed-milliseconds value onto the InstanceConfig it builds, so
// api.handleSystemMetrics has something other than the "not yet recorded"
// zero sentinel to report for a real instance.
func TestCreateInstance_RecordsCreateReadyMillis(t *testing.T) {
	log := newTestLogger(t)
	mgr := NewManager(&config.Config{
		Ports:    config.PortConfig{Base: 0, Max: 0},
		Defaults: config.InstanceDefaults{Protocol: agent.ProtocolACP},
	}, log)
	t.Cleanup(func() { _ = mgr.Shutdown(context.Background()) })

	var captured *config.InstanceConfig
	mgr.SetServerFactory(func(cfg *config.InstanceConfig, procMgr *process.Manager, log *logger.Logger) http.Handler {
		captured = cfg
		return http.NotFoundHandler()
	})

	resp, err := mgr.CreateInstance(context.Background(), &CreateRequest{WorkspacePath: t.TempDir()})
	if err != nil {
		t.Fatalf("CreateInstance: %v", err)
	}
	t.Cleanup(func() { _ = mgr.StopInstance(context.Background(), resp.ID) })

	if captured == nil {
		t.Fatal("server factory was never invoked; no InstanceConfig captured")
	}
	if captured.CreateReadyMillis == nil {
		t.Fatal("CreateReadyMillis is nil on a config built by NewInstanceConfig")
	}
	if millis := captured.CreateReadyMillis.Load(); millis <= 0 {
		t.Errorf("CreateReadyMillis = %d, want > 0 after a completed creation", millis)
	}
}

func TestCreateInstance_PropagatesMCPToolNamePresentationCapability(t *testing.T) {
	log := newTestLogger(t)
	mgr := NewManager(&config.Config{
		Ports:    config.PortConfig{Base: 0, Max: 0},
		Defaults: config.InstanceDefaults{Protocol: agent.ProtocolACP},
	}, log)
	t.Cleanup(func() { _ = mgr.Shutdown(context.Background()) })

	var captured *config.InstanceConfig
	mgr.SetServerFactory(func(cfg *config.InstanceConfig, procMgr *process.Manager, log *logger.Logger) http.Handler {
		captured = cfg
		return http.NotFoundHandler()
	})

	resp, err := mgr.CreateInstance(context.Background(), &CreateRequest{
		WorkspacePath:              t.TempDir(),
		NamespacesMCPToolsByServer: true,
	})
	if err != nil {
		t.Fatalf("CreateInstance: %v", err)
	}
	t.Cleanup(func() { _ = mgr.StopInstance(context.Background(), resp.ID) })

	if captured == nil || !captured.NamespacesMCPToolsByServer {
		t.Fatalf("captured instance config = %+v, want namespaced MCP tools", captured)
	}
}