message_thread-8.x-1.x-dev/src/MessageThreadViewBuilder.php

<?php

namespace Drupal\message_thread;

use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Entity\EntityViewBuilder;

/**
 * Render controller for Messages.
 */
class MessageThreadViewBuilder extends EntityViewBuilder {

  /**
   * {@inheritdoc}
   */
  public function view(EntityInterface $entity, $view_mode = 'full', $langcode = NULL) {
    $build = parent::view($entity, $view_mode, $langcode);

    if (!$langcode) {
      $langcode = \Drupal::languageManager()->getCurrentLanguage()->getId();
    }
    else {
      if (\Drupal::moduleHandler()->moduleExists('config_translation') && !isset($partials[$langcode])) {
        $langcode = \Drupal::languageManager()->getCurrentLanguage()->getId();
      }
    }

    // Load the partials in the correct language.
    /* @var \Drupal\message\Entity\Message $entity  */
    if ($langcode) {
      $entity->setLanguage($langcode);
    }

    $extra = '';

    // Get the partials the user selected for the current view mode.
    $extra_fields = \Drupal::service('entity_display.repository')->getViewDisplay('message', $entity->bundle(), $view_mode);
    foreach ($extra_fields->getComponents() as $field_name => $settings) {
      $display = $this->getSingleFieldDisplay($entity, $field_name, $settings);
      $build += $display->build($entity);
    }

    $build['#markup'] = $extra;

    return $build;
  }

}