newznab-tmux/newznab/controllers/ReleaseImage.php

<?php

use newznab\db\Settings;
use newznab\utility\Utility;

/**
 * Resize/save/delete images to disk.
 *
 * Class ReleaseImage
 */
class ReleaseImage
{
	/**
	 * Path to save ogg audio samples.
	 *
	 * @var string
	 */
	public $audSavePath;

	/**
	 * Path to save video preview jpg pictures.
	 *
	 * @var string
	 */
	public $imgSavePath;

	/**
	 * Path to save large jpg pictures(xxx).
	 *
	 * @var string
	 */
	public $jpgSavePath;

	/**
	 * Path to save movie jpg covers.
	 *
	 * @var string
	 */
	public $movieImgSavePath;

	/**
	 * Path to save video ogv files.
	 *
	 * @var string
	 */
	public $vidSavePath;

	/**
	 * Construct.
	 * @param \DB()
	 */
	public function __construct(&$pdo = null)
	{
		// Creates the NN_COVERS constant
		if ($pdo === null) {
			$pdo = new Settings();
		}
		//                                                            Table    |  Column
		$this->audSavePath    = NN_COVERS . 'audiosample' . DS; // releases    guid
		$this->imgSavePath    = NN_COVERS . 'preview'     . DS; // releases    guid
		$this->jpgSavePath    = NN_COVERS . 'sample'      . DS; // releases    guid
		$this->movieImgSavePath = NN_COVERS . 'movies'      . DS; // releases    imdbid
		$this->vidSavePath    = NN_COVERS . 'video'       . DS; // releases    guid

		/* For reference. *
		$this->anidbImgPath   = NN_COVERS . 'anime'       . DS; // anidb       anidbid | used in populate_anidb.php, not anidb.php
		$this->bookImgPath    = NN_COVERS . 'book'        . DS; // bookinfo    id
		$this->consoleImgPath = NN_COVERS . 'console'     . DS; // consoleinfo id
		$this->musicImgPath   = NN_COVERS . 'music'       . DS; // musicinfo   id
		$this->tvRageImgPath  = NN_COVERS . 'tvrage'      . DS; // tvrage      id (not rageid)

		$this->audioImgPath   = NN_COVERS . 'audio'       . DS; // unused folder, music folder already exists.
		**/
	}

	/**
	 * Get a URL or file image and convert it to string.
	 *
	 * @param string $imgLoc URL or file location.
	 *
	 * @return bool|mixed|string
	 */
	protected function fetchImage($imgLoc)
	{
		$img = false;

		if (strpos(strtolower($imgLoc), 'http:') === 0 || strpos(strtolower($imgLoc), 'https:') === 0) {
			$img = Utility::getUrl(['url' => $imgLoc]);
		} else if (is_file($imgLoc)) {
			$img = @file_get_contents($imgLoc);
		}

		if ($img !== false) {
			$im = @imagecreatefromstring($img);
			if ($im !== false) {
				imagedestroy($im);
				return $img;
			}
		}
		return false;
	}

	/**
	 * Save an image to disk, optionally resizing it.
	 * @param string $imgName      What to name the new image.
	 * @param string $imgLoc       URL or location on the disk the original image is in.
	 * @param string $imgSavePath  Folder to save the new image in.
	 * @param string $imgMaxWidth  Max width to resize image to.   (OPTIONAL)
	 * @param string $imgMaxHeight Max height to resize image to.  (OPTIONAL)
	 * @param bool   $saveThumb    Save a thumbnail of this image? (OPTIONAL)
	 *
	 * @return int 1 on success, 0 on failure Used on site to check if there is an image.
	 */
	public function saveImage($imgName, $imgLoc, $imgSavePath, $imgMaxWidth='', $imgMaxHeight='', $saveThumb=false)
	{
		// Try to get the image as a string.
		$cover = $this->fetchImage($imgLoc);
		if ($cover === false) {
			return 0;
		}

		// Check if we need to resize it.
		if ($imgMaxWidth != '' && $imgMaxHeight != '') {
			$im = @imagecreatefromstring($cover);
			$width = @imagesx($im);
			$height = @imagesy($im);
			$ratio = min($imgMaxHeight/$height, $imgMaxWidth/$width);
			// New dimensions
			$new_width = intval($ratio*$width);
			$new_height = intval($ratio*$height);
			if ($new_width < $width && $new_width > 10 && $new_height > 10) {
				$new_image = @imagecreatetruecolor($new_width, $new_height);
				@imagecopyresampled($new_image, $im, 0, 0, 0, 0, $new_width, $new_height, $width, $height);
				ob_start();
				@imagejpeg($new_image, null, 85);
				$thumb = ob_get_clean();
				@imagedestroy($new_image);

				if ($saveThumb) {
					@file_put_contents($imgSavePath.$imgName.'_thumb.jpg', $thumb);
				} else {
					$cover = $thumb;
				}

				unset($thumb);
			}
			@imagedestroy($im);
		}
		// Store it on the hard drive.
		$coverPath = $imgSavePath.$imgName.'.jpg';
		$coverSave = @file_put_contents($coverPath, $cover);

		// Check if it's on the drive.
		if ($coverSave === false || !is_file($coverPath)) {
			return 0;
		}
		return 1;
	}

	/**
	 * Delete images for the release.
	 *
	 * @param string      $guid   The GUID of the release.
	 *
	 * @return void
	 */
	public function delete($guid)
	{
		$thumb = $guid . '_thumb.jpg';

		// Audiosample folder.
		@unlink($this->audSavePath . $guid . '.ogg');

		// Preview folder.
		@unlink($this->imgSavePath . $thumb);

		// Sample folder.
		@unlink($this->jpgSavePath . $thumb);

		// Video folder.
		@unlink($this->vidSavePath . $guid . '.ogg');
	}
}