package synctree

import (
	"context"
	"github.com/anyproto/any-sync/commonspace/object/tree/treechangeproto"
	"github.com/anyproto/any-sync/commonspace/peermanager"
	"github.com/anyproto/any-sync/commonspace/requestmanager"
	"github.com/anyproto/any-sync/commonspace/spacesyncproto"
	"go.uber.org/zap"
)

type SyncClient interface {
	RequestFactory
	Broadcast(msg *treechangeproto.TreeSyncMessage)
	SendUpdate(peerId, objectId string, msg *treechangeproto.TreeSyncMessage) (err error)
	QueueRequest(peerId, objectId string, msg *treechangeproto.TreeSyncMessage) (err error)
	SendRequest(ctx context.Context, peerId, objectId string, msg *treechangeproto.TreeSyncMessage) (reply *spacesyncproto.ObjectSyncMessage, err error)
}

type syncClient struct {
	RequestFactory
	spaceId        string
	requestManager requestmanager.RequestManager
	peerManager    peermanager.PeerManager
}

func NewSyncClient(spaceId string, requestManager requestmanager.RequestManager, peerManager peermanager.PeerManager) SyncClient {
	return &syncClient{
		RequestFactory: &requestFactory{},
		spaceId:        spaceId,
		requestManager: requestManager,
		peerManager:    peerManager,
	}
}
func (s *syncClient) Broadcast(msg *treechangeproto.TreeSyncMessage) {
	objMsg, err := MarshallTreeMessage(msg, s.spaceId, msg.RootChange.Id, "")
	if err != nil {
		return
	}
	err = s.peerManager.Broadcast(context.Background(), objMsg)
	if err != nil {
		log.Debug("broadcast error", zap.Error(err))
	}
}

func (s *syncClient) SendUpdate(peerId, objectId string, msg *treechangeproto.TreeSyncMessage) (err error) {
	objMsg, err := MarshallTreeMessage(msg, s.spaceId, objectId, "")
	if err != nil {
		return
	}
	return s.peerManager.SendPeer(context.Background(), peerId, objMsg)
}

func (s *syncClient) SendRequest(ctx context.Context, peerId, objectId string, msg *treechangeproto.TreeSyncMessage) (reply *spacesyncproto.ObjectSyncMessage, err error) {
	objMsg, err := MarshallTreeMessage(msg, s.spaceId, objectId, "")
	if err != nil {
		return
	}
	return s.requestManager.SendRequest(ctx, peerId, objMsg)
}

func (s *syncClient) QueueRequest(peerId, objectId string, msg *treechangeproto.TreeSyncMessage) (err error) {
	objMsg, err := MarshallTreeMessage(msg, s.spaceId, objectId, "")
	if err != nil {
		return
	}
	return s.requestManager.QueueRequest(peerId, objMsg)
}

func MarshallTreeMessage(message *treechangeproto.TreeSyncMessage, spaceId, objectId, replyId string) (objMsg *spacesyncproto.ObjectSyncMessage, err error) {
	payload, err := message.Marshal()
	if err != nil {
		return
	}
	objMsg = &spacesyncproto.ObjectSyncMessage{
		ReplyId:  replyId,
		Payload:  payload,
		ObjectId: objectId,
		SpaceId:  spaceId,
	}
	return
}