📁 Captain File Manager

✏️ Editing: components-repository.php

<?php

namespace Elementor\Modules\Components;

use Elementor\Core\Utils\Collection;
use Elementor\Modules\Components\Documents\Component as Component_Document;
use Elementor\Plugin;
use Elementor\Core\Base\Document;

if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly.
}

class Components_Repository {

public static function make(): Components_Repository {
		return new self();
	}

public function all(): Collection {
		// Components count is limited to 100, if we increase this number, we need to iterate the posts in batches.
		$posts = get_posts( [
			'post_type' => Component_Document::TYPE,
			'post_status' => 'any',
			'posts_per_page' => Components_REST_API::MAX_COMPONENTS,
		] );

$components = [];

foreach ( $posts as $post ) {
			$component = $this->get( $post->ID );

if ( ! $component ) {
				continue;
			}

$components[] = [
				'id' => $component->get_main_id(),
				'title' => $component->get_post()->post_title,
				'uid' => $component->get_component_uid(),
				'is_archived' => $component->get_is_archived(),
				'styles' => $this->extract_styles( $component->get_elements_data() ),
			];
		}

return Collection::make( $components );
	}

public function get( $id, bool $include_autosave = true ) {
		$doc = $include_autosave
			? Plugin::$instance->documents->get_doc_or_auto_save( $id, get_current_user_id() )
			: Plugin::$instance->documents->get( $id );

if ( ! $doc instanceof Component_Document ) {
			return null;
		}

return $doc;
	}

public function create( string $title, array $content, string $status, string $uid, array $settings = [] ) {
		$document = Plugin::$instance->documents->create(
			Component_Document::get_type(),
			[
				'post_title' => $title,
				'post_status' => $status,
			],
			[
				Component_Document::COMPONENT_UID_META_KEY => $uid,
			]
		);

try {
			$saved = $document->save( [
				'elements' => $content,
				'settings' => $settings,
			] );
		} catch ( \Exception $e ) {
			$document->force_delete();
			throw $e;
		}

if ( ! $saved ) {
			$document->force_delete();
			throw new \Exception( 'Failed to create component' );
		}

return $document->get_main_id();
	}
	private function extract_styles( array $elements, array $styles = [] ) {
		foreach ( $elements as $element ) {
			if ( isset( $element['styles'] ) ) {
				$styles = array_merge( $styles, $element['styles'] );
			}

if ( isset( $element['elements'] ) ) {
				$styles = $this->extract_styles( $element['elements'], $styles );
			}
		}

return $styles;
	}

public function archive( array $ids, string $status ) {
		$failed_ids = [];
		$success_ids = [];

foreach ( $ids as $id ) {
			try {
				$component = $this->get_component_for_edit( $id, $status );

if ( ! $component ) {
					$failed_ids[] = $id;
					continue;
				}

$component->archive();
				$success_ids[] = $id;
			} catch ( \Exception $e ) {
				$failed_ids[] = $id;
			}
		}

return [
			'failedIds' => $failed_ids,
			'successIds' => $success_ids,
		];
	}

public function update_title( int $component_id, string $title, string $status ): bool {
		$component = $this->get_component_for_edit( $component_id, $status );

if ( ! $component ) {
			return false;
		}

return $component->update_title( $title );
	}

/**
	 * Get the component for edit.
	 *
	 * @param int    $component_id The component ID.
	 * @param string $target_status The target status, means the status the component should be saved as.
	 * @return ?Component_Document The component document for edit.
	 *
	 * If target status is an autosave / draft:
	 * - If the component main document is autosave / draft, it will return the main document.
	 * - If the component main document is published, it will create a new autosave document and return it.

* If target status is publish:
	 * - Will return the main document. If it's an autosave, it will be published later by the publish_component method.
	 */
	private function get_component_for_edit( int $component_id, string $target_status ): ?Component_Document {
		$component = $this->get( $component_id );

if ( ! $component ) {
			return null;
		}

$autosave_statuses = [ Document::STATUS_AUTOSAVE, Document::STATUS_DRAFT ];
		$autosave_exists = $component->is_autosave();
		$should_create_autosave = in_array( $target_status, $autosave_statuses, true ) && ! $autosave_exists;

if ( ! $should_create_autosave ) {
			return $component;
		}

// Create a new autosave document, based on the published version.
		return $component->get_autosave( 0, true );
	}

public function publish_component( Component_Document $component ): bool {
		try {
			$main_id = $component->get_main_id();
			$main_component = $this->get( $main_id, false );

$autosave = $main_component->get_newer_autosave();

if ( $autosave ) {
				$success = $this->copy_autosave_data_to_main_component_document_and_publish( $autosave, $main_component, $main_id );
			} else {
				$success = $main_component->update_status( Document::STATUS_PUBLISH );
			}

if ( ! $success ) {
				throw new \Exception( 'Failed to publish component' );
			}
		} catch ( \Exception $e ) {
			return false;
		}

return true;
	}

private function copy_autosave_data_to_main_component_document_and_publish( Component_Document $autosave, Component_Document $main_component_document, int $main_id ): bool {
		$autosave_id = $autosave->get_post()->ID;

// Copy component custom meta keys from the autosave to the main component.
		Plugin::$instance->db->copy_elementor_meta( $autosave_id, $main_id, Component_Document::COMPONENT_CUSTOM_META_KEYS );

$autosave_elements = $autosave->get_elements_data();
		$autosave_title = $autosave->get_post()->post_title;

return $main_component_document->save( [
			'elements' => $autosave_elements,
			'settings' => [
				'post_status' => Document::STATUS_PUBLISH,
				'post_title' => $autosave_title,
			],
		] );
	}
}