<?php

namespace Drupal\block_visibility_groups;

use Drupal\Core\Form\FormStateInterface;

/**
 * Provides functionality to redirect conditions form to correct location.
 *
 * Either group edit form or block layout form.
 */
trait ConditionRedirectTrait {

  /**
   * Ensure form redirects to the correct route.
   *
   * @param \Drupal\Core\Form\FormStateInterface $form_state
   *   The form state.
   */
  protected function setConditionRedirect(FormStateInterface $form_state): void {
    $redirect = $form_state->getValue('bvg_redirect');
    if ($redirect == 'edit') {
      $form_state->setRedirectUrl($this->block_visibility_group->toUrl('edit-form'));
    }
    elseif ($redirect == 'layout') {
      $query = [
        'block_visibility_group' => $this->block_visibility_group->id(),
        'show_conditions' => 1,
      ];

      $form_state->setRedirect(
        'block.admin_display',
        [],
        ['query' => $query]
      );
    }
  }

  /**
   * Set value for redirect.
   *
   * @param array $form
   *   The form array.
   * @param string|null $redirect
   *   The redirect option: layout, edit, or null.
   */
  protected function setRedirectValue(array &$form, ?string $redirect): void {
    $form['bvg_redirect'] = [
      '#type' => 'value',
      '#value' => $redirect,
    ];
  }

}