PluginDeletedMessageTable.class.php

<?php

/**
 * This file is part of the OpenPNE package.
 * (c) OpenPNE Project (http://www.openpne.jp/)
 *
 * For the full copyright and license information, please view the LICENSE
 * file and the NOTICE file that were distributed with this source code.
 */

class PluginDeletedMessageTable extends Doctrine_Table
{
 /**
  * add delete message query
  *
  * @param Doctrine_Query $q
  * @param integer $memberId
  */
  public function addDeleteMessageQuery(Doctrine_Query $q, $memberId = null)
  {
    if (is_null($memberId))
    {
      $memberId = sfContext::getInstance()->getUser()->getMemberId();
    }

    $q = $q->where('member_id = ?', $memberId)
      ->andWhere('is_deleted = ?', false);

    return $q;
  }

  /**
   * 削除済みメッセージ一覧
   * @param integer $memberId
   * @param integer $page
   * @param integer $size
   * @return DeletedMessage object（の配列）
   */
  public function getDeletedMessagePager($memberId = null, $page = 1, $size = 20)
  {
    $q = $this->addDeleteMessageQuery($this->createQuery(), $memberId);
    $q->orderBy('created_at DESC');
    $pager = new sfDoctrinePager('DeletedMessage', $size);
    $pager->setQuery($q);
    $pager->setPage($page);
    $pager->init();

    return $pager;
  }

  /**
   * message_idから削除済みメッセージを取得する
   * @param $member_id
   * @param $message_id
   * @return DeletedMessage
   */
  public function getDeletedMessageByMessageId($member_id, $message_id)
  {
    $q = $this->createQuery()
      ->where('member_id = ?', $member_id)
      ->addwhere('message_id = ?', $message_id);
    $obj = $q->fetchOne();

    if (!$obj) {
      return null;
    }
    return $obj;
  }

  /**
   * message_send_list_idから削除済みメッセージを取得する
   * @param $member_id
   * @param $message_send_list_id
   * @return DeletedMessage
   */
  public function getDeletedMessageByMessageSendListId($member_id, $message_send_list_id)
  {
    $q = $this->createQuery()
      ->where('member_id = ?', $member_id)
      ->addwhere('message_send_list_id = ?', $message_send_list_id);
    $obj = $q->fetchOne();

    if (!$obj) {
      return null;
    }
    return $obj;
  }

  /**
   * delete message 
   * @param int $member_id
   * @param int $message_id
   * @param str $object_name
   * @return boolean 
   */
  public function deleteMessage($member_id, $message_id, $object_name)
  {
    if ($object_name == 'MessageSendList') {
        $message = Doctrine::getTable('MessageSendList')->findOneByIdAndMemberId($message_id, $member_id);
        $deleted_message = $this->getDeletedMessageByMessageSendListId($member_id, $message_id);
        if (!$deleted_message) {
          $deleted_message = new DeletedMessage();
        }
        $deleted_message->setMessageSendListId($message_id);
      } elseif ($object_name == 'SendMessageData') {
        $message = Doctrine::getTable('SendMessageData')->findOneByIdAndMemberId($message_id, $member_id);
        $deleted_message = $this->getDeletedMessageByMessageId($member_id, $message_id);
        if (!$deleted_message) {
          $deleted_message = new DeletedMessage();
        }
        $deleted_message->setMessageId($message_id);
      } elseif ($object_name == 'DeletedMessage') {
        $message = $this->findOneByIdAndMemberId($message_id, $member_id);
        $deleted_message = null;
      }
      if (!$message) {
        return false;
      }
      if ($deleted_message) {
        $deleted_message->setMemberId($member_id);
        $deleted_message->save();
      }
      $message->setIsDeleted(1);
      /* @todo 完全削除の場合ファイルも削除すべきかも */
      $message->save();
      return true;
  }

  /**
   * restore message 
   * @param int $message_id
   * @return boolean 
   */
  public function restoreMessage($message_id, $member_id = null)
  {
    $member_id = $member_id ? $member_id : sfContext::getInstance()->getUser()->getMemberId();
    $deleted_message = $this->findOneByIdAndMemberId($message_id, $member_id);
    if (!$deleted_message) {
      return false;
    }
    if ($deleted_message->getMessageSendListId() != 0) {
        $message = Doctrine::getTable('MessageSendList')->find($deleted_message->getMessageSendListId());
     } else if ($deleted_message->getMessageId() != 0) {
        $message = Doctrine::getTable('SendMessageData')->find($deleted_message->getMessageId());
    }
    if (!$message) {
      return false;
    }
    $deleted_message->delete();
    $message->setIsDeleted(0);
    $message->save();
   return true;
  }

  public function getPreviousSendMessageData(SendMessageData $message, $myMemberId)
  {
    $q = $this->addDeleteMessageQuery($this->createQuery(), $myMemberId);
    $q->andWhere('message_id < ?', $message->id)
      ->orderBy('message_id DESC');

    $deleted = $q->fetchOne();
    if ($deleted && $deleted->message_id)
    {
      return Doctrine::getTable('SendMessageData')->find($deleted->message_id);
    }

    return false;
  }

  public function getNextSendMessageData(SendMessageData $message, $myMemberId)
  {
    $q = $this->addDeleteMessageQuery($this->createQuery(), $myMemberId);
    $q->andWhere('message_id > ?', $message->id)
      ->orderBy('message_id ASC');

    $deleted = $q->fetchOne();
    if ($deleted && $deleted->message_id)
    {
      return Doctrine::getTable('SendMessageData')->find($deleted->message_id);
    }

    return false;
  }
}