metron-interface/metron-alerts/src/app/service/update.service.ts

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import {Injectable} from '@angular/core';
import { HttpClient } from '@angular/common/http';
import {Observable} from 'rxjs';
import { forkJoin as observableForkJoin, Subject } from 'rxjs';
import { catchError, map } from 'rxjs/operators';

import {HttpUtil} from '../utils/httpUtil';
import {Alert} from '../model/alert';
import {PatchRequest} from '../model/patch-request';
import {Utils} from '../utils/utils';
import {Patch} from '../model/patch';
import { GlobalConfigService } from './global-config.service';
import {CommentAddRemoveRequest} from "../model/comment-add-remove-request";
import { AppConfigService } from './app-config.service';
import {AlertSource} from "../model/alert-source";

@Injectable()
export class UpdateService {

  alertChangedSource = new Subject<AlertSource>();
  alertChanged$ = this.alertChangedSource.asObservable();
  sourceType = 'source:type';
  alertCommentChangedSource = new Subject<AlertSource>();
  alertCommentChanged$ = this.alertCommentChangedSource.asObservable();

  constructor(private http: HttpClient, private globalConfigService: GlobalConfigService, private appConfigService: AppConfigService) {
    this.globalConfigService.get().subscribe((config: {}) => {
      this.sourceType = config['source.type.field'];
    });
  }

  public addComment(commentRequest: CommentAddRemoveRequest, fireChangeListener = true): Observable<AlertSource> {
    let url = this.appConfigService.getApiRoot() + '/update/add/comment';
    return this.http.post(url, commentRequest).pipe(
    catchError(HttpUtil.handleError),
    map(result => {
      let alertSource = result['document'];
      if (fireChangeListener) {
        this.alertCommentChangedSource.next(alertSource);
      }
      return alertSource;
    }));
  }

  public removeComment(commentRequest: CommentAddRemoveRequest, fireChangeListener = true): Observable<AlertSource> {
    let url = this.appConfigService.getApiRoot() + '/update/remove/comment';
    return this.http.post(url, commentRequest).pipe(
    catchError(HttpUtil.handleError),
    map(result => {
      let alertSource = result['document'];
      if (fireChangeListener) {
        this.alertCommentChangedSource.next(alertSource);
      }
      return alertSource;
    }));
  }

  public patch(patchRequest: PatchRequest, fireChangeListener = true): Observable<AlertSource> {
    let url = this.appConfigService.getApiRoot() + '/update/patch';
    return this.http.patch(url, patchRequest).pipe(
    catchError(HttpUtil.handleError),
    map(result => {
      let alertSource = result['document'];
      if (fireChangeListener) {
        this.alertChangedSource.next(alertSource);
      }
      return alertSource;
    }),);
  }

  public updateAlertState(alerts: Alert[], state: string, fireChangeListener = true): Observable<{}> {
    let patchRequests: PatchRequest[] = alerts.map(alert => {
      let patchRequest = new PatchRequest();
      patchRequest.guid = alert.source.guid;
      patchRequest.sensorType = Utils.getAlertSensorType(alert, this.sourceType);
      patchRequest.patch = [new Patch('add', '/alert_status', state)];
      return patchRequest;
    });
    let patchObservables = [];
    for (let patchRequest of patchRequests) {
      patchObservables.push(this.patch(patchRequest, fireChangeListener));
    }
    return observableForkJoin(patchObservables);
  }
}