mirror of https://github.com/byecorps/id.git
159 lines
3.8 KiB
PHP
159 lines
3.8 KiB
PHP
<?php
|
|
|
|
use kornrunner\Blurhash\Blurhash;
|
|
|
|
function mime_to_extension($mime) {
|
|
return match ($mime) {
|
|
'image/gif' => '.gif',
|
|
'image/jpeg' => '.jpg',
|
|
'image/png' => '.png',
|
|
'image/webp' => '.webp',
|
|
default => ''
|
|
};
|
|
}
|
|
|
|
/**
|
|
*
|
|
* Squishes an image into a 128x128 square, and converts it to JPEG
|
|
* if it is not a JPEG, PNG, WEBP or GIF.
|
|
*
|
|
* @param string $path
|
|
* @return array
|
|
*/
|
|
function turn_image_into_avatar(string $path): array
|
|
{
|
|
$manager = new \Intervention\Image\ImageManager(
|
|
new \Intervention\Image\Drivers\Gd\Driver()
|
|
);
|
|
|
|
$image = $manager->read($path);
|
|
|
|
// Get mimetype
|
|
$mime = mime_content_type($path);
|
|
$filetype = mime_to_extension($mime);
|
|
|
|
$enc = $image->resize(width: 128, height: 128);
|
|
|
|
if ($filetype=='') {
|
|
$enc = $image->encodeByMediaType('image/webp');
|
|
$filetype = '.webp';
|
|
} else {
|
|
$enc = $image->encodeByMediaType();
|
|
}
|
|
|
|
$image_data = (string) $enc;
|
|
|
|
return [
|
|
"data" => $image_data,
|
|
"mime" => $enc->mimetype(),
|
|
'filetype' => $filetype
|
|
];
|
|
}
|
|
|
|
function get_blurhash_for_image($image): string
|
|
{
|
|
// Copied shamelessly from https://github.com/kornrunner/php-blurhash
|
|
$width = imagesx($image);
|
|
$height = imagesy($image);
|
|
|
|
$pixels = [];
|
|
for ($y = 0; $y < $height; ++$y) {
|
|
$row = [];
|
|
for ($x = 0; $x < $width; ++$x) {
|
|
$index = imagecolorat($image, $x, $y);
|
|
$colors = imagecolorsforindex($image, $index);
|
|
|
|
$row[] = [$colors['red'], $colors['green'], $colors['blue']];
|
|
}
|
|
$pixels[] = $row;
|
|
}
|
|
|
|
$components_x = 4;
|
|
$components_y = 3;
|
|
return Blurhash::encode($pixels, $components_x, $components_y);
|
|
}
|
|
|
|
function upload_file($filename, $target, $owner=null) {
|
|
global $bunny_client;
|
|
|
|
$bunny_client->upload($filename, $target);
|
|
|
|
// Get file mime time
|
|
$mime = mime_content_type($filename);
|
|
|
|
$blurhash = '';
|
|
if (str_starts_with('image/', $mime)) {
|
|
$blurhash = get_blurhash_for_image(imagecreatefromstring(file_get_contents($filename)));
|
|
}
|
|
|
|
db_execute(
|
|
'INSERT INTO files (path, uploader, blurhash) VALUES (?, ?, ?)',
|
|
[$target, $owner, $blurhash]
|
|
);
|
|
}
|
|
|
|
function upload_raw_data($data, $target, $owner=null) {
|
|
$filename = '/tmp/'.uniqid(more_entropy: true);
|
|
file_put_contents($filename, $data);
|
|
|
|
upload_file($filename, $target, $owner);
|
|
|
|
unlink($filename);
|
|
}
|
|
|
|
/**
|
|
* @throws \Bunny\Storage\Exception
|
|
* @throws ImagickException
|
|
*/
|
|
function upload_avatar($data, $user): string
|
|
{
|
|
global $bunny_client;
|
|
|
|
$img = turn_image_into_avatar($data['tmp_name']);
|
|
|
|
$uuid = uniqid(prefix: 'avatar', more_entropy: true);
|
|
|
|
$remote_file_name = 'avatars/'.$uuid.$img['filetype'];
|
|
|
|
upload_raw_data($img['data'], $remote_file_name);
|
|
|
|
$file = db_execute('SELECT id FROM files WHERE path = ? LIMIT 1', [$remote_file_name]);
|
|
|
|
$existing_avatar = db_execute('SELECT * FROM avatars WHERE owner = ?', [$user['id']]);
|
|
|
|
if (empty($existing_avatar)) {
|
|
db_execute(
|
|
'INSERT INTO avatars (file_id, owner) VALUES (?, ?)',
|
|
[$file['id'], $user['id']]
|
|
);
|
|
} else {
|
|
db_execute(
|
|
'UPDATE avatars SET file_id = ? WHERE owner = ?',
|
|
[$file['id'], $user['id']]
|
|
);
|
|
}
|
|
|
|
return 'https://'.BUNNY_STORAGE_ZONE.'.b-cdn.net/'.$remote_file_name;
|
|
}
|
|
|
|
/**
|
|
* @param $id int
|
|
* @return void
|
|
*/
|
|
function delete_file_by_id(int $id): void
|
|
{
|
|
global $bunny_client;
|
|
|
|
// Get remote path
|
|
$file = db_execute('select * from files where id = ?', [$id]);
|
|
|
|
if (empty($file)) {
|
|
return;
|
|
}
|
|
|
|
// Remove reference to the file in the database
|
|
db_execute('delete from files where id = ?', [$id]);
|
|
// Then remove the file
|
|
$bunny_client->delete($file['path']);
|
|
}
|