File Coverage

blib/lib/Mail/Karmasphere/Publisher.pm

Criterion	Covered	Total	%
statement	36	77	46.7
branch	0	14	0.0
condition	0	2	0.0
subroutine	12	16	75.0
pod	0	3	0.0
total	48	112	42.8

package Mail::Karmasphere::Publisher;

use strict;
use Carp qw(cluck);
use warnings;
use vars qw(@ISA @EXPORT_OK %EXPORT_TAGS);

use Exporter;
use Data::Dumper;
use Time::HiRes;
use File::Temp;
use IO::File;
use LWP::UserAgent;
use HTTP::Request::Common;

BEGIN {
	@ISA = qw(Exporter);
	@EXPORT_OK = qw();
	%EXPORT_TAGS = (
		'all' => \@EXPORT_OK,
		'ALL' => \@EXPORT_OK,
	);
}

sub new {
	my $class = shift;
	my $self = ($#_ == 0) ? { %{ (shift) } } : { @_ };
	# Check for Principal, Credentials
	return bless $self, $class;
}

sub _output_file {
	my ($input, $output, $index) = @_;
	return $output if ref $output;	# An IO::File
	return new IO::File("> $output") if defined $output;
	my $temp = $input;
	return new File::Temp(
				TEMPLATE	=> "$input.$$.$index.XXXXXX",
				SUFFIX		=> ".ktmp",
					);
}

# Do not fuck with this method, Karma-Syndicator calls it.
# we output to a temporary file and write to it.
# once we're done writing to it, we rename the temp file to the real filename.
sub parse {
	my ($self, $input, $class, $outputs, %args) = @_;

	eval qq{ require $class; };
	die $@ if $@;

	my $fh = new IO::File("< $input");
	my $parser = $class->new(fh => $fh, %args);
	my $streams = $parser->streams;

	$outputs ||= [];	# An array of filenames.

	# print STDERR "outputs are " . Dumper($outputs);
	my @files = map { _output_file($outputs->[$_], undef, $_) } (0..$#$streams);
	# print STDERR ">>> parse temp files are " . Dumper(\@files);

	for my $i (0..$#$streams) {
	    if (not $outputs->[$i]) {
		cluck ("parser $parser for input $input defines stream $i, but there is no corresponding output filename!");
	    }
	}

	# this is a ridiculously slow inner loop.  can we optimize?
	while (my @records = $parser->parse) {
		for my $record (@records) {
			next if not defined $record;
			my $file = $files[$record->stream];
			print $file $record->as_string, "\n";
		}
	}

	foreach my $i (0 .. $#$streams) {
	    my $fh = $files[$i];
	    print STDERR " >> stream $i: renaming " . $fh->filename() . " to " . $outputs->[$i] . "\n";
	    rename($fh->filename, $outputs->[$i]);
	    chmod 0644, $outputs->[$i];
	}

	return 1;
}

sub publish {
    my ($self, $file, $class, $params) = @_;

    my $ua = LWP::UserAgent->new;

    my $url  = $params->{url};
    my $feed = $params->{feed};

    my $req = POST ($url,
		    Content_Type => "form-data",
		    Content =>
		    [ feed_id  => $feed,
		      login    => $params->{user},
		      password => $params->{pass},
		      data_source => "upload",
		      data_file => [ $file ],
		      ]);

    if (defined $params->{htuser}) {
	$req->headers->authorization_basic($params->{htuser},
					   $params->{htpass});
    }

    my $res = $ua->request($req);
	return $res;
}



1;

__END__