wazuh-kibana-app/public/factories/vis-handlers.js

/*
 * Wazuh app - Factory to store visualizations handlers
 * 
 * Copyright (C) 2018 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.
 */
import { uiModules } from 'ui/modules'
import dateMath from '@kbn/datemath';
const app = uiModules.get('app/wazuh', []);

app.factory('visHandlers', function() {
    let list = [];

    const addItem = item => {
        list.push(item);
    };

    const getList = () => {
        return list;
    };

    const getAppliedFilters = syscollector => {
        const appliedFilters = {};

        if(syscollector){
            Object.assign(appliedFilters, {
                filters: syscollector,
                time:{
                    from: 'now-1d/d',
                    to: 'now'
                },
                searchBar: false,
                tables:[]
            });
            return appliedFilters;
        }
        
        // Check raw response from all rendered tables
        const tables = list.filter(item => item._scope &&
                                           item._scope.savedObj &&
                                           item._scope.savedObj.vis &&
                                           item._scope.savedObj.vis._state &&
                                           item._scope.savedObj.vis._state.type === 'table'
                            )
                            .map(item => {
                                const columns = []; 
                                for(const agg of item._scope.savedObj._source.visState.aggs){
                                    if(agg.type === 'count') continue;
                                    if(agg.params && agg.params.customLabel){
                                        columns.push(agg.params.customLabel);
                                    } else {
                                        columns.push('Column');
                                    }
                                }
                                columns.push('Count');

                                return (item._scope &&
                                        item._scope.savedObj &&
                                        item._scope.savedObj.vis &&
                                        item._scope.savedObj.vis.searchSource &&
                                        item._scope.savedObj.vis.searchSource.rawResponse) ? 
                                        {
                                            rawResponse: item._scope.savedObj.vis.searchSource.rawResponse,
                                            title: (item._scope && item._scope.savedObj && item._scope.savedObj.title) ?
                                                    item._scope.savedObj.title :
                                                    'Table',
                                            columns
                                        } : 
                                        false;
                            });
        
        if(list && list.length) {
            // Parse applied filters for the first visualization
            const filters = list[0]._scope.savedObj.vis.API.queryFilter.getFilters();
     
            // Parse current time range
            const { from, to } = list[0]._scope.savedObj.vis.API.timeFilter.time;

            Object.assign(appliedFilters, {
                filters,
                time:{
                    from: dateMath.parse(from),
                    to: dateMath.parse(to)
                },
                searchBar: list[0] && list[0]._scope && list[0]._scope.appState && list[0]._scope.appState.query && list[0]._scope.appState.query.query ? 
                           list[0]._scope.appState.query.query :
                           false,
                tables
            });
        }

        return appliedFilters;
    };

    const hasData = () => {
        for(const item of list) {
            if(item && item._scope && item._scope.savedObj && item._scope.savedObj.searchSource && 
                item._scope.savedObj.searchSource.rawResponse &&
                item._scope.savedObj.searchSource.rawResponse.hits &&
                item._scope.savedObj.searchSource.rawResponse.hits.total){
                    return true;
                }
        }
        return false;
    };

    const removeAll = () => {
        for(const item of list){
            if(item && item._scope){
                item._scope.$destroy();
            }
            if(item && item._scope && item._scope.savedObj){
                item._scope.savedObj.destroy();
            }
        }
        list = [];
    };
  
    return {
      addItem,
      getList,
      removeAll,
      hasData,
      getAppliedFilters
    };
});