require 'common/plugin' module RoundcubePlugin # Code that all Roundcube plugins (Prune, Rm, Mv...) will share. # That is, we implement the Plugin interface. include Plugin def initialize(cfg) @db_host = cfg.roundcube_dbhost @db_port = cfg.roundcube_dbport @db_opts = cfg.roundcube_dbopts @db_tty = cfg.roundcube_dbtty @db_name = cfg.roundcube_dbname @db_user = cfg.roundcube_dbuser @db_pass = cfg.roundcube_dbpass end def describe_domain(domain) # Roundcube doesn't have a concept of domains. return domain end def describe_user(user) user_id = self.get_user_id(user) if user_id.nil? return 'User not found' else return "User ID: #{user_id}" end end def list_users() # Produce a list of Roundcube users. This is used in prune/rm, and # is public because it is useful in testing. usernames = [] # Just assume PostgreSQL for now. begin connection = PGconn.connect(@db_host, @db_port, @db_opts, @db_tty, @db_name, @db_user, @db_pass) sql_query = "SELECT username FROM users;" connection.query(sql_query) do |result| usernames = result.field_values('username') end connection.close() rescue PGError => e # Pretend like we're database-agnostic in case we ever are. raise DatabaseError.new(e) end return usernames end protected; def get_user_id(user) user_id = nil begin connection = PGconn.connect(@db_host, @db_port, @db_opts, @db_tty, @db_name, @db_user, @db_pass) sql_query = "SELECT user_id FROM users WHERE username = $1;" connection.query(sql_query, [user]) do |result| if result.num_tuples > 0 user_id = result[0]['user_id'] end end connection.close() rescue PGError => e # Pretend like we're database-agnostic in case we ever are. raise DatabaseError.new(e) end return user_id end end