[gnome-shell-extensions/extension-live-disable] alternative-status-menu: fix for live disable

[Giovanni Campagna <gcampagna src gnome org>]

commit d7e13b4abd88ca91cc7eb397728a38e605f72826
Author: Giovanni Campagna <gcampagna src gnome org>
Date:   Thu Aug 18 19:35:47 2011 +0200

    alternative-status-menu: fix for live disable
    
    After recreating the menu, we need to update the visibility of
    items affected by lockdown, and we need to update the presence
    indicator.

 extensions/alternative-status-menu/extension.js |   14 +++++++++++++-
 1 files changed, 13 insertions(+), 1 deletions(-)
---
diff --git a/extensions/alternative-status-menu/extension.js b/extensions/alternative-status-menu/extension.js
index d65c9d3..6ed0aa2 100644
--- a/extensions/alternative-status-menu/extension.js
+++ b/extensions/alternative-status-menu/extension.js
@@ -63,6 +63,7 @@ function createSubMenu() {
     item = new PopupMenu.PopupMenuItem(_("Lock Screen"));
     item.connect('activate', Lang.bind(this, this._onLockScreenActivate));
     this.menu.addMenuItem(item);
+    this._lockScreenItem = item;
 
     item = new PopupMenu.PopupMenuItem(_("Switch User"));
     item.connect('activate', Lang.bind(this, this._onLoginScreenActivate));
@@ -72,9 +73,11 @@ function createSubMenu() {
     item = new PopupMenu.PopupMenuItem(_("Log Out..."));
     item.connect('activate', Lang.bind(this, this._onQuitSessionActivate));
     this.menu.addMenuItem(item);
+    this._logoutItem = item;
 
     item = new PopupMenu.PopupSeparatorMenuItem();
     this.menu.addMenuItem(item);
+    this._sessionSeparator = item;
 
     item = new PopupMenu.PopupMenuItem(_("Suspend"));
     item.connect('activate', Lang.bind(this, onSuspendActivate));
@@ -96,14 +99,22 @@ function createSubMenu() {
 }
 
 // Put your extension initialization code here
-function main(metadata) {
+function init(metadata) {
     imports.gettext.bindtextdomain('gnome-shell-extensions', metadata.localedir);
 }
 
+function reset(statusMenu) {
+    statusMenu._updateSwitchUser();
+    statusMenu._updateLogout();
+    statusMenu._updateLockScreen();
+    statusMenu._presence.getStatus(Lang.bind(statusMenu, statusMenu._updatePresenceIcon));
+}
+
 function enable() {
     let statusMenu = Main.panel._userMenu;
     statusMenu.menu.removeAll();
     createSubMenu.call(statusMenu);
+    reset(statusMenu);
 }
 
 function disable() {
@@ -111,4 +122,5 @@ function disable() {
     let statusMenu = Main.panel._userMenu;
     statusMenu.menu.removeAll();
     statusMenu._createSubMenu();
+    reset(statusMenu);
 }
\ No newline at end of file