]> git.mxchange.org Git - quix0rs-gnu-social.git/blobdiff - classes/File_thumbnail.php
Merge branch 'master' of git.gnu.io:gnu/gnu-social into nightly
[quix0rs-gnu-social.git] / classes / File_thumbnail.php
index af3551028e804182f8abacd230d73e3da465018e..68cb2a737fb2a939e34beebbcfb14e394de78ca2 100644 (file)
@@ -27,19 +27,23 @@ class File_thumbnail extends Managed_DataObject
 {
     public $__table = 'file_thumbnail';                  // table name
     public $file_id;                         // int(4)  primary_key not_null
-    public $url;                             // varchar(191)  unique_key   not 255 because utf8mb4 takes more space
-    public $filename;                        // varchar(191)   not 255 because utf8mb4 takes more space
+    public $urlhash;                         // varchar(64) indexed
+    public $url;                             // text
+    public $filename;                        // text
     public $width;                           // int(4)  primary_key
     public $height;                          // int(4)  primary_key
     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
 
+    const URLHASH_ALG = 'sha256';
+
     public static function schemaDef()
     {
         return array(
             'fields' => array(
                 'file_id' => array('type' => 'int', 'not null' => true, 'description' => 'thumbnail for what URL/file'),
-                'url' => array('type' => 'varchar', 'length' => 191, 'description' => 'URL of thumbnail'),
-                'filename' => array('type' => 'varchar', 'length' => 191, 'description' => 'if stored locally, filename is put here'),
+                'urlhash' => array('type' => 'varchar', 'length' => 64, 'description' => 'sha256 of url field if non-empty'),
+                'url' => array('type' => 'text', 'description' => 'URL of thumbnail'),
+                'filename' => array('type' => 'text', 'description' => 'if stored locally, filename is put here'),
                 'width' => array('type' => 'int', 'description' => 'width of thumbnail'),
                 'height' => array('type' => 'int', 'description' => 'height of thumbnail'),
                 'modified' => array('type' => 'timestamp', 'not null' => true, 'description' => 'date this record was modified'),
@@ -47,6 +51,7 @@ class File_thumbnail extends Managed_DataObject
             'primary key' => array('file_id', 'width', 'height'),
             'indexes' => array(
                 'file_thumbnail_file_id_idx' => array('file_id'),
+                'file_thumbnail_urlhash_idx' => array('urlhash'),
             ),
             'foreign keys' => array(
                 'file_thumbnail_file_id_fkey' => array('file', array('file_id' => 'id')),
@@ -80,13 +85,25 @@ class File_thumbnail extends Managed_DataObject
 
     /**
      * Fetch an entry by using a File's id
+     *
+     * @param   File    $file       The File object we're getting a thumbnail for.
+     * @param   boolean $notNullUrl Originally remote thumbnails have a URL stored, we use this to find the "original"
+     *
+     * @return  File_thumbnail
+     * @throws  NoResultException if no File_thumbnail matched the criteria
      */
-    static function byFile(File $file) {
-        $file_thumbnail = self::getKV('file_id', $file->id);
-        if (!$file_thumbnail instanceof File_thumbnail) {
-            throw new ServerException(sprintf('No File_thumbnail entry for File id==%u', $file->id));
+    static function byFile(File $file, $notNullUrl=true) {
+        $thumb = new File_thumbnail();
+        $thumb->file_id = $file->getID();
+        if ($notNullUrl) {
+            $thumb->whereAdd('url IS NOT NULL');
+        }
+        $thumb->orderBy('modified ASC');    // the first created, a somewhat ugly hack
+        $thumb->limit(1);
+        if (!$thumb->find(true)) {
+            throw new NoResultException($thumb);
         }
-        return $file_thumbnail;
+        return $thumb;
     }
 
     /**
@@ -113,57 +130,139 @@ class File_thumbnail extends Managed_DataObject
 
     static function path($filename)
     {
-        // TODO: Store thumbnails in their own directory and don't use File::path here
-        return File::path($filename);
+        File::tryFilename($filename);
+
+        // NOTE: If this is left empty in default config, it will be set to File::path('thumb')
+        $dir = common_config('thumbnail', 'dir');
+
+        if (!in_array($dir[mb_strlen($dir)-1], ['/', '\\'])) {
+            $dir .= DIRECTORY_SEPARATOR;
+        }
+
+        return $dir . $filename;
+    }
+
+    static function url($filename)
+    {
+        File::tryFilename($filename);
+
+        // FIXME: private site thumbnails?
+
+        $path = common_config('thumbnail', 'path');
+        if (empty($path)) {
+            return File::url('thumb')."/{$filename}";
+        }
+
+        $protocol = (GNUsocial::useHTTPS() ? 'https' : 'http');
+        $server = common_config('thumbnail', 'server') ?: common_config('site', 'server');
+
+        if ($path[mb_strlen($path)-1] != '/') {
+            $path .= '/';
+        }
+        if ($path[0] != '/') {
+            $path = '/'.$path;
+        }
+
+        return $protocol.'://'.$server.$path.$filename;
+    }
+
+    public function getFilename()
+    {
+        return File::tryFilename($this->filename);
     }
 
+    /**
+     *
+     * @return  string  full filesystem path to the locally stored thumbnail file
+     * @throws  
+     */
     public function getPath()
     {
-        return self::path($this->filename);
+        $oldpath = File::path($this->getFilename());
+        $thumbpath = self::path($this->getFilename());
+
+        // If we have a file in our old thumbnail storage path, move (or copy) it to the new one
+        // (if the if/elseif don't match, we have a $thumbpath just as we should and can return it)
+        if (file_exists($oldpath) && !file_exists($thumbpath)) {
+            try {
+                // let's get the filename of the File, to check below if it happens to be identical
+                $file_filename = $this->getFile()->getFilename();
+            } catch (NoResultException $e) {
+                // reasonably the function calling us will handle the following as "File_thumbnail entry should be deleted"
+                throw new FileNotFoundException($thumbpath);
+            } catch (InvalidFilenameException $e) {
+                // invalid filename in getFile()->getFilename(), just
+                // means the File object isn't stored locally and that
+                // means it's safe to move it below.
+                $file_filename = null;
+            }
+
+            if ($this->getFilename() === $file_filename) {
+                // special case where thumbnail file exactly matches stored File filename
+                common_debug('File filename and File_thumbnail filename match on '.$this->file_id.', copying instead');
+                copy($oldpath, $thumbpath);
+            } elseif (!rename($oldpath, $thumbpath)) {
+                common_log(LOG_ERR, 'Could not move thumbnail from '._ve($oldpath).' to '._ve($thumbpath));
+                throw new ServerException('Could not move thumbnail from old path to new path.');
+            } else {
+                common_log(LOG_DEBUG, 'Moved thumbnail '.$this->file_id.' from '._ve($oldpath).' to '._ve($thumbpath));
+            }
+        } elseif (!file_exists($thumbpath)) {
+            throw new FileNotFoundException($thumbpath);
+        }
+
+        return $thumbpath;
     }
 
     public function getUrl()
     {
-        if (!empty($this->getFile()->filename)) {
-            // A locally stored File, so let's generate a URL for our instance.
-            $url = File::url($this->filename);
-            if ($url != $this->url) {
-                // For indexing purposes, in case we do a lookup on the 'url' field.
-                // also we're fixing possible changes from http to https, or paths
-                $this->updateUrl($url);
+        if (!empty($this->filename) || $this->getFile()->isLocal()) {
+            // A locally stored File, so we can dynamically generate a URL.
+            $url = common_local_url('attachment_thumbnail', array('attachment'=>$this->file_id));
+            if (strpos($url, '?') === false) {
+                $url .= '?';
             }
-            return $url;
+            return $url . http_build_query(array('w'=>$this->width, 'h'=>$this->height));
         }
 
-        // No local filename available, return the URL we have stored
+        // No local filename available, return the remote URL we have stored
         return $this->url;
     }
 
-    public function updateUrl($url)
+    public function getHeight()
     {
-        $file = File_thumbnail::getKV('url', $url);
-        if ($file instanceof File_thumbnail) {
-            throw new ServerException('URL already exists in DB');
-        }
-        $sql = 'UPDATE %1$s SET url=%2$s WHERE url=%3$s;';
-        $result = $this->query(sprintf($sql, $this->__table,
-                                             $this->_quote((string)$url),
-                                             $this->_quote((string)$this->url)));
-        if ($result === false) {
-            common_log_db_error($this, 'UPDATE', __FILE__);
-            throw new ServerException("Could not UPDATE {$this->__table}.url");
-        }
+        return $this->height;
+    }
+
+    public function getWidth()
+    {
+        return $this->width;
+    }
 
-        return $result;
+    /**
+     * @throws UseFileAsThumbnailException from File_thumbnail->getUrl() for stuff like animated GIFs
+     */
+    public function getHtmlAttrs(array $orig=array(), $overwrite=true)
+    {
+        $attrs = [
+                'height' => $this->getHeight(),
+                'width'  => $this->getWidth(),
+                'src'    => $this->getUrl(),
+            ];
+        return $overwrite ? array_merge($orig, $attrs) : array_merge($attrs, $orig);
     }
 
     public function delete($useWhere=false)
     {
-        if (!empty($this->filename) && file_exists(File_thumbnail::path($this->filename))) {
-            $deleted = @unlink(self::path($this->filename));
+        try {
+            $thumbpath = self::path($this->getFilename());
+            // if file does not exist, try to delete it
+            $deleted = !file_exists($thumbpath) || @unlink($thumbpath);
             if (!$deleted) {
-                common_log(LOG_ERR, sprintf('Could not unlink existing file: "%s"', self::path($this->filename)));
+                common_log(LOG_ERR, 'Could not unlink existing thumbnail file: '._ve($thumbpath));
             }
+        } catch (InvalidFilenameException $e) {
+            common_log(LOG_ERR, 'Deleting object but not attempting deleting file: '._ve($e->getMessage()));
         }
 
         return parent::delete($useWhere);
@@ -171,11 +270,37 @@ class File_thumbnail extends Managed_DataObject
 
     public function getFile()
     {
-        $file = new File();
-        $file->id = $this->file_id;
-        if (!$file->find(true)) {
-            throw new NoResultException($file);
+        return File::getByID($this->file_id);
+    }
+
+    public function getFileId()
+    {
+        return $this->file_id;
+    }
+
+    static public function hashurl($url)
+    {
+        if (!mb_strlen($url)) {
+            throw new Exception('No URL provided to hash algorithm.');
+        }
+        return hash(self::URLHASH_ALG, $url);
+    }
+
+    public function onInsert()
+    {
+        $this->setUrlhash();
+    }
+
+    public function onUpdate($dataObject=false)
+    {
+        // if we have nothing to compare with OR it has changed from previous entry
+        if (!$dataObject instanceof Managed_DataObject || $this->url !== $dataObject->url) {
+            $this->setUrlhash();
         }
-        return $file;
+    }
+
+    public function setUrlhash()
+    {
+        $this->urlhash = mb_strlen($this->url)>0 ? self::hashurl($this->url) : null;
     }
 }