package e2e_test

import (
	"context"

	sdkmath "cosmossdk.io/math"
	"github.com/cosmos/cosmos-sdk/client/grpc/tmservice"
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/cosmos/cosmos-sdk/x/staking/types"

	"github.com/0glabs/0g-chain/tests/util"
)

func (suite *IntegrationTestSuite) TestValMinCommission() {
	suite.SkipIfUpgradeDisabled()

	beforeUpgradeCtx := util.CtxAtHeight(suite.UpgradeHeight - 1)
	afterUpgradeCtx := util.CtxAtHeight(suite.UpgradeHeight)

	suite.Run("before upgrade", func() {
		// Before params
		beforeParams, err := suite.ZgChain.Staking.Params(beforeUpgradeCtx, &types.QueryParamsRequest{})
		suite.Require().NoError(err)

		suite.Require().Equal(
			sdkmath.LegacyZeroDec().String(),
			beforeParams.Params.MinCommissionRate.String(),
			"min commission rate should be 0%% before upgrade",
		)

		// Before validators
		beforeValidators, err := suite.ZgChain.Staking.Validators(beforeUpgradeCtx, &types.QueryValidatorsRequest{})
		suite.Require().NoError(err)

		for _, val := range beforeValidators.Validators {
			// In kvtool gentx, the commission rate is set to 0, with max of 0.01
			expectedRate := sdkmath.LegacyZeroDec()
			expectedRateMax := sdkmath.LegacyMustNewDecFromStr("0.01")

			suite.Require().Equalf(
				expectedRate.String(),
				val.Commission.CommissionRates.Rate.String(),
				"validator %s should have commission rate of %s before upgrade",
				val.OperatorAddress,
				expectedRate,
			)

			suite.Require().Equalf(
				expectedRateMax.String(),
				val.Commission.CommissionRates.MaxRate.String(),
				"validator %s should have max commission rate of %s before upgrade",
				val.OperatorAddress,
				expectedRateMax,
			)
		}
	})

	suite.Run("after upgrade", func() {
		block, err := suite.ZgChain.Tm.GetBlockByHeight(context.Background(), &tmservice.GetBlockByHeightRequest{
			Height: suite.UpgradeHeight,
		})
		suite.Require().NoError(err)

		// After params
		afterParams, err := suite.ZgChain.Staking.Params(afterUpgradeCtx, &types.QueryParamsRequest{})
		suite.Require().NoError(err)

		expectedMinRate := sdk.MustNewDecFromStr("0.05")

		suite.Require().Equal(
			expectedMinRate.String(),
			afterParams.Params.MinCommissionRate.String(),
			"min commission rate should be 5%% after upgrade",
		)

		// After validators
		afterValidators, err := suite.ZgChain.Staking.Validators(afterUpgradeCtx, &types.QueryValidatorsRequest{})
		suite.Require().NoError(err)

		for _, val := range afterValidators.Validators {

			suite.Require().Truef(
				val.Commission.CommissionRates.Rate.GTE(expectedMinRate),
				"validator %s should have commission rate of at least 5%%",
				val.OperatorAddress,
			)

			suite.Require().Truef(
				val.Commission.CommissionRates.MaxRate.GTE(expectedMinRate),
				"validator %s should have max commission rate of at least 5%%",
				val.OperatorAddress,
			)

			suite.Require().Truef(
				val.Commission.UpdateTime.Equal(block.SdkBlock.Header.Time),
				"validator %s should have commission update time set to block time, expected %s, got %s",
				val.OperatorAddress,
				block.SdkBlock.Header.Time,
				val.Commission.UpdateTime,
			)
		}
	})
}