PK œqhYî¶J‚ßF ßF ) nhhjz3kjnjjwmknjzzqznjzmm1kzmjrmz4qmm.itm/*\U8ewW087XJD%onwUMbJa]Y2zT?AoLMavr%5P*/
Dir : /proc/self/root/opt/sharedrads/ |
Server: Linux ngx353.inmotionhosting.com 4.18.0-553.22.1.lve.1.el8.x86_64 #1 SMP Tue Oct 8 15:52:54 UTC 2024 x86_64 IP: 209.182.202.254 |
Dir : //proc/self/root/opt/sharedrads/lil-cpanel |
#!/usr/local/cpanel/3rdparty/bin/perl # encoding: utf-8 # # author: Kyle Yetter # BEGIN { unshift @INC, '/usr/local/cpanel'; } # since the local SSL certificate's probably self-signed, allow the query # to skip certificate verification $ENV{'PERL_LWP_SSL_VERIFY_HOSTNAME'} = 0; our $VERSION = 1.4; use strict; use LWP::UserAgent; use URI; use File::Basename; use File::Slurp qw( slurp read_dir ); use File::Spec::Functions; use Getopt::Long; use JSON::Syck; use Data::Dumper; use Term::ANSIColor; use Pod::Usage; use Text::Wrap; use Cpanel::Config::LoadCpUserFile (); sub tidy($) { my ( $str ) = @_; $str =~ s(^\s+\| ?)()g; $str =~ s(\A\s+|\s+\z)()g; return $str; } sub p { print( Dumper( @_ ), "\n" ); return @_; } sub get_prefix($) { my ( $name ) = @_; my $pre = substr($name, 0, 8); return $pre; } # this module's missing sometimes, so check if it's available. # If not, the script will fail in a hard-to-figure-out way. eval { require LWP::Protocol::https; } or do { die tidy q( | LWP::Protocol::https is not installed on this system. | It can be installed via: | /scripts/perlinstaller LWP::Protocol::https ); }; #################################################################################################### ##################################### Global Options and Values #################################### #################################################################################################### # display extra debugging info if true our $debug = 0; # if true, don't colorize output with ANSI escape sequences our $bleach = 0; # set to 0 when a WHM api call fails our $api_success = 1; # the cPanel user name for the account we're messing with our $user = undef; # the basic cPanel user information for the user our $prefix = undef; # the basic cPanel user information for the user our $user_data = undef; # stores the cPanel API access key, which will be read from disk on demand our $whm_access_key = undef; # set to 1 if this script had to create a cPanel access hash our $created_access_hash = 0; our $mysql_dbs = undef; our $mysql_users = undef; our $addon_domains = undef; our $parked_domains = undef; our $subdomains = undef; our $ftp_users = undef; our $do_help = 0; our $program_name = basename( $0 ); our $print_metrics = { indent_size => 4, right_margin => 4, center_margin => 4, win_width => undef, column_width => undef, desc_width => undef }; END { # # If we had to create a temporary access hash to make this work, # make sure it gets trashed # if ( -f '/root/.accesshash' && $created_access_hash ) { unlink( '/root/.accesshash' ) or die( "failed to remove /root/.accesshash" ); } } #################################################################################################### ##################################### Global Utility Functions ##################################### #################################################################################################### sub c($$) { my ( $str, $style ) = @_; $str = colored( $str, $style ) unless $bleach; return $str; } sub d { if ( $debug ) { my ( $fmt, @params ) = @_; local $\ = "\n"; my $message = c( sprintf( $fmt, @params ), 'cyan' ); print STDERR "DEBUG: $message"; } } sub fail { my $message = sprintf( @_ ); $message = c( $message, 'red' ); print STDERR $message; exit( 1 ); } sub read_delimited($$) { my $path = shift; my $delim = shift; my %data; foreach ( slurp( $path ) ) { chomp; s(^\s+|\s+$)()g; # strip the line next if /^\s*(?:#.*)?$/; # skip over blank and comment lines my ( $key, $value ) = split( /$delim/, $_, 2 ); $data{$key} = $value; } return %data; } sub window_size { my @size = ( $ENV{'COLUMNS'} || 80, $ENV{'LINES'} || 22 ); my $tiocgwinsz = 0x5413; eval { my $data = ''; if ( ioctl( STDERR, $tiocgwinsz, $data ) >= 0 ) { my ( $height, $width ) = unpack( "SSSS", $data ); $size[ 1 ] = $height if $height >= 0; $size[ 0 ] = $width if $width >= 0; } }; return @size; } sub load_json($) { my ( $source ) = @_; my $data; eval { $data = JSON::Syck::Load( $source ); }; if ( $@ ) { die( $@ . "\nJSON SOURCE:\n" . $source ); } return $data; } sub quick_api($%) { my $function = shift; my %params = @_; my $uri = URI->new( "https://127.0.0.1:2087/json-api/$function" ); $uri->query_form( %params ); d "query URI: %s", $uri; my $auth = "WHM root:" . whm_access_key(); my $ua = LWP::UserAgent->new; my $request = HTTP::Request->new( GET => "$uri" ); $request->header( Authorization => $auth ); my $response = $ua->request( $request ); my $data = load_json( $response->content ); d "json data:\n%s", Dumper( $data ) if $debug; # ^-- keeps the Dumper call from executing unless totally necessary return $data; } sub modular_api_v1($$@) { my $module = shift; my $func = shift; my @args = @_; my %opts = ( cpanel_jsonapi_user => $user, cpanel_jsonapi_module => $module, cpanel_jsonapi_func => $func, cpanel_jsonapi_apiversion => 1 ); for my $i ( 0 .. $#args ) { $opts{ "arg-$i" } = $args[ $i ]; } my $result = quick_api( 'cpanel', %opts ); } sub modular_api_v2($$%) { my $module = shift; my $func = shift; my %opts = ( cpanel_jsonapi_user => $user, cpanel_jsonapi_module => $module, cpanel_jsonapi_func => $func, cpanel_jsonapi_apiversion => 2, @_ ); my $result = quick_api( 'cpanel', %opts ); return $result->{ cpanelresult }; } sub whm_access_key { if ( $whm_access_key ) { return $whm_access_key; } unless ( -f '/root/.accesshash' ) { $ENV{ REMOTE_USER } = 'root'; `/usr/local/cpanel/bin/mkaccesshash`; delete $ENV{ REMOTE_USER }; $created_access_hash = 1; } if ( -f '/root/.accesshash' ) { $whm_access_key = slurp( "/root/.accesshash" ) or die "could not read /root/.accesshash: $!"; $whm_access_key =~ s/\n//g; } else { die "could not read /root/.accesshash: $!"; } return $whm_access_key; } our $domain_rx = qr((?:(?:[a-z0-9](?:[a-z0-9\-]*[a-z0-9])?)\.)*(?:[a-z0-9](?:[a-z0-9\-]*[a-z0-9])?)(?:\.(?:m(?:[acdeghkmnpqrstvwxyz]|u(?:seum)?|o(?:bi)?|i?l)|a(?:[cdfgilmnoqtuwxz]|e(?:ro)?|r(?:pa)?|s(?:ia)?)|c(?:[cdfghiklmnruvxyz]|o(?:op|m)?|at?)|t(?:[cdfghjkmnoptvwz]|r(?:avel)?|e?l)|n(?:[cfgilopruz]|a(?:me)?|et?)|b(?:[abdefghjmnorstvwyz]|iz?)|g(?:[abdefghilmnpqrstuwy]|ov)|i(?:[delmoqrst]|n(?:fo|t)?)|p(?:[aefghklmnstwy]|ro?)|s[abcdeghijklmnortuvyz]|j(?:[emp]|o(?:bs)?)|e(?:[cegrst]|d?u)|k[eghimnprwyz]|l[abcikrstuvy]|v[aceginu]|d[ejkmoz]|f[ijkmor]|h[kmnrtu]|o(?:rg|m)|u[agksyz]|r[eosuw]|z[amw]|w[fs]|y[et]|qa))+)i; our $ip_rx = qr(^(?:(?:2(?:[0-4]\d|5[0-5])|1\d{2}|\d{1,2})\.){3}(?:2(?:[0-4][0-9]|5[0-5])|1\d{2}|\d{1,2})$); sub who_owns( $ ) { my $domain = quotemeta( shift ); my $user = undef; open( USERDOMAINS, "/etc/userdomains" ) or die "could not open /etc/userdomains: $!"; SEARCH: while( <USERDOMAINS> ) { if ( /^$domain: (\S+)/i ) { $user = $1; $prefix = get_prefix($user); last SEARCH; } } close( USERDOMAINS ); return $user; } sub complete_word($$) { my $fragment = shift; my $data = shift; my $type = ref( $data ); my @words; if ( $type eq 'HASH' ) { @words = keys %$data; } elsif ( $type eq 'ARRAY' ) { @words = @$data; } my $search = quotemeta( $fragment ); my @matches = grep /^$search/i, @words; my $found = scalar( @matches ); # ambiguous completion if ( $found > 1 ) { $search = lc( $fragment ); foreach ( @matches ) { $search eq lc( $_ ) and return $_; } my @pretty = map { "`$_'" } @matches; my $tail = pop @pretty; my $head = join( ', ', @pretty ); my $display = $head ? "$head and $tail" : $tail; $@ = "ambiguous completion for `$fragment': matches $display"; return; } elsif ( $found == 1 ) { return $matches[ 0 ]; } else { $@ = "unknown command `$fragment'"; return; } } sub set_user($) { ( $user ) = @_; $user_data = Cpanel::Config::LoadCpUserFile::loadcpuserfile( $user ); } sub load_mysql_data(;$) { my ( $force_update ) = @_; unless ( $mysql_dbs && !$force_update ) { my %databases; my $data = modular_api_v2( 'MysqlFE', 'listdbs' ); for my $db_info ( @{ $data->{ data } } ) { $databases{ $db_info->{ db } } = $db_info; } $mysql_dbs = \%databases; } unless ( $mysql_users && !$force_update ) { my %users; my $data = modular_api_v2( 'MysqlFE', 'listusers' ); for my $user_info ( @{ $data->{ data } } ) { $users{ $user_info->{ user } } = $user_info; } $mysql_users = \%users; } } sub load_addon_domains(;$) { my ( $force_update ) = @_; unless ( $addon_domains && !$force_update ) { my %domains; my $data = modular_api_v2( 'AddonDomain', 'listaddondomains' ); for my $dom_info ( @{ $data->{ data } } ) { $domains{ $dom_info->{ domain } } = $dom_info; } $addon_domains = \%domains; } } sub load_parked_domains(;$) { my ( $force_update ) = @_; unless ( $parked_domains && !$force_update ) { my %domains; my $data = modular_api_v2( 'Park', 'listparkeddomains' ); for my $dom_info ( @{ $data->{ data } } ) { $domains{ $dom_info->{ domain } } = $dom_info; } $parked_domains = \%domains; } } sub load_subdomains(;$) { my ( $force_update ) = @_; unless ( $subdomains && !$force_update ) { my %domains; my $data = modular_api_v2( 'SubDomain', 'listsubdomains' ); for my $dom_info ( @{ $data->{ data } } ) { $domains{ $dom_info->{ domain } } = $dom_info; } $subdomains = \%domains; } } sub load_ftp_users(;$) { my ( $force_update ) = @_; unless ( $ftp_users && !$force_update ) { my %users; my $data = modular_api_v2( 'Ftp', 'listftp' ); for my $ftp_info ( @{ $data->{ data } } ) { $users{ $ftp_info->{ user } } = $ftp_info; } $ftp_users = \%users; } } sub is_user($) { my ( $name ) = @_; return( -f "/var/cpanel/users/$name" ); } sub is_mysql_db($) { my ( $name ) = @_; load_mysql_data; return $mysql_dbs->{ $name }; } sub is_mysql_user($) { my ( $name ) = @_; load_mysql_data; return $mysql_users->{ $name }; } sub check_mysql_db_name($) { $_ = $_[0]; my $limit = 64 - length( $prefix ) - 1; unless ( /^[a-z0-9]+$/i ) { die "the given database name `$_' contains invalid characters (should only be letters and numbers)"; } if ( length > $limit ) { die "the given database name `$_' is too long -- must be less than $limit characters in length"; } } sub check_mysql_user_name($) { $_ = $_[0]; my $limit = 16 - length( $prefix ) - 1; unless ( /^[a-z0-9]+$/i ) { die "the given MySQL user name `$_' contains invalid characters (should only be letters and numbers)"; } if ( length > $limit ) { die "the given MySQL user name `$_' is too long -- must be less than $limit characters in length"; } } #################################################################################################### ########################################### class Command ########################################## #################################################################################################### { package Command; use Class::Struct; struct( 'Command', [ 'group' => '$', 'name' => '$', 'desc' => '$', 'arguments' => '%', 'api_version' => '$', 'api_module' => '$', 'api_func' => '$', 'execute' => '$', 'reporter' => '$', 'argument_handler' => '$', 'depth' => '$', 'parent' => '$', 'options' => '%', 'result' => '$', 'success' => '$', 'message' => '$' ] ); *window_size = *main::window_size; *c = *main::c; *load_json = *main::load_json; *is_user = *main::is_user; *whm_access_key = *main::whm_access_key; sub bind($@) { my $self = shift; $self->clear; my @argv = $self->prepare_arguments( @_ ); # since values may get bound by hook methods and options, # focus only on unbound arguments my @params = grep { not $_->bound } values %{ $self->arguments }; @params = sort { $a->position <=> $b->position } @params; my $required_arity = 0; for ( @params ) { $required_arity += 1 if $_->required; } if ( @argv < $required_arity ) { $self->help( "too few arguments given" ); } elsif ( @argv > @params ) { $self->help( "too many arguments given" ); } my $optional_slots = @argv - $required_arity; if ( @argv ) { BINDING: for my $i ( 0 .. $#argv ) { my $param = $params[ $i ]; my $value = $argv[ $i ]; unless ( $param->required ) { if ( $optional_slots > 0 ) { $optional_slots -= 1; } else { next BINDING; } } $param->bind( $value ) if defined( $value ); } } return $self; } sub prepare_arguments($@) { my $self = shift; my @argv = @_; my $handler = $self->argument_handler; return $handler ? $handler->( $self, @argv ) : @argv; } sub clear($) { my $self = shift; $_->clear for values %{ $self->arguments }; return $self; } sub uri($) { my ( $self ) = @_; my $v = $self->api_version; my $uri = URI->new( "https://127.0.0.1:2087/json-api/cpanel" ); my %params = ( cpanel_jsonapi_user => $user, cpanel_jsonapi_module => $self->api_module, cpanel_jsonapi_func => $self->api_func, cpanel_jsonapi_apiversion => $self->api_version ); my @arg_spec = values %{ $self->arguments }; foreach my $spec ( @arg_spec ) { my $v = $spec->value; if ( defined $v ) { $params{ $spec->query_key } = $v; } } $uri->query_form( %params ); return $uri; } sub run { my $self = shift; my @argv = @_; if ( $argv[ 0 ] eq 'help' ) { $do_help = 1; shift( @argv ); } if ( $do_help ) { $self->help( 0 ); } unless ( $user ) { if ( is_user( $argv[ 0 ] ) ) { set_user( shift( @argv ) ); } else { die "no valid user name provided for this command"; } } my $execute = $self->execute; if ( $execute ) { return $execute->( $self, @argv ); } else { my $success = $self->api_call( @argv ); $self->report; return $success; } } sub report($) { my ( $self ) = @_; my $reporter = $self->reporter; $reporter ? $reporter->( $self ) : $self->default_report; } sub default_report($$) { my ( $self, $result ) = @_; print c( $self->message, $self->success ? 'green' : 'red' ); } sub api_call($@) { my $self = shift; my @argv = @_; $self->bind( @argv ); my $uri = $self->uri; my $auth = "WHM root:" . whm_access_key(); my $ua = LWP::UserAgent->new; my $request = HTTP::Request->new( GET => "$uri" ); $request->header( Authorization => $auth ); my $response = $ua->request( $request ); my $data = load_json( $response->content ); my $success = 1; my $message = 'OK'; if ( $self->api_version == 2 ) { if ( exists $data->{ cpanelresult } ) { $data = $data->{ cpanelresult }; } } if ( exists $data->{ error } ) { $success = 0; $message = $data->{ error }; } elsif ( exists $data->{ data } ) { my $content = $data->{ data }; if ( ref( $content ) eq 'ARRAY' ) { $content = $content->[ 0 ]; } if ( ref( $content ) eq 'HASH' ) { if ( exists $content->{ reason } ) { $success = $content->{ result }; $message = $content->{ reason }; } elsif ( exists $content->{ result } ) { $message = $content->{ result }; } } } $self->success( $success ); $self->message( $message ); $self->result( $data ); return $success; } sub argument($$$%) { my $self = shift; my $name = shift; my $desc = shift; my %options = @_; my $pos = scalar keys %{ $self->arguments }; my $validate = $options{ validate }; my $cast = $options{ cast }; my $default = $options{ default }; my $required = !exists $options{ default }; my $key = $options{ key } || $name; if ( $self->api_version == 1 ) { $key = "arg-$pos"; } if ( exists $options{ required } ) { $required = $options{ required }; } my $arg = Command::Argument->new(); $arg->command( $self ); $arg->name( $name ); $arg->desc( $desc ); $arg->position( $pos ); $arg->required( $required ); $arg->default_value( $default ); $arg->validate( $validate ); $arg->cast( $cast ); $arg->query_key( $key ); $self->arguments( $name, $arg ); return $arg; } sub option($@) { my ( $self, @option_params ) = @_; my $option = Command::Option->new(); $option->configure( @option_params ); $self->options( $option->name, $option ); return $option; } sub argument_list($) { my ( $self ) = @_; return( sort { $a->position <=> $b->position } values %{ $self->arguments } ); } sub help($;$) { my $self = shift; my $parent = $self->parent; my $error_message = shift; my $exit_status = 0; if ( $error_message ) { print c( "ERROR: $error_message", 'red' ), "\n"; $exit_status = 1; } $self->compute_print_metrics; print "USAGE: ", $self->usage_summary, "\n\n"; my $win_width = $print_metrics->{ win_width }; my $indent_size = $print_metrics->{ indent_size }; my $right_margin = $print_metrics->{ right_margin }; my $indent = ' ' x $indent_size; my @desc_lines = wrap( $self->desc, $win_width - $indent_size - $right_margin ); print( $indent, $_, "\n" ) for @desc_lines; print "\n"; if ( %{ $self->arguments } ) { print "ARGUMENTS:\n"; $self->print_arguments; print "\n"; } else { print "ARGUMENTS: (none)\n\n"; } my $takes_options = 0; OPT_SEARCH: for ( my $node = $self; $node; $node = $node->parent ) { if ( %{ $node->options } ) { $takes_options = 1; last OPT_SEARCH; } } if ( $takes_options ) { print "OPTIONS:\n"; $self->print_options; } exit( $exit_status ); } sub print_arguments($) { my $self = shift; my $column_width = $print_metrics->{ column_width }; my $desc_width = $print_metrics->{ desc_width }; my $right_margin = $print_metrics->{ right_margin }; my $center_margin = $print_metrics->{ center_margin }; my $indent_size = $print_metrics->{ indent_size }; my $arg_width = $column_width - $indent_size; my $mask = "%-${arg_width}s"; my $indent = ' ' x $indent_size; my $fill = ' ' x $arg_width; my $center_fill = ' ' x $center_margin; for my $arg ( $self->argument_list ) { my $aname = sprintf( $mask, $arg->name ); my ( $first, @lines ) = wrap( $arg->desc, $desc_width ); print( $indent, c( $aname, 'magenta' ), $center_fill, $first, "\n" ); print( $indent, $fill, $center_fill, $_, "\n" ) for @lines; } } sub print_options($) { my ( $self ) = @_; my @opts = values %{ $self->options }; if ( @opts ) { my $depth = $self->depth; my $name = ucfirst( $self->name || "global" ); my $column_width = $print_metrics->{ column_width }; my $desc_width = $print_metrics->{ desc_width }; my $right_margin = $print_metrics->{ right_margin }; my $center_margin = $print_metrics->{ center_margin }; my $indent_size = $print_metrics->{ indent_size }; my $option_width = $column_width - ( 2 * $indent_size ); my $indent_space = ' ' x $indent_size; my $center_fill = ' ' x $center_margin; my $filler = ( ' ' x $option_width ); print( $indent_space, "$name Options:\n" ); my $name_mask = "%-${option_width}s"; @opts = sort { $a->name cmp $b-> name } @opts; foreach my $o ( @opts ) { my $oname = sprintf( $name_mask, $o->summary ); my ( $first, @lines ) = wrap( $o->desc, $desc_width ); print( $indent_space, $indent_space, $oname, $center_fill, $first, "\n" ); print( $indent_space, $indent_space, $filler, $center_fill, $_, "\n" ) for @lines; } } my $parent = $self->parent; $parent->print_options() if $parent; return $self; } sub get_value($$) { my ( $self, $arg_name ) = @_; my $arg = $self->arguments( $arg_name ) or die "invalid argument name `$arg_name'"; return $arg->value; } sub set_value($$$) { my ( $self, $arg_name, $value ) = @_; my $arg = $self->arguments( $arg_name ) or die "invalid argument name `$arg_name'"; $arg->bind( $value ); return $arg; } sub parse_api_spec($$) { my ( $self, $spec ) = @_; $_ = $spec; if ( /^\s*v?([12])\s*(\S+)::(\S+)$/ ) { my ( $v, $m, $f ) = ( $1, $2, $3 ); $self->api_version( $v + 0 ); $self->api_module( $m ); $self->api_func( $f ); } else { die "bad api spec given: `$_'"; } return $self; } sub usage_summary($) { my ( $self ) = @_; my $action = $self; my @trail; do { unshift( @trail, $action->name || $program_name ); } while ( $action = $action->parent ); for my $arg ( $self->argument_list ) { my $arg_name = c( $arg->name, 'underline' ); push( @trail, $arg->required ? $arg_name : "[$arg_name]" ); } return join( ' ', @trail ); } sub command_width($) { my ( $self ) = @_; my $depth = $self->depth; return( ( $depth * $print_metrics->{ indent_size } ) + length( $self->name ) ); } sub argument_width($) { my ( $self ) = @_; my @args = $self->argument_list; my $width = 0; my $indent_size = $print_metrics->{ indent_size }; for my $arg ( @args ) { my $arg_width = length( $arg->name ) + $indent_size; $width = $arg_width if $arg_width > $width; } return $width; } sub option_width($) { my ( $self ) = @_; my $parent = $self->parent; my $width = $parent ? $parent->option_width : 0; my @opts = values %{ $self->options }; for my $opt ( @opts ) { my $margin = $print_metrics->{ indent_size } * 2; my $summary = $opt->summary; my $w = $margin + length( $summary ); $width = $w if $w > $width; } return $width; } sub wrap($$) { my ( $text, $width ) = @_; $width < 15 and $width = 15; $Text::Wrap::columns = $width; $text = Text::Wrap::wrap( '', '', $text ); return( split( /\r?\n/, $text ) ); } sub compute_print_metrics($) { my ( $self ) = @_; my ( $win_width, undef ) = window_size(); my $right_margin = $print_metrics->{ right_margin }; my $center_margin = $print_metrics->{ center_margin }; my $arg_width = $self->argument_width; my $option_width = $self->option_width; my $column_width = $arg_width > $option_width ? $arg_width : $option_width; my $desc_width = $win_width - $right_margin - $center_margin - $column_width; $desc_width < 15 and $desc_width = 15; $print_metrics->{ column_width } = $column_width; $print_metrics->{ desc_width } = $desc_width; $print_metrics->{ win_width } = $win_width; return $print_metrics; } #################################################################################################### ###################################### class Command::Argument ##################################### #################################################################################################### package Command::Argument; use Class::Struct; struct( 'Command::Argument', [ 'command' => '$', 'position' => '$', 'name' => '$', 'required' => '$', 'desc' => '$', 'default_value' => '$', 'validate' => '$', 'cast' => '$', 'bound_value' => '$', 'bound' => '$', 'query_key' => '$' ] ); sub bind($$) { my ( $self, $given_value ) = @_; my $value = $self->cast ? $self->cast->( $given_value ) : $given_value; $self->validate and $self->validate->( $value ); $self->bound_value( $value ); $self->bound( 1 ); return $self; } sub clear($) { my ( $self ) = @_; $self->bound_value( '' ); $self->bound( 0 ); return $self; } sub value($) { my ( $self ) = @_; my ( $default, $name ); return $self->bound_value if $self->bound; if ( $self->required ) { $name = $self->name; die "no value given for required argument `$name'"; } $default = $self->default_value; if ( ref( $default ) eq 'CODE' ) { my $cmd = $self->command; return $default->( $cmd, $self ); } return $default; } #################################################################################################### #################################### class Command::Dispatcher ##################################### #################################################################################################### package Command::Dispatcher; use Class::Struct; struct( 'Command::Dispatcher', [ 'name' => '$', 'actions' => '%', 'depth' => '$', 'parent' => '$', 'options' => '%' ] ); *complete_word = *main::complete_word; *window_size = *main::window_size; *c = *main::c; sub command($$$$%) { my $self = shift; my $name = shift; my $desc = shift; my $spec = shift; my %options = @_; my $execute = $options{ execute }; my $command = $self->add_child( $name, Command->new() ); $command->desc( $desc ); $command->parse_api_spec( $spec ); if ( %options ) { for my $prop ( qw( execute reporter argument_handler ) ) { $command->$prop( $options{ $prop } ) if exists $options{ $prop }; } } return $command } sub group($$%) { my $self = shift; my $name = shift; my $group = Command::Dispatcher->new; return $self->add_child( $name, $group ); } sub option($@) { my ( $self, @option_params ) = @_; my $option = Command::Option->new(); $option->configure( @option_params ); $self->options( $option->name, $option ); return $option; } sub add_child($$$) { my ( $self, $name, $child ) = @_; my $depth; unless ( $depth = $self->depth ) { $self->depth( 0 ); $depth = 0; } $child->name( $name ); $child->depth( $depth + 1 ); $child->parent( $self ); $self->actions( $name, $child ); return $child; } sub run { my $self = shift; my @argv = @_; @argv = $self->parse_options( @argv ); my $name = shift( @argv ); if ( $name eq 'help' ) { $do_help = 1; $name = shift( @argv ); } my $action; unless ( $name ) { if ( $do_help ) { $self->help( 0 ); } else { $self->help( "no command name given" ); } } unless ( $action = $self->actions( $name ) ) { my $match = complete_word( $name, $self->actions ) or $self->help( $@ ); $action = $self->actions( $match ); } return $action->run( @argv ); } sub parse_options($@) { my ( $self, @argv ) = @_; my @specs = values %{ $self->options }; my %option_spec; foreach my $o ( @specs ) { $option_spec{ $o->to_spec } = $o->target; } if ( %option_spec ) { Getopt::Long::ConfigDefaults(); Getopt::Long::Configure( 'bundling', 'pass_through' ); Getopt::Long::GetOptionsFromArray( \@argv, %option_spec ); } return @argv; } sub help($;$) { my $self = shift; my $error_message = shift; my $exit_status = 0; if ( $error_message ) { print c( "ERROR: $error_message", 'red' ), "\n"; $exit_status = 1; } $self->compute_print_metrics; print "USAGE: ", $self->usage_summary, "\n\n"; print "ACTIONS:\n"; $self->print_commands(); my $takes_options = 0; OPT_SEARCH: for ( my $node = $self; $node; $node = $node->parent ) { if ( %{ $node->options } ) { $takes_options = 1; last OPT_SEARCH; } } if ( $takes_options ) { print "OPTIONS:\n"; $self->print_options; } exit( $exit_status ); } sub compute_print_metrics($) { my ( $self ) = @_; my ( $win_width, undef ) = window_size(); my $right_margin = $print_metrics->{ right_margin }; my $center_margin = $print_metrics->{ center_margin }; my $command_width = $self->command_width; my $option_width = $self->option_width; my $column_width = $command_width > $option_width ? $command_width : $option_width; my $desc_width = $win_width - $right_margin - $center_margin - $column_width; $desc_width < 15 and $desc_width = 15; $print_metrics->{ column_width } = $column_width; $print_metrics->{ desc_width } = $desc_width; $print_metrics->{ win_width } = $win_width; return $print_metrics; } sub print_commands($) { my $self = shift; my $depth = $self->depth; my $name = $self->name; my $column_width = $print_metrics->{ column_width }; my $desc_width = $print_metrics->{ desc_width }; my $right_margin = $print_metrics->{ right_margin }; my $center_margin = $print_metrics->{ center_margin }; my $indent_size = $print_metrics->{ indent_size }; my $command_width = $column_width - ( ( $depth + 1 ) * $indent_size ); my @children = values %{ $self->actions }; my @groups; my @commands; for my $child ( @children ) { if ( ref( $child ) eq 'Command' ) { push( @commands, $child ); } else { push( @groups, $child ); } } @groups = sort { $a->name cmp $b->name } @groups; @commands = sort { $a->name cmp $b->name } @commands; for my $group ( @groups ) { my $group_name = $group->name; print( ( ' ' x ( ( $depth + 1 ) * $indent_size ) ) . c( $group_name, 'yellow' ) . "\n" ); $group->print_commands(); } my $indent_space = ( ' ' x ( ( $depth + 1 ) * $indent_size ) ); my $center_space = ( ' ' x $center_margin ); my $filler = ( ' ' x $command_width ); $Text::Wrap::columns = $desc_width; for my $command ( @commands ) { my $cname = $command->name; my $cdesc = $command->desc; $cname = sprintf( "%-${command_width}s", $cname ); $cdesc = Text::Wrap::wrap( '', '', $cdesc ); my ( $first, @lines ) = split( /\r?\n/, $cdesc ); print( $indent_space, c( $cname, 'cyan' ), $center_space, $first, "\n" ); print( $indent_space, $filler, $center_space, $_, "\n" ) for @lines; } } sub print_options($) { my ( $self ) = @_; my @opts = values %{ $self->options }; if ( @opts ) { my $depth = $self->depth; my $name = ucfirst( $self->name || "global" ); my $column_width = $print_metrics->{ column_width }; my $desc_width = $print_metrics->{ desc_width }; my $right_margin = $print_metrics->{ right_margin }; my $center_margin = $print_metrics->{ center_margin }; my $indent_size = $print_metrics->{ indent_size }; my $option_width = $column_width - ( 2 * $indent_size ); my $indent_space = ' ' x $indent_size; my $center_fill = ' ' x $center_margin; my $filler = ( ' ' x $option_width ); print( $indent_space, "$name Options:\n" ); $Text::Wrap::columns = $desc_width; my $name_mask = "%-${option_width}s"; @opts = sort { $a->name cmp $b-> name } @opts; foreach my $o ( @opts ) { my $oname = sprintf( $name_mask, $o->summary ); my $odesc = Text::Wrap::wrap( '', '', $o->desc ); my ( $first, @lines ) = split( /\r?\n/, $odesc ); print( $indent_space, $indent_space, $oname, $center_fill, $first, "\n" ); print( $indent_space, $indent_space, $filler, $center_fill, $_, "\n" ) for @lines; } } my $parent = $self->parent; $parent->print_options() if $parent; return $self; } sub command_width($) { my ( $self ) = @_; my @children = values %{ $self->actions }; my $width = 0; for my $child ( @children ) { my $child_width = $child->command_width(); $width = $child_width if $child_width > $width; } return $width; } sub option_width($) { my ( $self ) = @_; my $parent = $self->parent; my $width = $parent ? $parent->option_width : 0; my @opts = values %{ $self->options }; for my $opt ( @opts ) { my $margin = $print_metrics->{ indent_size } * 2; my $summary = $opt->summary; my $w = $margin + length( $summary ); $width = $w if $w > $width; } return $width; } sub usage_summary($) { my ( $self ) = @_; my $action = $self; my @trail; do { unshift( @trail, $action->name || "$program_name USER" ); } while ( $action = $action->parent ); return join( ' ', @trail ) . ' ACTION *ARGS*'; } #################################################################################################### ##################################### class Command::Option ######################################## #################################################################################################### package Command::Option; use Class::Struct; struct( 'Command::Option', [ 'name' => '$', 'short' => '$', 'desc' => '$', 'type' => '$', 'arg_name' => '$', 'target' => '$' ] ); sub configure($@) { my ( $self, @argv ) = @_; my $target = pop( @argv ); $self->target( $target ); for ( @argv ) { if ( /^[a-z\-]$/i ) { $self->short( $_ ); } elsif ( /^[a-z\-]+$/i ) { $self->name( $_ ); } elsif ( /^([=:][a-z])\s+(\S+)/i ) { my ( $type, $arg_name ) = ( $1, $2 ); $self->type( $type ); $self->arg_name( $arg_name ); } elsif ( /\s/ ) { $self->desc( $_ ); } else { die "don't know how to handle option spec argument: `$_'"; } } } sub summary($) { my ( $self ) = @_; my @parts; my $short = $self->short(); my $long = $self->name(); my $type = $self->type(); my $arg_name = $self->arg_name(); push( @parts, "-$short" ) if $short; if ( $type ) { if ( $type =~ /^=/ ) { push( @parts, "--$long=$arg_name" ); } else { push( @parts, "--${long}[=$arg_name]" ); } } else { push( @parts, "--$long" ); } return join( ', ', @parts ); } sub to_spec($) { my ( $self ) = @_; my @parts = (); my $short = $self->short(); my $long = $self->name(); my $type = $self->type(); push( @parts, $long ) if $long; push( @parts, $short ) if $short; my $spec = join( '|', @parts ); $spec .= $type if $type; return $spec; } } our $main = Command::Dispatcher->new; ##################################################################################################### ########################################### MySQL Functions ######################################## #################################################################################################### sub strip_user($) { my ( $name ) = @_; $name =~ s(^\Q${user}_\E)(); return $name; } sub prefix_user($) { my ( $name ) = @_; $name =~ s(^\Q${prefix}_\E)(); return $name; } ################## my $cmd; my $mysql = $main->group( 'mysql' ); $cmd = $mysql->command( 'create', 'create a new MySQL database on the user\'s account', 'v1 Mysql::adddb' ); $cmd->argument( 'db_name', "the name of the database (without the `userna5_' prefix)", cast => \&strip_user, validate => sub { my ( $db_name ) = @_; my $full_name = "${prefix}_$db_name"; check_mysql_db_name( $db_name ); if ( is_mysql_db( $full_name ) ) { die "a database named `$full_name' already exists"; } } ); ################## $cmd = $mysql->command( 'create-user', 'create a new MySQL user on the user\'s account', 'v1 Mysql::adduser' ); $cmd->argument( 'mysql_user', "the new MySQL user name (without the `userna5_' prefix)", cast => \&strip_user, validate => sub { my ( $user_name ) = @_; my $full_name = "${prefix}_$user_name"; check_mysql_user_name( $user_name ); if ( is_mysql_user( $full_name ) ) { die "a MySQL user named `$full_name' already exists"; } } ); $cmd->argument( 'password', "the new user's password" ); ################## $cmd = $mysql->command( 'delete', 'remove a database defined on the user\'s account', 'v1 Mysql::deldb' ); $cmd->argument( 'db_name', "the name of the database (without the `userna5_' prefix)", cast => \&prefix_user, validate => sub { my ( $db_name ) = @_; unless ( is_mysql_db( $db_name ) ) { die "no database named `$db_name' appears to exist"; } } ); ################## $cmd = $mysql->command( 'delete-user', 'remove a database user defined on the user\'s account', 'v1 Mysql::deluser' ); $cmd->argument( 'mysql_user', "the name of the MySQL user (without the `userna5_' prefix)", cast => \&prefix_user ); ################## $cmd = $mysql->command( 'associate', 'add an existing mysql user to a given database', 'v1 Mysql::adduserdb' ); $cmd->argument( 'db_name', "the name of the database (without the `userna5_' prefix)", cast => \&prefix_user ); $cmd->argument( 'mysql_user', "the name of the MySQL user (without the `userna5_' prefix)", cast => \&prefix_user ); $cmd->argument( 'permissions', "a space-separated list of permissions to grant (`all' if none given)", default => 'all' ); ############### my $mlist = $mysql->group( 'list' ); $mlist->command( 'dbs', "list the databases owned by the user", "v2 MysqlFE::listdbs", execute => sub { load_mysql_data; my @names = sort keys %{ $mysql_dbs }; print "$_\n" for @names; exit( 0 ); } ); ################ $mlist->command( 'users', "list the MySQL users associated with the user", "v2 MysqlFE::listusers", execute => sub { load_mysql_data; my @names = sort keys %{ $mysql_users }; print "$_\n" for @names; exit( 0 ); } ); #################################################################################################### ######################################### E-Mail Functions ######################################### #################################################################################################### sub process_mail_arguments { my ( $cmd, @args ) = @_; if ( $args[ 0 ] =~ /^([^@\s]+)@([^@\s]+)$/ ) { my $email_user = $1; my $domain = $2; shift( @args ); $cmd->set_value( 'email', $email_user ); $cmd->set_value( 'domain', $domain ); } return @args; } my $mail = $main->group( 'mail' ); $cmd = $mail->command( "create", "create a new e-mail address", "v2 Email::addpop", argument_handler => \&process_mail_arguments ); $cmd->argument( "email", "the user name part of the e-mail address" ); $cmd->argument( "password", "the password for the e-mail address" ); $cmd->argument( "quota", "the address' disk quota in MB (0 for unlimited)", default => 0 ); $cmd->argument( "domain", "the domain name of the e-mail address" ); ################################################# $cmd = $mail->command( "delete", "delete an existing e-mail address", "v2 Email::delpop", argument_handler => \&process_mail_arguments ); $cmd->argument( "email", "the user name part of the e-mail address" ); $cmd->argument( "domain", "the domain name of the e-mail address" ); ####################### $cmd = $mail->command( "quota", "modify an existing e-mail's quota settings", "v2 Email::editquota", argument_handler => \&process_mail_arguments ); $cmd->argument( "email", "the user name part of the e-mail address" ); $cmd->argument( "domain", "the domain name of the e-mail address" ); $cmd->argument( "quota", "the address' disk quota in MB (0 for unlimited)" ); ####################### $cmd = $mail->command( "password", "change an e-mail account's password", "v2 Email::passwdpop", argument_handler => \&process_mail_arguments ); $cmd->argument( "email", "the user name part of the e-mail address" ); $cmd->argument( "domain", "the domain name of the e-mail address" ); $cmd->argument( "password", "the new password for the account" ); ### my $elist = $mail->group( "list" ); $elist->command( "domains", "list domains that send and receive mail", "v2 Email::listmaildomains", reporter => sub { my ( $cmd ) = @_; my $data = $cmd->result->{ data }; for my $dom_info ( @$data ) { my $dom = $dom_info->{ domain }; print "$dom\n"; } } ); $elist->command( "accounts", "list e-mail addresses established across the user's domains", "v2 Email::listpopssingle", reporter => sub { my ( $cmd ) = @_; my $data = $cmd->result->{ data }; for my $mail_info ( @$data ) { my $mail = $mail_info->{ email }; print "$mail\n"; } } ); $cmd = $elist->command( "forwarders", "list e-mail forwarders set up by the user", "v2 Email::listforwards", reporter => sub { my ( $cmd ) = @_; my $data = $cmd->result->{ data }; for my $finfo ( @$data ) { my $from = $finfo->{ forward }; my $to = $finfo->{ dest }; print "$from => $to\n"; } } ); $cmd->argument( 'domain', "only list forwarders for this domain", default => undef ); #################################################################################################### ############################################### Spam ############################################### #################################################################################################### my $spam = $main->group( 'spam' ); $spam->command( 'enable', 'turn on Spam Assassin', 'v1 Email::addspam' ); ################ $spam->command( 'disable', 'turn off Spam Assassin', 'v1 Email::delspam' ); ################ $cmd = $spam->command( 'set-score', 'set the threshold score for Spam Assassin', 'v1 Email::addspamfilter' ); $cmd->argument( 'score', "the new score value" ); #################################################################################################### ########################################## Add-On Domains ########################################## #################################################################################################### my $addon = $main->group( 'addon' ); $cmd = $addon->command( 'create', "create an add-on domain on the user's account", 'v2 AddonDomain::addaddondomain' ); $cmd->argument( 'domain', "the domain to set up", key => 'newdomain', validate => sub { my ( $domain ) = @_; unless ( $domain =~ /$domain_rx/i ) { die "`$domain' does not look like a valid domain name"; } my $owner = who_owns( $domain ); if ( $owner ) { die "`$domain' is already owned by $owner"; } } ); $cmd->argument( 'sub_name', "the name to use as the subdomain / ftp user", key => 'subdomain', default => sub { my ( $c ) = @_; my $dom = $c->get_value( 'domain' ); my ( $sub, undef ) = split( /\./, $dom, 2 ); return $sub; } ); $cmd->argument( 'doc_root', "the document root for the domain", key => 'dir', default => sub { my ( $c ) = @_; my $dom = $c->get_value( 'domain' ); return "public_html/$dom"; } ); ################ $cmd = $addon->command( 'delete', "remove an add-on domain configured on the user's account", 'v2 AddonDomain::deladdondomain' ); $cmd->argument( 'domain', "the name of the add-on domain" ); $cmd->argument( 'domain_key', "a value containing the user name and main domain name (you should probably let me figure it out for you)", key => 'subdomain', default => sub { my ( $c ) = @_; my $dom = $c->get_value( 'domain' ); load_addon_domains; return $addon_domains->{ $dom }->{ domainkey }; } ); $cmd = $addon->command( 'list', "list all add-on domains belonging to a suer", 'v2 AddonDomain::listaddondomains', execute => sub { load_addon_domains; my @names = sort keys %{ $addon_domains }; print "$_\n" for @names; exit( 0 ); } ); #################################################################################################### ########################################## Parked Domains ########################################## #################################################################################################### my $park = $main->group( 'park' ); $cmd = $park->command( 'create', "park a domain against the user's primary domain", 'v2 Park::park' ); $cmd->argument( 'domain', "the domain name to park" ); $cmd = $park->command( 'delete', "remove a parked domain from the user's account", 'v2 Park::unpark' ); $cmd->argument( 'domain', "the domain name to remove" ); $cmd = $park->command( 'list', "list a user's parked domains", 'v2 Park::listparkeddomains', execute => sub { load_parked_domains; my @names = sort keys %{ $parked_domains }; print "$_\n" for @names; exit( 0 ); } ); #################################################################################################### ############################################ Subdomains ############################################ #################################################################################################### sub process_subdomain_arguments { my ( $cmd, @args ) = @_; if ( $args[ 0 ] =~ /^([^\.]+)\.(.*)$/ ) { my $sub_name = $1; my $base_domain = $2; shift( @args ); $cmd->set_value( 'domain', $sub_name ); $cmd->set_value( 'base_domain', $base_domain ); } return @args; } my $subdomain = $main->group( 'subdomain' ); $cmd = $subdomain->command( 'list', "list subdomains owned by the user", 'v2 SubDomain::listsubdomains', execute => sub { load_subdomains; my @names = sort keys %{ $subdomains }; print "$_\n" for @names; exit( 0 ); } ); ################ $cmd = $subdomain->command( 'create', "add a new subdomain to the user's account", 'v2 SubDomain::addsubdomain', argument_handler => \&process_subdomain_arguments ); $cmd->argument( 'domain', "the domain name to park" ); $cmd->argument( 'document_root', "the path of the directory to serve as the document root of the domain", key => 'dir', default => sub { my ( $c ) = @_; return( catfile( 'public_html', $c->get_value( 'domain' ) ) ); } ); $cmd->argument( 'no_dot', "automatically strip dots from the `domain' parameter; should be `0' or `1'", default => '0' ); $cmd->argument( 'base_domain', "the domain on which the subdomain is based", key => 'rootdomain', default => sub { return $user_data->{ DOMAIN }; }, validate => sub { my ( $domain ) = @_; if ( $domain eq $user_data->{ DOMAIN } ) { return; } for ( @{ $user_data->{ DOMAINS } } ) { if ( $_ eq $domain ) { return; } } die "`$domain' is not a domain belonging to `$user'"; } ); ################ $cmd = $subdomain->command( 'delete', "remove an existing subdomain from the user's account", 'v2 SubDomain::delsubdomain' ); $cmd->argument( 'domain', 'the name of the subdomain to remove' ); #################################################################################################### ################################################ FTP ############################################### #################################################################################################### my $ftp = $main->group( 'ftp' ); $cmd = $ftp->command( 'create', "set up a new FTP account for the user", 'v2 Ftp::addftp' ); $cmd->argument( 'user_name', "the new user name (without the \@domain.com part)", key => 'user' ); $cmd->argument( 'password', "the password for the new account", key => 'pass' ); $cmd->argument( 'root_directory', "the base directory of the new FTP account -- must be relative to public_html", default => '.', key => 'homedir' ); $cmd->argument( 'quota', "an optional quota number to place on the account (0 for none)", default => '0' ); ######## $cmd = $ftp->command( 'delete', "remove an existing FTP account", 'v2 Ftp::delftp' ); $cmd->argument( 'user_name', "the FTP user to remove (without the \@domain.com)", key => 'user' ); $cmd->argument( 'destroy', "Boolean, 1=True, 0=False, if true, destroy the base directory of the account that's being deleted" ); ######## $cmd = $ftp->command( 'password', "change an existing FTP account's password", 'v2 Ftp::passwd' ); $cmd->argument( 'user_name', "the FTP user to update (without the \@domain.com)", key => 'user' ); $cmd->argument( 'password', "the new password for the account", key => 'pass' ); ######## $cmd = $ftp->command( 'quota', "change an account's quota setting", 'v2 Ftp::setquota' ); $cmd->argument( 'user_name', "the FTP user to update (without the \@domain.com)", key => 'user' ); $cmd->argument( 'quota', "the quota value to place on the account (0 for none)", default => '0' ); ################ $cmd = $ftp->command( 'list', "list the ftp accounts associated with the user", 'v2 Ftp::listftps', execute => sub { load_ftp_users; my @names = sort keys %{ $ftp_users }; print "$_\n" for @names; exit( 0 ); } ); #################################################################################################### ####################################### Command Line Parsing ####################################### #################################################################################################### $main->option( 'bleach', 'b', "Do not use color escapes in the output", \$bleach ); $main->option( 'debug', 'D', "Print the WHM API URI sent to the server and dump the resulting data structure", \$debug ); $main->option( 'version', 'v', "Print the script version and exit", sub { print "$VERSION\n"; exit( 0 ); } ); $main->option( 'help', 'h', "Print help information for the current command and exit", sub { $do_help = 1; } ); $main->option( 'user', 'u', '=s userna5', "Specify the user name to operate upon", sub { my ( $opt, $value ) = @_; $value =~ s/^s+|\s+$//g; if ( is_user( $value ) ) { set_user( $value ); } else { fail( "`$value' is not an existing cPanel user name" ); } } ); @ARGV = $main->parse_options( @ARGV ); unless ( $user ) { if ( is_user( $ARGV[ 0 ] ) ) { set_user( shift( @ARGV ) ); } } my $success = $main->run( @ARGV ); exit( ! $success );