[gnome-packagekit] Show the correct actions for the new security restart states

[Richard Hughes <rhughes src gnome org>]

commit 4c0d9869bc7b984eeeae1461a0817e158cea7b5c
Author: Richard Hughes <richard hughsie com>
Date:   Sun Jul 26 20:17:50 2009 +0100

    Show the correct actions for the new security restart states

 src/gpk-watch.c |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)
---
diff --git a/src/gpk-watch.c b/src/gpk-watch.c
index 5284c77..0a0e690 100644
--- a/src/gpk-watch.c
+++ b/src/gpk-watch.c
@@ -1280,7 +1280,8 @@ gpk_watch_activate_status_cb (GtkStatusIcon *status_icon, GpkWatch *watch)
 	}
 
 	/* log out session */
-	if (watch->priv->restart == PK_RESTART_ENUM_SESSION) {
+	if (watch->priv->restart == PK_RESTART_ENUM_SESSION ||
+	    watch->priv->restart == PK_RESTART_ENUM_SECURITY_SESSION) {
 		/* TRANSLATORS: log out of the session */
 		widget = gtk_image_menu_item_new_with_mnemonic (_("_Log out"));
 		image = gtk_image_new_from_icon_name ("system-log-out", GTK_ICON_SIZE_MENU);
@@ -1292,7 +1293,8 @@ gpk_watch_activate_status_cb (GtkStatusIcon *status_icon, GpkWatch *watch)
 	}
 
 	/* restart computer */
-	if (watch->priv->restart == PK_RESTART_ENUM_SYSTEM) {
+	if (watch->priv->restart == PK_RESTART_ENUM_SYSTEM ||
+	    watch->priv->restart == PK_RESTART_ENUM_SECURITY_SYSTEM) {
 		/* TRANSLATORS: this menu item restarts the computer after an update */
 		widget = gtk_image_menu_item_new_with_mnemonic (_("_Restart computer"));
 		image = gtk_image_new_from_icon_name ("system-shutdown", GTK_ICON_SIZE_MENU);