elementor / includes / template-library / classes / class-import-images.php

class-import-images.php in Elementor Website Builder – more than just a page builder 3.0.4, at includes/template-library/classes/class-import-images.php

177 lines 4.0 KB

Raw Download Zip

1	<?php
2	namespace Elementor\TemplateLibrary\Classes;
3
4	if ( ! defined( 'ABSPATH' ) ) {
5	exit; // Exit if accessed directly.
6	}
7
8	/**
9	* Elementor template library import images.
10	*
11	* Elementor template library import images handler class is responsible for
12	* importing remote images used by the template library.
13	*
14	* @since 1.0.0
15	*/
16	class Import_Images {
17
18	/**
19	* Replaced images IDs.
20	*
21	* The IDs of all the new imported images. An array containing the old
22	* attachment ID and the new attachment ID generated after the import.
23	*
24	* @since 1.0.0
25	* @access private
26	*
27	* @var array
28	*/
29	private $_replace_image_ids = [];
30
31	/**
32	* Get image hash.
33	*
34	* Retrieve the sha1 hash of the image URL.
35	*
36	* @since 2.0.0
37	* @access private
38	*
39	* @param string $attachment_url The attachment URL.
40	*
41	* @return string Image hash.
42	*/
43	private function get_hash_image( $attachment_url ) {
44	return sha1( $attachment_url );
45	}
46
47	/**
48	* Get saved image.
49	*
50	* Retrieve new image ID, if the image has a new ID after the import.
51	*
52	* @since 2.0.0
53	* @access private
54	*
55	* @param array $attachment The attachment.
56	*
57	* @return false\|array New image ID or false.
58	*/
59	private function get_saved_image( $attachment ) {
60	global $wpdb;
61
62	if ( isset( $this->_replace_image_ids[ $attachment['id'] ] ) ) {
63	return $this->_replace_image_ids[ $attachment['id'] ];
64	}
65
66	$post_id = $wpdb->get_var(
67	$wpdb->prepare(
68	'SELECT `post_id` FROM `' . $wpdb->postmeta . '`
69	WHERE `meta_key` = \'_elementor_source_image_hash\'
70	AND `meta_value` = %s
71	;',
72	$this->get_hash_image( $attachment['url'] )
73	)
74	);
75
76	if ( $post_id ) {
77	$new_attachment = [
78	'id' => $post_id,
79	'url' => wp_get_attachment_url( $post_id ),
80	];
81	$this->_replace_image_ids[ $attachment['id'] ] = $new_attachment;
82
83	return $new_attachment;
84	}
85
86	return false;
87	}
88
89	/**
90	* Import image.
91	*
92	* Import a single image from a remote server, upload the image WordPress
93	* uploads folder, create a new attachment in the database and updates the
94	* attachment metadata.
95	*
96	* @since 1.0.0
97	* @access public
98	*
99	* @param array $attachment The attachment.
100	*
101	* @return false\|array Imported image data, or false.
102	*/
103	public function import( $attachment ) {
104	$saved_image = $this->get_saved_image( $attachment );
105
106	if ( $saved_image ) {
107	return $saved_image;
108	}
109
110	// Extract the file name and extension from the url.
111	$filename = basename( $attachment['url'] );
112
113	$file_content = wp_remote_retrieve_body( wp_safe_remote_get( $attachment['url'] ) );
114
115	if ( empty( $file_content ) ) {
116	return false;
117	}
118
119	$upload = wp_upload_bits(
120	$filename,
121	null,
122	$file_content
123	);
124
125	$post = [
126	'post_title' => $filename,
127	'guid' => $upload['url'],
128	];
129
130	$info = wp_check_filetype( $upload['file'] );
131	if ( $info ) {
132	$post['post_mime_type'] = $info['type'];
133	} else {
134	// For now just return the origin attachment
135	return $attachment;
136	// return new \WP_Error( 'attachment_processing_error', __( 'Invalid file type.', 'elementor' ) );
137	}
138
139	$post_id = wp_insert_attachment( $post, $upload['file'] );
140
141	// On REST requests.
142	if ( ! function_exists( 'wp_generate_attachment_metadata' ) ) {
143	require_once ABSPATH . '/wp-admin/includes/image.php';
144	}
145
146	wp_update_attachment_metadata(
147	$post_id,
148	wp_generate_attachment_metadata( $post_id, $upload['file'] )
149	);
150	update_post_meta( $post_id, '_elementor_source_image_hash', $this->get_hash_image( $attachment['url'] ) );
151
152	$new_attachment = [
153	'id' => $post_id,
154	'url' => $upload['url'],
155	];
156	$this->_replace_image_ids[ $attachment['id'] ] = $new_attachment;
157	return $new_attachment;
158	}
159
160	/**
161	* Template library import images constructor.
162	*
163	* Initializing the images import class used by the template library through
164	* the WordPress Filesystem API.
165	*
166	* @since 1.0.0
167	* @access public
168	*/
169	public function __construct() {
170	if ( ! function_exists( 'WP_Filesystem' ) ) {
171	require_once ABSPATH . 'wp-admin/includes/file.php';
172	}
173
174	WP_Filesystem();
175	}
176	}
177