plugins/VideoThumbnails/VideoThumbnailsPlugin.php

   1 <?php
   2 /**
   3  * GNU social - a federating social network
   4  *
   5  * Plugin to make thumbnails of video files with ffmpeg/avconv
   6  *
   7  * PHP version 5
   8  *
   9  * LICENCE: This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU Affero General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU Affero General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Affero General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  *
  22  * @category  Plugin
  23  * @package   GNUsocial
  24  * @author    Mikael Nordfeldth <mmn@hethane.se>
  25  * @copyright 2014 Free Software Foundation http://fsf.org
  26  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  27  * @link      https://www.gnu.org/software/social/
  28  */
  29
  30 if (!defined('GNUSOCIAL')) { exit(1); }
  31
  32 /*
  33  * Dependencies:
  34  *  ffmpeg/avconv (external program call)
  35  *  php5-gd
  36  *
  37  * Todo:
  38  *  Make sure we support ffmpeg too, so we're not super Debian oriented.
  39  *
  40  * Video support will depend on your ffmpeg/avconv.
  41  */
  42
  43 class VideoThumbnailsPlugin extends Plugin
  44 {
  45     /*
  46      * This function should only extract an image from the video stream
  47      * and disregard any cropping or scaling in the resulting file, as
  48      * that will be handled in the core thumbnail algorithm.
  49      */
  50     public function onCreateFileImageThumbnailSource(File $file, &$imgPath, $media=null)
  51     {
  52         // The calling function might accidentally pass application/ogg videos.
  53         // If that's a problem, let's fix it in the calling function.
  54         if ($media !== 'video' || empty($file->filename)) {
  55             return true;
  56         }
  57
  58         try {
  59             // Exception thrown if no thumbnail found
  60             $thumb = File_thumbnail::byFile($file, false);
  61             $imgPath = $thumb->getPath();
  62             // If getPath didn't throw an exception, we have a working locally stored thumbnail
  63             return false;
  64         } catch (NoResultException $e) {
  65             // Alright, no thumbnail found, so let's create one.
  66         } catch (InvalidFilenameException $e) {
  67             // I guess this means $thumb->filename is null? Shouldn't happen because $file->filename is not null, so delete it
  68             $thumb->delete();
  69         } catch (FileNotFoundException $e) {
  70             // Thumb file was not found, let's delete it.
  71             $thumb->delete();
  72         }
  73
  74         // Let's save our frame to a temporary file. If we fail, remove it.
  75         $tmp_imgPath = tempnam(sys_get_temp_dir(), 'socialthumb-');
  76
  77         $cmd = null;
  78         if (shell_exec('which ffmpeg')) {
  79             $cmd = 'ffmpeg';
  80         } elseif (shell_exec('which avconv')) {
  81             $cmd = 'avconv';
  82         } else {
  83             common_log(LOG_ERR, 'Neither ffmpeg nor avconv was found in your PATH. Cannot create video thumbnail.');
  84             return true;
  85         }
  86         $fullcmd = $cmd.' -y -i '.escapeshellarg($file->getPath()).' -vcodec mjpeg -vframes 1 -f image2 -an '.escapeshellarg($tmp_imgPath);
  87         common_debug(__METHOD__ . ' executing: '._ve($fullcmd));
  88         $result = exec($fullcmd);
  89
  90         if (!getimagesize($tmp_imgPath)) {
  91             common_debug('exec of "avconv" produced a bad/nonexisting image it seems');
  92             @unlink($tmp_imgPath);
  93             return true;
  94         }
  95         $imgPath = $tmp_imgPath;
  96         return false;
  97     }
  98
  99     public function onPluginVersion(array &$versions)
 100     {
 101         $versions[] = array('name' => 'Video Thumbnails',
 102                             'version' => GNUSOCIAL_VERSION,
 103                             'author' => 'Mikael Nordfeldth',
 104                             'homepage' => 'https://www.gnu.org/software/social/',
 105                             'rawdescription' =>
 106                             // TRANS: Plugin description.
 107                             _m('Video thumbnail preview support.'));
 108         return true;
 109     }
 110 }
 111