package commands_test

import (
	"bytes"
	"context"
	"flag"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"

	"github.com/MichaelMure/git-bug/commands"
	"github.com/MichaelMure/git-bug/repository"
	"github.com/spf13/cobra"
	"github.com/stretchr/testify/require"
)

var update = flag.Bool("update", false, "pass -update to the test runner to update golden files")

type testEnv struct {
	cwd  string
	repo *repository.GoGitRepo
	cmd  *cobra.Command
	out  *bytes.Buffer
}

func newTestEnv(t *testing.T) *testEnv {
	t.Helper()

	cwd, err := ioutil.TempDir("", "")
	require.NoError(t, err)
	t.Cleanup(func() {
		require.NoError(t, os.RemoveAll(cwd))
	})

	repo, err := repository.InitGoGitRepo(cwd)
	require.NoError(t, err)
	t.Cleanup(func() {
		require.NoError(t, repo.Close())
	})

	out := new(bytes.Buffer)
	cmd := commands.NewRootCommand()
	cmd.SetArgs([]string{})
	cmd.SetErr(out)
	cmd.SetOut(out)

	return &testEnv{
		cwd:  cwd,
		repo: repo,
		cmd:  cmd,
		out:  out,
	}
}

func (e *testEnv) Execute(t *testing.T) {
	t.Helper()

	ctx := context.WithValue(context.Background(), "cwd", e.cwd)
	require.NoError(t, e.cmd.ExecuteContext(ctx))
}

func requireGoldenFileEqual(t *testing.T, path string, act []byte) {
	t.Helper()

	path = filepath.Join("testdata", path)

	if *update {
		require.NoError(t, ioutil.WriteFile(path, act, 0644))
	}

	exp, err := ioutil.ReadFile(path)
	require.NoError(t, err)
	require.Equal(t, exp, act)
}

func TestNewRootCommand(t *testing.T) {
	testEnv := newTestEnv(t)
	testEnv.Execute(t)

	requireGoldenFileEqual(t, "root_out_golden.txt", testEnv.out.Bytes())
}