package handlers

import (
	"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/proto"
	"github.com/ledgerwatch/erigon/cmd/lightclient/sentinel/proto/p2p"
	"github.com/ledgerwatch/erigon/cmd/lightclient/utils"
	"github.com/ledgerwatch/log/v3"
)

// These below are the handlers for heartbeat functions

func (c *ConsensusHandlers) goodbyeHandler(ctx *proto.StreamContext, dat *p2p.Goodbye) error {
	//log.Info("[Lightclient] Received", "goodbye", dat.Reason)
	_, err := ctx.Codec.WritePacket(dat)
	if err != nil {
		return err
	}
	c.peers.DisconnectPeer(ctx.Stream.Conn().RemotePeer())
	return nil
}

// type safe handlers which all have access to the original stream & decompressed data
// ping handler
func pingHandler(ctx *proto.StreamContext, dat *p2p.Ping) error {
	// since packets are just structs, they can be resent with no issue
	_, err := ctx.Codec.WritePacket(dat)
	if err != nil {
		return err
	}
	return nil
}

// TODO: respond with proper metadata
func metadataHandler(ctx *proto.StreamContext, dat *proto.EmptyPacket) error {
	return nil
}

func statusHandler(ctx *proto.StreamContext, dat *p2p.Status) error {
	log.Info("[ReqResp] Status",
		"epoch", dat.FinalizedEpoch,
		"final root", utils.BytesToHex(dat.FinalizedRoot),
		"head root", utils.BytesToHex(dat.HeadRoot),
		"head slot", dat.HeadSlot,
		"fork digest", utils.BytesToHex(dat.ForkDigest),
	)
	return nil
}