import { BehaviorSubject, distinctUntilChanged } from 'rxjs';
import { AudioDeviceManagerState } from './AudioDeviceManagerState';
import { AudioBitrateProfile } from '../gen/video/sfu/models/models';
import { ScreenShareSettings } from '../types';
import { RxUtils } from '../store';

export class ScreenShareState extends AudioDeviceManagerState<DisplayMediaStreamOptions> {
  private audioEnabledSubject = new BehaviorSubject<boolean>(true);
  private settingsSubject = new BehaviorSubject<
    ScreenShareSettings | undefined
  >(undefined);

  /**
   * An Observable that emits the current screen share audio status.
   */
  audioEnabled$ = this.audioEnabledSubject
    .asObservable()
    .pipe(distinctUntilChanged());

  /**
   * An Observable that emits the current screen share settings.
   */
  settings$ = this.settingsSubject.asObservable();

  /**
   * Constructs a new ScreenShareState instance.
   */
  constructor() {
    super('stop-tracks', undefined, AudioBitrateProfile.MUSIC_HIGH_QUALITY);
  }

  /**
   * @internal
   */
  protected override getDeviceIdFromStream = (
    stream: MediaStream,
  ): string | undefined => {
    const [track] = stream.getTracks();
    return track?.getSettings().deviceId;
  };

  /**
   * The current screen share audio status.
   */
  get audioEnabled() {
    return RxUtils.getCurrentValue(this.audioEnabled$);
  }

  /**
   * Set the current screen share audio status.
   */
  setAudioEnabled(isEnabled: boolean) {
    RxUtils.setCurrentValue(this.audioEnabledSubject, isEnabled);
  }

  /**
   * The current screen share settings.
   */
  get settings() {
    return RxUtils.getCurrentValue(this.settings$);
  }

  /**
   * Set the current screen share settings.
   *
   * @param settings the screen share settings to set.
   */
  setSettings(settings: ScreenShareSettings | undefined) {
    RxUtils.setCurrentValue(this.settingsSubject, settings);
  }
}