File Coverage

blib/lib/Mojolicious/Plugin/Google/Cloud/UserAgent.pm

Criterion	Covered	Total	%
statement	9	33	27.2
branch	0	4	0.0
condition	0	14	0.0
subroutine	3	5	60.0
pod	1	1	100.0
total	13	57	22.8

package Mojolicious::Plugin::Google::Cloud::UserAgent;
use Mojo::Base 'Mojolicious::Plugin';
use Mojo::JWT::Google;

our $VERSION = '0.03';

has oauth_url     => 'https://www.googleapis.com/oauth2/v4/token';
has grant_type    => 'urn:ietf:params:oauth:grant-type:jwt-bearer';
has web_token     => undef;
has web_token_enc => undef;

sub register {
    my ($self, $app, $config) = @_;

    $app->helper(
        jwt => sub {
            my $c = shift;
            my $scopes = shift // $config->{scopes};

            Mojo::JWT::Google->new(
                from_json  => $config->{gcp_auth_file},
                target     => $config->{oauth_url} // $self->oauth_url,
                scopes     => Mojo::Collection->new($scopes)->flatten,
                issue_at   => time,
                expires_in => $config->{duration} // 3600
            );
        }
    );

    $app->helper(
        gcp_ua => sub {
            my $c      = shift;
            my $err_cb = pop;
            my $cb     = pop;
            my @args   = @_;

            if (!$self->web_token
                or ($self->web_token->issue_at + $self->web_token->expires_in) < time)
            {
                $self->web_token($c->jwt);
                $self->web_token_enc($self->web_token->encode);
            }

            Mojo::IOLoop::Delay->new->steps(
                sub {
                    my $d = shift;
                    $c->ua->post(
                        $self->oauth_url,
                        form => {
                            grant_type => $config->{grant_type} // $self->grant_type,
                            assertion => $self->web_token_enc
                        },
                        $d->begin
                    );
                },

                sub {
                    my $d  = shift;
                    my $tx = pop;

                    if ($tx->res->json('/access_token')) {
                        $d->pass($tx->res->json);
                    }

                    else {
                        $d->remaining([$err_cb]);
                        $d->pass($tx);
                    }
                },

                sub {
                    my $d     = shift;
                    my $token = pop;

                    my $tx = $c->ua->transactor->tx(@args);
                    $tx->req->headers->header(Authorization => $token->{token_type} . ' ' . $token->{access_token});
                    $c->ua->start($tx, $d->begin);
                },

                $cb
            );
        }
    );
}

1;
__END__