Grep/Action/Fetch.pm

use strict;
use warnings;

=head1 NAME

Grep::Action::Fetch

=cut

package Grep::Action::Fetch;
use base qw/Grep::Action Jifty::Action/;

use XML::Feed;
use LWP::UserAgent;

use Data::Dump qw/dump/;

use Jifty::Param::Schema;
use Jifty::Action schema {

        param q =>
                type is 'text',
                label is 'Search for',
                hint is 'enter few words to search for';

        param feed =>
                label is 'From feed',
                render as 'select',
                available are defer {
                        my $feeds = Grep::Model::FeedCollection->new;
                        $feeds->order_by({ column => 'title', order => 'ASC' });
                        $feeds->unlimit;
                        warn "feeds ", $feeds->build_select_query;
                        [{
                                display_from => 'title',
                                value_from => 'id',
                                collection => $feeds,
                        }];
                };

        param item_fragment =>
                label is 'Show',
                render as 'select',
                valid are qw/result result_short/,
                available are qw/result result_short/;

};

=head2 take_action

=cut

sub take_action {
    my $self = shift;
    
    # Custom action code

        my $feed = Grep::Model::Feed->new();
        my $feed_id = $self->argument_value('feed');

        if (! $feed_id) {
                $self->result->message("Need feed ID");
                return 0;
        }

        $feed->load_by_cols( id => $feed_id );

        if (! $feed->id) {
        $self->result->message("Can't fetch feed $feed_id");
                return 0;
        }

        my $uri = $feed->uri;
        if ($uri =~ m/%s/) {
                $uri = $feed->search_uri( $self->argument_value('q') );
                Jifty->log->info("Searching ", $feed->title, " at $uri");
        } else {
                Jifty->log->info("Fetching ", $feed->title, " at $uri");
        }

        my $ua = LWP::UserAgent->new;
        $ua->default_header( 'Cookie' => $feed->cookie );
        my $r = $ua->get( $uri );
        die $feed->title . " returned " . $r->status_line . " for $uri\n" unless ( $r->is_success );

        my $content = $r->content;

        die "No content returned from $uri" unless length( $content ) > 1;

        my $xml_feed = XML::Feed->parse( \$content )
                or die $feed->title, " returned ", XML::Feed->errstr, " for $uri\n";

        warn "getting entries from ", $xml_feed->title, "\n";

        my @items;

        for my $entry ($xml_feed->entries) {
                my $i = Grep::Model::Item->new();

                $i->load_or_create(
                        in_feed => $feed,
                        title => $entry->title,
                        link => $entry->link,
                        content => $entry->content->body,
                        summary => $entry->summary->body,
                        category => $entry->category,
                        author => $entry->author,
                        issued => $entry->issued ? $entry->issued->strftime("%Y-%m-%d %H:%M:%S %z") : undef,
                        modified => $entry->modified ? $entry->modified->strftime("%Y-%m-%d %H:%M:%S %z") : undef,
                );

                if ( $i->id ) {
                        push @items, $i;

        #               Grep::Event::Result->new( $i )->publish;

                        Jifty->log->debug("published ", $i->id ) ; # dump( $entry, $i ) );
                } else {
                        warn "can't create item from entry ", dump( $entry ) unless ( $i->id );
                }

        }

        if ( @items ) {

                $self->result->message( $self->argument_value('q') . ' => ' .
                        $xml_feed->entries . ' items: ' . join(",", map { $_->id } @items)
                );

                $self->result->content( items => \@items );
                return 1;

        } else {

                $self->result->error( 'No results found' );
                return 0;

        }
}

1;

1	dpavlin	3	use strict;
2			use warnings;
3
4			=head1 NAME
5
6			Grep::Action::Fetch
7
8			=cut
9
10			package Grep::Action::Fetch;
11			use base qw/Grep::Action Jifty::Action/;
12
13			use XML::Feed;
14	dpavlin	21	use LWP::UserAgent;
15	dpavlin	3
16			use Data::Dump qw/dump/;
17
18			use Jifty::Param::Schema;
19			use Jifty::Action schema {
20
21			param q =>
22			type is 'text',
23			label is 'Search for',
24			hint is 'enter few words to search for';
25
26			param feed =>
27			label is 'From feed',
28	dpavlin	7	render as 'select',
29	dpavlin	3	available are defer {
30			my $feeds = Grep::Model::FeedCollection->new;
31			$feeds->order_by({ column => 'title', order => 'ASC' });
32			$feeds->unlimit;
33			warn "feeds ", $feeds->build_select_query;
34			[{
35			display_from => 'title',
36			value_from => 'id',
37			collection => $feeds,
38			}];
39			};
40
41	dpavlin	25	param item_fragment =>
42			label is 'Show',
43			render as 'select',
44			valid are qw/result result_short/,
45			available are qw/result result_short/;
46
47	dpavlin	3	};
48
49			=head2 take_action
50
51			=cut
52
53			sub take_action {
54			my $self = shift;
55
56			# Custom action code
57
58			my $feed = Grep::Model::Feed->new();
59			my $feed_id = $self->argument_value('feed');
60
61			if (! $feed_id) {
62			$self->result->message("Need feed ID");
63			return 0;
64			}
65
66			$feed->load_by_cols( id => $feed_id );
67
68			if (! $feed->id) {
69			$self->result->message("Can't fetch feed $feed_id");
70			return 0;
71			}
72
73	dpavlin	7	my $uri = $feed->uri;
74			if ($uri =~ m/%s/) {
75	dpavlin	19	$uri = $feed->search_uri( $self->argument_value('q') );
76	dpavlin	7	Jifty->log->info("Searching ", $feed->title, " at $uri");
77			} else {
78			Jifty->log->info("Fetching ", $feed->title, " at $uri");
79			}
80	dpavlin	3
81	dpavlin	21	my $ua = LWP::UserAgent->new;
82			$ua->default_header( 'Cookie' => $feed->cookie );
83			my $r = $ua->get( $uri );
84	dpavlin	25	die $feed->title . " returned " . $r->status_line . " for $uri\n" unless ( $r->is_success );
85	dpavlin	21
86			my $content = $r->content;
87
88			die "No content returned from $uri" unless length( $content ) > 1;
89
90			my $xml_feed = XML::Feed->parse( \$content )
91	dpavlin	19	or die $feed->title, " returned ", XML::Feed->errstr, " for $uri\n";
92	dpavlin	3
93	dpavlin	21	warn "getting entries from ", $xml_feed->title, "\n";
94	dpavlin	7
95	dpavlin	9	my @items;
96	dpavlin	7
97	dpavlin	3	for my $entry ($xml_feed->entries) {
98			my $i = Grep::Model::Item->new();
99
100			$i->load_or_create(
101			in_feed => $feed,
102			title => $entry->title,
103			link => $entry->link,
104	dpavlin	5	content => $entry->content->body,
105			summary => $entry->summary->body,
106	dpavlin	3	category => $entry->category,
107			author => $entry->author,
108			issued => $entry->issued ? $entry->issued->strftime("%Y-%m-%d %H:%M:%S %z") : undef,
109			modified => $entry->modified ? $entry->modified->strftime("%Y-%m-%d %H:%M:%S %z") : undef,
110			);
111	dpavlin	5
112	dpavlin	25	if ( $i->id ) {
113			push @items, $i;
114	dpavlin	7
115	dpavlin	25	# Grep::Event::Result->new( $i )->publish;
116	dpavlin	7
117	dpavlin	25	Jifty->log->debug("published ", $i->id ) ; # dump( $entry, $i ) );
118			} else {
119			warn "can't create item from entry ", dump( $entry ) unless ( $i->id );
120			}
121	dpavlin	9
122	dpavlin	3	}
123
124	dpavlin	9	if ( @items ) {
125	dpavlin	7
126			$self->result->message( $self->argument_value('q') . ' => ' .
127	dpavlin	9	$xml_feed->entries . ' items: ' . join(",", map { $_->id } @items)
128	dpavlin	7	);
129
130	dpavlin	9	$self->result->content( items => \@items );
131	dpavlin	7	return 1;
132
133			} else {
134
135	dpavlin	21	$self->result->error( 'No results found' );
136			return 0;
137	dpavlin	8
138	dpavlin	7	}
139	dpavlin	3	}
140
141			1;
142