version 1.3, 2005/03/22 23:07:23 |
version 1.24, 2010/03/08 20:19:37 |
|
|
#!/usr/bin/perl -T |
#!/usr/bin/perl |
#$Id$ |
# -T |
|
#$RedRiver: MakeTorrents.pl,v 1.23 2010/02/25 17:54:13 andrew Exp $ |
use strict; |
use strict; |
use warnings; |
use warnings; |
use diagnostics; |
use diagnostics; |
|
|
|
use lib 'lib'; |
|
use BT::MetaInfo::Cached; |
|
use OpenBSDTorrents; |
|
|
%ENV = (); |
%ENV = (); |
|
|
use YAML; |
chdir($OBT->{DIR_FTP}) || die "Couldn't change dir to " . $OBT->{DIR_FTP} . ": $!"; |
|
|
my $BaseDir = '/home/ftp/pub'; |
my $StartDir = ''; |
my $BaseName = 'OpenBSD'; |
if (@ARGV) { |
my $OutDir = '/home/andrew/torrents'; |
foreach (@ARGV) { |
my $BTMake = '/usr/local/bin/btmake'; |
s#/$##; |
my $Tracker = 'http://OpenBSD.somedomain.net/announce.php'; |
Process_Dir($_); |
|
} |
|
} else { |
|
$StartDir = $OBT->{BASENAME}; |
|
Process_Dir($StartDir); |
|
} |
|
|
my $MinFiles = 5; |
|
|
|
# These are regexes that tell what files to skip; |
|
my $SkipDirs = qr/\/patches$/; |
|
my $SkipFiles = qr/^\./; |
|
|
|
|
|
my $StartDir = shift || $BaseName; |
|
$StartDir =~ s#/$##; |
|
|
|
chdir($BaseDir) || die "Couldn't change dir to $BaseDir"; |
|
|
|
Process_Dir($StartDir); |
|
|
|
sub Process_Dir |
sub Process_Dir |
{ |
{ |
my $basedir = shift; |
my $basedir = shift; |
|
|
|
#return undef if $basedir =~ /packages/; |
|
|
my ($dirs, $files) = Get_Files_and_Dirs($basedir); |
my ($dirs, $files) = Get_Files_and_Dirs($basedir); |
if (@$files) { |
if (@$files) { |
Make_Torrent($basedir, $files); |
my $torrent = Make_Torrent($basedir, $files); |
} |
} |
|
|
# don't recurse if we were called on a specific directory |
# don't recurse if we were started with a specific directory |
return 1 if $StartDir ne $BaseName; |
return 1 if $StartDir ne $OBT->{BASENAME}; |
|
|
foreach my $subdir (@$dirs) { |
foreach my $subdir (@$dirs) { |
#next if $subdir eq '.'; |
next if $subdir =~ /^\./; |
#next if $subdir eq '..'; |
|
Process_Dir("$basedir/$subdir") |
Process_Dir("$basedir/$subdir") |
} |
} |
} |
} |
|
|
sub Make_Torrent |
sub Make_Torrent |
{ |
{ |
my $basedir = shift; |
my $basedir = shift; |
my $files = shift; |
my $files = shift; |
|
my $ignore_too_few = shift || 0; |
|
|
if ($#files < $MinFiles) { |
my $torrent_base = $basedir; |
print "Too few files in $basedir, skipping . . .\n"; |
my $comment = "Files from $torrent_base\n"; |
return undef; |
if ($ignore_too_few && $files->[0] =~ /$INSTALL_ISO_REGEX/) { |
|
$torrent_base = "$basedir/$files->[0]"; |
|
$comment = "$torrent_base\n"; |
} |
} |
|
elsif ($#{ $files } < $OBT->{MIN_FILES}) { |
|
print "Too few files in $basedir, skipping . . .\n"; |
|
return undef; |
|
} |
|
|
if ($basedir !~ /\.\./ && $basedir =~ /^([\w\/\.-]+)$/) { |
if ($basedir !~ /\.\./ && $basedir =~ /^([\w\/\.-]+)$/) { |
$basedir = $1; |
$basedir = $1; |
} else { |
} else { |
die "Invalid characters in dir '$basedir'"; |
die "Invalid characters in dir '$basedir'"; |
} |
} |
|
|
foreach (@$files) { |
my @good_files; |
if (/^([^\/]+)$/) { |
foreach my $file (@$files) { |
$_ = "$basedir/$1"; |
if ((!$ignore_too_few) && $file =~ /$INSTALL_ISO_REGEX/) { |
} else { |
my @f = ($file); |
die "Invalid characters in file '$_' in '$basedir'"; |
#foreach my $f (@$files) { |
|
# if ($f =~ /INSTALL|MD5/) { |
|
# push @f, $f; |
|
# } |
|
#} |
|
Make_Torrent($basedir, \@f, 1) |
} |
} |
} |
elsif ($file =~ /^([^\/]+)$/) { |
|
push @good_files, "$basedir/$1"; |
|
} |
|
else { |
|
die "Invalid characters in file '$file' in '$basedir'"; |
|
} |
|
} |
|
|
my $date = Torrent_Date(); |
my $torrent = Name_Torrent($torrent_base); |
|
|
my $torrent = $basedir; |
print "Creating $torrent\n"; |
$torrent =~ s/\W/_/g; |
|
$torrent .= '-' . $date; |
|
$torrent .= '.torrent'; |
|
|
|
#print Dump $torrent, $basedir, $files; |
$comment .= "Created by andrew fresh (andrew\@afresh1.com)\n" . |
print "Creating $torrent\n"; |
"http://OpenBSD.somedomain.net/"; |
|
|
my $comment = "Files from $basedir\n" . |
eval { btmake($torrent, $comment, \@good_files); }; |
"Created by andrew fresh (andrew\@mad-techies.org)\n" . |
if ($@) { |
"http://OpenBSD.somedomain.net/", |
print "Error creating $torrent\n$@\n"; |
|
} |
|
|
system($BTMake, |
# system($BTMake, |
'-C', |
# '-C', |
'-c', $comment, |
# '-c', $comment, |
'-n', $BaseName, |
# '-n', $OBT->{BASENAME}, |
'-o', "$OutDir/$torrent", |
# '-o', $OBT->{DIR_TORRENT} . "/$torrent", |
'-a', $Tracker, |
# '-a', $Tracker, |
@$files |
# @$files |
);# || die "Couldn't system $BTMake $torrent: $!"; |
# );# || die "Couldn't system $BTMake $torrent: $!"; |
|
|
|
return $torrent; |
} |
} |
|
|
|
|
sub Get_Files_and_Dirs |
# Stole and modified from btmake to work for this. |
{ |
sub btmake { |
my $basedir = shift; |
no locale; |
opendir DIR, $basedir or die "Couldn't opendir $basedir: $!"; |
|
my @contents = grep { ! /^\.\.$/ } grep { ! /^\.$/ } readdir DIR; |
|
closedir DIR; |
|
my @dirs = grep { -d "$basedir/$_" } @contents; |
|
|
|
my %dirs; # lookup table |
my $torrent = shift; |
my @files;# answer |
my $comment = shift; |
|
my $files = shift; |
|
|
# build lookup table |
my $name = $OBT->{BASENAME}; |
@dirs{@dirs} = (); |
my $announce = $OBT->{URL_TRACKER}; |
|
my $piece_len = 2 << ($OBT->{PIECE_LENGTH} - 1); |
|
|
foreach my $item (@contents) { |
my $torrent_with_path = $OBT->{DIR_NEW_TORRENT} . "/$torrent"; |
push(@files, $item) unless exists $dirs{$item}; |
|
} |
|
|
|
@dirs = grep { ! /$SkipDirs/ } @dirs if $SkipDirs; |
#if (@$files == 1) { |
@files = grep { ! /$SkipFiles/ } @files if $SkipFiles; |
#$name = $files->[0]; |
|
#} |
|
|
return \@dirs, \@files; |
my $t = BT::MetaInfo::Cached->new( |
} |
{ |
|
cache_root => '/tmp/OBTFileCache' |
|
} |
|
); |
|
|
sub Torrent_Date |
$t->name($name); |
{ |
$t->announce($announce); |
my ($min, $hour, $mday, $mon, $year) = (gmtime)[1..5]; |
unless ($announce =~ m!^http://[^/]+/!i) { |
$mon++; |
warn " [ WARNING: announce URL does not look like: http://hostname/ ]\n"; |
$year += 1900; |
} |
foreach ($min, $hour, $mday, $mon) { |
$t->comment($comment); |
if (length $_ == 1) { |
#foreach my $pair (split(/;/, $::opt_f)) { |
$_ = '0' . $_; |
# if (my($key, $val) = split(/,/, $pair, 2)) { |
} |
# $t->set($key, $val); |
} |
# } |
return join '-', ($year, $mon, $mday, $hour . $min); |
#} |
|
$t->piece_length($piece_len); |
|
$t->creation_date(time); |
|
#print "Checksumming files. This may take a little while...\n"; |
|
|
|
# Can't use this, have to do this manually because |
|
# we need to have the multi-file type of torrent |
|
# even when we have only one file. |
|
#$t->set_files(@$files); |
|
|
|
my @file_list; |
|
foreach my $f (@$files) { |
|
my $l = (stat("$OBT->{DIR_FTP}/$f"))[7]; |
|
my @p = split /\//, $f; |
|
shift @p; |
|
push @file_list, { |
|
length => $l, |
|
path => \@p, |
|
} |
|
} |
|
$t->files(\@file_list); |
|
$t->make_pieces(@$files); |
|
|
|
if ($t->total_size < $OBT->{MIN_SIZE}) { |
|
print "Skipping smaller than minimum size\n"; |
|
return 0; |
|
} |
|
|
|
my $hash = $t->info_hash; |
|
$hash = unpack("H*", $hash); |
|
|
|
$t->save($torrent_with_path); |
|
print "Created: $torrent_with_path\n"; |
} |
} |
|
|