===================================================================
RCS file: /cvs/openbsd/OpenBSDTorrents/MakeTorrents.pl,v
retrieving revision 1.6
retrieving revision 1.25
diff -u -r1.6 -r1.25
--- openbsd/OpenBSDTorrents/MakeTorrents.pl	2005/03/23 00:14:48	1.6
+++ openbsd/OpenBSDTorrents/MakeTorrents.pl	2010/03/22 21:15:06	1.25
@@ -1,43 +1,47 @@
-#!/usr/bin/perl -T
-#$Id: MakeTorrents.pl,v 1.6 2005/03/23 00:14:48 andrew Exp $
+#!/usr/bin/perl
+# -T
+#$RedRiver: MakeTorrents.pl,v 1.24 2010/03/08 20:19:37 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);
+		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")
 	}
 }
@@ -47,45 +51,71 @@
         my $basedir = shift;
         my $files   = shift;
 
-        if ($#{ $files } < $MinFiles) {
-                print "Too few files in $basedir, skipping . . .\n";
-                return undef;
-        }
-
         if ($basedir !~ /\.\./ && $basedir =~ /^([\w\/\.-]+)$/) {
                 $basedir = $1;
         } else {
                 die "Invalid characters in dir '$basedir'";
         }
 
-        foreach (@$files) {
-                if (/^([^\/]+)$/) {
-                        $_ = "$basedir/$1";
-                } else {
-                        die "Invalid characters in file '$_' in '$basedir'";
+        if ($#{ $files } < $OBT->{MIN_FILES}) {
+                print "Too few files in $basedir, skipping . . .\n";
+                return undef;
+        }
+
+	my $torrent = Name_Torrent($basedir);
+	my $comment = "Files from $basedir";
+
+	my %torrents;
+        foreach my $file (@$files) {
+		if ($file =~ /^([^\/]+)$/) {
+			$file = $1;
+
+			my $t = $torrent;
+			my $c = $comment;
+
+			if ($file =~ /$INSTALL_ISO_REGEX/xms) {
+				$t = Name_Torrent("$basedir/$file");
+				$c = "$basedir/$file";
+			}
+			elsif (my ($ext) = $file =~ /$SONG_REGEX/xms) {
+				$t = Name_Torrent("$basedir/$ext");
+				$c = "$ext files from $basedir";
+			}
+
+			$torrents{$t}{comment} = $c;
+                       	push @{ $torrents{$t}{files} }, "$basedir/$file";
+                } 
+		else {
+                        die "Invalid characters in file '$file' in '$basedir'";
                 }
         }
 
-        my $torrent = Name_Torrent($basedir);
 
-        print "Creating $torrent\n";
+	foreach my $t (keys %torrents) {
 
-        my $comment = "Files from $basedir\n" .
-                      "Created by andrew fresh (andrew\@mad-techies.org)\n" .
-                      "http://OpenBSD.somedomain.net/";
+        	print "Creating $t (" 
+			. (scalar @{ $torrents{$t}{files} }) . " files)\n";
 
-	btmake($torrent, $comment, $files);
+		my $c = $torrents{$t}{comment};
+		$c .= "\nCreated by andrew fresh (andrew\@afresh1.com)\n" .
+                      	"http://OpenBSD.somedomain.net/";
 
+		eval { btmake($t, $c, $torrents{$t}{files}); };
+		if ($@) {
+			print "Error creating $t\n$@\n";
+		}
+
 #        system($BTMake,
 #               '-C',
 #               '-c', $comment,
-#               '-n', $BaseName,
-#               '-o', "$TorrentDir/$torrent",
+#               '-n', $OBT->{BASENAME},
+#               '-o', $OBT->{DIR_TORRENT} . "/$t",
 #               '-a', $Tracker,
 #               @$files
-#        );# || die "Couldn't system $BTMake $torrent: $!";
+#        );# || die "Couldn't system $BTMake $t: $!";
+	}
 
-        return $torrent;
+        return [ keys %torrents ];
 }
 
 
@@ -97,13 +127,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) {
@@ -117,10 +156,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";
 }