151 lines
4.2 KiB
PHP
151 lines
4.2 KiB
PHP
<?php
|
|
|
|
class WPML_Media_Set_Posts_Media_Flag implements IWPML_Action {
|
|
const HAS_MEDIA_POST_FLAG = '_wpml_media_has_media';
|
|
|
|
const BATCH_SIZE = 100;
|
|
/**
|
|
* @var wpdb $wpdb
|
|
*/
|
|
private $wpdb;
|
|
/**
|
|
* @var WPML_Notices
|
|
*/
|
|
private $wpml_notices;
|
|
/**
|
|
* @var WPML_Media_Post_Media_Usage
|
|
*/
|
|
private $post_media_usage;
|
|
/**
|
|
* @var WPML_Media_Post_With_Media_Files_Factory
|
|
*/
|
|
private $post_with_media_files_factory;
|
|
|
|
public function __construct(
|
|
wpdb $wpdb,
|
|
WPML_Notices $wpml_notices,
|
|
WPML_Media_Post_Media_Usage $post_media_usage,
|
|
WPML_Media_Post_With_Media_Files_Factory $post_with_media_files_factory
|
|
) {
|
|
$this->wpdb = $wpdb;
|
|
$this->wpml_notices = $wpml_notices;
|
|
$this->post_media_usage = $post_media_usage;
|
|
$this->post_with_media_files_factory = $post_with_media_files_factory;
|
|
}
|
|
|
|
public function add_hooks() {
|
|
add_action( 'wp_ajax_' . WPML_Media_Posts_Media_Flag_Notice::PREPARE_ACTION, array( $this, 'clear_flags' ) );
|
|
add_action( 'wp_ajax_' . WPML_Media_Posts_Media_Flag_Notice::PROCESS_ACTION, array( $this, 'process_batch' ) );
|
|
|
|
add_action( 'save_post', array( $this, 'update_post_flag' ) );
|
|
}
|
|
|
|
public function clear_flags() {
|
|
if ( $this->verify_nonce( WPML_Media_Posts_Media_Flag_Notice::PREPARE_ACTION ) ) {
|
|
|
|
if ( ! WPML_Media::has_setup_started() ) {
|
|
$this->wpdb->delete( $this->wpdb->postmeta, array( 'meta_key' => self::HAS_MEDIA_POST_FLAG ), array( '%s' ) );
|
|
}
|
|
wp_send_json_success( array( 'status' => __( 'Running setup...', 'wpml-media' ) ) );
|
|
|
|
} else {
|
|
wp_send_json_error( array( 'status' => 'Invalid nonce' ) );
|
|
}
|
|
}
|
|
|
|
public function process_batch() {
|
|
if ( $this->verify_nonce( WPML_Media_Posts_Media_Flag_Notice::PROCESS_ACTION ) ) {
|
|
$this->mark_started();
|
|
|
|
$continue = false;
|
|
$status = __( 'Setup complete!', 'wpml-media' );
|
|
$offset = isset( $_POST['offset'] ) ? (int) $_POST['offset'] : 0;
|
|
|
|
if ( ! WPML_Media::has_setup_run() ) {
|
|
|
|
$sql = $this->wpdb->prepare( "
|
|
SELECT SQL_CALC_FOUND_ROWS ID, post_content FROM {$this->wpdb->posts} p
|
|
JOIN {$this->wpdb->prefix}icl_translations t
|
|
ON t.element_id = p.ID AND t.element_type LIKE 'post_%'
|
|
LEFT JOIN {$this->wpdb->prefix}postmeta m ON p.ID = m.post_id AND m.meta_key='%s'
|
|
WHERE p.post_type NOT IN ( 'auto-draft', 'attachment', 'revision' )
|
|
AND t.source_language_code IS NULL AND m.meta_id IS NULL
|
|
ORDER BY ID ASC
|
|
LIMIT %d, %d
|
|
", self::HAS_MEDIA_POST_FLAG, $offset, self::BATCH_SIZE );
|
|
|
|
$posts = $this->wpdb->get_results( $sql );
|
|
|
|
$total_posts_found = $this->wpdb->get_var( 'SELECT FOUND_ROWS()' );
|
|
|
|
if ( $continue = ( count( $posts ) > 0 ) ) {
|
|
$this->flag_posts( $posts );
|
|
$this->record_media_usage( $posts );
|
|
$progress = round( 100*min( $offset, $total_posts_found )/$total_posts_found );
|
|
$status = sprintf( __( 'Setup in progress: %d%% complete...', 'wpml-media' ), $progress );
|
|
}
|
|
}
|
|
|
|
if ( ! $continue ) {
|
|
$this->mark_complete();
|
|
}
|
|
|
|
wp_send_json_success( array(
|
|
'status' => $status,
|
|
'offset' => $offset + self::BATCH_SIZE,
|
|
'continue' => $continue
|
|
) );
|
|
|
|
} else {
|
|
wp_send_json_error( array( 'status' => 'Invalid nonce' ) );
|
|
}
|
|
}
|
|
|
|
private function verify_nonce( $action ) {
|
|
return isset( $_POST['nonce'] ) && wp_verify_nonce( $_POST['nonce'], $action );
|
|
}
|
|
|
|
/**
|
|
* @param array $posts
|
|
*/
|
|
private function flag_posts( $posts ) {
|
|
foreach ( $posts as $post ) {
|
|
$this->update_post_flag( $post->ID );
|
|
}
|
|
}
|
|
|
|
public function update_post_flag( $post_id ) {
|
|
|
|
$post_with_media_files = $this->post_with_media_files_factory->create( $post_id );
|
|
|
|
if ( $post_with_media_files->get_media_ids() ) {
|
|
update_post_meta( $post_id, self::HAS_MEDIA_POST_FLAG, 1 );
|
|
} else {
|
|
delete_post_meta( $post_id, self::HAS_MEDIA_POST_FLAG );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param array $posts
|
|
*/
|
|
private function record_media_usage( $posts ) {
|
|
foreach ( $posts as $post ) {
|
|
$this->post_media_usage->update_media_usage( $post->ID );
|
|
}
|
|
}
|
|
|
|
private function mark_complete() {
|
|
WPML_Media::set_setup_run();
|
|
$this->wpml_notices->remove_notice(
|
|
WPML_Media_Posts_Media_Flag_Notice::NOTICE_GROUP,
|
|
WPML_Media_Posts_Media_Flag_Notice::NOTICE_ID
|
|
);
|
|
}
|
|
|
|
private function mark_started() {
|
|
WPML_Media::set_setup_started();
|
|
}
|
|
|
|
|
|
}
|