upstream/nextcloud
1
0
Fork 0
mirror of https://github.com/nextcloud/server.git synced 2026-02-20 00:12:30 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions 38

Avatar export and import

Browse source 
Signed-off-by: Christopher Ng <chrng8@gmail.com>
This commit is contained in:
Christopher Ng 2022-03-01 02:44:30 +00:00
parent 10db84caca
commit 7bdb15208d
1 changed files with 50 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
lib/private/Accounts/AccountMigrator.php
View file

@ -27,28 +27,45 @@ declare(strict_types=1);
namespace OC\Accounts;
use InvalidArgumentException;
use OC\NotSquareException;
use OCP\Accounts\IAccountManager;
use OCP\IAvatarManager;
use OCP\IUser;
use OCP\UserMigration\IExportDestination;
use OCP\UserMigration\IImportSource;
use OCP\UserMigration\IMigrator;
use OCP\UserMigration\TMigratorBasicVersionHandling;
use Symfony\Component\Console\Output\OutputInterface;
use Throwable;
class AccountMigrator implements IMigrator {
use TMigratorBasicVersionHandling;
use TAccountsHelper;
private IAccountManager $accountManager;
private IAvatarManager $avatarManager;
private const EXPORT_FILE = 'account.json';
public function __construct(
IAccountManager $accountManager
IAccountManager $accountManager,
IAvatarManager $avatarManager
) {
$this->accountManager = $accountManager;
$this->avatarManager = $avatarManager;
}
private function getExtension(string $mimeType): string {
switch ($mimeType) {
case 'image/jpeg':
return 'jpg';
case 'image/png':
return 'png';
default:
throw new AccountMigratorException("Invalid avatar mimetype: \"$mimeType\"");
}
}
/**
@ -60,6 +77,16 @@ class AccountMigrator implements IMigrator {
if ($exportDestination->addFileContents(AccountMigrator::EXPORT_FILE, json_encode($this->accountManager->getAccount($user))) === false) {
throw new AccountMigratorException('Could not export account information');
}
$avatar = $this->avatarManager->getAvatar($user->getUID());
if ($avatar->isCustomAvatar()) {
$avatarData = $avatar->get(-1)->data();
$ext = $this->getExtension($avatar->get(-1)->dataMimeType());
$output->writeln('Exporting avatar to avatar.' . $ext . '…');
if ($exportDestination->addFileContents("avatar.$ext", $avatarData) === false) {
throw new AccountMigratorException('Could not export avatar');
}
}
}
/**
@ -99,5 +126,26 @@ class AccountMigrator implements IMigrator {
} catch (InvalidArgumentException $e) {
throw new AccountMigratorException('Failed to import account information');
}
foreach ($importSource->getFolderListing('') as $filename) {
if (str_starts_with($filename, 'avatar.')) {
$avatarFilename = $filename;
}
}
if (isset($avatarFilename)) {
$output->writeln('Importing avatar from ' . $avatarFilename . '…');
$avatar = $importSource->getFileContents($avatarFilename);
$image = new \OC_Image();
$image->loadFromData($avatar);
try {
$avatar = $this->avatarManager->getAvatar($user->getUID());
$avatar->set($image);
} catch (NotSquareException $e) {
throw new AccountMigratorException('Avatar image must be square');
} catch (Throwable $e) {
throw new AccountMigratorException('Failed to import avatar', 0, $e);
}
}
}
}