react_comments-8.x-1.0-beta2/src/Model/Comment.php

<?php

namespace Drupal\react_comments\Model;

use Drupal\comment\Entity\Comment as CommentEntity;

class Comment extends CommentBase {

  /**
   * Load an individual comment.
   */
  public function load($show_unpublished = FALSE) {
    $query = \Drupal::database()->select('comment_field_data', 'c');
    $query->join('comment__comment_body', 'b', 'c.cid = b.entity_id');
    $query->fields('c', ['uid', 'pid', 'entity_id', 'subject', 'created', 'changed', 'status', 'hostname', 'name', 'mail'])
      ->fields('b', ['comment_body_value'])
      ->condition('c.cid', $this->getId());
    $record = $query->execute()
      ->fetchObject();

    if (empty($record->comment_body_value)) return NULL;

    $user = new User();
    $user->setId($record->uid)->load();

    if (empty($user->getEmail()) && !empty($record->mail)) {
      $user->setEmail($record->mail);
    }

    if ($status = $this->loadCustomStatus()) {
      $this->setStatus($status);
    }
    else {
      $this->setStatus($record->status);
    }

    $this->setReplyId($record->pid)
      ->setEntityId($record->entity_id)
      ->setUser($user)
      ->setName($user->getName() ?: $record->name)
      ->setEmail($record->mail)
      ->setComment($record->comment_body_value)
      ->setIpAddress($record->hostname)
      ->setCreatedAt($record->created)
      ->setChangedAt($record->changed)
      ->setPublishedStatus($record->status);

    if (\Drupal::service('module_handler')->moduleExists('comment_notify')) {
      \Drupal::moduleHandler()->loadInclude('comment_notify', 'inc', 'comment_notify');

      $notify = comment_notify_get_notification_type($this->getId());
      if (is_string($notify)) {
        $this->setNotify(intval($notify));
      }
    }

    $replies = $this->loadReplies($this->getId(), $show_unpublished);
    $this->setReplies($replies);

    return $this;
  }

  /**
   * Save a comment.
   */
  public function save() {
    $comment = CommentEntity::create([
      // @todo entity_type shouldn't be hardcoded to node
      'entity_type'    => 'node',
      'entity_id'      => $this->getEntityId(),
      'pid'            => $this->getReplyId() ?: null,
      'field_name'     => $this->getFieldName(),
      'uid'            => $this->getUser()->getId(),
      'comment_type'   => $this->getCommentType(),
      'subject'        => $this->getSubject(),
      'comment_body'   => [
        'value'  => $this->getComment(),
        'format' => 'basic_html'
      ],
      'status'         => $this->getStatus(),
      'published'      => $this->isPublished(),
      'name'           => $this->getName(),
      'mail'          => $this->getEmail()
    ]);
    $comment->save();

    if (\Drupal::service('module_handler')->moduleExists('comment_notify')) {
      \Drupal::moduleHandler()->loadInclude('comment_notify', 'inc', 'comment_notify');

      $user = \Drupal::currentUser();
      $user_settings = \Drupal::service('comment_notify.user_settings');
      $notify = $this->getNotify();

      $status = $user_settings->getSetting($user->id(), 'comment_notify');
      if (!is_null($notify)) {
        $status = $notify > 0 ? $notify : COMMENT_NOTIFY_DISABLED;
        if (!$user->isAnonymous()) {
          $user_settings->saveSettings($user->id(), NULL, $status);
        }
      }

      $hostname = !$comment->getHostname() ? $comment->getHostname() : (isset($user->hostname) ? $user->hostname : '');
      $notify_hash = \Drupal::csrfToken()->get($hostname . $comment->id());
      comment_notify_add_notification($comment->id(), $status, $notify_hash, $comment->notified);
    }

    $this->setId($comment->id())
      ->updateCustomStatus();
    return $this->getId();
  }

  /**
   * Update a comment.
   */
  public function update() {
    // @todo consider paring this whole class way down and relying more on core Comment entity.
    $status =  $this->getStatus();
    if (is_numeric($status)) {
      $comment = CommentEntity::load($this->getId());

      $comment->set('pid',        $this->getReplyId() ?: null);
      $comment->set('entity_id',  $this->getEntityId());
      $comment->set('uid',        $this->getUser()->getId());
      $comment->set('subject',    $this->getSubject());
      $this->updateCustomStatus();
      if (is_numeric($this->customStatusToDrupalStatus($this->getStatus()))) {
        $comment->set('status', $this->customStatusToDrupalStatus($this->getStatus()));
      }
      else {
        $comment->set('status', $comment->isPublished());
      }
      $comment->set('comment_body', [
        'value'  => $this->getComment(),
        'format' => 'basic_html'
      ]);

      if (\Drupal::service('module_handler')->moduleExists('comment_notify')) {
        \Drupal::moduleHandler()->loadInclude('comment_notify', 'inc', 'comment_notify');

        $user = \Drupal::currentUser();
        $user_settings = \Drupal::service('comment_notify.user_settings');
        $notify = $this->getNotify();

        $status = $user_settings->getSetting($user->id(), 'comment_notify');
        if (!is_null($notify)) {
          $status = $notify > 0 ? $notify : COMMENT_NOTIFY_DISABLED;
          if (!$user->isAnonymous()) {
            $user_settings->saveSettings($user->id(), NULL, $status);
          }
        }

        comment_notify_update_notification($comment->id(), $status);
      }

      return $comment->save();
    }
    return NULL;
  }

 /**
  * Delete a comment.
  */
 public function delete() {
   $comment = CommentEntity::load($this->getId());
   $user = $this->getUser();
   if ( $this->isAdmin() || ($user->hasPermission('edit own comments') && ($comment->get('uid')->target_id == $user->getId())) ) {
     if (\Drupal::config('react_comments.settings')->get('full_delete')) {
       $comment->delete();
     }
     else {
       $this->setStatus( RC_COMMENT_DELETED )->update();
     }
   }
   return NULL;
 }

  /**
   * Update additional comment status.
   */
  private function loadCustomStatus() {
    $query = \Drupal::database()->select('react_comments_status', 's');
    $query->fields('s', ['status'])
      ->condition('s.cid', $this->getId());
    return $query->execute()
      ->fetchField();
  }

  /**
   * Update additional comment status.
   */
  private function updateCustomStatus() {
    \Drupal::database()->merge('react_comments_status')
      ->key(['cid' => $this->getId()])
      ->fields([
        'cid'        => $this->getId(),
        'uid'        => $this->getUser()->getId(),
        'status'     => $this->getStatus(),
        'created_at' => time(),
      ])
      ->execute();
  }

  /**
   * Load a comment with its replies recursively.
   */
  private function loadReplies($pid = 0, $show_unpublished = FALSE) {
    $query = \Drupal::database()->select('comment_field_data', 'c')
      ->fields('c')
      ->condition('c.pid', $pid)
      ->orderBy('c.created')
      ->addTag('react_comments_load_replies');

    if (!$show_unpublished) {
      $query->condition('c.status', 1);
    }

    $result = $query->execute();

    $thread = [];
    foreach ($result as $item) {
      if ($comment = (new Comment())->setId($item->cid)->load($show_unpublished)) {
        $thread[] = $comment->model();
        $this->loadReplies($item->cid, $show_unpublished);
      }
    }
    return $thread;
  }

  /**
   *  Convert react comments custom status into drupal published status.
   *  Use with is_numeric to decide whether to change the drupal status.
   */
  private function customStatusToDrupalStatus($custom_status) {
    switch($custom_status) {
      case RC_COMMENT_PUBLISHED:
      case RC_COMMENT_DELETED:
        return 1;
      case RC_COMMENT_UNPUBLISHED:
        return 0;
      case RC_COMMENT_FLAGGED:
      default:
        // changing to flagged state shouldn't effect whether the comment is published or unpublished
        return NULL;
    }
  }

}