#!/bin/sh

if [ $# -lt 1 ]; then
	cat <<EOF
Usage: dbcheck.sh PREFIX
EOF
	exit 1
fi

PREFIX="$1"
shift 1
ARGS=$@

. $(dirname $0)/subunit.sh

dbcheck()
{
	$PYTHON $BINDIR/samba-tool dbcheck --cross-ncs $ARGS
}

# This list of attributes can be freely extended
dbcheck_fix_one_way_links()
{
	$PYTHON $BINDIR/samba-tool dbcheck --quiet --fix --yes fix_all_old_dn_string_component_mismatch --attrs="lastKnownParent defaultObjectCategory fromServer rIDSetReferences" --cross-ncs $ARGS
}

# This list of attributes can be freely extended
dbcheck_fix_stale_links()
{
	$PYTHON $BINDIR/samba-tool dbcheck --quiet --fix --yes remove_plausible_deleted_DN_links --attrs="member msDS-NC-Replica-Locations msDS-NC-RO-Replica-Locations msDS-RevealOnDemandGroup msDS-NeverRevealGroup msDS-RevealedUsers" --cross-ncs $ARGS
}

# This list of attributes can be freely extended
dbcheck_fix_crosspartition_backlinks()
{
	# we may not know the target yet when we receive a cross-partition link,
	# which can result in a missing backlink
	$PYTHON $BINDIR/samba-tool dbcheck --quiet --fix --yes fix_all_missing_backlinks --attrs="serverReference" --cross-ncs $ARGS
}

# This test shows that this does not do anything to a current
# provision (that would be a bug)
dbcheck_reset_well_known_acls()
{
	$PYTHON $BINDIR/samba-tool dbcheck --cross-ncs --reset-well-known-acls $ARGS
}

reindex()
{
	$PYTHON $BINDIR/samba-tool dbcheck --reindex $ARGS
}

fixed_attrs()
{
	$PYTHON $BINDIR/samba-tool dbcheck --attrs=cn $ARGS
}

force_modules()
{
	$PYTHON $BINDIR/samba-tool dbcheck --force-modules $ARGS
}

dbcheck_fix_one_way_links
dbcheck_fix_stale_links
dbcheck_fix_crosspartition_backlinks
testit "dbcheck" dbcheck || failed=$(expr $failed + 1)
testit "reindex" reindex || failed=$(expr $failed + 1)
testit "fixed_attrs" fixed_attrs || failed=$(expr $failed + 1)
testit "force_modules" force_modules || failed=$(expr $failed + 1)
testit "reset_well_known_acls" dbcheck_reset_well_known_acls || failed=$(expr $failed + 1)

exit $failed