package releasepr

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/apricote/releaser-pleaser/internal/git"
	"github.com/apricote/releaser-pleaser/internal/testdata"
	"github.com/apricote/releaser-pleaser/internal/versioning"
)

func TestReleasePullRequest_GetOverrides(t *testing.T) {
	tests := []struct {
		name    string
		pr      ReleasePullRequest
		want    ReleaseOverrides
		wantErr assert.ErrorAssertionFunc
	}{
		{
			name:    "empty",
			pr:      ReleasePullRequest{},
			want:    ReleaseOverrides{},
			wantErr: assert.NoError,
		},
		{
			// TODO: Test for multiple version flags
			name: "single version flag",
			pr: ReleasePullRequest{
				Labels: []Label{LabelNextVersionTypeAlpha},
			},
			want: ReleaseOverrides{
				NextVersionType: versioning.NextVersionTypeAlpha,
			},
			wantErr: assert.NoError,
		},
		{
			name: "prefix in description",
			pr: ReleasePullRequest{
				PullRequest: git.PullRequest{
					Description: testdata.MustReadFileString(t, "description-prefix.txt"),
				},
			},
			want: ReleaseOverrides{
				Prefix: testdata.MustReadFileString(t, "prefix.txt"),
			},
			wantErr: assert.NoError,
		},
		{
			name: "suffix in description",
			pr: ReleasePullRequest{
				PullRequest: git.PullRequest{
					Description: testdata.MustReadFileString(t, "description-suffix.txt"),
				},
			},
			want: ReleaseOverrides{
				Suffix: testdata.MustReadFileString(t, "suffix.txt"),
			},
			wantErr: assert.NoError,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := tt.pr.GetOverrides()
			if !tt.wantErr(t, err, "GetOverrides()") {
				return
			}
			assert.Equalf(t, tt.want, got, "GetOverrides()")
		})
	}
}

func TestReleasePullRequest_ChangelogText(t *testing.T) {
	tests := []struct {
		name        string
		description string
		want        string
		wantErr     assert.ErrorAssertionFunc
	}{
		{
			name:        "no section",
			description: "# Foo\n",
			want:        "",
			wantErr:     assert.NoError,
		},
		{
			name:        "with section",
			description: testdata.MustReadFileString(t, "changelog.txt"),
			want:        testdata.MustReadFileString(t, "changelog-content.txt"),
			wantErr:     assert.NoError,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			pr := &ReleasePullRequest{
				PullRequest: git.PullRequest{
					Description: tt.description,
				},
			}
			got, err := pr.ChangelogText()
			if !tt.wantErr(t, err, "ChangelogText()") {
				return
			}
			assert.Equalf(t, tt.want, got, "ChangelogText()")
		})
	}
}

func TestReleasePullRequest_SetTitle(t *testing.T) {
	type args struct {
		branch  string
		version string
	}
	tests := []struct {
		name string
		pr   *ReleasePullRequest
		args args
		want string
	}{
		{
			name: "simple update",
			pr: &ReleasePullRequest{
				PullRequest: git.PullRequest{
					Title: "foo: bar",
				},
			},
			args: args{
				branch:  "main",
				version: "v1.0.0",
			},
			want: "chore(main): release v1.0.0",
		},
		{
			name: "no previous title",
			pr:   &ReleasePullRequest{},
			args: args{
				branch:  "release-1.x",
				version: "v1.1.1-rc.0",
			},
			want: "chore(release-1.x): release v1.1.1-rc.0",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.pr.SetTitle(tt.args.branch, tt.args.version)

			assert.Equal(t, tt.want, tt.pr.Title)
		})
	}
}

func TestReleasePullRequest_SetDescription(t *testing.T) {

	tests := []struct {
		name           string
		changelogEntry string
		overrides      ReleaseOverrides
		want           string
		wantErr        assert.ErrorAssertionFunc
	}{
		{
			name:           "no overrides",
			changelogEntry: `## v1.0.0`,
			overrides:      ReleaseOverrides{},
			want:           testdata.MustReadFileString(t, "description-no-overrides.txt"),
			wantErr:        assert.NoError,
		},
		{
			name:           "existing overrides",
			changelogEntry: `## v1.0.0`,
			overrides: ReleaseOverrides{
				Prefix: testdata.MustReadFileString(t, "prefix.txt"),
				Suffix: testdata.MustReadFileString(t, "suffix.txt"),
			},
			want:    testdata.MustReadFileString(t, "description-overrides.txt"),
			wantErr: assert.NoError,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			pr := &ReleasePullRequest{}
			err := pr.SetDescription(tt.changelogEntry, tt.overrides)
			if !tt.wantErr(t, err) {
				return
			}

			assert.Equal(t, tt.want, pr.Description)
		})
	}
}