123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267 |
- <?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 OCP\Image;
- use Psr\Log\LoggerInterface;
- use Sabre\CardDAV\Card;
- use Sabre\VObject\Document;
- use Sabre\VObject\Parameter;
- use Sabre\VObject\Property\Binary;
- use Sabre\VObject\Reader;
- class PhotoCache {
-
- public const ALLOWED_CONTENT_TYPES = [
- 'image/png' => 'png',
- 'image/jpeg' => 'jpg',
- 'image/gif' => 'gif',
- 'image/vnd.microsoft.icon' => 'ico',
- 'image/webp' => 'webp',
- ];
-
- public function __construct(
- protected IAppData $appData,
- protected LoggerInterface $logger,
- ) {
- }
-
- public function get(int $addressBookId, string $cardUri, int $size, Card $card): ISimpleFile {
- $folder = $this->getFolder($addressBookId, $cardUri);
- 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);
- }
- private function isEmpty(ISimpleFolder $folder): bool {
- return $folder->getDirectoryListing() === [];
- }
-
- private function init(ISimpleFolder $folder, Card $card): void {
- $data = $this->getPhoto($card);
- if ($data === false || !isset($data['Content-Type'])) {
- $folder->newFile('nophoto', '');
- return;
- }
- $contentType = $data['Content-Type'];
- $extension = self::ALLOWED_CONTENT_TYPES[$contentType] ?? null;
- if ($extension === null) {
- $folder->newFile('nophoto', '');
- return;
- }
- $file = $folder->newFile('photo.' . $extension);
- $file->putContent($data['body']);
- }
- private function hasPhoto(ISimpleFolder $folder): bool {
- return !$folder->fileExists('nophoto');
- }
-
- private function getFile(ISimpleFolder $folder, $size): ISimpleFile {
- $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 Image();
-
- $file = $folder->getFile('photo.' . $ext);
- $photo->loadFromData($file->getContent());
- $ratio = $photo->width() / $photo->height();
- if ($ratio < 1) {
- $ratio = 1 / $ratio;
- }
- $size = (int)($size * $ratio);
- if ($size !== -1) {
- $photo->resize($size);
- }
- try {
- $file = $folder->newFile($path);
- $file->putContent($photo->data());
- } catch (NotPermittedException $e) {
- }
- }
- return $file;
- }
-
- private function getFolder(int $addressBookId, string $cardUri, bool $createIfNotExists = true): ISimpleFolder {
- $hash = md5($addressBookId . ' ' . $cardUri);
- try {
- return $this->appData->getFolder($hash);
- } catch (NotFoundException $e) {
- if ($createIfNotExists) {
- return $this->appData->newFolder($hash);
- } else {
- throw $e;
- }
- }
- }
-
- private function getExtension(ISimpleFolder $folder): string {
- foreach (self::ALLOWED_CONTENT_TYPES as $extension) {
- if ($folder->fileExists('photo.' . $extension)) {
- return $extension;
- }
- }
- throw new NotFoundException('Avatar not found');
- }
-
- private function getPhoto(Card $node) {
- try {
- $vObject = $this->readCard($node->get());
- return $this->getPhotoFromVObject($vObject);
- } catch (\Exception $e) {
- $this->logger->error('Exception during vcard photo parsing', [
- 'exception' => $e
- ]);
- }
- return false;
- }
-
- public function getPhotoFromVObject(Document $vObject) {
- try {
- if (!$vObject->PHOTO) {
- return false;
- }
- $photo = $vObject->PHOTO;
- $val = $photo->getValue();
-
- if ($photo->getValueType() === 'URI') {
- $parsed = \Sabre\URI\parse($val);
-
- if ($parsed['scheme'] !== 'data') {
- return false;
- }
- if (substr_count($parsed['path'], ';') === 1) {
- [$type] = explode(';', $parsed['path']);
- }
- $val = file_get_contents($val);
- } else {
-
- $type = $this->getBinaryType($photo);
- }
- if (empty($type) || !isset(self::ALLOWED_CONTENT_TYPES[$type])) {
- $type = 'application/octet-stream';
- }
- return [
- 'Content-Type' => $type,
- 'body' => $val
- ];
- } catch (\Exception $e) {
- $this->logger->error('Exception during vcard photo parsing', [
- 'exception' => $e
- ]);
- }
- return false;
- }
- private function readCard(string $cardData): Document {
- return Reader::read($cardData);
- }
-
- private function getBinaryType(Binary $photo) {
- $params = $photo->parameters();
- if (isset($params['TYPE']) || isset($params['MEDIATYPE'])) {
-
- $typeParam = isset($params['TYPE']) ? $params['TYPE'] : $params['MEDIATYPE'];
- $type = (string)$typeParam->getValue();
- if (str_starts_with($type, 'image/')) {
- return $type;
- } else {
- return 'image/' . strtolower($type);
- }
- }
- return '';
- }
-
- public function delete($addressBookId, $cardUri) {
- try {
- $folder = $this->getFolder($addressBookId, $cardUri, false);
- $folder->delete();
- } catch (NotFoundException $e) {
-
- }
- }
- }
|