===================================================================
RCS file: /cvs/openbsd/OpenBSDTorrents/MakeTorrents.pl,v
retrieving revision 1.3
retrieving revision 1.6
diff -u -r1.3 -r1.6
--- openbsd/OpenBSDTorrents/MakeTorrents.pl	2005/03/22 23:07:23	1.3
+++ openbsd/OpenBSDTorrents/MakeTorrents.pl	2005/03/23 00:14:48	1.6
@@ -1,26 +1,21 @@
 #!/usr/bin/perl -T
-#$Id: MakeTorrents.pl,v 1.3 2005/03/22 23:07:23 andrew Exp $
+#$Id: MakeTorrents.pl,v 1.6 2005/03/23 00:14:48 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;
 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#/$##;
 
@@ -34,101 +29,98 @@
 
 	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 ($#files < $MinFiles) {
-		print "Too few files in $basedir, skipping . . .\n";
-		return undef;
-	}
+        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'";
-	}
+        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'";
-		}
-	}
+        foreach (@$files) {
+                if (/^([^\/]+)$/) {
+                        $_ = "$basedir/$1";
+                } else {
+                        die "Invalid characters in file '$_' in '$basedir'";
+                }
+        }
 
-	my $date = Torrent_Date();
+        my $torrent = Name_Torrent($basedir);
 
-	my $torrent = $basedir;
-	$torrent =~ s/\W/_/g;
-	$torrent .= '-' . $date;
-	$torrent .= '.torrent';
+        print "Creating $torrent\n";
 
-	#print Dump $torrent, $basedir, $files;
-	print "Creating $torrent\n";
+        my $comment = "Files from $basedir\n" .
+                      "Created by andrew fresh (andrew\@mad-techies.org)\n" .
+                      "http://OpenBSD.somedomain.net/";
 
-	my $comment = "Files from $basedir\n" .
-	              "Created by andrew fresh (andrew\@mad-techies.org)\n" . 
-	              "http://OpenBSD.somedomain.net/",
+	btmake($torrent, $comment, $files);
 
-	system($BTMake, 
-	       '-C',
-	       '-c', $comment,
-	       '-n', $BaseName,
-	       '-o', "$OutDir/$torrent",
-	       '-a', $Tracker,
-	       @$files
-	);# || die "Couldn't system $BTMake $torrent: $!";
+#        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;
+# Stole and modified from btmake to work for this.
+sub btmake {
+    no locale;
 
-	my %dirs; # lookup table
-	my @files;# answer
+    my $torrent = shift;
+    my $comment = shift;
+    my $files = shift;
 
-	# build lookup table
-	@dirs{@dirs} = ();
+    my $name = $BaseName;
+    my $announce = $Tracker;
+    my $piece_len = 2 << ($Piece_Length - 1);
 
-	foreach my $item (@contents) {
-    		push(@files, $item) unless exists $dirs{$item};
-	}
+    $torrent = "$TorrentDir/$torrent";
 
-	@dirs  = grep { ! /$SkipDirs/  } @dirs  if $SkipDirs;
-	@files = grep { ! /$SkipFiles/ } @files if $SkipFiles;
-
-	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);
+    $t->save("$torrent");
+    print "Created: $torrent\n";
+    #system("btinfo $torrent") if ($::opt_I);
 }
 
-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);
-}