Changeset 203 for AutoIPBan

Timestamp:

05/14/06 02:55:15 (3 years ago)

Author:

ogawa

Message:

Add MTIPBanList container tag and MTIPBanListIP tag to be able to access IPBanList from MT Templates.
Add context menu for adding hosts to IPBanList.

Files:

• AutoIPBan/trunk/AutoIPBan.pl (modified) (7 diffs)

AutoIPBan/trunk/AutoIPBan.pl

@@ -12 +12 @@
 use strict;
 use MT;
+use MT::Template::Context;
 use base 'MT::Plugin';
 
@@ -17 +18 @@
 
 sub BEGIN {
-    $VERSION = 0.01;
+    $VERSION = 0.02;
     my $plugin = __PACKAGE__->new({
         name => 'AutoIPBan',
@@ -28 +29 @@
     MT->add_plugin($plugin);
     MT->add_callback('TBPingThrottleFilter', 1, $plugin, \&tbping_auto_ipban);
+    MT::Template::Context->add_container_tag(IPBanList => \&ipbanlist);
+    MT::Template::Context->add_tag(IPBanListIP => \&ipbanlist_ip);
+}
+
+sub init_app {
+    my ($plugin, $app) = @_;
+    return unless $app->isa('MT::App::CMS');
+    $app->add_itemset_action({
+        type => 'ping',
+        key => 'add_to_ipbanlist_ping',
+        label => 'Add To IPBanList',
+        code => \&add_to_ipbanlist_ping
+        });
+    $app->add_itemset_action({
+        type => 'comment',
+        key => 'add_to_ipbanlist_comment',
+        label => 'Add To IPBanList',
+        code => \&add_to_ipbanlist_comment
+        });
 }
 
@@ -48 +68 @@
                                   { range => { created_on => 1 } });
     if ($count >= $app->config('OneHourMaxPings')) {
-        _add_ipbanlist($blog_id, $ip);
+        _add_to_ipbanlist($blog_id, $ip);
         return 0;
     }
@@ -60 +80 @@
                                { range => { created_on => 1 } });
     if ($count >= $app->config('OneDayMaxPings')) {
-        _add_ipbanlist($blog_id, $ip);
+        _add_to_ipbanlist($blog_id, $ip);
         return 0;
     }
@@ -66 +86 @@
 }
 
-sub _add_ipbanlist {
+sub add_to_ipbanlist_ping {
+    my ($app) = @_;
+    my @ids = $app->param('id')
+        or return $app->error("Need pings to add to IPBanList");
+    for my $id (@ids) {
+        my $ping = MT::TBPing->load($id, { cache_ok => 1 });
+        _add_to_ipbanlist($ping->blog_id, $ping->ip);
+    }
+    $app->call_return;
+}
+
+sub add_to_ipbanlist_comment {
+    my ($app) = @_;
+    my @ids = $app->param('id')
+        or return $app->error("Need comments to add to IPBanList");
+    for my $id (@ids) {
+        my $comment = MT::Comment->load($id, { cache_ok => 1 });
+        _add_to_ipbanlist($comment->blog_id, $comment->ip);
+    }
+    $app->call_return;
+}
+
+sub _add_to_ipbanlist {
     my ($blog_id, $ip) = @_;
     unless (MT::IPBanList->load({ blog_id => $blog_id, ip => $ip })) {
@@ -76 +118 @@
 }
 
+sub ipbanlist {
+    my ($ctx, $args) = @_;
+    my @blog_ids = defined $args->{blog_id} ?
+        split /\s*,\s*/, $args->{blog_id} : [ $ctx->stash('blog_id') ];
+    my %ips;
+    for my $blog_id (@blog_ids) {
+        my @list = MT::IPBanList->load({ blog_id => $blog_id });
+        %ips = map { $_->ip => 1 } @list;
+    }
+    my @res;
+    my $builder = $ctx->stash('builder');
+    my $tokens = $ctx->stash('tokens');
+    for my $ip (keys %ips) {
+        local $ctx->{__stash}{'ipbanlist_ip'} = $ip;
+        defined(my $out = $builder->build($ctx, $tokens))
+            or return $ctx->error($ctx->errstr);
+        push @res, $out;
+    }
+    my $glue = $args->{glue} || '';
+    join $glue, @res;
+}
+
+sub ipbanlist_ip {
+    $_[0]->stash('ipbanlist_ip') || '';
+}
+
 1;