package robin

import (
	"context"
	"errors"

	"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
	logging "github.com/tim-beatham/wgmesh/pkg/log"
	"github.com/tim-beatham/wgmesh/pkg/rpc"
)

type RobinRpc struct {
	rpc.UnimplementedMeshCtrlServerServer
	Server *ctrlserver.MeshCtrlServer
}

func nodeToRpcNode(node ctrlserver.MeshNode) *rpc.MeshNode {
	return &rpc.MeshNode{
		PublicKey:  node.PublicKey,
		WgEndpoint: node.WgEndpoint,
		WgHost:     node.WgHost,
		Endpoint:   node.HostEndpoint,
	}
}

func nodesToRpcNodes(nodes map[string]ctrlserver.MeshNode) []*rpc.MeshNode {
	n := len(nodes)
	meshNodes := make([]*rpc.MeshNode, n)

	var i int = 0

	for _, v := range nodes {
		meshNodes[i] = nodeToRpcNode(v)
		i++
	}

	return meshNodes
}

func (m *RobinRpc) GetMesh(ctx context.Context, request *rpc.GetMeshRequest) (*rpc.GetMeshReply, error) {
	mesh := m.Server.MeshManager.GetMesh(request.MeshId)

	if mesh == nil {
		return nil, errors.New("mesh does not exist")
	}

	meshBytes := mesh.Save()

	reply := rpc.GetMeshReply{
		Mesh: meshBytes,
	}

	return &reply, nil
}

func (m *RobinRpc) JoinMesh(ctx context.Context, request *rpc.JoinMeshRequest) (*rpc.JoinMeshReply, error) {
	mesh := m.Server.MeshManager.GetMesh(request.MeshId)

	logging.InfoLog.Println("[JOINING MESH]: " + request.MeshId)

	if mesh == nil {
		return nil, errors.New("mesh does not exist")
	}

	err := m.Server.MeshManager.UpdateMesh(request.MeshId, request.Changes)

	if err != nil {
		return nil, err
	}

	return &rpc.JoinMeshReply{Success: true}, nil
}