trait WorkspaceSafeFormTrait

trait WorkspaceSafeFormTrait {
    use EntityWorkspaceSafeFormTrait;
    
    /**
     * Determines whether the current form is safe to be submitted in a workspace.
     *
     * @param array $form
     *   An associative array containing the structure of the form.
     * @param \Drupal\Core\Form\FormStateInterface $form_state
     *   The current state of the form.
     *
     * @return bool
     *   TRUE if the form is workspace-safe, FALSE otherwise.
     */
    public function isWorkspaceSafeForm(array $form, FormStateInterface $form_state) : bool {
        $section_storage = $this->sectionStorage ?: $this->getSectionStorageFromFormState($form_state);
        if ($section_storage) {
            $context_definitions = $section_storage->getContextDefinitions();
            if (!empty($context_definitions['entity'])) {
                
                /** @var \Drupal\Core\Entity\EntityInterface $entity */
                $entity = $section_storage->getContextValue('entity');
                return $this->isWorkspaceSafeEntity($entity);
            }
        }
        return FALSE;
    }
    
    /**
     * Retrieves the section storage from a form state object, if it exists.
     *
     * @param \Drupal\Core\Form\FormStateInterface $form_state
     *   The form state object.
     *
     * @return \Drupal\layout_builder\SectionStorageInterface|null
     *   The section storage or NULL if it doesn't exist.
     */
    protected function getSectionStorageFromFormState(FormStateInterface $form_state) : ?SectionStorageInterface {
        foreach ($form_state->getBuildInfo()['args'] as $argument) {
            if ($argument instanceof SectionStorageInterface) {
                return $argument;
            }
        }
        return NULL;
    }

}