#! /bin/bash
#################################################
#       Title:  mk-install-iso                  #
#        Date:  2014-11-26                      #
#     Version:  1.0                             #
#      Author:  dthaluru@vmware.com             #
#     Options:                                  #
#################################################
#   Overview
#       Generates a photon iso
#   End
#

set +x                 # disable hashall
source config.inc       #   configuration parameters
source function.inc     #   commonn functions
PRGNAME=${0##*/}    # script name minus the path
LOGFILE=/var/log/"${PRGNAME}-${LOGFILE}"    #   set log file name


# Grab the name of the iso file 
if [ $# -lt 2 ]
then
    echo "Usage : " $0 " <output-iso-with-path>  <rpms-path> <pkg-list-path>"
    exit 1
fi
ISO_OUTPUT_NAME=$1
RPMS_PATH=$2
PACKAGE_LIST_FILE=$3
RPM_LIST=$4
STAGE_PATH=$5
ADDITIONAL_FILES_TO_COPY_FROM_STAGE=$6
LIVE_CD=$7
OUTPUT_DATA_PATH=$8
PHOTON_COMMON_DIR=$(dirname "${PACKAGE_LIST_FILE}")
PACKAGE_LIST_FILE_BASE_NAME=$(basename "${PACKAGE_LIST_FILE}")
#- Step 3 Setting up the boot loader
WORKINGDIR=${BUILDROOT}
BUILDROOT=${BUILDROOT}/photon-chroot

run_command "cp isolinux to working directory: ${WORKINGDIR}" "cp -r BUILD_DVD/isolinux ${WORKINGDIR}/" "${LOGFILE}"
run_command "cp boot/grub2 to working directory: ${WORKINGDIR}" "cp -r BUILD_DVD/boot ${WORKINGDIR}/" "${LOGFILE}"
run_command "# 1" "mkdir ${WORKINGDIR}/boot/grub2/fonts/" "${LOGFILE}"
run_command "# 2" "cp boot/ascii.pf2 ${WORKINGDIR}/boot/grub2/fonts/" "${LOGFILE}"
run_command "# 3" "mkdir -p ${WORKINGDIR}/boot/grub2/themes/photon/" "${LOGFILE}"
run_command "# 4" "cp boot/splash.png ${WORKINGDIR}/boot/grub2/themes/photon/photon.png" "${LOGFILE}"
run_command "# 5" "cp boot/terminal_*.tga ${WORKINGDIR}/boot/grub2/themes/photon/" "${LOGFILE}"
run_command "# 6" "cp boot/theme.txt ${WORKINGDIR}/boot/grub2/themes/photon/" "${LOGFILE}"
run_command "echo : ${WORKINGDIR}" "echo ${WORKINGDIR}" "${LOGFILE}"
cp BUILD_DVD/isolinux/splash.png ${BUILDROOT}/installer/boot/.
mkdir -p ${BUILDROOT}/installer/EFI/BOOT
cp EFI_$(uname -m)/BOOT/* ${BUILDROOT}/installer/EFI/BOOT/

#Generate efiboot image
# efiboot is a fat16 image that has at least EFI/BOOT/bootx64.efi

# As a bootx64.efi we use shimx64.efi from shim-12
# # make VENDOR_CERT_FILE=<VMware cert> EFI_PATH=/usr/lib 'DEFAULT_LOADER=\\\\grubx64.efi' shimx64.efi
# # mv shimx64.efi bootx64.efi

# grubx64.efi is generated on Photon OS by using grub2-efi >= 2.02-7:
# # grub2-efi-mkimage -o grubx64.efi -p /boot/grub2 -O x86_64-efi  fat iso9660 part_gpt part_msdos  normal boot linux configfile loopback chain  efifwsetup efi_gop efi_uga  ls search search_label search_fs_uuid search_fs_file  gfxterm gfxterm_background gfxterm_menu test all_video loadenv  exfat ext2 udf halt gfxmenu png tga lsefi help linuxefi

# both bootx64.efi and grubx64.efi are signed with VMware key
EFI_IMAGE=boot/grub2/efiboot.img
EFI_FOLDER=`readlink -f ${STAGE_PATH}/efiboot`
dd if=/dev/zero of=${WORKINGDIR}/${EFI_IMAGE} bs=3K count=1024
mkdosfs ${WORKINGDIR}/${EFI_IMAGE}
mkdir $EFI_FOLDER
mount -o loop ${WORKINGDIR}/${EFI_IMAGE} $EFI_FOLDER
mkdir $EFI_FOLDER/EFI
cp -r ./EFI_$(uname -m)/BOOT $EFI_FOLDER/EFI
ls -lR $EFI_FOLDER
umount $EFI_FOLDER
rm -rf $EFI_FOLDER
#mcopy -s -i ${WORKINGDIR}/${EFI_IMAGE} ./EFI '::/'

if [ "$LIVE_CD" = true ] ; then
    mv ${WORKINGDIR}/isolinux/live-menu.cfg ${WORKINGDIR}/isolinux/menu.cfg
fi

cp sample_ks.cfg ${WORKINGDIR}/isolinux/

find ${BUILDROOT} -name linux-[0-9]*.rpm | head -1 | xargs rpm2cpio | cpio -iv --to-stdout ./boot/vmlinuz* > ${WORKINGDIR}/isolinux/vmlinuz

rm -f ${BUILDROOT}/installer/*.pyc
rm -rf ${BUILDROOT}/installer/BUILD_DVD
# Copy package list json files, dereference symlinks
cp -rf -L $OUTPUT_DATA_PATH/*.json ${BUILDROOT}/installer/
#ID in the initrd.gz now is PHOTON_VMWARE_CD . This is how we recognize that the cd is actually ours. touch this file there.
touch ${WORKINGDIR}/PHOTON_VMWARE_CD

# Step 4.5 Create necessary devices
mkfifo ${BUILDROOT}/dev/initctl
mknod ${BUILDROOT}/dev/ram0 b 1 0
mknod ${BUILDROOT}/dev/ram1 b 1 1
mknod ${BUILDROOT}/dev/ram2 b 1 2
mknod ${BUILDROOT}/dev/ram3 b 1 3
mknod ${BUILDROOT}/dev/sda b 8 0


#- Step 5 - Creating the boot script
mkdir -p ${BUILDROOT}/etc/systemd/scripts

# Step 6 create fstab

cp BUILD_DVD/fstab ${BUILDROOT}/etc/fstab

mkdir -p ${BUILDROOT}/etc/yum.repos.d
cat >> ${BUILDROOT}/etc/yum.repos.d/photon-iso.repo << EOF
[photon-iso]
name=VMWare Photon Linux 1.0(x86_64)
baseurl=file:///mnt/cdrom/RPMS
gpgkey=file:///etc/pki/rpm-gpg/VMWARE-RPM-GPG-KEY
gpgcheck=1
enabled=1
skip_if_unavailable=True
EOF

#- Step 7 - Create installer script
if [ "$LIVE_CD" = false ] ; then

cat >> ${BUILDROOT}/bin/bootphotoninstaller << EOF
#!/bin/bash
cd /installer
[ \`tty\` == '/dev/tty1' ] && ./isoInstaller.py --json-file=$PACKAGE_LIST_FILE_BASE_NAME 2> /var/log/installer && shutdown -r now
/bin/bash
EOF

chmod 755 ${BUILDROOT}/bin/bootphotoninstaller

fi

cat >> ${BUILDROOT}/init << EOF
mount -t proc proc /proc
/lib/systemd/systemd
EOF
chmod 755 ${BUILDROOT}/init

#adding autologin to the root user
# and set TERM=linux for installer
sed -i "s/ExecStart.*/ExecStart=-\/sbin\/agetty --autologin root --noclear %I linux/g" ${BUILDROOT}/lib/systemd/system/getty@.service

#- Step 7 - Create installer script
if [ "$LIVE_CD" = false ] ; then

    sed -i "s/root:.*/root:x:0:0:root:\/root:\/bin\/bootphotoninstaller/g" ${BUILDROOT}/etc/passwd

fi

mkdir -p ${BUILDROOT}/mnt/photon-root/photon-chroot
rm -rf ${BUILDROOT}/RPMS

run_command " echo ${RPMS_PATH}" "echo ${RPMS_PATH}" "${LOGFILE}"
#cp -r ${RPMS_PATH} ${WORKINGDIR}/
(
cd ${RPMS_PATH}
mkdir ${WORKINGDIR}/RPMS
for rpm_name in $RPM_LIST; do
    if [ -f "$rpm_name" ]; then
        cp --parent $rpm_name ${WORKINGDIR}/RPMS/;
    fi
done
)

# Work in sub-shell using ( ... ) to come back to original folder.
(
cd $STAGE_PATH
for file_name in $ADDITIONAL_FILES_TO_COPY_FROM_STAGE; do
    if [ -n "$file_name" ]; then
        cp $file_name ${WORKINGDIR}/;
    fi
done
)

#creating rpm repo in cd..
createrepo --database ${WORKINGDIR}/RPMS

repodatadir=${WORKINGDIR}/RPMS/repodata
if [ -d $repodatadir ]; then
    pushd $repodatadir
    metaDataFile=`find -type f -name "*primary.xml.gz"`
    ln -sfv $metaDataFile primary.xml.gz
    popd
fi

rm -rf ${BUILDROOT}/LOGS

if [ "$LIVE_CD" = false ] ; then
    # Cleaning up
    #Remove our rpm database as it fills up the ramdisk
    for filename in ${BUILDROOT}/usr/lib/*; do 
        #run_command " echo ${filename}" "echo ${filename}" "${LOGFILE}"
        if [[ -f ${filename} ]]; then
            file ${filename} | grep ELF >/dev/null 2>&1
            #run_command " file ${filename}" "echo ${filename}" "${LOGFILE}"
            if [[ $? -eq 0 ]]; then
                run_command " strip ${filename}" "strip ${filename}" "${LOGFILE}"
            fi;
        fi;
    done

    #Remove our rpm database as it fills up the ramdisk
    for filename in $(find ${BUILDROOT}/usr/lib/modules); do 
        #run_command " echo ${filename}" "echo ${filename}" "${LOGFILE}"
        if [[ -f ${filename} ]]; then
            file ${filename} | grep ELF >/dev/null 2>&1
            #run_command " file ${filename}" "echo ${filename}" "${LOGFILE}"
            if [[ $? -eq 0 ]]; then
                run_command " strip ${filename}" "strip ${filename}" "${LOGFILE}"
            fi;
        fi;
    done
    rm -rf ${BUILDROOT}/home/*
    rm -rf ${BUILDROOT}/var/lib/rpm

    # Remove the boot directory
    rm -rf ${BUILDROOT}/boot

    #Remove the include files.
    rm -rf ${BUILDROOT}/usr/include

    rm ${BUILDROOT}/lib64/libmvec*
    rm ${BUILDROOT}/usr/sbin/sln
    rm ${BUILDROOT}/usr/bin/oldfind

    rm ${BUILDROOT}/usr/bin/localedef
    rm ${BUILDROOT}/usr/bin/systemd-nspawn
    rm ${BUILDROOT}/usr/bin/systemd-analyze
    rm -rf ${BUILDROOT}/usr/lib64/gconv
    rm ${BUILDROOT}/usr/bin/sqlite3

    rm ${BUILDROOT}/usr/bin/bsdcpio
    rm ${BUILDROOT}/usr/bin/bsdtar
    rm ${BUILDROOT}/usr/bin/networkctl
    rm ${BUILDROOT}/usr/bin/machinectl
    rm ${BUILDROOT}/usr/bin/pkg-config
    rm ${BUILDROOT}/usr/bin/openssl
    rm ${BUILDROOT}/usr/bin/timedatectl
    rm ${BUILDROOT}/usr/bin/localectl
    rm ${BUILDROOT}/usr/bin/systemd-cgls
    rm ${BUILDROOT}/usr/bin/systemd-inhibit
    rm ${BUILDROOT}/usr/bin/systemd-studio-bridge
    rm ${BUILDROOT}/usr/bin/iconv

    rm -rf ${BUILDROOT}/usr/lib/python2.7/lib2to3
    rm -rf ${BUILDROOT}/usr/lib/python2.7/lib-tk
    rm -rf ${BUILDROOT}/usr/lib/python2.7/ensurepip
    rm -rf ${BUILDROOT}/usr/lib/python2.7/distutils
    rm -rf ${BUILDROOT}/usr/lib/python2.7/pydoc_data
    rm -rf ${BUILDROOT}/usr/lib/python2.7/idlelib
    rm -rf ${BUILDROOT}/usr/lib/python2.7/unittest 

    rm ${BUILDROOT}/usr/lib/librpmbuild.so*
    rm ${BUILDROOT}/usr/lib/libdb_cxx*
    rm ${BUILDROOT}/usr/lib/libnss_compat*

    rm ${BUILDROOT}/usr/bin/grub2-*
    rm ${BUILDROOT}/usr/lib/grub/i386-pc/*.module
    rm ${BUILDROOT}/usr/lib/grub/x86_64-efi/*.module

    for j in `ls ${BUILDROOT}/usr/sbin/grub2*`; do
        bsname=$(basename "$j")
        if [ $bsname != 'grub2-install' ]; then
            rm $j
        fi
    done

    # TODO: mbassiouny, Find a clean way to do that
    for i in `ls ${BUILDROOT}/usr/share/`; do
        if [ $i != 'terminfo' -a $i != 'cracklib' -a $i != 'grub' -a $i != 'factory' -a $i != 'dbus-1' ]; then
            rm -rf ${BUILDROOT}/usr/share/$i
        fi
    done

fi

# Set password max days to 99999 (disable aging)
sed -i '/^root.*/root::0:0:99999:7:::' "${BUILDROOT}/etc/shadow"

# Generate the intird
pushd $BUILDROOT
(find . | cpio -o -H newc --quiet | gzip -9 ) > ${WORKINGDIR}/isolinux/initrd.img
popd
rm -rf $BUILDROOT

#Step 9 Generate the ISO!!!!
pushd $WORKINGDIR
mkisofs -R -l -L -D -b isolinux/isolinux.bin -c isolinux/boot.cat \
        -no-emul-boot -boot-load-size 4 -boot-info-table \
        -eltorito-alt-boot -e ${EFI_IMAGE} -no-emul-boot \
        -V "PHOTON_$(date +%Y%m%d)" \
        $WORKINGDIR >$ISO_OUTPUT_NAME
popd