wazuh-kibana-app/public/factories/tab-visualizations.js

/*
 * Wazuh app - Factory to store visualization tabs
 *
 * Copyright (C) 2015-2019 Wazuh, Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Find more information about this on the LICENSE file.
 */
export class TabVisualizations {
  /**
   * Class constructor
   */
  constructor() {
    this.agents = {
      welcome: 0,
      general: 7,
      fim: 7,
      pm: 4,
      vuls: 10,
      oscap: 13,
      ciscat: 11,
      audit: 15,
      gdpr: 6,
      pci: 6,
      virustotal: 6,
      configuration: 0,
      osquery: 5
    };

    this.overview = {
      welcome: 0,
      general: 13,
      fim: 8,
      pm: 5,
      vuls: 10,
      oscap: 11,
      ciscat: 11,
      audit: 15,
      pci: 5,
      gdpr: 5,
      aws: 6,
      virustotal: 7,
      osquery: 5
    };

    this.tabVisualizations = {};
    this.currentTab = '';
  }

  /**
   * Set given tab as current
   * @param {Object} tab
   */
  setTab(tab) {
    this.currentTab = tab;
  }

  /**
   * Get current tab
   */
  getTab() {
    return this.currentTab;
  }

  /**
   * Get given item
   * @param {String} item
   */
  getItem(item) {
    return this.tabVisualizations[item];
  }

  /**
   * Assign tab visualization with given tabs
   * @param {Object} tabs
   */
  assign(tabs) {
    if (typeof tabs === 'object') {
      this.tabVisualizations = tabs;
    } else if (typeof tabs === 'string') {
      this.tabVisualizations =
        tabs === 'overview' ? this.overview : this.agents;
    }
  }

  /**
   * Remove all tab Visualizations
   */
  removeAll() {
    this.tabVisualizations = {};
  }
}