123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371 |
- <?php
- namespace OC\Core\Controller;
- use OC\AppFramework\Utility\TimeFactory;
- use OCP\AppFramework\Controller;
- use OCP\AppFramework\Http;
- use OCP\AppFramework\Http\Attribute\FrontpageRoute;
- use OCP\AppFramework\Http\DataDisplayResponse;
- use OCP\AppFramework\Http\FileDisplayResponse;
- use OCP\AppFramework\Http\JSONResponse;
- use OCP\Files\File;
- use OCP\Files\IRootFolder;
- use OCP\IAvatarManager;
- use OCP\ICache;
- use OCP\IL10N;
- use OCP\IRequest;
- use OCP\IUserManager;
- use Psr\Log\LoggerInterface;
- class AvatarController extends Controller {
- public function __construct(
- string $appName,
- IRequest $request,
- protected IAvatarManager $avatarManager,
- protected ICache $cache,
- protected IL10N $l10n,
- protected IUserManager $userManager,
- protected IRootFolder $rootFolder,
- protected LoggerInterface $logger,
- protected ?string $userId,
- protected TimeFactory $timeFactory,
- ) {
- parent::__construct($appName, $request);
- }
-
-
- public function getAvatarDark(string $userId, int $size) {
- if ($size <= 64) {
- if ($size !== 64) {
- $this->logger->debug('Avatar requested in deprecated size ' . $size);
- }
- $size = 64;
- } else {
- if ($size !== 512) {
- $this->logger->debug('Avatar requested in deprecated size ' . $size);
- }
- $size = 512;
- }
- try {
- $avatar = $this->avatarManager->getAvatar($userId);
- $avatarFile = $avatar->getFile($size, true);
- $response = new FileDisplayResponse(
- $avatarFile,
- Http::STATUS_OK,
- ['Content-Type' => $avatarFile->getMimeType(), 'X-NC-IsCustomAvatar' => (int)$avatar->isCustomAvatar()]
- );
- } catch (\Exception $e) {
- return new JSONResponse([], Http::STATUS_NOT_FOUND);
- }
-
- $response->cacheFor(60 * 60 * 24, false, true);
- return $response;
- }
-
-
- public function getAvatar(string $userId, int $size) {
- if ($size <= 64) {
- if ($size !== 64) {
- $this->logger->debug('Avatar requested in deprecated size ' . $size);
- }
- $size = 64;
- } else {
- if ($size !== 512) {
- $this->logger->debug('Avatar requested in deprecated size ' . $size);
- }
- $size = 512;
- }
- try {
- $avatar = $this->avatarManager->getAvatar($userId);
- $avatarFile = $avatar->getFile($size);
- $response = new FileDisplayResponse(
- $avatarFile,
- Http::STATUS_OK,
- ['Content-Type' => $avatarFile->getMimeType(), 'X-NC-IsCustomAvatar' => (int)$avatar->isCustomAvatar()]
- );
- } catch (\Exception $e) {
- return new JSONResponse([], Http::STATUS_NOT_FOUND);
- }
-
- $response->cacheFor(60 * 60 * 24, false, true);
- return $response;
- }
-
-
- public function postAvatar(?string $path = null): JSONResponse {
- $files = $this->request->getUploadedFile('files');
- if (isset($path)) {
- $path = stripslashes($path);
- $userFolder = $this->rootFolder->getUserFolder($this->userId);
-
- $node = $userFolder->get($path);
- if (!($node instanceof File)) {
- return new JSONResponse(['data' => ['message' => $this->l10n->t('Please select a file.')]]);
- }
- if ($node->getSize() > 20 * 1024 * 1024) {
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('File is too big')]],
- Http::STATUS_BAD_REQUEST
- );
- }
- if ($node->getMimeType() !== 'image/jpeg' && $node->getMimeType() !== 'image/png') {
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('The selected file is not an image.')]],
- Http::STATUS_BAD_REQUEST
- );
- }
- try {
- $content = $node->getContent();
- } catch (\OCP\Files\NotPermittedException $e) {
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('The selected file cannot be read.')]],
- Http::STATUS_BAD_REQUEST
- );
- }
- } elseif (!is_null($files)) {
- if (
- $files['error'][0] === 0 &&
- is_uploaded_file($files['tmp_name'][0]) &&
- !\OC\Files\Filesystem::isFileBlacklisted($files['tmp_name'][0])
- ) {
- if ($files['size'][0] > 20 * 1024 * 1024) {
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('File is too big')]],
- Http::STATUS_BAD_REQUEST
- );
- }
- $this->cache->set('avatar_upload', file_get_contents($files['tmp_name'][0]), 7200);
- $content = $this->cache->get('avatar_upload');
- unlink($files['tmp_name'][0]);
- } else {
- $phpFileUploadErrors = [
- UPLOAD_ERR_OK => $this->l10n->t('The file was uploaded'),
- UPLOAD_ERR_INI_SIZE => $this->l10n->t('The uploaded file exceeds the upload_max_filesize directive in php.ini'),
- UPLOAD_ERR_FORM_SIZE => $this->l10n->t('The uploaded file exceeds the MAX_FILE_SIZE directive that was specified in the HTML form'),
- UPLOAD_ERR_PARTIAL => $this->l10n->t('The file was only partially uploaded'),
- UPLOAD_ERR_NO_FILE => $this->l10n->t('No file was uploaded'),
- UPLOAD_ERR_NO_TMP_DIR => $this->l10n->t('Missing a temporary folder'),
- UPLOAD_ERR_CANT_WRITE => $this->l10n->t('Could not write file to disk'),
- UPLOAD_ERR_EXTENSION => $this->l10n->t('A PHP extension stopped the file upload'),
- ];
- $message = $phpFileUploadErrors[$files['error'][0]] ?? $this->l10n->t('Invalid file provided');
- $this->logger->warning($message, ['app' => 'core']);
- return new JSONResponse(
- ['data' => ['message' => $message]],
- Http::STATUS_BAD_REQUEST
- );
- }
- } else {
-
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('No image or file provided')]],
- Http::STATUS_BAD_REQUEST
- );
- }
- try {
- $image = new \OCP\Image();
- $image->loadFromData($content);
- $image->readExif($content);
- $image->fixOrientation();
- if ($image->valid()) {
- $mimeType = $image->mimeType();
- if ($mimeType !== 'image/jpeg' && $mimeType !== 'image/png') {
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('Unknown filetype')]],
- Http::STATUS_OK
- );
- }
- if ($image->width() === $image->height()) {
- try {
- $avatar = $this->avatarManager->getAvatar($this->userId);
- $avatar->set($image);
-
- $this->cache->remove('tmpAvatar');
- return new JSONResponse(['status' => 'success']);
- } catch (\Throwable $e) {
- $this->logger->error($e->getMessage(), ['exception' => $e, 'app' => 'core']);
- return new JSONResponse(['data' => ['message' => $this->l10n->t('An error occurred. Please contact your admin.')]], Http::STATUS_BAD_REQUEST);
- }
- }
- $this->cache->set('tmpAvatar', $image->data(), 7200);
- return new JSONResponse(
- ['data' => 'notsquare'],
- Http::STATUS_OK
- );
- } else {
- return new JSONResponse(
- ['data' => ['message' => $this->l10n->t('Invalid image')]],
- Http::STATUS_OK
- );
- }
- } catch (\Exception $e) {
- $this->logger->error($e->getMessage(), ['exception' => $e, 'app' => 'core']);
- return new JSONResponse(['data' => ['message' => $this->l10n->t('An error occurred. Please contact your admin.')]], Http::STATUS_OK);
- }
- }
-
-
- public function deleteAvatar(): JSONResponse {
- try {
- $avatar = $this->avatarManager->getAvatar($this->userId);
- $avatar->remove();
- return new JSONResponse();
- } catch (\Exception $e) {
- $this->logger->error($e->getMessage(), ['exception' => $e, 'app' => 'core']);
- return new JSONResponse(['data' => ['message' => $this->l10n->t('An error occurred. Please contact your admin.')]], Http::STATUS_BAD_REQUEST);
- }
- }
-
-
- public function getTmpAvatar() {
- $tmpAvatar = $this->cache->get('tmpAvatar');
- if (is_null($tmpAvatar)) {
- return new JSONResponse(['data' => [
- 'message' => $this->l10n->t("No temporary profile picture available, try again")
- ]],
- Http::STATUS_NOT_FOUND);
- }
- $image = new \OCP\Image();
- $image->loadFromData($tmpAvatar);
- $resp = new DataDisplayResponse(
- $image->data() ?? '',
- Http::STATUS_OK,
- ['Content-Type' => $image->mimeType()]);
- $resp->setETag((string)crc32($image->data() ?? ''));
- $resp->cacheFor(0);
- $resp->setLastModified(new \DateTime('now', new \DateTimeZone('GMT')));
- return $resp;
- }
-
-
- public function postCroppedAvatar(?array $crop = null): JSONResponse {
- if (is_null($crop)) {
- return new JSONResponse(['data' => ['message' => $this->l10n->t("No crop data provided")]],
- Http::STATUS_BAD_REQUEST);
- }
- if (!isset($crop['x'], $crop['y'], $crop['w'], $crop['h'])) {
- return new JSONResponse(['data' => ['message' => $this->l10n->t("No valid crop data provided")]],
- Http::STATUS_BAD_REQUEST);
- }
- $tmpAvatar = $this->cache->get('tmpAvatar');
- if (is_null($tmpAvatar)) {
- return new JSONResponse(['data' => [
- 'message' => $this->l10n->t("No temporary profile picture available, try again")
- ]],
- Http::STATUS_BAD_REQUEST);
- }
- $image = new \OCP\Image();
- $image->loadFromData($tmpAvatar);
- $image->crop($crop['x'], $crop['y'], (int)round($crop['w']), (int)round($crop['h']));
- try {
- $avatar = $this->avatarManager->getAvatar($this->userId);
- $avatar->set($image);
-
- $this->cache->remove('tmpAvatar');
- return new JSONResponse(['status' => 'success']);
- } catch (\OC\NotSquareException $e) {
- return new JSONResponse(['data' => ['message' => $this->l10n->t('Crop is not square')]],
- Http::STATUS_BAD_REQUEST);
- } catch (\Exception $e) {
- $this->logger->error($e->getMessage(), ['exception' => $e, 'app' => 'core']);
- return new JSONResponse(['data' => ['message' => $this->l10n->t('An error occurred. Please contact your admin.')]], Http::STATUS_BAD_REQUEST);
- }
- }
- }
|