#!/bin/bash

while [[ $# -gt 0 ]]; do
  case $1 in
    --print-moduledir)
      [[ -f config.mk ]] && source config.mk
      [[ -z $DRACUT_MODULEDIR ]] && exit 1
      echo $DRACUT_MODULEDIR
      exit 0
      ;;
    --dracut)
      DRACUT=$2
      shift
      ;;
    --root-home)
      ROOTHOME=$2
      shift
      ;;
  esac
  shift
done

echo -n "Searching for dracut... "
if [[ -z $DRACUT ]]; then
  DRACUT=$(which dracut)
  if [[ $? -eq 0 ]]; then
    echo $DRACUT
  else
    echo "not found"
    exit 1
  fi
else
  echo "[provided] $DRACUT"
fi

echo -n "Checking dracut version... "
DRACUT_VERSION=$(${DRACUT} --help | egrep "Version:[ \t]+[0-9]+" | egrep -o [0-9]+)
if [[ $? -ne 0 ]]; then
	echo -n "[old version - fallback] "
	DRACUT_VERSION=004
else
	DRACUT_VERSION=$(echo ${DRACUT_VERSION} | cut -f 1 -d " ")
fi
echo ${DRACUT_VERSION}

echo -n "Checking dracut module prefix... "
DRACUT_MODULEDIR=""
if [[ -d /usr/lib/dracut/modules.d ]]; then
	DRACUT_MODULEDIR=/usr/lib/dracut/modules.d
elif [[ -d /usr/share/dracut/modules.d ]]; then
	DRACUT_MODULEDIR=/usr/share/dracut/modules.d
else
	echo "failed"
	exit 1
fi
echo ${DRACUT_MODULEDIR}

NEED_CRYPTSETTLE=0
CRYPTPARSE=${DRACUT_MODULEDIR}/90crypt/parse-crypt.sh
if [[ -f $CRYPTPARSE ]]; then
  echo -n "Checking whether crypt-settle patch is needed... "
  grep -q -- "--settled" ${CRYPTPARSE}
  if [[ $? -ne 0 ]]; then
    NEED_CRYPTSETTLE=1
    echo "yes"
  else
    echo "no"
  fi
fi

echo -n "Checking for old-style modules... "
OLDSTYLE=1
if [[ -f ${DRACUT_MODULEDIR}/99base/module-setup.sh ]]; then
  OLDSTYLE=0
  echo "no"
else
  echo "yes"
fi

echo -n "Checking for root's home... "
if [[ -z ${ROOTHOME} ]]; then
  ROOTHOME=/
  module=module-setup.sh
  [[ ${OLDSTYLE} -eq 1 ]] && module=install
  MODULEBASE=${DRACUT_MODULEDIR}/99base/${module}
  if [[ -f $MODULEBASE ]]; then
    grep -c "root:" ${MODULEBASE} >/dev/null
    if [[ $? -eq 0 ]]; then
      ROOTHOME=/root
    fi
  fi
else
  echo -n "[provided] "
fi
echo ${ROOTHOME}

cat >config.mk <<EOF
export DRACUT=${DRACUT}
export DRACUT_VERSION=${DRACUT_VERSION}
export DRACUT_MODULEDIR=${DRACUT_MODULEDIR}
export NEED_CRYPTSETTLE=${NEED_CRYPTSETTLE}
export ROOTHOME=${ROOTHOME}
export OLDSTYLE=${OLDSTYLE}
EOF
