=================================================================== RCS file: /cvs/openbsd/OpenBSDTorrents/lib/OpenBSDTorrents.pm,v retrieving revision 1.3 retrieving revision 1.12 diff -u -r1.3 -r1.12 --- openbsd/OpenBSDTorrents/lib/OpenBSDTorrents.pm 2005/03/23 01:32:30 1.3 +++ openbsd/OpenBSDTorrents/lib/OpenBSDTorrents.pm 2010/03/22 20:08:48 1.12 @@ -1,5 +1,5 @@ package OpenBSDTorrents; -#$Id: OpenBSDTorrents.pm,v 1.3 2005/03/23 01:32:30 andrew Exp $ +#$RedRiver: OpenBSDTorrents.pm,v 1.11 2010/03/03 21:15:20 andrew Exp $ use 5.008005; use strict; use warnings; @@ -11,23 +11,40 @@ our $VERSION = '0.01'; our @EXPORT = qw( - $BaseDir - $TorrentDir - $BaseName - $Tracker + $OBT + $INSTALL_ISO_REGEX + $SONGS_REGEX &Name_Torrent &Get_Files_and_Dirs + &justme ); - -our $BaseDir = '/home/ftp/pub'; -our $TorrentDir = '/home/andrew/torrents'; -our $BaseName = 'OpenBSD'; -our $Tracker = 'http://OpenBSD.somedomain.net/announce.php'; -# These are regexes that tell what files to skip; -our $SkipDirs = qr/\/patches$/; -our $SkipFiles = qr/^\./; +my $config_file = '/etc/OpenBSDTorrents.conf'; +our $OBT = Config(); +our $INSTALL_ISO_REGEX = qr/install\d+\.iso/xms; +our $SONGS_REGEX = qr/^(song.*\.([^\.]+))$/xms; +sub Config +{ + my %config; + open FILE, $config_file or die "Couldn't open FILE $config_file: $!"; + while () { + chomp; + s/#.*$//; + s/\s+$//; + next unless $_; + my ($name, $val) = split /=/, $_, 2; + $name =~ s/^OBT_//; + # This should really look for contents that are a + # bit safer, but I can't think of what would work here. + if ($val =~ /^(.*)$/) { + $config{$name} = $1; + $config{$name} =~ s/^['"]|["']$//gxms; + } + } + close FILE; + return \%config; +} sub Name_Torrent { @@ -47,23 +64,28 @@ { my $basedir = shift; opendir DIR, $basedir or die "Couldn't opendir $basedir: $!"; - my @contents = grep { ! /^\.\.$/ } grep { ! /^\.$/ } readdir DIR; + my @contents = sort grep { ! /^\.\.$/ } grep { ! /^\.$/ } readdir DIR; closedir DIR; - my @dirs = grep { -d "$basedir/$_" } @contents; - my %dirs; # lookup table - my @files;# answer - - # build lookup table - @dirs{@dirs} = (); - - foreach my $item (@contents) { - push(@files, $item) unless exists $dirs{$item}; + my @dirs; + my @files; + ITEM: foreach my $item (@contents) { + if ( -d "$basedir/$item" ) { + if ( $OBT->{SKIP_DIRS} + && $item =~ /$OBT->{SKIP_DIRS}/) { + next ITEM; + } + push @dirs, $item; + } + else { + if ( $OBT->{SKIP_FILES} + && $item =~ /$OBT->{SKIP_FILES}/) { + next ITEM; + } + push @files, $item; + } } - @dirs = grep { ! /$SkipDirs/ } @dirs if $SkipDirs; - @files = grep { ! /$SkipFiles/ } @files if $SkipFiles; - return \@dirs, \@files; } @@ -79,6 +101,40 @@ } return join '-', ($year, $mon, $mday, $hour . $min); } + +# "There can be only one." --the Highlander +sub justme { + + my $myname; + + if ($0 =~ m#([^/]+$)#) { + $myname = $1; + } else { + die "Couldn't figure out myname"; + } + + my $SEMA = $OBT->{DIR_HOME} . "/run/$myname.pid"; + if (open SEMA, "<", $SEMA) { + my $pid = ; + if (defined $pid) { + chomp $pid; + if ($pid =~ /^(\d+)$/) { + $pid = $1; + } else { + die "invalid pid read '$pid'"; + } + if (kill(0, $pid)) { + print "$0 already running (pid $pid), bailing out\n"; + exit 253; + } + } + close SEMA; + } + open (SEMA, ">", $SEMA) or die "can't write $SEMA: $!"; + print SEMA "$$\n"; + close(SEMA) or die "can't close $SEMA: $!"; +} + 1; __END__