gnu-social/plugins/ActivitySpam/ActivitySpamPlugin.php

261 lines
7.8 KiB
PHP
Raw Normal View History

<?php
/**
* StatusNet - the distributed open-source microblogging tool
2012-03-02 16:20:01 +00:00
* Copyright (C) 2011,2012, StatusNet, Inc.
*
* ActivitySpam Plugin
*
* PHP version 5
*
* 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/>.
*
* @category Spam
* @package StatusNet
* @author Evan Prodromou <evan@status.net>
2012-03-02 16:20:01 +00:00
* @copyright 2011,2012 StatusNet, Inc.
* @license http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
* @link http://status.net/
*/
if (!defined('STATUSNET')) {
// This check helps protect against security problems;
// your code file can't be executed directly from the web.
exit(1);
}
/**
* Check new notices with activity spam service.
*
* @category Spam
* @package StatusNet
* @author Evan Prodromou <evan@status.net>
2012-03-02 16:20:01 +00:00
* @copyright 2011,2012 StatusNet, Inc.
* @license http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
* @link http://status.net/
*/
class ActivitySpamPlugin extends Plugin
{
const PLUGIN_VERSION = '2.0.0';
public $server = null;
2012-03-21 01:37:35 +00:00
public $hideSpam = false;
const REVIEWSPAM = 'ActivitySpamPlugin::REVIEWSPAM';
const TRAINSPAM = 'ActivitySpamPlugin::TRAINSPAM';
/**
* Initializer
*
* @return boolean hook value; true means continue processing, false means stop.
*/
function initialize()
{
2012-03-17 05:02:41 +00:00
$this->filter = new SpamFilter(common_config('activityspam', 'server'),
common_config('activityspam', 'consumerkey'),
common_config('activityspam', 'secret'));
2012-03-05 15:58:57 +00:00
2012-03-21 01:37:35 +00:00
$this->hideSpam = common_config('activityspam', 'hidespam');
2012-05-05 03:16:46 +01:00
// Let DB_DataObject find Spam_score
common_config_set('db', 'class_location',
2012-05-05 03:20:47 +01:00
common_config('db', 'class_location') .':'.dirname(__FILE__));
2012-05-05 03:16:46 +01:00
return true;
}
/**
* Database schema setup
*
* @see Schema
* @see ColumnDef
*
* @return boolean hook value; true means continue processing, false means stop.
*/
function onCheckSchema()
{
$schema = Schema::get();
$schema->ensureTable('spam_score', Spam_score::schemaDef());
2012-03-02 16:37:20 +00:00
Spam_score::upgrade();
return true;
}
/**
2012-03-02 16:20:01 +00:00
* When a notice is saved, check its spam score
*
2012-03-02 16:20:01 +00:00
* @param Notice $notice Notice that was just saved
*
* @return boolean hook value; true means continue processing, false means stop.
*/
function onEndNoticeSave($notice)
{
2012-03-05 15:58:57 +00:00
try {
2012-03-05 15:58:57 +00:00
$result = $this->filter->test($notice);
2012-03-05 15:58:57 +00:00
$score = Spam_score::saveNew($notice, $result);
2012-03-05 15:58:57 +00:00
$this->log(LOG_INFO, "Notice " . $notice->id . " has spam score " . $score->score);
2012-03-05 15:58:57 +00:00
} catch (Exception $e) {
// Log but continue
2012-03-05 15:58:57 +00:00
$this->log(LOG_ERR, $e->getMessage());
}
return true;
}
2012-03-05 15:09:14 +00:00
function onNoticeDeleteRelated($notice) {
$score = Spam_score::getKV('notice_id', $notice->id);
2012-03-05 15:09:14 +00:00
if (!empty($score)) {
$score->delete();
}
return true;
}
function onUserRightsCheck($profile, $right, &$result) {
switch ($right) {
case self::REVIEWSPAM:
case self::TRAINSPAM:
$result = ($profile->hasRole(Profile_role::MODERATOR) || $profile->hasRole('modhelper'));
return false;
default:
return true;
}
}
function onGetSpamFilter(&$filter) {
$filter = $this->filter;
return false;
}
2012-03-08 13:01:37 +00:00
function onEndShowNoticeOptionItems($nli)
{
$profile = Profile::current();
if (!empty($profile) && $profile->hasRight(self::TRAINSPAM)) {
$notice = $nli->getNotice();
$out = $nli->getOut();
if (!empty($notice)) {
$score = Spam_score::getKV('notice_id', $notice->id);
if (empty($score)) {
2012-05-16 17:26:29 +01:00
// If it's empty, we can train it.
$form = new TrainSpamForm($out, $notice);
$form->show();
} else if ($score->is_spam) {
$form = new TrainHamForm($out, $notice);
$form->show();
} else if (!$score->is_spam) {
$form = new TrainSpamForm($out, $notice);
$form->show();
}
}
}
return true;
}
2012-03-07 15:08:05 +00:00
/**
* Map URLs to actions
*
* @param URLMapper $m path-to-action mapper
2012-03-07 15:08:05 +00:00
*
* @return boolean hook value; true means continue processing, false means stop.
*/
public function onRouterInitialized(URLMapper $m)
2012-03-07 15:08:05 +00:00
{
$m->connect('main/train/spam',
array('action' => 'train', 'category' => 'spam'));
$m->connect('main/train/ham',
array('action' => 'train', 'category' => 'ham'));
$m->connect('main/spam',
array('action' => 'spam'));
2012-03-07 15:08:05 +00:00
return true;
}
2012-03-07 20:32:05 +00:00
function onEndShowStyles($action)
{
$action->element('style', null,
2012-03-07 21:06:27 +00:00
'.form-train-spam input.submit { background: url('.$this->path('icons/bullet_black.png').') no-repeat 0px 0px } ' . "\n" .
'.form-train-ham input.submit { background: url('.$this->path('icons/exclamation.png').') no-repeat 0px 0px } ');
2012-03-07 20:32:05 +00:00
return true;
}
function onEndPublicGroupNav($nav)
{
$user = common_current_user();
if (!empty($user) && $user->hasRight(self::REVIEWSPAM)) {
$nav->out->menuItem(common_local_url('spam'),
_m('MENU','Spam'),
// TRANS: Menu item title in search group navigation panel.
_('Notices marked as spam'),
$nav->actionName == 'spam',
'nav_timeline_spam');
}
return true;
}
function onPluginVersion(array &$versions)
{
$versions[] = array('name' => 'ActivitySpam',
'version' => self::PLUGIN_VERSION,
'author' => 'Evan Prodromou',
'homepage' => 'https://git.gnu.io/gnu/gnu-social/tree/master/plugins/ActivitySpam',
'description' =>
_m('Test notices against the Activity Spam service.'));
return true;
}
2012-03-19 14:15:33 +00:00
2012-03-21 00:48:47 +00:00
function onEndNoticeInScope($notice, $profile, &$bResult)
{
2012-03-21 01:37:35 +00:00
if ($this->hideSpam) {
if ($bResult) {
2012-03-21 00:48:47 +00:00
$score = Spam_score::getKV('notice_id', $notice->id);
2012-03-21 00:48:47 +00:00
if (!empty($score) && $score->is_spam) {
2012-03-21 01:37:35 +00:00
if (empty($profile) ||
($profile->id !== $notice->profile_id &&
!$profile->hasRight(self::REVIEWSPAM))) {
$bResult = false;
}
2012-03-21 00:48:47 +00:00
}
}
}
2012-03-21 16:37:00 +00:00
return true;
2012-03-21 00:48:47 +00:00
}
/**
* Pre-cache our spam scores if needed.
*/
function onEndNoticeListPrefill(array &$notices, array &$profiles, array $notice_ids, Profile $scoped=null) {
if ($this->hideSpam) {
Spam_score::multiGet('notice_id', $notice_ids);
}
return true;
}
}