package main

import (
	"errors"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

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

func TestWhatsAppStatusFollowsTheConnection(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	clock := newTestClock()
	whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)

	<-fake.connects
	require.Equal(t, "WhatsApp is linked and not connected yet.", whatsapp.Status())

	require.True(t, fake.Emit(&events.Connected{}))
	require.Equal(t, "WhatsApp is linked and connected since 2026-01-01 12:00.", whatsapp.Status())

	clock.now = clock.now.Add(20 * time.Minute)
	require.True(t, fake.Emit(&events.Disconnected{}))
	clock.now = clock.now.Add(time.Minute)
	require.True(t, fake.Emit(&events.Disconnected{}))
	require.Equal(t, "WhatsApp is linked and disconnected since 2026-01-01 12:20. The last connection started at 2026-01-01 12:00.", whatsapp.Status())
}

func TestWhatsAppStatusOfAFirstConnectionThatFailed(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)

	<-fake.connects
	require.True(t, fake.Emit(&events.Disconnected{}))
	require.Equal(t, "WhatsApp is linked and disconnected since 2026-01-01 12:00.", whatsapp.Status())
	require.Equal(t, "first connect failed, retrying\n", logs.String())
}

func TestWhatsAppLogsEachChangeOfTheConnectionOnce(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	clock := newTestClock()
	whatsapp, fake, logs := whatsappStartTest(t, dir, clock)
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)

	<-fake.connects
	require.True(t, fake.Emit(&events.Connected{}))
	clock.now = clock.now.Add(time.Minute)
	require.True(t, fake.Emit(&events.Disconnected{}))
	clock.now = clock.now.Add(time.Minute)
	require.True(t, fake.Emit(&events.Disconnected{}))
	clock.now = clock.now.Add(25 * time.Second)
	require.True(t, fake.Emit(&events.Connected{}))

	require.Equal(t, []string{
		"connected",
		"disconnected, retrying",
		"connected again after 1m25s",
	}, strings.Split(strings.TrimSuffix(logs.String(), "\n"), "\n"))
	require.NotContains(t, logs.String(), whatsappTestNumber)
	require.NotContains(t, logs.String(), whatsappTestLID)
}

func TestWhatsAppIsNotLinkedWithoutALinkedDevice(t *testing.T) {
	testCases := []struct {
		name    string
		session func(t *testing.T, dir string)
	}{
		{
			name: "no session file",
			session: func(t *testing.T, dir string) {
			},
		},
		{
			name: "a session of a pairing that did not end",
			session: func(t *testing.T, dir string) {
				whatsappTestSession(t, dir, false)
			},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			dir := t.TempDir()
			tc.session(t, dir)
			whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
			require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
		})
	}
}

func TestWhatsAppRefusesASessionThatOthersCanRead(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	require.NoError(t, os.Chmod(filepath.Join(dir, whatsappDirName, whatsappSessionFileName), 0o644))

	_, err := openWhatsApp(t.Context(), dir, time.Now, whatsappDiscardLog(), whatsappNewClient)
	require.EqualError(t, err, "whatsapp/session.db holds the WhatsApp session, so its mode must be 0600")
}

func TestWhatsAppCommandsAnswerOnlyTheClientThatAsked(t *testing.T) {
	server := newModelFake(t)
	telegram := newTelegramFake(t)
	bot, _ := telegramStartTestBot(t, telegram)
	socket := daemonStartTestWithBot(t, server, bot)
	asking := terminalStartTestClient(t, socket)
	other := terminalStartTestClient(t, socket)
	asking.WaitFor(t, "[connected to OneChat]")
	other.WaitFor(t, "[connected to OneChat]")

	asking.Type("/whatsapp status")
	asking.WaitFor(t, "WhatsApp is not linked.")
	asking.Type(" /whatsapp  unknown ")
	asking.WaitFor(t, whatsappUsage)
	asking.Type("/whatsapp pair 0123")
	asking.WaitFor(t, "PHONE must be the number of your WhatsApp account")

	telegram.Add(TelegramUpdate{
		UpdateID: 1,
		Message:  telegramOwnerMessage("/whatsapp status"),
	})

	telegramWaitForSent(t, telegram, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.")
	require.Len(t, telegram.Sent(), 1)
	require.Equal(t, "[connected to OneChat]\n", other.Output())
	require.Zero(t, server.Calls())
}

func TestWhatsAppPairsWithACode(t *testing.T) {
	dir := t.TempDir()
	whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
	server := newModelFake(t)
	socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
	asking := terminalStartTestClient(t, socket)
	other := terminalStartTestClient(t, socket)
	asking.WaitFor(t, "[connected to OneChat]")
	other.WaitFor(t, "[connected to OneChat]")

	asking.Type("/whatsapp pair +1 555-000-1111")
	items := <-fake.pairings
	items <- whatsmeow.QRChannelItem{
		Event: whatsmeow.QRChannelEventCode,
	}

	asking.WaitFor(t, whatsappPairSteps+"\n"+whatsappTestCode+"\n")
	require.True(t, fake.Emit(&events.PairSuccess{
		ID:  types.NewADJID(whatsappTestNumber, 0, 1),
		LID: types.NewADJID(whatsappTestLID, 0, 1),
	}))
	items <- whatsmeow.QRChannelSuccess
	asking.WaitFor(t, "WhatsApp is linked to +15550001111.")

	// The library connects again after the link with the context of the
	// connection, so the end of the pairing must not end it.
	connections := fake.Connections()
	require.Len(t, connections, 1)
	require.Never(t, func() bool {
		return connections[0].Err() != nil
	}, 200*time.Millisecond, time.Millisecond)

	require.Equal(t, []string{whatsappTestNumber}, fake.Phones())
	require.Equal(t, "WhatsApp is linked and not connected yet.", whatsapp.Status())
	require.Equal(t, "[connected to OneChat]\n", other.Output())
	require.Zero(t, server.Calls())
	require.Equal(t, []string{"WhatsApp is linked."}, whatsappTestLogTexts(t, memory))

	info, err := os.Stat(filepath.Join(dir, whatsappDirName))
	require.NoError(t, err)
	require.Equal(t, os.FileMode(0o700), info.Mode().Perm())

	info, err = os.Stat(filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
	require.NoError(t, err)
	require.Equal(t, os.FileMode(0o600), info.Mode().Perm())
}

func TestWhatsAppPairingCodeIsATelegramMessageAlone(t *testing.T) {
	whatsapp, fake, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
	telegram := newTelegramFake(t)
	bot, _ := telegramStartTestBot(t, telegram)
	daemonStartTestWithWhatsApp(t, newModelFake(t), bot, whatsapp)

	telegram.Add(TelegramUpdate{
		UpdateID: 1,
		Message:  telegramOwnerMessage("/whatsapp pair 15550001111"),
	})

	items := <-fake.pairings
	items <- whatsmeow.QRChannelItem{
		Event: whatsmeow.QRChannelEventCode,
	}

	telegramWaitForSent(t, telegram, whatsappPairSteps, whatsappTestCode)
}

func TestWhatsAppPairingEnds(t *testing.T) {
	testCases := []struct {
		name   string
		end    func(items chan whatsmeow.QRChannelItem)
		answer string
	}{
		{
			name: "the code expires",
			end: func(items chan whatsmeow.QRChannelItem) {
				items <- whatsmeow.QRChannelTimeout
			},
			answer: whatsappPairExpired,
		},
		{
			name: "the library closes the pairing",
			end: func(items chan whatsmeow.QRChannelItem) {
				close(items)
			},
			answer: whatsappPairExpired,
		},
		{
			name: "the pairing fails",
			end: func(items chan whatsmeow.QRChannelItem) {
				items <- whatsmeow.QRChannelItem{
					Event: whatsmeow.QRChannelEventError,
					Error: errors.New("bad handshake"),
				}
			},
			answer: "The pairing failed: bad handshake. Run /whatsapp pair PHONE again.",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			whatsapp, fake, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
			socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
			client := terminalStartTestClient(t, socket)
			client.Type("/whatsapp pair 15550001111")

			items := <-fake.pairings
			items <- whatsmeow.QRChannelItem{
				Event: whatsmeow.QRChannelEventCode,
			}

			client.WaitFor(t, whatsappTestCode)
			tc.end(items)
			client.WaitFor(t, tc.answer)
			require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
		})
	}
}

func TestWhatsAppNewPairingEndsTheOneBefore(t *testing.T) {
	whatsapp, fake, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
	socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	client := terminalStartTestClient(t, socket)

	client.Type("/whatsapp pair 15550001111")
	<-fake.pairings
	client.Type("/whatsapp pair 15550002222")
	items := <-fake.pairings
	items <- whatsmeow.QRChannelItem{
		Event: whatsmeow.QRChannelEventCode,
	}

	client.WaitFor(t, whatsappTestCode)
	require.Equal(t, []string{"15550002222"}, fake.Phones())
	require.NotContains(t, client.Output(), "did not start the pairing")
}

func TestWhatsAppPairWhenLinked(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
	socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	client := terminalStartTestClient(t, socket)

	client.Type("/whatsapp pair 15550001111")
	client.WaitFor(t, "WhatsApp is linked already. Run /whatsapp unlink first.")
}

func TestWhatsAppUnlink(t *testing.T) {
	testCases := []struct {
		name      string
		logoutErr error
		answer    string
	}{
		{
			name:   "with a logout",
			answer: "WhatsApp is unlinked.",
		},
		{
			name:      "with no logout",
			logoutErr: errors.New("not connected"),
			answer:    "WhatsApp is unlinked here, but the phone did not get the logout. In WhatsApp on your phone, open Linked devices and remove the device.",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			dir := t.TempDir()
			whatsappTestSession(t, dir, true)
			whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
			fake.logoutErr = tc.logoutErr
			socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
			client := terminalStartTestClient(t, socket)
			<-fake.connects

			client.Type("/whatsapp unlink")
			client.WaitFor(t, tc.answer)
			require.Equal(t, 1, fake.Logouts())
			require.NoFileExists(t, filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
			require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
			require.Equal(t, []string{"WhatsApp is unlinked."}, whatsappTestLogTexts(t, memory))

			client.Type("/whatsapp unlink")
			client.WaitFor(t, "WhatsApp is not linked.\n")
		})
	}
}

func TestWhatsAppLogoutFromThePhone(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
	telegram := newTelegramFake(t)
	bot, _ := telegramStartTestBot(t, telegram)
	socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), bot, whatsapp)
	client := terminalStartTestClient(t, socket)
	client.WaitFor(t, "[connected to OneChat]")
	<-fake.connects

	require.True(t, fake.Emit(&events.LoggedOut{}))
	notice := "WhatsApp was unlinked from the phone. Run /whatsapp pair PHONE to link it again."
	client.WaitFor(t, notice)
	telegramWaitForSent(t, telegram, notice)
	require.NoFileExists(t, filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
	require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
	require.Equal(t, []string{"WhatsApp was unlinked from the phone."}, whatsappTestLogTexts(t, memory))
}

func TestWhatsAppLongDisconnectionGivesOneNotice(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	clock := newTestClock()
	whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
	whatsapp.noticeDelay = 300 * time.Millisecond
	socket, _ := 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(&events.Disconnected{}))
	require.True(t, fake.Emit(&events.Connected{}))
	time.Sleep(400 * time.Millisecond)
	require.Equal(t, "[connected to OneChat]\n", client.Output())

	clock.now = clock.now.Add(time.Hour)
	require.True(t, fake.Emit(&events.Disconnected{}))
	require.True(t, fake.Emit(&events.Disconnected{}))
	client.WaitFor(t, "WhatsApp is disconnected since 2026-01-01 13:00. OneChat tries again.\n")
	require.True(t, fake.Emit(&events.Disconnected{}))
	time.Sleep(400 * time.Millisecond)

	require.True(t, fake.Emit(&events.Connected{}))
	client.WaitFor(t, "WhatsApp is connected again.\n")
	require.Equal(t, 1, strings.Count(client.Output(), "WhatsApp is disconnected since"))
}

func TestWhatsAppDisconnectionWithNoRetryGivesANoticeAtOnce(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
	socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	client := terminalStartTestClient(t, socket)
	client.WaitFor(t, "[connected to OneChat]")
	<-fake.connects

	require.True(t, fake.Emit(&events.StreamReplaced{}))
	client.WaitFor(t, "WhatsApp is disconnected and does not connect again by itself: stream replaced. Restart the daemon to try again.")
}

func TestWhatsAppLogsNoOwnNumber(t *testing.T) {
	dir := t.TempDir()
	whatsappTestSession(t, dir, true)
	whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
	fake.connectErr = errors.New("no route for " + whatsappTestNumber + "@s.whatsapp.net and " + whatsappTestLID + "@lid")
	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
	<-fake.connects

	require.Eventually(t, func() bool {
		return logs.String() != ""
	}, 10*time.Second, time.Millisecond)
	require.Equal(t, "connect failed, restart to retry: no route for OWN@s.whatsapp.net and OWN@lid\n", logs.String())
}

func TestWhatsAppPhone(t *testing.T) {
	testCases := []struct {
		name  string
		phone string
		valid bool
	}{
		{
			name:  "a number with a plus and dashes",
			phone: "+1-555-000-1111",
			valid: true,
		},
		{
			name:  "a number with no country code",
			phone: "0115550001111",
			valid: false,
		},
		{
			name:  "a short number",
			phone: "123456",
			valid: false,
		},
		{
			name:  "a long number",
			phone: "1234567890123456",
			valid: false,
		},
		{
			name:  "a number with letters",
			phone: "1555ABC1111",
			valid: false,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			_, valid := whatsappPhone(tc.phone)
			require.Equal(t, tc.valid, valid)
		})
	}
}

func TestWhatsAppCommandArgs(t *testing.T) {
	testCases := []struct {
		name  string
		line  string
		args  string
		found bool
	}{
		{
			name:  "a command with its arguments",
			line:  " /whatsapp  pair +1 555 ",
			args:  "pair +1 555",
			found: true,
		},
		{
			name:  "the command alone",
			line:  "/whatsapp",
			args:  "",
			found: true,
		},
		{
			name:  "a word that starts with the command",
			line:  "/whatsappstatus",
			args:  "",
			found: false,
		},
		{
			name:  "a message that names the command",
			line:  "what does /whatsapp status do?",
			args:  "",
			found: false,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			args, found := whatsappCommandArgs(tc.line)
			require.Equal(t, tc.args, args)
			require.Equal(t, tc.found, found)
		})
	}
}

func TestSubagentHasNoWhatsAppTools(t *testing.T) {
	box := &Toolbox{}
	require.Equal(t, `error: unknown tool "whatsapp_find"`, box.run(t.Context(), "whatsapp_find", `{"name":"ana"}`))
	require.Equal(t, `error: unknown tool "whatsapp_send"`, box.run(t.Context(), "whatsapp_send", `{"id":"a","chat":"15550002222@s.whatsapp.net","text":"hi"}`))
}