trait ConditionRedirectTrait in Block Visibility Groups 8

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) {
    $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 $redirect
   *   The redirect option: layout, edit, or null.
   */
  protected function setRedirectValue(array &$form, $redirect) {
    $form['bvg_redirect'] = [
      '#type' => 'value',
      '#value' => $redirect,
    ];
  }

}