Strategy.php 1.68 KB
<?php
declare( strict_types=1 );

namespace ACA\MLA\Export;

use AC\ListTable;
use AC\ThirdParty\MediaLibraryAssistant\WpListTableFactory;
use ACA\MLA\ListScreen;
use ACA\MLA\ListTableFactory;
use ACP;

class Strategy extends ACP\Export\Strategy {

	public function __construct( ListScreen\MediaLibrary $list_screen ) {
		parent::__construct( $list_screen );
	}

	private function get_wp_list_table_factory(): WpListTableFactory {
		return new WpListTableFactory();
	}

	protected function get_list_table(): ?ListTable {
		return ( new ListTableFactory( $this->get_wp_list_table_factory() ) )->create();
	}

	public function get_total_items(): ?int {
		// will be populated through JS
		return 1;
	}

	protected function ajax_export(): void {
		add_filter( 'mla_list_table_query_final_terms', [ $this, 'query' ], 1e6 );
		add_action( 'mla_list_table_prepare_items', [ $this, 'prepare_items' ], 10, 2 );
		add_filter( 'posts_clauses', [ $this, 'filter_ids' ] );

		// Trigger above hooks early by initiating list table. This prevents "headers already sent".
		$this->get_wp_list_table_factory()->create();
	}

	public function filter_ids( $clauses ) {
		global $wpdb;

		$ids = $this->get_requested_ids();

		if ( $ids ) {
			$clauses['where'] .= sprintf( "\nAND $wpdb->posts.ID IN( %s )", implode( ',', $ids ) );
		}

		return $clauses;
	}

	public function query( $request ) {
		$per_page = $this->get_num_items_per_iteration();

		$request['offset'] = $this->get_export_counter() * $per_page;
		$request['posts_per_page'] = $per_page;
		$request['posts_per_archive_page'] = $per_page;

		return $request;
	}

	public function prepare_items( $query ): void {
		$this->export( wp_list_pluck( $query->items, 'ID' ) );
	}

}