package libfastimport_test

import (
	"io"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"

	"git.lukeshu.com/go/libfastimport"
)

func IdentPtr(ident libfastimport.Ident) *libfastimport.Ident {
	return &ident
}

func TestParser(t *testing.T) {
	MustParseIdent := func(str string) libfastimport.Ident {
		t.Helper()
		ident, err := libfastimport.ParseIdent(str)
		if err != nil {
			t.Fatal(err)
		}
		return ident
	}
	type resp struct {
		libfastimport.Cmd
		error
	}
	type testcase struct {
		input    string
		expected []resp
	}
	testcases := map[string]testcase{
		"TerminatingGetMark": {
			input: "\n" +
				"commit refs/heads/master\n" +
				"mark :1\n" +
				"author Luke Shumaker <lukeshu@lukeshu.com> 1561083373 -0400\n" +
				"committer Luke Shumaker <lukeshu@lukeshu.com> 1561083373 -0400\n" +
				"data 4\n" +
				"foo\n" +
				"from 0000000000000000000000000000000000000000\n" +
				"deleteall\n" +
				"M 100755 3bd4eeca3db66d8349fb0963f2caf2808a66e556 README.md\n" +
				"commit refs/heads/master\n" +
				"mark :2\n" +
				"author Luke Shumaker <lukeshu@lukeshu.com> 1561083374 -0400\n" +
				"committer Luke Shumaker <lukeshu@lukeshu.com> 1561083374 -0400\n" +
				"data 4\n" +
				"bar\n" +
				"deleteall\n" +
				"M 100755 505928d6061b54e26b48a9b1e391a7530e27b7ca README.md\n" +
				"get-mark :1\n" +
				"get-mark :2\n",
			expected: []resp{
				{libfastimport.CmdCommit{
					Ref:       "refs/heads/master",
					Mark:      1,
					Author:    IdentPtr(MustParseIdent("Luke Shumaker <lukeshu@lukeshu.com> 1561083373 -0400")),
					Committer: MustParseIdent("Luke Shumaker <lukeshu@lukeshu.com> 1561083373 -0400"),
					Msg:       "foo\n",
					From:      "0000000000000000000000000000000000000000",
				}, nil},
				{libfastimport.FileDeleteAll{}, nil},
				{libfastimport.FileModify{Mode: 0100755, Path: "README.md", DataRef: "3bd4eeca3db66d8349fb0963f2caf2808a66e556"}, nil},
				{libfastimport.CmdCommitEnd{}, nil},
				{libfastimport.CmdCommit{
					Ref:       "refs/heads/master",
					Mark:      2,
					Author:    IdentPtr(MustParseIdent("Luke Shumaker <lukeshu@lukeshu.com> 1561083374 -0400")),
					Committer: MustParseIdent("Luke Shumaker <lukeshu@lukeshu.com> 1561083374 -0400"),
					Msg:       "bar\n",
				}, nil},
				{libfastimport.FileDeleteAll{}, nil},
				{libfastimport.FileModify{Mode: 0100755, Path: "README.md", DataRef: "505928d6061b54e26b48a9b1e391a7530e27b7ca"}, nil},
				// OK, here's the interesting part of the testcase: `get-mark :1`
				// doesn't trigger CommitEnd, as get-mark is allowed inside of a
				// commit; BUT `get-mark :2` does, because the in-progress commit is
				// :2 and so asking for it implies that it's done.
				{libfastimport.CmdGetMark{Mark: 1}, nil},
				{libfastimport.CmdCommitEnd{}, nil},
				{libfastimport.CmdGetMark{Mark: 2}, nil},
				{nil, io.EOF},
			},
		},
	}

	t.Parallel()
	for tcName, tcData := range testcases {
		tcData := tcData
		t.Run(tcName, func(t *testing.T) {
			t.Parallel()
			parser := libfastimport.NewFrontend(strings.NewReader(tcData.input), nil, nil)
			var actual []resp
			for {
				cmd, err := parser.ReadCmd()
				actual = append(actual, resp{cmd, err})
				if err != nil {
					break
				}
			}
			assert.Equal(t, tcData.expected, actual)
		})
	}

}