From 92cd7f34b07e86062f2c024039e3309768b48ce6 Mon Sep 17 00:00:00 2001 From: Andy Wermke <andy@dev.next-step-software.com> Date: Thu, 04 Apr 2013 10:10:23 -0400 Subject: [PATCH] Merge branch 'master' of https://github.com/roundcube/roundcubemail --- plugins/virtuser_query/virtuser_query.php | 80 ++++++++++++++++++++++++++-------------- 1 files changed, 52 insertions(+), 28 deletions(-) diff --git a/plugins/virtuser_query/virtuser_query.php b/plugins/virtuser_query/virtuser_query.php index 073b4e2..675eb7c 100644 --- a/plugins/virtuser_query/virtuser_query.php +++ b/plugins/virtuser_query/virtuser_query.php @@ -17,6 +17,9 @@ * The email query can return more than one record to create more identities. * This requires identities_level option to be set to value less than 2. * + * By default Roundcube database is used. To use different database (or host) + * you can specify DSN string in $rcmail_config['virtuser_query_dsn'] option. + * * @version @package_version@ * @author Aleksander Machniak <alec@alec.pl> * @author Steffen Vogel @@ -25,11 +28,12 @@ { private $config; private $app; + private $db; function init() { - $this->app = rcmail::get_instance(); - $this->config = $this->app->config->get('virtuser_query'); + $this->app = rcmail::get_instance(); + $this->config = $this->app->config->get('virtuser_query'); if (!empty($this->config)) { if (is_string($this->config)) { @@ -53,35 +57,35 @@ */ function user2email($p) { - $dbh = $this->app->get_dbh(); + $dbh = $this->get_dbh(); - $sql_result = $dbh->query(preg_replace('/%u/', $dbh->escapeSimple($p['user']), $this->config['email'])); + $sql_result = $dbh->query(preg_replace('/%u/', $dbh->escapeSimple($p['user']), $this->config['email'])); - while ($sql_arr = $dbh->fetch_array($sql_result)) { - if (strpos($sql_arr[0], '@')) { - if ($p['extended'] && count($sql_arr) > 1) { - $result[] = array( - 'email' => rcube_idn_to_ascii($sql_arr[0]), - 'name' => $sql_arr[1], - 'organization' => $sql_arr[2], - 'reply-to' => rcube_idn_to_ascii($sql_arr[3]), - 'bcc' => rcube_idn_to_ascii($sql_arr[4]), - 'signature' => $sql_arr[5], - 'html_signature' => (int)$sql_arr[6], - ); - } - else { - $result[] = $sql_arr[0]; - } + while ($sql_arr = $dbh->fetch_array($sql_result)) { + if (strpos($sql_arr[0], '@')) { + if ($p['extended'] && count($sql_arr) > 1) { + $result[] = array( + 'email' => rcube_utils::idn_to_ascii($sql_arr[0]), + 'name' => $sql_arr[1], + 'organization' => $sql_arr[2], + 'reply-to' => rcube_utils::idn_to_ascii($sql_arr[3]), + 'bcc' => rcube_utils::idn_to_ascii($sql_arr[4]), + 'signature' => $sql_arr[5], + 'html_signature' => (int)$sql_arr[6], + ); + } + else { + $result[] = $sql_arr[0]; + } - if ($p['first']) - break; - } - } + if ($p['first']) + break; + } + } - $p['email'] = $result; + $p['email'] = $result; - return $p; + return $p; } /** @@ -89,7 +93,7 @@ */ function email2user($p) { - $dbh = $this->app->get_dbh(); + $dbh = $this->get_dbh(); $sql_result = $dbh->query(preg_replace('/%m/', $dbh->escapeSimple($p['email']), $this->config['user'])); @@ -105,7 +109,7 @@ */ function user2host($p) { - $dbh = $this->app->get_dbh(); + $dbh = $this->get_dbh(); $sql_result = $dbh->query(preg_replace('/%u/', $dbh->escapeSimple($p['user']), $this->config['host'])); @@ -116,5 +120,25 @@ return $p; } + /** + * Initialize database handler + */ + function get_dbh() + { + if (!$this->db) { + if ($dsn = $this->app->config->get('virtuser_query_dsn')) { + // connect to the virtuser database + $this->db = rcube_db::factory($dsn); + $this->db->set_debug((bool)$this->app->config->get('sql_debug')); + $this->db->db_connect('r'); // connect in read mode + } + else { + $this->db = $this->app->get_dbh(); + } + } + + return $this->db; + } + } -- Gitblit v1.9.1