#!/bin/sh
#-
# Copyright (c) 2010 iXsystems, Inc.  All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. 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 AUTHOR 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 AUTHOR 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.
#
# $FreeBSD: releng/11.0/usr.sbin/pc-sysinstall/backend/functions-installpackages.sh 247734 2013-03-03 23:01:46Z jpaetzel $

# Functions which check and load any optional packages specified in the config

. ${BACKEND}/functions.sh
. ${BACKEND}/functions-parse.sh

# Recursively determine all dependencies for this package
determine_package_dependencies()
{
  local PKGNAME="${1}"
  local DEPFILE="${2}"

  grep -q "${PKGNAME}" "${DEPFILE}"
  if [ $? -ne 0 ]
  then
    echo "${PKGNAME}" >> "${DEPFILE}"
    get_package_dependencies "${PKGNAME}" "1"

    local DEPS="${VAL}"
    for d in ${DEPS}
    do
      determine_package_dependencies "${d}" "${DEPFILE}"
    done
  fi
};

# Fetch packages dependencies from a file
fetch_package_dependencies()
{
  local DEPFILE
  local DEPS
  local SAVEDIR

  DEPFILE="${1}"
  DEPS=`cat "${DEPFILE}"`
  SAVEDIR="${2}"

  for d in ${DEPS}
  do
    get_package_short_name "${d}"
    SNAME="${VAL}"

    get_package_category "${SNAME}"
    CATEGORY="${VAL}"

    fetch_package "${CATEGORY}" "${d}" "${SAVEDIR}"
  done
};

# Check for any packages specified, and begin loading them
install_packages()
{
  echo "Checking for packages to install..."
  sleep 2

  # First, lets check and see if we even have any packages to install
  get_value_from_cfg installPackages

  # Nothing to do?
  if [ -z "${VAL}" ]; then return; fi

  echo "Installing packages..."
  sleep 3

  local PKGPTH

  HERE=`pwd`
  rc_halt "mkdir -p ${FSMNT}${PKGTMPDIR}"

  # Determine the directory we will install packages from
  get_package_location
  rc_halt "cd ${PKGDLDIR}"

  # Set the location of the INDEXFILE
  INDEXFILE="${TMPDIR}/INDEX"

  if [ ! -f "${INDEXFILE}" ]; then
    get_package_index
  fi

  if [ ! -f "${TMPDIR}/INDEX.parsed" -a "$INSTALLMEDIUM" = "ftp" ]; then
    parse_package_index
  fi

  # What extension are we using for pkgs?
  PKGEXT="txz"
  get_value_from_cfg pkgExt
  if [ -n "${VAL}" ]; then 
     strip_white_space ${VAL}
     PKGEXT="$VAL"
  fi
  export PKGEXT
  
  # We dont want to be bothered with scripts asking questions
  PACKAGE_BUILDING=yes
  export PACKAGE_BUILDING

  # Lets start by cleaning up the string and getting it ready to parse
  get_value_from_cfg_with_spaces installPackages
  PACKAGES="${VAL}"
  echo_log "Packages to install: `echo $PACKAGES | wc -w | awk '{print $1}'`"
  for i in $PACKAGES
  do
    if ! get_package_name "${i}"
    then
      echo_log "Unable to locate package ${i}"
      continue
    fi

    PKGNAME="${VAL}"

    # Fetch package + deps, but skip if installing from local media
    if [ "${INSTALLMEDIUM}" = "ftp" ] ; then
      DEPFILE="${FSMNT}/${PKGTMPDIR}/.${PKGNAME}.deps"
      rc_nohalt "touch ${DEPFILE}"
      determine_package_dependencies "${PKGNAME}" "${DEPFILE}"
      fetch_package_dependencies "${DEPFILE}" "${FSMNT}/${PKGTMPDIR}"
    fi

    # Set package location
    case "${INSTALLMEDIUM}" in
      usb|dvd|local) PKGPTH="${PKGTMPDIR}/All/${PKGNAME}" ;;
                  *) PKGPTH="${PKGTMPDIR}/${PKGNAME}" ;;
    esac

    # See if we need to determine the package format we are working with
    if [ -z "${PKGINFO}" ] ; then
      tar tqf "${FSMNT}${PKGPTH}" '+MANIFEST' >/dev/null 2>/dev/null	
      if [ $? -ne 0 ] ; then
        PKGADD="pkg_add -C ${FSMNT}" 
        PKGINFO="pkg_info" 
      else
        PKGADD="pkg -c ${FSMNT} add"
        PKGINFO="pkg info"
        bootstrap_pkgng
      fi
    fi

    # If the package is not already installed, install it!
    if ! run_chroot_cmd "${PKGINFO} -e ${PKGNAME}" >/dev/null 2>/dev/null
    then
      echo_log "Installing package: ${PKGNAME}"
      rc_nohalt "${PKGADD} ${PKGPTH}"
    fi

    if [ "${INSTALLMEDIUM}" = "ftp" ] ; then
      rc_nohalt "rm ${DEPFILE}"
    fi

  done

  echo_log "Package installation complete!"

  # Cleanup after ourselves
  rc_halt "cd ${HERE}"
  if [ "${INSTALLMEDIUM}" = "ftp" ] ; then
    rc_halt "rm -rf ${FSMNT}${PKGTMPDIR}" >/dev/null 2>/dev/null
  else
    rc_halt "umount ${FSMNT}${PKGTMPDIR}" >/dev/null 2>/dev/null
    rc_halt "rmdir ${FSMNT}${PKGTMPDIR}" >/dev/null 2>/dev/null
  fi
};
