1 files changed, 59 insertions, 4 deletions

diff --git a/lib/Net/Riak/Role/REST.pm b/lib/Net/Riak/Role/REST.pm
index 136ea88..dfab5a0 100644
--- a/lib/Net/Riak/Role/REST.pm
+++ b/lib/Net/Riak/Role/REST.pm
@@ -3,12 +3,36 @@ package Net::Riak::Role::REST;
 # ABSTRACT: role for REST operations
 
 use URI;
-use HTTP::Request;
+
 use Moose::Role;
+use MooseX::Types::Moose 'Bool';
+use Net::Riak::Types qw/HTTPResponse HTTPRequest/;
+use Data::Dump 'pp';
+with qw/Net::Riak::Role::REST::Bucket 
+    Net::Riak::Role::REST::Object 
+    Net::Riak::Role::REST::Link
+    Net::Riak::Role::REST::MapReduce
+    /;
+
+has http_request => (
+    is => 'rw',
+    isa => HTTPRequest,
+);
+
+has http_response => (
+    is => 'rw',
+    isa => HTTPResponse,
+    handles => {
+        is_success => 'is_success',
+        status => 'code',
+    }
+);
 
-requires 'http_request';
-requires 'http_response';
-requires 'useragent';
+has disable_return_body => (
+    is => 'rw',
+    isa => Bool,
+    default => 0
+);
 
 sub _build_path {
     my ($self, $path) = @_;
@@ -37,9 +61,40 @@ sub send_request {
 
     $self->http_request($req);
     my $r = $self->useragent->request($req);
+
     $self->http_response($r);
 
+    if ($ENV{RIAK_VERBOSE}) {
+        print STDERR pp($r);
+    }
+
     return $r;
 }
 
+sub is_alive {
+    my $self     = shift;
+    my $request  = $self->new_request('HEAD', ['ping']);
+    my $response = $self->send_request($request);
+    $self->is_success ? return 1 : return 0;
+}
+
+sub all_buckets {
+    my $self = shift;
+    my $request = $self->new_request('GET', [$self->prefix], {buckets => 'true'});
+    my $response = $self->send_request($request);
+    die "Failed to fetch buckets.. are you running riak 0.14+?" 
+        unless $response->is_success;
+    my $resp = JSON::decode_json($response->content);
+    return ref ($resp->{buckets}) eq 'ARRAY' ? @{$resp->{buckets}} : ();
+}
+
+sub server_info { die "->server_info not supported by the REST interface" }
+
+sub stats {
+    my $self = shift;
+    my $request = $self->new_request('GET', ["stats"]);
+    my $response = $self->send_request($request);
+    return JSON::decode_json($response->content);
+}
+
 1;