subgraph-gnome-shell-extension-tor/ui/tor_button.js

// ex: set sw=4:ts=4
/*
Copyright 2015 Frank Ploss <frank@fqxp.de>.

This file is part of gnome-shell-extension-tor.

gnome-shell-extension-tor 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 3 of the License, or
(at your option) any later version.

gnome-shell-extension-tor 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 gnome-shell-extension-tor.  If not, see <http://www.gnu.org/licenses/>./
*/
'use strict';

const Lang = imports.lang;
const Main = imports.ui.main;
const PanelMenu = imports.ui.panelMenu;
const PopupMenu = imports.ui.popupMenu;
const St = imports.gi.St;

const Me = imports.misc.extensionUtils.getCurrentExtension();
const TorDisconnectedMenu = Me.imports.ui.tor_disconnected_menu.TorDisconnectedMenu;
const TorPopupMenu = Me.imports.ui.tor_popup_menu.TorPopupMenu;

const TorConnectedIcon = 'tor-connected';
const TorDisconnectedIcon = 'tor-disconnected';

const TorButton = new Lang.Class({
    Name: 'TorButton',
    Extends: PanelMenu.Button,

    _init: function(torControlClient) {
        this.parent(null, this.Name);

        this._torControlClient = torControlClient;

        this._buildUi();
        this._bindEvents();

        this._currentState = null;
    },

    _buildUi: function() {
        this._icon = new St.Icon({
            icon_name: TorDisconnectedIcon,
            style_class: 'system-status-icon'
        });

        this.actor.add_child(this._icon);

        //var dummyMenu = new PopupMenu.PopupDummyMenu(this.actor);
        //this.setMenu(dummyMenu);
        //Main.panel.menuManager.addMenu(dummyMenu);
    },

    _bindEvents: function() {
        this._torControlClient.connect('changed-connection-state', Lang.bind(this, this._onChangedConnectionState));
        this._torControlClient.connect('switched-tor-identity', Lang.bind(this, this._onSwitchedTorIdentity));
        this._torControlClient.connect('protocol-error', Lang.bind(this, this._onProtocolError));
    },

    _onChangedConnectionState: function(source, state, message) {
        if (this._currentState == state)
            return;

        this._currentState = state;

        switch (state) {
            case 'connected':
                this._icon.icon_name = TorConnectedIcon;
                this._menu = new TorPopupMenu(this.actor, this._torControlClient);
                this.setMenu(this._menu);
                Main.panel.menuManager.addMenu(this._menu);
                break;
            case 'disconnected':
                this._icon.icon_name = TorDisconnectedIcon;
                this._menu = new TorDisconnectedMenu(this.actor, this._torControlClient);
                this.setMenu(this._menu);
                Main.panel.menuManager.addMenu(this._menu);
                break;
        }
    },

    _onSwitchedTorIdentity: function() {
        Main.notify('Switched to a new Tor identity!');
    },

    _onProtocolError: function(source, message, statusCode) {
        Main.notifyError(message);
        log('Tor control procotol error (status code ' + statusCode + '): ' + reason)
    }
});