monito/servers/icinga2api.js

/* -*- mode: js; js-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
  Monito Gnome-Shell extension
  Copyright (C) 2021  Benjamin Drieu

  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.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License along
  with this program; if not, write to the Free Software Foundation, Inc.,
  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

  SPDX-License-Identifier: GPL-2.0-or-later
*/

const { Soup } = imports.gi;

const ExtensionUtils = imports.misc.extensionUtils;
const Lang = imports.lang;
const Main = imports.ui.main;
const Me = ExtensionUtils.getCurrentExtension();
const Preferences = Me.imports.prefs;
const GenericServer = Me.imports.servers.genericserver.GenericServer;


class Icinga2API extends GenericServer {

    constructor ( _server, extension ) {
        super(_server, extension, 'Icinga2 API');
    }

    refresh ( ) {
        this.buildURL ( );
        this.prepareHttp ( );
    }

    prepareHttp ( )
    {
        super.prepareHttp ( );

        let message = Soup.form_request_new_from_hash ( 'GET', this.urlcgi + '/objects/services', { } );
        message.request_headers.append ( 'Accept', 'application/json' );

        this.authenticateAndSend ( message, this.handlePollMessage );
    }

    recheck ( entry )
    {
        let message = Soup.form_request_new_from_hash ( 'POST', this.urlcgi + '/actions/reschedule-check', { } );

        let params = '{ "type": "Service", "filter": "host.name==\\"%s\\" && service.name==\\"%s\\"", "force": true, "pretty": true }' . format ( entry.real_host_name, entry.real_service_display_name );

        message.request_body.truncate();
        message.request_body.append ( params );
        message.request_body.flatten();
        message.button = entry.button;

        message.request_headers.append ( 'Accept', 'application/json' );

        this.authenticateAndSend ( message, this.handleCMDMessage );
    }

    recheckAll ( button )
    {
        let message = Soup.form_request_new_from_hash ( 'POST', this.urlcgi + '/actions/reschedule-check', { } );

        let params = '{ "type": "Service", "force": true, "pretty": true }';

        message.request_body.truncate();
        message.request_body.append ( params );
        message.request_body.flatten();
        message.button = button;

        message.request_headers.append ( 'Accept', 'application/json' );

        this.authenticateAndSend ( message, this.handleCMDMessage );
    }

    handlePollMessage ( _httpSession, message )
    {
        log ( 'handlePollMessage' );

        let _data = super.handleMessage ( _httpSession, message );
        try
        {
            if ( this.error )
                log ( 'Parent error ' + this.error );
            else
            {
                let json = JSON.parse ( _data );
                let _statuses = [ 'OK', 'WARNING', 'CRITICAL', 'UNKNOWN', 'PENDING' ];
                
                for ( var entry of json.results )
                {
//                    log ( JSON.stringify(entry) );
                    this.status.service_status.push ( {
                        status: _statuses[entry.attrs.state],
                        host_name: entry.attrs.host_name,
                        service_display_name: entry.attrs.display_name,
                        has_been_acknowledged: parseInt(entry.attrs.acknowledgement),
                        attempts: '%d/%d'.format(entry.attrs.check_attempt,entry.attrs.max_check_attempts),
                        status_information: ( entry.attrs.last_check_result ? entry.attrs.last_check_result.output : _statuses[entry.attrs.state] ),
                        last_check: ( entry.attrs.last_check ? this.formatDate(parseInt(entry.attrs.last_check)) : '' ),
                    } );
                }
            }
        }
        catch ( e )
        {
            log ( '> ERROR: ' + e );
            log ( '> DATA: ' + _data );
        }

        this.extension.refreshUI ( this );
        return ! this.error;
    }

    getUrlForService ( service )
    {
        return this._settings.get_string ( 'url' ) ;
    }   

}