namespace OCA\DAV\CardDAV;
+use OCP\Files\NotFoundException;
use OCP\ILogger;
use Sabre\CardDAV\Card;
use Sabre\DAV\Server;
protected $server;
/** @var ILogger */
private $logger;
+ /** @var PhotoCache */
+ private $cache;
- public function __construct(ILogger $logger) {
+ /**
+ * ImageExportPlugin constructor.
+ *
+ * @param ILogger $logger
+ * @param PhotoCache $cache
+ */
+ public function __construct(ILogger $logger, PhotoCache $cache) {
$this->logger = $logger;
+ $this->cache = $cache;
}
/**
* @param Server $server
* @return void
*/
- function initialize(Server $server) {
-
+ public function initialize(Server $server) {
$this->server = $server;
$this->server->on('method:GET', [$this, 'httpGet'], 90);
}
*
* @param RequestInterface $request
* @param ResponseInterface $response
- * @return bool|void
+ * @return bool
*/
- function httpGet(RequestInterface $request, ResponseInterface $response) {
+ public function httpGet(RequestInterface $request, ResponseInterface $response) {
$queryParams = $request->getQueryParameters();
// TODO: in addition to photo we should also add logo some point in time
return true;
}
+ $size = -1;
+ if (isset($queryParams['size'])) {
+ $size = (int)$queryParams['size'];
+ }
+
$path = $request->getPath();
$node = $this->server->tree->getNodeForPath($path);
$aclPlugin->checkPrivileges($path, '{DAV:}read');
}
- if ($result = $this->getPhoto($node)) {
- // Allow caching
- $response->setHeader('Cache-Control', 'private, max-age=3600, must-revalidate');
- $response->setHeader('Etag', $node->getETag() );
- $response->setHeader('Pragma', 'public');
+ // Fetch addressbook
+ $addressbookpath = explode('/', $path);
+ array_pop($addressbookpath);
+ $addressbookpath = implode('/', $addressbookpath);
+ /** @var AddressBook $addressbook */
+ $addressbook = $this->server->tree->getNodeForPath($addressbookpath);
- $response->setHeader('Content-Type', $result['Content-Type']);
- $response->setHeader('Content-Disposition', 'attachment');
- $response->setStatus(200);
+ $hash = md5($addressbook->getResourceId() . $node->getName());
+
+ $response->setHeader('Cache-Control', 'private, max-age=3600, must-revalidate');
+ $response->setHeader('Etag', $node->getETag() );
+ $response->setHeader('Pragma', 'public');
- $response->setBody($result['body']);
+ try {
+ $file = $this->cache->get($hash, $size, $node);
+ $response->setHeader('Content-Type', $file->getMimeType());
+ $response->setHeader('Content-Disposition', 'inline');
+ $response->setStatus(200);
- // Returning false to break the event chain
- return false;
+ $response->setBody($file->getContent());
+ } catch (NotFoundException $e) {
+ $response->setStatus(404);
}
- return true;
+
+ return false;
}
function getPhoto(Card $node) {
--- /dev/null
+<?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());
+ 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 '';
+ }
+}
projects / nextcloud-server.git / commitdiff