package methods

import (
	"context"
	"encoding/json"
	"testing"

	"github.com/google/uuid"

	"github.com/nextlevelbuilder/goclaw/internal/store"
	"github.com/nextlevelbuilder/goclaw/pkg/protocol"
)

// ---- stub SkillStore ----

type stubSkillStore struct {
	skills  []store.SkillInfo
	content map[string]string
	version int64
}

func newStubSkillStore(skills []store.SkillInfo) *stubSkillStore {
	content := make(map[string]string, len(skills))
	for _, s := range skills {
		content[s.Name] = "# " + s.Name + " skill content"
	}
	return &stubSkillStore{skills: skills, content: content, version: 1}
}

func (s *stubSkillStore) ListSkills(_ context.Context) []store.SkillInfo { return s.skills }

func (s *stubSkillStore) GetSkill(_ context.Context, name string) (*store.SkillInfo, bool) {
	for i := range s.skills {
		if s.skills[i].Name == name {
			return &s.skills[i], true
		}
	}
	return nil, false
}

func (s *stubSkillStore) LoadSkill(_ context.Context, name string) (string, bool) {
	c, ok := s.content[name]
	return c, ok
}

func (s *stubSkillStore) LoadForContext(_ context.Context, _ []string) string { return "" }
func (s *stubSkillStore) BuildSummary(_ context.Context, _ []string) string   { return "" }
func (s *stubSkillStore) FilterSkills(_ context.Context, _ []string) []store.SkillInfo {
	return nil
}
func (s *stubSkillStore) Version() int64 { return s.version }
func (s *stubSkillStore) BumpVersion()   { s.version++ }
func (s *stubSkillStore) Dirs() []string { return nil }

// ---- helpers ----

func buildSkillMethods(t *testing.T, skills []store.SkillInfo) *SkillsMethods {
	t.Helper()
	return NewSkillsMethods(newStubSkillStore(skills), nil)
}

func skillReqFrame(t *testing.T, method string, params map[string]any) *protocol.RequestFrame {
	t.Helper()
	raw, err := json.Marshal(params)
	if err != nil {
		t.Fatalf("marshal: %v", err)
	}
	return &protocol.RequestFrame{
		Type:   protocol.FrameTypeRequest,
		ID:     "sk-req-1",
		Method: method,
		Params: raw,
	}
}

// ---- Tests: handleList ----

func TestSkillsList_EmptyStore_ReturnsEmptyList(t *testing.T) {
	m := buildSkillMethods(t, nil)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsList, map[string]any{})
	m.handleList(context.Background(), client, req)
	// No panic = success
}

func TestSkillsList_WithSkills_ReturnsAll(t *testing.T) {
	skills := []store.SkillInfo{
		{Name: "skill-a", Slug: "skill-a", Enabled: true, IsSystem: true},
		{Name: "skill-b", Slug: "skill-b", Enabled: false},
	}
	m := buildSkillMethods(t, skills)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsList, map[string]any{})
	m.handleList(context.Background(), client, req)
	// No panic = success
}

// ---- Tests: handleGet ----

func TestSkillsGet_MissingName_ReturnsInvalidRequest(t *testing.T) {
	m := buildSkillMethods(t, nil)
	client := nullClient()
	// Empty params — name is required
	req := skillReqFrame(t, protocol.MethodSkillsGet, map[string]any{})
	m.handleGet(context.Background(), client, req)
	// No panic = invalid-request path hit
}

func TestSkillsGet_SkillNotFound_ReturnsNotFound(t *testing.T) {
	m := buildSkillMethods(t, nil) // empty store
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsGet, map[string]any{"name": "missing-skill"})
	m.handleGet(context.Background(), client, req)
	// No panic = not-found path hit
}

func TestSkillsGet_ExistingSkill_ReturnsContent(t *testing.T) {
	skills := []store.SkillInfo{
		{Name: "my-skill", Slug: "my-skill", Description: "Test skill", Enabled: true},
	}
	m := buildSkillMethods(t, skills)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsGet, map[string]any{"name": "my-skill"})
	m.handleGet(context.Background(), client, req)
	// No panic = found path hit
}

// ---- Tests: handleUpdate ----

func TestSkillsUpdate_MissingNameAndID_ReturnsInvalidRequest(t *testing.T) {
	m := buildSkillMethods(t, nil)
	client := nullClient()
	// Neither name nor id provided
	req := skillReqFrame(t, protocol.MethodSkillsUpdate, map[string]any{
		"updates": map[string]any{"enabled": true},
	})
	m.handleUpdate(context.Background(), client, req)
	// No panic = invalid-request path hit
}

func TestSkillsUpdate_StoreNotUpdater_ReturnsNotFound(t *testing.T) {
	// stubSkillStore does NOT implement skillUpdater — update should return ErrNotFound
	skills := []store.SkillInfo{{Name: "some-skill", Slug: "some-skill"}}
	m := buildSkillMethods(t, skills)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsUpdate, map[string]any{
		"name":    "some-skill",
		"updates": map[string]any{"enabled": false},
	})
	m.handleUpdate(context.Background(), client, req)
	// No panic = "store does not support updates" path hit
}

func TestSkillsUpdate_EmptyUpdates_ReturnsInvalidRequest(t *testing.T) {
	// name provided but updates is empty
	skills := []store.SkillInfo{{Name: "sk", Slug: "sk"}}
	m := buildSkillMethods(t, skills)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsUpdate, map[string]any{
		"name":    "sk",
		"updates": map[string]any{},
	})
	m.handleUpdate(context.Background(), client, req)
	// No panic = invalid-request (empty updates) path hit
}

func TestSkillsUpdate_NormalizesEmptyVisibilityToPrivate(t *testing.T) {
	skillID := "0193a5b0-7000-7000-8000-000000000321"
	store := newUpdatingSkillStore([]store.SkillInfo{{Name: "sk", Slug: "sk", Path: skillID}})
	m := NewSkillsMethods(store, nil)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsUpdate, map[string]any{
		"id":      skillID,
		"updates": map[string]any{"visibility": ""},
	})

	m.handleUpdate(context.Background(), client, req)

	if got := store.lastUpdates["visibility"]; got != "private" {
		t.Fatalf("visibility update = %#v, want private", got)
	}
}

func TestSkillsUpdate_RejectsNonStringVisibilityBeforeUpdate(t *testing.T) {
	skillID := "0193a5b0-7000-7000-8000-000000000322"
	store := newUpdatingSkillStore([]store.SkillInfo{{Name: "sk", Slug: "sk", Path: skillID}})
	m := NewSkillsMethods(store, nil)
	client := nullClient()
	req := skillReqFrame(t, protocol.MethodSkillsUpdate, map[string]any{
		"id":      skillID,
		"updates": map[string]any{"visibility": true},
	})

	m.handleUpdate(context.Background(), client, req)

	if store.updateCalled {
		t.Fatalf("non-string visibility reached UpdateSkill: %+v", store.lastUpdates)
	}
}

type updatingSkillStore struct {
	*stubSkillStore
	lastUpdates  map[string]any
	updateCalled bool
}

func newUpdatingSkillStore(skills []store.SkillInfo) *updatingSkillStore {
	return &updatingSkillStore{stubSkillStore: newStubSkillStore(skills)}
}

func (s *updatingSkillStore) UpdateSkill(_ context.Context, _ uuid.UUID, updates map[string]any) error {
	s.updateCalled = true
	s.lastUpdates = updates
	return nil
}

func (s *updatingSkillStore) GetSkillOwnerID(context.Context, uuid.UUID) (string, bool) {
	return "", true
}