package v0_16

import (
	"io/ioutil"
	"path/filepath"
	"testing"
	"time"

	"github.com/cosmos/cosmos-sdk/codec"
	sdk "github.com/cosmos/cosmos-sdk/types"
	v036distr "github.com/cosmos/cosmos-sdk/x/distribution/legacy/v036"
	v040distr "github.com/cosmos/cosmos-sdk/x/distribution/types"
	v036gov "github.com/cosmos/cosmos-sdk/x/gov/legacy/v036"
	v040gov "github.com/cosmos/cosmos-sdk/x/gov/types"
	v036params "github.com/cosmos/cosmos-sdk/x/params/legacy/v036"
	v040params "github.com/cosmos/cosmos-sdk/x/params/types/proposal"
	v038upgrade "github.com/cosmos/cosmos-sdk/x/upgrade/legacy/v038"
	v040upgrade "github.com/cosmos/cosmos-sdk/x/upgrade/types"
	"github.com/stretchr/testify/suite"

	app "github.com/kava-labs/kava/app"
	v015committee "github.com/kava-labs/kava/x/committee/legacy/v0_15"
	v016committee "github.com/kava-labs/kava/x/committee/types"
	v015kavadist "github.com/kava-labs/kava/x/kavadist/legacy/v0_15"
	v016kavadist "github.com/kava-labs/kava/x/kavadist/types"
	v015pricefeed "github.com/kava-labs/kava/x/pricefeed/legacy/v0_15"
	v016pricefeedtypes "github.com/kava-labs/kava/x/pricefeed/types"
)

type migrateTestSuite struct {
	suite.Suite

	addresses            []sdk.AccAddress
	v15genstate          v015committee.GenesisState
	v15pricefeedgenstate v015pricefeed.GenesisState
	cdc                  codec.Codec
	legacyCdc            *codec.LegacyAmino
}

func (s *migrateTestSuite) SetupTest() {
	app.SetSDKConfig()

	s.v15genstate = v015committee.GenesisState{
		Committees:     v015committee.Committees{},
		Proposals:      []v015committee.Proposal{},
		NextProposalID: 1,
		Votes:          []v015committee.Vote{},
	}

	config := app.MakeEncodingConfig()
	s.cdc = config.Marshaler

	legacyCodec := codec.NewLegacyAmino()
	v015committee.RegisterLegacyAminoCodec(legacyCodec)
	v036distr.RegisterLegacyAminoCodec(legacyCodec)
	v038upgrade.RegisterLegacyAminoCodec(legacyCodec)
	v036params.RegisterLegacyAminoCodec(legacyCodec)
	v015kavadist.RegisterLegacyAminoCodec(legacyCodec)

	s.legacyCdc = legacyCodec

	_, accAddresses := app.GeneratePrivKeyAddressPairs(10)
	s.addresses = accAddresses
}

func (s *migrateTestSuite) TestMigrate_JSON() {
	file := filepath.Join("testdata", "v15-committee.json")
	data, err := ioutil.ReadFile(file)
	s.Require().NoError(err)

	err = s.legacyCdc.UnmarshalJSON(data, &s.v15genstate)
	s.Require().NoError(err)

	pricefeedFile := filepath.Join("testdata", "v15-pricefeed.json")
	pricefeedData, err := ioutil.ReadFile(pricefeedFile)
	s.Require().NoError(err)

	err = s.legacyCdc.UnmarshalJSON(pricefeedData, &s.v15pricefeedgenstate)
	s.Require().NoError(err)

	genstate := Migrate(s.v15genstate, s.v15pricefeedgenstate)
	actual := s.cdc.MustMarshalJSON(genstate)

	file = filepath.Join("testdata", "v16-committee.json")
	expected, err := ioutil.ReadFile(file)

	s.Require().NoError(err)
	s.Require().JSONEq(string(expected), string(actual))
}

func (s *migrateTestSuite) TestMigrate_PricefeedPermissions() {
	file := filepath.Join("testdata", "v15-committee.json")
	data, err := ioutil.ReadFile(file)
	s.Require().NoError(err)

	err = s.legacyCdc.UnmarshalJSON(data, &s.v15genstate)
	s.Require().NoError(err)

	pricefeedFile := filepath.Join("testdata", "v15-pricefeed.json")
	pricefeedData, err := ioutil.ReadFile(pricefeedFile)
	s.Require().NoError(err)

	err = s.legacyCdc.UnmarshalJSON(pricefeedData, &s.v15pricefeedgenstate)
	s.Require().NoError(err)

	genstate := Migrate(s.v15genstate, s.v15pricefeedgenstate)

	var uniqueMarkets = make(map[string]bool)

	for _, permission := range genstate.GetCommittees()[0].GetPermissions() {
		paramChangePermission, ok := permission.(v016committee.ParamsChangePermission)
		if !ok {
			continue
		}

		for _, allowedParamChanges := range paramChangePermission.AllowedParamsChanges {
			if allowedParamChanges.Subspace == v016pricefeedtypes.ModuleName {
				for _, req := range allowedParamChanges.MultiSubparamsRequirements {
					_, found := uniqueMarkets[req.Val]
					s.Require().Falsef(
						found,
						"pricefeed market MultiSubparamsRequirement for %v is duplicated",
						req.Val,
					)

					uniqueMarkets[req.Val] = true
				}
			}
		}
	}
}

func (s *migrateTestSuite) TestMigrate_TokenCommittee() {
	oldTokenCommittee := v015committee.TokenCommittee{
		BaseCommittee: v015committee.BaseCommittee{
			ID:               1,
			Description:      "test",
			Members:          s.addresses,
			Permissions:      []v015committee.Permission{},
			VoteThreshold:    sdk.NewDec(40),
			ProposalDuration: time.Hour * 24 * 7,
			TallyOption:      v015committee.Deadline,
		},
		Quorum:     sdk.NewDec(40),
		TallyDenom: "ukava",
	}

	expectedTokenCommittee, err := v016committee.NewTokenCommittee(1, "test", s.addresses, []v016committee.Permission{}, oldTokenCommittee.VoteThreshold, oldTokenCommittee.ProposalDuration, v016committee.TALLY_OPTION_DEADLINE, oldTokenCommittee.Quorum, oldTokenCommittee.TallyDenom)
	s.Require().NoError(err)

	s.v15genstate.Committees = []v015committee.Committee{oldTokenCommittee}
	genState := Migrate(s.v15genstate, s.v15pricefeedgenstate)
	s.Require().Len(genState.Committees, 1)
	s.Equal(expectedTokenCommittee, genState.GetCommittees()[0])
}

func (s *migrateTestSuite) TestMigrate_Committee_TallyOption() {
	testcases := []struct {
		name            string
		v015tallyOption v015committee.TallyOption
		v016tallyOption v016committee.TallyOption
	}{
		{
			name:            "null tally option",
			v015tallyOption: v015committee.NullTallyOption,
			v016tallyOption: v016committee.TALLY_OPTION_UNSPECIFIED,
		},
		{
			name:            "first past the post tally option",
			v015tallyOption: v015committee.FirstPastThePost,
			v016tallyOption: v016committee.TALLY_OPTION_FIRST_PAST_THE_POST,
		},
		{
			name:            "deadline tally",
			v015tallyOption: v015committee.Deadline,
			v016tallyOption: v016committee.TALLY_OPTION_DEADLINE,
		},
	}

	for _, tc := range testcases {
		s.Run(tc.name, func() {
			oldCommittee := v015committee.MemberCommittee{
				BaseCommittee: v015committee.BaseCommittee{
					ID:               2,
					Description:      "test",
					Members:          s.addresses,
					Permissions:      []v015committee.Permission{},
					VoteThreshold:    sdk.NewDec(40),
					ProposalDuration: time.Hour * 24 * 7,
					TallyOption:      tc.v015tallyOption,
				},
			}
			expectedProposal, err := v016committee.NewMemberCommittee(2, "test", s.addresses, []v016committee.Permission{}, oldCommittee.VoteThreshold, oldCommittee.ProposalDuration, tc.v016tallyOption)
			s.Require().NoError(err)
			s.v15genstate.Committees = []v015committee.Committee{oldCommittee}
			genState := Migrate(s.v15genstate, s.v15pricefeedgenstate)
			s.Require().Len(genState.Committees, 1)
			s.Equal(expectedProposal.GetTallyOption(), genState.GetCommittees()[0].GetTallyOption())
		})
	}
}

func (s *migrateTestSuite) TestMigrate_Committee_Permissions() {
	testcases := []struct {
		name           string
		v015permission v015committee.Permission
		v016permission v016committee.Permission
	}{
		{
			name:           "god permission",
			v015permission: v015committee.GodPermission{},
			v016permission: &v016committee.GodPermission{},
		},
		{
			name:           "text permission",
			v015permission: v015committee.TextPermission{},
			v016permission: &v016committee.TextPermission{},
		},
		{
			name:           "software upgrade permission",
			v015permission: v015committee.SoftwareUpgradePermission{},
			v016permission: &v016committee.SoftwareUpgradePermission{},
		},
		{
			name: "simple param change permission",
			v015permission: v015committee.SimpleParamChangePermission{
				AllowedParams: []v015committee.AllowedParam{
					{Subspace: "staking", Key: "bondDenom"},
					{Subspace: "test", Key: "testkey"},
				},
			},
			v016permission: &v016committee.ParamsChangePermission{
				AllowedParamsChanges: []v016committee.AllowedParamsChange{
					{Subspace: "staking", Key: "bondDenom"},
					{Subspace: "test", Key: "testkey"},
				},
			},
		},
		{
			name: "sub param change permission",
			v015permission: v015committee.SubParamChangePermission{
				AllowedParams: []v015committee.AllowedParam{
					{Subspace: "staking", Key: "bondDenom"},
					{Subspace: "test", Key: "testkey"},
				},
			},
			v016permission: &v016committee.ParamsChangePermission{
				AllowedParamsChanges: []v016committee.AllowedParamsChange{
					{Subspace: "staking", Key: "bondDenom"},
					{Subspace: "test", Key: "testkey"},
				},
			},
		},
	}

	for _, tc := range testcases {
		s.Run(tc.name, func() {
			oldCommittee := v015committee.MemberCommittee{
				BaseCommittee: v015committee.BaseCommittee{
					ID:               2,
					Description:      "test",
					Members:          s.addresses,
					Permissions:      []v015committee.Permission{tc.v015permission},
					VoteThreshold:    sdk.NewDec(40),
					ProposalDuration: time.Hour * 24 * 7,
					TallyOption:      v015committee.FirstPastThePost,
				},
			}
			expectedProposal, err := v016committee.NewMemberCommittee(2, "test", s.addresses, []v016committee.Permission{tc.v016permission}, oldCommittee.VoteThreshold, oldCommittee.ProposalDuration, v016committee.TALLY_OPTION_FIRST_PAST_THE_POST)
			s.Require().NoError(err)
			s.v15genstate.Committees = []v015committee.Committee{oldCommittee}
			genState := Migrate(s.v15genstate, s.v15pricefeedgenstate)
			s.Require().Len(genState.Committees, 1)
			s.Equal(expectedProposal, genState.GetCommittees()[0])
		})
	}
}

func (s *migrateTestSuite) TestMigrate_Proposals() {
	testcases := []struct {
		name         string
		v015proposal v015committee.PubProposal
		v016proposal v016committee.PubProposal
	}{
		{
			name:         "text proposal",
			v015proposal: v036gov.NewTextProposal("A Title", "A description of this proposal."),
			v016proposal: v040gov.NewTextProposal("A Title", "A description of this proposal."),
		},
		{
			name: "community poll spend proposal",
			v015proposal: v036distr.CommunityPoolSpendProposal{
				Title:       "Community Pool Spend",
				Description: "Fund the community pool.",
				Recipient:   s.addresses[0],
				Amount:      sdk.NewCoins(sdk.NewInt64Coin("ukava", 10)),
			},
			v016proposal: &v040distr.CommunityPoolSpendProposal{
				Title:       "Community Pool Spend",
				Description: "Fund the community pool.",
				Recipient:   s.addresses[0].String(),
				Amount:      sdk.NewCoins(sdk.NewInt64Coin("ukava", 10)),
			},
		},
		{
			name: "software upgrade with deprecated plan time",
			v015proposal: v038upgrade.SoftwareUpgradeProposal{
				Title:       "Test",
				Description: "Test",
				Plan:        v038upgrade.Plan{Name: "Test", Height: 100, Time: time.Now()},
			},
			v016proposal: &v040upgrade.SoftwareUpgradeProposal{
				Title:       "Test",
				Description: "Test",
				Plan:        v040upgrade.Plan{Name: "Test", Height: 100},
			},
		},
		{
			name: "param change proposal",
			v015proposal: v036params.ParameterChangeProposal{
				Title:       "Test",
				Description: "Test",
				Changes: []v036params.ParamChange{
					{Subspace: "Test", Key: "Test", Value: "Test"},
				},
			},
			v016proposal: &v040params.ParameterChangeProposal{
				Title:       "Test",
				Description: "Test",
				Changes: []v040params.ParamChange{
					{Subspace: "Test", Key: "Test", Value: "Test"},
				},
			},
		},
		{
			name: "kavadist community pool multi spend proposal",
			v015proposal: v015kavadist.CommunityPoolMultiSpendProposal{
				Title:       "Test",
				Description: "Test",
				RecipientList: v015kavadist.MultiSpendRecipients{
					v015kavadist.MultiSpendRecipient{
						Address: s.addresses[0],
						Amount:  sdk.NewCoins(sdk.NewInt64Coin("ukava", 10)),
					},
				},
			},
			v016proposal: &v016kavadist.CommunityPoolMultiSpendProposal{
				Title:       "Test",
				Description: "Test",
				RecipientList: []v016kavadist.MultiSpendRecipient{
					{
						Address: s.addresses[0].String(),
						Amount:  sdk.NewCoins(sdk.NewInt64Coin("ukava", 10)),
					},
				},
			},
		},
		{
			name: "cancel software upgrade proposal",
			v015proposal: v038upgrade.CancelSoftwareUpgradeProposal{
				Title:       "Test",
				Description: "Test",
			},
			v016proposal: &v040upgrade.CancelSoftwareUpgradeProposal{
				Title:       "Test",
				Description: "Test",
			},
		},
	}

	for _, tc := range testcases {
		s.Run(tc.name, func() {
			deadline := time.Now().Add(2 * time.Hour)
			oldProposal := v015committee.Proposal{
				PubProposal: tc.v015proposal,
				ID:          1,
				CommitteeID: 2,
				Deadline:    deadline,
			}
			expectedProposal, err := v016committee.NewProposal(tc.v016proposal, 1, 2, deadline)
			s.Require().NoError(err)
			s.v15genstate.Proposals = []v015committee.Proposal{oldProposal}
			genState := Migrate(s.v15genstate, s.v15pricefeedgenstate)
			s.Require().Len(genState.Proposals, 1)
			s.Equal(expectedProposal, genState.Proposals[0])
		})
	}
}

func (s *migrateTestSuite) TestMigrate_Votes() {
	testcases := []struct {
		name         string
		v015voteType v015committee.VoteType
		v016VoteType v016committee.VoteType
	}{
		{
			name:         "yes vote",
			v015voteType: v015committee.Yes,
			v016VoteType: v016committee.VOTE_TYPE_YES,
		},
		{
			name:         "no vote",
			v015voteType: v015committee.No,
			v016VoteType: v016committee.VOTE_TYPE_NO,
		},
		{
			name:         "null vote",
			v015voteType: v015committee.NullVoteType,
			v016VoteType: v016committee.VOTE_TYPE_UNSPECIFIED,
		},
		{
			name:         "abstain vote",
			v015voteType: v015committee.Abstain,
			v016VoteType: v016committee.VOTE_TYPE_ABSTAIN,
		},
	}

	for _, tc := range testcases {
		s.Run(tc.name, func() {
			oldVote := v015committee.Vote{
				ProposalID: 1,
				Voter:      s.addresses[0],
				VoteType:   tc.v015voteType,
			}
			expectedVote := v016committee.Vote{
				ProposalID: 1,
				Voter:      s.addresses[0],
				VoteType:   tc.v016VoteType,
			}
			s.v15genstate.Votes = []v015committee.Vote{oldVote}
			genState := Migrate(s.v15genstate, s.v15pricefeedgenstate)
			s.Require().Len(genState.Votes, 1)
			s.Equal(expectedVote, genState.Votes[0])
		})
	}
}

func TestMigrateTestSuite(t *testing.T) {
	suite.Run(t, new(migrateTestSuite))
}
