user.service.ts

import { Injectable } from '@angular/core';
import { HttpClient, HttpErrorResponse } from '@angular/common/http';

import { Observable, BehaviorSubject, of, EMPTY } from 'rxjs';
import { map, catchError } from 'rxjs/operators';

import { CookieService } from 'ngx-cookie-service';
import { StompConfig, StompRService } from '@stomp/ng2-stompjs';
import { Message } from '@stomp/stompjs';

import { SnackBarService } from './snackbar.service';
import { IdpService } from './idp.service';

import { User, Update, State, Deployment, SSHKey, NewSSHKey, Group, IdP, Service } from './types/types.module';


@Injectable()
export class UserService {
  private initialized = false;
  private loggedIn: boolean = false;

  private user: User;
  private user$: BehaviorSubject<User>;

  private sshKeys: SSHKey[] = new Array<SSHKey>();
  private sshKeys$ = new BehaviorSubject<SSHKey[]>([]);

  private deployments: Map<number, Deployment> = new Map([]);
  private deployments$ = new BehaviorSubject<Deployment[]>([]);

  public groupSelector = (user: User) => user ? user.groups : [];
  public serviceSelector = (user: User) => user ? user.services ? user.services : [] : [];

  constructor(
    private cookieService: CookieService,
    private http: HttpClient,
    private snackBar: SnackBarService,
    private idpService: IdpService,
    private stompService: StompRService,
  ) {
    this.user$ = new BehaviorSubject(null);
    this.connect();
  }

  // PRIVATE API
  private connect(): void {
    this.fetch();

    this.subscribeUser().subscribe(
      (newUser: User) => {
        console.log('user$:', newUser);

        if (newUser == undefined || newUser == null) {
          // LOGGED OUT
          // show the logout to the user
          if (this.loggedIn) {
            this.snackBar.open('Logged out');

            // purge all values
            this.stompService.disconnect();
            this.user = undefined;
            this.deployments = new Map([]);
            this.deployments$.next([]);
            this.sshKeys = [];
            this.sshKeys$.next([]);
          }


          this.loggedIn =  false;
        } else {
          // LOGGED IN
          // show the login to the user
          if (!this.loggedIn) {
            this.snackBar.open('Logged in');
          }


          if (newUser.id) {
            this.connectLiveUpdates(newUser.id);
          }
          if (newUser.ssh_keys) {
            this.sshKeys = newUser.ssh_keys;
            this.sshKeys$.next(this.sshKeys);
          }
          if (newUser.deployments) {
            newUser.deployments.forEach(
              (newDep: Deployment) => {
                this.deployments.set(newDep.id,  newDep);
              }
            );
            this.deployments$.next(Array.from(this.deployments.values()));
          }

          this.loggedIn = true;
          this.initialized = true;
        }
      },
      this.handleError(true),
      () => console.log('user$ is complete'),
    );
  }

  private connectLiveUpdates(userID: number): void {
    // handle with care
    let login = userID
    let passcode = this.cookieService.get('sessionid');

    const stompConfig: StompConfig = {
      // Which server?
      url: 'wss://'+window.location.host+'/ws',

      // Headers
      // Typical keys: login, passcode, host
      headers: {
        login: 'webpage-client:' + login,
        passcode: passcode,
      },

      // How often to heartbeat?
      // Interval in milliseconds, set to 0 to disable
      heartbeat_in: 0, // Typical value 0 - disabled
      heartbeat_out: 20000, // Typical value 20000 - every 20 seconds
      // Wait in milliseconds before attempting auto reconnect
      // Set to 0 to disable
      // Typical value 15000 (15 seconds)
      reconnect_delay: 15000,

      // Will log diagnostics on console
      debug: false,
    };
    this.stompService.config = stompConfig;
    this.stompService.initAndConnect();

    let subscription = this.stompService.subscribe(
      '/exchange/users/' + userID.toString()
    );

    subscription.subscribe(
      (message: Message) => {
        let update: Update = JSON.parse(message.body);
        console.log('update:', update);

        if (update.error && update.error != '') {
          this.snackBar.open(update.error);
        }

        if (update.deployment) {
          this.updateDeployment(update.deployment);
        }
      },
      this.logErrorAndFetch,
    );
  }

  private updateDeployment(dep: Deployment): void {
    if (dep) {
      this.deployments.set(dep.id, dep);
      this.deployments$.next(Array.from(this.deployments.values()));
    }
  }

  private updateState(update: State): void {
    if (update) {
      // report an occured error
      if (update.error) {
        this.snackBar.open(update.error);
      }

      this.user = update.user;
      this.user$.next(this.user);

    } else {
      this.user$.next(undefined);
    }
  }

  private fetch(): void {
    this.http.get<State>('/backend/api/state').subscribe(
      (state: State) => this.updateState(state),
      this.handleError(false, 'Error fetching state. Try again later'),
    );
  }

  private handleError(fetch: boolean, msg?: string) {
    return (error: any) => {
      if (error.status == 403) {
        this.login();
        return
      }

      if (msg) {
        this.snackBar.open(msg);
      }
      console.log('fetch:', error);
      if (fetch) {
        this.fetch();
      }
    }
  }

  private catchError(fetch: boolean, msg?: string) {
    return (error: any) => {
      this.handleError(fetch, msg)(error);
      return EMPTY;
    }
  }

  private logErrorAndFetch(error: any)  {
    if (error.status === 403) {
      this.login();
    }

    console.log(error);
    this.snackBar.open('Error');
    this.fetch();
    return Observable.throw(error);
  }

  // PUBLIC API
  public serviceDescription(service: Service): string {
    if (service.description != "") {
      return service.description;
    }
    return "No description";
  }

  public login(idp?: IdP): void {
    if (idp) {
      this.idpService.setIdPPreference(idp);
    }

    window.location.href = '/backend/auth/v1/request';
  }

  public logout(): void {
    this.http.post<State>('/backend/auth/v1/logout', {}).subscribe(
      (state: State) => this.updateState(state),
      this.logErrorAndFetch,
    );
  }

  public sentQuestionnaire(stateItemID: number, answers: Object) {
    return this.http.post<Deployment>('/backend/api/questionnaire?id='+String(stateItemID), answers).subscribe(
      (dep: Deployment) => this.updateDeployment(dep),
      this.logErrorAndFetch,
    );
  }

  public deleteUser() {
    return this.http.delete('/backend/api/delete_user').subscribe(
      (data: {deleted: boolean}) => {
        if (data && data.deleted) {
          this.user$.next(undefined);
          this.snackBar.open('Deleted user from server');
        }
      },
      this.logErrorAndFetch,
    );
  }

  public uploadSshKey(key: NewSSHKey) {
    console.log('Uploading key:', key);
    const body = {
      'type': 'add',
      'key': key,
    };
    return this.http.post<SSHKey>('/backend/api/sshkey', body).subscribe(
      (newKey: SSHKey) => {
        this.sshKeys.push(newKey);
        this.sshKeys$.next(this.sshKeys);
      },
      this.logErrorAndFetch,
    );
  }

  public removeSshKey(key: SSHKey) {
    console.log('Deleting key:', key);
    return this.http.post('/backend/api/sshkey', {
      'type': 'remove',
      'id': key.id,
    }).subscribe(
      (data: {deleted: boolean}) => {
        if (data && data.deleted) {
          this.sshKeys = this.sshKeys.filter(
            k => key.id != k.id
          );
          this.sshKeys$.next(this.sshKeys);
        }
      },
      this.logErrorAndFetch,
    );
  }

  public changeDeployment(action: string, group: Group): Observable<Deployment> {
    const body = {
      'type': action,
      'group': group.id,
    };
    return this.http.post<Deployment>('/backend/api/deployments', body).pipe(
      catchError(this.catchError(true, "Error changing deployment")),
    );
  }

  // DATA SERVICE API
  //
  public subscribeSpecific<T>(selector: (user: User) => T): Observable<T> {
    return this.subscribeUser().pipe(map(selector));
  }

  public subscribeSSHKeys(): Observable<SSHKey[]> {
    return this.sshKeys$.asObservable();
  }

  public subscribeUser(): Observable<User> {
    return this.user$.asObservable();
  }

  public subscribeDeployments(): Observable<Deployment[]> {
    return this.deployments$.asObservable();
  }

  public subscribeGroups(): Observable<Group[]> {
    let groupSelector = (user: User) => user ? user.groups : [];
    return this.subscribeSpecific<Group[]>(groupSelector);
  }

  public subscribeServiceDeployment(service: Service): Observable<Deployment> {
    return this.subscribeDeployments().pipe(
      map((deployments: Deployment[]) => {
          return deployments.find(
            (dep: Deployment) => dep.service ? dep.service.id == service.id : false
          );
        }
      ),
    );
  }

  public subscribeGroupDeployment(group: Group): Observable<Deployment> {
    return this.subscribeDeployments().pipe(
      map((deployments: Deployment[]) => {
          return deployments.find(
            (dep: Deployment) => dep.group == group.id
          );
        }
      ),
    );
  }
}