=================================================================== RCS file: /cvs/openbsd/OpenBSDTorrents/MakeTorrents.pl,v retrieving revision 1.4 retrieving revision 1.20 diff -u -r1.4 -r1.20 --- openbsd/OpenBSDTorrents/MakeTorrents.pl 2005/03/22 23:45:10 1.4 +++ openbsd/OpenBSDTorrents/MakeTorrents.pl 2006/07/24 19:03:53 1.20 @@ -1,37 +1,44 @@ -#!/usr/bin/perl -T -#$Id: MakeTorrents.pl,v 1.4 2005/03/22 23:45:10 andrew Exp $ +#!/usr/bin/perl +# -T +#$RedRiver: MakeTorrents.pl,v 1.19 2006/05/15 18:47:04 andrew Exp $ use strict; use warnings; use diagnostics; use lib 'lib'; +use BT::MetaInfo::Cached; use OpenBSDTorrents; %ENV = (); -use YAML; +chdir($OBT->{DIR_FTP}) || die "Couldn't change dir to " . $OBT->{DIR_FTP} . ": $!"; -my $BTMake = '/usr/local/bin/btmake'; -my $MinFiles = 5; +my $StartDir = ''; +if (@ARGV) { + foreach (@ARGV) { + s#/$##; + Process_Dir($_); + } +} else { + $StartDir = $OBT->{BASENAME}; + Process_Dir($StartDir); +} -my $StartDir = shift || $BaseName; -$StartDir =~ s#/$##; -chdir($BaseDir) || die "Couldn't change dir to $BaseDir"; -Process_Dir($StartDir); - sub Process_Dir { my $basedir = shift; + #return undef if $basedir =~ /packages/; + my ($dirs, $files) = Get_Files_and_Dirs($basedir); if (@$files) { my $torrent = Make_Torrent($basedir, $files); } - # don't recurse if we were called on a specific directory - return 1 if $StartDir ne $BaseName; + # don't recurse if we were started with a specific directory + return 1 if $StartDir ne $OBT->{BASENAME}; foreach my $subdir (@$dirs) { next if $subdir eq '.'; @@ -45,7 +52,7 @@ my $basedir = shift; my $files = shift; - if ($#{ $files } < $MinFiles) { + if ($#{ $files } < $OBT->{MIN_FILES}) { print "Too few files in $basedir, skipping . . .\n"; return undef; } @@ -72,15 +79,69 @@ "Created by andrew fresh (andrew\@mad-techies.org)\n" . "http://OpenBSD.somedomain.net/"; - system($BTMake, - '-C', - '-c', $comment, - '-n', $BaseName, - '-o', "$TorrentDir/$torrent", - '-a', $Tracker, - @$files - );# || die "Couldn't system $BTMake $torrent: $!"; + eval { btmake($torrent, $comment, $files); }; + if ($@) { + print "Error creating $torrent\n$@\n"; + } +# system($BTMake, +# '-C', +# '-c', $comment, +# '-n', $OBT->{BASENAME}, +# '-o', $OBT->{DIR_TORRENT} . "/$torrent", +# '-a', $Tracker, +# @$files +# );# || die "Couldn't system $BTMake $torrent: $!"; + return $torrent; +} + + +# Stole and modified from btmake to work for this. +sub btmake { + no locale; + + my $torrent = shift; + my $comment = shift; + my $files = shift; + + my $name = $OBT->{BASENAME}; + my $announce = $OBT->{URL_TRACKER}; + my $piece_len = 2 << ($OBT->{PIECE_LENGTH} - 1); + + my $torrent_with_path = $OBT->{DIR_NEW_TORRENT} . "/$torrent"; + + my $t = BT::MetaInfo::Cached->new( + { + cache_root => '/tmp/OBTFileCache' + } + ); + + $t->name($name); + $t->announce($announce); + unless ($announce =~ m!^http://[^/]+/!i) { + warn " [ WARNING: announce URL does not look like: http://hostname/ ]\n"; + } + $t->comment($comment); + #foreach my $pair (split(/;/, $::opt_f)) { + # if (my($key, $val) = split(/,/, $pair, 2)) { + # $t->set($key, $val); + # } + #} + $t->piece_length($piece_len); + $t->creation_date(time); + print "Checksumming files. This may take a little while...\n"; + $t->set_files(@$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"; }