MediaStream.js

'use strict';

const EventTarget = require('event-target-shim');
const WebRTCModule = require('react-native').NativeModules.WebRTCModule;

const MediaStreamTrackEvent = require('./MediaStreamTrackEvent');

import type MediaStreamTrack from './MediaStreamTrack';

const MEDIA_STREAM_EVENTS = [
  'active',
  'inactive',
  'addtrack',
  'removetrack',
];

class MediaStream extends EventTarget(MEDIA_STREAM_EVENTS) {
  id: number; // NOTE: spec wants string here
  active: boolean = true;

  onactive: ?Function;
  oninactive: ?Function;
  onaddtrack: ?Function;
  onremovetrack: ?Function;

  _tracks: Array<MediaStreamTrack> = [];

  constructor(streamId) {
    super();
    this.id = streamId;
  }

  addTrack(track: MediaStreamTrack) {
    this._tracks.push(track);
    this.dispatchEvent(new MediaStreamTrackEvent('addtrack', {track}));
  }

  removeTrack(track: MediaStreamTrack) {
    let index = this._tracks.indexOf(track);
    if (index === -1) {
      return;
    }
    this._tracks.splice(index, 1);
    this.dispatchEvent(new MediaStreamTrackEvent('removetrack', {track}));
  }

  getTracks(): Array<MediaStreamTrack> {
    return this._tracks.slice();
  }

  getTrackById(trackId): ?MediaStreamTrack {
    return this._tracks.find(track => track.id === trackId);
  }

  getAudioTracks(): Array<MediaStreamTrack> {
    return this._tracks.filter(track => track.kind === 'audio');
  }

  getVideoTracks(): Array<MediaStreamTrack> {
    return this._tracks.filter(track => track.kind === 'video');
  }

  clone() {
    throw new Error('Not implemented.');
  }

  toURL() {
    return this.id;
  }

  release() {
    WebRTCModule.mediaStreamRelease(this.id);
  }
}

module.exports = MediaStream;