File Coverage

blib/lib/WebService/Jina/UA.pm

Criterion	Covered	Total	%
statement	21	68	30.8
branch	0	14	0.0
condition			n/a
subroutine	7	14	50.0
pod	0	6	0.0
total	28	102	27.4

package WebService::Jina::UA;

use Moo;
use URI;
use LWP::UserAgent;
use JSON::Lines;
use MIME::Base64 qw/encode_base64/;

use WebService::Jina::Response;

has api_key => (
	is => 'ro',
	lazy => 1,
	required => 1,
);

has ua => (
	is => 'ro',
	default => sub {
		my $ua = LWP::UserAgent->new();
		use Data::Dumper;
		warn Dumper $_[0]->api_key;
		$ua->default_header('Authorization' => 'Bearer ' . $_[0]->api_key);
		$ua->default_header('User-Agent' => 'Mozilla/5.0');
		return $ua;
	}
);

has json => (
	is => 'ro',
	default => sub {
		JSON::Lines->new;
	}
);

sub get {
	shift->request(
		type => 'GET',
		@_
	);
}

sub post {
	shift->request(
		type => 'POST',
		@_
	);
}

sub delete {
	shift->request(
		type => 'DELETE',
		@_
	);
}

sub request {
	my ($self, %params) = @_;
	my $url = URI->new($params{url});
	my $res;
	my $stream_cb = delete $params{data}{stream_cb};
	$self->ua->remove_handler();
	if ($stream_cb) {
		$self->ua->add_handler(response_data => sub {
			my($response, $ua, $handler, $data) = @_;
			$data = $self->json->decode($data);
			for (@{$data}) {
				$stream_cb->(WebService::Jina::Response->new(%{$_}));
			}
			return 1;
		});
	}
	my %headers = %{ $params{headers} || {} };
	if ($params{type} eq 'GET') {
		$url->query_form($params{data});
		$res = $self->ua->get($url, %headers);
	} elsif ($params{type} eq 'DELETE') {
		$res = $self->ua->delete(
			$url,
			%headers,
			content => $self->json->encode([$params{data}]),
			'Content-Type' => 'application/json'
		);
	} else {
		$res = $self->ua->post(
			$url,
			%headers,
			content => $self->json->encode([$params{data}]),
			'Content-Type' => 'application/json'
		);
	}
	return $self->response($res);
}

sub response {
	my ($self, $res) = @_;
	if ($res->is_success) {
		if ($res->content_type eq 'text/plain') {
			return $res->decoded_content;
		}
		my $content = $self->json->decode($res->decoded_content);
		my @res = map {
			WebService::Jina::Response->new(%{$_})
		} @{$content};
		return scalar @res == 1 ? $res[0] : \@res;
	}
	die $res->decoded_content;
}

sub base64_images {
	my ($self, $images) = @_;

	my @out;
	for my $image (@{$images}) {
		open my $fh, '<', $image;
		my $content = do { local $/; <$fh> };
		close $fh;
		push @out, encode_base64($content);
	}
	return \@out;
}

1;