File Coverage

blib/lib/AnyEvent/FTP/Client/Role/FetchTransfer.pm

Criterion	Covered	Total	%
statement	39	44	88.6
branch	12	16	75.0
condition			n/a
subroutine	13	14	92.8
pod	0	3	0.0
total	64	77	83.1

package AnyEvent::FTP::Client::Role::FetchTransfer;

use strict;
use warnings;
use 5.010;
use Moo::Role;

# ABSTRACT: Fetch transfer interface for AnyEvent::FTP objects
our $VERSION = '0.20'; # VERSION

sub xfer
{
  my($self, $fh, $local) = @_;

  my $handle = $self->handle($fh);

  return unless defined $local;

  $handle->on_read(sub {
    $handle->push_read(sub {
      $local->($_[0]{rbuf});
      $_[0]{rbuf} = '';
    });
  });
}

sub convert_local
{
  my($self, $local) = @_;

  return unless defined $local;
  return $local if ref($local) eq 'CODE';

  if(ref($local) eq 'SCALAR')
  {
    return sub {
      $$local .= shift;
    };
  }
  elsif(ref($local) eq 'GLOB')
  {
    return sub {
      print $local shift;
    };
  }
  elsif(ref($local) eq '')
  {
    open my $fh, '>', $local;
    $self->on_close(sub { close $fh });
    return sub {
      print $fh shift;
    };
  }
  else
  {
    die 'unimplemented: ' . ref $local;
  }
}

sub push_command
{
  my $self = shift;
  my $cv = $self->{client}->push_command(
    @_,
  );

  $cv->cb(sub {
    eval { $cv->recv };
    my $err = $@;
    $self->{cv}->croak($err) if $err;
  });

  $self->on_eof(sub {
    $cv->cb(sub {
      my $res = eval { $cv->recv };
      $self->{cv}->send($res) unless $@;
    });
  });
}

1;

__END__