X-Git-Url: http://gitweb.michael.orlitzky.com/?p=mailshears.git;a=blobdiff_plain;f=lib%2Fcommon%2Fagendav_plugin.rb;h=9af7b01519b40aca2870176948ed4f0ac6f950cb;hp=ae9b11c0b05d89874ebdc6dedbc2e85af36c6b89;hb=20b843bddcd73833d41f98ff79d92ef59bb4d81e;hpb=bf7d0402eda27d9487ca9402156818545fdda286 diff --git a/lib/common/agendav_plugin.rb b/lib/common/agendav_plugin.rb index ae9b11c..9af7b01 100644 --- a/lib/common/agendav_plugin.rb +++ b/lib/common/agendav_plugin.rb @@ -1,4 +1,5 @@ require 'common/plugin' +require 'common/user' module AgendavPlugin # Code that all Agendav plugins (Prune, Rm, Mv...) will @@ -17,53 +18,27 @@ module AgendavPlugin end - def describe_domain(domain) - # AgenDAV doesn't have a concept of domains. - return domain - end - - - def describe_user(user) - if self.user_exists(user) - return "Username: #{user}" - else - return 'User not found' - end - end - - def list_users() # # Produce a list of AgenDAV users. This is public because it's # useful in testing. # - usernames = [] + users = [] - # Just assume PostgreSQL for now. - begin - connection = PGconn.connect(@db_host, - @db_port, - @db_opts, - @db_tty, - @db_name, - @db_user, - @db_pass) + connection = PGconn.connect(@db_host, @db_port, @db_opts, @db_tty, + @db_name, @db_user, @db_pass) - sql_query = '(SELECT username FROM prefs)' - sql_query += 'UNION' - sql_query += '(SELECT user_from FROM shared);' + sql_query = '(SELECT username FROM prefs)' + sql_query += 'UNION' + sql_query += '(SELECT user_from FROM shared);' - 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) + connection.query(sql_query) do |result| + users = result.field_values('username') end - return usernames + connection.close() + + return users.map{ |u| User.new(u) } end end