oneuptime/probe/workers/main.js

/**
 *
 * Copyright HackerBay, Inc.
 *
 */
const getApi = require('../utils/api').getApi;
const ApiMonitors = require('./apiMonitors');
const UrlMonitors = require('./urlMonitors');
const ScriptMonitors = require('./scriptMonitors');
const IPMonitors = require('./ipMonitors');
const ServerMonitors = require('./serverMonitors');
const ErrorService = require('../utils/errorService');
const ApplicationSecurity = require('./applicationSecurity');
const ContainerSecurity = require('./containerSecurity');
const IncomingHttpRequestMonitors = require('./incomingHttpRequestMonitors');
const KubernetesMonitors = require('./kubernetesMonitors');

module.exports = {
    runJob: async function() {
        try {
            let monitors = await getApi('probe/monitors');
            monitors = JSON.parse(monitors.data); // parse the stringified data
            await Promise.all(
                monitors.map(monitor => {
                    if (monitor.type === 'api') {
                        return ApiMonitors.ping(monitor);
                    } else if (monitor.type === 'url') {
                        return UrlMonitors.ping(monitor);
                    } else if (monitor.type === 'ip') {
                        return IPMonitors.ping(monitor);
                    } else if (monitor.type === 'script') {
                        return ScriptMonitors.run(monitor);
                    } else if (
                        monitor.type === 'server-monitor' &&
                        monitor.agentlessConfig
                    ) {
                        return ServerMonitors.run(monitor);
                    } else if (monitor.type === 'incomingHttpRequest') {
                        return IncomingHttpRequestMonitors.run(monitor);
                    } else if (monitor.type === 'kubernetes') {
                        return KubernetesMonitors.run(monitor);
                    }

                    return null;
                })
            );
        } catch (error) {
            ErrorService.log('getApi', error);
        }
    },
    runApplicationScan: async function() {
        try {
            const securities = await getApi('probe/applicationSecurities');
            if (securities && securities.length > 0) {
                await Promise.all(
                    securities.map(security => {
                        return ApplicationSecurity.scan(security);
                    })
                );
            }

            return;
        } catch (error) {
            ErrorService.log('runApplicationScan.getApi', error);
        }
    },
    runContainerScan: async function() {
        try {
            const securities = await getApi('probe/containerSecurities');
            if (securities && securities.length > 0) {
                await Promise.all(
                    securities.map(security => {
                        // send a stringified json over the network
                        // fix issue with iv key on the collection (obj.toObject is not a function)
                        return ContainerSecurity.scan(JSON.stringify(security));
                    })
                );
            }

            return;
        } catch (error) {
            ErrorService.log('runContainerScan.getApi', error);
        }
    },
};