package main

import (
	"fmt"
	"net/http"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
	"go.mau.fi/whatsmeow/types"
)

func TestWhatsAppFindMatchesNamesWithNoRegardToCaseAndAccents(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	fake.groups = []*types.GroupInfo{
		{
			JID: whatsappTestFamily,
			GroupName: types.GroupName{
				Name: "Família Souza",
			},
		},
	}

	whatsappTestContacts(t, whatsapp)
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	whatsappTestConnect(t, fake)

	testCases := []struct {
		name   string
		query  string
		answer string
	}{
		{
			name:   "no accent finds an accent",
			query:  "joao",
			answer: "15550004444@s.whatsapp.net | João Silva | contact",
		},
		{
			name:   "a part of a name finds contacts and groups",
			query:  "SOUZA",
			answer: "15550002222@s.whatsapp.net | Ana Souza | contact\n120363000000000001@g.us | Família Souza | group",
		},
		{
			name:   "a business name",
			query:  "bakery",
			answer: "15550005555@s.whatsapp.net | Corner Bakery | contact",
		},
		{
			name:   "no match",
			query:  "Zed",
			answer: "No contact or group has a name with \"Zed\".",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			answer, err := whatsapp.Find(t.Context(), tc.query)
			require.NoError(t, err)
			require.Equal(t, tc.answer, answer)
		})
	}
}

func TestWhatsAppFindGivesAtMostTwentyMatches(t *testing.T) {
	whatsapp, _ := whatsappStartTestLinked(t)
	for index := range 25 {
		jid := types.NewJID(fmt.Sprintf("1555001%04d", index), types.DefaultUserServer)
		require.NoError(t, whatsapp.device.Contacts.PutContactName(t.Context(), jid, "", fmt.Sprintf("Member %02d", index)))
	}

	answer, err := whatsapp.Find(t.Context(), "member")
	require.NoError(t, err)

	lines := strings.Split(answer, "\n")
	require.Len(t, lines, 22)
	require.Equal(t, "15550010000@s.whatsapp.net | Member 00 | contact", lines[0])
	require.Equal(t, "5 more: give a longer part of the name.", lines[20])
	require.Equal(t, "The groups are not in this answer, because WhatsApp is disconnected.", lines[21])
}

func TestWhatsAppFindBeforeTheNamesArrive(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	whatsappTestConnect(t, fake)

	answer, err := whatsapp.Find(t.Context(), "ana")
	require.NoError(t, err)
	require.Equal(t, whatsappNamesNotReady, answer)
}

func TestWhatsAppFindWithoutALink(t *testing.T) {
	whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), newTestClock())

	answer, err := whatsapp.Find(t.Context(), "ana")
	require.NoError(t, err)
	require.Equal(t, whatsappNotLinked, answer)
}

func TestAgentFindsAWhatsAppContact(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	whatsappTestContacts(t, whatsapp)
	server := newModelFake(t)
	server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
		if call.number == 0 {
			modelWriteEvents(w, modelCompletedEvent([]any{modelFunctionCallItem("fc_1", "call_1", "whatsapp_find", `{"name":"ana"}`)}, 100, 0, 5))
			return
		}

		modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("Found Ana.")}, 100, 0, 5))
	}

	socket, _ := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
	client := terminalStartTestClient(t, socket)
	whatsappTestConnect(t, fake)

	client.Type("Find Ana on WhatsApp.")
	client.WaitFor(t, "[result] 15550002222@s.whatsapp.net | Ana Souza | contact\nFound Ana.\n[done]")
}