#!/usr/bin/perl

use strict;
use warnings;
use RPM2;
use Data::Dumper;

sub compare($$);

my $test_repo = '/releases/7/smedev';
my $releases = 2;

my $rpm_flags =  RPM2->vsf_nodsa;
my @repositories = qw( 
		/releases/7/smeos/i386/CentOS/RPMS/
		/releases/7/smeupdates/i386/RPMS/
		/releases/7/smeupdates-testing/i386/RPMS/
		/releases/7/smeextras/i386/RPMS/
		/builds/rpms/RPMS/i386/
		/builds/rpms/RPMS/i586/
		/builds/rpms/RPMS/i686/
		/builds/rpms/RPMS/noarch/
		);

my %repos;
foreach my $repo (@repositories) {
	opendir RPMS, $repo;
	foreach my $rpm (grep { m#\.rpm# } readdir RPMS)
	{
		my $header = RPM2->open_package("$repo/$rpm", $rpm_flags);
		if ($repo =~ m#^/releases/7/sme#) {
			push @{$repos{$header->tag("NAME").".".$header->tag("ARCH")}{releases}}, $header;
		} else {
			push @{$repos{$header->tag("NAME")."-".$header->tag("VERSION").".".$header->tag("ARCH")}{builds}}, $header;
		}
	}
	closedir RPMS;
}

my @testing = map { RPM2->open_package("$test_repo/i386/RPMS/$_", $rpm_flags) } grep { m#\.rpm# } readdir RPMS if opendir RPMS, "$test_repo/i386/RPMS";
closedir RPMS;

my %pushed;
foreach my $rpm (keys %repos) {
	next unless $repos{$rpm}{builds};

	my @builds = reverse sort compare @{$repos{$rpm}{builds}};
	my $cnt = $#builds > $releases-1 ? $releases-1 : $#builds;

	if ($repos{$builds[0]->tag("NAME").".".$builds[0]->tag("ARCH")}{releases}) {
		my @released = reverse sort compare @{$repos{$builds[0]->tag("NAME").".".$builds[0]->tag("ARCH")}{releases}};
		push @{$pushed{$builds[0]->tag("NAME").".".$builds[0]->tag("ARCH")}}, grep { $_ gt $released[0] } @builds[0..$cnt];
	} else {
		push @{$pushed{$builds[0]->tag("NAME").".".$builds[0]->tag("ARCH")}}, @builds[0..$cnt];
        }
}

my $changed = 0;
my @pushed;
foreach my $package (sort keys %pushed)
{
	my @push = reverse sort compare @{$pushed{$package}};
	my $cnt = $#push > $releases*2-1 ? $releases*2-1 : $#push;

	foreach my $rpm (@push[0..$cnt]) {
		push @pushed, $rpm;
		unless (grep { ! compare($_, $rpm) } @testing) {
			print "adding ".$rpm->filename."\n";
			system(qw(cp -a), $rpm->filename, "$test_repo/i386/RPMS/");
			$changed++;
		}
	}
}

foreach my $rpm (sort compare @testing)
{
	unless (grep { ! compare($_,$rpm) } @pushed) {
		print "removing ".$rpm->filename."\n";
		unlink $rpm->filename;
	}
}

if ($changed) {
	system(qw(/builds/bin/linksource -s), "$test_repo/i386/RPMS", '-d', "$test_repo/SRPMS");
	system(qw(createrepo), "$test_repo/i386");
	system(qw(repoview), "$test_repo/i386");
	system(qw(yum-arch), "$test_repo/i386");
}

sub compare($$) {
	my ($a, $b) = @_;
	return $a->tag("NAME") cmp $b->tag("NAME") || $a cmp $b;
}