package tests

import (
	"context"
	"os"
	"strings"
	"testing"

	embeddedpostgres "github.com/fergusstrange/embedded-postgres"
	"github.com/hashicorp/consul/sdk/freeport"
	_ "github.com/jackc/pgx/v5/stdlib" // this is where we get our pgx database driver from
	"github.com/stretchr/testify/require"
	"gotest.tools/v3/golden"

	"cosmossdk.io/indexer/postgres"
	"cosmossdk.io/indexer/postgres/internal/testdata"
	"cosmossdk.io/schema/appdata"
	"cosmossdk.io/schema/indexer"
)

func TestInitSchema(t *testing.T) {
	t.Run("default", func(t *testing.T) {
		testInitSchema(t, false, "init_schema.txt")
	})

	t.Run("retain deletions disabled", func(t *testing.T) {
		testInitSchema(t, true, "init_schema_no_retain_delete.txt")
	})
}

func testInitSchema(t *testing.T, disableRetainDeletions bool, goldenFileName string) {
	t.Helper()
	connectionUrl := createTestDB(t)

	buf := &strings.Builder{}
	res, err := indexer.StartIndexing(indexer.IndexingOptions{
		Config: indexer.IndexingConfig{
			Target: map[string]indexer.Config{
				"postgres": {
					Type: "postgres",
					Config: postgres.Config{
						DatabaseURL:            connectionUrl,
						DisableRetainDeletions: disableRetainDeletions,
					},
				},
			},
		},
		Context: context.Background(),
		Logger:  prettyLogger{buf},
	})
	require.NoError(t, err)
	listener := res.Listener

	require.NotNil(t, listener.InitializeModuleData)
	require.NoError(t, listener.InitializeModuleData(appdata.ModuleInitializationData{
		ModuleName: "test",
		Schema:     testdata.ExampleSchema,
	}))

	require.NotNil(t, listener.Commit)
	cb, err := listener.Commit(appdata.CommitData{})
	require.NoError(t, err)
	if cb != nil {
		require.NoError(t, cb())
	}

	golden.Assert(t, buf.String(), goldenFileName)
}

func createTestDB(t *testing.T) (connectionUrl string) {
	t.Helper()
	tempDir, err := os.MkdirTemp("", "postgres-indexer-test")
	require.NoError(t, err)
	t.Cleanup(func() {
		require.NoError(t, os.RemoveAll(tempDir))
	})

	dbPort := freeport.GetOne(t)
	pgConfig := embeddedpostgres.DefaultConfig().
		Port(uint32(dbPort)).
		DataPath(tempDir)

	connectionUrl = pgConfig.GetConnectionURL()
	pg := embeddedpostgres.NewDatabase(pgConfig)
	require.NoError(t, pg.Start())
	t.Cleanup(func() {
		require.NoError(t, pg.Stop())
	})

	return
}