package mutator_test

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

	"github.com/databricks/bricks/bundle/config"
	"github.com/databricks/bricks/bundle/config/mutator"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func touch(t *testing.T, path, file string) {
	f, err := os.Create(filepath.Join(path, file))
	require.NoError(t, err)
	f.Close()
}

func TestProcessRootIncludesEmpty(t *testing.T) {
	root := &config.Root{Path: "."}
	_, err := mutator.ProcessRootIncludes().Apply(root)
	require.NoError(t, err)
}

func TestProcessRootIncludesAbs(t *testing.T) {
	root := &config.Root{
		Path: ".",
		Include: []string{
			"/tmp/*.yml",
		},
	}
	_, err := mutator.ProcessRootIncludes().Apply(root)
	require.Error(t, err)
	assert.Contains(t, err.Error(), "must be relative paths")
}

func TestProcessRootIncludesSingleGlob(t *testing.T) {
	root := &config.Root{
		Path: t.TempDir(),
		Include: []string{
			"*.yml",
		},
	}

	touch(t, root.Path, "bundle.yml")
	touch(t, root.Path, "a.yml")
	touch(t, root.Path, "b.yml")

	ms, err := mutator.ProcessRootIncludes().Apply(root)
	require.NoError(t, err)

	var names []string
	for _, m := range ms {
		names = append(names, m.Name())
	}

	assert.NotContains(t, names, "ProcessInclude(bundle.yml)")
	assert.Contains(t, names, "ProcessInclude(a.yml)")
	assert.Contains(t, names, "ProcessInclude(b.yml)")
}

func TestProcessRootIncludesMultiGlob(t *testing.T) {
	root := &config.Root{
		Path: t.TempDir(),
		Include: []string{
			"a*.yml",
			"b*.yml",
		},
	}

	touch(t, root.Path, "a1.yml")
	touch(t, root.Path, "b1.yml")

	ms, err := mutator.ProcessRootIncludes().Apply(root)
	require.NoError(t, err)

	var names []string
	for _, m := range ms {
		names = append(names, m.Name())
	}

	assert.Contains(t, names, "ProcessInclude(a1.yml)")
	assert.Contains(t, names, "ProcessInclude(b1.yml)")
}

func TestProcessRootIncludesRemoveDups(t *testing.T) {
	root := &config.Root{
		Path: t.TempDir(),
		Include: []string{
			"*.yml",
			"*.yml",
		},
	}

	touch(t, root.Path, "a.yml")

	ms, err := mutator.ProcessRootIncludes().Apply(root)
	require.NoError(t, err)
	assert.Len(t, ms, 1)
	assert.Equal(t, "ProcessInclude(a.yml)", ms[0].Name())
}