lib/Grep/Source.pm

# Dobrica Pavlinusic, <dpavlin@rot13.org> 02/22/07 20:30:00 CET

use strict;
use warnings;

package Grep::Source;

use Carp qw/verbose/;
use Module::Pluggable search_path => 'Grep::Source', sub_name => 'sources', require => 1;
use base qw(Class::Accessor Jifty::Object);
Grep::Source->mk_accessors( qw(feed uri q new_items collection search_obj) );

use HTML::TreeBuilder;
use WWW::Mechanize;
use XML::Feed;
use URI;
use HTML::ResolveLink;

use Data::Dump qw/dump/;

=head1 NAME

Grep::Source - base class for implementation of different sources for Grep

=head1 METHODS

This is mostly documentation because most of methods are implemented by plugins.

=head2 sources

  my @sources = Grep::Source->sources();

Returns all available sources.

=cut

Jifty->log->debug("Found source plugins: ", join(", ", __PACKAGE__->sources() ) );

=head2 new

  my $source = Grep::Source->new({ feed => $feed_record });

This will also setup:

=head2 feed

isa L<Grep::Model::Feed>

=head2 search

  my $collection = $source->search( 'query string' );

It will also setup following accessors:

=head2 q

Search query 

=head2 uri

URI of feed with embedded search query

=head2 new_items

Number of new items in result collection

=head2 collection

Actuall results which is L<Grep::Model::ItemCollection>, so following will
work:

  print "and ", $self->collection->count, " total items";


Also setups number of new items

  print $source->new_items, " items new";

=cut

sub search {
        my $self = shift;

        my $q = shift;

        $q ? $self->q( $q ) : $q = $self->q;

        die "no q?" unless ( $self->q );
        die "no feed?" unless ( $self->feed );
        die "feed not Grep::Model::Feed" unless ( $self->feed->isa('Grep::Model::Feed') );

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

        $self->uri( $uri );

        $self->log->info( $message );

        $self->collection( Grep::Model::ItemCollection->new() );

        my $class = $self->feed->source || 'Grep::Source::Feed';
        $self->log->debug("using $class");

        $self->search_obj( Grep::Search->new() );
        $self->log->debug("created " . $self->search_obj);

        $class->fetch( $self );

        $self->search_obj->finish;

        return $self->collection;
}

=head2 add_record

Plugins will be called with parametar C<$parent> so they can call this method to add
record into result collection (and store in cache and index).

  $parent->add_record( id => 42, foo => 'bar', ... );

This will also update L</new_items>

=cut

sub add_record {
        my $self = shift;

        $self->log->confess("no search_obj") unless ($self->search_obj);

        my $i = Grep::Model::Item->new();

        my $rec = {@_};

        $self->log->debug("resolving links using base ", $rec->{link});
        my $resolver = HTML::ResolveLink->new( base => $rec->{link} );
        $rec->{content} = $resolver->resolve( $rec->{content} );

        my ($ok,$msg) = $i->load_or_create( %$rec );

        $msg ||= '';

        if ( $ok ) {
                $self->log->debug("item ", $i->id, ": $msg");
                $self->collection->add_record( $i );

                # is new record?
                if ( $msg !~ m/^Found/ ) {
                        $self->search_obj->add( $i );
                        $self->new_items( ( $self->new_items || 0 ) + 1 );
                }
        } else {
                warn "can't add entry ", dump( @_ ), "\n";
        }
}

=head2 content_class

Return class registred for particular content.

  my $class = $source->content_class( $content );

=cut

sub content_class {
        my $self = shift;

        my $content = shift or die "no content?";

        foreach my $s ( $self->sources ) {
                $self->log->debug("testing source class $s");
                if ( $s->can('content_have') ) {
                        my $regex =     $s->content_have( $content ) or
                                die "${s}->content_have didn't return anything";
                        die "${s}->content_have didn't return regex but ", dump( $regex ), " ref ", ref( $regex )
                                unless ( ref($regex) eq 'Regexp' );
                        if ( $content =~ $regex ) {
                                $self->log->debug("${s}->content_have succesful");
                                return $s;
                        }
                }
        }
}

=head2 scrape

Create semi-complex L<WWW::Mechanize> rules to scrape page


=cut

sub scrape {
        my $self = shift;

        my $args = {@_};

        $self->log->debug("scrape with args ",dump($args));

        my ($feed,$uri,$q) = ($self->feed, $self->uri,$self->q);
        die "no uri" unless ($uri);
        die "feed is not a Grep::Model::Feed but ", ref $feed unless $feed->isa('Grep::Model::Feed');

        sub mech_warn {
                my $m = shift || return;
                warn $m;
        }

        my $mech = WWW::Mechanize->new(
                cookie_jar => {},
                onwarn => \&mech_warn,
                onerror => \&mech_warn,
        );

        $mech->get( $uri );

        $self->save( 'get.html', $mech->content );

        if ( my $form = $args->{submit_form} ) {
                $self->log->debug("submit form on $uri with ", dump( $form ));
                $mech->submit_form( %$form ) or die "can't submit form ", dump( $form );
                $self->save( 'submit.html', $mech->content );
        }

        $self->log->debug("parse result page");

        my $tree = HTML::TreeBuilder->new or die "can't create html tree";
        $tree->parse( $mech->content ) or die "can't parse fetched content";

        die "wrapper doesn't have 3 elements but ", $#{ $args->{wrapper} } unless ( $#{ $args->{wrapper} } == 2 );
        my ( $el,$attr,$value ) = @{ $args->{wrapper} };

        $self->log->debug("looking for <$el $attr=\"$value\">");

        my $div = $tree->look_down( '_tag', $el, sub {
                        ( $_[0]->attr( $attr ) || '' ) eq $value;
        });

        if ( ! $div ) {
                warn "can't find results wrapper <$el $attr=\"$value\">";
                return;
        }

        my $max = 15;
        my $nr = 1;

        my $base_uri = $uri;
        $base_uri =~ s!\?.*$!!;

        foreach my $dt ( $div->look_down( '_tag', $args->{results} ) ) {
                my $a = $dt->look_down( '_tag', 'a', sub { $_[0]->attr('href') } );
                if ( $a ) {

                        my $href = $a->attr('href') or die "can't find href inside <", $args->{results}, ">";
                        my $page_uri = URI->new_abs( $a->attr('href'), $base_uri );
                        $page_uri->query( undef );
                        $page_uri = $page_uri->canonical;

                        $self->log->debug("fetching page: ",$a->as_text," from $page_uri");
                        if ( $mech->follow_link( url => $a->attr('href') ) ) {

                                $self->save( "page-${nr}.html", $mech->content );

                                my $page_tree = HTML::TreeBuilder->new or die "can't create page tree";
                                $page_tree->parse( $mech->content ) or die "can't parse page at $page_uri";

                                ( $el,$attr,$value ) = @{ $args->{scrape} };
                                $div = $page_tree->look_down( '_tag', $el, sub { ( $_[0]->attr( $attr ) || '' ) eq $value } );

                                die "can't find <$el $attr=\"$value\">" unless ($div);

                                $self->add_record(
                                        in_feed => $feed,
                                        title => $mech->title,
                                        link => $page_uri,
                                        content => $div->as_HTML,
#                                       summary => 
#                                       category => 
#                                       author => 
#                                       issued => 
#                                       modified => 
                                );

                                $mech->back;
                                $page_tree->delete;

                        } else {
                                warn "can't follow uri $page_uri: $!\n";
                        }
                }

                last if ($nr == $max);
                $nr++;
        }

        $tree->delete; # clear memory!

}

=head2 save

  save( 'name', $content );

Save dumps into C</tmp/grep> if writable

=cut

sub save {
        my $self = shift;
        my ( $file, $content ) = @_;
        return unless ( defined($file) && defined($content) );
        if ( -w '/tmp/grep' ) {
                open(my $f, '>', "/tmp/grep/$file") or die "can't open $file: $!";
                print $f $content or die "can't write to $file: $!";
                close $f or die "can't close $file: $!";
                $self->log->debug("saved $file ",length($content)," bytes");
        }
}

1;
1	dpavlin	72	# Dobrica Pavlinusic, <dpavlin@rot13.org> 02/22/07 20:30:00 CET
2
3			use strict;
4			use warnings;
5
6			package Grep::Source;
7
8			use Carp qw/verbose/;
9			use Module::Pluggable search_path => 'Grep::Source', sub_name => 'sources', require => 1;
10	dpavlin	100	use base qw(Class::Accessor Jifty::Object);
11	dpavlin	110	Grep::Source->mk_accessors( qw(feed uri q new_items collection search_obj) );
12	dpavlin	72
13	dpavlin	85	use HTML::TreeBuilder;
14			use WWW::Mechanize;
15			use XML::Feed;
16			use URI;
17	dpavlin	96	use HTML::ResolveLink;
18	dpavlin	85
19	dpavlin	72	use Data::Dump qw/dump/;
20
21			=head1 NAME
22
23			Grep::Source - base class for implementation of different sources for Grep
24
25			=head1 METHODS
26
27			This is mostly documentation because most of methods are implemented by plugins.
28
29			=head2 sources
30
31			my @sources = Grep::Source->sources();
32
33			Returns all available sources.
34
35			=cut
36
37	dpavlin	73	Jifty->log->debug("Found source plugins: ", join(", ", __PACKAGE__->sources() ) );
38	dpavlin	72
39			=head2 new
40
41			my $source = Grep::Source->new({ feed => $feed_record });
42
43	dpavlin	73	This will also setup:
44
45			=head2 feed
46
47			isa L<Grep::Model::Feed>
48
49	dpavlin	72	=head2 search
50
51			my $collection = $source->search( 'query string' );
52
53			It will also setup following accessors:
54
55	dpavlin	73	=head2 q
56	dpavlin	72
57			Search query
58
59	dpavlin	73	=head2 uri
60	dpavlin	72
61			URI of feed with embedded search query
62
63	dpavlin	73	=head2 new_items
64	dpavlin	72
65			Number of new items in result collection
66
67			=head2 collection
68
69			Actuall results which is L<Grep::Model::ItemCollection>, so following will
70			work:
71
72			print "and ", $self->collection->count, " total items";
73
74
75			Also setups number of new items
76
77			print $source->new_items, " items new";
78
79			=cut
80
81			sub search {
82			my $self = shift;
83
84			my $q = shift;
85
86			$q ? $self->q( $q ) : $q = $self->q;
87
88			die "no q?" unless ( $self->q );
89			die "no feed?" unless ( $self->feed );
90			die "feed not Grep::Model::Feed" unless ( $self->feed->isa('Grep::Model::Feed') );
91
92			my $message;
93			my $uri = $self->feed->uri;
94			if ($uri =~ m/%s/) {
95			$uri = $self->feed->search_uri( $q );
96			$message = 'Searching';
97			} else {
98			$message = 'Fetching';
99			}
100			$message .= ' ' . $self->feed->title . " at $uri";
101
102			$self->uri( $uri );
103
104	dpavlin	100	$self->log->info( $message );
105	dpavlin	72
106			$self->collection( Grep::Model::ItemCollection->new() );
107
108	dpavlin	73	my $class = $self->feed->source \|\| 'Grep::Source::Feed';
109	dpavlin	100	$self->log->debug("using $class");
110	dpavlin	72
111	dpavlin	110	$self->search_obj( Grep::Search->new() );
112			$self->log->debug("created " . $self->search_obj);
113	dpavlin	72
114	dpavlin	110	$class->fetch( $self );
115
116			$self->search_obj->finish;
117
118	dpavlin	72	return $self->collection;
119			}
120
121			=head2 add_record
122
123	dpavlin	73	Plugins will be called with parametar C<$parent> so they can call this method to add
124			record into result collection (and store in cache and index).
125
126	dpavlin	72	$parent->add_record( id => 42, foo => 'bar', ... );
127
128	dpavlin	73	This will also update L</new_items>
129
130	dpavlin	72	=cut
131
132			sub add_record {
133			my $self = shift;
134
135	dpavlin	110	$self->log->confess("no search_obj") unless ($self->search_obj);
136	dpavlin	109
137	dpavlin	72	my $i = Grep::Model::Item->new();
138
139	dpavlin	96	my $rec = {@_};
140	dpavlin	72
141	dpavlin	102	$self->log->debug("resolving links using base ", $rec->{link});
142	dpavlin	96	my $resolver = HTML::ResolveLink->new( base => $rec->{link} );
143			$rec->{content} = $resolver->resolve( $rec->{content} );
144
145			my ($ok,$msg) = $i->load_or_create( %$rec );
146
147	dpavlin	72	$msg \|\|= '';
148
149			if ( $ok ) {
150	dpavlin	100	$self->log->debug("item ", $i->id, ": $msg");
151	dpavlin	72	$self->collection->add_record( $i );
152
153			# is new record?
154			if ( $msg !~ m/^Found/ ) {
155	dpavlin	110	$self->search_obj->add( $i );
156	dpavlin	74	$self->new_items( ( $self->new_items \|\| 0 ) + 1 );
157	dpavlin	72	}
158			} else {
159			warn "can't add entry ", dump( @_ ), "\n";
160			}
161			}
162
163	dpavlin	73	=head2 content_class
164
165			Return class registred for particular content.
166
167			my $class = $source->content_class( $content );
168
169			=cut
170
171			sub content_class {
172			my $self = shift;
173
174			my $content = shift or die "no content?";
175
176			foreach my $s ( $self->sources ) {
177	dpavlin	100	$self->log->debug("testing source class $s");
178	dpavlin	82	if ( $s->can('content_have') ) {
179			my $regex = $s->content_have( $content ) or
180			die "${s}->content_have didn't return anything";
181			die "${s}->content_have didn't return regex but ", dump( $regex ), " ref ", ref( $regex )
182			unless ( ref($regex) eq 'Regexp' );
183			if ( $content =~ $regex ) {
184	dpavlin	100	$self->log->debug("${s}->content_have succesful");
185	dpavlin	82	return $s;
186			}
187	dpavlin	73	}
188			}
189			}
190
191	dpavlin	85	=head2 scrape
192
193			Create semi-complex L<WWW::Mechanize> rules to scrape page
194
195
196			=cut
197
198			sub scrape {
199			my $self = shift;
200
201			my $args = {@_};
202
203	dpavlin	102	$self->log->debug("scrape with args ",dump($args));
204	dpavlin	85
205			my ($feed,$uri,$q) = ($self->feed, $self->uri,$self->q);
206			die "no uri" unless ($uri);
207			die "feed is not a Grep::Model::Feed but ", ref $feed unless $feed->isa('Grep::Model::Feed');
208
209	dpavlin	92	sub mech_warn {
210			my $m = shift \|\| return;
211			warn $m;
212			}
213	dpavlin	85
214	dpavlin	92	my $mech = WWW::Mechanize->new(
215			cookie_jar => {},
216			onwarn => \&mech_warn,
217			onerror => \&mech_warn,
218			);
219
220	dpavlin	85	$mech->get( $uri );
221
222	dpavlin	86	$self->save( 'get.html', $mech->content );
223	dpavlin	85
224	dpavlin	92	if ( my $form = $args->{submit_form} ) {
225	dpavlin	102	$self->log->debug("submit form on $uri with ", dump( $form ));
226	dpavlin	92	$mech->submit_form( %$form ) or die "can't submit form ", dump( $form );
227	dpavlin	86	$self->save( 'submit.html', $mech->content );
228	dpavlin	85	}
229
230	dpavlin	102	$self->log->debug("parse result page");
231	dpavlin	85
232			my $tree = HTML::TreeBuilder->new or die "can't create html tree";
233			$tree->parse( $mech->content ) or die "can't parse fetched content";
234
235			die "wrapper doesn't have 3 elements but ", $#{ $args->{wrapper} } unless ( $#{ $args->{wrapper} } == 2 );
236			my ( $el,$attr,$value ) = @{ $args->{wrapper} };
237
238	dpavlin	102	$self->log->debug("looking for <$el $attr=\"$value\">");
239	dpavlin	85
240			my $div = $tree->look_down( '_tag', $el, sub {
241			( $_[0]->attr( $attr ) \|\| '' ) eq $value;
242			});
243
244	dpavlin	88	if ( ! $div ) {
245			warn "can't find results wrapper <$el $attr=\"$value\">";
246			return;
247			}
248	dpavlin	85
249	dpavlin	103	my $max = 15;
250	dpavlin	85	my $nr = 1;
251
252			my $base_uri = $uri;
253			$base_uri =~ s!\?.*$!!;
254
255			foreach my $dt ( $div->look_down( '_tag', $args->{results} ) ) {
256			my $a = $dt->look_down( '_tag', 'a', sub { $_[0]->attr('href') } );
257			if ( $a ) {
258
259			my $href = $a->attr('href') or die "can't find href inside <", $args->{results}, ">";
260			my $page_uri = URI->new_abs( $a->attr('href'), $base_uri );
261			$page_uri->query( undef );
262			$page_uri = $page_uri->canonical;
263
264	dpavlin	102	$self->log->debug("fetching page: ",$a->as_text," from $page_uri");
265	dpavlin	85	if ( $mech->follow_link( url => $a->attr('href') ) ) {
266
267	dpavlin	86	$self->save( "page-${nr}.html", $mech->content );
268	dpavlin	85
269			my $page_tree = HTML::TreeBuilder->new or die "can't create page tree";
270			$page_tree->parse( $mech->content ) or die "can't parse page at $page_uri";
271
272	dpavlin	92	( $el,$attr,$value ) = @{ $args->{scrape} };
273			$div = $page_tree->look_down( '_tag', $el, sub { ( $_[0]->attr( $attr ) \|\| '' ) eq $value } );
274	dpavlin	85
275			die "can't find <$el $attr=\"$value\">" unless ($div);
276
277			$self->add_record(
278			in_feed => $feed,
279			title => $mech->title,
280			link => $page_uri,
281			content => $div->as_HTML,
282			# summary =>
283			# category =>
284			# author =>
285			# issued =>
286			# modified =>
287			);
288
289			$mech->back;
290			$page_tree->delete;
291
292			} else {
293			warn "can't follow uri $page_uri: $!\n";
294			}
295			}
296
297			last if ($nr == $max);
298			$nr++;
299			}
300
301			$tree->delete; # clear memory!
302
303			}
304
305	dpavlin	86	=head2 save
306
307			save( 'name', $content );
308
309			Save dumps into C</tmp/grep> if writable
310
311			=cut
312
313			sub save {
314			my $self = shift;
315			my ( $file, $content ) = @_;
316	dpavlin	100	return unless ( defined($file) && defined($content) );
317	dpavlin	86	if ( -w '/tmp/grep' ) {
318			open(my $f, '>', "/tmp/grep/$file") or die "can't open $file: $!";
319			print $f $content or die "can't write to $file: $!";
320			close $f or die "can't close $file: $!";
321	dpavlin	100	$self->log->debug("saved $file ",length($content)," bytes");
322	dpavlin	86	}
323			}
324
325	dpavlin	72	1;