Detection of local requests

[friendica.git] / src / Worker / RemoveUnusedAvatars.php

diff --git a/src/Worker/RemoveUnusedAvatars.php b/src/Worker/RemoveUnusedAvatars.php
index bbfd3fdf8df3147ab949602e9cabdf8c1fd8e4be..58670c87350a45bc928e06a79aadb608d4427b0c 100644 (file)
--- a/src/Worker/RemoveUnusedAvatars.php
+++ b/src/Worker/RemoveUnusedAvatars.php
@@ -1,6 +1,6 @@
 <?php
 /**
- * @copyright Copyright (C) 2020, Friendica
+ * @copyright Copyright (C) 2010-2021, the Friendica project
  *
  * @license GNU AGPL version 3 or any later version
  *
@@ -22,9 +22,9 @@
 namespace Friendica\Worker;
 
 use Friendica\Core\Logger;
-use Friendica\Core\Protocol;
 use Friendica\Core\Worker;
 use Friendica\Database\DBA;
+use Friendica\Model\Contact;
 use Friendica\Model\Photo;
 
 /**
@@ -56,5 +56,73 @@ class RemoveUnusedAvatars
                }
                DBA::close($contacts);
                Logger::notice('Removal done', ['count' => $count, 'total' => $total]);
+
+               self::fixPhotoContacts();
+               self::deleteDuplicates();
+       }
+
+       private static function fixPhotoContacts()
+       {
+               $total = 0;
+               $deleted = 0;
+               $updated1 = 0;
+               $updated2 = 0;
+               Logger::notice('Starting contact fix');
+               $photos = DBA::select('photo', [], ["`uid` = ? AND `contact-id` IN (SELECT `id` FROM `contact` WHERE `uid` != ?) AND `contact-id` != ? AND `scale` IN (?, ?, ?)", 0, 0, 0, 4, 5, 6]);
+               while ($photo = DBA::fetch($photos)) {
+                       $total++;
+                       $photo_contact = Contact::getById($photo['contact-id']);
+                       $resource = Photo::ridFromURI($photo_contact['photo']);
+                       if ($photo['resource-id'] == $resource) {
+                               $contact = DBA::selectFirst('contact', [], ['nurl' => $photo_contact['nurl'], 'uid' => 0]);
+                               if (!empty($contact['photo']) && ($contact['photo'] == $photo_contact['photo'])) {
+                                       Logger::notice('Photo updated to public user', ['id' => $photo['id'], 'contact-id' => $contact['id']]);
+                                       DBA::update('photo', ['contact-id' => $contact['id']], ['id' => $photo['id']]);
+                                       $updated1++;
+                               }
+                       } else {
+                               $updated = false;
+                               $contacts = DBA::select('contact', [], ['nurl' => $photo_contact['nurl']]);
+                               while ($contact = DBA::fetch($contacts)) {
+                                       if ($photo['resource-id'] == Photo::ridFromURI($contact['photo'])) {
+                                               Logger::notice('Photo updated to given user', ['id' => $photo['id'], 'contact-id' => $contact['id'], 'uid' => $contact['uid']]);
+                                               DBA::update('photo', ['contact-id' => $contact['id'], 'uid' => $contact['uid']], ['id' => $photo['id']]);
+                                               $updated = true;
+                                               $updated2++;
+                                       }
+                               }               
+                               DBA::close($contacts);
+                               if (!$updated) {
+                                       Logger::notice('Photo deleted', ['id' => $photo['id']]);
+                                       Photo::delete(['id' => $photo['id']]);
+                                       $deleted++;
+                               }
+                       }
+               }
+               DBA::close($photos);
+               Logger::notice('Contact fix done', ['total' => $total, 'updated1' => $updated1, 'updated2' => $updated2, 'deleted' => $deleted]);
+       }
+
+       private static function deleteDuplicates()
+       {
+               $size = [4 => 'photo', 5 => 'thumb', 6 => 'micro'];
+
+               $total = 0;
+               $deleted = 0;
+               Logger::notice('Starting duplicate removal');
+               $photos = DBA::p("SELECT `photo`.`id`, `photo`.`uid`, `photo`.`scale`, `photo`.`album`, `photo`.`contact-id`, `photo`.`resource-id`, `contact`.`photo`, `contact`.`thumb`, `contact`.`micro` FROM `photo` INNER JOIN `contact` ON `contact`.`id` = `photo`.`contact-id` and `photo`.`contact-id` != ? AND `photo`.`scale` IN (?, ?, ?)", 0, 4, 5, 6);
+               while ($photo = DBA::fetch($photos)) {
+                       $resource = Photo::ridFromURI($photo[$size[$photo['scale']]]);
+                       if ($resource != $photo['resource-id'] && !empty($resource)) {
+                               $total++;
+                               if (DBA::exists('photo', ['resource-id' => $resource, 'scale' => $photo['scale']])) {
+                                       Logger::notice('Photo deleted', ['id' => $photo['id']]);
+                                       Photo::delete(['id' => $photo['id']]);
+                                       $deleted++;
+                               }
+                       }
+               }
+               DBA::close($photos);
+               Logger::notice('Duplicate removal done', ['total' => $total, 'deleted' => $deleted]);
        }
 }