wazuh-kibana-app/public/controllers/overview.js

import rison from 'rison-node';

let app = require('ui/modules').get('app/wazuh', []);

app.controller('overviewController', 
function ($scope, $q, $routeParams, $route, $location, $rootScope, appState, genericReq, timefilter) {
    $rootScope.page = 'overview';
    $scope.extensions = appState.getExtensions().extensions;

    // Check the url hash and retrivew the tabView information 
    if ($location.search().tabView) {
        $scope.tabView = $location.search().tabView;
    } else { // If tabView doesn't exist, default it to 'panels' view
        $scope.tabView = 'panels';
        $location.search('tabView', 'panels');
    }

    // Check the url hash and retrivew the tab information 
    if ($location.search().tab) {
        $scope.tab = $location.search().tab;
    } else { // If tab doesn't exist, default it to 'general' view
        $scope.tab = 'general';
        $location.search('tab', 'general');

        // Now we initialize the implicitFilter
        $rootScope.currentImplicitFilter = "";
    }

    // Getting the timefilter, defaulting it otherwise
    if (timefilter.time) {
    	$scope.timeGTE = timefilter.time.from;
    	$scope.timeLT  = timefilter.time.to;
    } else {
    	$scope.timeGTE = "now-15m";
    	$scope.timeLT  = "now";
    }

    // Object for matching nav items and Wazuh groups
    let tabFilters = {
        "general": {
            "group": ""
        },
        "fim": {
            "group": "syscheck"
        },
        "pm": {
            "group": "rootcheck"
        },
        "oscap": {
            "group": "oscap"
        },
        "audit": {
            "group": "audit"
        },
        "pci": {
            "group": ""
        }
    };

    // Listen for changes
    var updateSearchSource = $scope.$on('updateVis', function (event, query, filters) {
        $scope.checkAlerts()
        .then((data) => {
            $scope.results = data;
            $scope.loading = false;
        })
        .catch(() => {
            $scope.results = false;
            $scope.loading = false;
        });
    });

    $scope.hideRing = (items) => {
        return $(".vis-container").length >= items;
    };

    // Switch subtab
    $scope.switchSubtab = (subtab) => {
        $location.search('_a', null);
        $scope.tabView = subtab;
    };

    // Switch tab
    $scope.switchTab = (tab) => {
        // Deleing app state traces in the url
        $location.search('_a', null);

        $scope.loading = true;
        $scope.tab = tab;
        $scope.checkAlerts()
        .then((data) => {
            $scope.results = data;
            $scope.loading = false;
        })
        .catch(() => {
            $scope.results = false;
            $scope.loading = false;
        });
    };

    // Check if there are alerts.
    $scope.checkAlerts = () => {
        let group = null;
        if (tabFilters[$scope.tab].group === '')
            group = "*";
        else group = tabFilters[$scope.tab].group;
        let payload = {};
        let fields  = {
            "fields": [{
                "field": "rule.groups",
                "value": group
            }]
        };

        // No filter needed for general/pci
        if (group === '*'){
            fields = {
                "fields": []
            };
        }

        let clusterName = {
            "cluster": appState.getClusterInfo().cluster
        };

        let timeInterval = {
            "timeinterval": {
                "gte": $scope.timeGTE,
                "lt":  $scope.timeLT
            }
        };
        
        angular.extend(payload, fields, clusterName, timeInterval);

        let deferred = $q.defer();
        genericReq
            .request('POST', '/api/wazuh-elastic/alerts-count/', payload)
            .then((data) => {
                if (data.data.data !== 0){
                    deferred.resolve(true);
                } else {
                    deferred.resolve(false);
                }
            });

        return deferred.promise;
    };

    // Watch for timefilter changes
    $scope.$on('$routeUpdate', () => {
        if ($location.search()._g && $location.search()._g !== '()') {
            let currentTimeFilter = rison.decode($location.search()._g);
            // Check if timefilter has changed and update values
            if (currentTimeFilter.time && 
                ($scope.timeGTE != currentTimeFilter.time.from || 
                $scope.timeLT != currentTimeFilter.time.to)) {

                $scope.timeGTE = currentTimeFilter.time.from;
                $scope.timeLT  = currentTimeFilter.time.to;
                $scope.checkAlerts()
                .then((data) => $scope.results = data)
                .catch(() => $scope.results = false);

            }
        }
    });

    // Watchers
    $scope.$watch('tabView', () => $location.search('tabView', $scope.tabView));
    $scope.$watch('tab', () => {
        $location.search('tab', $scope.tab);

        // Update the implicit filter
        if (tabFilters[$scope.tab].group === "") $rootScope.currentImplicitFilter = "";
        else $rootScope.currentImplicitFilter = 'rule.groups : "' + tabFilters[$scope.tab].group + '"';
    });

    // Check alerts
    $scope.checkAlerts()
    .then((data) => {
        $scope.results = data;
        $scope.loading = false;
    })
    .catch(() => {
        $scope.results = false;
        $scope.loading = false;
    });
});

app.controller('overviewGeneralController', function ($scope) {
});

app.controller('overviewFimController', function ($scope) {
});

app.controller('overviewPMController', function ($scope) {
});

app.controller('overviewOSCAPController', function ($scope) {
});

app.controller('overviewAuditController', function ($scope) {
});

app.controller('overviewPCIController', function ($scope, genericReq) {
    let tabs = [];
    
    genericReq
        .request('GET', '/api/wazuh-api/pci/all')
        .then((data) => {
            angular.forEach(data.data, (value, key) => {
                tabs.push({
                    "title":   key,
                    "content": value
                });
            });
        });

    $scope.tabs = tabs;
    $scope.selectedIndex = 0;
});