wazuh-kibana-app/public/controllers/management/decoders.js

/*
 * Wazuh app - Ruleset controllers
 * 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 * as FileSaver from '../../services/file-saver'

const colors = [
    '#004A65', '#00665F', '#BF4B45', '#BF9037', '#1D8C2E', 'BB3ABF',
    '#00B1F1', '#00F2E2', '#7F322E', '#7F6025', '#104C19', '7C267F',
    '#0079A5', '#00A69B', '#FF645C', '#FFC04A', '#2ACC43', 'F94DFF',
    '#0082B2', '#00B3A7', '#401917', '#403012', '#2DD947', '3E1340',
    '#00668B', '#008C83', '#E55A53', '#E5AD43', '#25B23B', 'E045E5'
];

const app = uiModules.get('app/wazuh', []);

app.controller('decodersController', function ($scope, $rootScope, $sce, errorHandler, appState, csvReq, wzTableFilter) {
    $scope.setRulesTab = tab => $rootScope.globalRulesetTab = tab;

    $scope.appliedFilters = [];

    //Initialization
    $scope.searchTerm = '';
    $scope.viewingDetail = false;
    $scope.typeFilter = "all";
    $scope.setRulesTab('decoders');
    $scope.isArray = Array.isArray;

    $scope.includesFilter = filterName => $scope.appliedFilters.map(item => item.name).includes(filterName);
    
    $scope.getFilter      = filterName => {
        const filtered = $scope.appliedFilters.filter(item => item.name === filterName);
        return filtered.length ? filtered[0].value : '';
    }

    $scope.removeFilter   = filterName => {
        $scope.appliedFilters = $scope.appliedFilters.filter(item => item.name !== filterName);
        return $scope.$broadcast('wazuhRemoveFilter',{filterName});
    }

    $scope.colorRegex = regex => {
        regex = regex.toString();
        let valuesArray   = regex.match(/\(((?!<\/span>).)*?\)(?!<\/span>)/gmi);
        let coloredString = regex;
        for (let i = 0, len = valuesArray.length; i < len; i++) {
            coloredString = coloredString.replace(/\(((?!<\/span>).)*?\)(?!<\/span>)/mi, '<span style="color: ' + colors[i] + ' ">' + valuesArray[i] + '</span>');
        }
        return $sce.trustAsHtml(coloredString);
    };

    $scope.colorOrder = order => {
        order = order.toString();
        let valuesArray   = order.split(',');
        let coloredString = order;
        for (let i = 0, len = valuesArray.length; i < len; i++) {
            coloredString = coloredString.replace(valuesArray[i], '<span style="color: ' + colors[i] + ' ">' + valuesArray[i] + '</span>');
        }
        return $sce.trustAsHtml(coloredString);
    };

    // Reloading event listener
    $scope.$on('rulesetIsReloaded',() => {
        $scope.viewingDetail = false;
        if(!$scope.$$phase) $scope.$digest();
    });

    $scope.search = term => {
        if(term && term.startsWith('path:') && term.split('path:')[1].trim()) {
            $scope.custom_search = ''
            const filter = {name:'path',value:term.split('path:')[1].trim()};
            $scope.appliedFilters = $scope.appliedFilters.filter(item => item.name !== 'path');
            $scope.appliedFilters.push(filter)
            $scope.$broadcast('wazuhFilter',{filter})
        } else if(term && term.startsWith('file:') && term.split('file:')[1].trim()) {
            $scope.custom_search = ''
            const filter = {name:'file',value:term.split('file:')[1].trim()};
            $scope.appliedFilters = $scope.appliedFilters.filter(item => item.name !== 'file');
            $scope.appliedFilters.push(filter)
            $scope.$broadcast('wazuhFilter',{filter})
        } else {
            $scope.$broadcast('wazuhSearch',{term,removeFilters: true})
        }
    }

    $scope.onlyParents = typeFilter => {
        $scope.appliedFilters = [];
        if(typeFilter === 'all') $scope.$broadcast('wazuhUpdateInstancePath',{path:'/decoders'})
        else $scope.$broadcast('wazuhUpdateInstancePath',{path:'/decoders/parents'})
    }

    $scope.downloadCsv = async () => {
        try {
            const currentApi   = JSON.parse(appState.getCurrentAPI()).id;
            const output       = await csvReq.fetch('/decoders', currentApi, wzTableFilter.get());
            const blob         = new Blob([output], {type: 'text/csv'}); // eslint-disable-line

            FileSaver.saveAs(blob, 'decoders.csv');

            return;

        } catch (error) {
            errorHandler.handle(error,'Download CSV');
        }
        return;
    }

    /**
     * This function takes back to the list but adding a filter from the detail view
     */
    $scope.addDetailFilter = (name, value) => {
        $scope.appliedFilters.push({name,value})
        // Clear the autocomplete component
        $scope.searchTerm = '';
        // Go back to the list
        $scope.closeDetailView();
    }

    $scope.$on('wazuhShowDecoder',(event,parameters) => {
        $scope.currentDecoder = parameters.decoder;
        $scope.viewingDetail = true;
        if(!$scope.$$phase) $scope.$digest();
    })

    /**
     * This function changes to the decoders list view
     */
    $scope.closeDetailView = clear => {
        if(clear) $scope.appliedFilters = $scope.appliedFilters.slice(0,$scope.appliedFilters.length-1);
        $scope.viewingDetail = false;
        $scope.currentDecoder = false;
        if(!$scope.$$phase) $scope.$digest();
    }
});