package destination

import (
	pb "github.com/linkerd/linkerd2-proxy-api/go/destination"
	"github.com/sirupsen/logrus"
	v1 "k8s.io/api/core/v1"
)

type endpointProfileTranslator struct {
	pod      *v1.Pod
	port     uint32
	endpoint *pb.WeightedAddr
	stream   pb.Destination_GetProfileServer
	log      *logrus.Entry
}

// newEndpointProfileTranslator translates protocol updates to
// DestinationProfiles for endpoints. When a Server on the cluster is updated
// it is possible that it selects an endpoint that is being watched, if that
// is the case then an update will be sent to the client if the Server has
// changed the endpoint's supported protocol—mainly being opaque or not.
func newEndpointProfileTranslator(pod *v1.Pod, port uint32, endpoint *pb.WeightedAddr, stream pb.Destination_GetProfileServer, log *logrus.Entry) *endpointProfileTranslator {
	return &endpointProfileTranslator{
		pod:      pod,
		port:     port,
		endpoint: endpoint,
		stream:   stream,
		log:      log,
	}
}

func (ept *endpointProfileTranslator) UpdateProtocol(opaqueProtocol bool) {
	// The protocol for an endpoint should only be updated if there is a pod,
	// endpoint, and the endpoint has a protocol hint. If there is an endpoint
	// but it does not have a protocol hint, that means we could not determine
	// if it has a peer proxy so a opaque traffic would not be supported.
	if ept.pod != nil && ept.endpoint != nil && ept.endpoint.ProtocolHint != nil {
		if !opaqueProtocol {
			ept.endpoint.ProtocolHint.OpaqueTransport = nil
		} else if ept.endpoint.ProtocolHint.OpaqueTransport == nil {
			port, err := getInboundPort(&ept.pod.Spec)
			if err != nil {
				ept.log.Error(err)
			} else {
				ept.endpoint.ProtocolHint.OpaqueTransport = &pb.ProtocolHint_OpaqueTransport{
					InboundPort: port,
				}
			}
		}

	}
	profile := ept.createDefaultProfile(opaqueProtocol)
	ept.log.Debugf("sending protocol update: %+v", profile)
	ept.stream.Send(profile)
}

func (ept *endpointProfileTranslator) createDefaultProfile(opaqueProtocol bool) *pb.DestinationProfile {
	return &pb.DestinationProfile{
		RetryBudget:    defaultRetryBudget(),
		Endpoint:       ept.endpoint,
		OpaqueProtocol: opaqueProtocol,
	}
}