package loader_test

import (
	"context"
	"strings"
	"testing"

	"git.kapelle.org/niklas/s3browser/internal/loader"
	"git.kapelle.org/niklas/s3browser/internal/s3"
	"git.kapelle.org/niklas/s3browser/internal/types"
	"github.com/graph-gophers/dataloader"
	"github.com/stretchr/testify/assert"
)

func setup(t *testing.T) (context.Context, *loader.Loader, *assert.Assertions) {
	assert := assert.New(t)
	s3, _ := s3.NewMockS3([]string{"bucket1", "bucket2"})
	loader := loader.NewLoader(loader.CacheConfig{
		ListObjectsLoaderCache:          &dataloader.NoCache{},
		ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
		StatObjectLoaderCache:           &dataloader.NoCache{},
		ListBucketsLoaderCache:          &dataloader.NoCache{},
	})

	fillS3(s3)
	ctx := context.WithValue(context.Background(), "s3Client", s3)

	return ctx, loader, assert
}

func fillS3(s3 s3.S3Service) {
	ctx := context.Background()
	length := int64(len("content"))

	for _, v := range []string{
		"bucket1:/file1", "bucket1:/file2", "bucket1:/dir1/file1",
		"bucket1:/dir1/file2", "bucket1:/dir2/file1", "bucket1:/dir1/sub1/file1",
		"bucket1:/dir1/sub1/file2",
	} {
		s3.PutObject(ctx, *types.ParseID(v), strings.NewReader("content"), length)
	}
}

func TestCreateLoader(t *testing.T) {
	assert := assert.New(t)
	loader := loader.NewLoader(loader.CacheConfig{
		ListObjectsLoaderCache:          &dataloader.NoCache{},
		ListObjectsRecursiveLoaderCache: &dataloader.NoCache{},
		StatObjectLoaderCache:           &dataloader.NoCache{},
		ListBucketsLoaderCache:          &dataloader.NoCache{},
	})

	assert.NotNil(loader)
}

func TestGetBuckets(t *testing.T) {
	ctx, loader, assert := setup(t)

	buckets, err := loader.GetBuckets(ctx)
	assert.NoError(err)
	assert.Len(buckets, 2)
	assert.Contains(buckets, "bucket1")
	assert.Contains(buckets, "bucket2")
}

func TestGetFile(t *testing.T) {
	ctx, loader, assert := setup(t)

	file, err := loader.GetFile(ctx, *types.ParseID("bucket1:/dir1/file1"))
	assert.NoError(err)

	assert.Equal("bucket1:/dir1/file1", file.ID.String())
	assert.Equal("file1", file.Name)
	assert.Equal(int64(len("content")), file.Size)
}

func TestGetFiles(t *testing.T) {
	ctx, loader, assert := setup(t)

	id := types.ParseID("bucket1:/")

	files, err := loader.GetFiles(ctx, *id)
	assert.NoError(err)
	assert.Len(files, 2)
}

func TestGetDir(t *testing.T) {
	ctx, loader, assert := setup(t)

	id := types.ParseID("bucket1:/")

	dirs, err := loader.GetDirs(ctx, *id)
	assert.NoError(err)
	assert.Len(dirs, 2)
}

func Test(t *testing.T) {
	ctx, loader, assert := setup(t)

	id := types.ParseID("bucket1:/dir1/")

	files, err := loader.GetFilesRecursive(ctx, *id)
	assert.NoError(err)
	assert.Len(files, 4)
}