#!/bin/bash
#
# This program 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; version 2 of the License.
#
# This program 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.

[[ -r ${LIBDIR}/util-iso-image.sh ]] && source ${LIBDIR}/util-iso-image.sh
[[ -r ${LIBDIR}/util-iso-calamares.sh ]] && source ${LIBDIR}/util-iso-calamares.sh

# $1: function
run_log(){

    logfile=${cache_dir_iso}/${buildset_iso}.log

    logpipe=$(mktemp -u "/tmp/logpipe.XXXXXXXX")
    mkfifo "$logpipe"

    tee "$logfile" < "$logpipe" &
    local teepid=$!

    $1 &> "$logpipe"

    wait $teepid
    rm "$logpipe"
}

check_run_dir(){
    if [[ ! -f shared/Packages ]];then
        die "${0##*/} is not run in a valid iso-profiles folder!"
    fi
}

check_profile(){
    if [[ ! -f $1/initsys ]]; then
        die "$1 is not a valid profile!"
    fi
}

copy_initcpio(){
    cp /usr/lib/initcpio/hooks/miso* $1/usr/lib/initcpio/hooks
    cp /usr/lib/initcpio/install/miso* $1/usr/lib/initcpio/install
    cp mkinitcpio.conf $1/etc/mkinitcpio-${manjaroiso}.conf
}

copy_overlay_root(){
    msg2 "Copying overlay ..."
    cp -a --no-preserve=ownership overlay/* $1
}

copy_overlay_custom(){
    msg2 "Copying ${custom}-overlay ..."
    cp -a --no-preserve=ownership ${custom}-overlay/* ${work_dir}/${custom}-image
}

copy_overlay_livecd(){
	msg2 "Copying overlay-livecd ..."
	if [[ -L overlay-livecd ]];then
	    cp -a --no-preserve=ownership overlay-livecd/* $1
	else
	    msg2 "Copying custom overlay-livecd ..."
	    cp -LR overlay-livecd/* $1
	fi
}

copy_startup_scripts(){
    msg2 "Copying startup scripts ..."
    cp ${PKGDATADIR}/scripts/livecd $1
    cp ${PKGDATADIR}/scripts/mhwd-live $1

    # fix script permissions
    chmod +x $1/livecd
    chmod +x $1/mhwd-live

#     cp ${BINDIR}/chroot-run $1

    # fix paths
#     sed -e "s|${LIBDIR}|/opt/livecd|g" -i $1/chroot-run
}

copy_livecd_helpers(){
    msg2 "Copying livecd helpers ..."
    [[ ! -d $1 ]] && mkdir -p $1
    cp ${LIBDIR}/util-livecd.sh $1


    cp ${LIBDIR}/util-msg.sh $1
    cp ${LIBDIR}/util.sh $1
#     cp ${LIBDIR}/util-mount.sh $1

    if [[ -f ${USER_CONFIG}/manjaro-tools.conf ]]; then
	msg2 "Copying ${USER_CONFIG}/manjaro-tools.conf ..."
	cp ${USER_CONFIG}/manjaro-tools.conf $1
    else
	msg2 "Copying ${manjaro_tools_conf} ..."
	cp ${manjaro_tools_conf} $1
    fi
    # write the custom var to conf to be sourced for use in util-livecd
    echo '' >> $1/manjaro-tools.conf
    echo '#Custom image name' >> $1/manjaro-tools.conf
    echo "custom=${custom}" >> $1/manjaro-tools.conf
}

copy_cache_lng(){
    msg2 "Copying lng cache ..."
    cp ${cache_dir_lng}/* ${work_dir}/lng-image/opt/livecd/lng
}

copy_cache_xorg(){
    msg2 "Copying xorg pkgs cache ..."
    cp ${cache_dir_xorg}/* ${work_dir}/pkgs-image/opt/livecd/pkgs
}

prepare_cachedirs(){
    [[ ! -d "${cache_dir_iso}" ]] && mkdir -p "${cache_dir_iso}"
    [[ ! -d "${cache_dir_xorg}" ]] && mkdir -p "${cache_dir_xorg}"
    [[ ! -d "${cache_dir_lng}" ]] && mkdir -p "${cache_dir_lng}"
}

clean_cache(){
    msg "Cleaning [$1] ..."
    find "$1" -name '*.pkg.tar.xz' -delete &>/dev/null
}

clean_up(){
    if [[ -d ${work_dir} ]];then
	msg "Removing [${work_dir}] ..."
	rm -r ${work_dir}
    fi
}

configure_custom_image(){
    msg3 "Configuring [${custom}-image]"

    configure_plymouth "$1"

    configure_displaymanager "$1"

    configure_services "$1"

    msg3 "Done configuring [${custom}-image]"
}

configure_livecd_image(){
    msg3 "Configuring [livecd-image]"

    configure_hostname "$1"

    configure_hosts "$1"

    configure_accountsservice "$1" "${username}"

    configure_user "$1"

    configure_services_live "$1"

    configure_calamares "$1"

    configure_thus "$1"

    msg3 "Done configuring [livecd-image]"
}

# $1: work_dir
gen_boot_image(){
	local _kernver=$(cat $1/usr/lib/modules/*-MANJARO/version)
        chroot-run $1 \
		  /usr/bin/mkinitcpio -k ${_kernver} \
		  -c /etc/mkinitcpio-${manjaroiso}.conf \
		  -g /boot/${img_name}.img
}

make_repo(){
    repo-add ${work_dir}/pkgs-image/opt/livecd/pkgs/gfx-pkgs.db.tar.gz ${work_dir}/pkgs-image/opt/livecd/pkgs/*pkg*z
}

# $1: work dir
# $2: cache dir
# $3: pkglist
download_to_cache(){
    pacman -v --config "${pacman_conf}" \
	      --arch "${arch}" --root "$1" \
	      --cache $2 \
	      -Syw $3 --noconfirm
}

# Build ISO
make_iso() {
    msg "Start [Build ISO]"
    touch "${work_dir}/iso/.miso"

    mkiso ${iso_args[*]} iso "${work_dir}" "${cache_dir_iso}/${iso_file}"

    chown -R "${OWNER}:users" "${cache_dir_iso}"
    msg "Done [Build ISO]"
}

# $1: file
make_checksum(){
    cd ${cache_dir_iso}
        msg "Creating [${checksum_mode}sum] ..."
        local cs=$(${checksum_mode}sum $1)
        msg2 "${checksum_mode}sum: ${cs}"
        echo "${cs}" > $1.${checksum_mode}
        msg "Done [${checksum_mode}sum]"
    cd ..
}

# $1: new branch
aufs_mount_root_image(){
    msg2 "mount [root-image] on [${1##*/}]"
    mount -t aufs -o br="$1":${work_dir}/root-image=ro none "$1"
}

# $1: add branch
aufs_append_root_image(){
    msg2 "append [root-image] on [${1##*/}]"
    mount -t aufs -o remount,append:${work_dir}/root-image=ro none "$1"
}

# $1: add branch
aufs_mount_custom_image(){
    msg2 "mount [${1##*/}] on [${custom}-image]"
    mount -t aufs -o br="$1":${work_dir}/${custom}-image=ro none "$1"
}

# $1: del branch
aufs_remove_image(){
    if mountpoint -q "$1";then
        msg2 "unmount ${1##*/}"
	umount $1
    fi
}

umount_image_handler(){
    aufs_remove_image "${work_dir}/livecd-image"
    aufs_remove_image "${work_dir}/${custom}-image"
    aufs_remove_image "${work_dir}/root-image"
    aufs_remove_image "${work_dir}/pkgs-image"
    aufs_remove_image "${work_dir}/lng-image"
    aufs_remove_image "${work_dir}/boot-image"
}

mkiso_error_handler(){
    umount_image_handler
    die "Exiting..."
}

# Base installation (root-image)
make_image_root() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

	msg "Prepare [Base installation] (root-image)"

	local path="${work_dir}/root-image"

	mkiso ${create_args[*]} -p "${packages}" -i "root-image" create "${work_dir}" || mkiso_error_handler

	pacman -Qr "${path}" > "${path}/root-image-pkgs.txt"

	if [ -e ${path}/boot/grub/grub.cfg ] ; then
	    rm ${path}/boot/grub/grub.cfg
	fi
	if [ -e ${path}/etc/lsb-release ] ; then
	    sed -i -e "s/^.*DISTRIB_RELEASE.*/DISTRIB_RELEASE=${iso_version}/" ${path}/etc/lsb-release
	    sed -i -e "s/^.*DISTRIB_CODENAME.*/DISTRIB_CODENAME=${code_name}/" ${path}/etc/lsb-release
	fi

	copy_overlay_root "${path}"

	: > ${work_dir}/build.${FUNCNAME}

	msg "Done [Base installation] (root-image)"
    fi
}

make_image_custom() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

	msg "Prepare [${custom} installation] (${custom}-image)"

	local path="${work_dir}/${custom}-image"

	mkdir -p ${path}

	umount_image_handler

	aufs_mount_root_image "${path}"

	mkiso ${create_args[*]} -i "${custom}-image" -p "${packages}" create "${work_dir}" || mkiso_error_handler

	pacman -Qr "${path}" > "${path}/${custom}-image-pkgs.txt"

	cp "${path}/${custom}-image-pkgs.txt" ${cache_dir_iso}/${img_name}-${custom}-${iso_version}-${arch}-pkgs.txt

	[[ -d ${custom}-overlay ]] && copy_overlay_custom

	configure_custom_image "${path}"

        umount_image_handler

	rm -R ${path}/.wh*

	: > ${work_dir}/build.${FUNCNAME}

	msg "Done [${custom} installation] (${custom}-image)"
    fi
}

make_image_livecd() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

	msg "Prepare [livecd-image]"

	local path="${work_dir}/livecd-image"

	mkdir -p ${path}

        umount_image_handler

	if [[ -n "${custom}" ]] ; then
	    aufs_mount_custom_image "${path}"
	    aufs_append_root_image "${path}"
	else
	    aufs_mount_root_image "${path}"
	fi

	mkiso ${create_args[*]} -i "livecd-image" -p "${packages}" create "${work_dir}" || mkiso_error_handler

	pacman -Qr "${path}" > "${path}/livecd-image-pkgs.txt"

	copy_overlay_livecd "${path}"

	# copy over setup helpers and config loader
        copy_livecd_helpers "${path}/opt/livecd"

        copy_startup_scripts "${path}/usr/bin"

	configure_livecd_image "${path}"

	# Clean up GnuPG keys?
	rm -rf "${path}/etc/pacman.d/gnupg"

        umount_image_handler

	rm -R ${path}/.wh*

        : > ${work_dir}/build.${FUNCNAME}

	msg "Done [livecd-image]"
    fi
}

make_image_xorg() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

	msg "Prepare [pkgs-image]"

	local path="${work_dir}/pkgs-image"

	mkdir -p ${path}/opt/livecd/pkgs

	umount_image_handler

	if [[ -n "${custom}" ]] ; then
	    aufs_mount_custom_image "${path}"
	    aufs_append_root_image "${path}"
	else
	    aufs_mount_root_image "${path}"
	fi

	download_to_cache "${path}" "${cache_dir_xorg}" "${packages_xorg}"
	copy_cache_xorg

	if [[ -n "${packages_xorg_cleanup}" ]]; then
	    for xorg_clean in ${packages_xorg_cleanup}; do
		rm ${path}/opt/livecd/pkgs/${xorg_clean}
	    done
	fi

	cp ${PKGDATADIR}/pacman-gfx.conf ${path}/opt/livecd
	rm -r ${path}/var

	make_repo "${path}/opt/livecd/pkgs/gfx-pkgs" "${path}/opt/livecd/pkgs"

	configure_xorg_drivers "${path}"

        umount_image_handler

	rm -R ${path}/.wh*

	: > ${work_dir}/build.${FUNCNAME}

	msg "Done [pkgs-image]"
    fi
}

make_image_lng() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

	msg "Prepare [lng-image]"

	local path="${work_dir}/lng-image"

	mkdir -p ${path}/opt/livecd/lng

	umount_image_handler

	if [[ -n "${custom}" ]] ; then
	    aufs_mount_custom_image "${path}"
	    aufs_append_root_image "${path}"
	else
	    aufs_mount_root_image "${path}"
	fi

	if [[ -n ${packages_lng_kde} ]]; then
	    download_to_cache "${path}" "${cache_dir_lng}" "${packages_lng} ${packages_lng_kde}"
	    copy_cache_lng
	else
	    download_to_cache "${path}" "${cache_dir_lng}" "${packages_lng}"
	    copy_cache_lng
	fi

	if [[ -n "${packages_lng_cleanup}" ]]; then
	    for lng_clean in ${packages_lng_cleanup}; do
		rm ${path}/opt/livecd/lng/${lng_clean}
	    done
	fi

	cp ${PKGDATADIR}/pacman-lng.conf ${path}/opt/livecd
	rm -r ${path}/var

	make_repo ${path}/opt/livecd/lng/lng-pkgs ${path}/opt/livecd/lng

	umount_image_handler

	rm -R ${path}/.wh*

	: > ${work_dir}/build.${FUNCNAME}

	msg "Done [lng-image]"
    fi
}

# Prepare ${install_dir}/boot/
make_image_boot() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then
	msg "Prepare [${install_dir}/boot]"

	local path_iso="${work_dir}/iso/${install_dir}/boot"

	mkdir -p ${path_iso}/${arch}

        cp ${work_dir}/root-image/boot/memtest86+/memtest.bin ${path_iso}/${arch}/memtest

	cp ${work_dir}/root-image/boot/vmlinuz* ${path_iso}/${arch}/${manjaroiso}

        local path="${work_dir}/boot-image"

        mkdir -p ${path}

        umount_image_handler

        if [[ -n "${custom}" ]] ; then
	    aufs_mount_custom_image "${path}"
	    aufs_append_root_image "${path}"
	else
	    aufs_mount_root_image "${path}"
        fi

        copy_initcpio "${path}"

        gen_boot_image "${path}"

        mv ${path}/boot/${img_name}.img ${path_iso}/${arch}/${img_name}.img
        cp ${path}/boot/intel-ucode.img ${path_iso}/intel_ucode.img
        cp ${path}/usr/share/licenses/intel-ucode/LICENSE ${path_iso}/intel_ucode.LICENSE

        umount_image_handler

        rm -R ${path}

	: > ${work_dir}/build.${FUNCNAME}

	msg "Done [${install_dir}/boot]"
    fi
}

# Prepare /EFI
make_efi() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

        msg "Prepare [${install_dir}/boot/EFI]"

        local path_iso="${work_dir}/iso"
        local path_efi="${path_iso}/EFI"

        mkdir -p ${path_efi}/boot

        cp ${work_dir}/root-image/usr/lib/prebootloader/PreLoader.efi ${path_efi}/boot/bootx64.efi
        cp ${work_dir}/root-image/usr/lib/prebootloader/HashTool.efi ${path_efi}/boot/
        cp ${work_dir}/root-image/usr/lib/gummiboot/gummibootx64.efi ${path_efi}/boot/loader.efi

        mkdir -p ${path_iso}/loader/entries

        cp efiboot/loader/loader.conf ${path_iso}/loader/
        cp efiboot/loader/entries/uefi-shell-v2-x86_64.conf ${path_iso}/loader/entries/
        cp efiboot/loader/entries/uefi-shell-v1-x86_64.conf ${path_iso}/loader/entries/

        sed "s|%MISO_LABEL%|${iso_label}|g;
             s|%INSTALL_DIR%|${install_dir}|g" \
            efiboot/loader/entries/${manjaroiso}-x86_64-usb.conf > ${path_iso}/loader/entries/${manjaroiso}-x86_64.conf

        sed "s|%MISO_LABEL%|${iso_label}|g;
             s|%INSTALL_DIR%|${install_dir}|g" \
            efiboot/loader/entries/${manjaroiso}-x86_64-nonfree-usb.conf > ${path_iso}/loader/entries/${manjaroiso}-x86_64-nonfree.conf

        # EFI Shell 2.0 for UEFI 2.3+ ( http://sourceforge.net/apps/mediawiki/tianocore/index.php?title=UEFI_Shell )
        curl -k -o ${path_efi}/shellx64_v2.efi https://svn.code.sf.net/p/edk2/code/trunk/edk2/ShellBinPkg/UefiShell/X64/Shell.efi
        # EFI Shell 1.0 for non UEFI 2.3+ ( http://sourceforge.net/apps/mediawiki/tianocore/index.php?title=Efi-shell )
        curl -k -o ${path_efi}/shellx64_v1.efi https://svn.code.sf.net/p/edk2/code/trunk/edk2/EdkShellBinPkg/FullShell/X64/Shell_Full.efi

        : > ${work_dir}/build.${FUNCNAME}

	msg "Done [${install_dir}/boot/EFI]"
    fi
}

# Prepare kernel.img::/EFI for "El Torito" EFI boot mode
make_efiboot() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then

        msg "Prepare [${install_dir}/iso/EFI]"

        local path_iso="${work_dir}/iso"

        mkdir -p ${path_iso}/EFI/miso

        truncate -s 31M ${path_iso}/EFI/miso/${img_name}.img

        mkfs.vfat -n MISO_EFI ${path_iso}/EFI/miso/${img_name}.img

        mkdir -p ${work_dir}/efiboot

        mount ${path_iso}/EFI/miso/${img_name}.img ${work_dir}/efiboot

        local path_efi="${work_dir}/efiboot/EFI"

        mkdir -p ${path_efi}/miso

        cp ${path_iso}/${install_dir}/boot/x86_64/${manjaroiso} ${path_efi}/miso/${manjaroiso}.efi
        cp ${path_iso}/${install_dir}/boot/x86_64/${img_name}.img ${path_efi}/miso/${img_name}.img
        cp ${path_iso}/${install_dir}/boot/intel_ucode.img ${path_efi}/miso/intel_ucode.img

        mkdir -p ${path_efi}/boot

        cp ${work_dir}/root-image/usr/lib/prebootloader/PreLoader.efi ${path_efi}/boot/bootx64.efi
        cp ${work_dir}/root-image/usr/lib/prebootloader/HashTool.efi ${path_efi}/boot/
        cp ${work_dir}/root-image/usr/lib/gummiboot/gummibootx64.efi ${path_efi}/boot/loader.efi

        mkdir -p ${work_dir}/efiboot/loader/entries

        cp efiboot/loader/loader.conf ${work_dir}/efiboot/loader/
        cp efiboot/loader/entries/uefi-shell-v2-x86_64.conf ${work_dir}/efiboot/loader/entries/
        cp efiboot/loader/entries/uefi-shell-v1-x86_64.conf ${work_dir}/efiboot/loader/entries/

        sed "s|%MISO_LABEL%|${iso_label}|g;
             s|%INSTALL_DIR%|${install_dir}|g" \
            efiboot/loader/entries/${manjaroiso}-x86_64-dvd.conf > ${work_dir}/efiboot/loader/entries/${manjaroiso}-x86_64.conf

        sed "s|%MISO_LABEL%|${iso_label}|g;
             s|%INSTALL_DIR%|${install_dir}|g" \
            efiboot/loader/entries/${manjaroiso}-x86_64-nonfree-dvd.conf > ${work_dir}/efiboot/loader/entries/${manjaroiso}-x86_64-nonfree.conf

        cp ${path_iso}/EFI/shellx64_v2.efi ${path_efi}/
        cp ${path_iso}/EFI/shellx64_v1.efi ${path_efi}/

        umount ${work_dir}/efiboot

        : > ${work_dir}/build.${FUNCNAME}

	msg "Done [${install_dir}/iso/EFI]"
    fi
}

# Prepare /isolinux
make_isolinux() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then
	msg "Prepare [${install_dir}/iso/isolinux]"

	mkdir -p ${work_dir}/iso/isolinux
        cp -a --no-preserve=ownership isolinux/* ${work_dir}/iso/isolinux

        if [[ -e isolinux-overlay ]]; then
	    msg2 "isolinux overlay found. Overwriting files."
            cp -a --no-preserve=ownership isolinux-overlay/* ${work_dir}/iso/isolinux
        fi

        local path="${work_dir}/root-image/usr/lib/syslinux"

        if [[ -e ${path}/bios/ ]]; then
            cp ${path}/bios/isolinux.bin ${work_dir}/iso/isolinux/
            cp ${path}/bios/isohdpfx.bin ${work_dir}/iso/isolinux/
            cp ${path}/bios/ldlinux.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/gfxboot.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/whichsys.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/mboot.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/hdt.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/chain.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/libcom32.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/libmenu.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/libutil.c32 ${work_dir}/iso/isolinux/
            cp ${path}/bios/libgpl.c32 ${work_dir}/iso/isolinux/
        else
            cp ${path}/isolinux.bin ${work_dir}/iso/isolinux/
            cp ${path}/isohdpfx.bin ${work_dir}/iso/isolinux/
            cp ${path}/gfxboot.c32 ${work_dir}/iso/isolinux/
            cp ${path}/whichsys.c32 ${work_dir}/iso/isolinux/
            cp ${path}/mboot.c32 ${work_dir}/iso/isolinux/
            cp ${path}/hdt.c32 ${work_dir}/iso/isolinux/
            cp ${path}/chain.c32 ${work_dir}/iso/isolinux/
        fi

        sed -i "s|%MISO_LABEL%|${iso_label}|g;
                s|%INSTALL_DIR%|${install_dir}|g;
                s|%ARCH%|${arch}|g" ${work_dir}/iso/isolinux/isolinux.cfg

        : > ${work_dir}/build.${FUNCNAME}

	msg "Done [${install_dir}/iso/isolinux]"
    fi
}

gen_isomounts(){
        echo '# syntax: <img> <arch> <mount point> <type> <kernel argument>' > $1
        echo '# Sample kernel argument in syslinux: overlay=extra,extra2' >> $1
        echo '' >> $1

        msg2 "Writing livecd entry ..."
        echo "${arch}/livecd-image.sqfs ${arch} / squashfs" >> $1

        if [[ -f Packages-Lng ]] ; then
            msg2 "Writing lng entry ..."
            echo "${arch}/lng-image.sqfs ${arch} / squashfs" >> $1
        fi

        if [[ -f Packages-Xorg ]] ; then
            msg2 "Writing pkgs entry ..."
            echo "${arch}/pkgs-image.sqfs ${arch} / squashfs" >> $1
        fi

        if [[ -f "${packages_custom}" ]] ; then
            msg2 "Writing ${custom} entry ..."
            echo "${arch}/${custom}-image.sqfs ${arch} / squashfs" >> $1
        fi

        msg2 "Writing root entry ..."
        echo "${arch}/root-image.sqfs ${arch} / squashfs" >> $1
}

make_isomounts() {
    if [[ ! -e ${work_dir}/build.${FUNCNAME} ]]; then
	msg "Creating [isomounts]"

        gen_isomounts "${work_dir}/iso/${install_dir}/isomounts"

        : > ${work_dir}/build.${FUNCNAME}

	msg "Done creating [isomounts]"
    fi
}

# $1: file name
load_pkgs(){
    msg3 "Loading Packages: [$1] ..."

    if [[ "${arch}" == "i686" ]]; then
	packages=$(sed "s|#.*||g" "$1" | sed "s| ||g" | sed "s|>dvd.*||g"  | sed "s|>blacklist.*||g" | sed "s|>x86_64.*||g" | sed "s|>i686||g" | sed "s|KERNEL|$manjaro_kernel|g" | sed ':a;N;$!ba;s/\n/ /g')
    elif [[ "${arch}" == "x86_64" ]]; then
	packages=$(sed "s|#.*||g" "$1" | sed "s| ||g" | sed "s|>dvd.*||g"  | sed "s|>blacklist.*||g" | sed "s|>i686.*||g" | sed "s|>x86_64||g" | sed "s|KERNEL|$manjaro_kernel|g" | sed ':a;N;$!ba;s/\n/ /g')
    fi
}

load_pkgs_xorg(){
    msg3 "Loading Packages: [Packages-Xorg] ..."

    if [[ "${arch}" == "i686" ]]; then
	packages_xorg=$(sed "s|#.*||g" Packages-Xorg | sed "s| ||g" | sed "s|>dvd.*||g"  | sed "s|>blacklist.*||g" | sed "s|>cleanup.*||g" | sed "s|>x86_64.*||g" | sed "s|>i686||g" | sed "s|>free_x64.*||g" | sed "s|>free_uni||g" | sed "s|>nonfree_x64.*||g" | sed "s|>nonfree_uni||g" | sed "s|KERNEL|$manjaro_kernel|g" | sed ':a;N;$!ba;s/\n/ /g')
    elif [[ "${arch}" == "x86_64" ]]; then
	packages_xorg=$(sed "s|#.*||g" Packages-Xorg | sed "s| ||g" | sed "s|>dvd.*||g"  | sed "s|>blacklist.*||g" | sed "s|>cleanup.*||g" | sed "s|>i686.*||g" | sed "s|>x86_64||g" | sed "s|>free_x64||g" | sed "s|>free_uni||g" | sed "s|>nonfree_uni||g" | sed "s|>nonfree_x64||g" | sed "s|KERNEL|$manjaro_kernel|g" | sed ':a;N;$!ba;s/\n/ /g')
    fi
    packages_xorg_cleanup=$(sed "s|#.*||g" Packages-Xorg | grep cleanup | sed "s|>cleanup||g" | sed "s|KERNEL|$manjaro_kernel|g" | sed ':a;N;$!ba;s/\n/ /g')
}

load_pkgs_lng(){
    msg3 "Loading Packages: [Packages-Lng] ..."

    if [[ "${arch}" == "i686" ]]; then
	packages_lng=$(sed "s|#.*||g" Packages-Lng | sed "s| ||g" | sed "s|>dvd.*||g"  | sed "s|>blacklist.*||g" | sed "s|>cleanup.*||g" | sed "s|>x86_64.*||g" | sed "s|>i686||g" | sed "s|>kde.*||g" | sed ':a;N;$!ba;s/\n/ /g')
    elif [[ "${arch}" == "x86_64" ]]; then
	packages_lng=$(sed "s|#.*||g" Packages-Lng | sed "s| ||g" | sed "s|>dvd.*||g"  | sed "s|>blacklist.*||g" | sed "s|>cleanup.*||g" | sed "s|>i686.*||g" | sed "s|>x86_64||g" | sed "s|>kde.*||g" | sed ':a;N;$!ba;s/\n/ /g')
    fi
    packages_lng_cleanup=$(sed "s|#.*||g" Packages-Lng | grep cleanup | sed "s|>cleanup||g")
    packages_lng_kde=$(sed "s|#.*||g" Packages-Lng | grep kde | sed "s|>kde||g" | sed ':a;N;$!ba;s/\n/ /g')
}

# $1: profile
load_profile(){
    msg3 "Profile: [$1] ..."

    local files=$(ls Packages*)

    for f in ${files[@]};do
        case $f in
            Packages|Packages-Livecd*|Packages-Xorg*|Packages-Lng*) continue ;;
            *) packages_custom="$f"; msg2 "Packages-Custom: $f" ;;
        esac
    done

    custom=${packages_custom#*-}
    custom=${custom,,}

    displaymanager="$(cat displaymanager)"
    initsys="$(cat initsys)"

    iso_file="${img_name}-${custom}-${iso_version}-${arch}.iso"

    if [[ -f pacman-${pacman_conf_arch}.conf ]]; then
	pacman_conf="pacman-${pacman_conf_arch}.conf"
    else
	pacman_conf="${PKGDATADIR}/pacman-${pacman_conf_arch}.conf"
    fi
    create_args+=(-C ${pacman_conf})

    work_dir=${chroots_iso}/$1/${arch}

    is_plymouth=false
    source mkinitcpio.conf
    for h in ${HOOKS[@]};do
        if [[ $h == 'plymouth' ]];then
            is_plymouth=true
        fi
    done
}

compress_images(){
    make_iso
    make_checksum "${iso_file}"
}

build_images(){
    load_pkgs "Packages"
    make_image_root

    if [[ -f "${packages_custom}" ]] ; then
	load_pkgs "${packages_custom}"
	make_image_custom
    fi

    if [[ -f Packages-Xorg ]] ; then
	load_pkgs_xorg
	make_image_xorg
    fi

    if [[ -f Packages-Lng ]] ; then
	load_pkgs_lng
	make_image_lng
    fi

    if [[ -f Packages-Livecd ]]; then
	load_pkgs "Packages-Livecd"
	make_image_livecd
    fi

    make_image_boot
    if [[ "${arch}" == "x86_64" ]]; then
	make_efi
	make_efiboot
    fi
    make_isolinux
    make_isomounts
}

build_profile(){
    ${clean_first} && clean_up

    ${clean_cache_xorg} && clean_cache "${cache_dir_xorg}"
    ${clean_cache_lng} && clean_cache "${cache_dir_lng}"

    if ${iso_only}; then
	[[ ! -d ${work_dir} ]] && die "You need to create images first eg. buildiso -p <name> -i"
	compress_images
	exit 1
    fi

    if ${images_only}; then
	build_images
	warning "Continue with eg. buildiso -p <name> -sc ..."
	exit 1
    else
	build_images
	compress_images
    fi
}

build_iso(){
    if ${is_buildset};then
	msg3 "Start building [${buildset_iso}]"
	for prof in $(cat ${sets_dir_iso}/${buildset_iso}.set); do
# 	    [[ ! -f $prof/initsys ]] && break
            check_profile "$prof"
	    cd $prof
		load_profile "$prof"
		build_profile
	    cd ..
	done
	msg3 "Finished building [${buildset_iso}]"
    else
# 	[[ ! -f ${buildset_iso}/initsys ]] && die "${buildset_iso} is not a valid profile!"
        check_profile "${buildset_iso}"
	cd ${buildset_iso}
	    load_profile "${buildset_iso}"
	    build_profile
	cd ..
    fi
}