1package eu.siacs.conversations.xmpp.jingle;
 2
 3import android.util.Log;
 4
 5import com.google.common.base.CaseFormat;
 6import com.google.common.base.Optional;
 7import com.google.common.base.Preconditions;
 8
 9import org.webrtc.MediaStreamTrack;
10import org.webrtc.PeerConnection;
11import org.webrtc.RtpSender;
12import org.webrtc.RtpTransceiver;
13
14import java.util.UUID;
15
16import javax.annotation.Nonnull;
17import javax.annotation.Nullable;
18
19import eu.siacs.conversations.Config;
20
21class TrackWrapper<T extends MediaStreamTrack> {
22    public final T track;
23    public final RtpSender rtpSender;
24
25    private TrackWrapper(final T track, final RtpSender rtpSender) {
26        Preconditions.checkNotNull(track);
27        Preconditions.checkNotNull(rtpSender);
28        this.track = track;
29        this.rtpSender = rtpSender;
30    }
31
32    public static <T extends MediaStreamTrack> TrackWrapper<T> addTrack(
33            final PeerConnection peerConnection, final T mediaStreamTrack) {
34        final RtpSender rtpSender = peerConnection.addTrack(mediaStreamTrack);
35        return new TrackWrapper<>(mediaStreamTrack, rtpSender);
36    }
37
38    public static <T extends MediaStreamTrack> Optional<T> get(
39            @Nullable final PeerConnection peerConnection, final TrackWrapper<T> trackWrapper) {
40        if (trackWrapper == null) {
41            return Optional.absent();
42        }
43        final RtpTransceiver transceiver =
44                peerConnection == null ? null : getTransceiver(peerConnection, trackWrapper);
45        if (transceiver == null) {
46            Log.w(Config.LOGTAG, "unable to detect transceiver for " + trackWrapper.rtpSender.id());
47            return Optional.of(trackWrapper.track);
48        }
49        final RtpTransceiver.RtpTransceiverDirection direction = transceiver.getDirection();
50        if (direction == RtpTransceiver.RtpTransceiverDirection.SEND_ONLY
51                || direction == RtpTransceiver.RtpTransceiverDirection.SEND_RECV) {
52            return Optional.of(trackWrapper.track);
53        } else {
54            Log.d(Config.LOGTAG, "withholding track because transceiver is " + direction);
55            return Optional.absent();
56        }
57    }
58
59    public static <T extends MediaStreamTrack> RtpTransceiver getTransceiver(
60            @Nonnull final PeerConnection peerConnection, final TrackWrapper<T> trackWrapper) {
61        final RtpSender rtpSender = trackWrapper.rtpSender;
62        for (final RtpTransceiver transceiver : peerConnection.getTransceivers()) {
63            if (transceiver.getSender().id().equals(rtpSender.id())) {
64                return transceiver;
65            }
66        }
67        return null;
68    }
69
70    public static String id(final Class<? extends MediaStreamTrack> clazz) {
71        return String.format(
72                "%s-%s",
73                CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_HYPHEN, clazz.getSimpleName()),
74                UUID.randomUUID().toString());
75    }
76}