| version 1.16, 2005/05/05 02:09:43 |
version 1.27, 2010/01/05 19:43:44 |
|
|
| #!/usr/bin/perl -T |
#!/usr/bin/perl -T |
| #$Id$ |
#$RedRiver: CurrentTorrents.pl,v 1.26 2009/10/20 19:04:28 andrew Exp $ |
| use strict; |
use strict; |
| use warnings; |
use warnings; |
| use diagnostics; |
use diagnostics; |
| |
|
| use Time::Local; |
use Time::Local; |
| use Fcntl ':flock'; |
use Fcntl ':flock'; |
| |
use File::Basename; |
| |
|
| |
#use YAML; |
| |
|
| use lib 'lib'; |
use lib 'lib'; |
| use OpenBSDTorrents; |
use OpenBSDTorrents; |
| use BT::OBTMetaInfo; |
use BT::MetaInfo::Cached; |
| |
|
| %ENV = (); |
%ENV = (); |
| |
|
| use YAML; |
|
| |
|
| #justme(); |
#justme(); |
| |
|
| my $Name_Filter = shift || ''; |
my $Name_Filter = shift || ''; |
| if ($Name_Filter =~ /^(\w*)$/) { |
if ( $Name_Filter =~ /^(\w*)$/ ) { |
| $Name_Filter = $1; |
$Name_Filter = $1; |
| } else { |
|
| die "Invalid filter: $Name_Filter"; |
|
| } |
} |
| |
else { |
| |
die "Invalid filter: $Name_Filter"; |
| |
} |
| |
|
| my %Possible_Torrents; |
my %Possible_Torrents; |
| Process_Dir($OBT->{DIR_FTP}); |
Process_Dir( $OBT->{DIR_FTP} ); |
| |
|
| my %files; |
my %files; |
| my %keep; |
my %keep; |
| my @delete; |
my @delete; |
| foreach my $DIR ($OBT->{DIR_NEW_TORRENT}, $OBT->{DIR_TORRENT}) { |
foreach my $DIR ( $OBT->{DIR_NEW_TORRENT}, $OBT->{DIR_TORRENT} ) { |
| opendir DIR, $DIR |
opendir DIR, $DIR |
| or die "Couldn't opendir $DIR: $!"; |
or die "Couldn't opendir $DIR: $!"; |
| foreach (readdir DIR) { |
foreach ( readdir DIR ) { |
| next unless my ($ext) = /\.(torrent|$OBT->{META_EXT})$/; |
next unless my ($ext) = /\.(torrent|$OBT->{META_EXT})$/; |
| |
|
| if (/^([^\/]+)$/) { |
if (/^([^\/]+)$/) { |
| $_ = $1; |
$_ = $1; |
| } else { |
} |
| die "Invalid character in $_: $!"; |
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--; |
if ( my ( $base, $year, $mon, $mday, $hour, $min ) |
| my $epoch = timegm(0,$min,$hour,$mday,$mon,$year); |
= /^(.*)-(\d{4})-(\d{2})-(\d{2})-(\d{2})(\d{2})/ ) |
| |
{ |
| |
|
| #print "Adding $_\n"; |
$mon--; |
| |
$epoch = timegm( 0, $min, $hour, $mday, $mon, $year ); |
| |
$name = $base; |
| |
} |
| |
|
| $files{$ext}{$name}{$epoch} = { |
#print "Adding $_\n"; |
| file => $_, |
|
| dir => $DIR, |
|
| path => "$DIR/$_", |
|
| ext => $ext, |
|
| year => $year, |
|
| mon => $mon, |
|
| mday => $mday, |
|
| hour => $hour, |
|
| min => $min, |
|
| name => $name, |
|
| epoch => $epoch, |
|
| }; |
|
| |
|
| unless (exists $Possible_Torrents{$name}) { |
$files{$ext}{$name}{$epoch} = { |
| print "Would remove $_\n"; |
file => $_, |
| push @delete, $files{$ext}{$name}{$epoch}; |
dir => $DIR, |
| } |
path => "$DIR/$_", |
| } |
ext => $ext, |
| closedir DIR; |
|
| } |
|
| |
|
| foreach my $name (keys %{ $files{torrent} }) { |
#year => $year, |
| next unless $name =~ /^$Name_Filter/; |
#mon => $mon, |
| print "Checking $name\n"; |
#mday => $mday, |
| |
#hour => $hour, |
| |
#min => $min, |
| |
name => $name, |
| |
epoch => $epoch, |
| |
}; |
| |
|
| foreach my $epoch ( sort { $b <=> $a } keys %{ $files{torrent}{$name} } ) { |
if ( $name =~ m/\A $OBT->{BASENAME} /xms |
| #print "\t$epoch\n"; |
&& !exists $Possible_Torrents{$name} ) |
| my $torrent = $files{torrent}{$name}{$epoch}{path}; |
{ |
| |
print "Would remove $_\n"; |
| |
push @delete, $files{$ext}{$name}{$epoch}; |
| |
} |
| |
} |
| |
closedir DIR; |
| |
} |
| |
|
| if ( |
#print Dump \%files; |
| keys %{ $files{torrent}{$name} } == 1 && |
foreach my $name ( sort keys %{ $files{torrent} } ) { |
| $files{torrent}{$name}{$epoch}{dir} |
next unless $name =~ /^$Name_Filter/; |
| eq $OBT->{DIR_TORRENT} |
|
| ) { |
|
| #print "Skipping torrent for $name there is only one.\n"; |
|
| next; |
|
| } |
|
| |
|
| my $meta_file = $torrent; |
#print "Checking $name\n"; |
| $meta_file =~ s/\.torrent$/.$OBT->{META_EXT}/; |
|
| |
|
| my $hash = undef; |
foreach my $epoch ( sort { $b <=> $a } keys %{ $files{torrent}{$name} } ) |
| if (-e $meta_file) { |
{ |
| #print "Reading meta file: $meta_file\n"; |
|
| open my $meta, $meta_file |
|
| or die "Couldn't open $meta_file: $!"; |
|
| flock($meta, LOCK_SH); |
|
| binmode $meta; |
|
| |
|
| $hash = do { local $/; <$meta> }; |
#print "\t$epoch\n"; |
| |
my $torrent = $files{torrent}{$name}{$epoch}{path}; |
| |
|
| flock($meta, LOCK_UN); |
if ( keys %{ $files{torrent}{$name} } == 1 |
| close $meta; |
&& $files{torrent}{$name}{$epoch}{dir} eq $OBT->{DIR_TORRENT} ) |
| } else { |
{ |
| |
|
| my $t; |
#print "Skipping torrent for $name there is only one.\n"; |
| eval { $t = BT::OBTMetaInfo->new( $torrent ); }; |
next; |
| |
} |
| |
|
| if ($@) { |
my $t; |
| warn "Error reading torrent $torrent\n"; |
eval { |
| next; |
$t = BT::MetaInfo::Cached->new( |
| } |
$torrent, |
| |
{ cache_root => '/tmp/OBTFileCache' |
| |
|
| $files{torrent}{$name}{$epoch}{comment} = $t->{comment}; |
#$OBT->{DIR_HOME} . '/FileCache' |
| my ($path) = $t->{comment} =~ /Files from ([^\n]+)\n/s; |
} |
| |
); |
| |
}; |
| |
|
| unless (-d $OBT->{DIR_FTP} . "/$path") { |
if ($@) { |
| #print "Deleting $files{torrent}{$name}{$epoch}{file} the path doesn't exist.\n"; |
warn "Error reading torrent $torrent\n"; |
| push @delete, $files{torrent}{$name}{$epoch}; |
push @delete, $files{torrent}{$name}{$epoch}; |
| delete $files{torrent}{$name}{$epoch}; |
delete $files{torrent}{$name}{$epoch}; |
| next; |
next; |
| } |
} |
| |
|
| $hash = $t->info_hash_cached($torrent); |
$files{torrent}{$name}{$epoch}{comment} = $t->{comment}; |
| $hash = unpack("H*", $hash); |
my ($path) = $t->{comment} =~ /($OBT->{BASENAME}\/[^\n]+)\n/s; |
| |
|
| undef $t; |
unless ( -e $OBT->{DIR_FTP} . "/$path" ) { |
| |
print |
| |
"Deleting $files{torrent}{$name}{$epoch}{file} the path ($path) doesn't exist.\n"; |
| |
push @delete, $files{torrent}{$name}{$epoch}; |
| |
delete $files{torrent}{$name}{$epoch}; |
| |
next; |
| |
} |
| |
|
| } |
my $hash = $t->info_hash; |
| |
$hash = unpack( "H*", $hash ); |
| |
|
| $files{torrent}{$name}{$epoch}{info_hash} = $hash; |
undef $t; |
| |
|
| |
$files{torrent}{$name}{$epoch}{info_hash} = $hash; |
| |
|
| if (exists $keep{$name}) { |
if ( exists $keep{$name} ) { |
| if (exists $keep{$name}{$hash}) { |
if ( exists $keep{$name}{$hash} ) { |
| push @delete, $keep{$name}{$hash}; |
if ( $keep{$name}{$hash}{epoch} == $epoch ) { |
| delete $files{torrent}{ |
next; |
| $keep{$name}{$hash}{name} |
} |
| }{ |
print "Removing [$name] [$hash]\n\t", |
| $keep{$name}{$hash}{epoch} |
$keep{$name}{$hash}{path}, |
| }; |
"\n"; |
| $keep{$name}{$hash} = |
push @delete, $keep{$name}{$hash}; |
| $files{torrent}{$name}{$epoch}; |
delete $files{torrent}{ $keep{$name}{$hash}{name} } |
| } else { |
{ $keep{$name}{$hash}{epoch} }; |
| push @delete, $files{torrent}{$name}{$epoch}; |
$keep{$name}{$hash} = $files{torrent}{$name}{$epoch}; |
| delete $files{torrent}{$name}{$epoch}; |
print "Keeping additional instance of [$name] [$hash]\n\t", |
| } |
$keep{$name}{$hash}{path}, |
| } else { |
"\n"; |
| $keep{$name}{$hash} = |
} |
| $files{torrent}{$name}{$epoch}; |
else { |
| |
print "Removing old [$name] [$hash]\n"; |
| |
if ( $keep{$name}{$hash}{path} ) { |
| |
print "\t", $keep{$name}{$hash}{path}, "\n"; |
| |
} |
| |
push @delete, $files{torrent}{$name}{$epoch}; |
| |
delete $files{torrent}{$name}{$epoch}; |
| |
} |
| |
} |
| |
else { |
| |
print "Keeping first instance of $name [$hash]\n\t", |
| |
$files{torrent}{$name}{$epoch}{path}, |
| |
"\n"; |
| |
$keep{$name}{$hash} = $files{torrent}{$name}{$epoch}; |
| |
|
| } |
} |
| } |
} |
| } |
} |
| |
|
| #print Dump \%files, \%keep, \@delete; |
#print Dump \%files, \%keep, \@delete; |
| |
#exit; |
| |
|
| foreach (@delete) { |
foreach (@delete) { |
| print "Deleting '$_->{path}'\n"; |
print "Deleting '$_->{path}'\n"; |
| unlink $_->{path} or die "Couldn't unlink $_->{path}"; |
unlink $_->{path} or die "Couldn't unlink $_->{path}"; |
| } |
} |
| |
|
| foreach my $name (keys %{ $files{$OBT->{META_EXT} } }) { |
foreach my $name ( keys %{ $files{ $OBT->{META_EXT} } } ) { |
| foreach my $epoch (keys %{ $files{ $OBT->{META_EXT} }{$name} }) { |
foreach my $epoch ( keys %{ $files{ $OBT->{META_EXT} }{$name} } ) { |
| unless ( exists $files{torrent}{$name}{$epoch} ) { |
unless ( exists $files{torrent}{$name}{$epoch} ) { |
| my $path = $files{$OBT->{META_EXT}}{$name}{$epoch}{path}; |
my $path = $files{ $OBT->{META_EXT} }{$name}{$epoch}{path}; |
| print "Unlinking '$path'\n"; |
print "Unlinking '$path'\n"; |
| unlink $path or die "couldn't unlink '$path': $!"; |
unlink $path or die "couldn't unlink '$path': $!"; |
| } |
} |
| } |
} |
| } |
} |
| |
|
| |
|
| #print Dump \%keep; |
#print Dump \%keep; |
| foreach my $name (keys %keep) { |
foreach my $name ( keys %keep ) { |
| foreach my $hash (keys %{ $keep{$name} }) { |
foreach my $hash ( keys %{ $keep{$name} } ) { |
| my $file = $keep{$name}{$hash}{file}; |
my $file = $keep{$name}{$hash}{file} || q{}; |
| my $dir = $keep{$name}{$hash}{dir }; |
my $dir = $keep{$name}{$hash}{dir} || q{}; |
| if ($dir eq $OBT->{DIR_NEW_TORRENT}) { |
if ( $dir eq $OBT->{DIR_NEW_TORRENT} ) { |
| print "Moving $file to current torrents\n"; |
print "Moving $file to current torrents\n"; |
| rename("$dir/$file", $OBT->{DIR_TORRENT} . "/" . $file) |
rename( "$dir/$file", $OBT->{DIR_TORRENT} . "/" . $file ) |
| or die "Couldn't rename '$file': $!"; |
or die "Couldn't rename '$file': $!"; |
| |
|
| my $name = $keep{$name}{$hash}{name}; |
my $name = $keep{$name}{$hash}{name}; |
| my $epoch = $keep{$name}{$hash}{epoch}; |
my $epoch = $keep{$name}{$hash}{epoch}; |
| |
|
| if (exists $files{txt}{$name}{$epoch}) { |
if ( exists $files{txt}{$name}{$epoch} ) { |
| my $m_file = $files{txt}{$name}{$epoch}{file}; |
my $m_file = $files{txt}{$name}{$epoch}{file}; |
| my $m_dir = $files{txt}{$name}{$epoch}{dir }; |
my $m_dir = $files{txt}{$name}{$epoch}{dir}; |
| rename( |
rename( "$m_dir/$m_file", |
| "$m_dir/$m_file", |
$OBT->{DIR_TORRENT} . "/" . $m_file ) |
| $OBT->{DIR_TORRENT} . "/" . $m_file |
or die "Couldn't rename '$m_file': $!"; |
| ) or die "Couldn't rename '$m_file': $!"; |
} |
| } |
} |
| } |
} |
| } |
|
| } |
} |
| |
|
| sub Process_Dir |
sub Process_Dir { |
| { |
my $basedir = shift; |
| my $basedir = shift; |
|
| |
|
| my ($dirs, $files) = Get_Files_and_Dirs($basedir); |
my ( $dirs, $files ) = Get_Files_and_Dirs($basedir); |
| if (@$files) { |
if (@$files) { |
| my $dir = $basedir; |
my $dir = $basedir; |
| $dir =~ s/^$OBT->{DIR_FTP}\///; |
$dir =~ s/^$OBT->{DIR_FTP}\///; |
| my $torrent = Name_Torrent($dir); |
my $torrent = Name_Torrent($dir); |
| $torrent =~ s/-.*$//; |
$torrent =~ s/-.*$//; |
| $Possible_Torrents{$torrent} = 1; |
$Possible_Torrents{$torrent} = 1; |
| |
foreach my $file (@$files) { |
| |
if ( $file =~ /$INSTALL_ISO_REGEX/ ) { |
| |
$torrent = Name_Torrent("$dir/$file"); |
| |
$torrent =~ s/-.*$//; |
| |
$Possible_Torrents{$torrent} = 1; |
| |
} |
| } |
} |
| |
} |
| |
|
| foreach my $subdir (@$dirs) { |
foreach my $subdir (@$dirs) { |
| next if $subdir eq '.'; |
next if $subdir eq '.'; |
| next if $subdir eq '..'; |
next if $subdir eq '..'; |
| Process_Dir("$basedir/$subdir") |
Process_Dir("$basedir/$subdir"); |
| } |
} |
| } |
} |
| |
|