package service

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// ========== parseJSONResponse Tests ==========

func TestParseJSONResponse_DirectJSON(t *testing.T) {
	content := `{"suggestions": [{"labels": ["a", "b"], "priority": "high", "reason": "test reason"}]}`
	var result LabelSuggestionResult
	err := parseJSONResponse(content, &result)

	require.NoError(t, err)
	assert.Len(t, result.Suggestions, 1)
	assert.Equal(t, "high", result.Suggestions[0].Priority)
	assert.Len(t, result.Suggestions[0].Labels, 2)
}

func TestParseJSONResponse_MarkdownCodeBlock(t *testing.T) {
	content := "```json\n{\"suggestions\": [{\"labels\": [\"test\"], \"priority\": \"low\", \"reason\": \"test reason\"}]}\n```"
	var result LabelSuggestionResult
	err := parseJSONResponse(content, &result)

	require.NoError(t, err)
	assert.Len(t, result.Suggestions, 1)
	assert.Equal(t, "low", result.Suggestions[0].Priority)
	assert.Equal(t, "test reason", result.Suggestions[0].Reason)
}

func TestParseJSONResponse_PlainCodeBlock(t *testing.T) {
	content := "```json\n{\"suggestions\": [{\"labels\": [], \"priority\": \"medium\"}]}\n```"
	var result LabelSuggestionResult
	err := parseJSONResponse(content, &result)

	require.NoError(t, err)
	assert.Len(t, result.Suggestions, 1)
	assert.Equal(t, "medium", result.Suggestions[0].Priority)
}

func TestParseJSONResponse_EmbeddedJSON(t *testing.T) {
	content := "Here is the analysis:\n{\"participants\": [], \"summary\": \"test summary\"}\nEnd of analysis."
	var result ParticipantAnalysisResult
	err := parseJSONResponse(content, &result)

	require.NoError(t, err)
	assert.Equal(t, "test summary", result.Summary)
}

func TestParseJSONResponse_InvalidJSON(t *testing.T) {
	content := "this is not json at all"
	var result LabelSuggestionResult
	err := parseJSONResponse(content, &result)

	assert.Error(t, err)
}

func TestExtractLabelSuggestionFromText(t *testing.T) {
	content := "Label: bug, feature\nPriority: high\nReason: urgent issue"
	result := extractLabelSuggestionFromText(content)

	assert.NotEmpty(t, result.Suggestions)
	if len(result.Suggestions) > 0 {
		assert.Contains(t, result.Suggestions[0].Labels, "bug")
		assert.Contains(t, result.Suggestions[0].Labels, "feature")
		assert.Equal(t, "high", result.Suggestions[0].Priority)
		assert.Equal(t, "urgent issue", result.Suggestions[0].Reason)
	}
}