friendica-addons/birdavatar/birdavatar.php
2022-10-20 23:51:49 +02:00

274 lines
7.3 KiB
PHP

<?php
/**
* Name: Bird Avatar Generator
* Description: Generate a default avatar based on David Revoy's bird-avatar-generator https://www.peppercarrot.com/extras/html/2019_bird-generator/index.php
* Version: 1.0
* Author: Fabio <https://kirgroup.com/profile/fabrixxm>
*/
use Friendica\App;
use Friendica\Core\Hook;
use Friendica\Core\Logger;
use Friendica\Core\Renderer;
use Friendica\Database\DBA;
use Friendica\DI;
use Friendica\Model\Contact;
use Friendica\Model\Photo;
use Friendica\Model\Profile;
use Friendica\Network\HTTPException\NotFoundException;
define("BIRDAVATAR_SIZE", 256);
/**
* Installs the addon hook
*/
function birdavatar_install()
{
Hook::register('avatar_lookup', __FILE__, 'birdavatar_lookup');
Hook::register('addon_settings', __FILE__, 'birdavatar_addon_settings');
Hook::register('addon_settings_post', __FILE__, 'birdavatar_addon_settings_post');
Logger::info('registered birdavatar');
}
/**
* Bird avatar user settings page
*/
function birdavatar_addon_settings(App $a, array &$data)
{
if (!DI::userSession()->getLocalUserId()) {
return;
}
$t = Renderer::getMarkupTemplate('settings.tpl', 'addon/birdavatar/');
$html = Renderer::replaceMacros($t, [
'$uncache' => time(),
'$uid' => DI::userSession()->getLocalUserId(),
'$setrandomize' => DI::l10n()->t('Set default profile avatar or randomize the bird.'),
]);
$data = [
'addon' => 'birdavar',
'title' => DI::l10n()->t('Bird Avatar Settings'),
'html' => $html,
'submit' => [
'birdavatar-usebird' => DI::l10n()->t('Use Bird as Avatar'),
'birdavatar-morebird' => DI::l10n()->t('More Random Bird!'),
'birdavatar-emailbird' => DI::pConfig()->get(DI::userSession()->getLocalUserId(), 'birdavatar', 'seed', false) ? DI::l10n()->t('Reset to email Bird') : null,
],
];
}
/**
* Bird avatar user settings POST handle
*/
function birdavatar_addon_settings_post(App $a, &$s)
{
if (!DI::userSession()->getLocalUserId()) {
return;
}
if (!empty($_POST['birdavatar-usebird'])) {
$url = DI::baseUrl()->get() . '/birdavatar/' . DI::userSession()->getLocalUserId() . '?ts=' . time();
$self = DBA::selectFirst('contact', ['id'], ['uid' => DI::userSession()->getLocalUserId(), 'self' => true]);
if (!DBA::isResult($self)) {
DI::sysmsg()->addNotice(DI::l10n()->t("The bird has not found itself."));
return;
}
Photo::importProfilePhoto($url, DI::userSession()->getLocalUserId(), $self['id']);
$condition = ['uid' => DI::userSession()->getLocalUserId(), 'contact-id' => $self['id']];
$photo = DBA::selectFirst('photo', ['resource-id'], $condition);
if (!DBA::isResult($photo)) {
DI::sysmsg()->addNotice(DI::l10n()->t('There was an error, the bird flew away.'));
return;
}
DBA::update('photo', ['profile' => false], ['profile' => true, 'uid' => DI::userSession()->getLocalUserId()]);
$fields = ['profile' => true, 'album' => DI::l10n()->t('Profile Photos'), 'contact-id' => 0];
DBA::update('photo', $fields, ['uid' => DI::userSession()->getLocalUserId(), 'resource-id' => $photo['resource-id']]);
Photo::importProfilePhoto($url, DI::userSession()->getLocalUserId(), $self['id']);
Contact::updateSelfFromUserID(DI::userSession()->getLocalUserId(), true);
// Update global directory in background
Profile::publishUpdate(DI::userSession()->getLocalUserId());
DI::sysmsg()->addInfo(DI::l10n()->t('Meow!'));
return;
}
if (!empty($_POST['birdavatar-morebird'])) {
DI::pConfig()->set(DI::userSession()->getLocalUserId(), 'birdavatar', 'seed', time());
}
if (!empty($_POST['birdavatar-emailbird'])) {
DI::pConfig()->delete(DI::userSession()->getLocalUserId(), 'birdavatar', 'seed');
}
}
/**
* Returns the URL to the bird avatar
*
* @param $a array
* @param &$b array
*/
function birdavatar_lookup(App $a, array &$b)
{
$user = DBA::selectFirst('user', ['uid'], ['email' => $b['email']]);
if (DBA::isResult($user)) {
$url = DI::baseUrl()->get() . '/birdavatar/' . $user['uid'];
} else {
$url = DI::baseUrl()->get() . '/birdavatar/' . md5(trim(strtolower($b['email'])));
}
switch ($b['size']) {
case 300: $url .= "/4"; break;
case 80: $url .= "/5"; break;
case 48: $url .= "/6"; break;
}
$b['url'] = $url;
$b['success'] = true;
}
/**
* This is a statement rather than an actual function definition. The simple
* existence of this method is checked to figure out if the addon offers a
* module.
*/
function birdavatar_module() {}
/**
* Returns image for user id
*
* @throws NotFoundException
*
*/
function birdavatar_content(App $a)
{
if (DI::args()->getArgc() < 2 || DI::args()->getArgc() > 3) {
throw new NotFoundException(); // this should be catched on index and show default "not found" page.
}
if (is_numeric(DI::args()->getArgv()[1])) {
$uid = intval(DI::args()->getArgv()[1]);
$condition = ['uid' => $uid,
'account_expired' => false, 'account_removed' => false];
$user = DBA::selectFirst('user', ['email'], $condition);
if ($user === false) {
throw new NotFoundException();
}
$seed = DI::pConfig()->get($uid, "birdavatar", "seed", md5(trim(strtolower($user['email']))));
} elseif (!empty(DI::args()->getArgv()[1])) {
$seed = DI::args()->getArgv()[1];
} else {
throw new NotFoundException();
}
$size = 0;
if (DI::args()->getArgc() == 3) {
$size = intval(DI::args()->getArgv()[2]);
}
// start generation
ob_start();
// render the picture:
build_bird($seed, $size);
ob_end_flush();
exit();
}
/**
* ====================
* BIRD-AVATAR-GENERATOR
* ====================
*
* @authors: Andreas Gohr, David Revoy
*
* This PHP is licensed under the short and simple permissive:
* [MIT License](https://en.wikipedia.org/wiki/MIT_License)
*
**/
function build_bird($seed = '', $size = 0)
{
// init random seed
if ($seed) {
srand(hexdec(substr(md5($seed), 0, 6)));
}
// throw the dice for body parts
$parts = [
'tail' => rand(1,9),
'hoop' => rand(1,10),
'body' => rand(1,9),
'wing' => rand(1,9),
'eyes' => rand(1,9),
'bec' => rand(1,9),
'accessorie' => rand(1,20)
];
// create backgound
$bird = @imagecreatetruecolor(BIRDAVATAR_SIZE, BIRDAVATAR_SIZE)
or die("GD image create failed");
$white = imagecolorallocate($bird, 255, 255, 255);
imagefill($bird, 0, 0, $white);
// add parts
foreach ($parts as $part => $num) {
$file = dirname(__FILE__) . '/avatars/' . $part . '_' . $num . '.png';
$im = @imagecreatefrompng($file);
if (!$im) {
die('Failed to load ' . $file);
}
imageSaveAlpha($im, true);
imagecopy($bird, $im, 0, 0, 0, 0, BIRDAVATAR_SIZE, BIRDAVATAR_SIZE);
imagedestroy($im);
}
// scale image
if ($size > 3 && $size < 7) {
switch ($size) {
case 4:
$size = 300;
break;
case 5:
$size = 80;
break;
case 6:
$size = 48;
break;
}
$dest = imagecreatetruecolor($size, $size) or die("GD image create failed");
imagealphablending($dest, false);
imagesavealpha($dest, true);
imagecopyresampled($dest, $bird, 0, 0, 0, 0, $size, $size, BIRDAVATAR_SIZE, BIRDAVATAR_SIZE);
imagedestroy($bird);
$bird = $dest;
}
// restore random seed
if ($seed) {
srand();
}
header('Pragma: public');
header('Cache-Control: max-age=86400');
header('Expires: ' . gmdate('D, d M Y H:i:s \G\M\T', time() + 86400));
header('Content-Type: image/jpg');
imagejpeg($bird, null, 90);
imagedestroy($bird);
}