class FormOperations

class FormOperations {
  public function __construct(protected WorkspaceManagerInterface $workspaceManager, #[Autowire('@workspaces.negotiator.query_parameter')] protected QueryParameterWorkspaceNegotiator $queryParameterNegotiator) {
  }
  
  /**
   * Implements hook_form_alter().
   */
  public function formAlter(array &$form, FormStateInterface $form_state, $form_id) : void {
    $active_workspace = $this->workspaceManager
      ->getActiveWorkspace();
    // Ensure that the form's initial workspace (if any) is used for the current
    // request.
    $form_workspace_id = $form_state->getUserInput()['active_workspace_id'] ?? NULL;
    $form_workspace = $form_workspace_id ? Workspace::load($form_workspace_id) : NULL;
    if ($form_workspace && (!$active_workspace || $active_workspace->id() != $form_workspace->id())) {
      $this->workspaceManager
        ->setActiveWorkspace($form_workspace, FALSE);
      $active_workspace = $form_workspace;
    }
    // No alterations are needed if we're not in a workspace context.
    if (!$active_workspace) {
      return;
    }
    // If a form hasn't already been marked as safe or not to submit in a
    // workspace, check the generic interfaces.
    if (!$form_state->has('workspace_safe')) {
      $form_object = $form_state->getFormObject();
      $workspace_safe = $form_object instanceof WorkspaceSafeFormInterface || $form_object instanceof WorkspaceDynamicSafeFormInterface && $form_object->isWorkspaceSafeForm($form, $form_state);
      $form_state->set('workspace_safe', $workspace_safe);
    }
    // Add a validation step for every other form.
    if ($form_state->get('workspace_safe') !== TRUE) {
      $form['workspace_safe'] = [
        '#type' => 'value',
        '#value' => FALSE,
      ];
      $this->addWorkspaceValidation($form);
    }
    else {
      // Persist the active workspace for the entire lifecycle of the form,
      // including AJAX requests.
      $form['active_workspace_id'] = [
        '#type' => 'hidden',
        '#value' => $active_workspace->id(),
      ];
      $url_query_options = $this->queryParameterNegotiator
        ->getQueryOptions($active_workspace->id());
      $this->setAjaxWorkspace($form, $url_query_options + [
        'persist' => FALSE,
      ]);
    }
  }
  
  /**
   * Adds our validation handler recursively on each element of a form.
   *
   * @param array &$element
   *   An associative array containing the structure of the form.
   */
  protected function addWorkspaceValidation(array &$element) : void {
    // Recurse through all children and add our validation handler if needed.
    foreach (Element::children($element) as $key) {
      if (isset($element[$key]) && $element[$key]) {
        $this->addWorkspaceValidation($element[$key]);
      }
    }
    if (isset($element['#submit'])) {
      $element['#validate'][] = [
        static::class,
        'validateDefaultWorkspace',
      ];
      // Ensure that the workspace validation is always shown, even when the
      // form element is limiting validation errors.
      if (isset($element['#limit_validation_errors']) && $element['#limit_validation_errors'] !== FALSE) {
        $element['#limit_validation_errors'][] = [
          'workspace_safe',
        ];
      }
    }
  }
  
  /**
   * Validation handler which sets a validation error for all unsupported forms.
   */
  public static function validateDefaultWorkspace(array &$form, FormStateInterface $form_state) : void {
    if ($form_state->get('workspace_safe') !== TRUE && isset($form_state->getCompleteForm()['workspace_safe'])) {
      $form_state->setErrorByName('workspace_safe', new TranslatableMarkup('This form can only be submitted in the default workspace.'));
    }
  }
  
  /**
   * Ensures that the current workspace is persisted across AJAX interactions.
   *
   * @param array &$element
   *   An associative array containing the structure of the form.
   * @param array $url_query_options
   *   An array of URL query options used by the query parameter workspace
   *   negotiator.
   */
  protected function setAjaxWorkspace(array &$element, array $url_query_options) : void {
    // Recurse through all children if needed.
    foreach (Element::children($element) as $key) {
      if (isset($element[$key]) && $element[$key]) {
        $this->setAjaxWorkspace($element[$key], $url_query_options);
      }
    }
    if (isset($element['#ajax']) && !isset($element['#ajax']['options']['query']['workspace'])) {
      $element['#ajax']['options']['query'] = array_merge_recursive($url_query_options, $element['#ajax']['options']['query'] ?? []);
    }
  }

}