From 4c7cfbd70a9dbe1c0168520c6c600b019703f289 Mon Sep 17 00:00:00 2001 From: Christoph Date: Wed, 15 Jun 2016 12:03:40 +0200 Subject: add ability to completely remove a user from identities and user db --- destroy-user-db | 39 ++++++++++++++++++++++++++++++++++----- 1 file changed, 34 insertions(+), 5 deletions(-) diff --git a/destroy-user-db b/destroy-user-db index fdf917b..00b17f1 100755 --- a/destroy-user-db +++ b/destroy-user-db @@ -6,20 +6,46 @@ def main() parse_opts server = LEAP::Server.new users_db = LEAP::Users.new(server) + identities_db = server.database('identities') + all_users_db = server.database('users') record = users_db.find_by_login($username) + user_id = record.id + return_code = 0 if record.nil? puts "Username #{$username} does not exist." - exit 1 + return_code=1 end - dbname = "user-#{record.id}" + dbname = "user-#{user_id}" begin server.database(dbname).delete! puts "Deleted DB #{dbname}" - exit 0 rescue RestClient::ResourceNotFound puts "DB does not exist: #{dbname}" - exit 1 + return_code=1 end + if $destroy_identities == TRUE + identities_db.view('Identity/by_user_id', :reduce => false, :startkey => user_id, :endkey => user_id)['rows'].each do |row| + begin + doc = identities_db.get(row['id']) + identities_db.delete_doc(doc) + puts " deleted #{identities_db.name}/#{row['id']}" + rescue RestClient::ResourceNotFound + puts "identities DB does not exist: #{doc}" + return_code=1 + end + end + all_users_db.view('User/by_login', :reduce => false, :startkey => $username, :endkey => $username)['rows'].each do |row| + begin + doc = all_users_db.get(user_id) + all_users_db.delete_doc(doc) + puts " deleted #{user_id} from user db" + rescue RestClient::ResourceNotFound + puts "user not found in user db: #{user_id}" + return_code=1 + end + end + end + exit return_code end def parse_opts() @@ -28,6 +54,8 @@ def parse_opts() case cmd when "--username" $username = ARGV.shift + when "--destroy-identities" + $destroy_identities=TRUE else usage end @@ -41,7 +69,8 @@ def usage() puts "#{$0} COMMAND" puts "Commands:" puts " --username USERNAME - destroy the user db corresponding to USERNAME." + puts " --destroy-identities - re-enable user for future sign-up" exit(1) end -main() \ No newline at end of file +main() -- cgit v1.2.3