<?php

namespace DeliciousBrains\WPMDB\Common\Transfers\Files;

use DeliciousBrains\WPMDB\Common\FullSite\FullSiteExport;
use DeliciousBrains\WPMDB\Common\Queue\Manager;
use DeliciousBrains\WPMDB\Common\Transfers\Abstracts\TransferManagerAbstract;
use WP_Error;

/**
 * Class TransferManager
 *
 * @package WPMDB\Transfers\Files
 */
class TransferManager extends TransferManagerAbstract {
	/**
	 * @var FullSiteExport
	 */
	private $full_site_export;

	public function __construct(
		Manager $manager,
		Util $util,
		FullSiteExport $full_site_export
	) {
		parent::__construct( $manager, $util );

		$this->queueManager     = $manager;
		$this->util             = $util;
		$this->full_site_export = $full_site_export;
	}

	/**
	 * @param array $processed
	 * @param array $state_data
	 *
	 * @return array|WP_Error
	 */
	public function handle_savefile( $processed, $state_data ) {
		$added_to_zip = $this->full_site_export->add_batch_to_zip( $processed, $state_data );

		if ( is_wp_error( $added_to_zip ) ) {
			return $added_to_zip;
		}

		$this->queueManager->delete_data_from_queue( $added_to_zip['count'] );

		return [ 'total_transferred' => $added_to_zip['total_size'] ];
	}
}