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

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

app.controller('osseclogController', function ($scope, DataFactory, $sce, $interval, $mdToast, errlog) {
    //Initialization
    $scope.load = true;
    $scope.text = [];
    $scope.realtime = false;

    $scope.$parent.state.setManagerState('logs');

    var objectsArray = [];

    var _category = 'all';
    var _level = 'all';
    var _promise;

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

    //Functions
    $scope.textObj = {
        //Obj with methods for virtual scrolling
        getItemAtIndex: function (index) {
            if ($scope.blocked) {
                return null;
            }
            var _pos = index - DataFactory.getOffset(objectsArray['/manager/logs']);
            if (DataFactory.filters.flag(objectsArray['/manager/logs'])) {
                $scope.blocked = true;
                DataFactory.scrollTo(objectsArray['/manager/logs'], 75)
                    .then(function (data) {
                        $scope.text.length = 0;
                        $scope.text = data.data.items;
                        DataFactory.filters.unflag(objectsArray['/manager/logs']);
                        $scope.blocked = false;
                    }, printError);
            } else if ((_pos > 225) || (_pos < 0)) {
                $scope.blocked = true;
                DataFactory.scrollTo(objectsArray['/manager/logs'], index)
                    .then(function (data) {
                        $scope.text.length = 0;
                        $scope.text = data.data.items;
                        $scope.blocked = false;
                    }, printError);
            } else {
                return $scope.text[_pos];
            }
        },
        getLength: function () {
            if ($scope.realtime) {
                return 120;
            }
            var tmpLength = DataFactory.getTotalItems(objectsArray['/manager/logs']);
            if (tmpLength === 0) {
                return 120;
            }
            return tmpLength;
        },
    };

    $scope.categoryFilter = function (category) {
        if (category) {
            DataFactory.filters.set(objectsArray['/manager/logs'], 'category', category);
        } else {
            DataFactory.filters.unset(objectsArray['/manager/logs'], 'category');
        }
    };

    $scope.levelFilter = function (level) {
        if (level) {
            DataFactory.filters.set(objectsArray['/manager/logs'], 'type_log', level);
        } else {
            DataFactory.filters.unset(objectsArray['/manager/logs'], 'type_log');
        }
    };

    var loadSummary = function () {
        DataFactory.getAndClean('get', '/manager/logs/summary', {})
            .then(function (data) {
                $scope.summary = data.data;
                $scope.load = false;
            }, printError);
    };

    $scope.colorLine = function (line) {
        if ((!line) || (line == null)) {
            return $sce.trustAsHtml('-');
        }
        line = line.replace('INFO', '<span style="color:#0066ff">INFO</span>');
        line = line.replace('ERROR', '<span style="color:#ff0000">ERROR</span>');
        line = line.replace('ossec-remoted', '<span style="color:#827e7d">ossec-remoted</span>');
        line = line.replace('ossec-testrule', '<span style="color:#827e7d">ossec-testrule</span>');
        line = line.replace('wazuh-moduled', '<span style="color:#827e7d">wazuh-moduled</span>');
        line = line.replace('ossec-monitord', '<span style="color:#827e7d">ossec-monitord</span>');
        line = line.replace('ossec-logcollector', '<span style="color:#827e7d">ossec-logcollector</span>');
        line = line.replace('ossec-execd', '<span style="color:#827e7d">ossec-execd</span>');
        line = line.replace('ossec-rootcheck', '<span style="color:#827e7d">ossec-rootcheck</span>');
        line = line.replace('rootcheck', '<span style="color:#827e7d">rootcheck</span>');
        line = line.replace('ossec-syscheckd', '<span style="color:#827e7d">ossec-syscheckd</span>');
        line = line.replace('ossec-analysisd', '<span style="color:#827e7d">ossec-analysisd</span>');
        line = line.replace('ossec-maild', '<span style="color:#827e7d">ossec-maild</span>');
        line = line.replace(line.substring(0, 19), '<b>' + line.substring(0, 19) + '</b>');
        return $sce.trustAsHtml(line);
    };

    $scope.playRealtime = function () {
        $scope.realtime = !$scope.realtime;
        if (!$scope.realtime) {
            $interval.cancel(_promise);
        } else {
            _promise = $interval(function () {
                DataFactory.get(objectsArray['/manager/logs'])
                    .then(function (data) {
                        $scope.text.length = 0;
                        $scope.text = data.data.items;
                    }, printError);
            }, 1000);
        }
    };

    var load = function () {
        DataFactory.initialize('get', '/manager/logs', {}, 250, 0)
            .then(function (data) {
                objectsArray['/manager/logs'] = data;
                DataFactory.filters.register(objectsArray['/manager/logs'], 'category', 'string');
                DataFactory.filters.register(objectsArray['/manager/logs'], 'type_log', 'string');
                DataFactory.get(data).then(function (data) {
                    $scope.text = data.data.items;
                    loadSummary();
                }, 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 () {
        $interval.cancel(_promise);
        angular.forEach(objectsArray, function (value) {
            DataFactory.clean(value)
        });
    });
});