X-Git-Url: http://gitweb.michael.orlitzky.com/?p=mailshears.git;a=blobdiff_plain;f=lib%2Fcommon%2Fagendav_plugin.rb;h=9af7b01519b40aca2870176948ed4f0ac6f950cb;hp=032d057c24d8e25c899ea672ae5fca50a711a8ff;hb=20b843bddcd73833d41f98ff79d92ef59bb4d81e;hpb=08c59f98a558acd7f166e5382a1b2fec5b57d68c;ds=sidebyside diff --git a/lib/common/agendav_plugin.rb b/lib/common/agendav_plugin.rb index 032d057..9af7b01 100644 --- a/lib/common/agendav_plugin.rb +++ b/lib/common/agendav_plugin.rb @@ -25,30 +25,19 @@ module AgendavPlugin # users = [] - # 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 prefs)' - sql_query += 'UNION' - sql_query += '(SELECT user_from FROM shared);' - - connection.query(sql_query) do |result| - users = 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 = 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);' + + connection.query(sql_query) do |result| + users = result.field_values('username') end + connection.close() + return users.map{ |u| User.new(u) } end