package blobovniczatree

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

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobovnicza"
	"github.com/stretchr/testify/require"
)

func TestDBExtensionFix(t *testing.T) {
	root := t.TempDir()
	createTestTree(t, 0, 2, 3, root)
	t.Run("adds suffix if not exists", func(t *testing.T) {
		openAndCloseTestTree(t, 2, 3, root)
		validateTestTree(t, root)
	})

	t.Run("not adds second suffix if exists", func(t *testing.T) {
		openAndCloseTestTree(t, 2, 3, root)
		validateTestTree(t, root)
	})
}

func createTestTree(t *testing.T, currentDepth, depth, width uint64, path string) {
	if currentDepth == depth {
		var w uint64
		for ; w < width; w++ {
			dbPath := filepath.Join(path, u64ToHexString(w))
			b := blobovnicza.New(blobovnicza.WithPath(dbPath))
			require.NoError(t, b.Open())
			require.NoError(t, b.Init())
			require.NoError(t, b.Close())
		}
		return
	}

	var w uint64
	for ; w < width; w++ {
		createTestTree(t, currentDepth+1, depth, width, filepath.Join(path, u64ToHexString(w)))
	}
}

func openAndCloseTestTree(t *testing.T, depth, width uint64, path string) {
	blz := NewBlobovniczaTree(
		WithBlobovniczaShallowDepth(depth),
		WithBlobovniczaShallowWidth(width),
		WithRootPath(path),
	)
	require.NoError(t, blz.Open(false))
	require.NoError(t, blz.Init())
	require.NoError(t, blz.Close())
}

func validateTestTree(t *testing.T, path string) {
	entries, err := os.ReadDir(path)
	require.NoError(t, err)

	for _, entry := range entries {
		if entry.IsDir() {
			validateTestTree(t, filepath.Join(path, entry.Name()))
		} else {
			require.True(t, strings.HasSuffix(entry.Name(), dbExtension))
			require.False(t, strings.HasSuffix(strings.TrimSuffix(entry.Name(), dbExtension), dbExtension))
		}
	}
}