save collection_items in x_meta
[ILL-Zotero-RT] / zotero.pl
index bc240f9..a0a424b 100755 (executable)
--- a/zotero.pl
+++ b/zotero.pl
@@ -44,14 +44,18 @@ $url .= '&key=' . $key;
 my $file = $UserID . '.' . md5_hex($url) . '.atom';
 $FETCH = 1 if ! -e $file;
 
-warn "# $url -> $file\n";
+warn "# mirror $FETCH $url -> $file\n";
 if ( $FETCH && mirror( $url => $file ) == RC_NOT_MODIFIED ) {
        warn "not modified";
-#      exit 0;
 }
 
-my $feed = XMLin( $file );
-#warn "# feed ",dump($feed);
+my $xml = XML::Simple->new(ForceArray => [ qw( entry ) ]);
+my $feed = eval { $xml->XMLin( $file ) };
+if ( $! ) {
+       warn "ERROR $file $!\n";
+       goto skip_url;
+}
+warn "# feed ",dump($feed);
 
 sub link_to_id {
        my $link = shift;
@@ -60,12 +64,16 @@ sub link_to_id {
        return $link;
 }
 
+my @collection_items;
+
 foreach my $entry ( keys %{ $feed->{entry} } ) {
        warn "# entry $entry ",dump($entry);
        my $id = link_to_id $entry;
 
+       push @collection_items, $id if $url =~ m{/collections/(\w+)/items};
+
        my $item = $feed->{entry}->{$entry};
-       warn "# entry $entry ",dump($item);
+       warn "# item $id $entry ",dump($item),$/;
 
        foreach my $i ( 0 .. $#{ $item->{link} } ) {
                my $link = $item->{link}->[$i];
@@ -75,6 +83,9 @@ foreach my $entry ( keys %{ $feed->{entry} } ) {
 
                if ( $link->{rel} eq 'up' ) {
                        push @{ $tree->{$key} }, $id;
+               } elsif ( $link->{rel} eq 'self' && $link->{href} =~ m{/collections/} ) {
+                       warn "# get items in this collection";
+                       push @urls, "$link->{href}/items?content=json";
                }
        }
 
@@ -117,6 +128,15 @@ foreach my $entry ( keys %{ $feed->{entry} } ) {
 
 }
 
+if ( @collection_items ) {
+       my $id = $1 if $url =~ m{/collections/(\w+)/items};
+       $db->modify( "zotero_$UserID/$id" => sub {
+               my $doc = shift;
+               $doc->{x_meta}->{collection_items} = [ @collection_items ];
+               return $doc;
+       });
+}
+
 delete $feed->{entry};
 warn "# feed without entry ",dump( $feed );
 
@@ -126,6 +146,8 @@ if ( my @next = map { $_->{href} } grep { $_->{rel} eq 'next' && $_->{type} eq '
        goto restart;
 }
 
+skip_url:
+
 if ( $url = shift @urls ) {
        warn "## next url $url";
        goto restart;