<?php

define("IMAP_HTML", 1);
define("IMAP_UID",  2);

/**
 * Description of EMail
 *
 * @author programer19
 */
class Imap{
  
  /** @var string Строка имени сервера imap */
  private $_server;
  /** @var integer Номер порта сервера imap */
  private $_port;
  /** @var string Логин пользователя сервера imap */
  private $_user;
  /** @var string Пароль пользователя сервера imap */
  private $_pass;
  /** @var handle Идентификатор сокета imap */
  private $_socket;
  /** @var integer Номер команды imap для вставки при общении с сервером */
  private $_commandNum = 0;
  /** @var boolean Флаг соединения с сервером */
  private $_connected = false;
  /** @var string Кодировка текста на выходе из класса */
  private $_encoding;
  /** @var string Активная директория imap сервера */
  private $_dir;
  
  /** @var string Результат выполнения последней команды */
  private $_result;
  /** @var string Текст предыдущей ошибки */
  private $_error;
  /** @var string Текст статуса предыдущей команды */
  private $_status;
  
  /**
   * 
   * @param string $server Имя сервера
   * @param integer $port Порт сервера
   * @param string $user Логин пользователя
   * @param string $pass Пароль пользователя
   * @param string $encoding Кодировка текста на выходе из класса
   */
  public function __construct($server, $port, $user, $pass, $encoding = 'UTF-8'){
    $this->_encoding = $encoding;
    $this->connect($server, $port);
    $this->login($user, $pass);
  }
  
  public function __destruct() {
    $this->logout();
    fclose($this->_socket);
  }
  
  /**
   * Функция возвращает текст предыдущей ошибки
   * 
   * @return string Текст ошибки
   */
  public function getLastError(){
    return $this->_error;
  }
  
  /**
   * Функция возвращает текст статуса предыдущей команды
   * 
   * @return string Текст статуса
   */
  public function getLastStatus(){
    return $this->_status;
  }
  
  /**
   * 
   * @param string $dir Переход на сервере imap в указаную директорию
   * @return boolean False при недуачном выполнении, True - при удачном
   */
  public function select($dir){
    $this->_dir = $this->encode7bit($dir);
    return $this->sendCommand("SELECT {$this->_dir}");
  }
  
  /**
   * Функция возвращает тело письма (без вложений)
   * 
   * @param integer $messageNo Номер письма
   * @param integer $flags Флаги-модификаторы
   * IMAP_UID - вместо номера письма будет взят UID
   * IMAP_HTML - Предпочтительно читать text/html разделы, если таковые отсутствуют -
   * будет возвращён текст из разделов с типом text/plain
   * @return string Тело письма. False при неудаче
   */
  public function getBody($messageNo, $flags = 0){
    $res = $this->_getBody($messageNo, $flags);
    return ($res!==false ? $res["body"] : false);
  }
  
  /**
   * Функция возвращает тело письма (без вложений)
   * 
   * @param integer $messageNo Номер письма
   * @param integer $flags Флаги-модификаторы
   * IMAP_UID - вместо номера письма будет взят UID
   * IMAP_HTML - Предпочтительно читать text/html разделы, если таковые отсутствуют -
   * будет возвращён текст из разделов с типом text/plain
   * @param string $part Часть письма, которую надо прочитать (по стандарту RFC-822)
   * @return string Тело нужной части письма. False при неудаче
   */
  private function _getBody($messageNo, $flags = 0, $part = ''){
    $headers = $this->getHeaders($messageNo, $flags & IMAP_UID, $part);
    
    //если не указан тип контента - эта часть нас явно не интересует
    if(empty($headers["content-type"])){
      return false;
    }
    
    //вытягиваю тип и подтип контента
    $contentTypeLines = explode("\n", $headers["content-type"]);
    $contentType = explode(";", str_replace(" ", "", $contentTypeLines[0]));
    $contentType[0] = mb_strtolower($contentType[0], $this->_encoding);
    
    if($contentType[0]!='multipart/mixed' 
    && $contentType[0]!='multipart/alternative'
    && $contentType[0]!='text/plain' 
    && $contentType[0]!='text/html'
    ){
      return false;
    }
    
    //обработка мультипартовых писем
    if($contentType[0]=='multipart/mixed' || $contentType[0]=='multipart/alternative'){
      $subPart = 1;
      
      //заголовки и тип контента мы укажем из данной части письма
      $res["headers"]=$headers;
      $res["contentType"]=$contentType[0];
      
      //а тело письма возьмём из подчастей
      do{
        $rs = $this->_getBody($messageNo, $flags, $part=="" ? strval($subPart) : $part.'.'.strval($subPart));
        if ($rs!==false) $res["body"][$rs["contentType"]] .= $rs["body"];
        $subPart++;
      }while($rs !== false);
      
      if (($flags & IMAP_HTML) && !empty($res["body"]["text/html"])){
        return array(
            "headers"=>$res["headers"],
            "contentType"=>$res["contentType"],
            "body"=>$res["body"]["text/html"]
        );
      }else{
        return array(
            "headers"=>$res["headers"],
            "contentType"=>$res["contentType"],
            "body"=>$res["body"]["text/plain"]
        );
      }
    //обработка обычных текстовых (и html) писем
    }elseif ($contentType[0]=='text/plain' || $contentType[0]=='text/html'){
      
      //читаем нужную часть письма
      $bodyPart = $part=='' ? 'TEXT' : $part;
      if (!$this->sendCommand("FETCH {$messageNo} BODY[$bodyPart]")){
        return false;
      }else{
        
        //вырезаем сервисные строки из ответа
        $body = str_replace("\r\n", "\n", $this->_result);
        $body = preg_replace(array("#^[^\n]*\n#is", "#\n[^\n]*\n$#is"), "", $body);

        //раскодируем тело части письма
        switch($headers["content-transfer-encoding"]){
          case "base64":
            $body = base64_decode($body);
            break;
          case "8bit":
            break;
          case "7bit":
            break;
          case "quoted-printable":
            $body = quoted_printable_decode($body);
            break;
        }

        //переводим тело в нужную кодировку
        preg_match("#charset=\"?([a-zA-Z0-9\\-]+)\"?#is", $contentType[1], $charset);
        $body = mb_convert_encoding($body, $this->_encoding, ($charset[1]=='') ? 'koi8-r' : $charset[1]);

        return array(
            "headers"=>$headers,
            "body"=>$body,
            "contentType"=>$contentType[0]
        );
      }
    }
    
  }
  
  /**
   * Функция возвращает заголовок нужной части письма (без вложений)
   * 
   * @param string $messageNo Номер письма в ящике.
   * доступна комбинаци читать_с:читать_до
   * @param integer $flags Флаги-модификаторы
   * IMAP_UID - вместо номера письма будет взят UID
   * IMAP_HTML - Предпочтительно читать text/html разделы, если таковые отсутствуют -
   * будет возвращён текст из разделов с типом text/plain
   * @param string $part Часть письма, которую надо прочитать (по стандарту RFC-822)
   * @return array Заголовок нужной части письма. False при неудаче или отсутсвии заголовка (части)
   * При считывании массива заголовков каждый набор параметров помещается в элемент массива
   * с номером соответствующего сообщения
   */
  public function getHeaders($num, $flags = 0, $part = '', $field = ""){
    $prefix = ($flags & IMAP_UID)!=0 ? "UID " : "";
    $answer_prefix = ($flags & IMAP_UID)!=0 ? "UID (\\d+) " : "";
    $oneMessage = is_numeric($num);
    
    $partHeader = $part==''   ? "HEADER" : "{$part}.HEADER";
    $partHeader = $field==''  ? $partHeader : "{$partHeader}.FIELDS ({$field})";
    
    $answerPartHeader = preg_quote($partHeader, "#");
    
    if (!$this->sendCommand("{$prefix}FETCH {$num} BODY[{$partHeader}]")){
      return false;
    }else{
      //удаляем из ответа первую и последние строки (стандартное обрамление серверого ответа)
      preg_match_all(
              "#(\\d+) FETCH \\({$answer_prefix}BODY\\[{$answerPartHeader}\\] \\{\\d+\\}[^\n]*\n"
              ."(.*?)\n\n\\)#si", 
              $this->_result,
              $headersMatches,
              PREG_SET_ORDER
      );
              
      $result = array();
      
      foreach($headersMatches as $headersMatch){
        
        $res = array();
        
        $headers = preg_replace(
                "#(\n\t+|\n +)#si", 
                " ", 
                ($flags & IMAP_UID)!=0 ? $headersMatch[3] : $headersMatch[2]
        );

        //распарсиваем заголвоки и заталкиваем в массив
        preg_match_all("#^([^:]+):[ \t]*([^\n]+)$#ims", $headers, $matches, PREG_SET_ORDER);
        foreach($matches as $match){
          $key = mb_strtolower($match[1], $this->_encoding);
          if (!isset($res[$key])){
            $res[$key]=preg_replace("#\n$#si", "", $match[2]);
          }else{
            $res[$key].="\n".preg_replace("#\n$#si", "", $match[2]);
          }
        }
        
        $messageNum = ($flags & IMAP_UID)!=0 ? $headersMatch[2] : $headersMatch[1];
        if(!$oneMessage) $result[$messageNum] = $res;
        
      }
      
      if ($oneMessage) $result = $res;
      
      return $result;
    }
  }
  
  /**
   * 
   * @return integer Количество писем в ящике. False при ошибке выполнения
   */
  public function getMessagesNum(){
    if(!$this->sendCommand("STATUS {$this->_dir} (MESSAGES)")){
      return false;
    }else{
      preg_match("#STATUS \"?{$this->_dir}\"? \\(MESSAGES (\d+)\\)#i", $this->_result, $status);
      return $status[1];
    }
  }
  
  /**
   * Функция возвращает идентификатор письма по его номеру в ящике
   * 
   * @param string $num Номер письма в ящике
   * доступна комбинаци читать_с:читать_до
   * @param integer $flags Флаги-модификаторы
   * IMAP_UID - вместо номера письма будет взят UID
   * @return mixed Возвращает массив, в котором ключ - номер письма в ящике,
   * а значение - идентификатор письма
   * False при ошибке выполнения
   */
  public function getIdByNum($num, $flags = 0){
    $prefix = ($flags & IMAP_UID)!=0 ? "UID " : "";
    $answer_prefix = ($flags & IMAP_UID)!=0 ? "UID (\\d+) " : "";
    
    if(!$this->sendCommand("{$prefix}FETCH {$num} BODY[HEADER.FIELDS (Message-ID)]")){
      return false;
    }else{
      
      preg_match_all(
              "#(\\d+) FETCH \\({$answer_prefix}BODY\\[HEADER\\.FIELDS \\(Message\\-ID\\)\\] \\{\\d+\\}[^\n]*\n"
              ."message\\-id: *<([^>]+)>[^\n]*\n\n"
              ."\\)#is", 
              $this->_result, 
              $idMatches,
              PREG_SET_ORDER
      );
      
      $res = array();
      $keyNum     = ($flags & IMAP_UID)!=0 ? 2 : 1;
      $valueNum   = ($flags & IMAP_UID)!=0 ? 3 : 2;
      
      foreach($idMatches as $idMatch){
        $res[$idMatch[$keyNum]]=$idMatch[$valueNum];
      }
      return $res;
    }
  }
  
  /**
   * Функция возвращает номер письма в ящике по его идентификатору
   * 
   * 
   * @param string $id Идентификатор письма
   * @param integer $flags Флаги-модификаторы
   * IMAP_UID - вместо номера письма будет возвращён UID
   * @return integer Номер письма в ящике.
   * False при неудаче или ошибке выполнения
   */
  public function getNumById($id, $flags = 0){

    if(($flags & IMAP_UID)==0){
      $messagesLastNum = $this->getMessagesNum();
    }else{
      $messagesLastNum = $this->getUID($this->getMessagesNum());
      if($messagesLastNum===false) return false;
    }
    
    $prefix =         ($flags & IMAP_UID)!=0 ? "UID "         : "";
    $answer_prefix =  ($flags & IMAP_UID)!=0 ? "UID (\\d+) "  : "";
    
    if(!$this->sendCommand("{$prefix}FETCH 1:{$messagesLastNum} BODY[HEADER.FIELDS (Message-ID)]")){
      return false;
    }else{
      
      $preg_id = preg_quote($id, "#");
      preg_match(
              "#(\\d+) FETCH \\({$answer_prefix}BODY\\[HEADER\\.FIELDS \\(Message\\-ID\\)\\] \\{\\d+\\}[^\n]*\n"
              ."message\\-id: *<{$preg_id}>[^\n]*\n\n"
              ."\\)#is", 
              $this->_result, 
              $idMatches
      );
      return ($flags & IMAP_UID)!=0 ? $idMatches[2] : $idMatches[1];
    }
  }
  
  /**
   * Функция возвращает UID письма по его номеру в ящике
   * 
   * 
   * @param integer $num Номер письма в ящике
   * @return integer Номер письма в ящике.
   * False при неудаче или ошибке выполнения
   */
  public function getUID($num){
    
    if(!$this->sendCommand("UID SEARCH {$num}")){
      return false;
    }else{
      
      preg_match(
              "#SEARCH (\\d+)#is", 
              $this->_result, 
              $idMatches
      );
      return $idMatches[1];
    }
  }
  public function getNumByUID($uid){
    
    if(!$this->sendCommand("SEARCH UID {$uid}")){
      return false;
    }else{
      
      preg_match(
              "#SEARCH (\\d+)#is", 
              $this->_result, 
              $idMatches
      );
      return $idMatches[1];
    }
  }
  
  /**
   * Функция удаления письма из ящика по его номеру
   * 
   * @param string $num Номер письма в ящике
   * @param integer $flags Флаги-модификаторы
   * IMAP_UID - вместо номера письма будет взят UID
   * @return boolean. False при ошибке. True при удачном выполнении
   */
  public function deleteMessage($num, $flags = 0){
    $prefix = ($flags & IMAP_UID)!=0 ? "UID " : "";
    if(!$this->sendCommand("{$prefix}STORE {$num} +FLAGS (\\Deleted)")){
      return false;
    }else{
      return $this->sendCommand("EXPUNGE");
    }
  }
  public function createMessage($headers, $body){
    $commandData = '';
    foreach($headers as $name=>$value){
      $commandData .= "{$name}: {$value}\r\n";
    }
    $commandData .= "Content-Transfer-Encoding: base64\r\n\r\n";
    
    $commandData .= base64_encode($body)."\r\n";
    $size = mb_strlen($commandData, $this->_encoding)-2;
    if(!$this->sendCommand("append {$this->_dir} (\\Seen) {{$size}}", $commandData)){
      return false;
    }else{
      preg_match("#^\\[APPENDUID (\\d+) (\\d+)\\]#i", $this->getLastStatus(), $statusMatches);
      return $statusMatches[2];
    }
    
  }
  
  /**
   * Возвращает основную (служебную) часть строки для таких значений как
   * from, to, message-id
   * 
   * @param string $str Строка для парсинга
   * @return string Обрезаная строка с служебным значением из начальной строки
   */
  public function getMainPartOfString($str){
    return preg_replace("#^[^<]*<([^>]+)>.*?$#ui", "\\1", $str);
  }
  
  /**
   * Функция кодирует строку в UTF7-IMAP для правильного общения с imap сервером
   * (требуется для русских символов)
   * 
   * @param string $str строка для кодирования
   * @return string Закодированая строка
   */
  public function encode7bit($str){
    return mb_convert_encoding($str, "UTF7-IMAP", $this->_encoding);
  }
  
  /**
   * Функция декодирует строку из UTF7-IMAP для правильного общения с imap сервером
   * (требуется для русских символов)
   * 
   * @param string $str строка для декодирования
   * @return string Декодированая строка
   */
  public function decode7bit($str){
    return mb_convert_encoding($str, $this->_encoding, "UTF7-IMAP");
  }
  
  /**
   * Функция для декодирования не англоязычных заголовков письма таких как
   * тема, от кого письмо (подпись), кому письмо (подпись) и т.д.
   * 
   * @param string $str Строка для декодирования
   * @param string $encoding Кодировка письма
   * @return string Декодированая строка
   */
  public function decodeLine($str, $encoding = "UTF-8"){
    preg_match_all("#=\\?([^\\?]+)\\?([^\\?]+)\\?([^\\?]+)\\?=#uis", $str, $matches, PREG_SET_ORDER);
    if (empty($matches)){
      $res = $this->_encoding!=$encoding ? iconv($encoding, $this->_encoding, $str) : $str;
      return $res;
    }
    
    $res = '';
    foreach($matches as $match){
      $rs = $match[3];
      
      switch($match[2]){
        case 'Q': $rs = quoted_printable_decode($rs);
          break;
        case 'B': $rs = base64_decode($rs);
          break;
      }
      
      if($match[1]!=$this->_encoding) $rs = iconv($match[1], $this->_encoding, $rs);
      $res .= $rs;
    }
    
    return $res;
  }
  
  /**
   * Рекурсивная функция возвращающая текст определённой части письма
   * (ВРЕМЕННО НЕ ВОСТРЕБОВАНА)
   * 
   * @param array $contentType Разбитый по ";" вышестоящий заголовок
   * @param string $body Текст вышестоящей части письма
   * @return array возвращет массив 
   * array(
   *   'text' => письмо в формате text/plain,
   *   'html' => письмо в формате text/html
   * )
   */
  private function getMultiPart($contentType, $body){
    $res = array("text"=>"", "html"=>"");
    
    preg_match("#boundary=\"?([a-zA-Z0-9\\-_=]+)\"?#s", $contentType, $boundary);
    $boundary[1] = preg_quote($boundary[1], "#");
    
    preg_match_all("#\\-\\-{$boundary[1]}\n((?:[^\n]+\n)+)\n(.*?)(?=\n\\-\\-{$boundary[1]})#s", $body, $parts, PREG_SET_ORDER);
    foreach($parts as $part){
      $part[1] = preg_replace("#\n[ \t]+#is", " ", $part[1]);
      
      $subHeaders = array();
      preg_match_all("#^([^:]+): *([^\n]+)$#ims", $part[1], $matches, PREG_SET_ORDER);
      foreach($matches as $match){
        $subHeaders[mb_strtolower($match[1], $this->_encoding)]=$match[2];
      }
      $subContentType = explode(";", str_replace(" ", "", $subHeaders["content-type"]));
      $subContentType[0] = mb_strtolower($subContentType[0], $this->_encoding);
      
      $partBody=$part[2];
      if ($subContentType[0]=="text/plain" || $subContentType[0]=="text/html"){
        switch($subHeaders["content-transfer-encoding"]){
          case "base64":
            $partBody = base64_decode($partBody);
            break;
          case "8bit":
            break;
          case "7bit":
            break;
          case "quoted-printable":
            $partBody = quoted_printable_decode($partBody);
            break;
        }
      }
      
      switch ($subContentType[0]){
        case "multipart/alternative":
          $rs = $this->getMultiPart(implode(" ", $subContentType), $partBody);
          $res['text'] .= $rs['text'];
          $res['html'] .= $rs['html'];
          break;
        case "text/plain":
          preg_match("#charset=\"?([a-zA-Z0-9\\-_]+)\"?#is", implode(" ", $subContentType), $subCharset);
          $res["text"] .= mb_convert_encoding($partBody, $this->_encoding, ($subCharset[1]=='') ? 'koi8-r' : $subCharset[1]);
          break;
        case "text/html":
          preg_match("#charset=\"?([a-zA-Z0-9\\-_]+)\"?#is", implode(" ", $subContentType), $subCharset);
          $res["html"] .= mb_convert_encoding($partBody, $this->_encoding, ($subCharset[1]=='') ? 'koi8-r' : $subCharset[1]);
          break;
      }
    }
    
    return $res;
  }
  
  /**
   * Функция соединения с imap сервером
   * 
   * @param string $server Имя imap сервера
   * @param integer $port Порт imap сервера
   * @return boolean Результат соединения. True - удачно, False - при ошибке
   */
  private function connect($server, $port){
    $this->_socket = fsockopen($server, $port, $errno, $error);
    if(!$this->_socket){
      $this->_error = $error;
      $this->_connected = false;
      return false;
    }else{
      $this->_server = $server;
      $this->_port = $port;
      $this->_connected = true;
      return true;
    }
  }
  
  /**
   * Функция для авторизации на сервере imap
   * 
   * @param string $user Логин пользователя
   * @param string $pass Пароль пользователя
   * @return boolean Результат логина. True - удачно. False - ошибка
   */
  private function login($user, $pass){
    return $this->sendCommand("LOGIN {$user} {$pass}");
  }
  
  /**
   * Функция для разавторизации на сервере imap
   * 
   * @return boolean Результат выполнения. True - удачно. False - оишбка
   */
  private function logout(){
    return $this->sendCommand("LOGOUT");
  }
  
  /**
   * Функция для отправки команды для выполнения на сервере
   * 
   * @param string $command Команда для выполнения
   * @return boolean Результат выполнения команды. True - удачно. False - не удачно
   */
  private function sendCommand($command, $data=''){
    if(!$this->_connected){
      $this->_error = 'You must to connect first';
      return false;
    }else{
      
      $this->_commandNum++;
      $commandNumLength = mb_strlen("{$this->_commandNum}", $this->_encoding);
      fputs($this->_socket, "{$this->_commandNum} {$command}\r\n");
      
      $res = '';
      /*
       * читаем строку с сокета и проверяем начинается ли она с номера отправленной комманды
       * если да - значит это статус выполнения и мы прекращаем чтение и переходим к обработке
       * полученого статуса
       */
      while($line=fgets($this->_socket)){
        if(mb_substr($line, 0, $commandNumLength, $this->_encoding)=="{$this->_commandNum}") break;
        if (mb_substr($line, 0, 2, $this->_encoding)=="* "){
          $line = mb_substr($line, 2, null, $this->_encoding);
        }elseif(mb_substr($line, 0, 1, $this->_encoding)=="+"){
          fputs($this->_socket, $data);
          $line = '';
        }
        $res .= str_replace("\r\n", "\n", $line);
      }
      
      /*
       * удаляем номер комманды с пробелом и читаем статус
       * если статус начинается с OK - значит комманда выполнилась удачно, если с NO или BAD -
       * значит выводим ошибку
       */
      $line = mb_substr($line, $commandNumLength+1, null, $this->_encoding);
      if (mb_substr($line, 0, 2, $this->_encoding)=="OK"){
        $this->_result = $res;
        $this->_status = preg_replace("#^OK #i", "", $line);
        return true;
      }else{
        $this->_error = preg_replace("#^(NO|BAD) #i", "", $line);
        $this->_status = '';
        return false;
      }
    }
  }
  
}