hastic-server/server/src/services/notification_service.ts

import { Segment } from '../models/segment_model';
import * as AnalyticUnit from '../models/analytic_unit_model';
import { HASTIC_WEBHOOK_URL, HASTIC_WEBHOOK_TYPE, HASTIC_WEBHOOK_SECRET, HASTIC_INSTANCE_NAME } from '../config';

import axios from 'axios';
import * as querystring from 'querystring';

enum ContentType {
  JSON = 'application/json',
  URLENCODED ='application/x-www-form-urlencoded'
}

export enum WebhookType {
  DETECT = 'DETECT',
  FAILURE = 'FAILURE',
  RECOVERY = 'RECOVERY',
  MESSAGE = 'MESSAGE'
}

export declare type AnalyticAlert = {
  type: WebhookType,
  analyticUnitType: string,
  analyticUnitName: string,
  analyticUnitId: AnalyticUnit.AnalyticUnitId,
  grafanaUrl: string,
  from: number,
  to: number
  params?: any,
  regionImage?: any
}

export declare type InfoAlert = {
  type: WebhookType,
  message: string,
  from: number,
  to: number,
  params?: any
}

// TODO: send webhook with payload without dep to AnalyticUnit
export async function sendAnalyticWebhook(alert: AnalyticAlert) {
  const fromTime = new Date(alert.from).toLocaleTimeString();
  const toTime = new Date(alert.to).toLocaleTimeString();
  console.log(`Sending alert unit:${alert.analyticUnitName} from: ${fromTime} to: ${toTime}`);

  let payload;
  if(HASTIC_WEBHOOK_TYPE === ContentType.JSON) {
    payload = JSON.stringify(alert);
  } else if(HASTIC_WEBHOOK_TYPE === ContentType.URLENCODED) {
    payload = querystring.stringify(alert);
  } else {
    throw new Error(`Unknown webhook type: ${HASTIC_WEBHOOK_TYPE}`);
  }
  sendWebhook(payload);
}

export async function sendInfoWebhook(alert: InfoAlert) {
  if(alert && typeof alert === 'object') {
    console.log(`Sending info webhook ${JSON.stringify(alert.message)}`);
    sendWebhook(alert, ContentType.JSON);
  } else {
    console.error(`skip sending Info webhook, got corrupted message ${alert}`);
  }
}

export async function sendWebhook(payload: any, contentType = HASTIC_WEBHOOK_TYPE) {
  if(HASTIC_WEBHOOK_URL === null) {
    throw new Error(`Can't send alert, HASTIC_WEBHOOK_URL is undefined`);
  }

  payload.instanceName = HASTIC_INSTANCE_NAME;

  // TODO: use HASTIC_WEBHOOK_SECRET
  const options = {
    method: 'POST',
    url: HASTIC_WEBHOOK_URL,
    data: payload,
    headers: { 'Content-Type': contentType }
  };

  try {
    await axios(options);
  } catch(err) {
    console.error(`Can't send alert to ${HASTIC_WEBHOOK_URL}. Error: ${err.message}`);
  }
}