plugins/ActivitySpam/scripts/testuser.php

   1 <?php
   2 /*
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2012 StatusNet, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 define('INSTALLDIR', realpath(dirname(__FILE__) . '/../../..'));
  21
  22 $shortoptions = 'i:n:a';
  23 $longoptions = array('id=', 'nickname=', 'all');
  24
  25 $helptext = <<<END_OF_TESTUSER_HELP
  26 testuser.php [options]
  27 Test user activities against the spam filter
  28
  29   -i --id       ID of user to export
  30   -n --nickname nickname of the user to export
  31   -a --all      All users
  32 END_OF_TESTUSER_HELP;
  33
  34 require_once INSTALLDIR.'/scripts/commandline.inc.php';
  35
  36 function testAllUsers($filter) {
  37     $found = false;
  38     $offset = 0;
  39     $limit  = 1000;
  40
  41     do {
  42
  43         $user = new User();
  44         $user->orderBy('created');
  45         $user->limit($offset, $limit);
  46
  47         $found = $user->find();
  48
  49         if ($found) {
  50             while ($user->fetch()) {
  51                 try {
  52                     testUser($filter, $user);
  53                 } catch (Exception $e) {
  54                     printfnq("ERROR testing user %s\n: %s", $user->nickname, $e->getMessage());
  55                 }
  56             }
  57             $offset += $found;
  58         }
  59
  60     } while ($found > 0);
  61 }
  62
  63 function testUser($filter, $user) {
  64
  65     printfnq("Testing user %s\n", $user->nickname);
  66
  67     $profile = Profile::getKV('id', $user->id);
  68
  69     $str = new ProfileNoticeStream($profile, $profile);
  70
  71     $offset = 0;
  72     $limit  = 100;
  73
  74     do {
  75         $notice = $str->getNotices($offset, $limit);
  76         while ($notice->fetch()) {
  77             try {
  78                 printfv("Testing notice %d...", $notice->id);
  79                 $result = $filter->test($notice);
  80                 Spam_score::save($notice, $result);
  81                 printfv("%s\n", ($result->isSpam) ? "SPAM" : "HAM");
  82             } catch (Exception $e) {
  83                 printfnq("ERROR testing notice %d: %s\n", $notice->id, $e->getMessage());
  84             }
  85         }
  86         $offset += $notice->N;
  87     } while ($notice->N > 0);
  88 }
  89
  90 try {
  91     $filter = null;
  92     Event::handle('GetSpamFilter', array(&$filter));
  93     if (empty($filter)) {
  94         throw new Exception(_("No spam filter."));
  95     }
  96     if (have_option('a', 'all')) {
  97         testAllUsers($filter);
  98     } else {
  99         $user = getUser();
 100         testUser($filter, $user);
 101     }
 102 } catch (Exception $e) {
 103     print $e->getMessage()."\n";
 104     exit(1);
 105 }