trait LayoutBuilderEntityFormTrait

trait LayoutBuilderEntityFormTrait {
  use PreviewToggleTrait;

  /**
   * {@inheritdoc}
   */
  public function getBaseFormId() : string {
    return $this
      ->getEntity()
      ->getEntityTypeId() . '_layout_builder_form';
  }

  /**
   * Build the message container.
   *
   * @param \Drupal\Core\StringTranslation\TranslatableMarkup $message
   *   The message to display.
   * @param string $type
   *   The form type this is being attached to.
   *
   * @return array
   *   The render array.
   */
  protected function buildMessageContainer(TranslatableMarkup $message, string $type) : array {
    return [
      '#type' => 'container',
      '#attributes' => [
        'class' => [
          'layout-builder__message',
          sprintf('layout-builder__message--%s', $type),
        ],
      ],
      'message' => [
        '#theme' => 'status_messages',
        '#message_list' => [
          'status' => [
            $message,
          ],
        ],
        '#status_headings' => [
          'status' => $this
            ->t('Status message'),
        ],
      ],
      '#weight' => -900,
    ];
  }

  /**
   * Form submission handler.
   */
  public function redirectOnSubmit(array $form, FormStateInterface $form_state) {
    $form_state
      ->setRedirectUrl($this->sectionStorage
      ->getLayoutBuilderUrl($form_state
      ->getTriggeringElement()['#redirect']));
  }

  /**
   * Retrieves the section storage object.
   *
   * @return \Drupal\layout_builder\SectionStorageInterface
   *   The section storage for the current form.
   */
  public function getSectionStorage() : SectionStorageInterface {
    return $this->sectionStorage;
  }

  /**
   * Builds the actions for the form.
   *
   * @param array $actions
   *   The actions array to modify.
   *
   * @return array
   *   The modified actions array.
   */
  protected function buildActions(array $actions) : array {
    $actions['#attributes']['role'] = 'region';
    $actions['#attributes']['aria-label'] = $this
      ->t('Layout Builder tools');
    $actions['submit']['#value'] = $this
      ->t('Save layout');
    $actions['#weight'] = -1000;
    $actions['discard_changes'] = [
      '#type' => 'submit',
      '#value' => $this
        ->t('Discard changes'),
      '#submit' => [
        '::redirectOnSubmit',
      ],
      '#redirect' => 'discard_changes',
    ];
    $actions['preview_toggle'] = $this
      ->buildContentPreviewToggle();
    return $actions;
  }

  /**
   * Performs tasks that are needed during the save process.
   *
   * @param \Drupal\Core\Form\FormStateInterface $formState
   *   The form state.
   * @param \Drupal\Core\StringTranslation\TranslatableMarkup $message
   *   The message to display.
   */
  protected function saveTasks(FormStateInterface $formState, TranslatableMarkup $message) : void {
    $this->layoutTempstoreRepository
      ->delete($this
      ->getSectionStorage());
    $this
      ->messenger()
      ->addStatus($message);
    $formState
      ->setRedirectUrl($this
      ->getSectionStorage()
      ->getRedirectUrl());
  }

}