Wordpress 4.6

[autoinstalls/wordpress.git] / wp-admin / includes / class-file-upload-upgrader.php

diff --git a/wp-admin/includes/class-file-upload-upgrader.php b/wp-admin/includes/class-file-upload-upgrader.php
new file mode 100644 (file)
index 0000000..35baa7f
--- /dev/null
+++ b/wp-admin/includes/class-file-upload-upgrader.php
@@ -0,0 +1,125 @@
+<?php
+/**
+ * Upgrade API: File_Upload_Upgrader class
+ *
+ * @package WordPress
+ * @subpackage Upgrader
+ * @since 4.6.0
+ */
+
+/**
+ * Core class used for handling file uploads.
+ *
+ * This class handles the upload process and passes it as if it's a local file
+ * to the Upgrade/Installer functions.
+ *
+ * @since 2.8.0
+ * @since 4.6.0 Moved to its own file from wp-admin/includes/class-wp-upgrader.php.
+ */
+class File_Upload_Upgrader {
+
+       /**
+        * The full path to the file package.
+        *
+        * @since 2.8.0
+        * @access public
+        * @var string $package
+        */
+       public $package;
+
+       /**
+        * The name of the file.
+        *
+        * @since 2.8.0
+        * @access public
+        * @var string $filename
+        */
+       public $filename;
+
+       /**
+        * The ID of the attachment post for this file.
+        *
+        * @since 3.3.0
+        * @access public
+        * @var int $id
+        */
+       public $id = 0;
+
+       /**
+        * Construct the upgrader for a form.
+        *
+        * @since 2.8.0
+        * @access public
+        *
+        * @param string $form      The name of the form the file was uploaded from.
+        * @param string $urlholder The name of the `GET` parameter that holds the filename.
+        */
+       public function __construct( $form, $urlholder ) {
+
+               if ( empty($_FILES[$form]['name']) && empty($_GET[$urlholder]) )
+                       wp_die(__('Please select a file'));
+
+               //Handle a newly uploaded file, Else assume it's already been uploaded
+               if ( ! empty($_FILES) ) {
+                       $overrides = array( 'test_form' => false, 'test_type' => false );
+                       $file = wp_handle_upload( $_FILES[$form], $overrides );
+
+                       if ( isset( $file['error'] ) )
+                               wp_die( $file['error'] );
+
+                       $this->filename = $_FILES[$form]['name'];
+                       $this->package = $file['file'];
+
+                       // Construct the object array
+                       $object = array(
+                               'post_title' => $this->filename,
+                               'post_content' => $file['url'],
+                               'post_mime_type' => $file['type'],
+                               'guid' => $file['url'],
+                               'context' => 'upgrader',
+                               'post_status' => 'private'
+                       );
+
+                       // Save the data.
+                       $this->id = wp_insert_attachment( $object, $file['file'] );
+
+                       // Schedule a cleanup for 2 hours from now in case of failed install.
+                       wp_schedule_single_event( time() + 2 * HOUR_IN_SECONDS, 'upgrader_scheduled_cleanup', array( $this->id ) );
+
+               } elseif ( is_numeric( $_GET[$urlholder] ) ) {
+                       // Numeric Package = previously uploaded file, see above.
+                       $this->id = (int) $_GET[$urlholder];
+                       $attachment = get_post( $this->id );
+                       if ( empty($attachment) )
+                               wp_die(__('Please select a file'));
+
+                       $this->filename = $attachment->post_title;
+                       $this->package = get_attached_file( $attachment->ID );
+               } else {
+                       // Else, It's set to something, Back compat for plugins using the old (pre-3.3) File_Uploader handler.
+                       if ( ! ( ( $uploads = wp_upload_dir() ) && false === $uploads['error'] ) )
+                               wp_die( $uploads['error'] );
+
+                       $this->filename = $_GET[$urlholder];
+                       $this->package = $uploads['basedir'] . '/' . $this->filename;
+               }
+       }
+
+       /**
+        * Delete the attachment/uploaded file.
+        *
+        * @since 3.2.2
+        * @access public
+        *
+        * @return bool Whether the cleanup was successful.
+        */
+       public function cleanup() {
+               if ( $this->id )
+                       wp_delete_attachment( $this->id );
+
+               elseif ( file_exists( $this->package ) )
+                       return @unlink( $this->package );
+
+               return true;
+       }
+}