/* -*- 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 { GObject, St, Clutter, Gio } = imports.gi; const SETTINGS_SCHEMA = "org.gnome.shell.extensions.monito@drieu.org"; 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); const Indicator = GObject.registerClass( class Indicator extends PanelMenu.Button { _init() { super._init(0.0, _('Monito Checker')); let box = new St.BoxLayout ( { } ); this.add_child(box); this.initStatus ( ); this.namesBoxes = { }; this.warningBoxes = { }; this.criticalBoxes = { }; this.unknownBoxes = { }; for ( let _server of Preferences.getServersList() ) { log ( '> Server ' + _server ); let _account_settings = Preferences.getAccountSettings ( _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); let warning_box = new St.BoxLayout({ style_class: 'monito-warning-box monito-box' }); 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' }); 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 ( ); } 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 ) { log ( '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 ( '…' ); } } } } 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); }