#!/usr/bin/perl

use strict;
use warnings;
use RPM2;
use File::Find;
use File::Basename;
use Getopt::Std;
use Term::ANSIColor;
use Data::Dumper;

delete $ENV{PATH};
my $HOME=$ENV{HOME};
umask 002;

my $rpm_flags = RPM2->vsf_nosha1header | RPM2->vsf_nomd5header | RPM2->vsf_nodsaheader | RPM2->vsf_norsaheader;
$rpm_flags |= RPM2->vsf_nosha1 | RPM2->vsf_nomd5 | RPM2->vsf_nodsa | RPM2->vsf_norsa;

$Term::ANSIColor::AUTORESET = 1;
$| = 1;

my @archs = ( 'i386', 'x86_64' );

my $repos = { 'smeos'              => { prio => 18, inc => 1, ver => 1, rel => 1, os => 1 },
	      'smeupdates'         => { prio => 17, inc => 1, ver => 1, rel => 1 },
	      'smeupdates-testing' => { prio => 16, inc => 1, ver => 1, rel => 1 },
	      'smeextras'          => { prio => 15, inc => 0, ver => 1, rel => 1 },
	      'smeaddons'          => { prio => 14, inc => 0, ver => 1, rel => 1 },
	      'smecontribs'        => { prio => 13, inc => 0, ver => 1, rel => 1, contribs => 1 },
	      'smetest'            => { prio => 12, inc => 1, ver => 2, rel => 2, devel => 2 },
	      'smedev'             => { prio => 11, inc => 0, ver => 2, rel => 2, devel => 1 },
	      'centos'             => { prio => 10, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'fedora/epel'        => { prio =>  9, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'fedora/extras'      => { prio =>  8, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'rpmforge'           => { prio =>  7, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'atrpms'             => { prio =>  6, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'fedora/core'        => { prio =>  5, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'base'               => { prio =>  4, inc => 0, ver => 0, rel => 0, orig => 1, base => 1 },
	      'builds'             => { prio =>  3, inc => 0, ver => 0, rel => 0, orig => 1 },
	      'contribs'           => { prio =>  2, inc => 0, ver => 0, rel => 0, orig => 1 },
	      'stage'              => { prio =>  1, inc => 0, ver => 0, rel => 0, stage => 1, os => 1 },
	    };

my $distrepo = { '7' => { active   => 1,
			  centos   => 4,
			  fedora   => 3,
			  repo     => '/releases/7/',
			  os       => 'SME/RPMS/',
			  builds   => '/builds/smeserver-7-core/',
			  contribs => '/builds/smeserver-7-contribs/',
			  stage    => '/stage/7/',
			},
		 '8' => { active   => 1,
			  centos   => 5,
			  fedora   => 6,
			  repo     => '/releases/testing/8/',
			  os       => 'SME/',
			  builds   => '/builds/smeserver-8-core/',
			  contribs => '/builds/smeserver-8-contribs/',
			  stage    => '/stage/8/',
			},
	       };

my @baserepo = ( ( map { "/mirrors/centos/~C/$_/" } ('os','updates','fasttrack','extras') ),
		 ( map { "/mirrors/fedora/epel$_/~C/" } ('','/testing') ),
		 '/mirrors/fedora/extras/~F/',
		 '/mirrors/rpmforge/dag/redhat/el~C/en/~A/RPMS.dag/',
		 '/mirrors/rpmforge/dag/source/',
		 ( map { "/mirrors/atrpms$_/el~C-~A/atrpms/stable/" } ('','/src') ),
		 ( map { "/mirrors/atrpms$_/el~C-~A/atrpms/testing/" } ('','/src') ),
		 ( map {"/mirrors/fedora/core$_/~F/" } ('','/updates') ),
	       );

my $skippkg = { '7' => { 'centos'           => ['horde','imp-h3','ingo-h3','turba-h3','vim'],
			 'fedora/epel'      => ['perl-Razor-Agent'],
			 'rpmforge'         => ['perl-Test-Inline'],
			 'atrpms'           => ['dovecot','yum'],
		       },
		'8' => { 'centos'           => ['horde','imp-h3','ingo-h3','turba-h3'],
			 'fedora/epel'      => ['perl-Razor-Agent','smolt'],
			 'fedora/extras'    => ['dstat','gocr','horde','oidentd','perl-Test-Inline','perl-Razor-Agent','smolt'],
			 'rpmforge'         => ['perl-Test-Inline'],
			 'atrpms'           => [],
			 'fedora/core'      => [],
		       },
	      };

my %opts;
getopts( 'dtvr:', \%opts );
$opts{r} ||= '';

my ($devel1, $devel2) = sort { $repos->{$a}->{devel} <=> $repos->{$b}->{devel} } grep { $repos->{$_}->{devel} } keys %$repos;
$devel2 ||= $devel1;
my ($contribs) = sort { $repos->{$a}->{contribs} <=> $repos->{$b}->{contribs} } grep { $repos->{$_}->{contribs} } keys %$repos;
my ($stage) = sort { $repos->{$a}->{stage} <=> $repos->{$b}->{stage} } grep { $repos->{$_}->{stage} } keys %$repos;

my ($rel, @rpms, %latest, %sources, $archs, $rebuild, $newline);
foreach my $smever ( sort { $a <=> $b } grep { $distrepo->{$_}->{active} } keys %$distrepo ) {
    $rel = $smever;
    %latest = ();
    %sources = ();
    @rpms = ();
    $rebuild = ();

    # Load distro packages
    find( { wanted => \&loadpkg, no_chdir => 1, follow_fast => 1 },  map { $distrepo->{$smever}->{$_} } ('repo','builds','contribs','stage') );
    printline('white', 0, "Distro (SME Server $rel) packages loaded.\n");

    # Load base packages
    foreach my $base ( @baserepo ) {
	my $bdir = $base;
	$bdir =~ s/~C/$distrepo->{$smever}->{centos}/;
	$bdir =~ s/~F/$distrepo->{$smever}->{fedora}/;
	if ( $bdir =~ m{~A} ) {
	    for my $arch ( @archs ) {
		my $basearch = $bdir;
		$basearch =~ s/~A/$arch/;
		find( { wanted => \&loadbase, no_chdir => 1, follow_fast => 1 }, $basearch ) if -r $basearch;
	    }
	} else {
	    find( { wanted => \&loadbase, no_chdir => 1, follow_fast => 1 }, $bdir ) if -r $bdir;
	}
    }
    printline('white', 0, 'Base (Centos: ', $distrepo->{$smever}->{centos}, ', Fedora: ', $distrepo->{$smever}->{fedora}, ") packages loaded.\n");

    %latest = ();
    foreach my $name ( sort keys %{{ map { $_->{base} => 1 } @rpms }} ) {
	printline('white', 0, "Processing: $name");
	my @pkgs = sort pkgsrt grep { $_->{base} eq $name } @rpms;
	my %seen = ();
	my $count = ();
	$archs = ();

	foreach my $pkg ( @pkgs ) {
	    next if $pkg->{done} || $pkg->{obsolete} || $pkg->{rpm}->is_source_package || ! $repos->{$pkg->{repo}};
	    printline('bold black', 1, ' - ', $pkg->{rpm}->filename) if $opts{d};
	    my $orig = 0;

	    # Find names of packages in same or higher repos
	    my %names = names($pkg, @pkgs);

	    # Stage packages should never be most current
	    if ( rprop($pkg, 'stage') ) {
		printline('dark blue', 1, '    - checking stage') if $opts{d};
		tagpkg($pkg, 'obsolete');
		next;
	    }

	    # Move packages from base/builds to correct area
	    elsif ( rprop($pkg, 'orig') ) {
		printline('dark blue', 1, '    - checking base/builds/contribs') if $opts{d};
		my $repo = $names{$pkg->{name}} && $names{$pkg->{name}} > 1 ? $devel2 : $devel1;
		$repo = $contribs if $pkg->{repo} eq 'contribs';
		if ( $pkg->{nonbase} ) {
		    foreach my $up ( @pkgs ) {
			next if rprop($up, 'orig');
			next if $up->{name} ne $pkg->{name} || ! $up->{obsolete} || $up->{rpm} ge $pkg->{rpm};
			$repo = $up->{repo};
			last;
		    }
		} 
		next if rprop($pkg, 'base') && $repo eq $devel1;
		movepkg($pkg, $repo);
		%names = names($pkg, @pkgs);
		$seen{$pkg->{nvra}}++;
		$orig++;
	    }

	    # Check version/release counts
	    unless ( $pkg->{nonbase} ) {
		printline('dark blue', 1, '    - checking version/release counts') if $opts{d};
		$count->{$pkg->{repo}}->{$pkg->{version}}->{$pkg->{release}}++;
		if ( rprop($pkg, 'ver') && scalar keys %{$count->{$pkg->{repo}}} > rprop($pkg, 'ver') ) {
		    movepkg($_, 'delete') foreach grep { $_->{rpm} eq $pkg->{rpm} } @pkgs;
		    next;
		}
		if ( rprop($pkg, 'rel') && scalar keys %{$count->{$pkg->{repo}}->{$pkg->{version}}} > rprop($pkg, 'rel') ) {
		    movepkg($_, 'delete') foreach grep { $_->{rpm} eq $pkg->{rpm} } @pkgs;
		    next;
		}
	    }

	    foreach my $cmp ( grep { $_->{rpm}->is_source_package || ( ! $_->{done} && $_->{rpm} le $pkg->{rpm} && rprop($_, 'prio') <= rprop($pkg, 'prio') ) } @pkgs ) {
		printline('dark blue', 1, '    - ', $cmp->{rpm}->filename) if $opts{d};
		if ( $cmp->{rpm}->is_source_package ) {
		    if ( $cmp->{name} eq $pkg->{base} ) {
			if ( $cmp->{rpm} eq $pkg->{rpm} ) {
			    if ( $pkg->{repo} eq $cmp->{repo} || ! $pkg->{srpm} || $pkg->{srpm}->{rpm} ne $pkg->{rpm} ) {
				delete $pkg->{srpm}->{latest} if $pkg->{latest} && $pkg->{srpm} && $pkg->{srpm}->{latest};
				$pkg->{srpm} = $cmp;
				$pkg->{srpm}->{latest}++ if $pkg->{latest};
			    }
			    if ( $pkg->{latest} && rprop($cmp, 'stage') ) {
				tagpkg($cmp, 'inuse');
				$pkg->{srpm}->{latest}++;
			    }
			    if ( $pkg->{repo} eq $cmp->{repo} || ( $cmp->{oldrepo} && $pkg->{repo} eq $cmp->{oldrepo} ) || rprop($cmp, 'orig') ) {
				tagpkg($cmp, 'inuse')
			    }
			}
			next if $pkg->{srpm} && $pkg->{srpm}->{rpm} eq $pkg->{rpm};

			my $src = $pkg->{rpm}->sourcerpm;
			my $source = basename $cmp->{rpm}->filename;
			SRC: {
			    $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{\.el\d\.at(\.src)}{$1}                 && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{\.el\d(\.rf\.src)}{$1}                 && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{(\d+)\.\d+\.el\d(\.rf\.src)}{$1$2}     && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{\.el\d\.rf(\.src)}{.dag$1}             && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{(\d+)\.\d+\.el\d\.rf(\.src)}{$1.dag$2} && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{\.el\d(\.src)}{$1}                     && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{(-\d+)\.\d+\.el\d(\.src)}{$1$2}        && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{\.centos\d(\.src)}{$1}                 && $src eq $source && last SRC;
			    ($src = $pkg->{rpm}->sourcerpm) =~ s{\.el\d\.sme(\.src)}{$1}                && $src eq $source && last SRC;
			    $src = '';
			}
			if ( $src eq $source && ( ! $pkg->{srpm} || $pkg->{srpm}->{rpm} eq $cmp->{rpm} ) ) {
			    if ( ! $pkg->{srpm} || $pkg->{repo} eq $cmp->{repo} ) {
				delete $pkg->{srpm}->{latest} if $pkg->{latest} && $pkg->{srpm} && $pkg->{srpm}->{latest};
				$pkg->{srpm} = $cmp;
				$pkg->{srpm}->{latest}++ if $pkg->{latest};
			    }
			    if ( $pkg->{latest} && rprop($cmp, 'stage') ) {
				tagpkg($cmp, 'inuse');
				$pkg->{srpm}->{latest}++;
			    }
			    tagpkg($cmp, 'inuse') if $pkg->{repo} eq $cmp->{repo} || rprop($cmp, 'orig');
			}
		    }
		} elsif ( $cmp->{rpm} eq $pkg->{rpm} ) {
		    if ( $names{$cmp->{name}} ) {
			if ( $pkg->{repo} eq $cmp->{repo} ) {
			    tagpkg($cmp, 'correct', 1);
			    $seen{$cmp->{nvra}}++;
			} elsif ( $seen{$cmp->{nvra}} && rprop($cmp, 'stage') && $pkg->{latest} ) {
			    $_->{latest}++ foreach grep { $_->{nvra} eq $cmp->{nvra} && $_->{latest} && $_->{done} } @pkgs;
			    tagpkg($cmp, 'correct', 1);
			} elsif ( $seen{$cmp->{nvra}} ) {
			    $orig++ if rprop($cmp, 'orig');
			    movepkg($cmp, 'delete', 1);
			} else {
			    movepkg($cmp, $pkg->{repo}, 1, 1);
			    $seen{$cmp->{nvra}}++;
			}
		    } else {
			tagpkg($cmp, 'nonbase');
		    }
		} elsif ( rprop($cmp, 'prio') < rprop($pkg, 'prio') ) {
		    tagpkg($cmp, 'obsolete');
		}
	    }
	    printline('dark blue', 1, '    - checking source packages') if $opts{d};
	    if ( ! $pkg->{srpm} ) {
		printline('bold yellow on_red', 1, ' - ', $pkg->{rpm}->filename, ' (missing source ', $pkg->{rpm}->sourcerpm, ')');
	    } elsif ( $pkg->{srpm}->{repo} ne $pkg->{repo} && ( ! $pkg->{srpm}->{oldrepo} || $pkg->{srpm}->{oldrepo} ne $pkg->{repo} ) ) {
		movepkg($pkg->{srpm}, $pkg->{repo}, 0);
	    }
	    unless ( $orig ) {
		foreach ( grep { $_->{base} eq $pkg->{base} && $_->{repo} eq $pkg->{repo} } @pkgs ) {
		    printline('bold yellow on_red', 1, ' - ', $_->{rpm}->filename, ' (missing original)'); 
		}
	    }
	}
	printline('bold black', 1, ' - arch count check') if $opts{d};
	foreach my $r ( keys %$archs ) {
	    foreach my $p ( keys %{$archs->{$r}} ) {
		foreach my $a ( keys %{$archs->{$r}->{$p}} ) {
		    my $p2 = ${$archs->{$r}->{$p}->{$a}}[0];
		    if ( $a eq 'noarch' ) {
			movepkg($p2, $p2->{repo}, 0) if ! $p2->{repos} && scalar @{$archs->{$r}->{$p}->{$a}} != scalar @archs;
		    } elsif ( scalar @{$archs->{$r}->{$p}->{$a}} != 1 ) {
			printline('bold yellow on_red', 1, ' - ', $p2->{rpm}->filename, ' (many arch)');
		    }
		}
	    }
	}

	printline('bold black', 1, ' - missing stage packages') if $opts{d};
	movepkg($_, $stage, 0) foreach grep { $_->{latest} && $_->{latest} == 1 } @pkgs;

	printline('bold black', 1, ' - obsolete packages') if $opts{d};
	movepkg($_, 'delete') foreach grep { ! rprop($_, 'base') && ! $_->{done} && ( $_->{obsolete} || ( $_->{rpm}->is_source_package && ! $_->{inuse} ) ) } @pkgs;
    }
    printline('white', 0, "Finished processing.\n");

    unless ( $opts{t} || $opts{r} ) {
	my %repochg = ();
	# Move / Copy / Delete packages and tag repo for rebuild.
	foreach my $pkg ( grep { $_->{oldrepo} } @rpms ) {
	    printline('white', 0, 'Copy/move: ', $pkg->{rpm}->filename);
	    my $src = $pkg->{rpm}->filename;

	    qx(cat $HOME/.rpmpass | setsid rpm --addsign $src >& /dev/null) if $pkg->{nosig};
	    if ( $pkg->{repos} ) {
		foreach my $dest ( dest($pkg, @{$pkg->{repos}}) ) {
		    qx(cp --preserve=timestamps $src $dest) unless -f "$dest";
		}
		$repochg{$_}++ foreach @{$pkg->{repos}};
	    }
	    if ( $pkg->{remove} ) {
		qx(rm -f $src);
		$repochg{$pkg->{oldrepo}}++;
	    }
	}
	finddepth(sub{rmdir}, $distrepo->{$rel}->{builds});
	finddepth(sub{rmdir}, $distrepo->{$rel}->{contribs});
	printline('white', 0, "Finished copying/moving.\n");

	foreach my $repo ( sort keys %repochg ) {
	    next if ! $repos->{$repo} || $repos->{$repo}->{orig} || $repos->{$repo}->{stage};
	    foreach my $arch ( @archs ) {
		my $dir = $distrepo->{$rel}->{repo} . "$repo/$arch";
		$dir = qx(readlink -f $dir);
		chomp $dir;
		printline('white', 0, "Createrepo: $repo/$arch");
		qx(createrepo $dir);
		printline('white', 0, "Repoview: $repo/$arch");
		qx(repoview $dir);
		qx(rm -rf $dir/.olddata) if -d "$dir/.olddata";
		qx(rm -rf $dir/.repodata) if -d "$dir/.repodata";
	    }
	}
	printline('white', 0, "Finished rebuilding repos.\n");
    }

    print "\n================================================================================\n\n";
}

sub pkgsrt {
    return $a->{rpm}->is_source_package <=> $b->{rpm}->is_source_package || $b->{rpm} cmp $a->{rpm} || rprop($b, 'prio') <=> rprop($a, 'prio');
}

sub rprop {
    my ($pkg, $prop) = @_;
    return $repos->{$pkg->{repo}} ? $repos->{$pkg->{repo}}->{$prop} : 0;
}

sub names {
    my $pkg = shift;
    my @rpms = @_;
    return map { $_->{name} => rprop($_, 'inc') ? 2 : 1 } grep { $_->{base} eq $pkg->{base} && 
								 rprop($_, 'prio') >= rprop($pkg, 'prio') && 
								 ! rprop($_, 'orig') &&
								 ! $_->{rpm}->is_source_package
							       } @rpms;
}

sub dest {
    my ($pkg, @repos) = @_;
    my @d = ();
    foreach my $repo ( @repos ) {
	my $base = $distrepo->{$rel}->{repo} . $repo;
	$base = $distrepo->{$rel}->{stage} if $repos->{$repo}->{stage};
	if ( $pkg->{arch} eq 'noarch' ) {
	    push @d, "$base/i386/" . 
		     ( $repos->{$repo}->{os} ? $distrepo->{$rel}->{os} : 'RPMS/' ) . 
		     basename $pkg->{rpm}->filename;
	    push @d, "$base/x86_64/" . 
		     ( $repos->{$repo}->{os} ? $distrepo->{$rel}->{os} : 'RPMS/' ) . 
		     basename $pkg->{rpm}->filename;
	} elsif ( $pkg->{arch} =~ m{^(i[356]86)$} ) {
	    push @d, "$base/i386/" . 
		     ( $repos->{$repo}->{os} ? $distrepo->{$rel}->{os} : 'RPMS/' ) . 
		     basename $pkg->{rpm}->filename;
	} elsif ( $pkg->{arch} eq 'x86_64' ) {
	    push @d, "$base/x86_64/" . 
		     ( $repos->{$repo}->{os} ? $distrepo->{$rel}->{os} : 'RPMS/' ) . 
		     basename $pkg->{rpm}->filename;
	} elsif ( $pkg->{arch} eq 'zsrc' ) {
	    push @d, "$base/SRPMS/" . 
		     basename $pkg->{rpm}->filename;
	} else {
	    printline('bold yellow on_red', 1, ' - ', $pkg->{rpm}->filename, ' (bad arch: ', $pkg->{arch}, ')');
	}
    }
    return @d;
}

sub movepkg {
    my ($pkg, $repo, $done, $remove) = @_;
    defined $done or $done = 2;
    defined $remove or $remove = 0;
    $remove = 0 if rprop($pkg, 'orig');
    $remove = 1 if $repo eq 'delete';

    if ( $repo eq 'delete' && ( rprop($pkg, 'base') || (  rprop($pkg, 'orig') && $done == 1 ) ) ) {
	tagpkg($pkg, 'inuse', 1);
    } else {
	printline('bold magenta', 1, "       - ($repo) ", $pkg->{rpm}->filename) if $opts{d};
	my $color = 'bold white';
	$color = 'bold yellow' if rprop($pkg, 'orig');
	$color = 'dark green' if $repos->{$repo} && $repos->{$repo}->{devel};
	$color = 'dark red' if $repo eq 'delete';
	push @{$pkg->{repos}}, $repo unless $repo eq 'delete';
	$pkg->{oldrepo} = $pkg->{repo} unless $pkg->{oldrepo};
	$pkg->{repo} = $repo;
	$pkg->{remove}++ if $remove;
	printline($color, 1, ' - ', $pkg->{rpm}->filename, ' (', $pkg->{oldrepo}, " to $repo)") if $opts{v} || $opts{d};
	done($pkg) if $done;
    }
}

sub tagpkg {
    my ($pkg, $tag, $done) = @_;
    defined $done or $done = 0;

    printline('bold magenta', 1, "       - ($tag) ", $pkg->{rpm}->filename) if $opts{d};
    delete $pkg->{obsolete} if $pkg->{obsolete};
    $pkg->{$tag}++;
    done($pkg) if $done;
}

sub done {
    my $pkg = shift;
    printline('bold yellow on_red', 1, " - ", $pkg->{rpm}->filename, " (already processed)") if $pkg->{done};
    $pkg->{done}++;
    delete $pkg->{obsolete} if $pkg->{obsolete};

    if ( $repos->{$pkg->{repo}} && ! rprop($pkg, 'orig') ) {
	# FIXME: Nasty hack for comps package
	if ( $pkg->{name} eq 'comps' ) {
	    if ( ! $latest{$pkg->{name}.'.'.$pkg->{arch}} || $latest{$pkg->{name}.'.'.$pkg->{arch}}->{rpm} eq $pkg->{rpm} ) { 
		if ( rprop($pkg, 'inc') && ! $pkg->{rpm}->is_source_package ) {
		    $latest{$pkg->{name}.'.'.$pkg->{arch}} = $pkg if ! $latest{$pkg->{name}.'.'.$pkg->{arch}};
		    $pkg->{latest}++;
		}
	    }
	} elsif ( ! $latest{$pkg->{name}} || $latest{$pkg->{name}}->{rpm} eq $pkg->{rpm} ) { 
	    if ( rprop($pkg, 'inc') && ! $pkg->{rpm}->is_source_package ) {
		$latest{$pkg->{name}} = $pkg if ! $latest{$pkg->{name}};
		$pkg->{latest}++;
	    }
	}
	unless ( $pkg->{oldrepo} ) {
	    my $goodarch = 0;
	    $goodarch++ if $pkg->{rpm}->filename =~ m{/(i386|x86_64)/.*\.noarch\.rpm$};
	    $goodarch++ if $pkg->{rpm}->filename =~ m{/i386/.*\.(i[56]86)\.rpm$};
	    $goodarch++ if $pkg->{rpm}->filename =~ m{/(noarch|i[356]86|x86_64)/.*\.\1\.rpm$};
	    printline('bold yellow on_red', 1, ' - ', $pkg->{rpm}->filename, ' (wrong arch)') unless $goodarch;
	}
	push @{$archs->{$pkg->{repo}}->{$pkg->{rpm}->as_nvre}->{$pkg->{rpm}->is_source_package ? 'src' : $pkg->{rpm}->arch}}, $pkg;
    }
}

sub printline {
    my $color = shift;
    my $nline = shift;
    my $string = join '', @_;

    if ( $nline ) {
	print "\n" unless $newline;
	$newline++; 
    } else {
	$newline = 0;
	print "\r\e[0K";
    }
    print color $color;
    print $string;
    print color 'reset';
    print "\n" if $nline;
}

sub loadpkg {
    printline('white', 0, "Loading: $_") if -d _;
    return unless m{/([^/]*)-[^-]+-[^-]+\.\w+.rpm} && ( ! $opts{r} || $1 =~ m{$opts{r}} );

    my $pkg;
    eval { $pkg = RPM2->open_package($_, $rpm_flags); };
    if ($@) { printline('bold yellow on_red', 1, "Corrupt package $_"); return; }
    eval { my $sig = $pkg->siggpg };
    my $nosig = ! $@;
    
    my ($srcname, $version, $release) = ($pkg->is_source_package ? basename $pkg->filename : $pkg->sourcerpm) =~ m{^([^/]*)-([^-]+)-([^-]+)\.\w+.rpm};
    foreach my $repo ( keys %{$skippkg->{$rel}} ) { return if (dirname $pkg->filename) =~ m{/$repo/} && grep { $_ eq $srcname } @{$skippkg->{$rel}->{$repo}}; }
    my $reponame = 'unknown';
    if ( substr($_, 0, length($distrepo->{$rel}->{builds})) eq $distrepo->{$rel}->{builds} ) {
	$reponame = 'builds';
    } elsif ( substr($_, 0, length($distrepo->{$rel}->{contribs})) eq $distrepo->{$rel}->{contribs} ) {
	$reponame = 'contribs';
    } elsif ( substr($_, 0, length($distrepo->{$rel}->{stage})) eq $distrepo->{$rel}->{stage} ) {
	$reponame = 'stage';
    } elsif ( (dirname $pkg->filename) =~ m{\d/(sme[^/]+)/} ) {
	$reponame = $1;
    }
    if ( $nosig && ( ! -f "$HOME/.rpmpass" || ! $repos->{$reponame}->{orig} ) ) {
	printline('bold yellow on_red', 1, ' - ', $pkg->filename, ' (missing signature)');
	return;
    }
    push @rpms, { base    => $srcname,
		  name    => $pkg->name,
		  repo    => $reponame,
		  nvra    => $pkg->name.'-'.$pkg->version.'-'.$pkg->release.'.'.( $pkg->is_source_package ? 'src' : $pkg->tag('ARCH') ),
		  arch    => $pkg->is_source_package ? 'zsrc' : $pkg->tag('ARCH'),
		  version => $version,
		  release => $release,
		  rpm     => $pkg,
		  nosig   => $nosig,
		};
    $sources{$srcname}++;
    if ( ! $pkg->is_source_package && ! $repos->{$reponame}->{stage} ) {
	$latest{$pkg->name} = $rpms[$#rpms] if ! $latest{$pkg->name} || $pkg gt $latest{$pkg->name}->{rpm};
    }
}

sub loadbase {
    printline('white', 0, "Loading: $_") if -d _;
    return unless m{/([^/]*)-[^-]+-[^-]+\.\w+.rpm} && ($latest{$1} || $sources{$1});

    my $pkg;
    eval { $pkg = RPM2->open_package($_, $rpm_flags); };
    if ($@) { printline('bold yellow on_red', 1, "Corrupt package $_"); return; }
    eval { my $sig = $pkg->siggpg };
    unless ($@) { printline('bold yellow on_red', 1, ' - ', $pkg->filename, ' (missing signature)'); return; }

    my $cmp = $latest{$pkg->name};
    my ($srcname, $version, $release) = ($pkg->is_source_package ? basename $pkg->filename : $pkg->sourcerpm) =~ m{^([^/]*)-([^-]+)-([^-]+)\.\w+.rpm};
    my $pkgrepo = 'base';
    foreach my $repo ( keys %{$skippkg->{$rel}} ) { 
	if ( (dirname $pkg->filename) =~ m{/$repo/} ) {
	    return if grep { $_ eq $srcname } @{$skippkg->{$rel}->{$repo}}; 
	    $pkgrepo = $repo;
	    return unless $pkg->is_source_package || $cmp;
	    return if ! $pkg->is_source_package && $pkg ge $cmp->{rpm} && rprop($cmp, 'base') && rprop($cmp, 'prio') > $repos->{$pkgrepo}->{prio};
	    last;
	}
    }

    push @rpms, { base    => $srcname,
		  name    => $pkg->name,
		  repo    => $pkgrepo,
		  nvra    => $pkg->name.'-'.$pkg->version.'-'.$pkg->release.'.'.( $pkg->is_source_package ? 'src' : $pkg->tag('ARCH') ),
		  arch    => $pkg->is_source_package ? 'zsrc' : $pkg->tag('ARCH'),
		  version => $version,
		  release => $release,
		  rpm     => $pkg,
		  nosig   => 0,
		};
    if ( ! $pkg->is_source_package && $pkg ge $cmp->{rpm} && ( $repos->{$pkgrepo}->{prio} >= rprop($cmp, 'prio') || ! rprop($cmp, 'base') ) ) {
	$latest{$pkg->name} = $rpms[$#rpms];
    }
}