wazuh-kibana-app/public/services/api-tester.js

/*
 * Wazuh app - API test service
 * Copyright (C) 2018 Wazuh, Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Find more information about this on the LICENSE file.
 */

import chrome from 'ui/chrome';
const app = require('ui/modules').get('app/wazuh', []);

app.service('testAPI', function ($http, $location, $rootScope, appState, genericReq) {
    return {
        check_stored: async data => {
            try {
                const headers = {headers:{ "Content-Type": 'application/json' },timeout: $rootScope.userTimeout || 8000};

                /** Checks for outdated cookies */
                const current     = appState.getCreatedAt();
                const lastRestart = $rootScope.lastRestart;

                if(current && lastRestart && lastRestart > current){
                    appState.removeCurrentPattern();
                    appState.removeCurrentAPI();
                    appState.removeClusterInfo();
                    appState.removeCreatedAt();
                    delete $rootScope.lastRestart;

                    const configuration = await genericReq.request('GET', '/api/wazuh-api/configuration', {})

                    appState.setPatternSelector(typeof configuration.data.data['ip.selector'] !== 'undefined' ? configuration.data.data['ip.selector'] : true)

                    return 'cookies_outdated';
                    /** End of checks for outdated cookies */

                } else {
                    if(appState.getUserCode()) headers.headers.code = appState.getUserCode();

                    const result = await Promise.all([
                        genericReq.request('GET', '/api/wazuh-api/configuration', {}),
                        $http.post(chrome.addBasePath('/api/wazuh-api/checkStoredAPI'), data,headers)
                    ]);

                    appState.setPatternSelector(typeof result[0].data.data['ip.selector'] !== 'undefined' ? result[0].data.data['ip.selector'] : true)

                    if(result[1].error) {
                        return Promise.reject(result[1])
                    }
                    return result[1];
                }
            } catch (error) {
                if(error.status && error.status === -1){
                    $rootScope.apiIsDown = true;
                    return Promise.reject({data: 'request_timeout_checkstored'});
                } else {
                    return Promise.reject(error);
                }
            }
        },
        check: async data => {
            try {
                const headers = {headers:{ "Content-Type": 'application/json' },timeout: $rootScope.userTimeout || 8000};
                if(appState.getUserCode()) headers.headers.code = appState.getUserCode();

                const url = chrome.addBasePath("/api/wazuh-api/checkAPI");
                const response = await $http.post(url, data, headers);

                if (response.error) {
                    return Promise.reject(response);
                }

                return response;

            } catch(error) {
                if(error.data && error.data.message && error.data.message.includes('ENOTFOUND')) {
                    return Promise.reject({data: 'invalid_url'});
                } else if(error.data && error.data.message && error.data.message.includes('ECONNREFUSED')) {
                    return Promise.reject({data: 'invalid_port'});
                } else if(error.status && error.status === -1){
                    return Promise.reject({data: 'request_timeout_checkapi'});
                } else if (error.data && error.data.message && error.data.message === 'wrong_credentials') {
                    return Promise.reject({data: 'wrong_credentials'});
                } else if(error.data && ((error.data.message && error.data.message === 'socket hang up') || (parseInt(error.data.error) === 5))) {
                    return Promise.reject({data:'socket_hang_up',extraMessage: `Wazuh API throws ${error.data.message}`, https: (data.url && data.url.includes('https'))});
                } else {
                    return Promise.reject(error);
                }
            }
        }
    };
});