classes/File_thumbnail.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2008, 2009, StatusNet, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.     See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.     If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 if (!defined('GNUSOCIAL')) { exit(1); }
  21
  22 /**
  23  * Table Definition for file_thumbnail
  24  */
  25
  26 class File_thumbnail extends Managed_DataObject
  27 {
  28     public $__table = 'file_thumbnail';                  // table name
  29     public $file_id;                         // int(4)  primary_key not_null
  30     public $urlhash;                         // varchar(64) indexed
  31     public $url;                             // text
  32     public $filename;                        // text
  33     public $width;                           // int(4)  primary_key
  34     public $height;                          // int(4)  primary_key
  35     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
  36
  37     const URLHASH_ALG = 'sha256';
  38
  39     public static function schemaDef()
  40     {
  41         return array(
  42             'fields' => array(
  43                 'file_id' => array('type' => 'int', 'not null' => true, 'description' => 'thumbnail for what URL/file'),
  44                 'urlhash' => array('type' => 'varchar', 'length' => 64, 'description' => 'sha256 of url field if non-empty'),
  45                 'url' => array('type' => 'text', 'description' => 'URL of thumbnail'),
  46                 'filename' => array('type' => 'text', 'description' => 'if stored locally, filename is put here'),
  47                 'width' => array('type' => 'int', 'description' => 'width of thumbnail'),
  48                 'height' => array('type' => 'int', 'description' => 'height of thumbnail'),
  49                 'modified' => array('type' => 'timestamp', 'not null' => true, 'description' => 'date this record was modified'),
  50             ),
  51             'primary key' => array('file_id', 'width', 'height'),
  52             'indexes' => array(
  53                 'file_thumbnail_file_id_idx' => array('file_id'),
  54                 'file_thumbnail_urlhash_idx' => array('urlhash'),
  55             ),
  56             'foreign keys' => array(
  57                 'file_thumbnail_file_id_fkey' => array('file', array('file_id' => 'id')),
  58             )
  59         );
  60     }
  61
  62     /**
  63      * Save oEmbed-provided thumbnail data
  64      *
  65      * @param object $data
  66      * @param int $file_id
  67      */
  68     public static function saveNew($data, $file_id) {
  69         if (!empty($data->thumbnail_url)) {
  70             // Non-photo types such as video will usually
  71             // show us a thumbnail, though it's not required.
  72             self::saveThumbnail($file_id,
  73                                 $data->thumbnail_url,
  74                                 $data->thumbnail_width,
  75                                 $data->thumbnail_height);
  76         } else if ($data->type == 'photo') {
  77             // The inline photo URL given should also fit within
  78             // our requested thumbnail size, per oEmbed spec.
  79             self::saveThumbnail($file_id,
  80                                 $data->url,
  81                                 $data->width,
  82                                 $data->height);
  83         }
  84     }
  85
  86     /**
  87      * Fetch an entry by using a File's id
  88      *
  89      * @param   File    $file       The File object we're getting a thumbnail for.
  90      * @param   boolean $notNullUrl Originally remote thumbnails have a URL stored, we use this to find the "original"
  91      *
  92      * @return  File_thumbnail
  93      * @throws  NoResultException if no File_thumbnail matched the criteria
  94      */
  95     static function byFile(File $file, $notNullUrl=true) {
  96         $thumb = new File_thumbnail();
  97         $thumb->file_id = $file->getID();
  98         if ($notNullUrl) {
  99             $thumb->whereAdd('url IS NOT NULL');
 100         }
 101         $thumb->limit(1);
 102         if (!$thumb->find(true)) {
 103             throw new NoResultException($thumb);
 104         }
 105         return $thumb;
 106     }
 107
 108     /**
 109      * Save a thumbnail record for the referenced file record.
 110      *
 111      * FIXME: Add error handling
 112      *
 113      * @param int $file_id
 114      * @param string $url
 115      * @param int $width
 116      * @param int $height
 117      */
 118     static function saveThumbnail($file_id, $url, $width, $height, $filename=null)
 119     {
 120         $tn = new File_thumbnail;
 121         $tn->file_id = $file_id;
 122         $tn->url = $url;
 123         $tn->filename = $filename;
 124         $tn->width = intval($width);
 125         $tn->height = intval($height);
 126         $tn->insert();
 127         return $tn;
 128     }
 129
 130     static function path($filename)
 131     {
 132         if (!File::validFilename($filename)) {
 133             // TRANS: Client exception thrown if a file upload does not have a valid name.
 134             throw new ClientException(_('Invalid filename.'));
 135         }
 136
 137         $dir = common_config('thumbnail', 'dir') ?: File::path('thumb');
 138
 139         if (!in_array($dir[mb_strlen($dir)-1], ['/', '\\'])) {
 140             $dir .= DIRECTORY_SEPARATOR;
 141         }
 142
 143         return $dir . $filename;
 144     }
 145
 146     static function url($filename)
 147     {
 148         if (!File::validFilename($filename)) {
 149             // TRANS: Client exception thrown if a file upload does not have a valid name.
 150             throw new ClientException(_('Invalid filename.'));
 151         }
 152
 153         // FIXME: private site thumbnails?
 154
 155         $path = common_config('thumbnail', 'path');
 156         if (empty($path)) {
 157             return File::url('thumb')."/{$filename}";
 158         }
 159
 160         $protocol = (GNUsocial::useHTTPS() ? 'https' : 'http');
 161         $server = common_config('thumbnail', 'server') ?: common_config('site', 'server');
 162
 163         if ($path[mb_strlen($path)-1] != '/') {
 164             $path .= '/';
 165         }
 166         if ($path[0] != '/') {
 167             $path = '/'.$path;
 168         }
 169
 170         return $protocol.'://'.$server.$path.$filename;
 171     }
 172
 173     public function getFilename()
 174     {
 175         if (!File::validFilename($this->filename)) {
 176             // TRANS: Client exception thrown if a file upload does not have a valid name.
 177             throw new ClientException(_("Invalid filename."));
 178         }
 179         return $this->filename;
 180     }
 181
 182     public function getPath()
 183     {
 184         $oldpath = File::path($this->getFilename());
 185         $thumbpath = self::path($this->getFilename());
 186
 187         // If we have a file in our old thumbnail storage path, move it to the new one
 188         if (file_exists($oldpath) && !file_exists($thumbpath)) {
 189             if ($this->getFilename() === $this->getFile()->filename) {
 190                 // special case where thumbnail file exactly matches stored File
 191                 common_debug('File filename and File_thumbnail filename match on '.$this->file_id);
 192             } elseif (!rename($oldpath, $thumbpath)) {
 193                 common_log(LOG_ERR, 'Could not move thumbnail from '._ve($oldpath).' to '._ve($thumbpath));
 194                 throw new ServerException('Could not move thumbnail from old path to new path.');
 195             } else {
 196                 common_log(LOG_DEBUG, 'Moved thumbnail '.$this->file_id.' from '._ve($oldpath).' to '._ve($thumbpath));
 197             }
 198         } elseif (!file_exists($thumbpath)) {
 199             throw new FileNotFoundException($thumbpath);
 200         }
 201         return $thumbpath;
 202     }
 203
 204     public function getUrl()
 205     {
 206         if (!empty($this->filename) || $this->getFile()->isLocal()) {
 207             // A locally stored File, so we can dynamically generate a URL.
 208             $url = common_local_url('attachment_thumbnail', array('attachment'=>$this->file_id));
 209             if (strpos($url, '?') === false) {
 210                 $url .= '?';
 211             }
 212             return $url . http_build_query(array('w'=>$this->width, 'h'=>$this->height));
 213         }
 214
 215         // No local filename available, return the remote URL we have stored
 216         return $this->url;
 217     }
 218
 219     public function getHeight()
 220     {
 221         return $this->height;
 222     }
 223
 224     public function getWidth()
 225     {
 226         return $this->width;
 227     }
 228
 229     /**
 230      * @throws UseFileAsThumbnailException from File_thumbnail->getUrl() for stuff like animated GIFs
 231      */
 232     public function getHtmlAttrs(array $orig=array(), $overwrite=true)
 233     {
 234         $attrs = [
 235                 'height' => $this->getHeight(),
 236                 'width'  => $this->getWidth(),
 237                 'src'    => $this->getUrl(),
 238             ];
 239         return $overwrite ? array_merge($orig, $attrs) : array_merge($attrs, $orig);
 240     }
 241
 242     public function delete($useWhere=false)
 243     {
 244         if (!empty($this->filename)) {
 245             try {
 246                 $deleted = @unlink($this->getPath());
 247                 if (!$deleted) {
 248                     common_log(LOG_ERR, 'Could not unlink existing thumbnail file: '._ve($this->getPath()));
 249                 }
 250             } catch (FileNotFoundException $e) {
 251                 common_log(LOG_INFO, 'Thumbnail already gone from '._ve($e->path));
 252             }
 253         }
 254
 255         return parent::delete($useWhere);
 256     }
 257
 258     public function getFile()
 259     {
 260         return File::getByID($this->file_id);
 261     }
 262
 263
 264     static public function hashurl($url)
 265     {
 266         if (!mb_strlen($url)) {
 267             throw new Exception('No URL provided to hash algorithm.');
 268         }
 269         return hash(self::URLHASH_ALG, $url);
 270     }
 271
 272     public function onInsert()
 273     {
 274         $this->setUrlhash();
 275     }
 276
 277     public function onUpdate($dataObject=false)
 278     {
 279         // if we have nothing to compare with OR it has changed from previous entry
 280         if (!$dataObject instanceof Managed_DataObject || $this->url !== $dataObject->url) {
 281             $this->setUrlhash();
 282         }
 283     }
 284
 285     public function setUrlhash()
 286     {
 287         $this->urlhash = mb_strlen($this->url)>0 ? self::hashurl($this->url) : null;
 288     }
 289 }