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

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

app.controller('managerController', function ($scope, DataFactory, genericReq, $mdDialog, $mdToast, errlog) {
    //Initialisation
    $scope.load = true;
    $scope.$parent.state.setManagerState('status');
    $scope.timeFilter = "24h";

    $scope.stats = [];
    $scope.stats['/top/agent'] = '-';
    $scope.stats['/overview/alerts'] = { "alerts": 0, "ip": "-", "group": "-" };
    $scope.stats['/overview/fim'] = { "alerts": 0, "agent": "-", "file": "-" };

    //Print Error
    var printError = function (error) {
        $mdToast.show({
            template: '<md-toast>' + error.html + '</md-toast>',
            position: 'bottom left',
            hideDelay: 5000,
        });
    };

    //Functions
    $scope.getDaemonStatusClass = function (daemonStatus) {
        if (daemonStatus == "running")
            return "status green"
        else if (daemonStatus == "stopped")
            return "status red";
        else
            return "status red";
    };

    $scope.setTimer = function (time) {
        if (time == "24h") {
            $scope.timerFilterValue = "24h";
        } else if (time == "48h") {
            $scope.timerFilterValue = "48h";
        } else {
            $scope.timerFilterValue = "7d";
        }
    };

    var load_tops = function () {
        var daysAgo = 1;
        if ($scope.timerFilterValue == "7d") {
            var daysAgo = 7;
        } else if ($scope.timerFilterValue == "48h") {
            var daysAgo = 2;
        } else {
            var daysAgo = 1;
        }
        var date = new Date();
        date.setDate(date.getDate() - daysAgo);
        var timeAgo = date.getTime();
        //timeAgo = "";
        genericReq.request('GET', '/api/wazuh-elastic/top/srcuser/' + timeAgo)
            .then(function (data) {
                $scope.topsrcuser = data.data;
            }, printError);
        genericReq.request('GET', '/api/wazuh-elastic/top/srcip/' + timeAgo)
            .then(function (data) {
                $scope.topsrcip = data.data;
            }, printError);
        genericReq.request('GET', '/api/wazuh-elastic/top/rule.groups/' + timeAgo)
            .then(function (data) {
                $scope.topgroup = data.data;
            }, printError);
        genericReq.request('GET', '/api/wazuh-elastic/top/rule.PCI_DSS/' + timeAgo)
            .then(function (data) {
                $scope.toppci = data.data;
            }, printError);
    };

    var load = function () {
        DataFactory.getAndClean('get', '/agents/summary', {})
            .then(function (data) {
                $scope.agentsCountActive = data.data.active;
                $scope.agentsCountDisconnected = data.data.disconnected;
                $scope.agentsCountNeverConnected = data.data.neverConnected;
                $scope.agentsCountTotal = data.data.total;
                DataFactory.getAndClean('get', '/manager/status', {})
                    .then(function (data) {
                        $scope.daemons = data.data;
                        $scope.load = false;
                    }, printError);
            }, printError);
    };

    //Load
    try {
        load();
        load_tops();
    } catch (e) {
        $mdToast.show({
            template: '<md-toast> Unexpected exception loading controller </md-toast>',
            position: 'bottom left',
            hideDelay: 5000,
        });
        errlog.log('Unexpected exception loading controller', e);
    }

    // Timer filter watch
    var loadWatch = $scope.$watch(function () {
        return $scope.$parent.timeFilter;
    }, function () {
        $scope.setTimer($scope.$parent.timeFilter);
        load_tops();
    });


    //Destroy
    $scope.$on("$destroy", function () {
        $scope.stats.length = 0;
        loadWatch();
    });

});

app.controller('managerConfigurationController', function ($scope, DataFactory, errlog) {
    //Initialisation
    $scope.load = true;
    $scope.$parent.state.setManagerState('configuration');

    //Print Error
    var printError = function (error) {
        $mdToast.show({
            template: '<md-toast>' + error.html + '</md-toast>',
            position: 'bottom left',
            hideDelay: 5000,
        });
    };

    //Functions
    var parseConfiguration = function () {
        if ($scope.managerConfiguration.rules.decoder) {
            if (angular.isString($scope.managerConfiguration.rules.decoder)) { $scope.managerConfiguration.rules.decoder = [$scope.managerConfiguration.rules.decoder] }
        }
        if ($scope.managerConfiguration.rules.rule_dir) {
            if (angular.isString($scope.managerConfiguration.rules.rule_dir)) { $scope.managerConfiguration.rules.rule_dir = [$scope.managerConfiguration.rules.rule_dir] }
        }
        if ($scope.managerConfiguration.rules.list) {
            if (angular.isString($scope.managerConfiguration.rules.list)) { $scope.managerConfiguration.rules.list = [$scope.managerConfiguration.rules.list] }
        }
        if ($scope.managerConfiguration.rootcheck.system_audit) {
            if (angular.isString($scope.managerConfiguration.rootcheck.system_audit)) { $scope.managerConfiguration.rootcheck.system_audit = [$scope.managerConfiguration.rootcheck.system_audit] }
        }
    };

    var load = function () {
        DataFactory.getAndClean('get', '/manager/status', {})
            .then(function (data) {
                $scope.daemons = data.data;
                DataFactory.getAndClean('get', '/manager/configuration', {})
                    .then(function (data) {
                        $scope.managerConfiguration = data.data;
                        parseConfiguration();
                        $scope.load = false;
                    }, printError);
            }, printError);
    };

    //Load
    try {
        load();
    } catch (e) {
        $mdToast.show({
            template: '<md-toast> Unexpected exception loading controller </md-toast>',
            position: 'bottom left',
            hideDelay: 5000,
        });
        errlog.log('Unexpected exception loading controller', e);
    }

    //Destroy
    $scope.$on("$destroy", function () {
        $scope.managerConfiguration.length = 0;
    });

});