package main

import (
	"net/http"
	"testing"
	"time"

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

func TestWhatsAppPrivateBatchStartsATurnWithAnEmptyReply(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	require.NoError(t, whatsapp.device.Contacts.PutContactName(t.Context(), whatsappTestAna, "Ana", "Ana Souza"))

	server := newModelFake(t)
	server.reply = func(w http.ResponseWriter, _ ModelFakeCall) {
		modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("")}, 100, 0, 5))
	}

	telegram := newTelegramFake(t)
	bot, _ := telegramStartTestBot(t, telegram)
	socket, memory := daemonStartTestWithWhatsApp(t, server, bot, whatsapp)
	client := terminalStartTestClient(t, socket)
	client.WaitFor(t, "[connected to OneChat]")
	<-fake.connects

	// The catch-up holds the batch open until its end, so the two messages
	// form one batch also on a slow machine.
	require.True(t, fake.Emit(&events.Connected{}))

	at := time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC)
	require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "Are you at home?", at)))
	require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestOwn, "m2", "Yes", at)))
	require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))

	text := "[WhatsApp] Ana Souza, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n12:00 Ana Souza: Are you at home?\n12:00 You: Yes\n" + whatsappBatchEmptyReply
	client.WaitFor(t, text+"\n[working]\n[done]\n")
	require.Equal(t, 1, server.Calls())
	require.Contains(t, string(server.Call(0).raw), "Are you at home?")
	require.Equal(t, []string{text}, whatsappTestLogTexts(t, memory))
	require.Empty(t, telegram.Sent())
	require.Eventually(t, func() bool {
		return len(whatsapp.state.Closed()) == 0
	}, 10*time.Second, time.Millisecond)
}

func TestWhatsAppGroupTurns(t *testing.T) {
	testCases := []struct {
		name    string
		message func(event *events.Message)
		line    string
		turn    bool
	}{
		{
			name: "plain",
			message: func(_ *events.Message) {
			},
			line: "12:00 Bruno Lima: Who brings the cake?",
		},
		{
			name: "mention",
			message: func(event *events.Message) {
				event.Message = &waE2E.Message{
					ExtendedTextMessage: &waE2E.ExtendedTextMessage{
						Text: new("@" + whatsappTestLID + " can you bring the cake? Call " + whatsappTestNumber + "."),
						ContextInfo: &waE2E.ContextInfo{
							MentionedJID: []string{whatsappTestLID + "@lid"},
						},
					},
				}
			},
			line: "12:00 Bruno Lima (mentions you): @you can you bring the cake? Call OWN.",
			turn: true,
		},
		{
			name: "reply",
			message: func(event *events.Message) {
				event.Message = &waE2E.Message{
					ExtendedTextMessage: &waE2E.ExtendedTextMessage{
						Text: new("Who brings the cake?"),
						ContextInfo: &waE2E.ContextInfo{
							Participant: new(whatsappTestOwn.String()),
						},
					},
				}
			},
			line: "12:00 Bruno Lima (replies to you): Who brings the cake?",
			turn: true,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			whatsapp, fake := whatsappStartTestLinked(t)
			fake.groups = []*types.GroupInfo{
				{
					JID: whatsappTestFamily,
					GroupName: types.GroupName{
						Name: "Family",
					},
				},
			}

			server := newModelFake(t)
			socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
			client := terminalStartTestClient(t, socket)
			client.WaitFor(t, "[connected to OneChat]")
			whatsappTestConnect(t, fake)

			event := whatsappTestMessage(whatsappTestFamily, whatsappTestBruno, "m1", "Who brings the cake?", time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC))
			event.Info.PushName = "Bruno Lima"
			tc.message(event)
			require.True(t, fake.Emit(event))

			text := "[WhatsApp] Family, group 120363000000000001@g.us, Thursday 2026-01-01\n" + tc.line
			if tc.turn {
				text += "\n" + whatsappBatchEmptyReply

				client.WaitFor(t, "[done]")
				require.Equal(t, 1, server.Calls())
			} else {
				client.WaitFor(t, text)
				time.Sleep(100 * time.Millisecond)
				require.Zero(t, server.Calls())
			}

			require.Equal(t, text, whatsappTestLogTexts(t, memory)[0])
		})
	}
}

func TestWhatsAppSkipsExcludedMessages(t *testing.T) {
	at := time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC)
	testCases := []struct {
		name   string
		change func(event *events.Message)
	}{
		{
			name: "an edit",
			change: func(event *events.Message) {
				event.IsEdit = true
			},
		},
		{
			name: "old history",
			change: func(event *events.Message) {
				event.SourceWebMsg = &waWeb.WebMessageInfo{}
			},
		},
		{
			name: "a view-once message",
			change: func(event *events.Message) {
				event.IsViewOnce = true
			},
		},
		{
			name: "a disappearing message",
			change: func(event *events.Message) {
				event.IsEphemeral = true
			},
		},
		{
			name: "the chat of the user with itself",
			change: func(event *events.Message) {
				event.Info.Chat = whatsappTestOwn
			},
		},
		{
			name: "a status broadcast",
			change: func(event *events.Message) {
				event.Info.Chat = types.StatusBroadcastJID
			},
		},
		{
			name: "a text of spaces",
			change: func(event *events.Message) {
				event.Message.Conversation = new("  ")
			},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			whatsapp, _ := whatsappStartTestLinked(t)
			event := whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "hi", at)
			tc.change(event)
			require.True(t, whatsapp.handle(event))
			require.NoError(t, whatsapp.state.CloseAll())
			require.Empty(t, whatsapp.state.Closed())
		})
	}
}

func TestWhatsAppCatchupGivesOneTurn(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	server := newModelFake(t)
	socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
	client := terminalStartTestClient(t, socket)
	client.WaitFor(t, "[connected to OneChat]")
	<-fake.connects

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

	at := time.Date(2026, 1, 1, 8, 0, 0, 0, time.UTC)
	for index, event := range []*events.Message{
		whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "Good morning", at),
		whatsappTestMessage(whatsappTestBruno, whatsappTestBruno, "m2", "Call me", at.Add(time.Hour)),
		whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m3", "Are you there?", at.Add(2*time.Hour)),
	} {
		event.Info.PushName = []string{"Ana", "Bruno", "Ana"}[index]
		require.True(t, fake.Emit(event))
	}

	time.Sleep(100 * time.Millisecond)
	require.Zero(t, memory.Len())

	require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
	client.WaitFor(t, "[done]")
	require.Equal(t, []string{
		"[WhatsApp] Ana, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n08:00 Ana: Good morning\n" + whatsappBatchEmptyReply,
		"[WhatsApp] Bruno, private chat 15550003333@s.whatsapp.net, Thursday 2026-01-01\n09:00 Bruno: Call me\n" + whatsappBatchEmptyReply,
		"[WhatsApp] Ana, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n10:00 Ana: Are you there?\n" + whatsappBatchEmptyReply,
	}, whatsappTestLogTexts(t, memory)[:3])

	time.Sleep(100 * time.Millisecond)
	require.Equal(t, 1, server.Calls())

	for _, text := range []string{"Good morning", "Call me", "Are you there?"} {
		require.Contains(t, string(server.Call(0).raw), text)
	}
}

func TestWhatsAppCatchupEndsAfterItsLimit(t *testing.T) {
	whatsapp, fake := whatsappStartTestLinked(t)
	whatsapp.inbox.catchupLimit = 100 * time.Millisecond
	socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	client := terminalStartTestClient(t, socket)
	client.WaitFor(t, "[connected to OneChat]")
	<-fake.connects

	require.True(t, fake.Emit(&events.Connected{}))
	require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestOwn, "m1", "On my way.", time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC))))
	client.WaitFor(t, "12:00 You: On my way.")
	require.Len(t, whatsappTestLogTexts(t, memory), 1)
}

func TestWhatsAppStateOfTheStateFileWaitForTheCatchup(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)

	before, _, _ := whatsappStartTest(t, dir, newTestClock())
	_, err := before.state.Add(whatsappTestChat, WhatsAppMessage{
		ID:     "m1",
		Sender: "Ana Souza",
		Time:   time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC),
		Text:   "Did you see my message?",
	}, time.Now().Add(-time.Hour))
	require.NoError(t, err)

	whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
	socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	client := terminalStartTestClient(t, socket)
	client.WaitFor(t, "[connected to OneChat]")
	<-fake.connects

	time.Sleep(100 * time.Millisecond)
	require.Zero(t, memory.Len())

	require.True(t, fake.Emit(&events.Connected{}))
	require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
	client.WaitFor(t, "12:00 Ana Souza: Did you see my message?")
	client.WaitFor(t, "[done]")
}