Implement user_exists() everywhere and use it to correct the console output.
[mailshears.git] / lib / rm / plugins / roundcube.rb
1 require 'pg'
2
3 require 'common/roundcube_plugin'
4 require 'rm/rm_plugin'
5
6 class RoundcubeRm
7
8 include RoundcubePlugin
9 include RmPlugin
10
11 def delete_account(account)
12 # Delete the given username and any records in other tables
13 # belonging to it.
14 raise NonexistentAccountError.new(account) if not user_exists(account)
15
16 user_id = self.get_user_id(account)
17
18 # The Roundcube developers were nice enough to include
19 # DBMS-specific install and upgrade scripts, so Postgres can take
20 # advantage of ON DELETE triggers. Here's an example:
21 #
22 # ...
23 # user_id integer NOT NULL
24 # REFERENCES users (user_id) ON DELETE CASCADE ON UPDATE CASCADE
25 #
26 # This query is of course necessary with any DBMS:
27 sql_queries = ['DELETE FROM users WHERE user_id = $1::int;']
28
29 begin
30 connection = PGconn.connect(@db_host,
31 @db_port,
32 @db_opts,
33 @db_tty,
34 @db_name,
35 @db_user,
36 @db_pass)
37
38 sql_queries.each do |sql_query|
39 connection.query(sql_query, [user_id])
40 end
41
42 connection.close()
43
44 rescue PGError => e
45 # Pretend like we're database-agnostic in case we ever are.
46 raise DatabaseError.new(e)
47 end
48
49 end
50
51 end