345 lines
11 KiB
JavaScript
345 lines
11 KiB
JavaScript
/* -*- 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 Gettext = imports.gettext.domain(GETTEXT_DOMAIN);
|
|
const _ = Gettext.gettext;
|
|
const Lang = imports.lang;
|
|
|
|
const ExtensionUtils = imports.misc.extensionUtils;
|
|
const Me = ExtensionUtils.getCurrentExtension();
|
|
const Main = imports.ui.main;
|
|
const PanelMenu = imports.ui.panelMenu;
|
|
const PopupMenu = imports.ui.popupMenu;
|
|
const Mainloop = imports.mainloop;
|
|
|
|
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 Preferences = Me.imports.prefs;
|
|
const Icinga = Me.imports.servers.icinga.Icinga;
|
|
const Icinga2 = Me.imports.servers.icinga2.Icinga2;
|
|
|
|
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 ( );
|
|
}
|
|
|
|
createBin ( 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.Label({ style_class: 'monito-label', text: text })
|
|
});
|
|
return _bin;
|
|
}
|
|
|
|
refreshUI ( serverLogic ) {
|
|
this.initStatus ( ); // Specialize !
|
|
|
|
this._box.remove_all_children();
|
|
|
|
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) );
|
|
}
|