scripts/updateavatarurl.php

   1 #!/usr/bin/env php
   2 <?php
   3 /*
   4  * StatusNet - a distributed open-source microblogging tool
   5  * Copyright (C) 2008, 2009, StatusNet, Inc.
   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 published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (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 <http://www.gnu.org/licenses/>.
  19  */
  20
  21 define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
  22
  23 $shortoptions = 'i:n:a';
  24 $longoptions = array('id=', 'nickname=', 'all');
  25
  26 $helptext = <<<END_OF_UPDATEAVATARURL_HELP
  27 updateavatarurl.php [options]
  28 update the URLs of all avatars in the system
  29
  30   -i --id       ID of user to update
  31   -n --nickname nickname of the user to update
  32   -a --all      update all
  33
  34 END_OF_UPDATEAVATARURL_HELP;
  35
  36 require_once INSTALLDIR.'/scripts/commandline.inc';
  37
  38 try {
  39     $user = null;
  40
  41     if (have_option('i', 'id')) {
  42         $id = get_option_value('i', 'id');
  43         $user = User::staticGet('id', $id);
  44         if (empty($user)) {
  45             throw new Exception("Can't find user with id '$id'.");
  46         }
  47         updateAvatars($user);
  48     } else if (have_option('n', 'nickname')) {
  49         $nickname = get_option_value('n', 'nickname');
  50         $user = User::staticGet('nickname', $nickname);
  51         if (empty($user)) {
  52             throw new Exception("Can't find user with nickname '$nickname'");
  53         }
  54         updateAvatars($user);
  55     } else if (have_option('a', 'all')) {
  56         $user = new User();
  57         if ($user->find()) {
  58             while ($user->fetch()) {
  59                 updateAvatars($user);
  60             }
  61         }
  62     } else {
  63         throw new Exception("You have to provide an ID or nickname or 'all'.");
  64     }
  65 } catch (Exception $e) {
  66     print $e->getMessage()."\n";
  67     exit(1);
  68 }
  69
  70 function updateAvatars($user)
  71 {
  72     $touched = false;
  73
  74     if (!have_option('q', 'quiet')) {
  75         print "Updating avatars for user '".$user->nickname."' (".$user->id.")...";
  76     }
  77
  78     $avatar = new Avatar();
  79
  80     $avatar->profile_id = $user->id;
  81
  82     if (!$avatar->find()) {
  83         if (have_option('v', 'verbose')) {
  84                 print "(none found)...";
  85         }
  86     } else {
  87         while ($avatar->fetch()) {
  88             if (have_option('v', 'verbose')) {
  89                 if ($avatar->original) {
  90                     print "original...";
  91                 } else {
  92                     print $avatar->width."...";
  93                 }
  94             }
  95
  96             $orig = clone($avatar);
  97
  98             $avatar->url = Avatar::url($avatar->filename);
  99
 100             if ($avatar->url != $orig->url) {
 101                 $sql =
 102                   "UPDATE avatar SET url = '" . $avatar->url . "' ".
 103                   "WHERE profile_id = " . $avatar->profile_id . " ".
 104                   "AND width = " . $avatar->width . " " .
 105                   "AND height = " . $avatar->height . " ";
 106
 107                 if ($avatar->original) {
 108                     $sql .= "AND original = 1 ";
 109                 }
 110
 111                 if (!$avatar->query($sql)) {
 112                     throw new Exception("Can't update avatar for user " . $user->nickname . ".");
 113                 } else {
 114                     $touched = true;
 115                 }
 116             }
 117         }
 118     }
 119
 120     if ($touched) {
 121         $profile = $user->getProfile();
 122         common_broadcast_profile($profile);
 123     }
 124
 125     if (have_option('v', 'verbose')) {
 126         print "DONE.\n";
 127     }
 128 }