Add list-toplevels subcommand.

2 files changed, 19 insertions, 0 deletions

diff --git a/obnamlib/plugins/encryption_plugin.py b/obnamlib/plugins/encryption_plugin.py
index 6ee4f362..eee56152 100644
--- a/obnamlib/plugins/encryption_plugin.py
+++ b/obnamlib/plugins/encryption_plugin.py
@@ -41,6 +41,7 @@ class EncryptionPlugin(obnamlib.ObnamPlugin):
         
         self.app.register_command('client-keys', self.client_keys)
         self.app.register_command('list-keys', self.list_keys)
+        self.app.register_command('list-toplevels', self.list_toplevels)
 
     @property
     def keyid(self):
@@ -132,6 +133,21 @@ class EncryptionPlugin(obnamlib.ObnamPlugin):
             for toplevel in keys[keyid]:
                 print '  %s' % toplevel
 
+    def list_toplevels(self, args):
+        repo = self.app.open_repository()
+        toplevels = repo.fs.listdir('.')
+        keys = dict()
+        tops = dict()
+        for toplevel in toplevels:
+            userkeys = self.read_keyring(repo, toplevel)
+            for keyid in userkeys.keyids():
+                keys[keyid] = keys.get(keyid, []) + [toplevel]
+                tops[toplevel] = tops.get(toplevel, []) + [keyid]
+        for toplevel in tops:
+            print 'toplevel: %s' % toplevel
+            for keyid in tops[toplevel]:
+                print '  %s' % keyid
+
 #    def add_client(self, repo, client_public_key):
 #        self.add_to_userkeys(repo, 'metadata', client_public_key)
 #        self.add_to_userkeys(repo, 'clientlist', client_public_key)
diff --git a/test-encrypted-repo b/test-encrypted-repo
index 2193bd38..618232a2 100755
--- a/test-encrypted-repo
+++ b/test-encrypted-repo
@@ -24,3 +24,6 @@ $cmd client-keys
 
 echo "list-keys:"
 $cmd list-keys
+
+echo "list-toplevels:"
+$cmd list-toplevels