gnu-social/components/Avatar/Controller/Avatar.php

131 lines
5.3 KiB
PHP
Raw Normal View History

<?php
// {{{ License
// This file is part of GNU social - https://www.gnu.org/software/social
//
// GNU social 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.
//
// GNU social 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 GNU social. If not, see <http://www.gnu.org/licenses/>.
// }}}
namespace Component\Avatar\Controller;
use App\Core\Controller;
2021-04-30 03:12:32 +09:00
use App\Core\DB\DB;
use App\Core\Event;
use App\Core\Form;
2021-04-30 03:12:32 +09:00
use App\Core\GSFile;
use App\Core\GSFile as M;
use function App\Core\I18n\_m;
use App\Core\Log;
2021-04-30 03:12:32 +09:00
use App\Util\Common;
use App\Util\Exception\ClientException;
2021-04-30 03:12:32 +09:00
use App\Util\Exception\NotFoundException;
use App\Util\TemporaryFile;
use Component\Avatar\Entity\Avatar as AvatarEntity;
use Exception;
use Symfony\Component\Form\Extension\Core\Type\CheckboxType;
use Symfony\Component\Form\Extension\Core\Type\FileType;
use Symfony\Component\Form\Extension\Core\Type\HiddenType;
use Symfony\Component\Form\Extension\Core\Type\SubmitType;
2021-04-30 03:12:32 +09:00
use Symfony\Component\Form\FormError;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
class Avatar extends Controller
{
public function default_avatar_view(Request $request, string $size): Response
{
return $this->avatar_view($request, 0, $size);
}
/**
* @throws Exception
*/
2021-09-18 11:22:27 +09:00
public function avatar_view(Request $request, int $actor_id, string $size): Response
{
switch ($size) {
case 'full':
2021-09-18 11:22:27 +09:00
$res = \Component\Avatar\Avatar::getAvatarFileInfo($actor_id);
return M::sendFile($res['filepath'], $res['mimetype'], $res['title']);
default:
throw new Exception('Not implemented');
}
}
/**
* Local user avatar panel
*/
public static function settings_avatar(Request $request): array
{
$form = Form::create([
['avatar', FileType::class, ['label' => _m('Avatar'), 'help' => _m('You can upload your personal avatar. The maximum file size is 2MB.'), 'multiple' => false, 'required' => false]],
2021-04-30 03:12:32 +09:00
['remove', CheckboxType::class, ['label' => _m('Remove avatar'), 'help' => _m('Remove your avatar and use the default one'), 'required' => false, 'value' => false]],
['hidden', HiddenType::class, []],
['save_avatar', SubmitType::class, ['label' => _m('Submit')]],
]);
$form->handleRequest($request);
if ($form->isSubmitted() && $form->isValid()) {
2021-09-18 11:22:27 +09:00
$data = $form->getData();
$user = Common::user();
$actor_id = $user->getId();
if ($data['remove'] == true) {
2021-04-30 03:12:32 +09:00
try {
2021-09-18 11:22:27 +09:00
$avatar = DB::findOneBy('avatar', ['actor_id' => $actor_id]);
2021-04-30 03:12:32 +09:00
$avatar->delete();
2021-08-05 11:15:07 +09:00
Event::handle('AvatarUpdate', [$user->getId()]);
2021-04-30 03:12:32 +09:00
} catch (NotFoundException) {
$form->addError(new FormError(_m('No avatar set, so cannot delete')));
}
} else {
$attachment = null;
if (isset($data['hidden'])) {
// Cropped client side
$matches = [];
if (!empty(preg_match('/data:([^;]*)(;(base64))?,(.*)/', $data['hidden'], $matches))) {
list(, , , $encoding_user, $data_user) = $matches;
if ($encoding_user === 'base64') {
$data_user = base64_decode($data_user);
$tempfile = new TemporaryFile(['prefix' => 'gs-avatar']);
$tempfile->write($data_user);
$attachment = GSFile::sanitizeAndStoreFileAsAttachment($tempfile);
} else {
Log::info('Avatar upload got an invalid encoding, something\'s fishy and/or wrong');
}
}
} elseif (isset($data['avatar'])) {
// Cropping failed (e.g. disabled js), use file as uploaded
$file = $data['avatar'];
$attachment = GSFile::sanitizeAndStoreFileAsAttachment($file);
} else {
throw new ClientException('Invalid form');
}
2021-08-05 11:15:07 +09:00
// Delete current avatar if there's one
2021-09-18 11:22:27 +09:00
$avatar = DB::find('avatar', ['actor_id' => $actor_id]);
2021-08-05 11:15:07 +09:00
$avatar?->delete();
2021-04-30 03:12:32 +09:00
DB::persist($attachment);
// Can only get new id after inserting
DB::flush();
2021-09-18 11:22:27 +09:00
DB::persist(AvatarEntity::create(['actor_id' => $actor_id, 'attachment_id' => $attachment->getId()]));
DB::flush();
2021-08-05 11:15:07 +09:00
Event::handle('AvatarUpdate', [$user->getId()]);
}
}
return ['_template' => 'settings/avatar.html.twig', 'avatar' => $form->createView()];
}
}