OwlCyberSecurity - MANAGER

Edit File: st-importer-image-importer.php

<?php
/**
 * Image Importer
 *
 * @see https://github.com/elementor/elementor/blob/master/includes/template-library/classes/class-import-images.php
 *
 * => How to use?
 *
 *  $image = array(
 *      'url' => '<image-url>',
 *      'id'  => '<image-id>',
 *  );
 *
 *  $downloaded_image = Astra_Sites_Image_Importer::get_instance()->import( $image );
 *
 * @package Astra Sites
 * @since 1.0.14
 */

namespace STImporter\Importer\Helpers;

use STImporter\Importer\ST_Importer_Helper;

if ( ! class_exists( 'ST_Image_Importer' ) ) :

/**
	 * Astra Sites Image Importer
	 *
	 * @since 1.0.14
	 */
	class ST_Image_Importer {

/**
		 * Instance
		 *
		 * @since 1.0.14
		 * @var object Class object.
		 * @access private
		 */
		private static $instance = null;

/**
		 * Images IDs
		 *
		 * @var array   The Array of already image IDs.
		 * @since 1.0.14
		 */
		private $already_imported_ids = array();

/**
		 * Initiator
		 *
		 * @since 1.0.14
		 * @return object initialized object of class.
		 */
		public static function get_instance() {
			if ( null === self::$instance ) {
				self::$instance = new self();
			}
			return self::$instance;
		}

/**
		 * Constructor
		 *
		 * @since 1.0.14
		 */
		public function __construct() {

add_action( 'init', array( $this, 'defer_image_processing_while_import' ) );

if ( ! function_exists( 'WP_Filesystem' ) ) {
				require_once ABSPATH . 'wp-admin/includes/file.php';
			}

WP_Filesystem();
		}

/**
		 * Add filters to defer image subsizes processing while importing.
		 */
		public function defer_image_processing_while_import() {
			if ( function_exists( 'astra_sites_has_import_started' ) && astra_sites_has_import_started() && 'ai' !== get_option( 'astra_sites_current_import_template_type' ) ) {
				$this->defer_image_subsizes();
			}
		}

/**
		 * Defer image subsizes.
		 *
		 * @return void
		 */
		public function defer_image_subsizes() {
			add_filter( 'intermediate_image_sizes_advanced', array( $this, 'buffer_images_for_processing' ), 10, 3 );
		}

/**
		 * Force attachment size geenration in the background.
		 *
		 * @param array   $new_sizes       Array of image sizes.
		 * @param array   $image_meta      Metadata of the image.
		 * @param integer $attachment_id Attachment id.
		 *
		 * @return array
		 */
		public function buffer_images_for_processing( $new_sizes, $image_meta, $attachment_id ) {
			$all_attachments = get_option( 'st_attachments', array() );
			// If the cron job is already scheduled, bail.
			if ( in_array( $attachment_id, $all_attachments, true ) ) {
				return $new_sizes;
			}

$all_attachments[] = $attachment_id;

update_option( 'st_attachments', $all_attachments, 'no' );

// Return blank array of sizes to not generate any sizes in this request.
			return array();
		}

/**
		 * Process Image Download
		 *
		 * @since 1.0.14
		 * @param  array $attachments Attachment array.
		 * @return array              Attachment array.
		 */
		public function process( $attachments ) {

$downloaded_images = array();

foreach ( $attachments as $key => $attachment ) {
				$downloaded_images[] = $this->import( $attachment );
			}

return $downloaded_images;
		}

/**
		 * Get Hash Image.
		 *
		 * @since 1.0.14
		 * @param  string $attachment_url Attachment URL.
		 * @return string                 Hash string.
		 */
		public function get_hash_image( $attachment_url ) {
			return sha1( $attachment_url );
		}

/**
		 * Get Saved Image.
		 *
		 * @since 1.0.14
		 * @param  string $attachment   Attachment Data.
		 * @return string                 Hash string.
		 */
		private function get_saved_image( $attachment ) {

if ( apply_filters( 'astra_sites_image_importer_skip_image', false, $attachment ) ) {
				return array(
					'status'     => true,
					'attachment' => $attachment,
				);
			}

global $wpdb;

// 1. Is already imported in Batch Import Process?
			$post_id = $wpdb->get_var( // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching -- We are checking if this image is already processed. WO_Query would have been overkill.
				$wpdb->prepare(
					'SELECT `post_id` FROM `' . $wpdb->postmeta . '`
						WHERE `meta_key` = \'_astra_sites_image_hash\'
							AND `meta_value` = %s
					;',
					$this->get_hash_image( $attachment['url'] )
				)
			);

// 2. Is image already imported though XML?
			if ( empty( $post_id ) ) {

// Get file name without extension.
				// To check it exist in attachment.
				$filename = basename( $attachment['url'] );

// Find the attachment by meta value.
				// Code reused from Elementor plugin.
				$post_id = $wpdb->get_var( // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching -- We are checking if this attachment is already processed. WO_Query would have been overkill.
					$wpdb->prepare(
						"SELECT post_id FROM {$wpdb->postmeta}
						WHERE meta_key = '_wp_attached_file'
						AND meta_value LIKE %s",
						'%/' . $filename . '%'
					)
				);
			}

if ( $post_id ) {
				$new_attachment               = array(
					'id'  => $post_id,
					'url' => wp_get_attachment_url( $post_id ),
				);
				$this->already_imported_ids[] = $post_id;

return array(
					'status'     => true,
					'attachment' => $new_attachment,
				);
			}

return array(
				'status'     => false,
				'attachment' => $attachment,
			);
		}

/**
		 * Import Image
		 *
		 * @since 1.0.14
		 * @param  array $attachment Attachment array.
		 * @throws \Exception Exception that is catched.
		 * @return array              Attachment array.
		 */
		public function import( $attachment ) {

if ( isset( $attachment['url'] ) && ! astra_sites_is_valid_url( $attachment['url'] ) ) {
				return $attachment;
			}

$saved_image = $this->get_saved_image( $attachment );

if ( $saved_image['status'] ) {
				return $saved_image['attachment'];
			}

$file_content = wp_remote_retrieve_body(
				wp_safe_remote_get(
					$attachment['url'],
					array(
						'timeout'   => '60',
						'sslverify' => false,
					)
				)
			);

// Empty file content?
			if ( empty( $file_content ) ) {

return $attachment;
			}

// Extract the file name and extension from the URL.
			$filename = basename( $attachment['url'] );

$upload = wp_upload_bits( $filename, null, $file_content );

astra_sites_error_log( $filename );
			astra_sites_error_log( wp_json_encode( $upload ) );

$post = array(
				'post_title' => $filename,
				'guid'       => $upload['url'],
			);
			astra_sites_error_log( wp_json_encode( $post ) );

$info = wp_check_filetype( $upload['file'] );
			if ( $info ) {
				$post['post_mime_type'] = $info['type'];
			} else {
				// For now just return the origin attachment.
				return $attachment;
			}

$post_id = wp_insert_attachment( $post, $upload['file'] );
			try {

if ( ! function_exists( 'wp_generate_attachment_metadata' ) ) {
					include ABSPATH . 'wp-admin/includes/image.php';
				}

wp_update_attachment_metadata(
					$post_id,
					wp_generate_attachment_metadata( $post_id, $upload['file'] )
				);
			} catch ( \Exception $e ) {
				throw $e;
			}

update_post_meta( $post_id, '_astra_sites_image_hash', $this->get_hash_image( $attachment['url'] ) );
			ST_Importer_Helper::track_post( $post_id );

$new_attachment = array(
				'id'  => $post_id,
				'url' => $upload['url'],
			);

$this->already_imported_ids[] = $post_id;

return $new_attachment;
		}

/**
		 * Is Image URL
		 *
		 * @since 1.3.10
		 *
		 * @param  string $url URL.
		 * @return boolean
		 */
		public function is_image_url( $url = '' ) {
			if ( empty( $url ) ) {
				return false;
			}

if ( astra_sites_is_valid_image( $url ) ) {
				return true;
			}

return false;
		}

}

/**
	 * Kicking this off by calling 'get_instance()' method
	 */
	ST_Image_Importer::get_instance();

endif;