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