Modify identity selection logic / fix bugs
[moonshot-ui.git] / src / moonshot-identity-manager-app.vala
index 3989807..a9a4c33 100644 (file)
@@ -1,9 +1,22 @@
+using Gee;
 using Gtk;
 
+#if IPC_DBUS
+[DBus (name = "org.janet.Moonshot")]
+interface IIdentityManager : GLib.Object {
+#if IPC_DBUS_GLIB
+    public abstract bool show_ui() throws DBus.Error;
+#else
+    public abstract bool show_ui() throws IOError;
+#endif
+}
+#endif
 
-class IdentityManagerApp {
+public class IdentityManagerApp {
     public IdentityManagerModel model;
-    private IdentityManagerView view;
+    public IdCard default_id_card;
+    public bool explicitly_launched;
+    public IdentityManagerView view;
     private MoonshotServer ipc_server;
 
 #if OS_MACOS
@@ -23,12 +36,34 @@ class IdentityManagerApp {
     private const int WINDOW_WIDTH = 400;
     private const int WINDOW_HEIGHT = 500;
     public void show() {
-        view.show();    
+        if (view != null) view.show();    
     }
        
-    public IdentityManagerApp () {
-        model = new IdentityManagerModel(this);
-        view = new IdentityManagerView(this);
+    public IdentityManagerApp (bool headless, bool use_flat_file_store) {
+        use_flat_file_store |= UserForcesFlatFileStore();
+#if GNOME_KEYRING
+        bool keyring_available = (!use_flat_file_store) && GnomeKeyring.is_available();
+#else
+        bool keyring_available = false;
+#endif
+        IIdentityCardStore.StoreType store_type;
+        if (headless || use_flat_file_store || !keyring_available)
+            store_type = IIdentityCardStore.StoreType.FLAT_FILE;
+        else
+            store_type = IIdentityCardStore.StoreType.KEYRING;
+
+        model = new IdentityManagerModel(this, store_type);
+        /* if headless, but we have nothing in the flat file store
+         * and keyring is available, switch to keyring */
+        if (headless && keyring_available && !use_flat_file_store && !model.HasNonTrivialIdentities())
+            model.set_store_type(IIdentityCardStore.StoreType.KEYRING);
+
+        if (!headless)
+            view = new IdentityManagerView(this);
+        LinkedList<IdCard> card_list = model.get_card_list() ;
+        if (card_list.size > 0)
+            this.default_id_card = card_list.first();
+
         init_ipc_server ();
 
 #if OS_MACOS
@@ -40,7 +75,157 @@ class IdentityManagerApp {
         Signal.connect(osxApp, "NSApplicationOpenFile", (GLib.Callback)(on_osx_open_files), this);
 
 #endif
-        view.show();
+    }
+
+    public bool add_identity (IdCard id, bool force_flat_file_store) {
+        if (view != null) return view.add_identity(id, force_flat_file_store);
+        model.add_card(id, force_flat_file_store);
+        return true;
+    }
+
+    public void select_identity (IdentityRequest request) {
+        IdCard identity = null;
+
+        if (request.select_default)
+        {
+            identity = default_id_card;
+        }
+
+        if (identity == null)
+        {
+            bool has_nai = request.nai != null && request.nai != "";
+            bool has_srv = request.service != null && request.service != "";
+            bool confirm = false;
+
+            foreach (IdCard id in model.get_card_list())
+            {
+                /* If NAI matches, use this id card */
+                if (has_nai && request.nai == id.nai)
+                {
+                    identity = id;
+                    break;
+                }
+
+                /* If any service matches we add id card to the candidate list */
+                if (has_srv)
+                {
+                    foreach (string srv in id.services)
+                    {
+                        if (request.service == srv)
+                        {
+                            request.candidates.append (id);
+                            continue;
+                        }
+                    }
+                }
+            }
+
+            /* If more than one candidate we dissasociate service from all ids */
+            if ((identity == null) && has_srv && request.candidates.length() > 1)
+            {
+                foreach (IdCard id in request.candidates)
+                {
+                    int i = 0;
+                    SList<string> services_list = null;
+                    bool has_service = false;
+
+                    foreach (string srv in id.services)
+                    {
+                        if (srv == request.service)
+                        {
+                            has_service = true;
+                            continue;
+                        }
+                        services_list.append (srv);
+                    }
+                    
+                    if (!has_service)
+                        continue;
+
+                    if (services_list.length () == 0)
+                    {
+                        id.services = {};
+                        continue;
+                    }
+
+                    string[] services = new string[services_list.length ()];
+                    foreach (string srv in services_list)
+                    {
+                        services[i] = srv;
+                        i++;
+                    }
+
+                    id.services = services;
+                }
+            }
+
+            /* If there are no candidates we use the service matching rules */
+            if ((identity==null) && (request.candidates.length () == 0))
+            {
+                foreach (IdCard id in model.get_card_list())
+                {
+                    foreach (Rule rule in id.rules)
+                    {
+                        if (!match_service_pattern (request.service, rule.pattern))
+                            continue;
+
+                        request.candidates.append (id);
+
+                        if (rule.always_confirm == "true")
+                            confirm = true;
+                    }
+                }
+            }
+            
+            if ((identity == null) && has_nai) {
+                // create a temp identity
+                string[] components = request.nai.split("@", 2);
+                identity = new IdCard();
+                identity.display_name = request.nai;
+                identity.username = components[0];
+                if (components.length > 1)
+                    identity.issuer = components[1];
+                identity.password = request.password;
+                identity.temporary = true;
+            }
+            if (identity == null) {
+                if (request.candidates.length () != 1) {
+                    confirm = true;
+                } else {
+                    identity = request.candidates.nth_data (0);                    
+                }
+            }
+
+            if (confirm && (view != null))
+            {
+                if (!explicitly_launched)
+                    show();
+               view.queue_identity_request(request);
+                return;
+            }
+        }
+        // Send back the identity (we can't directly run the
+        // callback because we may be being called from a 'yield')
+        Idle.add(
+            () => {
+                if (view != null) {
+                    identity = view.check_add_password(identity, request, model);
+                }
+                request.return_identity (identity);
+// The following occasionally causes the app to exit without sending the dbus
+// reply, so for now we just don't exit
+//                if (!explicitly_launched)
+//                    Idle.add( () => { Gtk.main_quit(); return false; } );
+                return false;
+            }
+        );
+        return;
+    }
+
+    private bool match_service_pattern (string service, string pattern)
+    {
+        var pspec = new PatternSpec (pattern);
+        return pspec.match_string (service);
     }   
     
 #if IPC_MSRPC
@@ -50,12 +235,11 @@ class IdentityManagerApp {
         // obtrusive message box, on Windows
         //
         this.ipc_server = MoonshotServer.get_instance ();
-        MoonshotServer.start (this.view);
+        MoonshotServer.start (this);
     }
 #elif IPC_DBUS_GLIB
     private void init_ipc_server ()
     {
         try {
             var conn = DBus.Bus.get (DBus.BusType.SESSION);
             dynamic DBus.Object bus = conn.get_object ("org.freedesktop.DBus",
@@ -64,10 +248,26 @@ class IdentityManagerApp {
 
             // try to register service in session bus
             uint reply = bus.request_name ("org.janet.Moonshot", (uint) 0);
-            assert (reply == DBus.RequestNameReply.PRIMARY_OWNER);
+            if (reply == DBus.RequestNameReply.PRIMARY_OWNER)
+            {
+                this.ipc_server = new MoonshotServer (this);
+                conn.register_object ("/org/janet/moonshot", ipc_server);
+            } else {
+                bool shown=false;
+                GLib.Error e;
+                DBus.Object manager_proxy = conn.get_object ("org.janet.Moonshot",
+                                                             "/org/janet/moonshot",
+                                                             "org.janet.Moonshot");
+                if (manager_proxy != null)
+                    manager_proxy.call("ShowUi", out e, GLib.Type.INVALID, typeof(bool), out shown, GLib.Type.INVALID);
 
-            this.ipc_server = new MoonshotServer (this.view);
-            conn.register_object ("/org/janet/moonshot", ipc_server);
+                if (!shown) {
+                    GLib.error ("Couldn't own name org.janet.Moonshot on dbus or show previously launched identity manager.");
+                } else {
+                    stdout.printf("Showed previously launched identity manager.\n");
+                    GLib.Process.exit(0);
+                }
+            }
         }
         catch (DBus.Error e)
         {
@@ -88,22 +288,73 @@ class IdentityManagerApp {
 
     private void init_ipc_server ()
     {
-        this.ipc_server = new MoonshotServer (this.view);
+        this.ipc_server = new MoonshotServer (this);
         GLib.Bus.own_name (GLib.BusType.SESSION,
                            "org.janet.Moonshot",
                            GLib.BusNameOwnerFlags.NONE,
                            bus_acquired_cb,
                            (conn, name) => {},
                            (conn, name) => {
-                               error ("Couldn't own name %s on DBus.", name);
+                               bool shown=false;
+                               try {
+                                   IIdentityManager manager = Bus.get_proxy_sync (BusType.SESSION, name, "/org/janet/moonshot");
+                                   shown = manager.show_ui();
+                               } catch (IOError e) {
+                               }
+                               if (!shown) {
+                                   GLib.error ("Couldn't own name %s on dbus or show previously launched identity manager.", name);
+                               } else {
+                                   stdout.printf("Showed previously launched identity manager.\n");
+                                   GLib.Process.exit(0);
+                               }
                            });
     }
 #endif
 }
 
+static bool explicitly_launched = true;
+static bool use_flat_file_store = false;
+const GLib.OptionEntry[] options = {
+    {"dbus-launched",0,GLib.OptionFlags.REVERSE,GLib.OptionArg.NONE,
+     ref explicitly_launched,"launch for dbus rpc use",null},
+    {"flat-file-store",0,0,GLib.OptionArg.NONE,
+     ref use_flat_file_store,"force use of flat file identity store (used by default only for headless operation)",null},
+    {null}
+};
+
 
 public static int main(string[] args){
-        Gtk.init(ref args);
+#if IPC_MSRPC
+       bool headless = false;
+#else
+        bool headless = GLib.Environment.get_variable("DISPLAY") == null;
+#endif
+
+        if (headless) {
+            try {
+                var opt_context = new OptionContext(null);
+                opt_context.set_help_enabled (true);
+                opt_context.add_main_entries (options, null);
+                opt_context.parse(ref args);
+            } catch (OptionError e) {
+                stdout.printf(_("error: %s\n"),e.message);
+                stdout.printf(_("Run '%s --help' to see a full list of available options\n"), args[0]);
+                return -1;
+            }
+            explicitly_launched = false;
+        } else {
+            try {
+                if (!Gtk.init_with_args(ref args, _(""), options, null)) {
+                    stdout.printf(_("unable to initialize window\n"));
+                    return -1;
+                }
+            } catch (GLib.Error e) {
+                stdout.printf(_("error: %s\n"),e.message);
+                stdout.printf(_("Run '%s --help' to see a full list of available options\n"), args[0]);
+                return -1;
+            }
+            gtk_available = true;
+        }
 
 #if OS_WIN32
         // Force specific theme settings on Windows without requiring a gtkrc file
@@ -117,11 +368,21 @@ public static int main(string[] args){
         Intl.textdomain (Config.GETTEXT_PACKAGE);
        
           
-        var app = new IdentityManagerApp();
+        var app = new IdentityManagerApp(headless, use_flat_file_store);
+        app.explicitly_launched = explicitly_launched;
         
-        app.show();
+       if (app.explicitly_launched) {
+            app.show();
+        }
 
-        Gtk.main();
+        if (headless) {
+#if !IPC_MSRPC
+            MainLoop loop = new MainLoop();
+            loop.run();
+#endif
+        } else {
+            Gtk.main();
+        }
 
         return 0;
     }