X-Git-Url: http://gitweb.michael.orlitzky.com/?p=mailshears.git;a=blobdiff_plain;f=lib%2Frm%2Fplugins%2Fagendav.rb;h=2c7709011ca3a421ced83d7d58e2f1aa3c6c2faf;hp=28f1eb4167a4a63bec61889ea47b97492d87f80f;hb=fa7782720ff15fce29b6f875678e9fd0c197485a;hpb=df4e02ebf6a4e28a58abcb298a4442a245ad0b15 diff --git a/lib/rm/plugins/agendav.rb b/lib/rm/plugins/agendav.rb index 28f1eb4..2c77090 100644 --- a/lib/rm/plugins/agendav.rb +++ b/lib/rm/plugins/agendav.rb @@ -16,22 +16,43 @@ class AgendavRm # Remove *user* from the Agendav database. This should remove him # from _every_ table in which he is referenced. # + # We do not raise an error if the user doesn't exist. This is due to + # an unfortunate problem with the "user exists" check in AgenDAV. + # The AgenDAV "shares" table is not tied directly to a username, so + # we are forced to use a regexp match to decide what rows to delete + # from that table. We do so regardless of whether or not the username + # exists in the "prefs" table, because that table stores only non- + # default preferences -- not all users' preferences. + # # @param user [User] the user to remove. # def remove_user(user) - raise NonexistentUserError.new(user.to_s()) if not user_exists(user) - sql_queries = ['DELETE FROM prefs WHERE username = $1;'] - sql_queries << 'DELETE FROM shared WHERE user_from = $1;' - connection = PGconn.connect(@db_host, @db_port, @db_opts, @db_tty, - @db_name, @db_user, @db_pass) - - sql_queries.each do |sql_query| - connection.query(sql_query, [user.to_s()]) + # The "shares" table contains principal URLs, and the "@" symbol + # is usually encoded to "%40". These queries do a regex match on + # the username after replacing the "%40" with a "@". + # + # As a precaution, I haven chosen not to delete based on the + # "calendar" field here. Nobody should have a calendar named + # "user%40example.com", but it's not impossible -- and we don't + # want to delete that calendar when the not-necessarily-related + # "user@example.com" account is removed. And the usual appearance + # of the user's email address in the "calendar" field happens when + # he is also the owner, so the calendar does get deleted in the + # normal situation. + sql_queries << "DELETE FROM shares WHERE REPLACE(owner,'%40','@') ~ $1;" + sql_queries << "DELETE FROM shares WHERE REPLACE(\"with\",'%40','@') ~ $1;" + + connection = PG::Connection.new(@db_hash) + begin + sql_queries.each do |sql_query| + connection.sync_exec_params(sql_query, [user.to_s()]) + end + ensure + # Make sure the connection gets closed even if a query explodes. + connection.close() end - - connection.close() end end