File Coverage

blib/lib/StreamRoot/Manifest.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 StreamRoot::Manifest;

use strict;
use warnings;
use WWW::Curl::Easy;
use JSON;

our $VERSION = '0.03';

sub new {
    my $class = shift;
    my $self = {};
    $self->{token} = shift||undef;
    $self->{curl} = WWW::Curl::Easy->new;
    $self->{curl}->setopt(CURLOPT_TIMEOUT, 180);
    $self->{curl}->setopt(CURLOPT_HEADER, 0);
    $self->{curl}->setopt(CURLOPT_USERAGENT, 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)');
    $self->{curl}->setopt(CURLOPT_AUTOREFERER, 1);
    $self->{curl}->setopt(CURLOPT_FOLLOWLOCATION, 1);
    $self->{curl}->setopt(CURLOPT_SSL_VERIFYPEER, 0);
    $self->{curl}->setopt(CURLOPT_VERBOSE, 0);
    return bless $self, $class;
}

sub curl {
    my ( $self, $url, $data, $referer ) = @_;
    my $params = "";
    if ($data) {
        if (ref($data) eq 'HASH') {
            for(keys %$data){
                $params .= "&" if $params;
                $params .= "$_=$data->{$_}";
            }
        }else{
            $params = "$data";
        }
    }
    $self->{curl}->setopt(CURLOPT_POSTFIELDS, $params);
    $self->{curl}->setopt(CURLOPT_URL, $url);
    my $content = "";
    open(my $writedata, ">", \$content);
    $self->{curl}->setopt(CURLOPT_WRITEDATA, $writedata);
    if ($self->{curl}->perform == 0) {
        if ($self->{curl}->getinfo(CURLINFO_HTTP_CODE) =~ /^2/) {
            decode_json($content) if $content !~ /^$/;
        }else{
            if ($self->{curl}->getinfo(CURLINFO_HTTP_CODE) =~ /^4/) {
                decode_json($content) if $content !~ /^$/;
            }else{
                die($content);
            }
        }
    }
}

sub setToken {
    my $self = shift;
    $self->{token} = shift||undef;
}

sub getToken {
    my $self = shift;
    return $self->{token};
}

sub authenticate {
    my ( $self, $username, $password ) = @_;
    my $data = {
        'username' => $username,
        'password' => $password
    };
    $self->{curl}->setopt(CURLOPT_CUSTOMREQUEST, 'POST');
    $self->{curl}->setopt(CURLOPT_HTTPHEADER(), ['Content-type: application/x-www-form-urlencoded']);
    my $content = $self->curl('https://manifests.streamroot.io/auth', $data);
    $self->setToken($content->{token}) if $content->{token};
    return $content;
}

sub addManifest {
    my $self = shift;
    my @values = @_;
    if (ref($values[0]) eq 'HASH') {
        return $self->loop('add', @values);
    }else{
        my %params = @values;
        my $data = {};
        $data->{url} = $params{url};
        $data->{status} = $params{status} if $params{status};
        $data->{ttl} = $params{ttl} if $params{ttl};
        $self->{curl}->setopt(CURLOPT_CUSTOMREQUEST, 'POST');
        $self->{curl}->setopt(CURLOPT_HTTPHEADER(), ['Authorization:Bearer '.$self->getToken]);
        my $content = $self->curl('https://manifests.streamroot.io', $data);
        return $content;
    }
}

sub updateManifest {
    my $self = shift;
    my @values = @_;
    if (ref($values[0]) eq 'HASH') {
        return $self->loop('update', @values);
    }else{
        my %params = @values;
        my $data = {};
        $data->{status} = $params{status} if $params{status};
        $data->{live} = $params{live} if $params{live};
        $self->{curl}->setopt(CURLOPT_CUSTOMREQUEST, 'PUT');
        $self->{curl}->setopt(CURLOPT_HTTPHEADER(), ['Authorization:Bearer '.$self->getToken]);
        my $content = $self->curl('https://manifests.streamroot.io/' . $params{id}, $data);
        return $content;
    }
}

sub removeManifest {
    my $self = shift;
    my @values = @_;
    if (scalar(@values) > 1) {
        return $self->loop('remove', @values);
    }else{
        my $id = $values[0];
        $self->{curl}->setopt(CURLOPT_CUSTOMREQUEST, 'DELETE');
        $self->{curl}->setopt(CURLOPT_HTTPHEADER(), ['Authorization:Bearer '.$self->getToken]);
        my $content = $self->curl('https://manifests.streamroot.io/' . $id);
        return $content;
    }
}

sub showManifest {
    my $self = shift;
    my @values = @_;
    if (scalar(@values) > 1) {
        return $self->loop('show', @values);
    }else{
        my $id = $values[0];
        $self->{curl}->setopt(CURLOPT_CUSTOMREQUEST, 'GET');
        $self->{curl}->setopt(CURLOPT_HTTPHEADER(), ['Authorization:Bearer '.$self->getToken]);
        my $content = $self->curl('https://manifests.streamroot.io/' . $id);
        return $content;
    }
}

sub showAllManifests {
    my ( $self, $pattern ) = @_;
    my $data = {};
    $data->{pattern} = $pattern if $pattern;
    $self->{curl}->setopt(CURLOPT_CUSTOMREQUEST, 'GET');
    $self->{curl}->setopt(CURLOPT_HTTPHEADER(), ['Authorization:Bearer '.$self->getToken]);
    my $content = $self->curl('https://manifests.streamroot.io', $data);
    return $content;
}

sub loop {
    my ($self, $type, @values) = @_;
    my @results;
    for(@values){
        my $content;
        if ($type eq 'add') {
            $content = $self->addManifest(%{$_});
        }
        if ($type eq 'update') {
            $content = $self->updateManifest(%{$_});
        }
        if ($type eq 'remove') {
            $content = $self->removeManifest($_);
        }
        if ($type eq 'show') {
            $content = $self->showManifest($_);
        }
        push(@results, $content);
    }
    return $results[0] if scalar(@results) == 1;
    return \@results if scalar(@results) > 1;
}

1;

__END__