spreed/lib/Controller/AvatarController.php
Vitor Mattos d014b4de14
Remove endpoint to get dark avatar, not necessary
Signed-off-by: Vitor Mattos <vitor@php.rio>
2022-12-07 15:00:29 +01:00

119 lines
3.4 KiB
PHP

<?php
declare(strict_types=1);
/**
* @copyright Copyright (c) 2022, Vitor Mattos <vitor@php.rio>
*
* @author Vitor Mattos <vitor@php.rio>
*
* @license GNU AGPL version 3 or any later version
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
namespace OCA\Talk\Controller;
use InvalidArgumentException;
use OC\Files\Filesystem;
use OCA\Talk\Service\AvatarService;
use OCP\AppFramework\Http;
use OCP\AppFramework\Http\DataResponse;
use OCP\AppFramework\Http\FileDisplayResponse;
use OCP\AppFramework\Http\Response;
use OCP\IL10N;
use OCP\IRequest;
use OCP\IUserSession;
class AvatarController extends AEnvironmentAwareController {
private AvatarService $avatarService;
private IUserSession $userSession;
private IL10N $l;
public function __construct(
string $appName,
IRequest $request,
AvatarService $avatarService,
IUserSession $userSession,
IL10N $l10n
) {
parent::__construct($appName, $request);
$this->avatarService = $avatarService;
$this->userSession = $userSession;
$this->l = $l10n;
}
/**
* @PublicPage
* @RequireModeratorParticipant
*/
public function uploadAvatar(): DataResponse {
try {
$file = $this->request->getUploadedFile('file');
if (is_null($file)) {
throw new InvalidArgumentException($this->l->t('No image file provided'));
}
if (
$file['error'] !== 0 ||
!is_uploaded_file($file['tmp_name']) ||
Filesystem::isFileBlacklisted($file['tmp_name'])
) {
throw new InvalidArgumentException($this->l->t('Invalid file provided'));
}
if ($file['size'] > 20 * 1024 * 1024) {
throw new InvalidArgumentException($this->l->t('File is too big'));
}
$content = file_get_contents($file['tmp_name']);
unlink($file['tmp_name']);
$this->avatarService->setAvatar($this->getRoom(), $content);
return new DataResponse();
} catch (InvalidArgumentException $e) {
return new DataResponse(['message' => $e->getMessage()], Http::STATUS_BAD_REQUEST);
} catch (\Exception $e) {
$this->logger->error('Failed to post avatar', [
'exception' => $e,
]);
return new DataResponse(['message' => $this->l->t('An error occurred. Please contact your administrator.')], Http::STATUS_BAD_REQUEST);
}
}
/**
* @PublicPage
* @NoCSRFRequired
* @RequireParticipant
*/
public function getAvatar(): Response {
$file = $this->avatarService->getAvatar($this->getRoom(), $this->userSession->getUser());
$response = new FileDisplayResponse($file);
$response->addHeader('Content-Type', $file->getMimeType());
// Cache for 1 day
$response->cacheFor(60 * 60 * 24, false, true);
return $response;
}
/**
* @PublicPage
* @RequireModeratorParticipant
*/
public function deleteAvatar(): DataResponse {
$this->avatarService->deleteAvatar($this->getRoom());
return new DataResponse([]);
}
}