package client_test

import (
	"context"
	"os"
	"testing"

	"github.com/keys-pub/keys"
	"github.com/keys-pub/keys-ext/http/api"
	"github.com/keys-pub/keys-ext/http/client"
	"github.com/keys-pub/keys/dstore/events"
	"github.com/keys-pub/keys/tsutil"
	"github.com/stretchr/testify/require"
)

// TODO: Test truncated

func TestMessages(t *testing.T) {
	env, closeFn := newEnv(t)
	defer closeFn()

	testMessages(t, env, testKeysSeeded())
}

func TestMessagesFirestore(t *testing.T) {
	if os.Getenv("TEST_FIRESTORE") != "1" {
		t.Skip()
	}
	env, closeFn := newEnvWithOptions(t, &envOptions{fi: testFirestore(t), clock: tsutil.NewTestClock()})
	defer closeFn()

	testMessages(t, env, testKeysRandom())
}

func testMessages(t *testing.T, env *env, tk testKeys) {
	// SetLogger(NewLogger(DebugLevel))
	// api.SetLogger(NewLogger(DebugLevel))
	// server.SetContextLogger(NewContextLogger(DebugLevel))

	aliceClient := newTestClient(t, env)
	bobClient := newTestClient(t, env)
	alice, bob, channel := tk.alice, tk.bob, tk.channel
	ctx := context.TODO()

	// Create channel
	info := &api.ChannelInfo{Name: "test"}
	_, err := aliceClient.ChannelCreate(ctx, channel, alice, info)
	require.NoError(t, err)

	// Messages
	msgs, err := aliceClient.Messages(ctx, channel, nil)
	require.NoError(t, err)
	require.Equal(t, int64(1), msgs.Index)
	require.Equal(t, 1, len(msgs.Events))
	require.False(t, msgs.Truncated)

	// MessageSend #1
	msg1 := api.NewMessage(alice.ID()).WithText("hi bob").WithTimestamp(env.clock.NowMillis())
	err = aliceClient.MessageSend(ctx, msg1, alice, channel)
	require.NoError(t, err)

	// MessageSend #2
	msg2 := api.NewMessage(bob.ID()).WithPrev(msg1.ID).WithText("what time are we meeting?").WithTimestamp(env.clock.NowMillis())
	err = bobClient.MessageSend(ctx, msg2, bob, channel)
	require.NoError(t, err)

	// Messages
	msgs, err = aliceClient.Messages(ctx, channel, nil)
	require.NoError(t, err)
	require.Equal(t, 3, len(msgs.Events))
	require.False(t, msgs.Truncated)
	out1, err := api.DecryptMessageFromEvent(msgs.Events[1], channel)
	require.NoError(t, err)
	require.Equal(t, msg1.Text, out1.Text)
	require.Equal(t, alice.ID(), out1.Sender)
	out2, err := api.DecryptMessageFromEvent(msgs.Events[2], channel)
	require.NoError(t, err)
	require.Equal(t, msg2.Text, out2.Text)
	require.Equal(t, bob.ID(), out2.Sender)
	require.NotEmpty(t, msgs.Events[2].Timestamp)
	require.NotEmpty(t, msgs.Events[2].Index)

	// MessageSend #3
	msg3 := api.NewMessage(alice.ID()).WithPrev(msg2.ID).WithText("3pm").WithTimestamp(env.clock.NowMillis())
	err = aliceClient.MessageSend(ctx, msg3, alice, channel)
	require.NoError(t, err)

	// Messages (from idx)
	msgs, err = aliceClient.Messages(ctx, channel, &client.MessagesOpts{Index: msgs.Index})
	require.NoError(t, err)
	require.Equal(t, 1, len(msgs.Events))
	out3, err := api.DecryptMessageFromEvent(msgs.Events[0], channel)
	require.NoError(t, err)
	require.Equal(t, msg3.Text, out3.Text)
	require.Equal(t, alice.ID(), out3.Sender)

	// Messages (desc)
	msgs, err = aliceClient.Messages(ctx, channel, &client.MessagesOpts{Order: events.Descending})
	require.NoError(t, err)
	require.Equal(t, 4, len(msgs.Events))
	out1, err = api.DecryptMessageFromEvent(msgs.Events[0], channel)
	require.NoError(t, err)
	require.Equal(t, msg3.Text, out1.Text)
	out2, err = api.DecryptMessageFromEvent(msgs.Events[1], channel)
	require.NoError(t, err)
	require.Equal(t, msg2.Text, out2.Text)
	out3, err = api.DecryptMessageFromEvent(msgs.Events[2], channel)
	require.NoError(t, err)
	require.Equal(t, msg1.Text, out3.Text)

	// Unknown channel
	unknown := keys.GenerateEdX25519Key()
	msgs, err = aliceClient.Messages(ctx, unknown, nil)
	require.NoError(t, err)
	require.Nil(t, msgs)
}