package keeper_test

import (
	"testing"
	"time"

	"github.com/0glabs/0g-chain/app"
	"github.com/0glabs/0g-chain/x/hard/keeper"
	"github.com/0glabs/0g-chain/x/hard/types"
	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/stretchr/testify/suite"
	tmprototypes "github.com/tendermint/tendermint/proto/tendermint/types"
)

type grpcQueryTestSuite struct {
	suite.Suite

	tApp        app.TestApp
	ctx         sdk.Context
	keeper      keeper.Keeper
	queryServer types.QueryServer
	addrs       []sdk.AccAddress
}

func (suite *grpcQueryTestSuite) SetupTest() {
	suite.tApp = app.NewTestApp()
	_, addrs := app.GeneratePrivKeyAddressPairs(2)

	suite.addrs = addrs

	suite.ctx = suite.tApp.NewContext(true, tmprototypes.Header{}).
		WithBlockTime(time.Now().UTC())
	suite.keeper = suite.tApp.GetHardKeeper()
	suite.queryServer = keeper.NewQueryServerImpl(suite.keeper, suite.tApp.GetAccountKeeper(), suite.tApp.GetBankKeeper())

	err := suite.tApp.FundModuleAccount(
		suite.ctx,
		types.ModuleAccountName,
		cs(
			c("usdx", 10000000000),
			c("busd", 10000000000),
		),
	)
	suite.Require().NoError(err)

	suite.tApp.InitializeFromGenesisStates(
		NewPricefeedGenStateMulti(suite.tApp.AppCodec()),
		NewHARDGenState(suite.tApp.AppCodec()),
		app.NewFundedGenStateWithSameCoins(
			suite.tApp.AppCodec(),
			cs(
				c("bnb", 10000000000),
				c("busd", 20000000000),
			),
			addrs,
		),
	)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryParams() {
	res, err := suite.queryServer.Params(sdk.WrapSDKContext(suite.ctx), &types.QueryParamsRequest{})
	suite.Require().NoError(err)

	var expected types.GenesisState
	defaultHARDState := NewHARDGenState(suite.tApp.AppCodec())
	suite.tApp.AppCodec().MustUnmarshalJSON(defaultHARDState[types.ModuleName], &expected)

	suite.Equal(expected.Params, res.Params, "params should equal test genesis state")
}

func (suite *grpcQueryTestSuite) TestGrpcQueryAccounts() {
	res, err := suite.queryServer.Accounts(sdk.WrapSDKContext(suite.ctx), &types.QueryAccountsRequest{})
	suite.Require().NoError(err)

	ak := suite.tApp.GetAccountKeeper()
	acc := ak.GetModuleAccount(suite.ctx, types.ModuleName)

	suite.Len(res.Accounts, 1)
	suite.Equal(acc, &res.Accounts[0], "accounts should include module account")
}

func (suite *grpcQueryTestSuite) TestGrpcQueryDeposits_EmptyResponse() {
	res, err := suite.queryServer.Deposits(sdk.WrapSDKContext(suite.ctx), &types.QueryDepositsRequest{})
	suite.Require().NoError(err)
	suite.Require().Empty(res)
}

func (suite *grpcQueryTestSuite) addDeposits() {
	deposits := []struct {
		Address sdk.AccAddress
		Coins   sdk.Coins
	}{
		{
			suite.addrs[0],
			cs(c("bnb", 100000000)),
		},
		{
			suite.addrs[1],
			cs(c("bnb", 20000000)),
		},
		{
			suite.addrs[0],
			cs(c("busd", 20000000)),
		},
		{
			suite.addrs[0],
			cs(c("busd", 8000000)),
		},
	}

	for _, dep := range deposits {
		suite.NotPanics(func() {
			err := suite.keeper.Deposit(suite.ctx, dep.Address, dep.Coins)
			suite.Require().NoError(err)
		})
	}
}

func (suite *grpcQueryTestSuite) addBorrows() {
	borrows := []struct {
		Address sdk.AccAddress
		Coins   sdk.Coins
	}{
		{
			suite.addrs[0],
			cs(c("usdx", 10000000)),
		},
		{
			suite.addrs[1],
			cs(c("usdx", 20000000)),
		},
		{
			suite.addrs[0],
			cs(c("usdx", 40000000)),
		},
		{
			suite.addrs[0],
			cs(c("busd", 80000000)),
		},
	}

	for _, dep := range borrows {
		suite.NotPanics(func() {
			err := suite.keeper.Borrow(suite.ctx, dep.Address, dep.Coins)
			suite.Require().NoErrorf(err, "borrow %s should not error", dep.Coins)
		})
	}
}

func (suite *grpcQueryTestSuite) TestGrpcQueryDeposits() {
	suite.addDeposits()

	tests := []struct {
		giveName          string
		giveRequest       *types.QueryDepositsRequest
		wantDepositCounts int
		shouldError       bool
		errorSubstr       string
	}{
		{
			"empty query",
			&types.QueryDepositsRequest{},
			2,
			false,
			"",
		},
		{
			"owner",
			&types.QueryDepositsRequest{
				Owner: suite.addrs[0].String(),
			},
			// Excludes the second address
			1,
			false,
			"",
		},
		{
			"invalid owner",
			&types.QueryDepositsRequest{
				Owner: "invalid address",
			},
			// No deposits
			0,
			true,
			"decoding bech32 failed",
		},
		{
			"owner and denom",
			&types.QueryDepositsRequest{
				Owner: suite.addrs[0].String(),
				Denom: "bnb",
			},
			// Only the first one
			1,
			false,
			"",
		},
		{
			"owner and invalid denom empty response",
			&types.QueryDepositsRequest{
				Owner: suite.addrs[0].String(),
				Denom: "invalid denom",
			},
			0,
			false,
			"",
		},
		{
			"denom",
			&types.QueryDepositsRequest{
				Denom: "bnb",
			},
			2,
			false,
			"",
		},
	}

	for _, tt := range tests {
		suite.Run(tt.giveName, func() {
			res, err := suite.queryServer.Deposits(sdk.WrapSDKContext(suite.ctx), tt.giveRequest)

			if tt.shouldError {
				suite.Error(err)
				suite.Contains(err.Error(), tt.errorSubstr)
			} else {
				suite.NoError(err)
				suite.Equal(tt.wantDepositCounts, len(res.Deposits))
			}
		})

		// Unsynced deposits should be the same
		suite.Run(tt.giveName+"_unsynced", func() {
			res, err := suite.queryServer.UnsyncedDeposits(sdk.WrapSDKContext(suite.ctx), &types.QueryUnsyncedDepositsRequest{
				Denom:      tt.giveRequest.Denom,
				Owner:      tt.giveRequest.Owner,
				Pagination: tt.giveRequest.Pagination,
			})

			if tt.shouldError {
				suite.Error(err)
				suite.Contains(err.Error(), tt.errorSubstr)
			} else {
				suite.NoError(err)
				suite.Equal(tt.wantDepositCounts, len(res.Deposits))
			}
		})
	}
}

func (suite *grpcQueryTestSuite) TestGrpcQueryBorrows() {
	suite.addDeposits()
	suite.addBorrows()

	tests := []struct {
		giveName          string
		giveRequest       *types.QueryBorrowsRequest
		wantDepositCounts int
		shouldError       bool
		errorSubstr       string
	}{
		{
			"empty query",
			&types.QueryBorrowsRequest{},
			2,
			false,
			"",
		},
		{
			"owner",
			&types.QueryBorrowsRequest{
				Owner: suite.addrs[0].String(),
			},
			// Excludes the second address
			1,
			false,
			"",
		},
		{
			"invalid owner",
			&types.QueryBorrowsRequest{
				Owner: "invalid address",
			},
			// No deposits
			0,
			true,
			"decoding bech32 failed",
		},
		{
			"owner and denom",
			&types.QueryBorrowsRequest{
				Owner: suite.addrs[0].String(),
				Denom: "usdx",
			},
			// Only the first one
			1,
			false,
			"",
		},
		{
			"owner and invalid denom empty response",
			&types.QueryBorrowsRequest{
				Owner: suite.addrs[0].String(),
				Denom: "invalid denom",
			},
			0,
			false,
			"",
		},
		{
			"denom",
			&types.QueryBorrowsRequest{
				Denom: "usdx",
			},
			2,
			false,
			"",
		},
	}

	for _, tt := range tests {
		suite.Run(tt.giveName, func() {
			res, err := suite.queryServer.Borrows(sdk.WrapSDKContext(suite.ctx), tt.giveRequest)

			if tt.shouldError {
				suite.Error(err)
				suite.Contains(err.Error(), tt.errorSubstr)
			} else {
				suite.NoError(err)
				suite.Equal(tt.wantDepositCounts, len(res.Borrows))
			}
		})

		// Unsynced deposits should be the same
		suite.Run(tt.giveName+"_unsynced", func() {
			res, err := suite.queryServer.UnsyncedBorrows(sdk.WrapSDKContext(suite.ctx), &types.QueryUnsyncedBorrowsRequest{
				Denom:      tt.giveRequest.Denom,
				Owner:      tt.giveRequest.Owner,
				Pagination: tt.giveRequest.Pagination,
			})

			if tt.shouldError {
				suite.Error(err)
				suite.Contains(err.Error(), tt.errorSubstr)
			} else {
				suite.NoError(err)
				suite.Equal(tt.wantDepositCounts, len(res.Borrows))
			}
		})
	}
}

func (suite *grpcQueryTestSuite) TestGrpcQueryTotalDeposited() {
	suite.addDeposits()

	totalDeposited, err := suite.queryServer.TotalDeposited(sdk.WrapSDKContext(suite.ctx), &types.QueryTotalDepositedRequest{})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryTotalDepositedResponse{
		SuppliedCoins: cs(
			c("bnb", 100000000+20000000),
			c("busd", 20000000+8000000),
		),
	}, totalDeposited)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryTotalDeposited_Empty() {
	totalDeposited, err := suite.queryServer.TotalDeposited(sdk.WrapSDKContext(suite.ctx), &types.QueryTotalDepositedRequest{})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryTotalDepositedResponse{
		SuppliedCoins: cs(),
	}, totalDeposited)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryTotalDeposited_Denom_Empty() {
	totalDeposited, err := suite.queryServer.TotalDeposited(sdk.WrapSDKContext(suite.ctx), &types.QueryTotalDepositedRequest{
		Denom: "bnb",
	})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryTotalDepositedResponse{
		SuppliedCoins: cs(),
	}, totalDeposited)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryTotalDeposited_Denom() {
	suite.addDeposits()

	totalDeposited, err := suite.queryServer.TotalDeposited(sdk.WrapSDKContext(suite.ctx), &types.QueryTotalDepositedRequest{
		Denom: "bnb",
	})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryTotalDepositedResponse{
		SuppliedCoins: cs(
			c("bnb", 100000000+20000000),
		),
	}, totalDeposited)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryTotalBorrowed() {
	suite.addDeposits()
	suite.addBorrows()

	totalDeposited, err := suite.queryServer.TotalBorrowed(sdk.WrapSDKContext(suite.ctx), &types.QueryTotalBorrowedRequest{})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryTotalBorrowedResponse{
		BorrowedCoins: cs(
			c("usdx", 10000000+20000000+40000000),
			c("busd", 80000000),
		),
	}, totalDeposited)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryTotalBorrowed_denom() {
	suite.addDeposits()
	suite.addBorrows()

	totalDeposited, err := suite.queryServer.TotalBorrowed(sdk.WrapSDKContext(suite.ctx), &types.QueryTotalBorrowedRequest{
		Denom: "usdx",
	})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryTotalBorrowedResponse{
		BorrowedCoins: cs(
			c("usdx", 10000000+20000000+40000000),
		),
	}, totalDeposited)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryInterestRate() {
	tests := []struct {
		giveName          string
		giveDenom         string
		wantInterestRates types.MoneyMarketInterestRates
		shouldError       bool
	}{
		{
			"no denom",
			"",
			types.MoneyMarketInterestRates{
				{
					Denom:              "usdx",
					SupplyInterestRate: "0.000000000000000000",
					BorrowInterestRate: "0.050000000000000000",
				},
				{
					Denom:              "bnb",
					SupplyInterestRate: "0.000000000000000000",
					BorrowInterestRate: "0.000000000000000000",
				},
				{
					Denom:              "busd",
					SupplyInterestRate: "0.000000000000000000",
					BorrowInterestRate: "0.000000000000000000",
				},
			},
			false,
		},
		{
			"denom",
			"usdx",
			types.MoneyMarketInterestRates{
				{
					Denom:              "usdx",
					SupplyInterestRate: "0.000000000000000000",
					BorrowInterestRate: "0.050000000000000000",
				},
			},
			false,
		},
		{
			"invalid denom",
			"bun",
			types.MoneyMarketInterestRates{},
			true,
		},
	}

	for _, tt := range tests {
		suite.Run(tt.giveName, func() {
			res, err := suite.queryServer.InterestRate(sdk.WrapSDKContext(suite.ctx), &types.QueryInterestRateRequest{
				Denom: tt.giveDenom,
			})

			if tt.shouldError {
				suite.Require().Error(err)
			} else {
				suite.Require().NoError(err)

				suite.ElementsMatch(tt.wantInterestRates, res.InterestRates)
			}
		})
	}
}

func (suite *grpcQueryTestSuite) TestGrpcQueryInterestFactors() {
	res, err := suite.queryServer.InterestFactors(sdk.WrapSDKContext(suite.ctx), &types.QueryInterestFactorsRequest{
		Denom: "usdx",
	})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryInterestFactorsResponse{
		InterestFactors: types.InterestFactors{
			{
				Denom:                "usdx",
				BorrowInterestFactor: "1.000000000000000000",
				SupplyInterestFactor: "1.000000000000000000",
			},
		},
	}, res)
}

func (suite *grpcQueryTestSuite) TestGrpcQueryReserves() {
	suite.addDeposits()
	suite.addBorrows()

	res, err := suite.queryServer.Reserves(sdk.WrapSDKContext(suite.ctx), &types.QueryReservesRequest{})
	suite.Require().NoError(err)

	suite.Equal(&types.QueryReservesResponse{
		Amount: sdk.Coins{},
	}, res)
}

func TestGrpcQueryTestSuite(t *testing.T) {
	suite.Run(t, new(grpcQueryTestSuite))
}