package rtc

import (
	"github.com/livekit/protocol/livekit"
	"github.com/livekit/protocol/logger"

	"github.com/livekit/livekit-server/pkg/rtc/types"
)

func HandleParticipantSignal(room types.Room, participant types.LocalParticipant, req *livekit.SignalRequest, pLogger logger.Logger) error {
	switch msg := req.Message.(type) {
	case *livekit.SignalRequest_Offer:
		participant.HandleOffer(FromProtoSessionDescription(msg.Offer))
	case *livekit.SignalRequest_Answer:
		participant.HandleAnswer(FromProtoSessionDescription(msg.Answer))
	case *livekit.SignalRequest_Trickle:
		candidateInit, err := FromProtoTrickle(msg.Trickle)
		if err != nil {
			pLogger.Warnw("could not decode trickle", err)
			return nil
		}
		participant.AddICECandidate(candidateInit, msg.Trickle.Target)
	case *livekit.SignalRequest_AddTrack:
		pLogger.Debugw("add track request", "trackID", msg.AddTrack.Cid)
		participant.AddTrack(msg.AddTrack)
	case *livekit.SignalRequest_Mute:
		participant.SetTrackMuted(livekit.TrackID(msg.Mute.Sid), msg.Mute.Muted, false)
	case *livekit.SignalRequest_Subscription:
		var err error
		// always allow unsubscribe
		if participant.CanSubscribe() || !msg.Subscription.Subscribe {
			updateErr := room.UpdateSubscriptions(
				participant,
				livekit.StringsAsTrackIDs(msg.Subscription.TrackSids),
				msg.Subscription.ParticipantTracks,
				msg.Subscription.Subscribe,
			)
			if updateErr != nil {
				err = updateErr
			}
		} else {
			err = ErrCannotSubscribe
		}
		if err != nil {
			pLogger.Warnw("could not update subscription", err,
				"trackID", msg.Subscription.TrackSids,
				"subscribe", msg.Subscription.Subscribe)
		} else {
			pLogger.Infow("updated subscription",
				"trackID", msg.Subscription.TrackSids,
				"subscribe", msg.Subscription.Subscribe)
		}
	case *livekit.SignalRequest_TrackSetting:
		for _, sid := range livekit.StringsAsTrackIDs(msg.TrackSetting.TrackSids) {
			err := participant.UpdateSubscribedTrackSettings(sid, msg.TrackSetting)
			if err != nil {
				pLogger.Errorw("failed to update subscribed track settings", err, "trackID", sid)
				continue
			}

			pLogger.Infow("updated subscribed track settings", "trackID", sid, "settings", msg.TrackSetting)
		}
	case *livekit.SignalRequest_Leave:
		pLogger.Infow("client leaving room")
		room.RemoveParticipant(participant.Identity(), types.ParticipantCloseReasonClientRequestLeave)
	case *livekit.SignalRequest_UpdateLayers:
		err := room.UpdateVideoLayers(participant, msg.UpdateLayers)
		if err != nil {
			pLogger.Warnw("could not update video layers", err,
				"update", msg.UpdateLayers)
			return nil
		}
	case *livekit.SignalRequest_SubscriptionPermission:
		err := room.UpdateSubscriptionPermission(participant, msg.SubscriptionPermission)
		if err != nil {
			pLogger.Warnw("could not update subscription permission", err,
				"permissions", msg.SubscriptionPermission)
		}
	case *livekit.SignalRequest_SyncState:
		err := room.SyncState(participant, msg.SyncState)
		if err != nil {
			pLogger.Warnw("could not sync state", err,
				"state", msg.SyncState)
		}
	case *livekit.SignalRequest_Simulate:
		err := room.SimulateScenario(participant, msg.Simulate)
		if err != nil {
			pLogger.Warnw("could not simulate scenario", err,
				"simulate", msg.Simulate)
		}
	case *livekit.SignalRequest_DataStreamRequest:
		pLogger.Infow("DataStream request")
		participant.HandleDataStreamRequest(msg.DataStreamRequest)
	}
	return nil
}