/* -*- 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 ) {
        super(_server, 'Icinga2 API');
      }

    refresh ( extension ) {
        this.extension = extension;

        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 ( encodeURI ( entry.real_host_name ), encodeURI ( entry.real_service_display_name ) );

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

        log ( '> Body: ' + message.request_body.data );

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

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

    handlePollMessage ( _httpSession, message )
    {
        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' ];
                
                for ( var entry of json.results )
                {
                    this.status.service_status.push ( {
                        status: _statuses [ entry.attrs.state ],
                        host_name: entry.attrs.host_name,
                        service_display_name: entry.attrs.display_name,
                        attempts: '%d/%d'.format(entry.attrs.check_attempt,entry.attrs.max_check_attempts),
                        status_information: entry.attrs.last_check_result.output,
                        last_check: new Date ( entry.attrs.last_state_change * 1000 ) . toString(),
                    } );
                }
            }
        }
        catch ( e )
        {
            log ( '> ERROR: ' + e );
            log ( '> DATA: ' + _data );
        }

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