===================================================================
RCS file: /cvs/openbsd/OpenBSDTorrents/MakeTorrents.pl,v
retrieving revision 1.2
retrieving revision 1.8
diff -u -r1.2 -r1.8
--- openbsd/OpenBSDTorrents/MakeTorrents.pl	2005/03/22 04:19:03	1.2
+++ openbsd/OpenBSDTorrents/MakeTorrents.pl	2005/03/24 19:49:09	1.8
@@ -1,22 +1,22 @@
 #!/usr/bin/perl -T
-#$Id: MakeTorrents.pl,v 1.2 2005/03/22 04:19:03 andrew Exp $
+#$Id: MakeTorrents.pl,v 1.8 2005/03/24 19:49:09 andrew Exp $
 use strict;
 use warnings;
 use diagnostics;
 
+use BT::MetaInfo;
+
+use lib 'lib';
+use OpenBSDTorrents;
+
 %ENV = ();
 
 use YAML;
 
-my $BaseDir  = '/home/ftp/pub';
-my $BaseName = 'OpenBSD';
-my $OutDir   = '/home/andrew/torrents';
-my $BTMake   = '/usr/local/bin/btmake';
-my $Tracker  = 'http://OpenBSD.somedomain.net/announce.php';
+my $Piece_Length = 18;
 
-# These are regexes that tell what files to skip;
-my $SkipDirs;
-my $SkipFiles = qr/^\./;
+my $MinFiles = 5;
+my $MinSize  = 50 * 1024 * 1024; # 50 MiB
 
 my $StartDir = shift || $BaseName;
 $StartDir =~ s#/$##;
@@ -31,86 +31,107 @@
 
 	my ($dirs, $files) = Get_Files_and_Dirs($basedir);
 	if (@$files) {
-		Make_Torrent($basedir, $files);
+		my $torrent = Make_Torrent($basedir, $files);
 	}
+
+	# don't recurse if we were called on a specific directory
+	return 1 if $StartDir ne $BaseName;
+
 	foreach my $subdir (@$dirs) {
-		#next if $subdir eq '.';
-		#next if $subdir eq '..';
+		next if $subdir eq '.';
+		next if $subdir eq '..';
 		Process_Dir("$basedir/$subdir")
 	}
 }
 
 sub Make_Torrent
 {
-	my $basedir = shift;
-	my $files   = shift;
+        my $basedir = shift;
+        my $files   = shift;
 
-	if ($basedir =~ /^([\w\/\.-]+)$/) {
-		$basedir = $1;
-	} else {
-		die "Invalid characters in dir '$basedir'";
-	}
+        if ($#{ $files } < $MinFiles) {
+                print "Too few files in $basedir, skipping . . .\n";
+                return undef;
+        }
 
-	foreach (@$files) {
-		if (/^([^\/]+)$/) {
-			$_ = "$basedir/$1";
-		} else {
-			die "Invalid characters in file '$_' in '$basedir'";
-		}
-	}
+        if ($basedir !~ /\.\./ && $basedir =~ /^([\w\/\.-]+)$/) {
+                $basedir = $1;
+        } else {
+                die "Invalid characters in dir '$basedir'";
+        }
 
-	my $torrent = $basedir;
-	$torrent =~ s/\W/_/g;
-	$torrent .= '-' . Torrent_Date();
-	$torrent .= '.torrent';
+        foreach (@$files) {
+                if (/^([^\/]+)$/) {
+                        $_ = "$basedir/$1";
+                } else {
+                        die "Invalid characters in file '$_' in '$basedir'";
+                }
+        }
 
-	print Dump $torrent, $basedir, $files;
-	print "Creating $torrent\n";
+        my $torrent = Name_Torrent($basedir);
 
-	system($BTMake, 
-	       '-C',
-	       '-c', "Created by andrew fresh <andrew\@mad-techies.org>\n" . 
-	             "See http://OpenBSD.somedomain.net/",
-	       '-n', $BaseName,
-	       '-o', "$OutDir/$torrent",
-	       '-a', $Tracker,
-	       @$files
-	);# || die "Couldn't system $BTMake $torrent: $!";
+        print "Creating $torrent\n";
+
+        my $comment = "Files from $basedir\n" .
+                      "Created by andrew fresh (andrew\@mad-techies.org)\n" .
+                      "http://OpenBSD.somedomain.net/";
+
+	eval { btmake($torrent, $comment, $files); };
+	if ($@) {
+		print "Error creating $torrent\n";
+	}
+
+#        system($BTMake,
+#               '-C',
+#               '-c', $comment,
+#               '-n', $BaseName,
+#               '-o', "$TorrentDir/$torrent",
+#               '-a', $Tracker,
+#               @$files
+#        );# || die "Couldn't system $BTMake $torrent: $!";
+
+        return $torrent;
 }
 
-sub Get_Files_and_Dirs
-{
-	my $basedir = shift;
-	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 @files;# answer
+# Stole and modified from btmake to work for this.
+sub btmake {
+    no locale;
 
-	# build lookup table
-	@dirs{@dirs} = ();
+    my $torrent = shift;
+    my $comment = shift;
+    my $files = shift;
 
-	foreach my $item (@contents) {
-    		push(@files, $item) unless exists $dirs{$item};
-	}
+    my $name = $BaseName;
+    my $announce = $Tracker;
+    my $piece_len = 2 << ($Piece_Length - 1);
 
-	@dirs  = grep { ! /$SkipDirs/  } @dirs  if $SkipDirs;
-	@files = grep { ! /$SkipFiles/ } @files if $SkipFiles;
+    $torrent = "$TorrentDir/$torrent";
 
-	return \@dirs, \@files;
-}
+    my $t = BT::MetaInfo->new();
+    $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);
+    warn "Checksumming files. This may take a little while...\n";
+    $t->set_files(@$files);
 
-sub Torrent_Date
-{
-	my ($min, $hour, $mday, $mon, $year) = (gmtime)[1..5];
-	$mon++;
-	$year += 1900;
-	foreach ($min, $hour, $mday, $mon) {
-		if (length $_ == 1) {
-			$_ = '0' . $_;
-		}
-	}
-	return join '-', ($year, $mon, $mday, $hour . $min);
+    if ($t->total_size < $MinSize) {
+        print "Skipping smaller than minimum size\n";
+        return 0;
+    }
+
+    $t->save("$torrent");
+    print "Created: $torrent\n";
+    #system("btinfo $torrent") if ($::opt_I);
 }
+