package store_test

import (
	"context"
	"fmt"
	"testing"

	"github.com/dchest/uniuri"
	"github.com/gopatchy/metadata"
	"github.com/gopatchy/store"
	"github.com/stretchr/testify/require"
)

func TestStore(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	dbname := fmt.Sprintf("file:%s?mode=memory&cache=shared", uniuri.New())

	st, err := store.NewStore(dbname)
	require.NoError(t, err)

	defer st.Close()

	err = st.Write(ctx, "storeTest", &storeTest{
		Metadata: metadata.Metadata{
			ID: "id1",
		},
		Opaque: "foo",
	})
	require.NoError(t, err)

	err = st.Write(ctx, "storeTest", &storeTest{
		Metadata: metadata.Metadata{
			ID: "id2",
		},
		Opaque: "bar",
	})
	require.NoError(t, err)

	err = st.Write(ctx, "storeTest", &storeTest{
		Metadata: metadata.Metadata{
			ID: "id2",
		},
		Opaque: "zig",
	})
	require.NoError(t, err)

	out1, err := st.Read(ctx, "storeTest", "id1", newStoreTest)
	require.NoError(t, err)
	require.NotNil(t, out1)
	require.Equal(t, "foo", out1.(*storeTest).Opaque)

	out2, err := st.Read(ctx, "storeTest", "id2", newStoreTest)
	require.NoError(t, err)
	require.NotNil(t, out1)
	require.Equal(t, "zig", out2.(*storeTest).Opaque)
}

func TestDelete(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	dbname := fmt.Sprintf("file:%s?mode=memory&cache=shared", uniuri.New())

	st, err := store.NewStore(dbname)
	require.NoError(t, err)

	defer st.Close()

	err = st.Write(ctx, "storeTest", &storeTest{
		Metadata: metadata.Metadata{
			ID: "id1",
		},
		Opaque: "foo",
	})
	require.NoError(t, err)

	out1, err := st.Read(ctx, "storeTest", "id1", newStoreTest)
	require.NoError(t, err)
	require.Equal(t, "foo", out1.(*storeTest).Opaque)

	err = st.Delete(ctx, "storeTest", "id1")
	require.NoError(t, err)

	out2, err := st.Read(ctx, "storeTest", "id1", newStoreTest)
	require.NoError(t, err)
	require.Nil(t, out2)
}

func TestList(t *testing.T) {
	t.Parallel()

	ctx := context.Background()

	dbname := fmt.Sprintf("file:%s?mode=memory&cache=shared", uniuri.New())

	st, err := store.NewStore(dbname)
	require.NoError(t, err)

	defer st.Close()

	objs, err := st.List(ctx, "storeTest", func() any { return &storeTest{} })
	require.NoError(t, err)
	require.Len(t, objs, 0)

	err = st.Write(ctx, "storeTest", &storeTest{
		Metadata: metadata.Metadata{
			ID: "id1",
		},
		Opaque: "foo",
	})
	require.NoError(t, err)

	err = st.Write(ctx, "storeTest", &storeTest{
		Metadata: metadata.Metadata{
			ID: "id2",
		},
		Opaque: "bar",
	})
	require.NoError(t, err)

	objs, err = st.List(ctx, "storeTest", func() any { return &storeTest{} })
	require.NoError(t, err)
	require.Len(t, objs, 2)
	require.ElementsMatch(t, []string{"foo", "bar"}, []string{objs[0].(*storeTest).Opaque, objs[1].(*storeTest).Opaque})
}

type storeTest struct {
	metadata.Metadata
	Opaque string
}

func newStoreTest() any {
	return &storeTest{}
}