gnu-social/plugins/FFmpeg/FFmpegPlugin.php
Alexei Sorokin 08145f635f Implement a class for automatic temporary file handling
And adopt it all over the code.
2020-09-04 13:15:23 +03:00

154 lines
5.0 KiB
PHP

<?php
// 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/>.
/**
* Animated GIF resize support via PHP-FFMpeg
*
* @package GNUsocial
* @author Bruno Casteleiro <up201505347@fc.up.pt>
* @copyright 2020 Free Software Foundation, Inc http://www.fsf.org
* @license https://www.gnu.org/licenses/agpl.html GNU AGPL v3 or later
* @link http://www.gnu.org/software/social/
*/
defined('GNUSOCIAL') || die();
class FFmpegPlugin extends Plugin
{
const PLUGIN_VERSION = '0.1.0';
public function onStartResizeImageFile(
ImageFile $imagefile,
string $outpath,
array $box
): bool {
switch ($imagefile->mimetype) {
case 'image/gif':
// resize only if an animated GIF
if ($imagefile->animated) {
return !$this->resizeImageFileAnimatedGif($imagefile, $outpath, $box);
}
break;
}
return true;
}
/**
* High quality GIF conversion.
*
* @see http://blog.pkh.me/p/21-high-quality-gif-with-ffmpeg.html
* @see https://github.com/PHP-FFMpeg/PHP-FFMpeg/pull/592
*/
public function resizeImageFileAnimatedGif(ImageFile $imagefile, string $outpath, array $box): bool
{
// Create FFMpeg instance
// Need to explictly tell the drivers location or it won't find them
$ffmpeg = FFMpeg\FFMpeg::create([
'ffmpeg.binaries' => exec("which ffmpeg"),
'ffprobe.binaries' => exec("which ffprobe")
]);
// FFmpeg can't edit existing files in place,
// generate temporary output file to avoid that
$tempfile = new TemporaryFile('gs-outpath');
// Generate palette file. FFmpeg explictly needs to be told the
// extension for PNG files outputs
$palette = $this->tempnam_sfx(sys_get_temp_dir(), '.png');
// Build filters
$filters = 'fps=30';
$filters .= ",crop={$box['w']}:{$box['h']}:{$box['x']}:{$box['y']}";
$filters .= ",scale={$box['width']}:{$box['height']}:flags=lanczos";
// Assemble commands for palette generation
$commands[] = $commands_2[] = '-f';
$commands[] = $commands_2[] = 'gif';
$commands[] = $commands_2[] = '-i';
$commands[] = $commands_2[] = $imagefile->filepath;
$commands[] = '-vf';
$commands[] = $filters . ',palettegen';
$commands[] = '-y';
$commands[] = $palette;
// Assemble commands for GIF generation
$commands_2[] = '-i';
$commands_2[] = $palette;
$commands_2[] = '-lavfi';
$commands_2[] = $filters . ' [x]; [x][1:v] paletteuse';
$commands_2[] = '-f';
$commands_2[] = 'gif';
$commands_2[] = '-y';
$commands_2[] = $tempfile->getRealPath();
$success = true;
// Generate the palette image
try {
$ffmpeg->getFFMpegDriver()->command($commands);
} catch (Exception $e) {
$this->log(LOG_ERR, 'Unable to generate the palette image');
$success = false;
}
// Generate GIF
try {
if ($success) {
$ffmpeg->getFFMpegDriver()->command($commands_2);
}
} catch (Exception $e) {
$this->log(LOG_ERR, 'Unable to generate the GIF image');
$success = false;
}
if ($success) {
$success = $tempfile->commit($outpath);
}
@unlink($palette);
return $success;
}
/**
* Suffix version of tempnam.
* Courtesy of tomas at slax dot org:
* @see https://www.php.net/manual/en/function.tempnam.php#98232
*/
private function tempnam_sfx(string $dir, string $suffix): string
{
do {
$file = $dir . "/" . mt_rand() . $suffix;
$fp = @fopen($file, 'x');
} while (!$fp);
fclose($fp);
return $file;
}
public function onPluginVersion(array &$versions): bool
{
$versions[] = ['name' => 'FFmpeg',
'version' => self::PLUGIN_VERSION,
'author' => 'Bruno Casteleiro',
'homepage' => 'https://notabug.org/diogo/gnu-social/src/nightly/plugins/FFmpeg',
'rawdescription' =>
// TRANS: Plugin description.
_m('Use PHP-FFMpeg for resizing animated GIFs')];
return true;
}
}