added cleanup-user-dbs

2 files changed, 13 insertions, 0 deletions

diff --git a/lib/leap/server.rb b/lib/leap/server.rb
index bc60455..05d76e3 100644
--- a/lib/leap/server.rb
+++ b/lib/leap/server.rb
@@ -2,10 +2,22 @@ require 'couchrest'
 
 module LEAP
   class Server < CouchRest::Server
+
     def initialize
       netrc = File.read('/root/.netrc').split(' ')
       auth = "%{username}:%{password}@" % {username: netrc[3], password: netrc[5]}
       super("http://#{auth}localhost:5984")
     end
+
+    #
+    # returns an array of the names of all the per-user storage
+    # databases.
+    #
+    def storage_dbs
+      self.databases.select { |db_name|
+        db_name =~ /^user-[a-f0-9]{32}$/
+      }
+    end
+
   end
 end
\ No newline at end of file
diff --git a/lib/leap/users.rb b/lib/leap/users.rb
index 14c9c4e..b9dc12c 100644
--- a/lib/leap/users.rb
+++ b/lib/leap/users.rb
@@ -30,6 +30,7 @@ module LEAP
     end
 
     def all_ids
+      self.all_docs["rows"].map {|row| row["id"]}
     end
 
   end