package main

import (
	"crypto/sha256"
	"encoding/hex"
	"errors"
	"net/http"
	"path/filepath"
	"strings"
	"testing"

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

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

	answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "The keys are under the mat.")
	require.NoError(t, err)
	require.Equal(t, "Sent to Ana Souza at 12:05.", answer)

	answer, err = whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "The keys are under the mat.")
	require.NoError(t, err)
	require.Equal(t, "Sent to Ana Souza at 12:05. This is the answer of the first call with this id: nothing was sent again.", answer)

	_, err = whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "Another text.")
	require.EqualError(t, err, "the id keys belongs to another message: give a new id for each message")

	require.Equal(t, []WhatsAppFakeSend{
		{
			to:   whatsappTestAna.String(),
			text: "The keys are under the mat.",
			id:   "MSG1",
		},
	}, fake.Sent())

	again, err := openWhatsAppState(filepath.Join(whatsapp.dir, whatsappStateFileName))
	require.NoError(t, err)

	record, found := again.Send("keys")
	require.True(t, found)
	require.Equal(t, whatsappSent, record.State)
	require.Equal(t, "MSG1", record.MessageID)
}

func TestWhatsAppSendToAGroup(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	fake.groups = []*types.GroupInfo{
		{
			JID: whatsappTestFamily,
			GroupName: types.GroupName{
				Name: "Family",
			},
		},
	}

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

	answer, err := whatsapp.Send(t.Context(), "cake", whatsappTestFamily.String(), "I bring the cake.")
	require.NoError(t, err)
	require.Equal(t, "Sent to Family at 12:05.", answer)
}

func TestWhatsAppSendThatFails(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	fake.sendErr = errors.New("server error for " + whatsappTestNumber)

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

	for range 2 {
		answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
		require.NoError(t, err)
		require.True(t, strings.HasPrefix(answer, "The send failed: server error for OWN."), answer)
	}

	require.Len(t, fake.Sent(), 1)
}

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

	digest := sha256.Sum256([]byte("hi"))
	require.NoError(t, whatsapp.state.PutSend(WhatsAppSend{
		ID:     "keys",
		Chat:   whatsappTestAna.String(),
		Digest: hex.EncodeToString(digest[:]),
		State:  whatsappSending,
	}))

	answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
	require.NoError(t, err)
	require.Contains(t, answer, "its result is unknown")
	require.Empty(t, fake.Sent())
}

func TestWhatsAppSendRefuses(t *testing.T) {
	testCases := []struct {
		name string
		id   string
		chat string
		text string
		err  string
	}{
		{
			name: "no id",
			id:   " ",
			chat: whatsappTestAna.String(),
			text: "hi",
			err:  "the id must have 1 to 100 bytes",
		},
		{
			name: "no text",
			id:   "a",
			chat: whatsappTestAna.String(),
			text: "",
			err:  "the text must have 1 to 65536 bytes",
		},
		{
			name: "a name for a chat",
			id:   "a",
			chat: "Ana Souza",
			text: "hi",
			err:  `"Ana Souza" is not the ID of a WhatsApp chat: find the ID with whatsapp_find`,
		},
		{
			name: "a status broadcast",
			id:   "a",
			chat: types.StatusBroadcastJID.String(),
			text: "hi",
			err:  `"status@broadcast" is not a private chat or a group`,
		},
		{
			name: "the chat of the user",
			id:   "a",
			chat: whatsappTestOwn.String(),
			text: "hi",
			err:  "this is the chat of the user with itself: send to another chat",
		},
	}

	whatsapp, fake := whatsappStartTestLinked(t)
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	whatsappTestConnect(t, fake)

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			_, err := whatsapp.Send(t.Context(), tc.id, tc.chat, tc.text)
			require.EqualError(t, err, tc.err)
		})
	}

	require.Empty(t, fake.Sent())
}

func TestWhatsAppSendWhileDisconnected(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	whatsappTestConnect(t, fake)
	require.True(t, fake.Emit(&events.Disconnected{}))

	_, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
	require.EqualError(t, err, "WhatsApp is disconnected, so nothing was sent: try again later")

	require.True(t, fake.Emit(&events.Connected{}))

	answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
	require.NoError(t, err)
	require.Equal(t, "Sent to +15550002222 at 12:05.", answer)
}

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

	answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
	require.NoError(t, err)
	require.Equal(t, whatsappNotLinked, answer)
}

func TestAgentSendsAWhatsAppMessage(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_send", `{"id":"keys","chat":"15550002222@s.whatsapp.net","text":"The keys are under the mat."}`)}, 100, 0, 5))

			return
		}

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

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

	client.Type("Tell Ana that the keys are under the mat.")
	client.WaitFor(t, "[result] Sent to Ana Souza at 12:05.\nSent.\n[done]")
	require.Len(t, fake.Sent(), 1)
}