src/Worker/RemoveUnusedAvatars.php

   1 <?php
   2 /**
   3  * @copyright Copyright (C) 2010-2021, the Friendica project
   4  *
   5  * @license GNU AGPL version 3 or any later version
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as
   9  * published by the Free Software Foundation, either version 3 of the
  10  * License, or (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  19  *
  20  */
  21
  22 namespace Friendica\Worker;
  23
  24 use Friendica\Core\Logger;
  25 use Friendica\Core\Worker;
  26 use Friendica\Database\DBA;
  27 use Friendica\Model\Contact;
  28 use Friendica\Model\Photo;
  29
  30 /**
  31  * Removes cached avatars from public contacts that aren't in use
  32  */
  33 class RemoveUnusedAvatars
  34 {
  35         public static function execute()
  36         {
  37                 $condition = ["`uid` = ? AND NOT `self` AND NOT `nurl` IN (SELECT `nurl` FROM `contact` WHERE `uid` != ?)
  38                         AND `id` IN (SELECT `contact-id` FROM `photo`) AND NOT `id` IN (SELECT `author-id` FROM `post-user`)
  39                         AND NOT `id` IN (SELECT `owner-id` FROM `post-user`) AND NOT `id` IN (SELECT `causer-id` FROM `post-user`)
  40                         AND NOT `id` IN (SELECT `cid` FROM `post-tag`) AND NOT `id` IN (SELECT `contact-id` FROM `post-user`)", 0, 0];
  41
  42                 $total = DBA::count('contact', $condition);
  43                 Logger::notice('Starting removal', ['total' => $total]);
  44                 $count = 0;
  45                 $contacts = DBA::select('contact', ['id'], $condition);
  46                 while ($contact = DBA::fetch($contacts)) {
  47                         DBA::update('contact', ['photo' => '', 'thumb' => '', 'micro' => ''], ['id' => $contact['id']]);
  48                         Photo::delete(['contact-id' => $contact['id'], 'album' => Photo::CONTACT_PHOTOS]);
  49                         if ((++$count % 1000) == 0) {
  50                                 if (!Worker::isInMaintenanceWindow()) {
  51                                         Logger::notice('We are outside of the maintenance window, quitting');
  52                                         return;
  53                                 }
  54                                 Logger::notice('In removal', ['count' => $count, 'total' => $total]);
  55                         }
  56                 }
  57                 DBA::close($contacts);
  58                 Logger::notice('Removal done', ['count' => $count, 'total' => $total]);
  59
  60                 self::fixPhotoContacts();
  61                 self::deleteDuplicates();
  62         }
  63
  64         private static function fixPhotoContacts()
  65         {
  66                 $total = 0;
  67                 $deleted = 0;
  68                 $updated1 = 0;
  69                 $updated2 = 0;
  70                 Logger::notice('Starting contact fix');
  71                 $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]);
  72                 while ($photo = DBA::fetch($photos)) {
  73                         $total++;
  74                         $photo_contact = Contact::getById($photo['contact-id']);
  75                         $resource = Photo::ridFromURI($photo_contact['photo']);
  76                         if ($photo['resource-id'] == $resource) {
  77                                 $contact = DBA::selectFirst('contact', [], ['nurl' => $photo_contact['nurl'], 'uid' => 0]);
  78                                 if (!empty($contact['photo']) && ($contact['photo'] == $photo_contact['photo'])) {
  79                                         Logger::notice('Photo updated to public user', ['id' => $photo['id'], 'contact-id' => $contact['id']]);
  80                                         DBA::update('photo', ['contact-id' => $contact['id']], ['id' => $photo['id']]);
  81                                         $updated1++;
  82                                 }
  83                         } else {
  84                                 $updated = false;
  85                                 $contacts = DBA::select('contact', [], ['nurl' => $photo_contact['nurl']]);
  86                                 while ($contact = DBA::fetch($contacts)) {
  87                                         if ($photo['resource-id'] == Photo::ridFromURI($contact['photo'])) {
  88                                                 Logger::notice('Photo updated to given user', ['id' => $photo['id'], 'contact-id' => $contact['id'], 'uid' => $contact['uid']]);
  89                                                 DBA::update('photo', ['contact-id' => $contact['id'], 'uid' => $contact['uid']], ['id' => $photo['id']]);
  90                                                 $updated = true;
  91                                                 $updated2++;
  92                                         }
  93                                 }
  94                                 DBA::close($contacts);
  95                                 if (!$updated) {
  96                                         Logger::notice('Photo deleted', ['id' => $photo['id']]);
  97                                         Photo::delete(['id' => $photo['id']]);
  98                                         $deleted++;
  99                                 }
 100                         }
 101                 }
 102                 DBA::close($photos);
 103                 Logger::notice('Contact fix done', ['total' => $total, 'updated1' => $updated1, 'updated2' => $updated2, 'deleted' => $deleted]);
 104         }
 105
 106         private static function deleteDuplicates()
 107         {
 108                 $size = [4 => 'photo', 5 => 'thumb', 6 => 'micro'];
 109
 110                 $total = 0;
 111                 $deleted = 0;
 112                 Logger::notice('Starting duplicate removal');
 113                 $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);
 114                 while ($photo = DBA::fetch($photos)) {
 115                         $resource = Photo::ridFromURI($photo[$size[$photo['scale']]]);
 116                         if ($resource != $photo['resource-id'] && !empty($resource)) {
 117                                 $total++;
 118                                 if (DBA::exists('photo', ['resource-id' => $resource, 'scale' => $photo['scale']])) {
 119                                         Logger::notice('Photo deleted', ['id' => $photo['id']]);
 120                                         Photo::delete(['id' => $photo['id']]);
 121                                         $deleted++;
 122                                 }
 123                         }
 124                 }
 125                 DBA::close($photos);
 126                 Logger::notice('Duplicate removal done', ['total' => $total, 'deleted' => $deleted]);
 127         }
 128 }