Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\Core\Ajax;
Chris@0: 
Chris@0: use Drupal\Core\Asset\AttachedAssets;
Chris@0: 
Chris@0: /**
Chris@0:  * Trait for Ajax commands that render content and attach assets.
Chris@0:  *
Chris@0:  * @ingroup ajax
Chris@0:  */
Chris@0: trait CommandWithAttachedAssetsTrait {
Chris@0: 
Chris@0:   /**
Chris@0:    * The attached assets for this Ajax command.
Chris@0:    *
Chris@0:    * @var \Drupal\Core\Asset\AttachedAssets
Chris@0:    */
Chris@0:   protected $attachedAssets;
Chris@0: 
Chris@0:   /**
Chris@0:    * Processes the content for output.
Chris@0:    *
Chris@0:    * If content is a render array, it may contain attached assets to be
Chris@0:    * processed.
Chris@0:    *
Chris@0:    * @return string|\Drupal\Component\Render\MarkupInterface
Chris@0:    *   HTML rendered content.
Chris@0:    */
Chris@0:   protected function getRenderedContent() {
Chris@0:     $this->attachedAssets = new AttachedAssets();
Chris@0:     if (is_array($this->content)) {
Chris@0:       $html = \Drupal::service('renderer')->renderRoot($this->content);
Chris@0:       $this->attachedAssets = AttachedAssets::createFromRenderArray($this->content);
Chris@0:       return $html;
Chris@0:     }
Chris@0:     else {
Chris@0:       return $this->content;
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Gets the attached assets.
Chris@0:    *
Chris@0:    * @return \Drupal\Core\Asset\AttachedAssets|null
Chris@0:    *   The attached assets for this command.
Chris@0:    */
Chris@0:   public function getAttachedAssets() {
Chris@0:     return $this->attachedAssets;
Chris@0:   }
Chris@0: 
Chris@0: }