#!/usr/bin/perl

use strict;
use warnings;
use RPM2;
use File::Copy;
use Getopt::Std;

my %opts;
getopts( "s:d:", \%opts );
my $src = $opts{s} || '/builds/RPMS';
my $dest = $opts{d} || '/builds/SRPMS';

my $rpm_flags =  RPM2->vsf_nodsa;
my @repositories = qw( 
	/mirrors/centos/4/fasttrack/SRPMS
	/mirrors/centos/4/updates/SRPMS
	/mirrors/centos/4/os/SRPMS
	/mirrors/centos/4/extras/SRPMS
	/mirrors/rpmforge/dag/source/
	/mirrors/atrpms/src/el4-i386/atrpms/stable/
	/mirrors/fedora-legacy/fedora/3/updates/SRPMS/
	/mirrors/fedora-legacy/fedora/3/os/SRPMS/
	/builds/rpms/SRPMS/
	);

opendir RPMS, "$src" or die "Couldn't opendir $src";
my @rpms = map { RPM2->open_package("$src/$_", $rpm_flags) } grep { /\.rpm$/ } readdir RPMS;
closedir RPMS;

opendir SRPMS, "$dest" or die "Couldn't opendir $dest";
my %srpms = map { $_ => 0 } grep { /\.rpm$/ } readdir SRPMS;
closedir SRPMS;

my %repos;
foreach my $repo (reverse @repositories)
{
    my %temp_repo = map { $_ => $repo } grep { /\.rpm$/ } readdir RPMS if ( opendir RPMS, $repo );
    closedir RPMS;

    %repos = (%repos, %temp_repo);
}

foreach my $rpm ( sort @rpms )
{
    my (@headers, $header);

    my $rpmfix = $rpm;
    $rpmfix =~ s@\+@\\+@g;

    my @sources;
    my $src = $rpm->tag('SOURCERPM');
    push @sources, $src;
    push @sources, $src if ($src =~ s/\.2\.el4\.rf\./\.rf\./);
    push @sources, $src if ($src =~ s/\.el4\.rf\./\.rf\./);
    push @sources, $src if ($src =~ s/\.el4\.at\./\.at\./);
    push @sources, $src if ($src =~ s/\.rf\./\.dag\./);

    my $found = 0;
    foreach my $source ( @sources )
    {
        if( -f "$dest/$source" )
        {
            $found = 1;
	    $srpms{$source}++;
            last;
        } else {
            if ($repos{$source})
            {
                my $fromfile = $repos{$source} . "/$source";
                $fromfile =~ s#//#/#g;
		print "copying $source\n";
                system(qw(cp --preserve=timestamps), $fromfile, "$dest/");
                $found = 1;
		$srpms{$source}++;
                last;
            }

        }
    }
    print "can't find SRPM ($sources[0]) for ", $rpm->as_nvre(), "\n" unless $found;
}

foreach my $srpm ( grep { $srpms{$_} == 0 } keys %srpms ) {
    print "removing unused SRPM ($srpm)\n";
    unlink "$dest/$srpm";
}