22

✏️ Editing: file-uploader.php

<?php

namespace JFB_Modules\Block_Parsers;

use Jet_Form_Builder\Classes\Resources\File;
use Jet_Form_Builder\Classes\Resources\File_Collection;
use Jet_Form_Builder\Classes\Resources\File_Tools;
use Jet_Form_Builder\Classes\Resources\Media_Block_Value;
use Jet_Form_Builder\Classes\Resources\Uploaded_Collection;
use Jet_Form_Builder\Classes\Resources\Uploaded_File;
use Jet_Form_Builder\Classes\Resources\Upload_Exception;
use Jet_Form_Builder\Classes\Resources\Upload_Permission_Exception;
use Jet_Form_Builder\Classes\Tools;
use JFB_Modules\Block_Parsers\Fields\Media_Field_Parser;

// If this file is called directly, abort.
if ( ! defined( 'WPINC' ) ) {
	die;
}

class File_Uploader {

protected $settings = array();

/** @var Uploaded_File|Uploaded_File[] */
	protected $preset = array();

/** @var File|File_Collection */
	protected $file;

protected $default_value;

protected $max_files;
	protected $max_size;
	protected $allowed_user_cap;
	protected $allowed_mimes;
	protected $insert_attachment;

/** @var Field_Data_Parser|Media_Field_Parser */
	protected $parser;

/**
	 * @return Media_Block_Value
	 * @throws Upload_Exception
	 * @throws Upload_Permission_Exception
	 */
	public function upload() {
		$this->sanitize();

return $this->upload_files();
	}

/**
	 * @throws Upload_Permission_Exception|Upload_Exception
	 */
	protected function sanitize() {
		$this->sanitize_permissions();
		$this->sanitize_max_files();
		$this->sanitize_files();
	}

/**
	 * @return Media_Block_Value
	 * @throws Upload_Exception
	 */
	protected function upload_files() {
		if ( $this->file instanceof File ) {
			return $this->upload_file( $this->file );
		}

$uploaded = array();

/** @var File $file */
		foreach ( $this->file as $file ) {
			try {
				$uploaded[] = $this->upload_file( $file );
			} catch ( Upload_Exception $exception ) {
				continue;
			}
		}

return new Uploaded_Collection( $uploaded );
	}

/**
	 * @param File $file
	 *
	 * @return Uploaded_File
	 * @throws Upload_Exception
	 */
	protected function upload_file( File $file ): Uploaded_File {
		$preset_uploaded = $this->get_uploaded( $file );

if ( $preset_uploaded instanceof Uploaded_File ) {
			return $preset_uploaded;
		}

$uploaded = new Uploaded_File();
		$uploaded->upload( $file );

if ( ! $this->insert_attachment ) {
			return $uploaded;
		}

$uploaded->add_attachment();

do_action( 'jet-form-builder/inserted-attachment', $uploaded, $this );

return $uploaded;
	}

protected function get_uploaded( File $file ) {
		if ( empty( $this->preset ) ) {
			return false;
		}

return File_Tools::get_uploaded( $file, $this->preset );
	}

/**
	 * @throws Upload_Permission_Exception
	 */
	protected function sanitize_permissions() {
		$cap = ! empty( $this->settings['allowed_user_cap'] ) ? $this->settings['allowed_user_cap'] : 'upload_files';

// phpcs:disable WordPress.Security.EscapeOutput.ExceptionNotEscaped
		if ( 'any_user' !== $cap && ! is_user_logged_in() ) {
			throw new Upload_Permission_Exception(
				__( 'You are not allowed to upload files', 'jet-form-builder' )
			);
		}

if ( ! in_array( $cap, array( 'all', 'any_user' ), true ) && ! current_user_can( $cap ) ) {
			throw new Upload_Permission_Exception(
				__( 'You are not allowed to upload files', 'jet-form-builder' )
			);
		}
		// phpcs:enable WordPress.Security.EscapeOutput.ExceptionNotEscaped
	}

/**
	 * @throws Upload_Exception
	 */
	protected function sanitize_max_files() {
		if ( $this->count_files() > $this->max_files ) {
			throw new Upload_Exception( 'upload_max_files' );
		}
	}

/**
	 * @throws Upload_Exception
	 */
	protected function sanitize_files() {
		if ( $this->file instanceof File ) {
			$this->sanitize_file( $this->file );

return;
		}

if ( ! ( $this->file instanceof File_Collection ) ) {
			return;
		}

/** @var File $file */
		foreach ( $this->file as $file ) {
			$this->sanitize_file( $file );
		}
	}

/**
	 * @param File $file
	 *
	 * @throws Upload_Exception
	 */
	protected function sanitize_file( File $file ) {
		if ( $file->get_size() > $this->max_size ) {
			throw new Upload_Exception( 'upload_max_size' );
		}

if ( empty( $this->allowed_mimes ) ) {
			return;
		}

if ( ! in_array( $file->get_type(), $this->allowed_mimes, true ) ) {
			throw new Upload_Exception( 'upload_mime_types' );
		}
	}

public function get_max_size(): int {
		$max_size = wp_max_upload_size();

if ( empty( $this->settings['max_size'] ) ) {
			return $max_size;
		}

$field_max_size = floatval(
			\JFB_Modules\Rich_Content\Module::rich( $this->settings['max_size'] ?? '' )
		);
		$field_max_size = intval( $field_max_size * MB_IN_BYTES );

if ( $field_max_size > $max_size ) {
			return $max_size;
		}

return $field_max_size;
	}

public function count_files(): int {
		$counter = 0;
		if ( $this->file instanceof File ) {
			++$counter;
		} elseif ( $this->file instanceof File_Collection ) {
			$counter += count( $this->file );
		}

return $counter;
	}

public function get_max_files(): int {
		$files = absint(
			\JFB_Modules\Rich_Content\Module::rich( $this->settings['max_files'] ?? '' )
		);

return empty( $files ) ? 1 : $files;
	}

public function get_mime_types(): array {
		return $this->settings['allowed_mimes'] ?? array();
	}

/**
	 * @return bool
	 */
	public function is_insert_attachment(): bool {
		// Prevent non logged-in users insert attachment
		if ( ! is_user_logged_in() ) {
			return $this->parser->get_context()->is_allowed_for_guest();
		}

return ! empty( $this->settings['insert_attachment'] );
	}

public function get_allowed_user_cap(): string {
		$capability = $this->settings['allowed_user_cap'] ?? 'upload_files';

return empty( $capability ) ? 'upload_files' : $capability;
	}

public function is_must_be_single(): bool {
		return 1 === $this->max_files;
	}

/**
	 * @param File|File_Collection $file
	 *
	 * @return File_Uploader
	 */
	public function set_file( $file ): File_Uploader {
		$this->file = $file;

return $this;
	}

public function set_preset( $preset ): File_Uploader {
		if ( empty( $preset ) || ! is_array( $preset ) ) {
			return $this;
		}

if ( $this->is_must_be_single() ) {
			$uploaded     = new Uploaded_File();
			$this->preset = $uploaded->set_from_array( $preset[0] ?? $preset );

return $this;
		}

$files = isset( $preset[0] ) ? $preset : array( $preset );

foreach ( $files as $index => $file ) {
			if ( ! is_array( $file ) ) {
				unset( $files[ $index ] );
				continue;
			}
			$files[ $index ] = ( new Uploaded_File() )->set_from_array( $file );
		}

$this->preset = $files;

return $this;
	}

/**
	 * @param array $settings
	 *
	 * @return File_Uploader
	 */
	public function set_settings( array $settings ): File_Uploader {
		$this->settings = $settings;

$this->max_files         = $this->get_max_files();
		$this->max_size          = $this->get_max_size();
		$this->allowed_user_cap  = $this->get_allowed_user_cap();
		$this->insert_attachment = $this->is_insert_attachment();
		$this->allowed_mimes     = $this->get_mime_types();

return $this;
	}

public function set_context( Field_Data_Parser $parser ): File_Uploader {
		$this->parser = $parser;

$this->set_settings( $parser->get_settings() );
		$this->set_preset( Tools::decode_json( $parser->get_value() ) );
		$this->set_file( $parser->get_file() );

return $this;
	}

public function get_settings(): array {
		return $this->settings;
	}

}

MR SKB