Leveraging the new PostMount in ComponentWithFormTrait!

Author: weaverryan

src/LiveComponent/src/ComponentWithFormTrait.php

@@ -17,6 +17,8 @@
 use Symfony\Component\HttpKernel\Exception\UnprocessableEntityHttpException;
 use Symfony\UX\LiveComponent\Attribute\BeforeReRender;
 use Symfony\UX\LiveComponent\Attribute\LiveProp;
+use Symfony\UX\LiveComponent\Util\ArrayUtil;
+use Symfony\UX\TwigComponent\Attribute\PostMount;
 
 /**
  * @author Ryan Weaver <[email protected]>
@@ -63,23 +65,19 @@ trait ComponentWithFormTrait
      */
     abstract protected function instantiateForm(): FormInterface;
 
-    /**
-     * Override in your class if you need extra mounted values.
-     *
-     * If you override, call $this->initializeFormValues() manually
-     * and also $this->formView = $form if you are passing in a Form.
-     *
-     * TODO: Refactor to PostMount when available.
-     * https://github.com/symfony/ux/pull/220
-     */
-    public function mount(FormView $form = null)
+    #[PostMount]
+    public function postMount(array $data): array
     {
-        if ($form) {
-            $this->formView = $form;
+        // allow the FormView object to be passed into the component() as "form"
+        if (array_key_exists('form', $data)) {
+            $this->formView = $data['form'];
+            unset($data['form']);
         }
 
         // set the formValues from the initial form view's data
         $this->initializeFormValues();
+
+        return $data;
     }
 
     /**