#! /bin/sh

PREREQ=""
DESCRIPTION="Setting up locales..."

. /scripts/casper-functions

prereqs()
{
       echo "$PREREQ"
}

case $1 in
# get pre-requisites
prereqs)
       prereqs
       exit 0
       ;;
esac

log_begin_msg "$DESCRIPTION"

if [ -e /root/etc/default/locale ]; then
    grep_file=/root/etc/default/locale
elif [ -e /root/etc/environment ]; then # Old locales policy
    grep_file=/root/etc/environment
fi

if [ -n "${grep_file}" ]; then
    # use rootfs configured locale
    locale=$(grep -s 'LANG=' ${grep_file} | sed s/'LANG='// | tr -d '"' )
else
    grep_file=/root/etc/default/locale
fi

# commandline
if [ ! -z "${LOCALE}" ]; then
    locale="${LOCALE}"
    set_locale="true"
fi

if [ -z "${locale}" ]; then
    # Set a default one
    locale=en_US.UTF-8
    set_locale="true"
fi

if [ "${set_locale}" ]; then
    if echo "${locale}" | grep -sqE '^[[:lower:]]{2}$' ; then
        # input is like "locale=it", so we will convert and setup also the keyboard if not already set
        if [ -z "${KBD}" ]; then
            # FIXME: look if this keyb is supported
            KBD="${locale}"
            really_export KBD
        fi
        uploc=$(echo "${locale}" | tr '[a-z]' '[A-Z]')
        locale="${locale}_${uploc}.UTF-8"
    fi

    LANG=$(grep "^${locale}" /root/usr/share/i18n/SUPPORTED | grep UTF-8 |sed -e 's, .*,,' -e q)
    if [ -z "${LANG}" ]; then
        log_warning_message "Locale ${locale} is unsupported."
        locale="en_US.UTF-8"
        LANG="${locale}"
    fi

    really_export LANG

    if [ "${BUILD_SYSTEM}" == "Ubuntu" ]; then
        printf 'LANG="%s"\n' "${LANG}" > "${grep_file}"
        chroot /root /usr/sbin/locale-gen "${LANG}"
        casper-preseed /root debian-installer/locale "${locale}"
    else
        printf 'LANG=%s\n' "${LANG}" > "${grep_file}"
        printf '%s UTF-8\n' "${LANG}" > /root/etc/locale.gen
        chroot /root /usr/sbin/locale-gen
    fi
fi

log_end_msg
