#!/bin/bash
# Move uploaded .warc.gz files to an archive directory.
# When the archive is large enough, make a tar and start with a
# new archive.
#
# Be careful: this script assumes that any file in the upload directory
# that has a name that ends with *.warc.gz is a fully uploaded file and
# can be moved somewhere else. Remember this when running Rsync.
#

INCOMING_UPLOADS_DIR="${1}" # /home/archiveteam/uploads
CHUNKER_WORKING_DIR="${2}"  # /home/archiveteam/processed
PACKING_QUEUE_DIR="${CHUNKER_WORKING_DIR}/archive"
MEGABYTES_PER_CHUNK=$((1024*25))

# if not specified in command-line arguments
if [ -z "${INCOMING_UPLOADS_DIR}" ]
then
  source ./config.sh || exit 1
fi

BYTES_PER_CHUNK=$((1024*1024*MEGABYTES_PER_CHUNK))

mkdir -p "${CHUNKER_WORKING_DIR}" || exit 1
mkdir -p "${PACKING_QUEUE_DIR}" || exit 1

mkdir -p "${CHUNKER_WORKING_DIR}/current" || exit 1
cur_size=$( du -B1 -s "${CHUNKER_WORKING_DIR}/current" | grep -oE "^[0-9]+" )

# find every .warc.gz in the upload directory
find "${INCOMING_UPLOADS_DIR}" -type f -regex ".+\.warc\.\(gz\|zst\)$" \
| while read filename
do
  # skip partial uploads
  if [[ "${filename}" =~ rsync-tmp ]]
  then
    continue
  fi

  cur_size=$((cur_size + $( du -B1 -s "${filename}" | grep -oE "^[0-9]+" )))

  # move to the current/ directory
  echo "Moving ${filename}"
  mkdir -p "${CHUNKER_WORKING_DIR}/current"
  mv "${filename}" "${CHUNKER_WORKING_DIR}/current/"

  # if the current/ directory is large enough,
  # rename it to archive-XXXXX and start a new current/
  if [[ "${cur_size}" -gt "${BYTES_PER_CHUNK}" ]]
  then
    timestamp=$( date +'%Y%m%d%H%M%S' )
    uuid=$(cat /proc/sys/kernel/random/uuid | cut -d- -f1)
    echo "Current archive is full, moving to ${timestamp}_${uuid}."
    mv "${CHUNKER_WORKING_DIR}/current" "${PACKING_QUEUE_DIR}/${timestamp}_${uuid}"
    cur_size=0
    sleep 3
  fi
done