alecpl
2010-04-23 bf67d60e435a6f01b835a0d5afb9737f81560433
program/include/rcube_user.php
@@ -16,7 +16,7 @@
 | Author: Thomas Bruederli <roundcube@gmail.com>                        |
 +-----------------------------------------------------------------------+
 $Id: rcube_user.inc 933 2007-11-29 14:17:32Z thomasb $
 $Id$
*/
@@ -47,7 +47,7 @@
    
    if ($id && !$sql_arr)
    {
      $sql_result = $this->db->query("SELECT * FROM ".get_table_name('users')." WHERE  user_id=?", $id);
      $sql_result = $this->db->query("SELECT * FROM ".get_table_name('users')." WHERE user_id=?", $id);
      $sql_arr = $this->db->fetch_assoc($sql_result);
    }
    
@@ -111,19 +111,22 @@
      if (!isset($old_prefs[$key]) && ($value == $config->get($key)))
        unset($save_prefs[$key]);
    }
    $save_prefs = serialize($save_prefs);
    $this->db->query(
      "UPDATE ".get_table_name('users')."
       SET    preferences=?,
              language=?
       WHERE  user_id=?",
      serialize($save_prefs),
      $save_prefs,
      $_SESSION['language'],
      $this->ID);
    $this->language = $_SESSION['language'];
    if ($this->db->affected_rows()) {
      $config->merge($a_user_prefs);
      $config->set_user_prefs($a_user_prefs);
      $this->data['preferences'] = $save_prefs;
      return true;
    }
@@ -135,12 +138,12 @@
   * Get default identity of this user
   *
   * @param int  Identity ID. If empty, the default identity is returned
   * @return array Hash array with all cols of the
   * @return array Hash array with all cols of the identity record
   */
  function get_identity($id = null)
  {
    $sql_result = $this->list_identities($id ? sprintf('AND identity_id=%d', $id) : '');
    return $this->db->fetch_assoc($sql_result);
    $result = $this->list_identities($id ? sprintf('AND identity_id=%d', $id) : '');
    return $result[0];
  }
  
  
@@ -154,13 +157,17 @@
    // get contacts from DB
    $sql_result = $this->db->query(
      "SELECT * FROM ".get_table_name('identities')."
       WHERE  del<>1
       AND    user_id=?
       WHERE del<>1 AND user_id=?
       $sql_add
       ORDER BY ".$this->db->quoteIdentifier('standard')." DESC, name ASC, identity_id ASC",
      $this->ID);
    
    return $sql_result;
    $result = array();
    while ($sql_arr = $this->db->fetch_assoc($sql_result)) {
      $result[] = $sql_arr;
    }
    return $result;
  }
  
  
@@ -176,23 +183,24 @@
    if (!$this->ID)
      return false;
    
    $write_sql = array();
    $query_cols = $query_params = array();
    
    foreach ((array)$data as $col => $value)
    {
      $write_sql[] = sprintf("%s=%s",
        $this->db->quoteIdentifier($col),
        $this->db->quote($value));
      $query_cols[] = $this->db->quoteIdentifier($col) . '=?';
      $query_params[] = $value;
    }
    $this->db->query(
      "UPDATE ".get_table_name('identities')."
       SET ".join(', ', $write_sql)."
    $query_params[] = $iid;
    $query_params[] = $this->ID;
    $sql = "UPDATE ".get_table_name('identities')."
       SET    changed=".$this->db->now().", ".join(', ', $query_cols)."
       WHERE  identity_id=?
       AND    user_id=?
       AND    del<>1",
      $iid,
      $this->ID);
       AND    del<>1";
    call_user_func_array(array($this->db, 'query'),
                        array_merge(array($sql), $query_params));
    
    return $this->db->affected_rows();
  }
@@ -209,20 +217,25 @@
    if (!$this->ID)
      return false;
    unset($data['user_id']);
    $insert_cols = $insert_values = array();
    foreach ((array)$data as $col => $value)
    {
      $insert_cols[] = $this->db->quoteIdentifier($col);
      $insert_values[] = $this->db->quote($value);
      $insert_values[] = $value;
    }
    $insert_cols[] = 'user_id';
    $insert_values[] = $this->ID;
    $this->db->query(
      "INSERT INTO ".get_table_name('identities')."
        (user_id, ".join(', ', $insert_cols).")
       VALUES (?, ".join(', ', $insert_values).")",
      $this->ID);
    $sql = "INSERT INTO ".get_table_name('identities')."
        (changed, ".join(', ', $insert_cols).")
       VALUES (".$this->db->now().", ".join(', ', array_pad(array(), sizeof($insert_values), '?')).")";
    return $this->db->insert_id(get_sequence_name('identities'));
    call_user_func_array(array($this->db, 'query'),
                        array_merge(array($sql), $insert_values));
    return $this->db->insert_id('identities');
  }
  
  
@@ -237,12 +250,9 @@
    if (!$this->ID)
      return false;
    if (!$this->ID || $this->ID == '')
      return false;
    $sql_result = $this->db->query("SELECT count(*) AS ident_count FROM " .
      get_table_name('identities') .
      " WHERE user_id = ? AND del <> 1",
    $sql_result = $this->db->query(
      "SELECT count(*) AS ident_count FROM ".get_table_name('identities')."
       WHERE user_id = ? AND del <> 1",
      $this->ID);
    $sql_arr = $this->db->fetch_assoc($sql_result);
@@ -251,7 +261,7 @@
    
    $this->db->query(
      "UPDATE ".get_table_name('identities')."
       SET    del=1
       SET    del=1, changed=".$this->db->now()."
       WHERE  user_id=?
       AND    identity_id=?",
      $this->ID,
@@ -350,17 +360,22 @@
    $user_email = '';
    $rcmail = rcmail::get_instance();
    $data = $rcmail->plugins->exec_hook('create_user', array('user'=>$user, 'user_name'=>$user_name, 'user_email'=>$user_email));
    // try to resolve user in virtuser table and file
    if ($email_list = self::user2email($user, false, true)) {
      $user_email = is_array($email_list[0]) ? $email_list[0]['email'] : $email_list[0];
    }
    $data = $rcmail->plugins->exec_hook('create_user',
   array('user'=>$user, 'user_name'=>$user_name, 'user_email'=>$user_email));
    // plugin aborted this operation
    if ($data['abort'])
      return false;
    $user_name = $data['user_name'];
    $user_email = $data['user_email'];
    $dbh = $rcmail->get_dbh();
    // try to resolve user in virtuser table and file
    if ($user_email != '' && !strpos($user, '@')) {
      if ($email_list = self::user2email($user, false))
        $user_email = $email_list[0];
    }
    $dbh->query(
      "INSERT INTO ".get_table_name('users')."
@@ -368,11 +383,15 @@
       VALUES (".$dbh->now().", ".$dbh->now().", ?, ?, ?, ?)",
      strip_newlines($user),
      strip_newlines($host),
      strip_newlines($user_email),
      strip_newlines($data['alias'] ? $data['alias'] : $user_email),
      $_SESSION['language']);
    if ($user_id = $dbh->insert_id(get_sequence_name('users')))
    if ($user_id = $dbh->insert_id('users'))
    {
      // create rcube_user instance to make plugin hooks work
      $user_instance = new rcube_user($user_id);
      $rcmail->user = $user_instance;
      $mail_domain = $rcmail->config->mail_domain($host);
      if ($user_email=='')
@@ -383,23 +402,35 @@
      }
      if (empty($email_list))
        $email_list[] = strip_newlines($user_email);
        $email_list[] = strip_newlines($user_email);
      // identities_level check
      if (count($email_list) > 1 && $rcmail->config->get('identities_level', 0) > 1)
      else if (count($email_list) > 1 && $rcmail->config->get('identities_level', 0) > 1)
        $email_list = array($email_list[0]);
      // create new identities records
      $standard = 1;
      foreach ($email_list as $email) {
        $dbh->query(
            "INSERT INTO ".get_table_name('identities')."
              (user_id, del, standard, name, email)
             VALUES (?, 0, ?, ?, ?)",
            $user_id,
            $standard,
            strip_newlines($user_name),
            preg_replace('/^@/', $user . '@', $email));
      foreach ($email_list as $row) {
   $record = array();
        if (is_array($row)) {
     $record = $row;
        }
        else {
          $record['email'] = $row;
        }
   if (empty($record['name']))
     $record['name'] = $user_name;
        $record['name'] = strip_newlines($record['name']);
        $record['user_id'] = $user_id;
        $record['standard'] = $standard;
        $plugin = $rcmail->plugins->exec_hook('create_identity',
     array('login' => true, 'record' => $record));
        if (!$plugin['abort'] && $plugin['record']['email']) {
          $rcmail->user->insert_identity($plugin['record']);
        }
        $standard = 0;
      }
    }
@@ -413,105 +444,42 @@
        'message' => "Failed to create new user"), true, false);
    }
    
    return $user_id ? new rcube_user($user_id) : false;
    return $user_id ? $user_instance : false;
  }
  
  
  /**
   * Resolve username using a virtuser file
   * Resolve username using a virtuser plugins
   *
   * @param string E-mail address to resolve
   * @return string Resolved IMAP username
   */
  static function email2user($email)
  {
    $r = self::findinvirtual('^' . quotemeta($email) . '[[:space:]]');
    $rcmail = rcmail::get_instance();
    $plugin = $rcmail->plugins->exec_hook('email2user',
      array('email' => $email, 'user' => NULL));
    for ($i=0; $i<count($r); $i++)
    {
      $data = trim($r[$i]);
      $arr = preg_split('/\s+/', $data);
      if (count($arr) > 0)
        return trim($arr[count($arr)-1]);
    }
    return NULL;
    return $plugin['user'];
  }
  /**
   * Resolve e-mail address from virtuser file/table
   * Resolve e-mail address from virtuser plugins
   *
   * @param string User name
   * @param boolean If true returns first found entry
   * @param boolean If true returns email as array (email and name for identity)
   * @return mixed Resolved e-mail address string or array of strings
   */
  static function user2email($user, $first=true)
  static function user2email($user, $first=true, $extended=false)
  {
    $result = array();
    $rcmail = rcmail::get_instance();
    $dbh = $rcmail->get_dbh();
    $plugin = $rcmail->plugins->exec_hook('user2email',
      array('email' => NULL, 'user' => $user,
        'first' => $first, 'extended' => $extended));
    // SQL lookup
    if ($virtuser_query = $rcmail->config->get('virtuser_query')) {
      $sql_result = $dbh->query(preg_replace('/%u/', $dbh->escapeSimple($user), $virtuser_query));
      while ($sql_arr = $dbh->fetch_array($sql_result))
        if (strpos($sql_arr[0], '@')) {
          $result[] = $sql_arr[0];
          if ($first)
            return $result[0];
        }
    }
    // File lookup
    $r = self::findinvirtual('[[:space:]]' . quotemeta($user) . '[[:space:]]*$');
    for ($i=0; $i<count($r); $i++)
    {
      $data = $r[$i];
      $arr = preg_split('/\s+/', $data);
      if (count($arr) > 0 && strpos($arr[0], '@'))
      {
        $result[] = trim(str_replace('\\@', '@', $arr[0]));
        if ($first)
          return $result[0];
      }
    }
    return empty($result) ? NULL : $result;
    return empty($plugin['email']) ? NULL : $plugin['email'];
  }
  
  /**
   * Find matches of the given pattern in virtuser file
   *
   * @param string Regular expression to search for
   * @return array Matching entries
   */
  private static function findinvirtual($pattern)
  {
    $result = array();
    $virtual = null;
    if ($virtuser_file = rcmail::get_instance()->config->get('virtuser_file'))
      $virtual = file($virtuser_file);
    if (empty($virtual))
      return $result;
    // check each line for matches
    foreach ($virtual as $line)
    {
      $line = trim($line);
      if (empty($line) || $line{0}=='#')
        continue;
      if (eregi($pattern, $line))
        $result[] = $line;
    }
    return $result;
  }
}