#!/usr/bin/perl
# cpanel - cphulkdwhitelist                       Copyright(c) 2010 cPanel, Inc.
#                                                           All rights Reserved.
# copyright@cpanel.net                                         http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited

BEGIN { unshift @INC, '/usr/local/cpanel'; }

use strict;
use warnings;
use Cpanel::MysqlUtils   ();
use Cpanel::Hulk::Admin  ();
use Cpanel::Config::Hulk ();
use DBI;

my $app  = $0;
my $list = 'white';
if ( grep( /-black/, @ARGV ) ) {
    $list = 'black';
    @ARGV = grep( !/-black/, @ARGV );
    $app =~ s/white/black/g;
}

if ( Cpanel::Config::Hulk::is_enabled() ) {
    if ( !$ARGV[0] || $ARGV[0] !~ /^\d+\.\d+\.\d+\.\d+(?:\/\d+)?$/ ) {
        print "Usage: $app <ip/cidr>\n";
        print "Examples:\n\t\t$app 1.1.1.1\n";
        print "\t\t$app 1.1.1.0/24\n";
        exit;
    }

    my $database  = 'cphulkd';
    my $mysqlpass = Cpanel::MysqlUtils::getmydbpass('root');
    my $mysqlhost = Cpanel::MysqlUtils::getmydbhost('root') || 'localhost';

    my $dbh = DBI->connect( "DBI:mysql:$database:$mysqlhost", 'root', $mysqlpass );
    Cpanel::Hulk::Admin::list_ip( { 'dbh' => $dbh }, $list, $ARGV[0] );
    print "$ARGV[0] has been ${list}listed\n";
}
else {
    print "cphulkd is not enabled\n";
}