forked from GNUsocial/gnu-social
		
	
		
			
				
	
	
		
			123 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			PHP
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			123 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			PHP
		
	
	
		
			Executable File
		
	
	
	
	
| #!/usr/bin/env php
 | |
| <?php
 | |
| /*
 | |
|  * StatusNet - a distributed open-source microblogging tool
 | |
|  * Copyright (C) 2009-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', realpath(dirname(__FILE__) . '/..'));
 | |
| 
 | |
| $shortoptions = 'i::n::y';
 | |
| $longoptions = array('id=', 'nickname=', 'yes', 'dry-run', 'all');
 | |
| 
 | |
| $helptext = <<<END_OF_HELP
 | |
| strip_geo.php [options]
 | |
| Removes geolocation info from the given user's notices.
 | |
| 
 | |
|   -i --id       ID of the user (may be a remote profile)
 | |
|   -n --nickname nickname of the user
 | |
|   -y --yes      do not wait for confirmation
 | |
|      --dry-run  list affected notices without deleting
 | |
|      --all      run over and decache all messages, even if they don't
 | |
|                 have geo data now (helps to fix cache bugs)
 | |
| 
 | |
| END_OF_HELP;
 | |
| 
 | |
| require_once INSTALLDIR.'/scripts/commandline.inc';
 | |
| 
 | |
| if (have_option('i', 'id')) {
 | |
|     $id = get_option_value('i', 'id');
 | |
|     $profile = Profile::staticGet('id', $id);
 | |
|     if (empty($profile)) {
 | |
|         print "Can't find local or remote profile with ID $id\n";
 | |
|         exit(1);
 | |
|     }
 | |
| } else if (have_option('n', 'nickname')) {
 | |
|     $nickname = get_option_value('n', 'nickname');
 | |
|     $user = User::staticGet('nickname', $nickname);
 | |
|     if (empty($user)) {
 | |
|         print "Can't find local user with nickname '$nickname'\n";
 | |
|         exit(1);
 | |
|     }
 | |
|     $profile = $user->getProfile();
 | |
| } else {
 | |
|     print "You must provide either an ID or a nickname.\n\n";
 | |
|     show_help();
 | |
|     exit(1);
 | |
| }
 | |
| 
 | |
| if (!have_option('y', 'yes') && !have_option('--dry-run')) {
 | |
|     print "About to PERMANENTLY remove geolocation data from user '{$profile->nickname}' ({$profile->id})'s notices. Are you sure? [y/N] ";
 | |
|     $response = fgets(STDIN);
 | |
|     if (strtolower(trim($response)) != 'y') {
 | |
|         print "Aborting.\n";
 | |
|         exit(0);
 | |
|     }
 | |
| }
 | |
| 
 | |
| // @fixme for a very prolific poster this could be too many.
 | |
| $notice = new Notice();
 | |
| $notice->profile_id = $profile->id;
 | |
| if (have_option('--all')) {
 | |
|     print "Finding all notices by $profile->nickname...";
 | |
| } else {
 | |
|     print "Finding notices by $profile->nickname with geolocation data...";
 | |
|     $notice->whereAdd("lat != ''");
 | |
| }
 | |
| $notice->find();
 | |
| 
 | |
| if ($notice->N) {
 | |
|     print " $notice->N found.\n";
 | |
|     while ($notice->fetch()) {
 | |
|         print "notice id $notice->id ";
 | |
|         if (have_option('v') || have_option('--verbose')) {
 | |
|             print "({$notice->lat},{$notice->lon}) ";
 | |
|             if ($notice->location_ns) {
 | |
|                 print "ns {$notice->location_ns} id {$notice->location_id} ";
 | |
|             }
 | |
|         }
 | |
|         if (have_option('--dry-run')) {
 | |
|             // sucka
 | |
|             echo "(skipped)";
 | |
|         } else {
 | |
|             // note: setting fields to null and calling update() doesn't save the nulled fields
 | |
|             $orig = clone($notice);
 | |
|             $update = clone($notice);
 | |
| 
 | |
|             // In theory we could hit a chunk of notices at once in the UPDATE,
 | |
|             // but we're going to have to decache them individually anyway and
 | |
|             // it doesn't hurt to make sure we don't hold up replication with
 | |
|             // what might be a very slow single UPDATE.
 | |
|             $query = sprintf('UPDATE notice ' .
 | |
|                              'SET lat=NULL,lon=NULL,location_ns=NULL,location_id=NULL ' .
 | |
|                              'WHERE id=%d', $notice->id);
 | |
|             $ok = $update->query($query);
 | |
|             if ($ok) {
 | |
|                 // And now we decache him manually, as query() doesn't know what we're doing...
 | |
|                 $orig->decache();
 | |
|                 echo "(removed)";
 | |
|             } else {
 | |
|                 echo "(unchanged?)";
 | |
|             }
 | |
|         }
 | |
|         print "\n";
 | |
|     }
 | |
| } else {
 | |
|     print " none found.\n";
 | |
| }
 | |
| 
 | |
| print "DONE.\n";
 |