package gitdiff

import (
	"testing"
	"time"
)

func TestParsePatchIdentity(t *testing.T) {
	tests := map[string]struct {
		Input  string
		Output PatchIdentity
		Err    interface{}
	}{
		"simple": {
			Input: "Morton Haypenny <mhaypenny@example.com>",
			Output: PatchIdentity{
				Name:  "Morton Haypenny",
				Email: "mhaypenny@example.com",
			},
		},
		"extraWhitespace": {
			Input: "   Morton Haypenny  <mhaypenny@example.com  >  ",
			Output: PatchIdentity{
				Name:  "Morton Haypenny",
				Email: "mhaypenny@example.com",
			},
		},
		"trailingCharacters": {
			Input: "Morton Haypenny <mhaypenny@example.com> unrelated garbage",
			Output: PatchIdentity{
				Name:  "Morton Haypenny",
				Email: "mhaypenny@example.com",
			},
		},
		"missingName": {
			Input: "<mhaypenny@example.com>",
			Err:   "invalid identity",
		},
		"missingEmail": {
			Input: "Morton Haypenny",
			Err:   "invalid identity",
		},
		"unclosedEmail": {
			Input: "Morton Haypenny <mhaypenny@example.com",
			Err:   "unclosed email",
		},
	}

	for name, test := range tests {
		t.Run(name, func(t *testing.T) {
			id, err := ParsePatchIdentity(test.Input)
			if test.Err != nil {
				assertError(t, test.Err, err, "parsing identity")
				return
			}
			if err != nil {
				t.Fatalf("unexpected error parsing identity: %v", err)
			}

			if test.Output != id {
				t.Errorf("incorrect identity: expected %#v, actual %#v", test.Output, id)
			}
		})
	}
}

func TestParsePatchDate(t *testing.T) {
	expected := time.Date(2020, 4, 9, 8, 7, 6, 0, time.UTC)

	tests := map[string]struct {
		Input  string
		Output time.Time
		Err    interface{}
	}{
		"default": {
			Input:  "Thu Apr 9 01:07:06 2020 -0700",
			Output: expected,
		},
		"defaultLocal": {
			Input:  "Thu Apr 9 01:07:06 2020",
			Output: time.Date(2020, 4, 9, 1, 7, 6, 0, time.Local),
		},
		"iso": {
			Input:  "2020-04-09 01:07:06 -0700",
			Output: expected,
		},
		"isoStrict": {
			Input:  "2020-04-09T01:07:06-07:00",
			Output: expected,
		},
		"rfc": {
			Input:  "Thu, 9 Apr 2020 01:07:06 -0700",
			Output: expected,
		},
		"short": {
			Input:  "2020-04-09",
			Output: time.Date(2020, 4, 9, 0, 0, 0, 0, time.Local),
		},
		"raw": {
			Input:  "1586419626 -0700",
			Output: expected,
		},
		"unix": {
			Input:  "1586419626",
			Output: expected,
		},
		"unknownFormat": {
			Input: "4/9/2020 01:07:06 PDT",
			Err:   "unknown date format",
		},
		"empty": {
			Input: "",
		},
	}

	for name, test := range tests {
		t.Run(name, func(t *testing.T) {
			d, err := ParsePatchDate(test.Input)
			if test.Err != nil {
				assertError(t, test.Err, err, "parsing date")
				return
			}
			if err != nil {
				t.Fatalf("unexpected error parsing date: %v", err)
			}
			if !test.Output.Equal(d) {
				t.Errorf("incorrect parsed date: expected %v, actual %v", test.Output, d)
			}
		})
	}
}

func TestParsePatchHeader(t *testing.T) {
	expectedSHA := "61f5cd90bed4d204ee3feb3aa41ee91d4734855b"
	expectedIdentity := &PatchIdentity{
		Name:  "Morton Haypenny",
		Email: "mhaypenny@example.com",
	}
	expectedDate := time.Date(2020, 04, 11, 15, 21, 23, 0, time.FixedZone("PDT", -7*60*60))
	expectedTitle := "A sample commit to test header parsing"
	expectedEmojiOneLineTitle := "🤖 Enabling auto-merging"
	expectedEmojiMultiLineTitle := "[IA64] Put ia64 config files on the Uwe Kleine-König diet"
	expectedBody := "The medium format shows the body, which\nmay wrap on to multiple lines.\n\nAnother body line."
	expectedBodyAppendix := "CC: Joe Smith <joe.smith@company.com>"

	tests := map[string]struct {
		Input  string
		Header PatchHeader
		Err    interface{}
	}{
		"prettyShort": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author: Morton Haypenny <mhaypenny@example.com>

    A sample commit to test header parsing
`,
			Header: PatchHeader{
				SHA:    expectedSHA,
				Author: expectedIdentity,
				Title:  expectedTitle,
			},
		},
		"prettyMedium": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author: Morton Haypenny <mhaypenny@example.com>
Date:   Sat Apr 11 15:21:23 2020 -0700

    A sample commit to test header parsing

    The medium format shows the body, which
    may wrap on to multiple lines.

    Another body line.
`,
			Header: PatchHeader{
				SHA:        expectedSHA,
				Author:     expectedIdentity,
				AuthorDate: expectedDate,
				Title:      expectedTitle,
				Body:       expectedBody,
			},
		},
		"prettyFull": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author: Morton Haypenny <mhaypenny@example.com>
Commit: Morton Haypenny <mhaypenny@example.com>

    A sample commit to test header parsing

    The medium format shows the body, which
    may wrap on to multiple lines.

    Another body line.
`,
			Header: PatchHeader{
				SHA:       expectedSHA,
				Author:    expectedIdentity,
				Committer: expectedIdentity,
				Title:     expectedTitle,
				Body:      expectedBody,
			},
		},
		"prettyFuller": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author:     Morton Haypenny <mhaypenny@example.com>
AuthorDate: Sat Apr 11 15:21:23 2020 -0700
Commit:     Morton Haypenny <mhaypenny@example.com>
CommitDate: Sat Apr 11 15:21:23 2020 -0700

    A sample commit to test header parsing

    The medium format shows the body, which
    may wrap on to multiple lines.

    Another body line.
`,
			Header: PatchHeader{
				SHA:           expectedSHA,
				Author:        expectedIdentity,
				AuthorDate:    expectedDate,
				Committer:     expectedIdentity,
				CommitterDate: expectedDate,
				Title:         expectedTitle,
				Body:          expectedBody,
			},
		},
		"prettyAppendix": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author:     Morton Haypenny <mhaypenny@example.com>
AuthorDate: Sat Apr 11 15:21:23 2020 -0700
Commit:     Morton Haypenny <mhaypenny@example.com>
CommitDate: Sat Apr 11 15:21:23 2020 -0700

    A sample commit to test header parsing

    The medium format shows the body, which
    may wrap on to multiple lines.

    Another body line.
    ---
    CC: Joe Smith <joe.smith@company.com>
`,
			Header: PatchHeader{
				SHA:           expectedSHA,
				Author:        expectedIdentity,
				AuthorDate:    expectedDate,
				Committer:     expectedIdentity,
				CommitterDate: expectedDate,
				Title:         expectedTitle,
				Body:          expectedBody + "\n---\n" + expectedBodyAppendix,
			},
		},
		"mailbox": {
			Input: `From 61f5cd90bed4d204ee3feb3aa41ee91d4734855b Mon Sep 17 00:00:00 2001
From: Morton Haypenny <mhaypenny@example.com>
Date: Sat, 11 Apr 2020 15:21:23 -0700
Subject: [PATCH] A sample commit to test header parsing

The medium format shows the body, which
may wrap on to multiple lines.

Another body line.
`,
			Header: PatchHeader{
				SHA:        expectedSHA,
				Author:     expectedIdentity,
				AuthorDate: expectedDate,
				Title:      expectedTitle,
				Body:       expectedBody,
			},
		},
		"mailboxEmojiOneLine": {
			Input: `From 61f5cd90bed4d204ee3feb3aa41ee91d4734855b Mon Sep 17 00:00:00 2001
From: Morton Haypenny <mhaypenny@example.com>
Date: Sat, 11 Apr 2020 15:21:23 -0700
Subject: [PATCH] =?UTF-8?q?=F0=9F=A4=96=20Enabling=20auto-merging?=

The medium format shows the body, which
may wrap on to multiple lines.

Another body line.
`,
			Header: PatchHeader{
				SHA:        expectedSHA,
				Author:     expectedIdentity,
				AuthorDate: expectedDate,
				Title:      expectedEmojiOneLineTitle,
				Body:       expectedBody,
			},
		},
		"mailboxEmojiMultiLine": {
			Input: `From 61f5cd90bed4d204ee3feb3aa41ee91d4734855b Mon Sep 17 00:00:00 2001
From: Morton Haypenny <mhaypenny@example.com>
Date: Sat, 11 Apr 2020 15:21:23 -0700
Subject: [PATCH] =?UTF-8?q?[IA64]=20Put=20ia64=20config=20files=20on=20the=20?=
 =?UTF-8?q?Uwe=20Kleine-K=C3=B6nig=20diet?=

The medium format shows the body, which
may wrap on to multiple lines.

Another body line.
`,
			Header: PatchHeader{
				SHA:        expectedSHA,
				Author:     expectedIdentity,
				AuthorDate: expectedDate,
				Title:      expectedEmojiMultiLineTitle,
				Body:       expectedBody,
			},
		},
		"mailboxAppendix": {
			Input: `From 61f5cd90bed4d204ee3feb3aa41ee91d4734855b Mon Sep 17 00:00:00 2001
From: Morton Haypenny <mhaypenny@example.com>
Date: Sat, 11 Apr 2020 15:21:23 -0700
Subject: [PATCH] A sample commit to test header parsing

The medium format shows the body, which
may wrap on to multiple lines.

Another body line.
---
CC: Joe Smith <joe.smith@company.com>
`,
			Header: PatchHeader{
				SHA:          expectedSHA,
				Author:       expectedIdentity,
				AuthorDate:   expectedDate,
				Title:        expectedTitle,
				Body:         expectedBody,
				BodyAppendix: expectedBodyAppendix,
			},
		},
		"mailboxMinimalNoName": {
			Input: `From: <mhaypenny@example.com>
Subject: [PATCH] A sample commit to test header parsing

The medium format shows the body, which
may wrap on to multiple lines.

Another body line.
`,
			Header: PatchHeader{
				Author: &PatchIdentity{expectedIdentity.Email, expectedIdentity.Email},
				Title:  expectedTitle,
				Body:   expectedBody,
			},
		},
		"mailboxMinimal": {
			Input: `From: Morton Haypenny <mhaypenny@example.com>
Subject: [PATCH] A sample commit to test header parsing

The medium format shows the body, which
may wrap on to multiple lines.

Another body line.
`,
			Header: PatchHeader{
				Author: expectedIdentity,
				Title:  expectedTitle,
				Body:   expectedBody,
			},
		},
		"unwrapTitle": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author: Morton Haypenny <mhaypenny@example.com>
Date:   Sat Apr 11 15:21:23 2020 -0700

    A sample commit to test header parsing with a long
	title that is wrapped.
`,
			Header: PatchHeader{
				SHA:        expectedSHA,
				Author:     expectedIdentity,
				AuthorDate: expectedDate,
				Title:      expectedTitle + " with a long title that is wrapped.",
			},
		},
		"normalizeBodySpace": {
			Input: `commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author: Morton Haypenny <mhaypenny@example.com>
Date:   Sat Apr 11 15:21:23 2020 -0700

    A sample commit to test header parsing


    The medium format shows the body, which
    may wrap on to multiple lines.


    Another body line.


`,
			Header: PatchHeader{
				SHA:        expectedSHA,
				Author:     expectedIdentity,
				AuthorDate: expectedDate,
				Title:      expectedTitle,
				Body:       expectedBody,
			},
		},
		"ignoreLeadingBlankLines": {
			Input: `

` + "    " + `
commit 61f5cd90bed4d204ee3feb3aa41ee91d4734855b
Author: Morton Haypenny <mhaypenny@example.com>

    A sample commit to test header parsing
`,
			Header: PatchHeader{
				SHA:    expectedSHA,
				Author: expectedIdentity,
				Title:  expectedTitle,
			},
		},
	}

	for name, test := range tests {
		t.Run(name, func(t *testing.T) {
			h, err := ParsePatchHeader(test.Input)
			if test.Err != nil {
				assertError(t, test.Err, err, "parsing patch header")
				return
			}
			if err != nil {
				t.Fatalf("unexpected error parsing patch header: %v", err)
			}
			if h == nil {
				t.Fatalf("expected non-nil header, but got nil")
			}

			exp := test.Header
			act := *h

			if exp.SHA != act.SHA {
				t.Errorf("incorrect parsed SHA: expected %q, actual %q", exp.SHA, act.SHA)
			}

			assertPatchIdentity(t, "author", exp.Author, act.Author)
			if !exp.AuthorDate.Equal(act.AuthorDate) {
				t.Errorf("incorrect parsed author date: expected %v, but got %v", exp.AuthorDate, act.AuthorDate)
			}

			assertPatchIdentity(t, "committer", exp.Committer, act.Committer)
			if !exp.CommitterDate.Equal(act.CommitterDate) {
				t.Errorf("incorrect parsed committer date: expected %v, but got %v", exp.CommitterDate, act.CommitterDate)
			}

			if exp.Title != act.Title {
				t.Errorf("incorrect parsed title:\n  expected: %q\n    actual: %q", exp.Title, act.Title)
			}
			if exp.Body != act.Body {
				t.Errorf("incorrect parsed body:\n  expected: %q\n    actual: %q", exp.Body, act.Body)
			}
			if exp.BodyAppendix != act.BodyAppendix {
				t.Errorf("incorrect parsed body appendix:\n  expected: %q\n    actual: %q",
					exp.BodyAppendix, act.BodyAppendix)
			}
		})
	}
}

func assertPatchIdentity(t *testing.T, kind string, exp, act *PatchIdentity) {
	switch {
	case exp == nil && act == nil:
	case exp == nil && act != nil:
		t.Errorf("incorrect parsed %s: expected nil, but got %+v", kind, act)
	case exp != nil && act == nil:
		t.Errorf("incorrect parsed %s: expected %+v, but got nil", kind, exp)
	case exp.Name != act.Name || exp.Email != act.Email:
		t.Errorf("incorrect parsed %s, expected %+v, bot got %+v", kind, exp, act)
	}
}

func TestCleanupSubject(t *testing.T) {
	exp := "A sample commit to test header parsing"
	tests := map[string]string{
		"plain":        "",
		"patch":        "[PATCH] ",
		"patchv5":      "[PATCH v5] ",
		"patchrfc":     "[PATCH RFC] ",
		"patchnospace": "[PATCH]",
		"space":        "   ",
		"re":           "re: ",
		"Re":           "Re: ",
		"RE":           "rE: ",
		"rere":         "re: re: ",
	}

	for name, prefix := range tests {
		gotprefix, gottitle := parseSubject(prefix + exp)
		if gottitle != exp {
			t.Errorf("%s: Incorrect parsing of prefix %s: got title %s, wanted %s",
				name, prefix, gottitle, exp)
		}
		if gotprefix != prefix {
			t.Errorf("%s: Incorrect parsing of prefix %s: got prefix %s",
				name, prefix, gotprefix)
		}
	}

	moretests := map[string]struct {
		in, eprefix, etitle string
	}{
		"Reimplement":       {"Reimplement something", "", "Reimplement something"},
		"patch-reimplement": {"[PATCH v5] Reimplement something", "[PATCH v5] ", "Reimplement something"},
		"Openbracket":       {"[Just to annoy people", "", "[Just to annoy people"},
	}

	for name, test := range moretests {
		prefix, title := parseSubject(test.in)
		if title != test.etitle {
			t.Errorf("%s: Incorrect parsing of %s: got title %s, wanted %s",
				name, test.in, title, test.etitle)
		}
		if prefix != test.eprefix {
			t.Errorf("%s: Incorrect parsing of %s: got prefix %s, wanted %s",
				name, test.in, title, test.etitle)
		}
	}
}