fastlane/ci

import {DOCUMENT} from '@angular/common';
import {Inject, Injectable} from '@angular/core';
import {Observable} from 'rxjs/Observable';
import {Observer} from 'rxjs/Observer';
import {Subject} from 'rxjs/Subject';
import {AuthService} from './auth.service';

export interface BuildLogMessageEvent extends MessageEvent {
  data: string;
}

@Injectable()
export class BuildLogWebsocketService {
  private readonly API_ROOT: string;
  private readonly authService: AuthService;

  constructor(@Inject(DOCUMENT) document: Document, authService: AuthService) {
    this.authService = authService;
    // TODO(snatchev): consider using wss once we have tls setup.
    this.API_ROOT = `ws://${document.location.host}`;
  }

  connect(projectId: string, buildNumber: number):
      Observable<BuildLogMessageEvent> {
    const socket = this.createSocket(projectId, buildNumber);
    const observable =
        Observable.create((observer: Observer<BuildLogMessageEvent>) => {
          socket.onmessage = observer.next.bind(observer);
          socket.onerror = observer.error.bind(observer);
          socket.onclose = observer.complete.bind(observer);
          return socket.close.bind(socket);
        });

    return observable;
  }

  /** Public for testing */
  createSocket(projectId: string, buildNumber: number): WebSocket {
    return new WebSocket(this.createSocketUrl(projectId, buildNumber));
  }

  private createSocketUrl(projectId: string, buildNumber: number) {
    const token = this.authService.token();
    return `${this.API_ROOT}/data/projects/${projectId}/build/${buildNumber}/log.ws?bearer_token=${token}`;
  }
}