perlbuut/plugins/pastebinadmin.pm

package Bot::BB3::Plugin::Pastebinadmin;
use POE::Component::IRC::Common qw/l_irc/;
use DBD::SQLite;
use Bot::BB3::DebugCrypt;
use strict;

sub new {
	my( $class ) = @_;
	my $self = bless {}, $class;
	$self->{name} = "pastebinadmin";
	$self->{opts} = {
		command => 1,
	};

	return $self;
}

sub dbh {
	my( $self, $env ) = @_;

	if( $self->{$env . "dbh"} and $self->{$env . "dbh"}->ping ) {
		return $self->{$env . "dbh"};
	}
  
  my $dbnames = {
    www => "perlbot_pastes",
    dev => "perlbot_pastes_dev",
  };

  $self->{$env . "dbh"} = DBI->connect( "dbi:Pg:dbname=".$dbnames->{$env}, "perlbot_pastebin", "ignored", { PrintError => 2, RaiseError => 1 } );

	return $self->{$env."dbh"};
}
sub postload {
	my( $self, $pm ) = @_;
	
	delete $self->{wwwdbh}; # UGLY HAX GO.
  delete $self->{devdbh};
	                     # Basically we delete the dbh we cached so we don't fork
                         # with one active
}

sub add_ban_word {
  my ($self, $env, $who, $where, $word) = @_;

  $self->dbh($env)->do(q{INSERT INTO banned_words (word, who, "where") VALUES (?, ?, ?)}, {}, $word, $who, $where);
}

sub get_ip_for_paste {
  my ($self, $env, $id) = @_;

  my ($ip) = @{$self->dbh($env)->selectrow_arrayref("SELECT ip FROM posts p JOIN slugs s ON s.post_id = p.id WHERE s.slug = ?", {}, $id) || ['0.0.0.0']};

  return sprintf("%03d.%03d.%03d.%03d", split(/\./,$ip));
}

sub get_asn_for_paste {
  my ($self, $env, $id) = @_;

  my $ip = $self->get_ip_for_paste($env, $id);

  my ($asn) = @{$self->dbh('www')->selectrow_arrayref('SELECT asn FROM asn WHERE ? >= start AND ? <= "end"', {}, $ip, $ip) || []}[0];
  return $asn;
}

sub ban_user_paste {
  my ($self, $env, $id, $who, $where) = @_;

  my $ip = $self->get_ip_for_paste($env, $id);

  if ($ip) {
    $self->dbh($env)->do(q{INSERT INTO banned_ips (ip, who, "where") VALUES (?, ?, ?);}, {}, $ip, $who, $where);
    return "USER WAS BANNED FOR THIS POST";
  } else {
    return "Failed to find IP for paste in db";
  }
}

sub ban_asn_paste {
  my ($self, $env, $id, $who, $where) = @_;

  my $ip  = $self->get_ip_for_paste($env, $id);
  my $asn = $self->get_asn_for_paste($env, $id);

  if ($asn) {
    $self->dbh($env)->do(q{INSERT INTO banned_asns (asn, who, "where") VALUES (?, ?, ?);}, {}, $asn, $who, $where);
    return "ISP WAS BANNED FOR THIS POST";
  } else {
    return "Failed to find ISP for paste in db. yell at simcop2387; ".encrypt("[$id][$ip][$asn]");
  }
}

sub command {
	my( $self, $said, $pm ) = @_;
	my( $cmd ) = join ' ', @{ $said->{recommended_args} };

  my ($env, $command, @args);
  if ($cmd =~ /^\s*(?<dev>--dev)?\s*(?<command>\S+)\s*(?<args>.*?)\s*$/i) {
    $env = $+{dev} ? "dev" : "www";
    $command = $+{command};
    @args = split ' ', $+{args};
  }

  my $where = $said->{server} . $said->{channel};
  my $who = $said->{sender_raw};

  if ($command eq 'banword') {
    $self->add_ban_word($env, $who, $where, $_) for (@args);
    use Data::Dumper;
    return ("handled", "Added words [".join(', ', @args)."] to ban list");
  } elsif ($command eq 'banuser') {
    my $paste = $args[0];
    
    my $id = $paste;
    $id =~ s/^\s+|\s+$//g;
    $id = $1 || $2 if ($paste =~ m{^(?:(?:https?://(?:[a-z\.]+)?(?:perlbot\.pl|perl\.bot)/p(?:astebin)?/([^/]{6,})/?)|([^/]+))$}g);

    if ($id) {
      my $response = $self->ban_user_paste($env, $id, $who, $where);
      return ("handled", $response);
    } else {
      return ("handled", "didn't find an id");
    }
  } elsif ($command eq 'banasn') {
    my $paste = $args[0];
    
    my $id = $paste;
    $id =~ s/^\s+|\s+$//g;
    $id = $1 || $2 if ($paste =~ m{^(?:(?:https?://(?:[a-z\.]+)?(?:perlbot\.pl|perl\.bot)/p(?:astebin)?/([^/]{6,})/?)|([^/]+))$}g);

    if ($id) {
      my $response = $self->ban_asn_paste($env, $id, $who, $where);
      return ("handled", $response);
    } else {
      return ("handled", "didn't find an id");
    }

###    $self->ban_user()
  } else {
    return ("handled", "Failed to parse [$env, $command, @args]");
  }
}

no warnings 'void';
"Bot::BB3::Plugin::Pastebinadmin";

__DATA__
The pastebinadmin plugin.  Lets operators change options on the https://perlbot.pl/ pastebin.  perlbot: pastebinadmin [--dev] <command> [<args>]. see https://github.com/perlbot/perlbuut-pastebin/wiki/Op-Tools