===================================================================
RCS file: /cvs/openbsd/OpenBSDTorrents/MakeTorrents.pl,v
retrieving revision 1.7
retrieving revision 1.22
diff -u -r1.7 -r1.22
--- openbsd/OpenBSDTorrents/MakeTorrents.pl	2005/03/24 00:21:43	1.7
+++ openbsd/OpenBSDTorrents/MakeTorrents.pl	2007/11/02 03:36:01	1.22
@@ -1,43 +1,47 @@
-#!/usr/bin/perl -T
-#$Id: MakeTorrents.pl,v 1.7 2005/03/24 00:21:43 andrew Exp $
+#!/usr/bin/perl
+# -T
+#$RedRiver: MakeTorrents.pl,v 1.21 2006/11/06 23:17:59 andrew Exp $
 use strict;
 use warnings;
 use diagnostics;
 
-use BT::MetaInfo;
-
 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 $Piece_Length = 18;
-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 '.';
-		next if $subdir eq '..';
+		next if $subdir =~ /^\./;
 		Process_Dir("$basedir/$subdir")
 	}
 }
@@ -46,8 +50,15 @@
 {
         my $basedir = shift;
         my $files   = shift;
+        my $ignore_too_few = shift || 0;
 
-        if ($#{ $files } < $MinFiles) {
+	my $torrent_base = $basedir;
+        my $comment = "Files from $torrent_base\n";
+	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;
         }
@@ -58,32 +69,42 @@
                 die "Invalid characters in dir '$basedir'";
         }
 
-        foreach (@$files) {
-                if (/^([^\/]+)$/) {
-                        $_ = "$basedir/$1";
-                } else {
-                        die "Invalid characters in file '$_' in '$basedir'";
+	my @good_files;
+        foreach my $file (@$files) {
+		if ((!$ignore_too_few) && $file =~ /$INSTALL_ISO_REGEX/) {
+			my @f = ($file);
+			#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 $torrent = Name_Torrent($basedir);
+        my $torrent = Name_Torrent($torrent_base);
 
         print "Creating $torrent\n";
 
-        my $comment = "Files from $basedir\n" .
-                      "Created by andrew fresh (andrew\@mad-techies.org)\n" .
+	$comment .= "Created by andrew fresh (andrew\@mad-techies.org)\n" .
                       "http://OpenBSD.somedomain.net/";
 
-	eval { btmake($torrent, $comment, $files); };
+	eval { btmake($torrent, $comment, \@good_files); };
 	if ($@) {
-		print "Error creating $torrent\n";
+		print "Error creating $torrent\n$@\n";
 	}
 
 #        system($BTMake,
 #               '-C',
 #               '-c', $comment,
-#               '-n', $BaseName,
-#               '-o', "$TorrentDir/$torrent",
+#               '-n', $OBT->{BASENAME},
+#               '-o', $OBT->{DIR_TORRENT} . "/$torrent",
 #               '-a', $Tracker,
 #               @$files
 #        );# || die "Couldn't system $BTMake $torrent: $!";
@@ -100,13 +121,22 @@
     my $comment = shift;
     my $files = shift;
 
-    my $name = $BaseName;
-    my $announce = $Tracker;
-    my $piece_len = 2 << ($Piece_Length - 1);
+    my $name = $OBT->{BASENAME};
+    my $announce = $OBT->{URL_TRACKER};
+    my $piece_len = 2 << ($OBT->{PIECE_LENGTH} - 1);
 
-    $torrent = "$TorrentDir/$torrent";
+    my $torrent_with_path = $OBT->{DIR_NEW_TORRENT} . "/$torrent";
 
-    my $t = BT::MetaInfo->new();
+    #if (@$files == 1) {
+        #$name = $files->[0];
+    #}
+
+    my $t = BT::MetaInfo::Cached->new(
+        {
+		cache_root => '/tmp/OBTFileCache'
+        }
+    );
+
     $t->name($name);
     $t->announce($announce);
     unless ($announce =~ m!^http://[^/]+/!i) {
@@ -120,10 +150,35 @@
     #}
     $t->piece_length($piece_len);
     $t->creation_date(time);
-    warn "Checksumming files. This may take a little while...\n";
-    $t->set_files(@$files);
-    $t->save("$torrent");
-    print "Created: $torrent\n";
-    #system("btinfo $torrent") if ($::opt_I);
+    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";
 }