Merge pull request #12385 from annando/more-blurhash

[friendica.git] / src / Model / Post / Link.php

diff --git a/src/Model/Post/Link.php b/src/Model/Post/Link.php
index 52ed5572ceab3204ef34059836f61e844e1f38a3..83e5bea99fe000a62649f5497e4e910254d2a1a2 100644 (file)
--- a/src/Model/Post/Link.php
+++ b/src/Model/Post/Link.php
@@ -28,7 +28,10 @@ use Friendica\Database\DBA;
 use Friendica\DI;
 use Friendica\Network\HTTPClient\Client\HttpClientAccept;
 use Friendica\Network\HTTPClient\Client\HttpClientOptions;
+use Friendica\Util\HTTPSignature;
+use Friendica\Util\Images;
 use Friendica\Util\Proxy;
+use Friendica\Object\Image;
 
 /**
  * Class Link
@@ -40,7 +43,7 @@ class Link
        /**
         * Check if the link is stored
         *
-        * @param int $uriId URI id
+        * @param int $uriId
         * @param string $url URL
         * @return bool Whether record has been found
         */
@@ -52,9 +55,9 @@ class Link
        /**
         * Returns URL by URI id and other URL
         *
-        * @param int $uriId URI id
-        * @param string $url URL
-        * @param string size Size
+        * @param int $uriId
+        * @param string $url
+        * @param string $size
         * @return string Found link URL + id on success, $url on failture
         */
        public static function getByLink(int $uriId, string $url, string $size = ''): string
@@ -72,12 +75,14 @@ class Link
                if (!empty($link['id'])) {
                        $id = $link['id'];
                        Logger::info('Found', ['id' => $id, 'uri-id' => $uriId, 'url' => $url]);
-               } else {
-                       $mime = self::fetchMimeType($url);
+               } else { 
+                       $fields = self::fetchMimeType($url);
+                       $fields['uri-id'] = $uriId;
+                       $fields['url'] = $url;
 
-                       DBA::insert('post-link', ['uri-id' => $uriId, 'url' => $url, 'mimetype' => $mime], Database::INSERT_IGNORE);
+                       DBA::insert('post-link', $fields, Database::INSERT_IGNORE);
                        $id = DBA::lastInsertId();
-                       Logger::info('Inserted', ['id' => $id, 'uri-id' => $uriId, 'url' => $url]);
+                       Logger::info('Inserted', $fields);
                }
 
                if (empty($id)) {
@@ -113,27 +118,36 @@ class Link
         * Fetches MIME type by URL and Accept: header
         *
         * @param string $url URL to fetch
-        * @param string $accept Accept: line
-        * @return string Discovered MIME type or empty string on failure
+        * @param string $accept Comma-separated list of expected response MIME type(s)
+        * @return array Discovered MIME type and blurhash or empty array on failure
         */
-       private static function fetchMimeType(string $url, string $accept = HttpClientAccept::DEFAULT): string
+       private static function fetchMimeType(string $url, string $accept = HttpClientAccept::DEFAULT): array
        {
                $timeout = DI::config()->get('system', 'xrd_timeout');
 
-               $curlResult = DI::httpClient()->head($url, [HttpClientOptions::TIMEOUT => $timeout, HttpClientOptions::ACCEPT_CONTENT => $accept]);
-
-               if ($curlResult->isSuccess() && empty($media['mimetype'])) {
-                       return $curlResult->getHeader('Content-Type')[0] ?? '';
+               $curlResult = HTTPSignature::fetchRaw($url, 0, [HttpClientOptions::TIMEOUT => $timeout, HttpClientOptions::ACCEPT_CONTENT => $accept]);
+               if (!$curlResult->isSuccess()) {
+                       return [];
+               }
+               $fields = ['mimetype' => $curlResult->getHeader('Content-Type')[0]];
+
+               $img_str = $curlResult->getBody();
+               $image = new Image($img_str, Images::getMimeTypeByData($img_str));
+               if ($image->isValid()) {
+                       $fields['mimetype'] = $image->getType();
+                       $fields['width']    = $image->getWidth();
+                       $fields['height']   = $image->getHeight();
+                       $fields['blurhash'] = $image->getBlurHash();
                }
 
-               return '';
+               return $fields;
        }
 
        /**
         * Add external links and replace them in the body
         *
-        * @param integer $uriId URI id
-        * @param string $body HTML body
+        * @param integer $uriId
+        * @param string $body Item body formatted with BBCodes
         * @return string Body with replaced links
         */
        public static function insertFromBody(int $uriId, string $body): string