#!/bin/sh
#-*-mode:  sh -*-

# Copyright (c) 2007-2015, Aleksey Cheusov <vle@gmx.net>
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above
#       copyright notice, this list of conditions and the following
#       disclaimer in the documentation and/or other materials provided
#       with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

set -e

export LC_ALL=C

############################################################

LIBEXECDIR=${LIBEXECDIR-/usr/pkg/libexec/psu}
. ${LIBEXECDIR}/sig_handler.sh
on_exit () {
	# Stupid test for stupid Solaris
	if test -n "$tmp_dir"; then rm -rf $tmp_dir; fi
}

tmp_dir=`mktemp -d ${TMPDIR-/tmp}/pkg_cleanup_distdir.XXXXXX`
test -n "$tmp_dir" || exit 1

############################################################

usage (){
    cat 1>&2 <<EOF
pkg_cleanup_distdir - analyses ALLSRCFILES field of src_summary
given on input and optionally removes files and directories in DISTDIR
not referenced by any package.

usage: pkg_cleanup_distdir [OPTIONS] [files...]
OPTIONS:
  -h     display this help message
  -r     remove unreferenced files and empty directories
EOF
}

delete=''
while getopts hr f; do
    case "$f" in
	h)   usage; exit 0;;
	r)   delete=1;;
	?)   printf "Run pkg_cleanup_distdir -h for details\n"; exit 2;;
    esac
done
shift `expr $OPTIND - 1`

DISTDIR=${DISTDIR:=/usr/pkgsrc/distfiles}
cd "$DISTDIR"

useful_distfiles (){
    awk '/^ALLSRCFILES=/ {
	$0 = substr($0, 13)
	for (i=1; i <= NF; ++i)
	    print $i
    }' "$@" |
    sort | uniq
}

existing_files (){
    find . -type f | sed "s,^[.]/,,"
}

ex_file=$tmp_dir/existing
us_file=$tmp_dir/useful
rm_file=$tmp_dir/to_be_removed
rmdir_file=$tmp_dir/dirs

existing_files > "$ex_file"

useful_distfiles "$@" > "$us_file"

sort "$ex_file" "$ex_file" "$us_file" | uniq -c |
awk '$1 == 2 {print $2}' > "$rm_file"

if test -n "$delete"; then
    xargs rm -f < "$rm_file"

    sed 's,/[^/]*$,,' "$rm_file" | sort | uniq |
    while read -r dir; do
	while test "$dir" != '.'; do
	    if ! rmdir "$dir" 2>/dev/null; then
		break
	    fi
	    dir=`dirname $dir`
	done
    done
else
    cat "$rm_file"
fi
