Sync with U. Desktop

2 files changed, 41 insertions, 11 deletions

diff --git a/src/apt-watcher.c b/src/apt-watcher.c
index 6e8a5c9..092942f 100644
--- a/src/apt-watcher.c
+++ b/src/apt-watcher.c
@@ -52,6 +52,7 @@ get_updates_complete (GObject *source_object,
   AptWatcher* self = APT_WATCHER (user_data);
 
   PkResults *results;
+  PkRestartEnum restart_required;
   GError *error = NULL;
   results = pk_client_generic_finish (PK_CLIENT(source_object), res, &error);
   
@@ -83,6 +84,20 @@ get_updates_complete (GObject *source_object,
                                         DBUSMENU_MENUITEM_PROP_LABEL,
                                         _("Software Up to Date"));       
   }
+
+  /* check if there was a restart required info in the signal */
+  restart_required = pk_results_get_require_restart_worst (results);
+  if (restart_required == PK_RESTART_ENUM_SYSTEM ||
+      restart_required == PK_RESTART_ENUM_SECURITY_SYSTEM) {
+     dbusmenu_menuitem_property_set (self->apt_item,
+                                     DBUSMENU_MENUITEM_PROP_LABEL,
+                                     _("Restart to Complete Updates…"));
+     dbusmenu_menuitem_property_set (self->apt_item,
+                                     DBUSMENU_MENUITEM_PROP_DISPOSITION,
+                                     DBUSMENU_MENUITEM_DISPOSITION_ALERT); 
+     session_dbus_restart_required (self->session_dbus_interface);
+  }
+
   g_ptr_array_unref (packages);
   g_object_unref (results);
   g_object_unref (source_object);
@@ -115,15 +130,6 @@ static void apt_watcher_signal_cb ( GDBusProxy* proxy,
     g_debug ("updates changed signal received");
     apt_watcher_check_for_updates (self);
   }
-  else if (g_strcmp0(signal_name, "RestartScheduled") == 0) {
-    g_debug ("RestartScheduled signal received");
-    dbusmenu_menuitem_property_set (self->apt_item,
-                                    DBUSMENU_MENUITEM_PROP_LABEL,
-                                    _("Restart to Complete Updates…"));
-    dbusmenu_menuitem_property_set (self->apt_item,
-                                    DBUSMENU_MENUITEM_PROP_DISPOSITION,
-                                    DBUSMENU_MENUITEM_DISPOSITION_ALERT);     
-  } 
 }
 
 static void
diff --git a/src/user-menu-mgr.c b/src/user-menu-mgr.c
index 7aad2af..3f4bdc4 100644
--- a/src/user-menu-mgr.c
+++ b/src/user-menu-mgr.c
@@ -285,11 +285,29 @@ check_new_session ()
 	return TRUE;
 }
 
+/* Check to see if the lockdown key is protecting from
+   locking the screen.  If not, lock it. */
+static void
+lock_if_possible (void) {
+	ensure_settings_client ();
+
+	if (!g_settings_get_boolean (settings, LOCKDOWN_KEY_SCREENSAVER)) {
+		lock_screen(NULL, 0, NULL);
+	}
+
+	return;
+}
+
+
 /* Starts a new generic session */
 static void
 activate_new_session (DbusmenuMenuitem * mi, guint timestamp, gpointer user_data)
 {
-	users_service_dbus_show_greeter (USERS_SERVICE_DBUS(user_data));
+	lock_if_possible();
+
+  users_service_dbus_show_greeter (USERS_SERVICE_DBUS(user_data));
+
+	return;
 }
 
 /* Activates a session for a particular user. */
@@ -299,6 +317,8 @@ activate_user_session (DbusmenuMenuitem *mi, guint timestamp, gpointer user_data
   UserData *user = (UserData *)user_data;
   UsersServiceDbus *service = user->service;
 
+  lock_if_possible();
+
   users_service_dbus_activate_user_session (service, user);
 }
 
@@ -386,7 +406,11 @@ activate_guest_session (DbusmenuMenuitem * mi, guint timestamp, gpointer user_da
   UserMenuMgr* user_mgr = USER_MENU_MGR(user_data);  
   UsersServiceDbus *service = user_mgr->users_dbus_interface;
 
-  users_service_dbus_activate_guest_session(service);
+	lock_if_possible();
+  
+  if (users_service_dbus_activate_guest_session(service)) {
+    return;
+  }
 }