WPS3Media/classes/pro/background-processes/woocommerce-product-urls-process.php

<?php

namespace DeliciousBrains\WP_Offload_Media\Pro\Background_Processes;

use DeliciousBrains\WP_Offload_Media\Integrations\Media_Library;
use DeliciousBrains\WP_Offload_Media\Items\Media_Library_Item;
use DeliciousBrains\WP_Offload_Media\Pro\Integrations\Woocommerce;
use DeliciousBrains\WP_Offload_Media\Pro\Items\Update_Acl_Handler;
use Exception;

class Woocommerce_Product_Urls_Process extends Background_Tool_Process {

	/**
	 * @var string
	 */
	protected $action = 'woocommerce_product_url';

	/**
	 * Process chunk of products
	 *
	 * @param string $source_type
	 * @param array  $source_ids
	 * @param int    $blog_id
	 *
	 * @return array
	 *
	 * @throws Exception
	 */
	protected function process_items_chunk( $source_type, $source_ids, $blog_id ) {
		foreach ( $source_ids as $source_id ) {
			$this->handle_attachment( $source_id, $blog_id );
		}

		// We processed every item.
		return $source_ids;
	}

	/**
	 * Process individual product or product variations by looking for downloadable
	 * files directly in the products meta data.
	 *
	 * @param int $product_id
	 * @param int $blog_id
	 *
	 * @return bool
	 * @throws Exception
	 */
	protected function handle_attachment( $product_id, $blog_id ) {
		/** @var Media_Library $media_library */
		$media_library = $this->as3cf->get_integration_manager()->get_integration( 'mlib' );
		$woocommerce   = new Woocommerce( $this->as3cf );

		$acl_handler = $this->as3cf->get_item_handler( Update_Acl_Handler::get_item_handler_key_name() );

		// Get all the downloadable files for this post. Straight
		// from the DB to avoid filters
		$downloads = get_post_meta( $product_id, '_downloadable_files', true );

		// If we don't get an array, there's nothing we can do with
		// this product / variation
		if ( ! $downloads || ! is_array( $downloads ) ) {
			return true;
		}

		$updated = false;

		foreach ( $downloads as &$download ) {
			$stored_file   = $download['file'];
			$size          = null;
			$update_needed = false;

			// Is this our shortcode?
			$attachment_id = $woocommerce->get_attachment_id_from_shortcode( $stored_file );
			if ( $attachment_id ) {
				$atts          = $woocommerce->get_shortcode_atts( $stored_file );
				$update_needed = isset( $atts['id'] );
			}

			// Is this a local URL?
			if ( false === $attachment_id ) {
				$attachment_id = $media_library->get_attachment_id_from_local_url( $stored_file );
			}

			// Is it a remote URL we recognize?
			if ( false === $attachment_id ) {
				$attachment_id = $media_library->get_attachment_id_from_provider_url( $stored_file );
				$update_needed = true;
			}

			// If we can't identify an offloaded item we have to give up
			if ( false === $attachment_id ) {
				continue;
			}

			$as3cf_item = Media_Library_Item::get_by_source_id( $attachment_id );

			// We couldn't find an item for this $attachment_id
			if ( ! $as3cf_item ) {
				continue;
			}

			// Ensure that it's private on the provider
			$size = $this->as3cf->filter_local->get_size_string_from_url( $as3cf_item->get_item_source_array(), $stored_file );

			$options = array(
				'object_keys' => array( $size ),
				'set_private' => true,
			);
			$result  = $acl_handler->handle( $as3cf_item, $options );

			if ( is_wp_error( $result ) ) {
				$error_msg = sprintf( __( 'Error updating object ACL for media library item %s', 'amazon-s3-and-cloudfront' ), $attachment_id );
				$this->record_error( $blog_id, Media_Library_Item::source_type(), $attachment_id, $error_msg );
			}

			// If this is not a local URL already, we update:
			if ( $update_needed ) {
				$download['file'] = $as3cf_item->get_local_url( $size );
				$updated          = true;
			}
		}

		if ( $updated ) {
			update_post_meta( $product_id, '_downloadable_files', $downloads );
		}

		return true;
	}

	/**
	 * Find all product and/or product variation IDs that have a _downloadable_files
	 * meta data item.
	 *
	 * @param string $source_type     Item source type
	 * @param int    $last_product_id The ID of the last item previously processed
	 * @param int    $limit           Maximum number of product IDs to return
	 * @param bool   $count           Just return the count, negates $limit, default false
	 *
	 * @return array|int
	 */
	protected function get_blog_items( $source_type, $last_product_id, $limit, $count = false ) {
		global $wpdb;

		$args = array();

		if ( $count ) {
			$sql = 'SELECT COUNT(DISTINCT post_id)';
		} else {
			$sql = 'SELECT DISTINCT post_id';
		}

		$sql .= " FROM {$wpdb->postmeta} ";
		$sql .= " WHERE meta_key='_downloadable_files' ";

		if ( ! empty( $last_product_id ) ) {
			$sql    .= ' AND post_id < %d';
			$args[] = $last_product_id;
		}

		if ( ! $count ) {
			$sql    .= ' ORDER BY post_id DESC LIMIT %d';
			$args[] = $limit;
		}

		if ( count( $args ) > 0 ) {
			$sql = $wpdb->prepare( $sql, $args );
		}

		if ( $count ) {
			return $wpdb->get_var( $sql );
		} else {
			return array_map( 'intval', $wpdb->get_col( $sql ) );
		}
	}

	/**
	 * Called when background process has been cancelled.
	 */
	protected function cancelled() {
		// Do nothing at the moment.
	}

	/**
	 * Called when background process has been paused.
	 */
	protected function paused() {
		// Do nothing at the moment.
	}

	/**
	 * Called when background process has been resumed.
	 */
	protected function resumed() {
		// Do nothing at the moment.
	}

	/**
	 * Called when background process has completed.
	 */
	protected function completed() {
		// Do nothing at the moment.
	}

	/**
	 * Get complete notice message.
	 *
	 * @return string
	 */
	protected function get_complete_message() {
		return __( 'Finished updating and verifying WooCommerce downloads.', 'amazon-s3-and-cloudfront' );
	}
}