package ibc

import (
	sdk "github.com/cosmos/cosmos-sdk/types"

	client "github.com/cosmos/ibc-go/v10/modules/core/02-client"
	clientv2 "github.com/cosmos/ibc-go/v10/modules/core/02-client/v2"
	connection "github.com/cosmos/ibc-go/v10/modules/core/03-connection"
	channel "github.com/cosmos/ibc-go/v10/modules/core/04-channel"
	channelv2 "github.com/cosmos/ibc-go/v10/modules/core/04-channel/v2"
	"github.com/cosmos/ibc-go/v10/modules/core/keeper"
	"github.com/cosmos/ibc-go/v10/modules/core/types"
)

// InitGenesis initializes the ibc state from a provided genesis
// state.
func InitGenesis(ctx sdk.Context, k keeper.Keeper, gs *types.GenesisState) {
	client.InitGenesis(ctx, k.ClientKeeper, gs.ClientGenesis)
	clientv2.InitGenesis(ctx, k.ClientV2Keeper, gs.ClientV2Genesis)
	connection.InitGenesis(ctx, k.ConnectionKeeper, gs.ConnectionGenesis)
	channel.InitGenesis(ctx, k.ChannelKeeper, gs.ChannelGenesis)
	channelv2.InitGenesis(ctx, k.ChannelKeeperV2, gs.ChannelV2Genesis)
}

// ExportGenesis returns the ibc exported genesis.
func ExportGenesis(ctx sdk.Context, k keeper.Keeper) *types.GenesisState {
	return &types.GenesisState{
		ClientGenesis:     client.ExportGenesis(ctx, k.ClientKeeper),
		ClientV2Genesis:   clientv2.ExportGenesis(ctx, k.ClientV2Keeper),
		ConnectionGenesis: connection.ExportGenesis(ctx, k.ConnectionKeeper),
		ChannelGenesis:    channel.ExportGenesis(ctx, k.ChannelKeeper),
		ChannelV2Genesis:  channelv2.ExportGenesis(ctx, k.ChannelKeeperV2),
	}
}