#! /bin/sh
set -e

# Copyright (C) 1999,2000,2001,2002,2003,2004,2005,2006,2007,2008,2009,2010,2011,2012  Free Software Foundation, Inc.
#
# GRUB is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# GRUB is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GRUB.  If not, see <http://www.gnu.org/licenses/>.

prefix="/usr"
exec_prefix="${prefix}"
datarootdir="${prefix}/share"
bindir="${exec_prefix}/bin"

libdir="${exec_prefix}/lib"
PACKAGE_NAME=GRUB2
PACKAGE_TARNAME=grub2
PACKAGE_VERSION=2.00
datadir="${datarootdir}"
if [ "x$pkgdatadir" = x ]; then
    pkgdatadir="${datadir}/grub2"
fi

self=`basename $0`

export TEXTDOMAIN=grub2
export TEXTDOMAINDIR="${datarootdir}/locale"

. "${pkgdatadir}/grub-mkconfig_lib"

modules=

pkglib_DATA="moddep.lst command.lst fs.lst partmap.lst parttool.lst \
handler.lst video.lst crypto.lst terminal.lst"

grub_mkimage="${bindir}/grub2-mkimage"

grub_compress_file () {
    if [ "$compressor" != "" ] ; then
        "$compressor" $compressor_opts "$1" > "$2"
    else
        cp -f "$1" "$2"
    fi
}

grub_install_files () {
    grub_install_files_source_directory="$1"
    grub_install_files_target_directory="$2"
    grub_install_files_platform="$3"
    
    mkdir -p "${grub_install_files_target_directory}"/"${grub_install_files_platform}"
    
    for file in "${grub_install_files_target_directory}"/*.mod \
"${grub_install_files_target_directory}"/*.lst \
"${grub_install_files_target_directory}"/*.img \
"${grub_install_files_target_directory}"/efiemu??.o \
"${grub_install_files_target_directory}"/"${grub_install_files_platform}"/*.mod \
"${grub_install_files_target_directory}"/"${grub_install_files_platform}"/*.lst \
"${grub_install_files_target_directory}"/"${grub_install_files_platform}"/*.img \
"${grub_install_files_target_directory}"/"${grub_install_files_platform}"/efiemu??.o;
    do
	if test -f "$file" && [ "`basename $file`" != menu.lst ]; then
	    rm -f "$file" || exit 1
	fi
    done

    if [ x"$install_modules" = xall ]; then
	for file in "${grub_install_files_source_directory}/"*.mod; do
	    grub_compress_file "$file" "${grub_install_files_target_directory}"/"${grub_install_files_platform}/$(basename "$file")"
	done
    else
	modules1=
	modules2="$install_modules"
	while [ x"$modules2" != x ]; do
	    modules3=
	    for x in $modules2; do
		modules3="$modules3 $(grep "^$x:" "${grub_install_files_source_directory}/moddep.lst" | sed 's,^[^:]*:,,')"
	    done
	    modules1="$modules1 $modules2"
	    modules2="$modules3"
	done
	for file in $(echo "$modules1" | sed 's, ,\n,g' |sort -u); do
	    grub_compress_file "${grub_install_files_source_directory}/$file.mod" "${grub_install_files_target_directory}"/"${grub_install_files_platform}/$file.mod"
	done
    fi
    
    for file in ${pkglib_DATA} efiemu32.o efiemu64.o; do
	if test -f "${grub_install_files_source_directory}/${file}"; then
            grub_compress_file "${grub_install_files_source_directory}/${file}" "${grub_install_files_target_directory}"/"${grub_install_files_platform}/${file}"
	fi
    done
    
    # Copy gettext files
    mkdir -p "${grub_install_files_target_directory}"/locale

    for file in "${grub_install_files_target_directory}"/locale/*.mo; do
	if test -f "$file"; then
	    rm -f "$file" || exit 1
	fi
    done

    if [ x"$install_locales" = xall ]; then
	for file in "${grub_install_files_source_directory}"/po/*.mo; do
	    if test -f "$file"; then
		grub_compress_file "$file" "${grub_install_files_target_directory}"/locale/"$(basename "$file")"
	    fi
	done
	for dir in "${localedir}"/*; do
	    if test -f "$dir/LC_MESSAGES/grub2.mo" && ! test -f "${grub_install_files_target_directory}"/locale/"${dir##*/}.mo"; then
		grub_compress_file "$dir/LC_MESSAGES/grub2.mo" "${grub_install_files_target_directory}"/locale/"${dir##*/}.mo"
	    fi
	done
    else
	for locale in $install_locales; do
	    if test -f "${grub_install_files_source_directory}"/po/$locale.mo; then
		grub_compress_file "${grub_install_files_source_directory}"/po/locale.mo "${grub_install_files_target_directory}"/locale/$locale.mo
	    elif test -f "${localedir}/$locale/LC_MESSAGES/grub2.mo"; then
		grub_compress_file "${localedir}/$locale/LC_MESSAGES/grub2.mo" "${grub_install_files_target_directory}"/locale/$locale.mo
	    fi
	done
    fi
    for theme in ${install_themes} ; do
	if test -f "${pkgdatadir}"/themes/"${theme}"/theme.txt; then
	    mkdir -p "${grub_install_files_target_directory}"/themes/"${theme}"
	    for file in "${pkgdatadir}"/themes/"${theme}"/*; do
		grub_compress_file "$file" "${grub_install_files_target_directory}"/themes/"${theme}"/"$(basename "$file")"
	    done
	fi
    done

    for font in ${install_fonts} ; do
	if test -f "${pkgdatadir}"/"$font".pf2; then
	    mkdir -p "${grub_install_files_target_directory}"/fonts
	    grub_compress_file "${pkgdatadir}"/"$font".pf2 "${grub_install_files_target_directory}"/fonts/"$font".pf2
	fi
    done
}

grub_print_install_files_help () {
    print_option_help "--modules=$(gettext "MODULES")" "$(gettext "pre-load specified modules MODULES")"
    print_option_help "--install-modules=$(gettext "MODULES")" "$(gettext "install only MODULES and their dependencies [default=all]")"
    print_option_help "--themes=THEMES" "$(gettext_printf "install THEMES [default=%s]" "starfield")"
    print_option_help "--fonts=FONTS" "$(gettext_printf "install FONTS [default=%s]" "unicode")"
    print_option_help "--locales=LOCALES" "$(gettext_printf "install only LOCALES [default=all]")"
    print_option_help "--compress[=no,xz,gz,lzo]" "$(gettext "compress GRUB files [optional]")"
    # TRANSLATORS: platform here isn't identifier. It can be translated.
    dir_msg="$(gettext_printf "use images and modules under DIR [default=%s/<platform>]" "${libdir}/grub2")"
    print_option_help "-d, --directory=$(gettext "DIR")" "$dir_msg"
    print_option_help  "--grub-mkimage=$(gettext "FILE")" "$(gettext "use FILE as grub-mkimage")"
    print_option_help "-v, --version" "$(gettext "print the version information and exit")"
}

install_modules=all
install_themes=starfield
install_fonts=unicode
install_locales=all
compress=no
grub_decompression_module=""
compressor=""
compressor_opts=""
source_directory=""

argument () {
  opt=$1
  shift

  if test $# -eq 0; then
      gettext_printf "%s: option requires an argument -- \`%s'\n" "$0" "$opt" 1>&2
      exit 1
  fi
  echo $1
}

grub_parse_compress () {
    compress="$1"
    case x"$compress" in
	xno) ;;
	xgz)
	    compressor=`which gzip || true`
	    grub_decompression_module="gzio"
	    compressor_opts="--best --stdout";;
	xxz)
	    compressor=`which xz || true`
	    grub_decompression_module="xzio gcry_crc"
	    compressor_opts="--lzma2=dict=128KiB --check=none --stdout";;
	xlzo)
	    compressor=`which lzop || true`
	    grub_decompression_module="lzopio adler32 gcry_crc"
	    compressor_opts="-9 -c";;
	*)
	    gettext_printf "Unrecognized compression \`%s'\n" "$compress" 1>&2
	    usage
	    exit 1
    esac
}

grub_process_install_options () {
    option=$1
    shift

    grub_process_install_options_consumed=0

    case "$option" in
	--install-modules)
	    install_modules=`argument $option "$@"`; grub_process_install_options_consumed=2; return ;;
	--install-modules=*)
            install_modules=`echo "$option" | sed 's/--install-modules=//'`; grub_process_install_options_consumed=1; return ;;
	--themes)
	    install_themes=`argument $option "$@"`; grub_process_install_options_consumed=2; return ;;
	--themes=*)
            install_themes=`echo "$option" | sed 's/--themes=//'`; grub_process_install_options_consumed=1; return ;;
	--fonts)
	    install_fonts=`argument $option "$@"`; grub_process_install_options_consumed=2; return ;;
	--fonts=*)
            install_fonts=`echo "$option" | sed 's/--fonts=//'`; grub_process_install_options_consumed=1; return ;;
	--locales)
	    install_locales=`argument $option "$@"`; grub_process_install_options_consumed=2; return ;;
	--locales=*)
            install_locales=`echo "$option" | sed 's/--locales=//'`; grub_process_install_options_consumed=1; return ;;
	--compress)
	    grub_parse_compress `argument $option "$@"`; grub_process_install_options_consumed=2; return ;;
        --compress=*)
            grub_parse_compress `echo "${option}" | sed 's/--compress=//'`; grub_process_install_options_consumed=1; return ;;
	--directory | -d)
	    source_directory=`argument $option "$@"`; grub_process_install_options_consumed=2 ;;
	--directory=*)
	    source_directory=`echo "$option" | sed 's/--directory=//'` grub_process_install_options_consumed=1;;

	# For backwards compatibility
	--override-directory)
	    source_directory=`argument $option "$@"`; grub_process_install_options_consumed=2 ;;
	--override-directory=*)
	    source_directory=`echo "$option" | sed 's/--override-directory=//'` grub_process_install_options_consumed=1;;

	--grub-mkimage)
	    grub_mkimage=`argument $option "$@"`; grub_process_install_options_consumed=2 ;;
        --grub-mkimage=*)
	    grub_mkimage=`echo "$option" | sed 's/--grub-mkimage=//'`;grub_process_install_options_consumed=1 ;;
        --modules)
	    modules=`argument $option "$@"`; grub_process_install_options_consumed=2;;
        --modules=*)
	    modules=`echo "$option" | sed 's/--modules=//'` grub_process_install_options_consumed=1;;
        -v | --version)
	    echo "$self (${PACKAGE_NAME}) ${PACKAGE_VERSION}"
	    exit 0 ;;
    esac
}

export grub_decompression_module
# Install GRUB on your drive.
# Copyright (C) 1999,2000,2001,2002,2003,2004,2005,2006,2007,2008,2009,2010  Free Software Foundation, Inc.
#
# GRUB is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# GRUB is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GRUB.  If not, see <http://www.gnu.org/licenses/>.

# Initialize some variables.
host_os=linux-gnu

rootdir=/srv/tftp
modules=

no_floppy=
recheck=no
debug=no
debug_image=
subdir="`echo '/boot/grub2' | sed "s,//*,/,g"`"
pc_dir="${libdir}/grub2/i386-pc"
ppc_dir="${libdir}/grub2/powerpc-ieee1275"
sparc_dir="${libdir}/grub2/sparc64-ieee1275"
i386_ieee1275_dir="${libdir}/grub2/i386-ieee1275"
efi32_dir="${libdir}/grub2/i386-efi"
efi64_dir="${libdir}/grub2/x86_64-efi"
itanium_dir="${libdir}/grub2/ia64-efi"

# Usage: usage
# Print the usage.
usage () {
    gettext_printf "Usage: %s [OPTION]\n" "$self"
    gettext; echo "Install GRUB on your drive."; echo
    echo
    print_option_help "-h, --help" "$(gettext "print this message and exit")"
    grub_print_install_files_help
    print_option_help "--net-directory=$(gettext "DIR")" "$(gettext "root directory of TFTP server")"
    print_option_help "--subdir=$(gettext "DIR")" "$(gettext "relative subdirectory on network server")"
    echo
    gettext_printf "%s copies GRUB images into net_directory/subdir/target_cpu-platform\n" "$self" 
    echo
    gettext "Report bugs to <bug-grub@gnu.org>."; echo
}

# Check the arguments.
while test $# -gt 0
do
    grub_process_install_options "$@"
    case "$grub_process_install_options_consumed" in
	1) shift; continue;;
	2) shift; shift; continue;;
    esac

    option=$1
    shift

    case "$option" in
    -h | --help)
	usage
	exit 0 ;;

    --net-directory)
	rootdir=`argument $option "$@"`; shift;;
    --net-directory=*)
	rootdir=`echo "$option" | sed 's/--net-directory=//'` ;;

    --subdir)
	subdir=`argument $option "$@"`; shift;;
    --subdir=*)
	subdir=`echo "$option" | sed 's/--subdir=//'` ;;

    # This is an undocumented feature...
    --debug)
	debug=yes ;;
    --debug-image)
	debug_image=`argument $option "$@"`; shift;;
    --debug-image=*)
	debug_image=`echo "$option" | sed 's/--debug-image=//'` ;;

    -*)
	gettext_printf "Unrecognized option \`%s'\n" "$option" 1>&2
	usage
	exit 1
	;;
    *)
	gettext_printf "Unknown extra argument \`%s'." "$option" 1>&2
	echo 1>&2
	usage
	exit 1
	;;
    esac
done

set $grub_mkimage dummy
if test -f "$1"; then
    :
else
    gettext_printf "%s: Not found.\n" "$1" 1>&2
    exit 1
fi

# Create the GRUB directory if it is not present.
mkdir -p "${rootdir}/${subdir}" || exit 1

process_input_dir ()
{ 
    input_dir="$1"
    platform="$2"
    grubdir="${rootdir}/${subdir}/${platform}"
    config_opt=

    grub_install_files "${input_dir}" "${rootdir}/${subdir}" "${platform}"

    rm -f "${grubdir}"/load.cfg

    if [ "x${debug_image}" != x ]; then
       echo "set debug='${debug_image}'" >> ${grubdir}/load.cfg
       config_opt="-c ${grubdir}/load.cfg "
    fi

    prefix="/${subdir}";
    case "${platform}" in
    	 i386-pc) mkimage_target=i386-pc-pxe;
	 	  netmodules="pxe";
		  ext=0 ;;
    	 sparc64-ieee1275) mkimage_target=sparc64-ieee1275-aout;
	 	  netmodules="ofnet";
		  ext=img ;;
    	 *-ieee1275) mkimage_target="${platform}";
	 	  netmodules="ofnet";
		  ext=elf ;;
    	 *-efi) mkimage_target="${platform}";
	 	  netmodules="efinet";
		  ext=efi ;;
    	 *) gettext_printf "Unsupported platform %s\n" ${platform};
    	    exit 1;;
    esac

    cat << EOF > ${grubdir}/grub.cfg
source ${subdir}/grub.cfg
EOF

    "$grub_mkimage" ${config_opt} -d "${input_dir}" -O ${mkimage_target} "--output=${grubdir}/core.$ext" "--prefix=$prefix" $modules $grub_decompression_module $netmodules tftp || exit 1
    # TRANSLATORS: First %s is replaced by platform name. Second one by filename.
    gettext_printf "Netboot directory for %s created. Configure your DHCP server to point to %s\n" "${platform}" "${subdir}/${platform}/core.$ext"
}

if [ "${source_directory}" = "" ] ; then
    if test -e "${pc_dir}" ; then
        process_input_dir "${pc_dir}" i386-pc
    fi
    if test -e "${ppc_dir}" ; then
        process_input_dir "${ppc_dir}" powerpc-ieee1275
    fi
    if test -e "${sparc_dir}" ; then
        process_input_dir ${sparc_dir} sparc64-ieee1275
    fi
    if test -e "${i386_ieee1275_dir}" ; then
        process_input_dir "${i386_ieee1275_dir}" i386-ieee1275
    fi
    if test -e "${efi32_dir}" ; then
        process_input_dir "${efi32_dir}" i386-efi
    fi
    if test -e "${efi64_dir}" ; then
        process_input_dir "${efi64_dir}" x86_64-efi
    fi
    if test -e "${itanium_dir}" ; then
        process_input_dir "${itanium_dir}" ia64-efi
    fi
else
    . "${source_directory}"/modinfo.sh
    process_input_dir "${source_directory}" ${grub_modinfo_target_cpu}-${grub_modinfo_platform}
fi


# Bye.
exit 0
