Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\contact;
Chris@0: 
Chris@0: use Drupal\Core\Entity\EntityInterface;
Chris@0: use Drupal\Core\Entity\EntityViewBuilder;
Chris@0: use Drupal\Core\Mail\MailFormatHelper;
Chris@0: use Drupal\Core\Render\Element;
Chris@0: 
Chris@0: /**
Chris@0:  * Render controller for contact messages.
Chris@0:  */
Chris@0: class MessageViewBuilder extends EntityViewBuilder {
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function getBuildDefaults(EntityInterface $entity, $view_mode) {
Chris@0:     $build = parent::getBuildDefaults($entity, $view_mode);
Chris@0:     // The message fields are individually rendered into email templates, so
Chris@0:     // the entity has no template itself.
Chris@0:     unset($build['#theme']);
Chris@0:     return $build;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function view(EntityInterface $entity, $view_mode = 'full', $langcode = NULL) {
Chris@0:     $build = parent::view($entity, $view_mode, $langcode);
Chris@0: 
Chris@0:     if ($view_mode == 'mail') {
Chris@0:       // Convert field labels into headings.
Chris@0:       // @todo Improve \Drupal\Core\Mail\MailFormatHelper::htmlToText() to
Chris@0:       // convert DIVs correctly.
Chris@0:       foreach (Element::children($build) as $key) {
Chris@0:         if (isset($build[$key]['#label_display']) && $build[$key]['#label_display'] == 'above') {
Chris@0:           $build[$key] += ['#prefix' => ''];
Chris@0:           $build[$key]['#prefix'] = $build[$key]['#title'] . ":\n";
Chris@0:           $build[$key]['#label_display'] = 'hidden';
Chris@0:         }
Chris@0:       }
Chris@0:       $build['#post_render'][] = function ($html, array $elements) {
Chris@0:         return MailFormatHelper::htmlToText($html);
Chris@0:       };
Chris@0:     }
Chris@0:     return $build;
Chris@0:   }
Chris@0: 
Chris@0: }