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; } }