2017-05-03 15:22:02 +03:00
|
|
|
<?php
|
|
|
|
|
|
|
|
namespace OCA\DAV\CardDAV;
|
|
|
|
|
|
|
|
use OCP\Files\IAppData;
|
|
|
|
use OCP\Files\NotFoundException;
|
|
|
|
use OCP\Files\NotPermittedException;
|
|
|
|
use OCP\Files\SimpleFS\ISimpleFile;
|
|
|
|
use OCP\Files\SimpleFS\ISimpleFolder;
|
|
|
|
use Sabre\CardDAV\Card;
|
|
|
|
use Sabre\VObject\Property\Binary;
|
|
|
|
use Sabre\VObject\Reader;
|
|
|
|
|
|
|
|
class PhotoCache {
|
|
|
|
|
|
|
|
/** @var IAppData $appData */
|
|
|
|
protected $appData;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* PhotoCache constructor.
|
|
|
|
*
|
|
|
|
* @param IAppData $appData
|
|
|
|
*/
|
|
|
|
public function __construct(IAppData $appData) {
|
|
|
|
$this->appData = $appData;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param string $hash
|
|
|
|
* @param int $size
|
|
|
|
* @param Card $card
|
|
|
|
*
|
|
|
|
* @return ISimpleFile
|
|
|
|
* @throws NotFoundException
|
|
|
|
*/
|
|
|
|
public function get($hash, $size, Card $card) {
|
|
|
|
$folder = $this->getFolder($hash);
|
|
|
|
|
|
|
|
if ($this->isEmpty($folder)) {
|
|
|
|
$this->init($folder, $card);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!$this->hasPhoto($folder)) {
|
|
|
|
throw new NotFoundException();
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($size !== -1) {
|
|
|
|
$size = 2 ** ceil(log($size) / log(2));
|
|
|
|
}
|
|
|
|
|
|
|
|
return $this->getFile($folder, $size);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param ISimpleFolder $folder
|
|
|
|
* @return bool
|
|
|
|
*/
|
|
|
|
private function isEmpty(ISimpleFolder $folder) {
|
|
|
|
return $folder->getDirectoryListing() === [];
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param ISimpleFolder $folder
|
|
|
|
* @param Card $card
|
|
|
|
*/
|
|
|
|
private function init(ISimpleFolder $folder, Card $card) {
|
|
|
|
$data = $this->getPhoto($card);
|
|
|
|
|
|
|
|
if ($data === false) {
|
|
|
|
$folder->newFile('nophoto');
|
|
|
|
} else {
|
|
|
|
switch ($data['Content-Type']) {
|
|
|
|
case 'image/png':
|
|
|
|
$ext = 'png';
|
|
|
|
break;
|
|
|
|
case 'image/jpeg':
|
|
|
|
$ext = 'jpg';
|
|
|
|
break;
|
|
|
|
case 'image/gif':
|
|
|
|
$ext = 'gif';
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
$file = $folder->newFile('photo.' . $ext);
|
|
|
|
$file->putContent($data['body']);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private function hasPhoto(ISimpleFolder $folder) {
|
|
|
|
return !$folder->fileExists('nophoto');
|
|
|
|
}
|
|
|
|
|
|
|
|
private function getFile(ISimpleFolder $folder, $size) {
|
|
|
|
$ext = $this->getExtension($folder);
|
|
|
|
|
|
|
|
if ($size === -1) {
|
|
|
|
$path = 'photo.' . $ext;
|
|
|
|
} else {
|
|
|
|
$path = 'photo.' . $size . '.' . $ext;
|
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
|
|
|
$file = $folder->getFile($path);
|
|
|
|
} catch (NotFoundException $e) {
|
|
|
|
if ($size <= 0) {
|
|
|
|
throw new NotFoundException;
|
|
|
|
}
|
|
|
|
|
|
|
|
$photo = new \OC_Image();
|
|
|
|
/** @var ISimpleFile $file */
|
|
|
|
$file = $folder->getFile('photo.' . $ext);
|
|
|
|
$photo->loadFromData($file->getContent());
|
2017-05-03 15:44:23 +03:00
|
|
|
|
|
|
|
$ratio = $photo->width() / $photo->height();
|
|
|
|
if ($ratio < 1) {
|
|
|
|
$ratio = 1/$ratio;
|
|
|
|
}
|
|
|
|
$size = (int)($size * $ratio);
|
|
|
|
|
2017-05-03 15:22:02 +03:00
|
|
|
if ($size !== -1) {
|
|
|
|
$photo->resize($size);
|
|
|
|
}
|
|
|
|
try {
|
|
|
|
$file = $folder->newFile($path);
|
|
|
|
$file->putContent($photo->data());
|
|
|
|
} catch (NotPermittedException $e) {
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return $file;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param $hash
|
|
|
|
* @return ISimpleFolder
|
|
|
|
*/
|
|
|
|
private function getFolder($hash) {
|
|
|
|
try {
|
|
|
|
return $this->appData->getFolder($hash);
|
|
|
|
} catch (NotFoundException $e) {
|
|
|
|
return $this->appData->newFolder($hash);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Get the extension of the avatar. If there is no avatar throw Exception
|
|
|
|
*
|
|
|
|
* @param ISimpleFolder $folder
|
|
|
|
* @return string
|
|
|
|
* @throws NotFoundException
|
|
|
|
*/
|
|
|
|
private function getExtension(ISimpleFolder $folder) {
|
|
|
|
if ($folder->fileExists('photo.jpg')) {
|
|
|
|
return 'jpg';
|
|
|
|
} elseif ($folder->fileExists('photo.png')) {
|
|
|
|
return 'png';
|
|
|
|
} elseif ($folder->fileExists('photo.gif')) {
|
|
|
|
return 'gif';
|
|
|
|
}
|
|
|
|
throw new NotFoundException;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function getPhoto(Card $node) {
|
|
|
|
try {
|
|
|
|
$vObject = $this->readCard($node->get());
|
|
|
|
if (!$vObject->PHOTO) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
$photo = $vObject->PHOTO;
|
|
|
|
$type = $this->getType($photo);
|
|
|
|
|
|
|
|
$val = $photo->getValue();
|
|
|
|
if ($photo->getValueType() === 'URI') {
|
|
|
|
$parsed = \Sabre\URI\parse($val);
|
|
|
|
//only allow data://
|
|
|
|
if ($parsed['scheme'] !== 'data') {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
if (substr_count($parsed['path'], ';') === 1) {
|
|
|
|
list($type,) = explode(';', $parsed['path']);
|
|
|
|
}
|
|
|
|
$val = file_get_contents($val);
|
|
|
|
}
|
|
|
|
|
|
|
|
$allowedContentTypes = [
|
|
|
|
'image/png',
|
|
|
|
'image/jpeg',
|
|
|
|
'image/gif',
|
|
|
|
];
|
|
|
|
|
|
|
|
if(!in_array($type, $allowedContentTypes, true)) {
|
|
|
|
$type = 'application/octet-stream';
|
|
|
|
}
|
|
|
|
|
|
|
|
return [
|
|
|
|
'Content-Type' => $type,
|
|
|
|
'body' => $val
|
|
|
|
];
|
|
|
|
} catch(\Exception $ex) {
|
|
|
|
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param string $cardData
|
|
|
|
* @return \Sabre\VObject\Document
|
|
|
|
*/
|
|
|
|
private function readCard($cardData) {
|
|
|
|
return Reader::read($cardData);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param Binary $photo
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
private function getType(Binary $photo) {
|
|
|
|
$params = $photo->parameters();
|
|
|
|
if (isset($params['TYPE']) || isset($params['MEDIATYPE'])) {
|
|
|
|
/** @var Parameter $typeParam */
|
|
|
|
$typeParam = isset($params['TYPE']) ? $params['TYPE'] : $params['MEDIATYPE'];
|
|
|
|
$type = $typeParam->getValue();
|
|
|
|
|
|
|
|
if (strpos($type, 'image/') === 0) {
|
|
|
|
return $type;
|
|
|
|
} else {
|
|
|
|
return 'image/' . strtolower($type);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return '';
|
|
|
|
}
|
|
|
|
}
|