[BACK]Return to CurrentTorrents.pl CVS log [TXT][DIR] Up to [local] / openbsd / OpenBSDTorrents

Diff for /openbsd/OpenBSDTorrents/CurrentTorrents.pl between version 1.16 and 1.30

version 1.16, 2005/05/05 02:09:43 version 1.30, 2010/02/25 18:03:01
Line 1 
Line 1 
 #!/usr/bin/perl -T  #!/usr/bin/perl -T
 #$Id$  #$RedRiver: CurrentTorrents.pl,v 1.29 2010/01/08 17:18:35 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 Transmission::Client;
   use Transmission::Utils;
   
   #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 @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;  
               #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;
 }  }
   
 foreach my $name (keys %{ $files{torrent} }) {  #print Dump \%files;
         next unless $name =~ /^$Name_Filter/;  
         print "Checking $name\n";  
   
         foreach my $epoch ( sort { $b <=> $a } keys %{ $files{torrent}{$name} } ) {  my %keep;
                 #print "\t$epoch\n";  my %seen;
                 my $torrent = $files{torrent}{$name}{$epoch}{path};  foreach my $name ( sort keys %{ $files{torrent} } ) {
       next unless $name =~ /^$Name_Filter/;
   
                 if (      #next if $name =~ /_packages_/xms;
                         keys %{ $files{torrent}{$name} } == 1 &&      #print "Checking $name\n";
                         $files{torrent}{$name}{$epoch}{dir}  
                                 eq $OBT->{DIR_TORRENT}  
                 ) {  
                         #print "Skipping torrent for $name there is only one.\n";  
                         next;  
                 }  
   
                 my $meta_file = $torrent;      my $cn = $files{torrent}{$name};
                 $meta_file =~ s/\.torrent$/.$OBT->{META_EXT}/;  
   
                 my $hash = undef;  EPOCH: foreach my $epoch ( sort { $a <=> $b } keys %{$cn} ) {
                 if (-e $meta_file) {          my $ct = $cn->{$epoch};
                         #print "Reading meta file: $meta_file\n";          my $cf = $ct->{path};
                         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 - $cf\n";
   
                         flock($meta, LOCK_UN);          my $t;
                         close $meta;          eval {
                 } else {              $t
                   = BT::MetaInfo::Cached->new( $cf,
                   { cache_root => '/tmp/OBTFileCache' } );
           };
   
                 my $t;          if ($@) {
                 eval { $t = BT::OBTMetaInfo->new( $torrent ); };              warn "Error reading torrent $cf\n";
               push @delete, $ct;
               delete $cn->{$epoch};
               next EPOCH;
           }
   
                 if ($@) {          $ct->{comment} = $t->{comment};
                         warn "Error reading torrent $torrent\n";          my ($path) = $t->{comment} =~ /($OBT->{BASENAME}\/[^\n]+)\n/s;
                         next;  
                 }  
   
                 $files{torrent}{$name}{$epoch}{comment}   = $t->{comment};          if ( !-e $OBT->{DIR_FTP} . "/$path" ) {
                 my ($path) = $t->{comment} =~ /Files from ([^\n]+)\n/s;              print
                   'Deleting ',
                   $cn->{$epoch}{file}, ' the path (', $path,
                   ") doesn't exist.\n";
               push @delete, $ct;
               delete $cn->{$epoch};
               next EPOCH;
           }
   
                 unless (-d $OBT->{DIR_FTP} . "/$path") {          my $hash = unpack( "H*", $t->info_hash );
                         #print "Deleting $files{torrent}{$name}{$epoch}{file} the path doesn't exist.\n";          $ct->{info_hash} = $hash;
                         push @delete, $files{torrent}{$name}{$epoch};  
                         delete $files{torrent}{$name}{$epoch};  
                         next;  
                 }  
   
                 $hash = $t->info_hash_cached($torrent);          undef $t;
                 $hash = unpack("H*", $hash);  
   
                 undef $t;          if ( $seen{$name} && $seen{$name} ne $hash ) {
               print "Removing older [$name] [$hash]\n";
               if ( $keep{$hash}{path} ) {
                   print "\t", $keep{$hash}{path}, "\n";
               }
               push @delete, $ct;
               delete $cn->{$epoch};
               next EPOCH;
           }
           $seen{$name} = $hash;
   
                 }          if ( keys %{$cn} == 1 && $ct->{dir} eq $OBT->{DIR_TORRENT} ) {
               $keep{$hash} = $ct;
   
                 $files{torrent}{$name}{$epoch}{info_hash} = $hash;              #print "Keeping only instance of [$name] [$hash]\n\t",
               #    $ct->{path},
               #    "\n";
               next EPOCH;
           }
           elsif ( $keep{$hash} ) {
               if ( $keep{$hash}{epoch} == $epoch ) {
                   next EPOCH;
               }
   
               print "Removing duplicate [$name] [$hash]\n\t",
                   $keep{$hash}{path}, "\n";
               push @delete, $keep{$hash};
               delete $files{torrent}{ $keep{$hash}{name} }
                   { $keep{$hash}{epoch} };
   
                 if (exists $keep{$name}) {              $keep{$hash} = $ct;
                         if (exists $keep{$name}{$hash}) {              print "Keeping additional instance of [$name] [$hash]\n\t",
                                 push @delete, $keep{$name}{$hash};                  $ct->{path},
                                 delete $files{torrent}{                  "\n";
                                         $keep{$name}{$hash}{name}          }
                                 }{          else {
                                         $keep{$name}{$hash}{epoch}              $keep{$hash} = $ct;
                                 };              print "Keeping first instance of [$name] [$hash]\n\t",
                                 $keep{$name}{$hash} =                  $ct->{path},
                                         $files{torrent}{$name}{$epoch};                  "\n";
                         } else {  
                                 push @delete, $files{torrent}{$name}{$epoch};  
                                 delete $files{torrent}{$name}{$epoch};  
                         }  
                 } else {  
                         $keep{$name}{$hash} =  
                                 $files{torrent}{$name}{$epoch};  
   
                 }          }
         }      }
 }  }
   
 #print Dump \%files, \%keep, \@delete;  #print Dump \%files, \%keep, \@delete;
   #exit;
   
 foreach (@delete) {  my $client = Transmission::Client->new;
         print "Deleting '$_->{path}'\n";  my %seeding;
         unlink $_->{path} or die "Couldn't unlink $_->{path}";  foreach my $torrent ( @{ $client->torrents } ) {
 }  
   
 foreach my $name (keys %{ $files{$OBT->{META_EXT} } }) {      #my $status = Transmission::Utils::from_numeric_status($torrent->status);
         foreach my $epoch (keys %{ $files{ $OBT->{META_EXT} }{$name} }) {      my $hash = $torrent->hash_string;
                 unless ( exists $files{torrent}{$name}{$epoch} ) {      if ( exists $keep{$hash} ) {
                         my $path = $files{$OBT->{META_EXT}}{$name}{$epoch}{path};          $seeding{$hash} = $torrent;
                         print "Unlinking '$path'\n";      }
                         unlink $path or die "couldn't unlink '$path': $!";      else {
                 }          print "No longer seeding [$hash]\n";
         }          $torrent->stop or warn $torrent->error_string;
           $client->remove( $torrent->id ) or warn $client->error;
       }
 }  }
   
   
 #print Dump \%keep;  #print Dump \%keep;
 foreach my $name (keys %keep) {  foreach my $hash ( keys %keep ) {
         foreach my $hash (keys %{ $keep{$name} }) {      my $file = $keep{$hash}{file} || q{};
                 my $file = $keep{$name}{$hash}{file};      my $dir  = $keep{$hash}{dir}  || q{};
                 my $dir  = $keep{$name}{$hash}{dir };      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{$hash}{name};
                         my $epoch = $keep{$name}{$hash}{epoch};          my $epoch = $keep{$hash}{epoch};
           $dir = $OBT->{DIR_TORRENT};
   
                         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", $OBT->{DIR_TORRENT} . "/" . $m_file )
                                         "$m_dir/$m_file",                  or die "Couldn't rename '$m_file': $!";
                                         $OBT->{DIR_TORRENT} . "/" . $m_file          }
                                 ) or die "Couldn't rename '$m_file': $!";      }
                         }  
                 }      if ( !$seeding{$hash} ) {
         }          print "Starting seed of [$file] [$hash]\n";
           if (!$client->add(
                   filename     => "$dir/$file",
                   download_dir => $OBT->{DIR_FTP},
               )
               )
           {
   
               #warn $client->error, ": $dir/$file\n";
               print "Removing invalid torrent\n\t", $keep{$hash}{path}, "\n";
               push @delete, $keep{$hash};
               delete $files{torrent}{ $keep{$hash}{name} }
                   { $keep{$hash}{epoch} };
           }
       }
 }  }
   
 sub Process_Dir  foreach (@delete) {
 {      if ( $_->{path} ) {
         my $basedir = shift;          print "Deleting '$_->{path}'\n";
           unlink $_->{path} or die "Couldn't unlink $_->{path}";
       }
       else {
           use Data::Dumper;
           print Dumper $_;
       }
   }
   
         my ($dirs, $files) = Get_Files_and_Dirs($basedir);  foreach my $name ( keys %{ $files{ $OBT->{META_EXT} } } ) {
         if (@$files) {      foreach my $epoch ( keys %{ $files{ $OBT->{META_EXT} }{$name} } ) {
                 my $dir = $basedir;          unless ( exists $files{torrent}{$name}{$epoch} ) {
                 $dir =~ s/^$OBT->{DIR_FTP}\///;              my $path = $files{ $OBT->{META_EXT} }{$name}{$epoch}{path};
                 my $torrent = Name_Torrent($dir);              print "Unlinking '$path'\n";
                 $torrent =~ s/-.*$//;              unlink $path or die "couldn't unlink '$path': $!";
                 $Possible_Torrents{$torrent} = 1;  
         }          }
       }
   }
   
         foreach my $subdir (@$dirs) {  $client->start;
                 next if $subdir eq '.';  
                 next if $subdir eq '..';  sub Process_Dir {
                 Process_Dir("$basedir/$subdir")      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 $file (@$files) {
               if ( $file =~ /$INSTALL_ISO_REGEX/ ) {
                   $torrent = Name_Torrent("$dir/$file");
                   $torrent =~ s/-.*$//;
                   $Possible_Torrents{$torrent} = 1;
               }
         }          }
       }
   
       foreach my $subdir (@$dirs) {
           next if $subdir eq '.';
           next if $subdir eq '..';
           Process_Dir("$basedir/$subdir");
       }
 }  }
   

Legend:
Removed from v.1.16  
changed lines
  Added in v.1.30

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>