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

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

app.controller('fimController', function ($scope, $q, DataFactory, $mdToast, errlog) {
    //Initialisation
    $scope.load = true;
    var objectsArray = [];
    var loadWatch;

    $scope._fimEvent = 'all'
    $scope.files = [];

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

    //Functions

	$scope.setTimer = function (time) {
        $scope.timerFilterValue = time;
    };
	
    $scope.setSort = function (field) {
        if ($scope._sort === field) {
            if ($scope._sortOrder) {
                $scope._sortOrder = false;
                $scope._sort = '';
                DataFactory.filters.unset(objectsArray['/files'], 'filter-sort');
            } else {
                $scope._sortOrder = true;
                DataFactory.filters.set(objectsArray['/files'], 'filter-sort', field);
            }
        } else {
            $scope._sortOrder = false;
            $scope._sort = field;
            DataFactory.filters.set(objectsArray['/files'], 'filter-sort', '-' + field);
        }
    }

    $scope.fileSearchFilter = function (search) {
        if (search) {
            DataFactory.filters.set(objectsArray['/files'], 'search', search);
        } else {
            DataFactory.filters.unset(objectsArray['/files'], 'search');
        }
    };

    $scope.fileEventFilter = function (event) {
        if (event == 'all') {
            DataFactory.filters.unset(objectsArray['/files'], 'event');
        } else {
            DataFactory.filters.set(objectsArray['/files'], 'event', event);
        }
    };

    $scope.filesObj = {
        //Obj with methods for virtual scrolling
        getItemAtIndex: function (index) {
            if ($scope._files_blocked) {
                return null;
            }
            var _pos = index - DataFactory.getOffset(objectsArray['/files']);
            if (DataFactory.filters.flag(objectsArray['/files'])) {
                $scope._files_blocked = true;
                DataFactory.scrollTo(objectsArray['/files'], 50)
                    .then(function (data) {
                        $scope.files.length = 0;
                        $scope.files = data.data.items;
                        DataFactory.filters.unflag(objectsArray['/files']);
                        $scope._files_blocked = false;
                    }, printError);
            } else if ((_pos > 70) || (_pos < 0)) {
                $scope._files_blocked = true;
                DataFactory.scrollTo(objectsArray['/files'], index)
                    .then(function (data) {
                        $scope.files.length = 0;
                        $scope.files = data.data.items;
                        $scope._files_blocked = false;
                    }, printError);
            } else {
                return $scope.files[_pos];
            }
        },
        getLength: function () {
            return DataFactory.getTotalItems(objectsArray['/files']);
        },
    };

    $scope.changeType = function () {
        $scope.showFilesRegistry = !$scope.showFilesRegistry;
        fileTypeFilter();
    };

    var fileTypeFilter = function () {
        DataFactory.filters.set(objectsArray['/files'], 'filetype', $scope.showFilesRegistry ? 'registry' : 'file');
    };

    var createWatch = function () {
        loadWatch = $scope.$watch(function () {
            return $scope.$parent._agent;
        }, function () {
            DataFactory.initialize('get', '/syscheck/' + $scope.$parent._agent.id, {}, 100, 0)
                .then(function (data) {
                    DataFactory.clean(objectsArray['/files']);
                    objectsArray['/files'] = data;
                    DataFactory.get(objectsArray['/files'])
                        .then(function (data) {
                            $scope.files.length = 0;
                            $scope.files = data.data.items;
                            DataFactory.filters.register(objectsArray['/files'], 'search', 'string');
                            DataFactory.filters.register(objectsArray['/files'], 'event', 'string');
                            DataFactory.filters.register(objectsArray['/files'], 'filter-sort', 'string');
                            DataFactory.filters.register(objectsArray['/files'], 'filetype', 'string');
                            $scope._sort = '';
                            $scope.fileSearchFilter($scope._fileSearch);
                            $scope.fileEventFilter($scope._fimEvent);
                            fileTypeFilter();
                        }, printError);
                }, printError);
        });
    };

    var load = function () {
        DataFactory.initialize('get', '/syscheck/' + $scope.$parent._agent.id, {}, 100, 0)
            .then(function (data) {
                objectsArray['/files'] = data;
                DataFactory.get(objectsArray['/files'])
                    .then(function (data) {
                        $scope.files = data.data.items;
                        DataFactory.filters.register(objectsArray['/files'], 'search', 'string');
                        DataFactory.filters.register(objectsArray['/files'], 'event', 'string');
                        DataFactory.filters.register(objectsArray['/files'], 'filter-sort', 'string');
                        DataFactory.filters.register(objectsArray['/files'], 'filetype', 'string');
                        DataFactory.filters.set(objectsArray['/files'], 'filetype', 'file');
                        createWatch();
                        $scope.load = false;
                    }, printError);
            }, printError);
    };

    //Load
    try {
        load();
		$scope.setTimer($scope.$parent.timeFilter);
    } 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 timerWatch = $scope.$watch(function () {
        return $scope.$parent.timeFilter;
    }, function () {
        $scope.setTimer($scope.$parent.timeFilter);
    });
	
    //Destroy
    $scope.$on("$destroy", function () {
        angular.forEach(objectsArray, function (value) {
            DataFactory.clean(value)
        });
        $scope.files.length = 0;
        loadWatch();
		timerWatch();
    });

});