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      private void load_id_cards() {
  28         id_card_list.clear();
  29         var key_file = new KeyFile ();
  30         var path = get_data_dir ();
  31         var filename = Path.build_filename (path, FILE_NAME);
  32
  33         try {
  34             key_file.load_from_file (filename, KeyFileFlags.NONE);
  35         }
  36         catch (Error e) {
  37             stdout.printf("Error: %s\n", e.message);
  38             return;
  39         }
  40
  41         var identities_uris = key_file.get_groups ();
  42         foreach (string identity in identities_uris) {
  43             try {
  44                 IdCard id_card = new IdCard ();
  45
  46                 id_card.issuer = key_file.get_string (identity, "Issuer");
  47                 id_card.username = key_file.get_string (identity, "Username");
  48                 id_card.password = key_file.get_string (identity, "Password");
  49                 id_card.services = key_file.get_string_list (identity, "Services");
  50                 id_card.display_name = key_file.get_string (identity, "DisplayName");
  51                 id_card.set_data ("pixbuf", find_icon ("avatar-default", 48));
  52
  53
  54                 if (key_file.has_key (identity, "Rules-Patterns") &&
  55                     key_file.has_key (identity, "Rules-AlwaysConfirm")) {
  56                     string [] rules_patterns =    key_file.get_string_list (identity, "Rules-Patterns");
  57                     string [] rules_always_conf = key_file.get_string_list (identity, "Rules-AlwaysConfirm");
  58
  59                     if (rules_patterns.length == rules_always_conf.length) {
  60                       Rule[] rules = new Rule[rules_patterns.length];
  61                       for (int i = 0; i < rules_patterns.length; i++) {
  62                         rules[i] = {rules_patterns[i], rules_always_conf[i]};
  63                       }
  64                       id_card.rules = rules;
  65                     }
  66                 }
  67
  68                 // Trust anchor
  69                 id_card.trust_anchor.ca_cert = key_file.get_string (identity, "CA-Cert");
  70                 id_card.trust_anchor.subject = key_file.get_string (identity, "Subject");
  71                 id_card.trust_anchor.subject_alt = key_file.get_string (identity, "SubjectAlt");
  72                 id_card.trust_anchor.server_cert = key_file.get_string (identity, "ServerCert");
  73
  74                 id_card_list.add (id_card);
  75             }
  76             catch (Error e) {
  77                 stdout.printf ("Error:  %s\n", e.message);
  78             }
  79         }
  80      }
  81
  82     private string get_data_dir() {
  83         string path;
  84         path = Path.build_filename (Environment.get_user_data_dir (),
  85                                     Config.PACKAGE_TARNAME);
  86
  87         if (!FileUtils.test (path, FileTest.EXISTS)) {
  88             DirUtils.create (path, 0700);
  89         }
  90         return path;
  91     }
  92
  93     public void store_id_cards () {
  94         var key_file = new KeyFile ();
  95         foreach (IdCard id_card in this.id_card_list) {
  96             string[] rules_patterns = new string[id_card.rules.length];
  97             string[] rules_always_conf = new string[id_card.rules.length];
  98
  99             for (int i=0; i<id_card.rules.length; i++) {
 100               rules_patterns[i] = id_card.rules[i].pattern;
 101               rules_always_conf[i] = id_card.rules[i].always_confirm;
 102             }
 103
 104             key_file.set_string (id_card.display_name, "Issuer", id_card.issuer ?? "");
 105             key_file.set_string (id_card.display_name, "DisplayName", id_card.display_name ?? "");
 106             key_file.set_string (id_card.display_name, "Username", id_card.username ?? "");
 107             key_file.set_string (id_card.display_name, "Password", id_card.password ?? "");
 108             key_file.set_string_list (id_card.display_name, "Services", id_card.services ?? {});
 109
 110             if (id_card.rules.length > 0) {
 111               key_file.set_string_list (id_card.display_name, "Rules-Patterns", rules_patterns);
 112               key_file.set_string_list (id_card.display_name, "Rules-AlwaysConfirm", rules_always_conf);
 113             }
 114
 115             // Trust anchor
 116             key_file.set_string (id_card.display_name, "CA-Cert", id_card.trust_anchor.ca_cert ?? "");
 117             key_file.set_string (id_card.display_name, "Subject", id_card.trust_anchor.subject ?? "");
 118             key_file.set_string (id_card.display_name, "SubjectAlt", id_card.trust_anchor.subject_alt ?? "");
 119             key_file.set_string (id_card.display_name, "ServerCert", id_card.trust_anchor.server_cert ?? "");
 120         }
 121
 122         var text = key_file.to_data (null);
 123
 124         try {
 125             var path = get_data_dir ();
 126             var filename = Path.build_filename (path, FILE_NAME);
 127             FileUtils.set_contents (filename, text, -1);
 128         }
 129         catch (Error e) {
 130             stdout.printf ("Error:  %s\n", e.message);
 131         }
 132
 133         load_id_cards();
 134     }
 135
 136      public LocalFlatFileStore () {
 137         id_card_list = new LinkedList<IdCard>();
 138         load_id_cards();
 139      }
 140  }
 141