package keeper

import (
	"context"

	"github.com/0glabs/0g-chain/crypto/bn254util"
	"github.com/0glabs/0g-chain/x/dasigners/v1/types"
	sdk "github.com/cosmos/cosmos-sdk/types"
	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
	"github.com/ethereum/go-ethereum/common"
	etherminttypes "github.com/evmos/ethermint/types"
)

var _ types.MsgServer = &Keeper{}

func (k Keeper) RegisterSigner(goCtx context.Context, msg *types.MsgRegisterSigner) (*types.MsgRegisterSignerResponse, error) {
	ctx := sdk.UnwrapSDKContext(goCtx)
	// validate sender
	valAddr, err := sdk.ValAddressFromHex(msg.Signer.Account)
	if err != nil {
		return nil, err
	}
	_, found := k.stakingKeeper.GetValidator(ctx, valAddr)
	if !found {
		return nil, stakingtypes.ErrNoValidatorFound
	}
	_, found, err = k.GetSigner(ctx, msg.Signer.Account)
	if err != nil {
		return nil, err
	}
	if found {
		return nil, types.ErrSignerExists
	}
	// validate signature
	chainID, err := etherminttypes.ParseChainID(ctx.ChainID())
	if err != nil {
		return nil, err
	}
	hash := types.PubkeyRegistrationHash(common.HexToAddress(msg.Signer.Account), chainID)
	if !msg.Signer.ValidateSignature(hash, bn254util.DeserializeG1(msg.Signature)) {
		return nil, types.ErrInvalidSignature
	}
	// save signer
	if err := k.SetSigner(ctx, *msg.Signer); err != nil {
		return nil, err
	}
	return &types.MsgRegisterSignerResponse{}, nil
}

func (k Keeper) UpdateSocket(goCtx context.Context, msg *types.MsgUpdateSocket) (*types.MsgUpdateSocketResponse, error) {
	ctx := sdk.UnwrapSDKContext(goCtx)
	signer, found, err := k.GetSigner(ctx, msg.Account)
	if err != nil {
		return nil, err
	}
	if !found {
		return nil, types.ErrSignerNotFound
	}
	signer.Socket = msg.Socket
	if err := k.SetSigner(ctx, signer); err != nil {
		return nil, err
	}
	return &types.MsgUpdateSocketResponse{}, nil
}

func (k Keeper) RegisterNextEpoch(goCtx context.Context, msg *types.MsgRegisterNextEpoch) (*types.MsgRegisterNextEpochResponse, error) {
	ctx := sdk.UnwrapSDKContext(goCtx)
	// get signer
	signer, found, err := k.GetSigner(ctx, msg.Account)
	if err != nil {
		return nil, err
	}
	if !found {
		return nil, types.ErrSignerNotFound
	}
	// validate signature
	epochNumber, err := k.GetEpochNumber(ctx)
	if err != nil {
		return nil, err
	}
	chainID, err := etherminttypes.ParseChainID(ctx.ChainID())
	if err != nil {
		return nil, err
	}
	hash := types.EpochRegistrationHash(common.HexToAddress(msg.Account), epochNumber+1, chainID)
	if !signer.ValidateSignature(hash, bn254util.DeserializeG1(msg.Signature)) {
		return nil, types.ErrInvalidSignature
	}
	// save registration
	k.SetRegistration(ctx, epochNumber+1, msg.Account, msg.Signature)
	return &types.MsgRegisterNextEpochResponse{}, nil
}