#!/bin/sh
#
# Process a distfiles/ directory, efficiently moving files from
# .pbtmp/<package>/* into . (relative to distfiles/)
#
# We do this in several stages
#
# 1. Remove incomplete downloads where .pbtmp/<package>/.done does not
#    exist (this is an incomplete transfer and may be corrupted)
#
# 2. Populate the directory hierarchy from .pbtmp/*/ into .
#
# 3. For each subdirectory under .pbtmp/*/, group them together by
#    subdirectory name and move all of the files at once.
#    e.g. .pbtmp/foo-1.0/dir/* and .pbtmp/bar-2.0/dir/* are
#    processed at the same time (contents moved to ./dir/).
#
# 4. Once we have handled the subdirectories, everything left
#    is a plain file in .pbtmp/*/ so we move those in bulk together
#    into .
#
# 5. Clean up

if [ $# -ne 3 ]; then
  echo "usage: $0 <arch> <branch> <buildid>"
  exit 1
fi

arch=$1
branch=$2
buildid=$3

pbd=${PORTBUILD_DATA:-/a/portbuild}

builddir=${pbd}/${arch}/${branch}/builds/${buildid}

distdir=${builddir}/distfiles

cd ${distdir} || exit 1

echo "Removing incomplete downloads"

if [ -d ${distdir}/.pbtmp/ ]; then
  find ${distdir}/.pbtmp/ -name .done -depth 2 | sed -e 's,/.done$,/,' | sort > .done || exit 1
  find -d ${distdir}/.pbtmp/ -type d -mindepth 1 |sed -E -e 's,([^/])$,\1/,' > .alldirs || exit 1
else
  cp /dev/null .done
  cp /dev/null .alldirs
fi

sed -E -e "s,^(${distdir}/.pbtmp/[^/]+/).*,\1," < .alldirs | sort -u > .pkgdirs

comm -1 -3 .done .pkgdirs | xargs rm -rf

# Full path of subdirectories under package dirs
grep -E "^${distdir}/.pbtmp/[^/]+/.+/" .alldirs > .pkgsubdirs

# All subdirectories under package dirs
sed -E -e "s,^${distdir}/.pbtmp/[^/]+/,," < .pkgsubdirs | grep -v '^$' | sort -u > .subdirs

echo "Making directories"
cat .subdirs | xargs mkdir -p

# Move files in each subdir
for i in `cat .pkgsubdirs`; do
  find ${i} -type f -depth 1 \! -name .done | xargs -J % mv % ${distdir}/${i#${distdir}/.pbtmp/*/}
#  rmdir ${i} || exit 1
#  rm -rf ${distdir}/.pbtmp/$i
done
cat .pkgsubdirs | xargs rmdir || exit 1

echo "Moving remaining distfiles"
if [ -d ${distdir}/.pbtmp/ ]; then
  find ${distdir}/.pbtmp/ -type f -depth 2 \! -name .done | xargs -J % mv % ${distdir}
fi

echo "Cleaning up"
sed -e 's,$,.done,' < .pkgdirs | xargs rm -f || exit 1
cat .pkgdirs | xargs rmdir || exit 1
rmdir .pbtmp 2> /dev/null || exit 1

rm -f .alldirs .done .pkgdirs .pkgsubdirs .subdirs || exit 1