gnu-social/scripts/fixup_deletions.php

168 lines
4.5 KiB
PHP
Raw Normal View History

#!/usr/bin/env php
<?php
/*
* StatusNet - a distributed open-source microblogging tool
* Copyright (C) 2010 StatusNet, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
define('INSTALLDIR', dirname(__DIR__));
define('PUBLICDIR', INSTALLDIR . DIRECTORY_SEPARATOR . 'public');
$longoptions = array('dry-run', 'start=', 'end=');
$helptext = <<<END_OF_USERROLE_HELP
fixup_deletions.php [options]
Finds notices posted by deleted users and cleans them up.
Stray incompletely deleted items cause various fun problems!
--dry-run look but don't touch
--start=N start looking at profile_id N instead of 1
--end=N end looking at profile_id N instead of the max
END_OF_USERROLE_HELP;
require_once INSTALLDIR.'/scripts/commandline.inc';
/**
* Find the highest profile_id currently listed in the notice table;
* this field is indexed and should return very quickly.
*
* We check notice.profile_id rather than profile.id because we're
* looking for notices left behind after deletion; if the most recent
* accounts were deleted, we wouldn't have them from profile.
*
* @return int
* @access private
*/
function get_max_profile_id()
{
$query = 'SELECT MAX(profile_id) AS id FROM notice';
$profile = new Profile();
$profile->query($query);
if ($profile->fetch()) {
return intval($profile->id);
} else {
die("Something went awry; could not look up max used profile_id.");
}
}
/**
* Check for profiles in the given id range that are missing, presumed deleted.
*
* @param int $start beginning profile.id, inclusive
* @param int $end final profile.id, inclusive
* @return array of integer profile.ids
* @access private
*/
function get_missing_profiles($start, $end)
{
$query = sprintf("SELECT id FROM profile WHERE id BETWEEN %d AND %d",
$start, $end);
$profile = new Profile();
$profile->query($query);
$all = range($start, $end);
$known = array();
while ($row = $profile->fetch()) {
$known[] = intval($profile->id);
}
unset($profile);
$missing = array_diff($all, $known);
return $missing;
}
/**
* Look for stray notices from this profile and, if present, kill them.
*
* @param int $profile_id
* @param bool $dry if true, we won't delete anything
*/
function cleanup_missing_profile($profile_id, $dry)
{
$notice = new Notice();
$notice->profile_id = $profile_id;
$notice->find();
if ($notice->N == 0) {
return;
}
$s = ($notice->N == 1) ? '' : 's';
print "Deleted profile $profile_id has $notice->N stray notice$s:\n";
while ($notice->fetch()) {
print " notice $notice->id";
if ($dry) {
print " (skipped; dry run)\n";
} else {
$victim = clone($notice);
try {
$victim->delete();
print " (deleted)\n";
} catch (Exception $e) {
print " FAILED: ";
print $e->getMessage();
print "\n";
}
}
}
}
$dry = have_option('dry-run');
$max_profile_id = get_max_profile_id();
$chunk = 1000;
if (have_option('start')) {
$begin = intval(get_option_value('start'));
} else {
$begin = 1;
}
if (have_option('end')) {
$final = min($max_profile_id, intval(get_option_value('end')));
} else {
$final = $max_profile_id;
}
if ($begin < 1) {
die("Silly human, you can't begin before profile number 1!\n");
}
if ($final < $begin) {
die("Silly human, you can't end at $final if it's before $begin!\n");
}
// Identify missing profiles...
for ($start = $begin; $start <= $final; $start += $chunk) {
$end = min($start + $chunk - 1, $final);
print "Checking for missing profiles between id $start and $end";
if ($dry) {
print " (dry run)";
}
print "...\n";
$missing = get_missing_profiles($start, $end);
foreach ($missing as $profile_id) {
cleanup_missing_profile($profile_id, $dry);
}
}
echo "done.\n";