#!/bin/bash

ac=$#
cmd=`basename $0`

rpmdb=/var/lib/rpm
dbfiles="`/bin/ls -1 $rpmdb/*`"
dbdump=/usr/lib/rpm/rpmdb_dump
dbload=/usr/bin/db_load

set `$dbdump -V` || {
    echo "The file $dbdump, usually in the rpm-devel package, is needed."
    echo " Please install the rpm-devel package and retry $cmd."
    exit 1
}
dumpfmt="${5%%:}"

set `$dbload -V` || {
    echo "The file $dbload, usually in the db4-utils package, is needed."
    echo " Please install the db4_utils package and retry $cmd."
    exit 1
}
loadfmt="${5%%:}"

[ "$dumpfmt" = "$loadfmt" ] && {
    echo "Both $dbdump and $dbload use db-$loadfmt format."
    echo "No conversion is possible."
    exit 0
}

[ "$ac" != "0" -o "`/usr/bin/id -u`" != "0" ] && {
    echo "$cmd: Convert $rpmdb files to db-$loadfmt format."
    echo "$cmd: Must be run as root, takes no arguments."
    exit 1
}

echo "Converting to db-$loadfmt compatible format (1-10 minutes per file) ..."

dbfile=
trap "
    [ -n "$dbfile" ] && {
	[ -f $dbfile-O ] && /bin/mv $dbfile-O $dbfile
	/bin/rm -f $dbfile-N
    }
    exit 1
" 1 2 3 15

rm -f $rpmdb/__db*

for dbfile in $dbfiles
do
    echo " `/usr/bin/file $dbfile`"

    $dbdump $dbfile | $dbload $dbfile-N || {
	echo "Failed conversion:"
	echo "    $dbdump $dbfile | $dbload $dbfile-N"
	echo "Cleaning up and exiting ..."
	/bin/rm -f $dbfile-N
	exit 1
    }

    /bin/mv $dbfile $dbfile-O
    /bin/mv $dbfile-N $dbfile
    /bin/rm -f $dbfile-O
done

echo "Done!"

exit 0