#!/usr/local/cpanel/3rdparty/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

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

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] || !Cpanel::Validate::IP::is_valid_ip_cidr_or_prefix( $ARGV[0] ) ) {
        print "Usage: $app <ip/cidr>\n";
        print "Examples:\n\t\t$app 1.1.1.1\n";
        print "Examples:\n\t\t$app 0:33:ad\n";
        print "\t\t$app 1.1.1.0/24\n";
        exit;
    }

    my $database  = 'cphulkd';
    my $mysqluser = Cpanel::MysqlUtils::getmydbuser('root') || 'root';
    my $mysqlpass = Cpanel::MysqlUtils::getmydbpass('root');
    my $mysqlhost = Cpanel::MysqlUtils::getmydbhost('root') || 'localhost';
    my $dbh;
    eval { $dbh = Cpanel::MysqlUtils::Connect::get_dbi_handle( 'dbuser' => $mysqluser, 'database' => $database, 'dbpass' => $mysqlpass, 'dbserver' => $mysqlhost, ); };
    if ( $@ || !$dbh ) {
        warn "Failed to connect to db: $database: $@";
        exit 1;
    }

    Cpanel::Hulk::Admin::add_ip_to_list( { 'dbh' => $dbh }, $ARGV[0], $list );
    print "$ARGV[0] has been ${list}listed\n";
}
else {
    print "cphulkd is not enabled\n";
}