File Coverage

blib/lib/WWW/MediaTemple.pm

Criterion	Covered	Total	%
statement	7	9	77.7
branch			n/a
condition			n/a
subroutine	3	3	100.0
pod			n/a
total	10	12	83.3

package WWW::MediaTemple;

use Carp qw(croak);
use JSON;
use XML::Simple;

use base qw(REST::Client);

our $VERSION = 0.02;

sub new {
    my ( $class, %args ) = @_;

    if ( !$args{api_key} ) { croak "new method requires an api key"; }

    my %parent_args = (
        host         => 'https://api.mediatemple.net/api/v1',
        pretty_print => 'true',
        wrap_root    => 'true',
        format       => 'json',
        raw_data     => '',
        key          => ''
    );

    # we prefer the following verbage
    $parent_args{host} = $args{base_host} if $args{base_host};
    $parent_args{key}  = $args{api_key}   if $args{api_key};

    # assign args to parent
    while ( my ( $key, $value ) = each %args ) {
        if ( exists $parent_args{$key} ) {
            $parent_args{$key} = $value;
        }
    }

    my $self = $class->SUPER::new( { host => $parent_args{host} } );
    bless $self, $class;

    foreach my $params ( keys %parent_args ) {
        $self->{$params} = $parent_args{$params};
    }

    $self->getUseragent()->agent("perl-WWW-MediaTemple/$VERSION");
    $self->{headers} = {
        'Content-type' => "application/$self->{format}",
        Accept         => "application/$self->{format}",
        Authorization  => "MediaTemple $self->{key}"
    };

    $self->{params} =
      "prettyPrint=$self->{pretty_print}&wrapRoot=$self->{wrap_root}";

    return $self;
}

sub services {
    my ($self) = @_;

    $self->GET( "/services?" . $self->{params}, $self->{headers} );

    return _dumper($self);
}

sub serviceIds {
    my ($self) = @_;

    $self->GET( "/services/ids?" . $self->{params}, $self->{headers} );

    return _dumper($self);
}

sub service {
    my ( $self, %args ) = @_;

    if ( !$args{serviceId} ) {
        croak "service method requires a serviceId parameter";
    }

    $self->GET( "/services/$args{serviceId}?" . $self->{params},
        $self->{headers} );

    return _dumper($self);
}

sub serviceTypes {
    my ($self) = @_;

    $self->GET( "/services/types?" . $self->{params}, $self->{headers} );

    return _dumper($self);
}

sub serviceOSTypes {
    my ($self) = @_;

    $self->GET( '/services/types/os?' . $self->{params}, $self->{headers} );

    return _dumper($self);
}

sub addService {
    my ( $self, %args ) = @_;

    my $b_content = undef;

    # prepare xml/json
    if ( $self->{format} eq 'xml' ) {
        $b_content = "$args{serviceType}"
          . "$args{primaryDomain};";

        if ( $args{operatingSystem} ) {
            $param .=
              "$args{operatingSystem}";
        }

        $param .= "";
    }
    else {
        $b_content = "{ \"serviceType\": $args{serviceType}, "
          . "\"primaryDomain\": \"$args{primaryDomain};\"";

        if ( $args{operatingSystem} ) {
            $param .= ", \"operatingSystem\": \"$args{operatingSystem}\"";
        }

        $param .= " }";
    }

    $self->POST( '/services' . $self->{params}, $b_content, $self->{headers} );

    return _dumper($self);
}

sub reboot {
    my ( $self, %args ) = @_;

    if ( !$args{serviceId} ) {
        croak 'reboot method requires a serviceId parameter';
    }

    $self->POST( "/services/$args{serviceId}/reboot?" . $self->{params},
        '', $self->{headers} );

    return _dumper($self);
}

sub flushFirewall {
    my ( $self, %args ) = @_;

    if ( !$args{serviceId} ) {
        croak 'flushFirewall method requires a serviceId parameter';
    }

    $self->POST( "/services/$args{serviceId}/firewall/flush?" . $self->{params},
        '', $self->{headers} );

    return _dumper($self);
}

sub addTempDisk {
    my ( $self, %args ) = @_;

    if ( !$args{serviceId} ) {
        croak 'addTempDisk method requires a serviceId parameter';
    }

    $self->POST( "/services/$args{serviceId}/disk/temp?" . $self->{params},
        '', $self->{headers} );

    return _dumper($self);
}

sub pleskPassword {
    my ( $self, %args ) = @_;

    if ( !$args{serviceId} ) {
        croak 'pleskPassword method requires a serviceId parameter';
    }

    my $b_content = undef;

    if ( $self->{format} eq 'xml' ) {
        $b_content = "$args{password}";
    }
    else {
        $b_content = "{\"password\": \"$args{password}\"}";
    }

    $self->PUT( "/services/$args{serviceId}/pleskPassword?" . $self->{params},
        $b_content, $self->{headers} );

    return _dumper($self);
}

sub setRootPass {
    my ( $self, %args ) = @_;

    if ( !$args{serviceId} ) {
        croak 'setRootPass method requires a serviceId parameter';
    }

    my $b_content = undef;

    if ( $self->{format} eq 'xml' ) {
        $b_content = "$args{password}";
    }
    else {
        $b_content = "{\"password\": \"$args{password}\"}";
    }

    $self->PUT( "/services/$args{serviceId}/rootPassword?" . $self->{params},
        $b_content, $self->{headers} );

    return _dumper($self);
}

sub stats {
    my ( $self, %args ) = @_;

    my ( $added_param, $extra_param ) = undef;

    if ( $args{precision} ) {
        $added_param .= "&precision=" . $arg{precision};
    }
    if ( $args{resolution} ) {
        $added_param .= "&resolution=" . $arg{resolution};
    }
    if ( $args{start} && $arg{end} ) {
        $added_param .= "&start=" . $arg{start} . "&end=" . $arg{end};
    }
    if ( $args{predefined} ) {
        $extra_param = "/$args{predefined}";
    }

    $self->GET(
        "/stats/$args{serviceId}"
          . $extra_param . "?"
          . $self->{params}
          . $added_param,
        $self->{headers}
    );

    return _dumper($self);
}

sub warnings {
    my ($self) = @_;

    $self->GET( "/stats/warnings?" . $self->{params}, $self->{headers} );

    return _dumper($self);
}

sub thresholds {
    my ($self) = @_;

    $self->GET( "/stats/warnings/thresholds?" . $self->{params},
        $self->{headers} );

    return _dumper($self);
}

sub _dumper {
    my ($self) = @_;

    my ( $init_mod, $ret_val );

    # return object based on constructor
    # ( xml object, json object, or raw xml/json text)
    if ( $self->{format} eq 'xml' and !$self->{raw_data} ) {
        $init_mod = XML::Simple->new();
        $ret_val  = $init_mod->XMLin(
            $self->responseContent,
            KeyAttr    => [],
            ForceArray => 0
        );
    }

    if ( $self->{format} eq 'json' and !$self->{raw_data} ) {
        $ret_val = JSON->new->utf8->decode( $self->responseContent );
    }

    if ( $self->{raw_data} ) {
        $ret_val = $self->responseContent;
    }

    return $ret_val;
}

1;

__END__