| @@ -45,8 +45,9 @@ | ||
| 45 | 45 | add_action( 'elementor/document/after_save', fn( Document $document, array $data ) => $this->set_component_overridable_props( $document, $data ), 10, 2 ); |
| 46 | 46 | add_filter( 'elementor/global_classes/additional_post_types', fn( $post_types ) => array_merge( $post_types, [ Component_Document::TYPE ] ) ); |
| 47 | 47 | |
| 48 | 48 | add_action( 'elementor/atomic-widgets/settings/transformers/register', fn ( $transformers ) => $this->register_settings_transformers( $transformers ) ); |
| 49 | + add_action( 'elementor/document/after_migrate', fn( Document $document, array $data ) => $this->after_component_migrate( $document, $data ), 10, 2 ); | |
| 49 | 50 | |
| 50 | 51 | ( Component_Lock_Manager::get_instance()->register_hooks() ); |
| 51 | 52 | ( new Component_Styles() )->register_hooks(); |
| 52 | 53 | ( new Components_REST_API() )->register_hooks(); |
| @@ -142,6 +143,14 @@ | ||
| 142 | 143 | private function register_settings_transformers( Transformers_Registry $transformers ) { |
| 143 | 144 | $transformers->register( Component_Instance_Prop_Type::get_key(), new Component_Instance_Transformer() ); |
| 144 | 145 | $transformers->register( Overridable_Prop_Type::get_key(), new Overridable_Transformer() ); |
| 145 | 146 | $transformers->register( Override_Prop_Type::get_key(), new Override_Transformer() ); |
| 147 | + } | |
| 148 | + | |
| 149 | + private function after_component_migrate( Document $document, array $data ) { | |
| 150 | + if ( ! $document instanceof Component_Document ) { | |
| 151 | + return; | |
| 152 | + } | |
| 153 | + | |
| 154 | + $document->align_overridable_props_with_elements(); | |
| 146 | 155 | } |
| 147 | 156 | } |