/* -*- 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
 */

/* exported init */

const GETTEXT_DOMAIN = 'monito';

let _httpSession;
let _status;
let _ok_text;

const ExtensionUtils = imports.misc.extensionUtils;
const Lang = imports.lang;
const Main = imports.ui.main;
const Mainloop = imports.mainloop;
const Me = ExtensionUtils.getCurrentExtension();
const PanelMenu = imports.ui.panelMenu;
const PopupMenu = imports.ui.popupMenu;

const Gettext = imports.gettext.domain(GETTEXT_DOMAIN);
const _ = Gettext.gettext;

const { GObject, St, Clutter, Gio } = imports.gi;

const SETTINGS_SCHEMA = "org.gnome.shell.extensions.monito";
const SETTINGS_SCHEMA_ACCOUNT = "org.gnome.shell.extensions.monito.account";
const SETTINGS_SCHEMA_ACCOUNT_PATH = "/org/gnome/shell/extensions/monito/account";

const Convenience = Me.imports.convenience;
const Icinga = Me.imports.servers.icinga.Icinga;
const Icinga2 = Me.imports.servers.icinga2.Icinga2;
const Preferences = Me.imports.prefs;

let settings = Convenience.getSettings(SETTINGS_SCHEMA);
let account_settings = [ ];


const Indicator = GObject.registerClass(
class Indicator extends PanelMenu.Button {
    _init() {
        super._init(0.0, _('Monito Checker'));

	this.initStatus ( );

        this.namesBoxes = { };
        this.warningBoxes = { };
        this.criticalBoxes = { };
        this.unknownBoxes = { };

        this.initUI ( );

        settings.connect("changed::servers", Lang.bind ( this, function(stgs, key) {
            monitoLog ( this );
            this.remove_all_children ( );
            this.initUI ( );
        } ) );
    }

    initUI ( ) {
	let box = new St.BoxLayout ( { } );
        this.add_child(box);

        for ( let _server of Preferences.getServersList() )
        {
            monitoLog ( '> Server ' + _server );
            account_settings [ _server ] = Preferences.getAccountSettings ( _server );
            let _account_settings = account_settings [ _server ];
            
	    let serverBox = new St.BoxLayout ( { style_class: 'monito-serverbox' } );
            box.add_child(serverBox);

	    let name_box = new St.BoxLayout( { style_class: 'monito-namebox' } );
	    this.namesBoxes [ _server ] = new St.Label ( { text: _account_settings.get_string ( 'name' ) } );
            name_box.add_child ( this.namesBoxes [ _server ] );
            serverBox.add_child(name_box);
            _account_settings.bind ( 'name', this.namesBoxes [ _server ], 'text', Gio.SettingsBindFlags.GET );

	    let warning_box = new St.BoxLayout({ style_class: 'monito-warning-box monito-box' });
            warning_box.set_style ( 'background-color: ' + _account_settings.get_string ( 'warning-color' ) );
            _account_settings.connect("changed::warning-color", Lang.bind ( { widget: warning_box }, setColor ) );

	    this.warningBoxes [ _server ] = new St.Label({ text: String(_status['WARNING']) })
            warning_box.add_child ( this.warningBoxes [ _server ] );
            serverBox.add_child(warning_box);

            let critical_box = new St.BoxLayout({ style_class: 'monito-critical-box monito-box' });
            critical_box.set_style ( 'background-color: ' + _account_settings.get_string ( 'critical-color' ) );
            _account_settings.connect("changed::critical-color", Lang.bind ( { widget: critical_box }, setColor ) );

	    this.criticalBoxes [ _server ] = new St.Label({ text: String(_status['CRITICAL']) })
            critical_box.add_child ( this.criticalBoxes [ _server ] );
            serverBox.add_child(critical_box);

        }

        box.add_child(PopupMenu.arrowIcon(St.Side.BOTTOM));

	// Menu
        this._buttonMenu = new PopupMenu.PopupBaseMenuItem({
            reactive: false,
            style_class: 'monito-menu-button-container',
        });
        this.menu.addMenuItem(this._buttonMenu);

//        let item = new PopupMenu.PopupMenuItem(_('Reload'));
//        item.connect('activate', () => {
//	    this.updateStatus ( );
//        });
//        this.menu.addMenuItem(item);

        let _path = Me.path + '/img/monito.png';
        let _icon = new St.Icon({
            gicon: Gio.icon_new_for_string(_path),
        });
        let _iconBin = new St.Bin();
        _iconBin.child = _icon;
        this._buttonMenu.actor.add_actor(_iconBin);

	this._mainLabel = new St.Label({ style_class: 'monito-title', text: 'Monito Checker', x_expand: true });
        this._buttonMenu.actor.add_actor(this._mainLabel);

	this._prefsButton = this._createButton ( 'preferences-system-symbolic', 'Preferences', this._onPreferencesActivate );
        this._buttonMenu.actor.add_child (this._prefsButton);

	this._reloadButton = this._createButton ( 'view-refresh-symbolic', 'Reload', this.updateStatus );
        this._buttonMenu.actor.add_child (this._reloadButton );

	let _intermediate = new PopupMenu.PopupBaseMenuItem ( {
	    style_class: 'monito-services',
            reactive: false
        });

        this.menu.addMenuItem(_intermediate);
        this._box = new St.BoxLayout({
	    style_class: 'monito-services',
	    vertical: true,
	    x_expand: true
	});
        _intermediate.actor.add_actor(this._box);

	this.updateStatus ( );
        this.setupTimeout ( );
    }

    setupTimeout ( )
    {
        monitoLog ( 'Setting up timeout of ' + settings.get_int ( "poll-delay" ) + ' secs' );
        if (this.timeout) {
            Mainloop.source_remove(this.timeout);
        }
        this.timeout = Mainloop.timeout_add ( settings.get_int ( "poll-delay" ) * 1000, Lang.bind(this, this.updateStatus ) );
    }

    initStatus ( ) {
	_status = { 'OK': 0,
		    'WARNING': 0,
		    'CRITICAL': 0,
		    'UNKNOWN': 0 };
    }

    updateStatus ( )
    {
        for ( let _server of Preferences.getServersList() )
        {
	    let _account_settings = Preferences.getAccountSettings ( _server );
            
	    let type = _account_settings.get_string("type");
	    let username = _account_settings.get_string("username");
	    let password = _account_settings.get_string("password");
	    let urlcgi = _account_settings.get_string("urlcgi");

	    if ( ! urlcgi )
	    {
	        monitoLog ( 'Not updating monito because no URL configured' );
	    }
            else
            {
                let _serverLogic;
                if ( type == 'Icinga' )
                    _serverLogic = new Icinga ( _server );
                else if ( type == 'Icinga2' )
                    _serverLogic = new Icinga2 ( _server );
                
                if ( ! _serverLogic.refresh ( this ) )
                {
	            this.warningBoxes[_server].set_text ( '…' );
	            this.criticalBoxes[_server].set_text ( '…' );
                    // TODO: Add display of error if any
                }
            }
        }

        this.setupTimeout ( );
    }

    createHeaderBin ( status, text, col ) {
	let _widths = [ 300, 300, 200, 50, 600 ];
	let _bin = new St.Bin({
	    style_class: 'monito-service-' + status,
	    width: _widths[col],
	    x_expand: ( col == 4 ? true : false ),
	    child: new St.Button ( {
	        x_align: Clutter.ActorAlign.FILL,
                y_align: Clutter.ActorAlign.CENTER,
                reactive: true,
                can_focus: true,
                track_hover: true,
                accessible_name: text,
                style_class: 'button',
                label: 'Foo',
            })
	});
	return _bin;
    }

    createBin ( status, text, col ) {
	let _widths = [ 300, 300, 200, 50, 600 ];
	let _bin = new St.Bin({
	    style_class: 'monito-service-' + status,
            track_hover: true,
	    width: _widths[col],
	    x_expand: ( col == 4 ? true : false ),
	    child: new St.Label({ style_class: 'monito-label', text: text })
	});
	return _bin;
    }

    refreshUI ( serverLogic ) {
	this.initStatus ( );    // Specialize !

	this._box.remove_all_children();
	
	let headerBox = new St.BoxLayout({
	    style_class: 'monito-service-line',
	    hover: true,
	    x_expand: true
	});
	this._box.add_child(headerBox);

	headerBox.add_child ( this.createHeaderBin ( '', 'Foo', 0 ) );
	headerBox.add_child ( this.createHeaderBin ( '', 'Foo', 1 ) );
	headerBox.add_child ( this.createHeaderBin ( '', 'Foo', 2 ) );
	headerBox.add_child ( this.createHeaderBin ( '', 'Foo', 3 ) );
	headerBox.add_child ( this.createHeaderBin ( '', 'Foo', 4 ) );

        for ( let i = 0 ; i < serverLogic.status.service_status.length ; i ++ )
	{
	    _status [ serverLogic.status.service_status[i].status ] ++;
	    if ( serverLogic.status.service_status[i].status != 'OK' )
	    {
		let infoBox = new St.BoxLayout({
		    style_class: 'monito-service-line monito-service-line-' + serverLogic.status.service_status[i].status,
		    hover: true,
		    x_expand: true
		});
		this._box.add_child(infoBox);

		infoBox.add_child ( this.createBin ( serverLogic.status.service_status[i].status,
						     serverLogic.status.service_status[i].host_name,
						     0 ) );
		infoBox.add_child ( this.createBin ( serverLogic.status.service_status[i].status,
						     serverLogic.status.service_status[i].service_display_name,
						     1 ) );
		infoBox.add_child ( this.createBin ( serverLogic.status.service_status[i].status,
						     serverLogic.status.service_status[i].last_check,
						     2) );
		infoBox.add_child ( this.createBin ( serverLogic.status.service_status[i].status,
						     serverLogic.status.service_status[i].attempts,
						     3 ) );
		infoBox.add_child ( this.createBin ( serverLogic.status.service_status[i].status,
						     serverLogic.status.service_status[i].status_information,
						     4 ) );
	    }
	}
        
	this.warningBoxes[serverLogic.server].set_text ( String(_status.WARNING) );
	this.criticalBoxes[serverLogic.server].set_text ( String(_status.CRITICAL) );
        
        return;
    }

    _onPreferencesActivate ( ) {
        this.menu.actor.hide();
        if (typeof ExtensionUtils.openPrefs === 'function') {
            ExtensionUtils.openPrefs();
        } else {
            Util.spawn([
                "gnome-shell-extension-prefs",
                Me.uuid
            ]);
        }
        return 0;
    }

    _createButton ( icon, text, callback ) {
        let button = new St.Button({
	    x_align: Clutter.ActorAlign.END,
            y_align: Clutter.ActorAlign.CENTER,
            reactive: true,
            can_focus: true,
            track_hover: true,
            accessible_name: text,
            style_class: 'button',
        });

        button.child = new St.Icon({
	    style_class: 'monito-button-icon',
            icon_name: icon,
	    icon_size: 24,
	    width: 24,
	    height: 24,
        });

        button.connect('clicked', Lang.bind(this, callback ) );

	return button;
    }

});

class Extension {
    constructor(uuid) {
        this._uuid = uuid;

        ExtensionUtils.initTranslations(GETTEXT_DOMAIN);
    }

    enable() {
        this._indicator = new Indicator();

        Main.panel.addToStatusArea(this._uuid, this._indicator);
    }

    disable() {
        this._indicator.destroy();
        this._indicator = null;

        if (_httpSession !== undefined)
            _httpSession.abort();

        _httpSession = undefined;
    }
}

function init(meta) {
    return new Extension(meta.uuid);
}


function monitoLog ( msg )
{
    log ( 'Monito: ' + msg );
}


function setColor (stgs, key) {
    monitoLog ( stgs.get_string(key) );
    this.widget.set_style ( 'background-color: ' + stgs.get_string(key) );
}