Ejemplo n.º 1
0
 /**
  * Given the filename of the temporary image, post-process the image to be the right size, format, etc.
  *
  * Returns an error code if there is an error or UPLOAD_ERR_OK if there were no errors.
  *
  * @param String $sTmpFile -- the full path to the temporary image file (will be deleted after processing).
  * @param $errorNo -- optional initial error-code state.
  * @param $errorMsg -- optional string containing details on what went wrong if there is an UPLOAD_ERR_EXTENSION.
  */
 private function postProcessImageInternal($sTmpFile, &$errorNo = UPLOAD_ERR_OK, &$errorMsg = '')
 {
     wfProfileIn(__METHOD__);
     $aImgInfo = getimagesize($sTmpFile);
     /**
      * check if mimetype is allowed
      */
     $aAllowMime = array('image/jpeg', 'image/pjpeg', 'image/gif', 'image/png', 'image/x-png', 'image/jpg');
     if (!in_array($aImgInfo['mime'], $aAllowMime)) {
         global $wgLang;
         // This seems to be the most appropriate error message to describe that the image type is invalid.
         // Available error codes; http://php.net/manual/en/features.file-upload.errors.php
         $errorNo = UPLOAD_ERR_EXTENSION;
         $errorMsg = wfMsg('blog-avatar-error-type', $aImgInfo['mime'], $wgLang->listToText($aAllowMime));
         //				Wikia::log( __METHOD__, 'mime', $errorMsg);
         wfProfileOut(__METHOD__);
         return $errorNo;
     }
     switch ($aImgInfo['mime']) {
         case 'image/gif':
             $oImgOrig = @imagecreatefromgif($sTmpFile);
             break;
         case 'image/pjpeg':
         case 'image/jpeg':
         case 'image/jpg':
             $oImgOrig = @imagecreatefromjpeg($sTmpFile);
             break;
         case 'image/x-png':
         case 'image/png':
             $oImgOrig = @imagecreatefrompng($sTmpFile);
             break;
     }
     $aOrigSize = array('width' => $aImgInfo[0], 'height' => $aImgInfo[1]);
     /**
      * generate new image to png format
      */
     $addedAvatars = array();
     $sFilePath = $this->getFullPath();
     $ioh = new ImageOperationsHelper();
     $oImg = $ioh->postProcess($oImgOrig, $aOrigSize);
     /**
      * save to new file ... but create folder for it first
      */
     if (!is_dir(dirname($sFilePath)) && !wfMkdirParents(dirname($sFilePath))) {
         wfDebugLog("avatar", __METHOD__ . sprintf(": Cannot create directory %s", dirname($sFilePath)), true);
         wfProfileOut(__METHOD__);
         return UPLOAD_ERR_CANT_WRITE;
     }
     if (!imagepng($oImg, $sFilePath)) {
         wfDebugLog("avatar", __METHOD__ . ": Cannot save png Avatar: {$sFilePath}", true);
         $errorNo = UPLOAD_ERR_CANT_WRITE;
     } else {
         /* remove tmp file */
         imagedestroy($oImg);
         $sUserText = $this->mUser->getName();
         $mUserPage = Title::newFromText($sUserText, NS_USER);
         $oLogPage = new LogPage(AVATAR_LOG_NAME);
         $oLogPage->addEntry('avatar_chn', $mUserPage, '');
         unlink($sTmpFile);
         /**
          * notify image replication system
          */
         global $wgEnableUploadInfoExt;
         if ($wgEnableUploadInfoExt) {
             UploadInfo::log($mUserPage, $sFilePath, $this->getLocalPath());
         }
         // remove generated thumbnails
         $this->purgeThumbnails();
         $errorNo = UPLOAD_ERR_OK;
     }
     return $errorNo;
 }
Ejemplo n.º 2
0
 /**
  * log starter images to upload_log table
  *
  * @access private
  * @author Piotr Molski (moli)
  *
  */
 private function addStarterImagesToUploadLog()
 {
     global $wgEnableUploadInfoExt;
     wfProfileIn(__METHOD__);
     if (empty($wgEnableUploadInfoExt)) {
         wfProfileOut(__METHOD__);
         return;
     }
     $files = array("Wiki.png", "Favicon.ico");
     $dbr = wfGetDB(DB_SLAVE);
     $oRes = $dbr->select(array('page'), array('page_title'), array('page_namespace' => NS_IMAGE), __METHOD__);
     while ($oRow = $dbr->fetchObject($oRes)) {
         if (!in_array($oRow->page_title, $files)) {
             $files[] = $oRow->page_title;
         }
     }
     $dbr->freeResult($oRes);
     $loop = 0;
     foreach ($files as $image) {
         $oTitle = Title::newFromText($image, NS_IMAGE);
         $file = wfLocalFile($oTitle);
         if (is_object($file)) {
             # add to upload_log
             UploadInfo::log($oTitle, $file->getPath(), $file->getRel(), "", "u");
             $loop++;
         }
     }
     Wikia::log(__METHOD__, "info", "added {$loop} images to upload_log table");
     wfProfileOut(__METHOD__);
 }
Ejemplo n.º 3
0
function uploadAvatar($oMasthead, $mUser, $nypath, $city_id) {
	global $wgTmpDirectory, $UNLINK_OLD, $TEST;
	
	$result = false;
	$filename = wfBasename($nypath);
	$sFilePath = $oMasthead->getFullPath();

	$user_id = $mUser->getId();
	$__path = str_replace($filename, "", $nypath);
	$__files = $__path . "/*_" . $user_id . "_*";
	
	if ( $TEST == 1 ) {
		saveDataInDB($user_id, $city_id, $__files, $sFilePath);
		return true;
	}

	if( !isset( $wgTmpDirectory ) || !is_dir( $wgTmpDirectory ) ) {
		$wgTmpDirectory = '/tmp';
	}

	$sTmpFile = sprintf("%s/avatars/%s", $wgTmpDirectory, $filename);
	if ( copy( $nypath, $sTmpFile ) ) {
		$iFileSize = filesize($sTmpFile);
		if ( empty( $iFileSize ) ) {
			__log("Empty file {$input} reported size {$iFileSize}");
			return $result;
		}
		
		__log("Temp file set to {$sTmpFile}");

		if ( file_exists( $sTmpFile ) ) {
			$aImgInfo = getimagesize($sTmpFile);

			$aAllowMime = array( 'image/jpeg', 'image/pjpeg', 'image/gif', 'image/png', 'image/x-png', 'image/jpg' );
			if (!in_array($aImgInfo['mime'], $aAllowMime)) {
				__log("Invalid mime type " . $aImgInfo['mime'] . ", allowed: " . implode(',', $aAllowMime));
				return $result;
			}

			switch ($aImgInfo['mime']) {
				case 'image/gif':
					$oImgOrig = @imagecreatefromgif($sTmpFile);
					break;
				case 'image/pjpeg':
				case 'image/jpeg':
				case 'image/jpg':
					$oImgOrig = @imagecreatefromjpeg($sTmpFile);
					break;
				case 'image/x-png':
				case 'image/png':
					$oImgOrig = @imagecreatefrompng($sTmpFile);
					break;
			}
			$aOrigSize = array('width' => $aImgInfo[0], 'height' => $aImgInfo[1]);

			/**
			 * generate new image to png format
			 */
			$addedAvatars = array();

			/**
			 * calculate new image size - should be 100 x 100
			 */
			$iImgW = AVATAR_DEFAULT_WIDTH;
			$iImgH = AVATAR_DEFAULT_HEIGHT;
			/* WIDTH > HEIGHT */
			if ( $aOrigSize['width'] > $aOrigSize['height'] ) {
				$iImgH = $iImgW * ( $aOrigSize['height'] / $aOrigSize['width'] );
			}
			/* HEIGHT > WIDTH */
			if ( $aOrigSize['width'] < $aOrigSize['height'] ) {
				$iImgW = $iImgH * ( $aOrigSize['width'] / $aOrigSize['height'] );
			}

			/* empty image with thumb size on white background */
			$oImg = @imagecreatetruecolor($iImgW, $iImgH);
			$white = imagecolorallocate($oImg, 255, 255, 255);
			imagefill($oImg, 0, 0, $white);

			imagecopyresampled(
				$oImg,
				$oImgOrig,
				floor ( ( AVATAR_DEFAULT_WIDTH - $iImgW ) / 2 ) /*dx*/,
				floor ( ( AVATAR_DEFAULT_HEIGHT - $iImgH ) / 2 ) /*dy*/,
				0 /*sx*/,
				0 /*sy*/,
				$iImgW /*dw*/,
				$iImgH /*dh*/,
				$aOrigSize['width']/*sw*/,
				$aOrigSize['height']/*sh*/
			);

			/**
			 * save to new file ... but create folder for it first
			 */
			if ( !is_dir( dirname( $sFilePath ) ) && !wfMkdirParents( dirname( $sFilePath ) ) ) {
				__log( sprintf("Cannot create directory %s", dirname( $sFilePath ) ) );
				return $result;
			}

			if ( !imagepng( $oImg, $sFilePath ) ) {
				__log( sprintf("Cannot save png Avatar: %s", $sFilePath ) );
				return $result;
			}

			/* remove tmp file */
			imagedestroy($oImg);

			$sUserText = $mUser->getName();
			$mUserPage = Title::newFromText( $sUserText, NS_USER );

			if ( $ADD_TO_USER_LOG == 1 ) {
				$oLogPage = new LogPage( AVATAR_LOG_NAME );
				$oLogPage->addEntry( 'avatar_chn', $mUserPage, '');
			}
			unlink($sTmpFile);

			/**
			 * notify image replication system
			 */
			$newAvatarPath = $oMasthead->getLocalPath();
			__log(sprintf("New avatar path -> %s", $newAvatarPath));
			global $wgEnableUploadInfoExt;
			if( $wgEnableUploadInfoExt ) {
				UploadInfo::log( $mUserPage, $sFilePath, $newAvatarPath );
			}
			$errorNo = UPLOAD_ERR_OK;

			if ( !empty($newAvatarPath) ) {
				/* set user option */
				__log(sprintf("Set %s = %s in user preferences", AVATAR_USER_OPTION_NAME, $newAvatarPath));
				$mUser->setOption( AVATAR_USER_OPTION_NAME, $newAvatarPath );
				$mUser->saveSettings();
			}

			if ( $UNLINK_OLD ) {
				$files = glob($__files);
				if ( $files ) {
					foreach ( $files as $file ) {
						__log("Unlink $file");
						unlink($file);
					}
				}
			}
			saveDataInDB($user_id, $city_id, $__files, $sFilePath);
			$result = true;
		}
		else {
			__log(sprintf("File %s doesn't exist", $sTmpFile ));
			return $result;
		}
	} else {
		__log("Cannot copy $nypath to $sTmpFile");
		return $result;
	}
	
	return $result;
}