File Coverage

blib/lib/WWW/VastAI/API/Endpoints.pm

Criterion	Covered	Total	%
statement	37	37	100.0
branch	8	16	50.0
condition	1	7	14.2
subroutine	10	10	100.0
pod	5	5	100.0
total	61	75	81.3

package WWW::VastAI::API::Endpoints;
our $VERSION = '0.001';
# ABSTRACT: Serverless endpoint management for Vast.ai

use Moo;
use Carp qw(croak);
use WWW::VastAI::Endpoint;
use namespace::clean;

has client => (
    is       => 'ro',
    required => 1,
    weak_ref => 1,
);

sub _wrap {
    my ($self, $data) = @_;
    return WWW::VastAI::Endpoint->new(
        client => $self->client,
        data   => $data,
    );
}

sub list {
    my ($self) = @_;
    my $result = $self->client->request_op('listEndpoints');
    my $endpoints = ref $result eq 'HASH' ? ($result->{endpoints} || $result->{results} || []) : ($result || []);
    return [ map { $self->_wrap($_) } @{$endpoints} ];
}

sub create {
    my ($self, %params) = @_;
    croak "endpoint_name required" unless $params{endpoint_name};
    croak "min_load required"      unless exists $params{min_load};
    croak "target_util required"   unless exists $params{target_util};

    my $result = $self->client->request_op('createEndpoint', body => \%params);
    my $endpoint = ref $result eq 'HASH' ? ($result->{endpoint} || $result) : $result;
    return $self->_wrap($endpoint);
}

sub delete {
    my ($self, $id) = @_;
    croak "endpoint id required" unless defined $id;
    return $self->client->request_op('deleteEndpoint', path => { id => $id });
}

sub logs {
    my ($self, $endpoint_name, %params) = @_;
    croak "endpoint_name required" unless $endpoint_name;
    my %body = ( endpoint_name => $endpoint_name, %params );
    return $self->client->request_op('getEndpointLogs', body => \%body);
}

sub workers {
    my ($self, $endpoint_id) = @_;
    croak "endpoint id required" unless defined $endpoint_id;
    return $self->client->request_op('getEndpointWorkers', body => { endpoint_id => $endpoint_id });
}

1;

__END__