#!/bin/bash

if [ -z "$1" ]; then
  echo "Usage: $0 /dev/sdx1 [/dev/sdx2 [https://mirror.site/and/path]]"
  echo "Variables: FORCE_UPDATE=1  -  remove files before download to force update"
  echo "           SIZE=2          -  target size"
  exit 1
fi

set -e -x

if [ -z "$3" ]; then
  MIRROR="https://ftp.upjs.sk/pub"
else
  MIRROR="$3"
fi

PART=$1
LIVE=$2
DEV=${PART:0:-1}
MOUNT=/mnt/t
SIZE=${SIZE:-4}
FEDORA=40
FEDORA_PREV=$((FEDORA-1))
PRODUCT=Server
HDT=0.5.2
MEMTEST=5.01
#PMAGIC=`grep item pmagic.ipxe | head -1 | cut -d" " -f2`
PMAGIC=""
CFG=$MOUNT/syslinux/syslinux.cfg
DIR=$MOUNT/syslinux
EFIGRUB=$MOUNT/EFI/BOOT/grub.cfg
SALGRUB=$MOUNT/EFI/BOOT/salstar.cfg

copy() {
  # Skip copy if already present
  #FN=`basename $1`
  #if [ -d $1 ]; then
  #  if [ -f $1/$FN ]; then
  #    echo "$1/$FN already present, use force to overrwrite."
  #    return
  #  fi
  #fi
  rsync -crtvP --inplace "$@"
}

unlink() {
  if [ "$FORCE_UPDATE" ]; then
    rm -f "$@" || :
  fi
}

addos() {
  NAME=$1
  VERSION=$2
  ARCH=$3
  URL=$4
  #mkdir -p $DIR/$NAME/$ARCH
  #pushd $DIR/$NAME/$ARCH/
  mkdir -p $MOUNT/$NAME/$NAME$VERSION/$ARCH
  pushd $MOUNT/$NAME/$NAME$VERSION/$ARCH
  for i in initrd.img vmlinuz isolinux.bin grub.conf isolinux.cfg boot.msg memtest splash.png vesamenu.c32; do
    wget -c -4 --tries=2 $URL/isolinux/$i \
      || wget -c -4 --tries=2 $URL/images/pxeboot/$i \
      || echo "File missing: $i"
  done
  # update root=live: path
  #sed -i "s|root=live:CDLABEL=[^ ]*|repo=$URL|" isolinux.cfg
  # update syslinux files
  #cp -f $DIR/*.c32 ./

  OSDIR=/$1/$1$2/$3
  (
    #source /$1/$1$2/$3/grub.conf
    echo "menuentry '$1 $2 $3' --class gnu-linux --class gnu --class os {"
    echo "      linuxefi $OSDIR/vmlinuz initrd=initrd.img inst.repo=$URL"
    echo "      initrdefi $OSDIR/initrd.img"
    echo "      boot"
    echo "}"
  ) >> $SALGRUB

  popd
}

cfgos() {
  #cat >> $CFG << EOF
  #LABEL $1$2
  #  MENU LABEL $3 $2
  #  CONFIG $1/$2/isolinux.cfg $1/$2/
  #EOF
  true
}

cfgpmagic() {
if [ ! -f $DIR/pmagic$2/bzImage$3 ]; then
  echo "PMagic image not found, skipping config option..."
  return
fi
cat >> $CFG << EOF
LABEL pmagic$1
  MENU LABEL $4Parted Magic $PMAGIC $1
  KERNEL pmagic$2/bzImage$3
  INITRD pmagic$2/initrd.img
  APPEND edd=off load_ramdisk=1 prompt_ramdisk=0 rw vga=normal loglevel=9 max_loop=256 firewall
EOF
}

getmods() {
  echo `cd /usr/lib/grub/i386-pc; ls -1 $1.mod | sed 's/\\.mod//g'`
}

fedora_live() {
  #GRUBDIR=$MOUNT/boot/grub2/i386-pc
  GRUBDIR=$DIR/grub2
  CONFIG=$GRUBDIR/grub.cfg
  #grub2-mknetdir --net-directory=$MOUNT
  mkdir -p $GRUBDIR
  UUID=`dumpe2fs -h $LIVE 2>/dev/null | awk '/Filesystem UUID:/ { print $3 }'`
  (
    echo "set prefix=/boot/grub2"
    echo "search --no-floppy --fs-uuid --set=root $UUID"
    echo "normal /boot/grub2/grub.cfg"
  ) > $CONFIG
  DISK_MODS="pata scsi"
  USB_MODS="ahci ohci uhci usb usbms"
  FS_MODS="biosdisk ext2 btrfs xfs fat exfat ntfs"
  PART_MODS="part_msdos part_gpt"
  HW_MODS="at_keyboard usb_keyboard keystatus keylayouts sendkey"
  MODS="$DISK_MODS $USB_MODS $FS_MODS $PART_MODS $HW_MODS $USB_MODS"
  MODS="$MODS search normal"
  MODS="$MODS ls cat echo test gzio hexdump lspci help hdparm reboot"
  MODS="$MODS linux linux16 chain boot"
  MODS="$MODS net pxe pxechain"
  MODS="$MODS lvm `getmods '*raid*'`"
  MODS="$MODS serial loopback"
  grub2-mkimage -o $GRUBDIR/grub.0 -O i386-pc-pxe \
    --prefix='/boot/grub2' --config=$CONFIG $MODS
  rm -f $CONFIG
}

cfg_fedora_live() {
  if [ "$LIVE" ]; then
    echo "LABEL fedora_live"
    echo "  MENU DEFAULT"
    echo "  MENU LABEL ^Fedora $FEDORA Live"
    echo "  KERNEL grub2/grub.0"
  elif [ -d $MOUNT/LiveOS/syslinux ]; then
    echo "LABEL fedora_live"
    echo "  MENU DEFAULT"
    echo "  MENU LABEL ^Fedora $FEDORA Live"
    echo "  CONFIG ../LiveOS/syslinux/syslinux.cfg ../LiveOS/syslinux"
  fi
}

umount $MOUNT 2>/dev/null || true
umount $PART 2>/dev/null || true
[ "$LIVE" ] && umount $LIVE 2>/dev/null || true
#mkfs.vfat $PART

mount $PART $MOUNT
[ -f $DIR ] && rm -f $DIR
mkdir -p $DIR
umount $MOUNT

# install syslinux
syslinux --install --mbr --active --directory syslinux ${PART}
# copy boot sector
dd if=/usr/share/syslinux/mbr.bin of=${DEV} conv=notrunc bs=440 count=1
sync
sleep 1s

mount $PART $MOUNT

# IPXE menu
copy /usr/share/syslinux/menu.c32 $MOUNT/syslinux
copy /usr/share/syslinux/memdisk $MOUNT/syslinux
unlink $MOUNT/syslinux/hdt.iso
wget -c -4 --timeout=5 --tries=2 -O $MOUNT/syslinux/hdt.iso \
  https://ftp.upjs.sk/pub/mirrors/hdt/hdt-$HDT.iso \
  || echo "HDT download error"
#copy /boot/memtest86+-${MEMTEST} $MOUNT/syslinux/memtest
cat > $CFG << EOF
DEFAULT menu.c32
PROMPT 0

MENU TITLE SAL's BOOT MENU
MENU ROWS 14
MENU TABMSGROW 21
MENU CMDLINEROW 21
MENU TIMEOUTROW 23
TIMEOUT 300
TOTALTIMEOUT 9000

LABEL ipxe
  MENU LABEL ^iPXE menu
  KERNEL ipxe.lkrn
  APPEND -

`cfg_fedora_live`

LABEL hdt
  MENU LABEL ^HDT
  KERNEL memdisk
  INITRD hdt.iso
  APPEND iso

LABEL memtest
  MENU LABEL ^MemTest86+
  KERNEL memtest
  APPEND -

EOF
unlink $MOUNT/syslinux/ipxe.lkrn
wget --no-check-certificate -4 -O $MOUNT/syslinux/ipxe.lkrn \
  https://www.salstar.sk/boot/ipxe/ipxe.lkrn
wget -4 -O $MOUNT/ipxe/ipxe.efi \
  https://www.salstar.sk/boot/ipxe/com1/ipxe.efi

grep -q salstar $EFIGRUB || \
  echo "source /EFI/BOOT/salstar.cfg" >> $EFIGRUB

cat << EOF > $SALGRUB
menuentry 'iPXE' --class ipxe --class gnu --class os {
        chainloader /ipxe/ipxe.efi
        boot
}
EOF

#cp -a ~ondrejj/svn/pxeboot/ipxe/ipxe.lkrn $MOUNT/syslinux/

RELEASES=$MIRROR/fedora/linux/releases
if [ $SIZE -gt 1 ]; then
  addos fedora $FEDORA x86_64 $RELEASES/$FEDORA/$PRODUCT/x86_64/os
  cfgos fedora $FEDORA x86_64 "^Fedora $FEDORA"
fi

if [ $SIZE -gt 3 ]; then
  addos fedora $FEDORA_PREV x86_64 $RELEASES/$FEDORA_PREV/$PRODUCT/x86_64/os
  cfgos fedora $FEDORA_PREV x86_64 "Fedora $FEDORA_PREV"
fi

RELEASES=$MIRROR/mirrors/almalinux
if [ $SIZE -gt 1 ]; then
  addos almalinux 9 x86_64 $RELEASES/9/BaseOS/x86_64/os
  cfgos almalinux 9 x86_64 "^AlmaLinux 9"

  addos almalinux 8 x86_64 $RELEASES/8/BaseOS/x86_64/os
  cfgos almalinux 8 x86_64 "^AlmaLinux 8"

  #addos centos 8 x86_64 $RELEASES/8/BaseOS/x86_64/os
  #cfgos centos 8 x86_64 "^CentOS 8"

  #addos centos 7 x86_64 $RELEASES/7/os/x86_64
  #cfgos centos 7 x86_64 "^CentOS 7"

  #addos centos 6 x86_64 $RELEASES/6/os/x86_64
  #cfgos centos 6 x86_64 "CentOS 6"

  #addos centos 6 i386 $RELEASES/6/os/i386
  #cfgos centos 6 i386 "CentOS 6"
fi

# Parted Magic
if [ "$PMAGIC" ]; then
  RELEASES=$MIRROR/mirrors/pmagic
  #if [ $SIZE -gt 3 ]; then
  #  PM_ARCHS='x86_64 i586'
  #else
  #  PM_ARCHS=''
  #fi
  #for arch in $PM_ARCHS ""; do
  #  mkdir -p $DIR/pmagic/$arch
  #  unlink $DIR/pmagic/$arch/bzImage $DIR/pmagic/$arch/initrd.img
  #  if [ -f $DIR/pmagic/$arch/bzImage ]; then
  #    echo "Pmagic bzImage present, use FORCE_UPDATE=1 to overwrite."
  #  else
  #    for pmfn in bzImage initrd.img; do
  #      wget -4 -O $DIR/pmagic/$arch/$pmfn \
  #        $RELEASES/$arch/pmagic_pxe_$PMAGIC/pmagic/$pmfn
  #    done
  #  fi
  #done
  #cfgpmagic x86_64 "" ^
  #cfgpmagic ""
  #cfgpmagic i586
  mkdir -p $DIR/pmagic
  unlink $DIR/pmagic/bzImage $DIR/pmagic/bzImage64 $DIR/pmagic/initrd.img
  for pmfn in bzImage bzImage64 initrd.img; do
    wget -4 -O $DIR/pmagic/$pmfn $RELEASES/pmagic_pxe_$PMAGIC/pmagic/$pmfn
  done
  cfgpmagic x86_64 "" 64 ^
  cfgpmagic i686
fi

# Fedora Live (installed on HDD)
if [ "$LIVE" ]; then
  fedora_live
fi

find $DIR
df -h $DIR

umount $MOUNT

sync
echo 3 > /proc/sys/vm/drop_caches
sync