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';
  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                 testUser($filter, $user);
  52             }
  53             $offset += $found;
  54         }
  55
  56     } while ($found > 0);
  57 }
  58
  59 function testUser($filter, $user) {
  60
  61     printfnq("Testing user %s\n", $user->nickname);
  62
  63     $profile = Profile::staticGet('id', $user->id);
  64
  65     $str = new ProfileNoticeStream($profile, $profile);
  66
  67     $offset = 0;
  68     $limit  = 100;
  69
  70     do {
  71         $notice = $str->getNotices($offset, $limit);
  72         while ($notice->fetch()) {
  73             printfv("Testing notice %d...", $notice->id);
  74             $result = $filter->test($notice);
  75             Spam_score::save($notice, $result);
  76             printfv("%s\n", ($result->isSpam) ? "SPAM" : "HAM");
  77         }
  78         $offset += $notice->N;
  79     } while ($notice->N > 0);
  80 }
  81
  82 try {
  83     $filter = null;
  84     Event::handle('GetSpamFilter', array(&$filter));
  85     if (empty($filter)) {
  86         throw new Exception(_("No spam filter."));
  87     }
  88     if (have_option('a', 'all')) {
  89         testAllUsers($filter);
  90     } else {
  91         $user = getUser();
  92         testUser($filter, $user);
  93     }
  94 } catch (Exception $e) {
  95     print $e->getMessage()."\n";
  96     exit(1);
  97 }