File Coverage

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

Criterion	Covered	Total	%
statement	40	41	97.5
branch	13	14	92.8
condition			n/a
subroutine	12	12	100.0
pod	0	3	0.0
total	65	70	92.8

package AnyEvent::FTP::Client::Role::StoreTransfer;

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

# ABSTRACT: Store 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_drain(sub {
    my $data = $local->();
    if(defined $data)
    {
      $handle->push_write($data);
    }
    else
    {
      $handle->push_shutdown;
      $handle->destroy;
    }
  });
}

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

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

  if(ref($local) eq '')
  {
    open my $fh, '<', $local;
    $self->on_close(sub { close $fh });
    return sub {
      local $/;
      <$fh>;
    };
  }
  elsif(ref($local) eq 'SCALAR')
  {
    my $buffer = $$local;
    return sub {
      my $tmp = $buffer;
      undef $buffer;
      $tmp;
    };
  }
  elsif(ref($local) eq 'GLOB')
  {
    sub {
      # TODO: for big files, maybe
      # break this up into batches
      local $/;
      <$local>;
    };
  }
  else
  {
    die 'bad local type';
  }
}

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

1;

__END__