Modify identity selection logic / fix bugs
[moonshot-ui.git] / src / moonshot-identities-manager.vala
index 90bbec9..cc57f43 100644 (file)
-class IdentitiesManager : Object {
+using Gee;
+
+public class Password {
+#if GNOME_KEYRING
+    private unowned string _password;
+    public string password {
+        get {
+            return _password;
+        }
+        set {
+            if (_password != null) {
+                GnomeKeyring.memory_free((void *)_password);
+                _password = null;
+            }
+            if (value != null)
+                _password = GnomeKeyring.memory_strdup(value); 
+        }
+    }
+#else
+    public string password { get; set; default = null; }
+#endif
 
-    public SList<IdCard> id_card_list;
+    public Password(string in_password) {
+        password = in_password;
+    }
 
-    private const string FILE_NAME = "identities.txt";
+    ~Password() {
+        password = null;
+    }
+}
 
-    public IdentitiesManager ()
-    {
-        id_card_list = new SList<IdCard>();
-        var key_file = new KeyFile ();
+public class PasswordHashTable : Object {
+    private HashTable<string, Password> password_table;
 
-        var path = get_data_dir ();
-        var filename = Path.build_filename (path, FILE_NAME);
+    private static string ComputeHashKey(IdCard card, IIdentityCardStore store) {
+        return "%s_store_%d".printf( card.display_name, store.get_store_type() );
+    }
 
-        try
-        {
-            key_file.load_from_file (filename, KeyFileFlags.NONE);
-        }
-        catch (Error e)
-        {
-            stdout.printf("Error: %s\n", e.message);
-            return;
+    public void CachePassword(IdCard card, IIdentityCardStore store) {
+        password_table.replace(ComputeHashKey(card, store), new Password(card.password));
+    }
+
+    public void RemovePassword(IdCard card, IIdentityCardStore store) {
+        password_table.remove(ComputeHashKey(card, store));
+    }
+    public void RetrievePassword(IdCard card, IIdentityCardStore store) {
+        weak Password password = password_table.lookup(ComputeHashKey(card, store));
+        if (password != null) {
+            card.password = password.password;
         }
+    }
+    public PasswordHashTable() {
+        password_table = new HashTable<string, Password>(GLib.str_hash, GLib.str_equal);
+    }
+}
+
+public class IdentityManagerModel : Object {
+    private const string FILE_NAME = "identities.txt";
+    private PasswordHashTable password_table;
+    private IIdentityCardStore store;
+    public LinkedList<IdCard>  get_card_list() {
+         var identities = store.get_card_list();
+         identities.sort( (a, b) => {
+             IdCard id_a = (IdCard )a;
+             IdCard id_b = (IdCard )b;
+             if (id_a.IsNoIdentity() && !id_b.IsNoIdentity()) {
+                return -1;
+             } else if (id_b.IsNoIdentity() && !id_a.IsNoIdentity()) {
+                return 1;
+             }
+             return strcmp(id_a.display_name, id_b.display_name);
+         });
+         if (identities.is_empty || !identities[0].IsNoIdentity())
+             identities.insert(0, IdCard.NewNoIdentity());
+         foreach (IdCard id_card in identities) {
+             if (!id_card.store_password) {
+                 password_table.RetrievePassword(id_card, store);
+             }
+         }
+         return identities;
+    }
+    public signal void card_list_changed();
 
-        var identities_uris = key_file.get_groups ();
-        foreach (string identity in identities_uris)
+    /* This method finds a valid display name */
+    public bool display_name_is_valid (string name,
+                                       out string? candidate)
+    {
+        candidate = null;
+        foreach (IdCard id_card in this.get_card_list())
         {
-            try
+          if (id_card.display_name == name)
+          {
+            if (&candidate != null)
             {
-                IdCard id_card = new IdCard ();
-
-                id_card.issuer = key_file.get_string (identity, "Issuer");
-                id_card.username = key_file.get_string (identity, "Username");
-                id_card.password = key_file.get_string (identity, "Password");
-                id_card.services = key_file.get_string_list (identity, "Services");
-                id_card.display_name = key_file.get_string (identity, "DisplayName");
-                id_card.set_data ("pixbuf", find_icon ("avatar-default", 48));
-                
-                
-                if (key_file.has_key (identity, "Rules-Patterns") &&
-                    key_file.has_key (identity, "Rules-AlwaysConfirm"))
+              for (int i=0; i<1000; i++)
+              {
+                string tmp = "%s %d".printf (name, i);
+                if (display_name_is_valid (tmp, null))
                 {
-                    string [] rules_patterns =    key_file.get_string_list (identity, "Rules-Patterns");
-                    string [] rules_always_conf = key_file.get_string_list (identity, "Rules-AlwaysConfirm");
-                    
-                    if (rules_patterns.length == rules_always_conf.length)
-                    {
-                      Rule[] rules = new Rule[rules_patterns.length];
-                      for (int i = 0; i < rules_patterns.length; i++)
-                      {
-                        rules[i] = {rules_patterns[i], rules_always_conf[i]};
-                      }
-                      id_card.rules = rules;
-                    }
+                  candidate = tmp;
+                  break;
                 }
-                // Trust anchor 
-                id_card.trust_anchor.ca_cert = key_file.get_string (identity, "CA-Cert");
-                id_card.trust_anchor.subject = key_file.get_string (identity, "Subject");
-                id_card.trust_anchor.subject_alt = key_file.get_string (identity, "SubjectAlt");
-                id_card.trust_anchor.server_cert = key_file.get_string (identity, "ServerCert");
-
-                id_card_list.prepend (id_card);
-            }
-            catch (Error e)
-            {
-                stdout.printf ("Error:  %s\n", e.message);
+              }
             }
+            return false;
+          }
         }
+        return true;
     }
 
-    public void store_id_cards ()
-    {
-        var key_file = new KeyFile ();
+    public void add_card(IdCard card, bool force_flat_file_store) {
+        if (card.temporary)
+            return;
 
-        foreach (IdCard id_card in this.id_card_list)
-        {
-            string[] rules_patterns = new string[id_card.rules.length];
-            string[] rules_always_conf = new string[id_card.rules.length];
-            
-            for (int i=0; i<id_card.rules.length; i++)
-            {
-              rules_patterns[i] = id_card.rules[i].pattern;
-              rules_always_conf[i] = id_card.rules[i].always_confirm;
-            }
+        string candidate;
+        IIdentityCardStore.StoreType saved_store_type = get_store_type();
 
-            key_file.set_string (id_card.display_name, "Issuer", id_card.issuer ?? "");
-            key_file.set_string (id_card.display_name, "DisplayName", id_card.display_name ?? "");
-            key_file.set_string (id_card.display_name, "Username", id_card.username ?? "");
-            key_file.set_string (id_card.display_name, "Password", id_card.password ?? "");
-            key_file.set_string_list (id_card.display_name, "Services", id_card.services ?? {});
+        if (force_flat_file_store)
+            set_store_type(IIdentityCardStore.StoreType.FLAT_FILE);
 
-            if (id_card.rules.length > 0)
-            {
-              key_file.set_string_list (id_card.display_name, "Rules-Patterns", rules_patterns);
-              key_file.set_string_list (id_card.display_name, "Rules-AlwaysConfirm", rules_always_conf);
-            }
-            // Trust anchor 
-            key_file.set_string (id_card.display_name, "CA-Cert", id_card.trust_anchor.ca_cert ?? "");
-            key_file.set_string (id_card.display_name, "Subject", id_card.trust_anchor.subject ?? "");
-            key_file.set_string (id_card.display_name, "SubjectAlt", id_card.trust_anchor.subject_alt ?? "");
-            key_file.set_string (id_card.display_name, "ServerCert", id_card.trust_anchor.server_cert ?? "");
-        }
-
-        var text = key_file.to_data (null);
-
-        try
+        if (!display_name_is_valid (card.display_name, out candidate))
         {
-            var path = get_data_dir ();
-            var filename = Path.build_filename (path, FILE_NAME);
-            FileUtils.set_contents (filename, text, -1);
+          card.display_name = candidate;
         }
-        catch (Error e)
-        {
-            stdout.printf ("Error:  %s\n", e.message);
-        }
-    }
 
-    private string get_data_dir()
-    {
-        string path;
-
-        path = Path.build_filename (Environment.get_user_data_dir (),
-                                    Config.PACKAGE_TARNAME);
-        if (!FileUtils.test (path, FileTest.EXISTS))
-        {
-            DirUtils.create (path, 0700);
+        if (!card.store_password)
+            password_table.CachePassword(card, store);
+        store.add_card(card);
+        set_store_type(saved_store_type);
+        card_list_changed();
+     }
+
+     public IdCard update_card(IdCard card) {
+        IdCard retval;
+        if (card.temporary) {
+            retval = card;
+            return retval;
         }
-
-        return path;
+            
+        if (!card.store_password)
+            password_table.CachePassword(card, store);
+        else
+            password_table.RemovePassword(card, store);
+        retval = store.update_card(card);
+        card_list_changed();
+        return retval;
+     }
+
+     public void remove_card(IdCard card) {
+        password_table.RemovePassword(card, store);
+        store.remove_card(card);
+        card_list_changed();
+     }
+
+     public void set_store_type(IIdentityCardStore.StoreType type) {
+         if ((store != null) && (store.get_store_type() == type))
+             return;
+         switch (type) {
+#if GNOME_KEYRING
+             case IIdentityCardStore.StoreType.KEYRING:
+                 store = new KeyringStore();
+                 break;
+#endif
+             case IIdentityCardStore.StoreType.FLAT_FILE:
+             default:
+                 store = new LocalFlatFileStore();
+                 break;
+         }
+     }
+
+     public IIdentityCardStore.StoreType get_store_type() {
+         return store.get_store_type();
+     }
+
+     public bool HasNonTrivialIdentities() {
+         foreach (IdCard card in this.store.get_card_list()) {
+             // The 'NoIdentity' card is non-trivial if it has services or rules.
+             // All other cards are automatically non-trivial.
+             if ((!card.IsNoIdentity()) || 
+                 (card.services.length > 0) ||
+                 (card.rules.length > 0)) {
+                 return true;
+             }
+         }
+         return false;
+     }
+
+
+    private IdentityManagerApp parent;
+
+    public IdentityManagerModel(IdentityManagerApp parent_app, IIdentityCardStore.StoreType store_type) {
+        parent = parent_app;
+        password_table = new PasswordHashTable();
+        set_store_type(store_type);
     }
 }