File Coverage

blib/lib/POE/Component/SmokeBox/Dists.pm

Criterion	Covered	Total	%
statement	127	197	64.4
branch	21	84	25.0
condition	6	17	35.2
subroutine	28	33	84.8
pod	4	4	100.0
total	186	335	55.5

package POE::Component::SmokeBox::Dists;
$POE::Component::SmokeBox::Dists::VERSION = '1.10';
#ABSTRACT: Search for CPAN distributions by cpanid or distribution name

use strict;
use warnings;
use Carp;
use Cwd;
use File::Spec ();
use File::Path (qw/mkpath/);
use URI;
use File::Fetch;
use CPAN::DistnameInfo;
use Sort::Versions;
use IO::Zlib;
use POE qw(Wheel::Run);

sub author {
  my $package = shift;
  return $package->_spawn( @_, command => 'author' );
}

sub distro {
  my $package = shift;
  return $package->_spawn( @_, command => 'distro' );
}

sub phalanx {
  my $package = shift;
  return $package->_spawn( @_, command => 'phalanx' );
}

sub random {
  my $package = shift;
  return $package->_spawn( @_, command => 'random' );
}

sub _spawn {
  my $package = shift;
  my %opts = @_;
  $opts{lc $_} = delete $opts{$_} for grep { !/^\_/ } keys %opts;

  $opts{pkg_time} = 21600 unless $opts{pkg_time};

  my @mandatory = qw(event);
  push @mandatory, 'search' unless $opts{command} eq 'phalanx' or $opts{command} eq 'random';
  foreach my $mandatory ( @mandatory ) {
     next if $opts{ $mandatory };
     carp "The '$mandatory' parameter is a mandatory requirement\n";
     return;
  }
  my $options = delete $opts{options};
  my $self = bless \%opts, $package;
  $self->{session_id} = POE::Session->create(
     package_states => [
	$self => [qw(
			_start
			_initialise
			_dispatch
			_spawn_fetch
			_fetch_err
			_fetch_close
			_fetch_sout
			_fetch_serr
			_spawn_process
			_proc_close
			_proc_sout
			_sig_child)],
     ],
     heap => $self,
     ( ref($options) eq 'HASH' ? ( options => $options ) : () ),
  )->ID();

  return $self;
}

sub _start {
  my ($kernel,$sender,$self) = @_[KERNEL,SENDER,OBJECT];
  $self->{session_id} = $_[SESSION]->ID();
  if ( $kernel == $sender and !$self->{session} ) {
	croak "Not called from another POE session and 'session' wasn't set\n";
  }
  my $sender_id;
  if ( $self->{session} ) {
    if ( my $ref = $kernel->alias_resolve( $self->{session} ) ) {
	$sender_id = $ref->ID();
    }
    else {
	croak "Could not resolve 'session' to a valid POE session\n";
    }
  }
  else {
    $sender_id = $sender->ID();
  }
  $kernel->refcount_increment( $sender_id, __PACKAGE__ );
  $self->{session} = $sender_id;
  $kernel->detach_myself() if $kernel != $sender;
  $kernel->yield( '_initialise' );
  return;
}

sub _initialise {
  my ($kernel,$self) = @_[KERNEL,OBJECT];
  my $return = { };

  my $smokebox_dir = File::Spec->catdir( _smokebox_dir(), '.smokebox' );

  mkpath $smokebox_dir if ! -d $smokebox_dir;
  if ( ! -d $smokebox_dir ) {
     $return->{error} = "Could not create smokebox directory '$smokebox_dir': $!";
     $kernel->yield( '_dispatch', $return );
     return;
  }

  $self->{return} = $return;
  $self->{sb_dir} = $smokebox_dir;

  my $packages_file = File::Spec->catfile( $smokebox_dir, '02packages.details.txt.gz' );

  $self->{pack_file} = $packages_file;

  if ( -e $packages_file ) {
     my $mtime = ( stat( $packages_file ) )[9];
     if ( $self->{force} or ( time() - $mtime > $self->{pkg_time} ) ) {
        $kernel->yield( '_spawn_fetch', $smokebox_dir, $self->{url} );
	return;
     }
  }
  else {
     $kernel->yield( '_spawn_fetch', $smokebox_dir, $self->{url} );
     return;
  }

  # if packages file exists but is older than $self->{pkg_time}, fetch.
  # if packages file does not exist, fetch.
  # otherwise it exists so spawn packages processing.

  $kernel->yield( '_spawn_process' );
  return;
}

sub _dispatch {
  my ($kernel,$self,$return) = @_[KERNEL,OBJECT,ARG0];
  $return->{$_} = $self->{$_} for grep { /^\_/ } keys %{ $self };
  $kernel->post( $self->{session}, $self->{event}, $return );
  $kernel->refcount_decrement( $self->{session}, __PACKAGE__ );
  return;
}

sub _sig_child {
  $poe_kernel->sig_handled();
}

sub _spawn_fetch {
  my ($kernel,$self) = @_[KERNEL,OBJECT];
  $self->{FETCH} = POE::Wheel::Run->new(
	Program     => \&_fetch,
	ProgramArgs => [ $self->{sb_dir}, $self->{url} ],
	StdoutEvent => '_fetch_sout',
	StderrEvent => '_fetch_serr',
	ErrorEvent  => '_fetch_err',             # Event to emit on errors.
	CloseEvent  => '_fetch_close',     # Child closed all output.
  );
  $kernel->sig_child( $self->{FETCH}->PID(), '_sig_chld' ) if $self->{FETCH};
  return;
}

sub _fetch_sout {
  return;
}

sub _fetch_serr {
  return;
}

sub _fetch_err {
  return;
}

sub _fetch_close {
  my ($kernel,$self) = @_[KERNEL,OBJECT];
  delete $self->{FETCH};
  if ( -e $self->{pack_file} ) {
     $kernel->yield( '_spawn_process' );
  }
  else {
     $self->{return}->{error} = 'Could not retrieve packages file';
     $kernel->yield( '_dispatch', $self->{return} );
  }
  return;
}

sub _spawn_process {
  my ($kernel,$self) = @_[KERNEL,OBJECT];
  $self->{dists} = [ ];
  $self->{PROCESS} = POE::Wheel::Run->new(
	Program     => \&_read_packages,
	ProgramArgs => [ $self->{pack_file}, $self->{command}, $self->{search} ],
	StdoutEvent => '_proc_sout',
	StderrEvent => '_fetch_serr',
	ErrorEvent  => '_fetch_err',             # Event to emit on errors.
	CloseEvent  => '_proc_close',     # Child closed all output.
  );
  $kernel->sig_child( $self->{PROCESS}->PID(), '_sig_chld' ) if $self->{PROCESS};
  return;
}

sub _proc_sout {
  my ($self,$line) = @_[OBJECT,ARG0];
  push @{ $self->{dists} }, $line;
  return;
}

sub _proc_close {
  my ($kernel,$self) = @_[KERNEL,OBJECT];
  delete $self->{PROCESS};
  $self->{return}->{dists} = delete $self->{dists};
  $kernel->yield( '_dispatch', $self->{return} );
  return;
}

sub _read_packages {
  my ($packages_file,$command,$search) = @_;
  my %phalanx;
  if ( $command eq 'phalanx' ) {
    $phalanx{ $_ } = undef for _phalanx();
  }
  my $fh = IO::Zlib->new( $packages_file, "rb" ) or die "$!\n";
  my %dists;
  while (<$fh>) {
    last if /^\s*$/;
  }
  while (<$fh>) {
    chomp;
    my $path = ( split ' ', $_ )[2];
    next unless $path;
    next if exists $dists{ $path };
    my $distinfo = CPAN::DistnameInfo->new( $path );
    next unless $distinfo->filename() =~ m!(\.tar\.gz|\.tgz|\.zip)$!i;
    if ( $command eq 'author' ) {
       next unless eval { $distinfo->cpanid() =~ /$search/ };
       print $path, "\n";
    }
    elsif ( $command eq 'phalanx' ) {
       next unless exists $phalanx{ $distinfo->dist };
       if ( defined $phalanx{ $distinfo->dist } ) {
	       my $exists = CPAN::DistnameInfo->new( $phalanx{ $distinfo->dist } );
	       if ( versioncmp( $distinfo->version, $exists->version ) == 1 ) {
		        $phalanx{ $distinfo->dist } = $path;
	       }
       }
       else {
	       $phalanx{ $distinfo->dist } = $path;
       }
    }
    elsif ( $command eq 'random' ) {
       $dists{ $path } = 1;
       next;
    }
    else {
       next unless eval { $distinfo->distvname() =~ /$search/ };
       print $path, "\n";
    }
    $dists{ $path } = 1;
  }
  if ( $command eq 'phalanx' ) {
    print $_, "\n" for grep { defined $_ } values %phalanx;
  }
  if ( $command eq 'random' ) {
    my @dists = keys %dists;
    my %picked;
    while ( scalar keys %picked < 100 ) {
      my $random = $dists[ rand( $#dists ) ];
      next if $picked{ $random };
      $picked{ $random } = $random;
      print $random, "\n";
    }
  }
  return;
}

sub _fetch {
  my $location = shift || return;
  my $url = shift;
  my @urls = qw(
    http://www.cpan.org/
    ftp://ftp.cpan.org/pub/CPAN/
    http://cpan.cpantesters.org/
    ftp://cpan.cpantesters.org/CPAN/
    ftp://ftp.funet.fi/pub/CPAN/
  );
  @urls = ( $url ) if $url;
  my $file;
  foreach my $url ( @urls ) {
    my $uri = URI->new( $url ) or next;
    my @segs = $uri->path_segments();
    pop @segs unless $segs[$#segs];
    $uri->path_segments( @segs, 'modules', '02packages.details.txt.gz' );
    local $File::Fetch::TIMEOUT = 30;
    my $ff = File::Fetch->new( uri => $uri->as_string() ) or next;
    $file = $ff->fetch( to => $location ) or next;
    last if $file;
  }
  return $file;
}

sub _smokebox_dir {
  return $ENV{PERL5_SMOKEBOX_DIR}
     if  exists $ENV{PERL5_SMOKEBOX_DIR}
     && defined $ENV{PERL5_SMOKEBOX_DIR};

  my @os_home_envs = qw( APPDATA HOME USERPROFILE WINDIR SYS$LOGIN );

  for my $env ( @os_home_envs ) {
      next unless exists $ENV{ $env };
      next unless defined $ENV{ $env } && length $ENV{ $env };
      return $ENV{ $env } if -d $ENV{ $env };
  }

  return cwd();
}

# List taken from Bundle::Phalanx100 v0.07
sub _phalanx {
  return qw(
	Test-Harness
	Test-Reporter
	Test-Simple
	Test-Builder-Tester
	Sub-Uplevel
	Test-Exception
	Test-Tester
	Test-NoWarnings
	Test-Tester
	Pod-Escapes
	Pod-Simple
	Test-Pod
	YAML
	PathTools
	Archive-Tar
	Module-Build
	Devel-Symdump
	Pod-Coverage
	Test-Pod-Coverage
	Compress-Zlib
	IO-Zlib
	Archive-Zip
	Archive-Tar
	Storable
	Digest-MD5
	URI
	HTML-Tagset
	HTML-Parser
	libwww-perl
	IPC-Run
	CPANPLUS
	DBI
	DBD-mysql
	GD
	MIME-Base64
	Net-SSLeay
	perl-ldap
	XML-Parser
	Apache-ASP
	CGI.pm
	Date-Manip
	DBD-Oracle
	DBD-Pg
	Digest-SHA1
	Digest-HMAC
	HTML-Tagset
	HTML-Template
	libnet
	MailTools
	MIME-tools
	Net-DNS
	Time-HiRes
	Apache-DBI
	Apache-Session
	Apache-Test
	AppConfig
	App-Info
	Authen-PAM
	Authen-SASL
	BerkeleyDB
	Bit-Vector
	Carp-Clan
	Chart
	Class-DBI
	Compress-Zlib-Perl
	Config-IniFiles
	Convert-ASN1
	Convert-TNEF
	Convert-UUlib
	CPAN
	Crypt-CBC
	Crypt-DES
	Crypt-SSLeay
	Data-Dumper
	Date-Calc
	DateTime
	DBD-DB2
	DBD-ODBC
	DBD-SQLite
	DBD-Sybase
	Device-SerialPort
	Digest-SHA
	Encode
	Event
	Excel-Template
	Expect
	ExtUtils-MakeMaker
	File-Scan
	File-Spec
	File-Tail
	File-Temp
	GDGraph
	GDTextUtil
	Getopt-Long
	HTML-Mason
	Image-Size
	IMAP-Admin
	Parse-RecDescent
	Inline
	IO
	Spiffy
	IO-All
	IO-Socket-SSL
	IO-String
	IO-stringy
	libxml-perl
	Mail-Audit
	Mail-ClamAV
	Mail-Sendmail
	Math-Pari
	MD5
	MIME-Lite
	MP3-Info
	Net-Daemon
	Net-FTP-Common
	Net-Ping
	Net-Server
	Net-SNMP
	Net-SSH-Perl
	Net-Telnet
	OLE-Storage_Lite
	Params-Validate
	PerlMagick
	PlRPC
	Pod-Parser
	POE
	SNMP
	SOAP-Lite
	Spreadsheet-ParseExcel
	Spreadsheet-WriteExcel
	Spreadsheet-WriteExcelXML
	Storable
	Template-Toolkit
	TermReadKey
	Term-ReadLine-Perl
	Text-Iconv
  TimeDate
  Time-modules
	Unicode-String
	Unix-Syslog
	Verilog-Perl
	WWW-Mechanize
	XML-DOM
	XML-Generator
	XML-LibXML
	XML-NamespaceSupport
	XML-SAX
	XML-Simple
	XML-Writer
  );
}

1;

__END__