// Copyright 2020 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package repo

import (
	"net/http"
	"testing"

	"gitea.dev/models/db"
	issues_model "gitea.dev/models/issues"
	"gitea.dev/models/organization"
	"gitea.dev/models/perm"
	"gitea.dev/models/unit"
	"gitea.dev/models/unittest"
	"gitea.dev/modules/references"
	"gitea.dev/modules/timeutil"
	"gitea.dev/modules/web"
	"gitea.dev/services/contexttest"
	"gitea.dev/services/forms"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestCombineXRefComments(t *testing.T) {
	none, closes, neutered := references.XRefActionNone, references.XRefActionCloses, references.XRefActionNeutered
	xref := func(id, refIssueID, refCommentID int64, action references.XRefAction) *issues_model.Comment {
		return &issues_model.Comment{ID: id, Type: issues_model.CommentTypeIssueRef, RefIssueID: refIssueID, RefCommentID: refCommentID, RefAction: action}
	}
	issue := issues_model.Issue{Comments: issues_model.CommentList{
		xref(1, 10, 1, neutered),
		xref(2, 11, 2, none),
		xref(3, 10, 3, none),
		xref(4, 11, 4, closes),
		xref(5, 11, 5, neutered),
		xref(6, 0, 0, none),
		xref(7, 0, 0, none),
	}}
	combineXRefComments(&issue)
	assert.Equal(t, issues_model.CommentList{
		xref(1, 10, 3, none),
		xref(2, 11, 2, closes),
		xref(6, 0, 0, none),
		xref(7, 0, 0, none),
	}, issue.Comments)
}

func TestCombineLabelComments(t *testing.T) {
	labelComment := func(posterID int64, content, labelName string, createdUnix timeutil.TimeStamp, addedLabels, removedLabels []*issues_model.Label) *issues_model.Comment {
		return &issues_model.Comment{
			Type:          issues_model.CommentTypeLabel,
			PosterID:      posterID,
			Content:       content,
			Label:         &issues_model.Label{Name: labelName},
			CreatedUnix:   createdUnix,
			AddedLabels:   addedLabels,
			RemovedLabels: removedLabels,
		}
	}
	comment := func(posterID int64, content string) *issues_model.Comment {
		return &issues_model.Comment{Type: issues_model.CommentTypeComment, PosterID: posterID, Content: content, CreatedUnix: 0}
	}
	kases := []struct {
		name           string
		beforeCombined []*issues_model.Comment
		afterCombined  []*issues_model.Comment
	}{
		{
			name: "kase 1",
			beforeCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, nil, nil),
				labelComment(1, "", "kind/bug", 0, nil, nil),
				comment(1, "test"),
			},
			afterCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, []*issues_model.Label{}, nil),
				comment(1, "test"),
			},
		},
		{
			name: "kase 2",
			beforeCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, nil, nil),
				labelComment(1, "", "kind/bug", 70, nil, nil),
				comment(1, "test"),
			},
			afterCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, []*issues_model.Label{{Name: "kind/bug"}}, nil),
				labelComment(1, "", "kind/bug", 70, nil, []*issues_model.Label{{Name: "kind/bug"}}),
				comment(1, "test"),
			},
		},
		{
			name: "kase 3",
			beforeCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, nil, nil),
				labelComment(2, "", "kind/bug", 0, nil, nil),
				comment(1, "test"),
			},
			afterCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, []*issues_model.Label{{Name: "kind/bug"}}, nil),
				labelComment(2, "", "kind/bug", 0, nil, []*issues_model.Label{{Name: "kind/bug"}}),
				comment(1, "test"),
			},
		},
		{
			name: "migrated events of different original authors",
			beforeCombined: []*issues_model.Comment{
				{Type: issues_model.CommentTypeComment, PosterID: 1},
				{Type: issues_model.CommentTypeLabel, PosterID: 1, OriginalAuthor: "alice", Content: "1", Label: &issues_model.Label{Name: "kind/bug"}},
				{Type: issues_model.CommentTypeLabel, PosterID: 1, OriginalAuthor: "bob", Content: "1", Label: &issues_model.Label{Name: "kind/ui"}},
			},
			afterCombined: []*issues_model.Comment{
				{Type: issues_model.CommentTypeComment, PosterID: 1},
				{Type: issues_model.CommentTypeLabel, PosterID: 1, OriginalAuthor: "alice", Content: "1", Label: &issues_model.Label{Name: "kind/bug"}, AddedLabels: []*issues_model.Label{{Name: "kind/bug"}}},
				{Type: issues_model.CommentTypeLabel, PosterID: 1, OriginalAuthor: "bob", Content: "1", Label: &issues_model.Label{Name: "kind/ui"}, AddedLabels: []*issues_model.Label{{Name: "kind/ui"}}},
			},
		},
		{
			name: "kase 4",
			beforeCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, nil, nil),
				labelComment(1, "1", "kind/backport", 10, nil, nil),
			},
			afterCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 10, []*issues_model.Label{{Name: "kind/bug"}, {Name: "kind/backport"}}, nil),
			},
		},
		{
			name: "kase 5",
			beforeCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, nil, nil),
				comment(2, "testtest"),
				labelComment(1, "", "kind/bug", 0, nil, nil),
			},
			afterCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, []*issues_model.Label{{Name: "kind/bug"}}, nil),
				comment(2, "testtest"),
				labelComment(1, "", "kind/bug", 0, nil, []*issues_model.Label{{Name: "kind/bug"}}),
			},
		},
		{
			name: "kase 6",
			beforeCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, nil, nil),
				labelComment(1, "1", "reviewed/confirmed", 0, nil, nil),
				labelComment(1, "", "kind/bug", 0, nil, nil),
				labelComment(1, "1", "kind/feature", 0, nil, nil),
			},
			afterCombined: []*issues_model.Comment{
				labelComment(1, "1", "kind/bug", 0, []*issues_model.Label{{Name: "reviewed/confirmed"}, {Name: "kind/feature"}}, nil),
			},
		},
		{
			name: "project workflow",
			beforeCombined: []*issues_model.Comment{
				{Type: issues_model.CommentTypeLabel, PosterID: 1, Content: "1", Label: &issues_model.Label{Name: "kind/bug"}},
				{Type: issues_model.CommentTypeLabel, PosterID: 1, Content: "1", Label: &issues_model.Label{Name: "status/done"}, CommentMetaData: &issues_model.CommentMetaData{SpecialDoerName: issues_model.SpecialDoerNameProjectWorkflow}},
			},
			afterCombined: []*issues_model.Comment{
				{Type: issues_model.CommentTypeLabel, PosterID: 1, Content: "1", Label: &issues_model.Label{Name: "kind/bug"}, AddedLabels: []*issues_model.Label{{Name: "kind/bug"}}},
				{Type: issues_model.CommentTypeLabel, PosterID: 1, Content: "1", Label: &issues_model.Label{Name: "status/done"}, AddedLabels: []*issues_model.Label{{Name: "status/done"}}, CommentMetaData: &issues_model.CommentMetaData{SpecialDoerName: issues_model.SpecialDoerNameProjectWorkflow}},
			},
		},
	}

	for _, kase := range kases {
		t.Run(kase.name, func(t *testing.T) {
			issue := issues_model.Issue{
				Comments: kase.beforeCombined,
			}
			combineLabelComments(&issue)
			assert.EqualValues(t, kase.afterCombined, issue.Comments)
		})
	}
}

func TestIssueActionsRequireItemUnitWrite(t *testing.T) {
	unittest.PrepareTestEnv(t)
	// user4's team on org3/repo3 uses per-unit permissions: pull requests write, issues read
	_, err := db.GetEngine(t.Context()).ID(2).Cols("authorize").Update(&organization.Team{AccessMode: perm.AccessModeNone})
	require.NoError(t, err)
	_, err = db.GetEngine(t.Context()).Where("team_id = ? AND `type` = ?", 2, unit.TypeIssues).Cols("access_mode").Update(&organization.TeamUnit{AccessMode: perm.AccessModeRead})
	require.NoError(t, err)

	closeIssues := func(issueIDs string) int {
		ctx, resp := contexttest.MockContext(t, "org3/repo3/issues/status")
		contexttest.LoadUser(t, ctx, 4)
		contexttest.LoadRepo(t, ctx, 3)
		ctx.Req.Form.Set("issue_ids", issueIDs)
		ctx.Req.Form.Set("action", "close")
		UpdateIssueStatus(ctx)
		return resp.Code
	}
	assert.Equal(t, http.StatusNotFound, closeIssues("6,12"))
	assert.False(t, unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 6}).IsClosed)
	assert.False(t, unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 12}).IsClosed)

	assert.Equal(t, http.StatusOK, closeIssues("12"))
	assert.True(t, unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 12}).IsClosed)

	lockIssue := func(index string) int {
		ctx, resp := contexttest.MockContext(t, "org3/repo3/issues/"+index+"/lock")
		contexttest.LoadUser(t, ctx, 4)
		contexttest.LoadRepo(t, ctx, 3)
		ctx.SetPathParam("index", index)
		web.SetForm(ctx, &forms.IssueLockForm{Reason: "Spam"})
		LockIssue(ctx)
		return resp.Code
	}
	assert.Equal(t, http.StatusNotFound, lockIssue("1"))
	assert.False(t, unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 6}).IsLocked)

	assert.Equal(t, http.StatusOK, lockIssue("2"))
	assert.True(t, unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 12}).IsLocked)
}