Chris@14: <?php
Chris@14: 
Chris@14: namespace Drupal\layout_builder\Controller;
Chris@14: 
Chris@17: use Drupal\Core\Ajax\AjaxHelperTrait;
Chris@14: use Drupal\Core\DependencyInjection\ContainerInjectionInterface;
Chris@14: use Drupal\layout_builder\LayoutTempstoreRepositoryInterface;
Chris@14: use Drupal\layout_builder\Section;
Chris@14: use Drupal\layout_builder\SectionStorageInterface;
Chris@14: use Symfony\Component\DependencyInjection\ContainerInterface;
Chris@14: use Symfony\Component\HttpFoundation\RedirectResponse;
Chris@14: 
Chris@14: /**
Chris@14:  * Defines a controller to add a new section.
Chris@14:  *
Chris@14:  * @internal
Chris@18:  *   Controller classes are internal.
Chris@14:  */
Chris@14: class AddSectionController implements ContainerInjectionInterface {
Chris@14: 
Chris@14:   use AjaxHelperTrait;
Chris@14:   use LayoutRebuildTrait;
Chris@14: 
Chris@14:   /**
Chris@14:    * The layout tempstore repository.
Chris@14:    *
Chris@14:    * @var \Drupal\layout_builder\LayoutTempstoreRepositoryInterface
Chris@14:    */
Chris@14:   protected $layoutTempstoreRepository;
Chris@14: 
Chris@14:   /**
Chris@14:    * AddSectionController constructor.
Chris@14:    *
Chris@14:    * @param \Drupal\layout_builder\LayoutTempstoreRepositoryInterface $layout_tempstore_repository
Chris@14:    *   The layout tempstore repository.
Chris@14:    */
Chris@18:   public function __construct(LayoutTempstoreRepositoryInterface $layout_tempstore_repository) {
Chris@14:     $this->layoutTempstoreRepository = $layout_tempstore_repository;
Chris@14:   }
Chris@14: 
Chris@14:   /**
Chris@14:    * {@inheritdoc}
Chris@14:    */
Chris@14:   public static function create(ContainerInterface $container) {
Chris@14:     return new static(
Chris@18:       $container->get('layout_builder.tempstore_repository')
Chris@14:     );
Chris@14:   }
Chris@14: 
Chris@14:   /**
Chris@14:    * Adds the new section.
Chris@14:    *
Chris@14:    * @param \Drupal\layout_builder\SectionStorageInterface $section_storage
Chris@14:    *   The section storage.
Chris@14:    * @param int $delta
Chris@14:    *   The delta of the section to splice.
Chris@14:    * @param string $plugin_id
Chris@14:    *   The plugin ID of the layout to add.
Chris@14:    *
Chris@14:    * @return \Symfony\Component\HttpFoundation\Response
Chris@14:    *   The controller response.
Chris@14:    */
Chris@14:   public function build(SectionStorageInterface $section_storage, $delta, $plugin_id) {
Chris@14:     $section_storage->insertSection($delta, new Section($plugin_id));
Chris@14: 
Chris@14:     $this->layoutTempstoreRepository->set($section_storage);
Chris@14: 
Chris@14:     if ($this->isAjax()) {
Chris@14:       return $this->rebuildAndClose($section_storage);
Chris@14:     }
Chris@14:     else {
Chris@14:       $url = $section_storage->getLayoutBuilderUrl();
Chris@14:       return new RedirectResponse($url->setAbsolute()->toString());
Chris@14:     }
Chris@14:   }
Chris@14: 
Chris@14: }