=================================================================== RCS file: /cvs/openbsd/OpenBSDTorrents/CurrentTorrents.pl,v retrieving revision 1.13 retrieving revision 1.23 diff -u -r1.13 -r1.23 --- openbsd/OpenBSDTorrents/CurrentTorrents.pl 2005/05/02 22:43:53 1.13 +++ openbsd/OpenBSDTorrents/CurrentTorrents.pl 2007/10/01 21:17:23 1.23 @@ -1,21 +1,22 @@ #!/usr/bin/perl -T -#$Id: CurrentTorrents.pl,v 1.13 2005/05/02 21:43:53 andrew Exp $ +#$RedRiver: CurrentTorrents.pl,v 1.22 2006/07/24 18:03:53 andrew Exp $ use strict; use warnings; use diagnostics; use Time::Local; +use Fcntl ':flock'; +use File::Basename; +#use YAML; use lib 'lib'; use OpenBSDTorrents; -use BT::OBTMetaInfo; +use BT::MetaInfo::Cached; %ENV = (); -use YAML; +#justme(); -justme(); - my $Name_Filter = shift || ''; if ($Name_Filter =~ /^(\w*)$/) { $Name_Filter = $1; @@ -23,7 +24,12 @@ die "Invalid filter: $Name_Filter"; } +my %Possible_Torrents; +Process_Dir($OBT->{DIR_FTP}); + my %files; +my %keep; +my @delete; foreach my $DIR ($OBT->{DIR_NEW_TORRENT}, $OBT->{DIR_TORRENT}) { opendir DIR, $DIR or die "Couldn't opendir $DIR: $!"; @@ -35,12 +41,17 @@ } else { die "Invalid character in $_: $!"; } - my ($name, $year, $mon, $mday, $hour, $min) = - /^(.*)-(\d{4})-(\d{2})-(\d{2})-(\d{2})(\d{2})/; + my $epoch = 0; + my $name = basename($_, '.torrent'); - $mon--; - my $epoch = timegm(0,$min,$hour,$mday,$mon,$year); + if (my ($base, $year, $mon, $mday, $hour, $min) = + /^(.*)-(\d{4})-(\d{2})-(\d{2})-(\d{2})(\d{2})/) { + $mon--; + $epoch = timegm(0,$min,$hour,$mday,$mon,$year); + $name = $base; + } + #print "Adding $_\n"; $files{$ext}{$name}{$epoch} = { @@ -48,31 +59,53 @@ dir => $DIR, path => "$DIR/$_", ext => $ext, - year => $year, - mon => $mon, - mday => $mday, - hour => $hour, - min => $min, + #year => $year, + #mon => $mon, + #mday => $mday, + #hour => $hour, + #min => $min, name => $name, epoch => $epoch, }; + if ( + $name =~ m/\A $OBT->{BASENAME} /xms && + ! exists $Possible_Torrents{$name} + ) { + print "Would remove $_\n"; + push @delete, $files{$ext}{$name}{$epoch}; + } } closedir DIR; } -my %keep; -my @delete; foreach my $name (keys %{ $files{torrent} }) { next unless $name =~ /^$Name_Filter/; - print "Checking $name\n"; + #print "Checking $name\n"; foreach my $epoch ( sort { $b <=> $a } keys %{ $files{torrent}{$name} } ) { #print "\t$epoch\n"; my $torrent = $files{torrent}{$name}{$epoch}{path}; + if ( + keys %{ $files{torrent}{$name} } == 1 && + $files{torrent}{$name}{$epoch}{dir} + eq $OBT->{DIR_TORRENT} + ) { + #print "Skipping torrent for $name there is only one.\n"; + next; + } + my $t; - eval { $t = BT::OBTMetaInfo->new( $torrent ); }; + eval { + $t = BT::MetaInfo::Cached->new( + $torrent, + { + cache_root => '/tmp/OBTFileCache' + #$OBT->{DIR_HOME} . '/FileCache' + } + ); + }; if ($@) { warn "Error reading torrent $torrent\n"; @@ -89,21 +122,13 @@ next; } - if ( - keys %{ $files{torrent}{$name} } == 1 && - $files{torrent}{$name}{$epoch}{dir} - eq $OBT->{DIR_TORRENT} - ) { - #print "Skipping torrent for $name there is only one.\n"; - next; - } - - my $hash = $t->info_hash_cached($torrent); + my $hash = $t->info_hash; $hash = unpack("H*", $hash); + undef $t; + $files{torrent}{$name}{$epoch}{info_hash} = $hash; - undef $t; if (exists $keep{$name}) { if (exists $keep{$name}{$hash}) { @@ -128,6 +153,7 @@ } #print Dump \%files, \%keep, \@delete; +#exit; foreach (@delete) { print "Deleting '$_->{path}'\n"; @@ -169,3 +195,24 @@ } } } + +sub Process_Dir +{ + my $basedir = shift; + + my ($dirs, $files) = Get_Files_and_Dirs($basedir); + if (@$files) { + my $dir = $basedir; + $dir =~ s/^$OBT->{DIR_FTP}\///; + my $torrent = Name_Torrent($dir); + $torrent =~ s/-.*$//; + $Possible_Torrents{$torrent} = 1; + } + + foreach my $subdir (@$dirs) { + next if $subdir eq '.'; + next if $subdir eq '..'; + Process_Dir("$basedir/$subdir") + } +} +