1 files changed, 126 insertions, 0 deletions
diff --git a/extensions/PdfHandler/CreatePdfThumbnailsJob.class.php b/extensions/PdfHandler/CreatePdfThumbnailsJob.class.php
new file mode 100644
index 00000000..aba204f2
--- /dev/null
+++ b/extensions/PdfHandler/CreatePdfThumbnailsJob.class.php
@@ -0,0 +1,126 @@
+<?php
+
+class CreatePdfThumbnailsJob extends Job {
+	/**
+	 * Flags for thumbnail jobs
+	 */
+	const BIG_THUMB = 1;
+	const SMALL_THUMB = 2;
+
+	/**
+	 * Construct a thumbnail job
+	 *
+	 * @param $title Title Title object
+	 * @param $params array Associative array of options:
+	 *     page:           page number for which the thumbnail will be created
+	 *     jobtype:        CreatePDFThumbnailsJob::BIG_THUMB or CreatePDFThumbnailsJob::SMALL_THUMB
+	 *                     BIG_THUMB will create a thumbnail visible for full thumbnail view,
+	 *                     SMALL_THUMB will create a thumbnail shown in "previous page"/"next page" boxes
+	 * 
+	 */
+	public function __construct( $title, $params ) {
+		parent::__construct( 'createPdfThumbnailsJob', $title, $params );
+	}
+	
+	/**
+	 * Run a thumbnail job on a given PDF file.
+	 * @return bool true
+	 */
+	public function run() {
+		if ( !isset( $this->params['page'] ) ) {
+			wfDebugLog('thumbnails', 'A page for thumbnails job of ' . $this->title->getText() . ' was not specified! That should never happen!');
+			return true; // no page set? that should never happen
+		}
+
+		$file = wfLocalFile( $this->title ); // we just want a local file
+		if ( !$file ) {
+			return true; // Just silently fail, perhaps the file was already deleted, don't bother
+		}
+
+		switch ($this->params['jobtype']) {
+			case self::BIG_THUMB:
+				global $wgImageLimits;
+				// Ignore user preferences, do default thumbnails
+				// everything here shamelessy copied and reused from includes/ImagePage.php
+				$sizeSel = User::getDefaultOption( 'imagesize' );
+
+				// The user offset might still be incorrect, specially if
+				// $wgImageLimits got changed (see bug #8858).
+				if ( !isset( $wgImageLimits[$sizeSel] ) ) {
+					// Default to the first offset in $wgImageLimits
+					$sizeSel = 0;
+				}
+				$max = $wgImageLimits[$sizeSel];
+				$maxWidth = $max[0];
+				$maxHeight = $max[1];
+
+				$width_orig = $file->getWidth( $this->params['page'] );
+				$width = $width_orig;
+				$height_orig = $file->getHeight( $this->params['page'] );
+				$height = $height_orig;
+				if ( $width > $maxWidth || $height > $maxHeight ) {
+					# Calculate the thumbnail size.
+					# First case, the limiting factor is the width, not the height.
+					if ( $width / $height >= $maxWidth / $maxHeight ) {
+						//$height = round( $height * $maxWidth / $width );
+						$width = $maxWidth;
+						# Note that $height <= $maxHeight now.
+					} else {
+						$newwidth = floor( $width * $maxHeight / $height );
+						//$height = round( $height * $newwidth / $width );
+						$width = $newwidth;
+						# Note that $height <= $maxHeight now, but might not be identical
+						# because of rounding.
+					}
+					$transformParams = array( 'page' => $this->params['page'], 'width' => $width );
+					$file->transform( $transformParams );
+				}
+				break;
+
+			case self::SMALL_THUMB:
+				Linker::makeThumbLinkObj( $this->title, $file, '', '', 'none', array( 'page' => $this->params['page'] ) );
+				break;
+		}
+
+		return true;
+	}
+
+	/**
+	 * @param $upload UploadBase
+	 * @param $mime
+	 * @param $error
+	 * @return bool
+	 */
+	public static function insertJobs( $upload, $mime, &$error ) {
+		global $wgPdfCreateThumbnailsInJobQueue;
+		if ( !$wgPdfCreateThumbnailsInJobQueue ) {
+			return true;
+		}
+		if (!MimeMagic::singleton()->isMatchingExtension('pdf', $mime)) {
+			return true; // not a PDF, abort
+		}
+
+		$title = $upload->getTitle();
+		$uploadFile = $upload->getLocalFile();
+		if ( is_null( $uploadFile ) ) {
+			wfDebugLog('thumbnails', '$uploadFile seems to be null, should never happen...');
+			return true; // should never happen, but it's better to be secure
+		}
+
+		$metadata = $uploadFile->getMetadata();
+		$unserialized = unserialize( $metadata );
+		$pages = intval( $unserialized['Pages'] );
+
+		$jobs = array();
+		for ( $i = 1; $i <= $pages; $i++ ) {
+			$jobs[] = new CreatePdfThumbnailsJob( $title, 
+								array( 'page' => $i, 'jobtype' => self::BIG_THUMB )
+							);
+			$jobs[] = new CreatePdfThumbnailsJob( $title, 
+								array( 'page' => $i, 'jobtype' => self::SMALL_THUMB )
+							);
+		}
+		Job::batchInsert( $jobs );
+		return true;
+	}
+}