summaryrefslogtreecommitdiff
path: root/program/include/rcmail_utils.php
diff options
context:
space:
mode:
Diffstat (limited to 'program/include/rcmail_utils.php')
-rw-r--r--program/include/rcmail_utils.php362
1 files changed, 362 insertions, 0 deletions
diff --git a/program/include/rcmail_utils.php b/program/include/rcmail_utils.php
new file mode 100644
index 000000000..50700b0e2
--- /dev/null
+++ b/program/include/rcmail_utils.php
@@ -0,0 +1,362 @@
+<?php
+
+/*
+ +-----------------------------------------------------------------------+
+ | program/include/rcmail_utils.php |
+ | |
+ | This file is part of the Roundcube PHP suite |
+ | Copyright (C) 2005-2015 The Roundcube Dev Team |
+ | |
+ | Licensed under the GNU General Public License version 3 or |
+ | any later version with exceptions for skins & plugins. |
+ | See the README file for a full license statement. |
+ | |
+ | CONTENTS: |
+ | Roundcube utilities |
+ | |
+ +-----------------------------------------------------------------------+
+ | Author: Thomas Bruederli <roundcube@gmail.com> |
+ | Author: Aleksander Machniak <alec@alec.pl> |
+ +-----------------------------------------------------------------------+
+*/
+
+/**
+ * Roundcube utilities
+ *
+ * @package Webmail
+ * @subpackage Utils
+ */
+class rcmail_utils
+{
+ public static $db;
+
+ /**
+ * Initialize database object and connect
+ *
+ * @return rcube_db Database instance
+ */
+ public static function db()
+ {
+ if (self::$db === null) {
+ $rc = rcube::get_instance();
+ $db = rcube_db::factory($rc->config->get('db_dsnw'));
+
+ $db->set_debug((bool)$rc->config->get('sql_debug'));
+
+ // Connect to database
+ $db->db_connect('w');
+
+ if (!$db->is_connected()) {
+ rcube::raise_error("Error connecting to database: " . $db->is_error(), false, true);
+ }
+
+ self::$db = $db;
+ }
+
+ return self::$db;
+ }
+
+ /**
+ * Initialize database schema
+ *
+ * @param string Directory with sql files
+ */
+ public static function db_init($dir)
+ {
+ $db = self::db();
+
+ $file = $dir . '/' . $db->db_provider . '.initial.sql';
+ if (!file_exists($file)) {
+ rcube::raise_error("DDL file $file not found", false, true);
+ }
+
+ echo "Creating database schema... ";
+
+ if ($sql = file_get_contents($file)) {
+ if (!$db->exec_script($sql)) {
+ $error = $db->is_error();
+ }
+ }
+ else {
+ $error = "Unable to read file $file or it is empty";
+ }
+
+ if ($error) {
+ echo "[FAILED]\n";
+ rcube::raise_error($error, false, true);
+ }
+ else {
+ echo "[OK]\n";
+ }
+ }
+
+ /**
+ * Update database schema
+ *
+ * @param string Directory with sql files
+ * @param string Component name
+ * @param string Optional current version number
+ * @param array Parameters (errors, quiet)
+ *
+ * @return True on success, False on failure
+ */
+ public static function db_update($dir, $package, $ver = null, $opts = array())
+ {
+ // Check if directory exists
+ if (!file_exists($dir)) {
+ if ($opts['errors']) {
+ rcube::raise_error("Specified database schema directory doesn't exist.", false, true);
+ }
+ return false;
+ }
+
+ $db = self::db();
+
+ // Read DB schema version from database (if 'system' table exists)
+ if (in_array($db->table_name('system'), (array)$db->list_tables())) {
+ $db->query("SELECT `value`"
+ . " FROM " . $db->table_name('system', true)
+ . " WHERE `name` = ?",
+ $package . '-version');
+
+ $row = $db->fetch_array();
+ $version = preg_replace('/[^0-9]/', '', $row[0]);
+ }
+
+ // DB version not found, but release version is specified
+ if (!$version && $ver) {
+ // Map old release version string to DB schema version
+ // Note: This is for backward compat. only, do not need to be updated
+ $map = array(
+ '0.1-stable' => 1,
+ '0.1.1' => 2008030300,
+ '0.2-alpha' => 2008040500,
+ '0.2-beta' => 2008060900,
+ '0.2-stable' => 2008092100,
+ '0.2.1' => 2008092100,
+ '0.2.2' => 2008092100,
+ '0.3-stable' => 2008092100,
+ '0.3.1' => 2009090400,
+ '0.4-beta' => 2009103100,
+ '0.4' => 2010042300,
+ '0.4.1' => 2010042300,
+ '0.4.2' => 2010042300,
+ '0.5-beta' => 2010100600,
+ '0.5' => 2010100600,
+ '0.5.1' => 2010100600,
+ '0.5.2' => 2010100600,
+ '0.5.3' => 2010100600,
+ '0.5.4' => 2010100600,
+ '0.6-beta' => 2011011200,
+ '0.6' => 2011011200,
+ '0.7-beta' => 2011092800,
+ '0.7' => 2011111600,
+ '0.7.1' => 2011111600,
+ '0.7.2' => 2011111600,
+ '0.7.3' => 2011111600,
+ '0.7.4' => 2011111600,
+ '0.8-beta' => 2011121400,
+ '0.8-rc' => 2011121400,
+ '0.8.0' => 2011121400,
+ '0.8.1' => 2011121400,
+ '0.8.2' => 2011121400,
+ '0.8.3' => 2011121400,
+ '0.8.4' => 2011121400,
+ '0.8.5' => 2011121400,
+ '0.8.6' => 2011121400,
+ '0.9-beta' => 2012080700,
+ );
+
+ $version = $map[$ver];
+ }
+
+ // Assume last version before the 'system' table was added
+ if (empty($version)) {
+ $version = 2012080700;
+ }
+
+ $dir .= '/' . $db->db_provider;
+ if (!file_exists($dir)) {
+ if ($opts['errors']) {
+ rcube::raise_error("DDL Upgrade files for " . $db->db_provider . " driver not found.", false, true);
+ }
+ return false;
+ }
+
+ $dh = opendir($dir);
+ $result = array();
+
+ while ($file = readdir($dh)) {
+ if (preg_match('/^([0-9]+)\.sql$/', $file, $m) && $m[1] > $version) {
+ $result[] = $m[1];
+ }
+ }
+ sort($result, SORT_NUMERIC);
+
+ foreach ($result as $v) {
+ if (!$opts['quiet']) {
+ echo "Updating database schema ($v)... ";
+ }
+
+ $error = self::db_update_schema($package, $v, "$dir/$v.sql");
+
+ if ($error) {
+ if (!$opts['quiet']) {
+ echo "[FAILED]\n";
+ }
+ if ($opts['errors']) {
+ rcube::raise_error("Error in DDL upgrade $v: $error", false, true);
+ }
+ return false;
+ }
+ else if (!$opts['quiet']) {
+ echo "[OK]\n";
+ }
+ }
+
+ return true;
+ }
+
+ /**
+ * Run database update from a single sql file
+ */
+ protected static function db_update_schema($package, $version, $file)
+ {
+ $db = self::db();
+
+ // read DDL file
+ if ($sql = file_get_contents($file)) {
+ if (!$db->exec_script($sql)) {
+ return $db->is_error();
+ }
+ }
+
+ // escape if 'system' table does not exist
+ if ($version < 2013011000) {
+ return;
+ }
+
+ $system_table = $db->table_name('system', true);
+
+ $db->query("UPDATE " . $system_table
+ . " SET `value` = ?"
+ . " WHERE `name` = ?",
+ $version, $package . '-version');
+
+ if (!$db->is_error() && !$db->affected_rows()) {
+ $db->query("INSERT INTO " . $system_table
+ ." (`name`, `value`) VALUES (?, ?)",
+ $package . '-version', $version);
+ }
+
+ return $db->is_error();
+ }
+
+ /**
+ * Removes all deleted records older than X days
+ *
+ * @param int Number of days
+ */
+ public static function db_clean($days)
+ {
+ // mapping for table name => primary key
+ $primary_keys = array(
+ 'contacts' => 'contact_id',
+ 'contactgroups' => 'contactgroup_id',
+ );
+
+ $db = self::db();
+
+ $threshold = date('Y-m-d 00:00:00', time() - $days * 86400);
+
+ foreach (array('contacts','contactgroups','identities') as $table) {
+ $sqltable = $db->table_name($table, true);
+
+ // also delete linked records
+ // could be skipped for databases which respect foreign key constraints
+ if ($db->db_provider == 'sqlite' && ($table == 'contacts' || $table == 'contactgroups')) {
+ $pk = $primary_keys[$table];
+ $memberstable = $db->table_name('contactgroupmembers');
+
+ $db->query(
+ "DELETE FROM " . $db->quote_identifier($memberstable)
+ . " WHERE `$pk` IN ("
+ . "SELECT `$pk` FROM $sqltable"
+ . " WHERE `del` = 1 AND `changed` < ?"
+ . ")",
+ $threshold);
+
+ echo $db->affected_rows() . " records deleted from '$memberstable'\n";
+ }
+
+ // delete outdated records
+ $db->query("DELETE FROM $sqltable WHERE `del` = 1 AND `changed` < ?", $threshold);
+
+ echo $db->affected_rows() . " records deleted from '$table'\n";
+ }
+ }
+
+ /**
+ * Reindex contacts
+ */
+ public static function indexcontacts()
+ {
+ $db = self::db();
+
+ // iterate over all users
+ $sql_result = $db->query("SELECT `user_id` FROM " . $db->table_name('users', true) . " ORDER BY `user_id`");
+ while ($sql_result && ($sql_arr = $db->fetch_assoc($sql_result))) {
+ echo "Indexing contacts for user " . $sql_arr['user_id'] . "...\n";
+
+ $contacts = new rcube_contacts($db, $sql_arr['user_id']);
+ $contacts->set_pagesize(9999);
+
+ $result = $contacts->list_records();
+ while ($result->count && ($row = $result->next())) {
+ unset($row['words']);
+ $contacts->update($row['ID'], $row);
+ }
+ }
+
+ echo "done.\n";
+ }
+
+ /**
+ * Modify user preferences
+ *
+ * @param string Option name
+ * @param string Option value
+ * @param int Optional user identifier
+ */
+ public static function mod_pref($name, $value, $userid = null)
+ {
+ $db = self::db();
+
+ if ($userid) {
+ $query = '`user_id` = ' . intval($userid);
+ }
+ else {
+ $query = '1=1';
+ }
+
+ // iterate over all users
+ $sql_result = $db->query("SELECT * FROM " . $db->table_name('users', true) . " WHERE $query");
+
+ while ($sql_result && ($sql_arr = $db->fetch_assoc($sql_result))) {
+ echo "Updating prefs for user " . $sql_arr['user_id'] . "...";
+
+ $user = new rcube_user($sql_arr['user_id'], $sql_arr);
+ $prefs = $old_prefs = $user->get_prefs();
+
+ $prefs[$name] = $value;
+
+ if ($prefs != $old_prefs) {
+ $user->save_prefs($prefs);
+ echo "saved.\n";
+ }
+ else {
+ echo "nothing changed.\n";
+ }
+ }
+ }
+}