1 files changed, 158 insertions, 0 deletions
diff --git a/src/session-list.vala b/src/session-list.vala
new file mode 100644
index 0000000..7127023
--- /dev/null
+++ b/src/session-list.vala
@@ -0,0 +1,158 @@
+/* -*- Mode: Vala; indent-tabs-mode: nil; tab-width: 4 -*-
+ *
+ * Copyright (C) 2012 Canonical Ltd
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 3 as
+ * published by the Free Software Foundation.
+ *
+ * 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, see <http://www.gnu.org/licenses/>.
+ *
+ * Authors: Michael Terry <michael.terry@canonical.com>
+ */
+
+public class SessionPrompt : PromptBox
+{
+    public string session { get; construct; }
+    public string default_session { get; construct; }
+
+    public SessionPrompt (string id, string? session, string? default_session)
+    {
+        Object (id: id, session: session, default_session: default_session);
+    }
+
+    private ToggleBox box;
+
+    construct
+    {
+        label = _("Select desktop environment");
+        name_label.vexpand = false;
+
+        box = new ToggleBox (default_session, session);
+
+        if (UnityGreeter.singleton.test_mode)
+        {
+            box.add_item ("gnome", "GNOME", SessionList.get_badge ("gnome"));
+            box.add_item ("kde", "KDE", SessionList.get_badge ("kde"));
+            box.add_item ("ubuntu", "Ubuntu", SessionList.get_badge ("ubuntu"));
+        }
+        else
+        {
+            foreach (var session in LightDM.get_sessions ())
+            {
+                debug ("Adding session %s (%s)", session.key, session.name);
+                box.add_item (session.key, session.name, SessionList.get_badge (session.key));
+            }
+        }
+
+        box.notify["selected-key"].connect (selected_cb);
+        box.show ();
+
+        attach_item (box);
+    }
+
+    private void selected_cb ()
+    {
+        respond ({ box.selected_key });
+    }
+}
+
+public class SessionList : GreeterList
+{
+    public signal void session_clicked (string session);
+    public string session { get; construct; }
+    public string default_session { get; construct; }
+
+    private SessionPrompt prompt;
+
+    public SessionList (Background bg, MenuBar mb, string? session, string? default_session)
+    {
+        Object (background: bg, menubar: mb, session: session, default_session: default_session);
+    }
+
+    construct
+    {
+        prompt = add_session_prompt ("session");
+    }
+
+    private SessionPrompt add_session_prompt (string id)
+    {
+        var e = new SessionPrompt (id, session, default_session);
+        e.respond.connect ((responses) => { session_clicked (responses[0]); });
+        add_entry (e);
+        return e;
+    }
+
+    protected override void add_manual_entry () {}
+    public override void show_authenticated (bool successful = true) {}
+
+    private static string? get_badge_name (string session)
+    {
+        switch (session)
+        {
+        case "ubuntu":
+        case "ubuntu-2d":
+            return "ubuntu_badge.png";
+        case "gnome-classic":
+        case "gnome-flashback-compiz":
+        case "gnome-flashback":
+        case "gnome-fallback-compiz":
+        case "gnome-fallback":
+        case "gnome-shell":
+        case "gnome":
+            return "gnome_badge.png";
+        case "kde":
+        case "kde-plasma":
+            return "kde_badge.png";
+        case "xterm":
+            return "recovery_console_badge.png";
+        case "remote-login":
+            return "remote_login_help.png";
+        default:
+            return null;
+        }
+    }
+
+    private static HashTable<string, Gdk.Pixbuf> badges; /* cache of badges */
+    public static Gdk.Pixbuf? get_badge (string session)
+    {
+        var name = get_badge_name (session);
+
+        if (name == null)
+        {
+            /* Not a known name, but let's see if we have a custom badge before
+               giving up entirely and using the unknown badget. */
+            var maybe_name = "custom_%s_badge.png".printf (session);
+            var maybe_path = Path.build_filename (Config.PKGDATADIR, maybe_name, null);
+            if (FileUtils.test (maybe_path, FileTest.EXISTS))
+                name = maybe_name;
+            else
+                name = "unknown_badge.png";
+        }
+
+        if (badges == null)
+            badges = new HashTable<string, Gdk.Pixbuf> (str_hash, str_equal);
+
+        var pixbuf = badges.lookup (name);
+        if (pixbuf == null)
+        {
+            try
+            {
+                pixbuf = new Gdk.Pixbuf.from_file (Path.build_filename (Config.PKGDATADIR, name, null));
+                badges.insert (name, pixbuf);
+            }
+            catch (Error e)
+            {
+                debug ("Error loading badge %s: %s", name, e.message);
+            }
+        }
+
+        return pixbuf;
+    }
+}