v3.33.1

modules/theme-builder/import-export-customization/export.php

@@ -0,0 +1,87 @@
+<?php
+namespace ElementorPro\Modules\ThemeBuilder\ImportExportCustomization;
+
+use Elementor\Core\Base\Document;
+use Elementor\TemplateLibrary\Source_Local;
+use ElementorPro\Modules\ThemeBuilder\Documents\Theme_Document;
+use ElementorPro\Plugin;
+
+if ( ! defined( 'ABSPATH' ) ) {
+	exit; // Exit if accessed directly
+}
+
+class Export {
+
+	public function add_theme_builder_to_export( $export_data, $data, $customization ) {
+		if ( ! isset( $customization['themeBuilder']['enabled'] ) || ! $customization['themeBuilder']['enabled'] ) {
+			return $export_data;
+		}
+
+		$theme_builder_data = $this->export_theme_builder_templates();
+
+		if ( empty( $theme_builder_data['files'] ) ) {
+			return $export_data;
+		}
+
+		$export_data['files'] = array_merge( $export_data['files'], $theme_builder_data['files'] );
+
+		if ( ! empty( $theme_builder_data['manifest'] ) ) {
+			$export_data['manifest'][0]['templates'] = ( $export_data['manifest'][0]['templates'] ?? [] ) + $theme_builder_data['manifest'];
+		}
+
+		return $export_data;
+	}
+
+	private function export_theme_builder_templates() {
+		$theme_builder_types = array_keys( Plugin::elementor()->documents->get_document_types( [
+			'is_editable' => true,
+			'export_group' => Theme_Document::EXPORT_GROUP,
+		] ) );
+
+		if ( empty( $theme_builder_types ) ) {
+			return [
+				'files' => [],
+				'manifest' => [],
+			];
+		}
+
+		$query_args = [
+			'post_type' => Source_Local::CPT,
+			'post_status' => 'publish',
+			'posts_per_page' => -1,
+			'meta_query' => [
+				[
+					'key' => Document::TYPE_META_KEY,
+					'value' => $theme_builder_types,
+				],
+			],
+		];
+
+		$theme_builder_query = new \WP_Query( $query_args );
+
+		$manifest_data = [];
+		$files = [];
+
+		foreach ( $theme_builder_query->posts as $template_post ) {
+			$template_id = $template_post->ID;
+
+			$template_document = Plugin::elementor()->documents->get( $template_id );
+
+			if ( ! $template_document || ! $template_document instanceof Theme_Document ) {
+				continue;
+			}
+
+			$manifest_data[ $template_id ] = $template_document->get_export_summary();
+
+			$files[] = [
+				'path' => 'templates/' . $template_id,
+				'data' => $template_document->get_export_data(),
+			];
+		}
+
+		return [
+			'files' => $files,
+			'manifest' => $manifest_data,
+		];
+	}
+}

modules/theme-builder/import-export-customization/import.php

@@ -0,0 +1,154 @@
+<?php
+namespace ElementorPro\Modules\ThemeBuilder\ImportExportCustomization;
+
+use Elementor\Core\Base\Document;
+use Elementor\TemplateLibrary\Source_Local;
+use Elementor\App\Modules\ImportExportCustomization\Utils as ImportExportCustomizationUtils;
+use ElementorPro\Modules\ThemeBuilder\Documents\Theme_Document;
+use ElementorPro\Plugin;
+
+if ( ! defined( 'ABSPATH' ) ) {
+	exit; // Exit if accessed directly
+}
+
+class Import {
+
+	private $templates_conditions = [];
+
+	public function add_theme_builder_to_import( $result, $data, $customization, $runner ) {
+		if ( $customization && ( ! isset( $customization['themeBuilder']['enabled'] ) || ! $customization['themeBuilder']['enabled'] ) ) {
+			return $result;
+		}
+
+		$theme_builder_result = $this->import_theme_builder_templates( $data, $runner );
+
+		if ( ! empty( $theme_builder_result ) ) {
+			$result['templates']['succeed'] = $this->array_merge_with_key_preservation( $result['templates']['succeed'], $theme_builder_result['succeed'] );
+			$result['templates']['failed'] = $this->array_merge_with_key_preservation( $result['templates']['failed'], $theme_builder_result['failed'] );
+
+			foreach ( $theme_builder_result['succeed_summary'] as $doc_type => $count ) {
+				$result['templates']['succeed_summary'][ $doc_type ] = ( $result['templates']['succeed_summary'][ $doc_type ] ?? 0 ) + $count;
+			}
+		}
+
+		return $result;
+	}
+
+	private function import_theme_builder_templates( $data, $runner ) {
+		$path = $data['extracted_directory_path'] . 'templates/';
+		$templates = $data['manifest']['templates'];
+
+		$result = [
+			'succeed' => [],
+			'failed' => [],
+			'succeed_summary' => [],
+		];
+
+		$theme_builder_types = array_keys( Plugin::elementor()->documents->get_document_types( [
+			'is_editable' => true,
+			'export_group' => Theme_Document::EXPORT_GROUP,
+		] ) );
+
+		foreach ( $templates as $id => $template_settings ) {
+			if ( ! in_array( $template_settings['doc_type'], $theme_builder_types, true ) ) {
+				continue;
+			}
+
+			try {
+				$template_data = ImportExportCustomizationUtils::read_json_file( $path . $id );
+				$imported_id = $this->import_theme_builder_template( $id, $template_settings, $template_data, $data );
+
+				$result['succeed'][ $id ] = $imported_id;
+				$result['succeed_summary'][ $template_settings['doc_type'] ] = ( $result['succeed_summary'][ $template_settings['doc_type'] ] ?? 0 ) + 1;
+			} catch ( \Exception $error ) {
+				$result['failed'][ $id ] = $error->getMessage();
+			}
+		}
+
+		if ( ! empty( $this->templates_conditions ) ) {
+			$runner->add_import_session_metadata( 'template_conditions', $this->templates_conditions );
+			$this->templates_conditions = [];
+		}
+
+		return $result;
+	}
+
+	private function import_theme_builder_template( $id, array $template_settings, array $template_data, array $import_data ) {
+		$new_document = Plugin::elementor()->documents->create(
+			$template_settings['doc_type'],
+			[
+				'post_title' => $template_settings['title'],
+				'post_type' => Source_Local::CPT,
+				'post_status' => 'publish',
+			]
+		);
+
+		if ( is_wp_error( $new_document ) ) {
+			throw new \Exception( esc_html( $new_document->get_error_message() ) );
+		}
+
+		$template_data['import_settings'] = $template_settings;
+		$template_data['id'] = $id;
+
+		$this->set_templates_conditions( $template_data );
+
+		if ( isset( $import_data['session_id'] ) ) {
+			$new_attachment_callback = function( $attachment_id ) use ( $import_data ) {
+				$this->set_session_post_meta( $attachment_id, $import_data['session_id'] );
+			};
+
+			add_filter( 'elementor/template_library/import_images/new_attachment', $new_attachment_callback );
+		}
+
+		$new_document->import( $template_data );
+
+		if ( isset( $new_attachment_callback ) ) {
+			remove_filter( 'elementor/template_library/import_images/new_attachment', $new_attachment_callback );
+		}
+
+		$document_id = $new_document->get_main_id();
+
+		if ( isset( $import_data['session_id'] ) ) {
+			$this->set_session_post_meta( $document_id, $import_data['session_id'] );
+		}
+
+		return $document_id;
+	}
+
+	private function set_session_post_meta( $post_id, $session_id ) {
+		update_post_meta( $post_id, '_elementor_import_session_id', $session_id );
+	}
+
+	private function set_templates_conditions( $template_data ) {
+		$conditions = $template_data['import_settings']['conditions'] ?? [];
+
+		if ( empty( $conditions ) ) {
+			return;
+		}
+
+		$condition = $conditions[0];
+
+		$condition = rtrim( implode( '/', $condition ), '/' );
+
+		/** @var ThemeBuilderModule $theme_builder_module */
+		$theme_builder_module = Plugin::instance()->modules_manager->get_modules( 'theme-builder' );
+		$conditions_manager = $theme_builder_module->get_conditions_manager();
+
+		$conflicts = $conditions_manager->get_conditions_conflicts_by_location(
+			$condition,
+			$template_data['import_settings']['location']
+		);
+
+		foreach ( $conflicts as $template ) {
+			$template_document = Plugin::elementor()->documents->get( $template['template_id'] );
+
+			$template_conditions = $theme_builder_module->get_conditions_manager()->get_document_conditions( $template_document );
+
+			$this->templates_conditions[ $template['template_id'] ] = $template_conditions;
+		}
+	}
+
+	public function array_merge_with_key_preservation( array $base_array, array $additional_array ): array {
+		return $base_array + $additional_array;
+	}
+}

modules/theme-builder/import-export-customization/revert.php

@@ -0,0 +1,20 @@
+<?php
+
+namespace ElementorPro\Modules\ThemeBuilder\ImportExportCustomization;
+
+use ElementorPro\Plugin;
+
+class Revert {
+
+	public function revert_theme_builder_templates_conditions( array $data ) {
+		$theme_builder_module = Plugin::instance()->modules_manager->get_modules( 'theme-builder' );
+
+		$theme_builder_module->get_conditions_manager()->clear_cache();
+
+		$old_conditions = $data['runners']['templates']['template_conditions'] ?? [];
+
+		foreach ( $old_conditions as $template_id => $conditions ) {
+			$theme_builder_module->get_conditions_manager()->save_conditions( $template_id, $conditions );
+		}
+	}
+}