<?php declare( strict_types=1 );

namespace FernleafSystems\Wordpress\Plugin\Shield\ActionRouter\Actions;

use FernleafSystems\Wordpress\Plugin\Shield\Modules\HackGuard\Lib\Utility\FileDownloadHandler;
use FernleafSystems\Wordpress\Plugin\Shield\Modules\Plugin\Lib\ImportExport\Export;
use FernleafSystems\Wordpress\Plugin\Shield\Utilities\Tool\DbTableExport;
use FernleafSystems\Wordpress\Services\Utilities\File\Download\IssueFileDownloadResponse;

class FileDownload extends BaseAction {

	public const SLUG = 'file_download';

	protected function exec() {
		try {
			$cat = $this->action_data[ 'download_category' ] ?? '';
			if ( empty( $cat ) ) {
				throw new \Exception( __( 'Invalid download request.', 'wp-simple-firewall' ) );
			}
			$contents = $this->getFileDownloadContents( $cat );

			( new IssueFileDownloadResponse( $contents[ 'name' ] ) )
				->fromString( $contents[ 'content' ], [ 'Set-cookie' => 'fileDownload=true; path=/' ] );
		}
		catch ( \Exception $e ) {
			$resp = $this->response();
			$resp->success = false;
			$resp->message = $e->getMessage();
		}
	}

	/**
	 * @return array{name:string, content:string}
	 * @throws \Exception
	 */
	private function getFileDownloadContents( string $downloadID ) :array {
		$con = self::con();

		switch ( $downloadID ) {

			case 'filelocker':
				$fileDetails = $con->comps->file_locker->handleFileDownloadRequest();
				break;

			case 'scan_file':
				$fileDetails = ( new FileDownloadHandler() )->downloadByItemId( (int)$this->action_data[ 'rid' ] ?? -1 );
				break;

			case 'db_ip':
				$fileDetails = ( new DbTableExport() )
					->setDbH( $con->db_con->ip_rules )
					->toCSV();
				break;

			case 'plugin_export':
				$fileDetails = ( new Export() )->toFile();
				break;

			case 'report_download_pdf':
				$fileDetails = [
					'name'    => wp_rand().'.pdf',
					'content' => self::con()->comps->reports->convertToPdf( (int)$this->action_data[ 'rid' ] ?? -1 )
				];
				break;

			default:
				throw new \Exception( __( 'Invalid download request.', 'wp-simple-firewall' ) );
		}

		return $fileDetails;
	}
}
