src/moonshot-local-flat-file-store.vala

   1 using Gee;
   2
   3 public class LocalFlatFileStore : Object, IIdentityCardStore {
   4     private LinkedList<IdCard> id_card_list;
   5     private const string FILE_NAME = "identities.txt";
   6
   7     public void add_card(IdCard card) {
   8         id_card_list.add(card);
   9         store_id_cards ();
  10     }
  11
  12     public void update_card(IdCard card) {
  13         id_card_list.remove(card);
  14         id_card_list.add(card);
  15         store_id_cards ();
  16      }
  17
  18      public void remove_card(IdCard card) {
  19         id_card_list.remove(card);
  20         store_id_cards ();
  21     }
  22
  23      public LinkedList<IdCard> get_card_list() {
  24           return id_card_list;
  25      }
  26
  27      public IIdentityCardStore.StoreType get_store_type() {
  28           return IIdentityCardStore.StoreType.FLAT_FILE;
  29      }
  30
  31      private void load_id_cards() {
  32         id_card_list.clear();
  33         var key_file = new KeyFile ();
  34         var path = get_data_dir ();
  35         var filename = Path.build_filename (path, FILE_NAME);
  36
  37         try {
  38             key_file.load_from_file (filename, KeyFileFlags.NONE);
  39         }
  40         catch (Error e) {
  41             stdout.printf("Error: %s\n", e.message);
  42             return;
  43         }
  44
  45         var identities_uris = key_file.get_groups ();
  46         foreach (string identity in identities_uris) {
  47             try {
  48                 IdCard id_card = new IdCard ();
  49
  50                 id_card.issuer = key_file.get_string (identity, "Issuer");
  51                 id_card.username = key_file.get_string (identity, "Username");
  52                 id_card.password = key_file.get_string (identity, "Password");
  53                 id_card.services = key_file.get_string_list (identity, "Services");
  54                 id_card.display_name = key_file.get_string (identity, "DisplayName");
  55                 if (key_file.has_key (identity, "StorePassword")) {
  56                     id_card.store_password = (key_file.get_string (identity, "StorePassword") == "yes");
  57                 } else {
  58                     id_card.store_password = (id_card.password != null) && (id_card.password != "");
  59                 }
  60
  61                 if (key_file.has_key (identity, "Rules-Patterns") &&
  62                     key_file.has_key (identity, "Rules-AlwaysConfirm")) {
  63                     string [] rules_patterns =    key_file.get_string_list (identity, "Rules-Patterns");
  64                     string [] rules_always_conf = key_file.get_string_list (identity, "Rules-AlwaysConfirm");
  65
  66                     if (rules_patterns.length == rules_always_conf.length) {
  67                       Rule[] rules = new Rule[rules_patterns.length];
  68                       for (int i = 0; i < rules_patterns.length; i++) {
  69                         rules[i] = {rules_patterns[i], rules_always_conf[i]};
  70                       }
  71                       id_card.rules = rules;
  72                     }
  73                 }
  74
  75                 // Trust anchor
  76                 id_card.trust_anchor.ca_cert = key_file.get_string (identity, "CA-Cert");
  77                 id_card.trust_anchor.subject = key_file.get_string (identity, "Subject");
  78                 id_card.trust_anchor.subject_alt = key_file.get_string (identity, "SubjectAlt");
  79                 id_card.trust_anchor.server_cert = key_file.get_string (identity, "ServerCert");
  80
  81                 id_card_list.add (id_card);
  82             }
  83             catch (Error e) {
  84                 stdout.printf ("Error:  %s\n", e.message);
  85             }
  86         }
  87      }
  88
  89     private string get_data_dir() {
  90         string path;
  91         path = Path.build_filename (Environment.get_user_data_dir (),
  92                                     Config.PACKAGE_TARNAME);
  93
  94         if (!FileUtils.test (path, FileTest.EXISTS)) {
  95             DirUtils.create_with_parents (path, 0700);
  96         }
  97         return path;
  98     }
  99
 100     public void store_id_cards () {
 101         var key_file = new KeyFile ();
 102         foreach (IdCard id_card in this.id_card_list) {
 103             string[] rules_patterns = new string[id_card.rules.length];
 104             string[] rules_always_conf = new string[id_card.rules.length];
 105
 106             for (int i=0; i<id_card.rules.length; i++) {
 107               rules_patterns[i] = id_card.rules[i].pattern;
 108               rules_always_conf[i] = id_card.rules[i].always_confirm;
 109             }
 110
 111             key_file.set_string (id_card.display_name, "Issuer", id_card.issuer ?? "");
 112             key_file.set_string (id_card.display_name, "DisplayName", id_card.display_name ?? "");
 113             key_file.set_string (id_card.display_name, "Username", id_card.username ?? "");
 114             if (id_card.store_password && (id_card.password != null))
 115               key_file.set_string (id_card.display_name, "Password", id_card.password);
 116             else
 117               key_file.set_string (id_card.display_name, "Password", "");
 118             key_file.set_string_list (id_card.display_name, "Services", id_card.services ?? {});
 119
 120             if (id_card.rules.length > 0) {
 121               key_file.set_string_list (id_card.display_name, "Rules-Patterns", rules_patterns);
 122               key_file.set_string_list (id_card.display_name, "Rules-AlwaysConfirm", rules_always_conf);
 123             }
 124             key_file.set_string (id_card.display_name, "StorePassword", id_card.store_password ? "yes" : "no");
 125
 126             // Trust anchor
 127             key_file.set_string (id_card.display_name, "CA-Cert", id_card.trust_anchor.ca_cert ?? "");
 128             key_file.set_string (id_card.display_name, "Subject", id_card.trust_anchor.subject ?? "");
 129             key_file.set_string (id_card.display_name, "SubjectAlt", id_card.trust_anchor.subject_alt ?? "");
 130             key_file.set_string (id_card.display_name, "ServerCert", id_card.trust_anchor.server_cert ?? "");
 131         }
 132
 133         var text = key_file.to_data (null);
 134
 135         try {
 136             var path = get_data_dir ();
 137             var filename = Path.build_filename (path, FILE_NAME);
 138             FileUtils.set_contents (filename, text, -1);
 139         }
 140         catch (Error e) {
 141             stdout.printf ("Error:  %s\n", e.message);
 142         }
 143
 144         load_id_cards();
 145     }
 146
 147      public LocalFlatFileStore () {
 148         id_card_list = new LinkedList<IdCard>();
 149         load_id_cards();
 150      }
 151  }
 152