###############################################################################
#                                                                             #
# IPFire.org - A linux based firewall                                         #
# Copyright (C) 2007-2018  IPFire Team  <info@ipfire.org>                     #
#                                                                             #
# 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, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# 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.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

###############################################################################
# Definitions
###############################################################################

include Config

VER = ipfire

THISAPP    = flash-image
TARGET     = $(DIR_INFO)/$(THISAPP)

ifeq "$(MACHINE)" "i586"
	EFI = 1
endif

DEVICE     = $(shell losetup -f)
PART_BOOT  = /dev/mapper/$(patsubst /dev/%,%,$(DEVICE))p1
ifeq "$(EFI)" "1"
  PART_EFI = /dev/mapper/$(patsubst /dev/%,%,$(DEVICE))p2
endif
PART_ROOT  = /dev/mapper/$(patsubst /dev/%,%,$(DEVICE))p3

IMAGE_FILE = /install/images/$(SNAME)-$(VERSION).2gb-ext4.$(BUILD_ARCH)-full-core$(CORE).img.xz

FSTAB_FMT = UUID=%s %-8s %-4s %-10s %d %d\n

###############################################################################
# Top-level Rules
###############################################################################

install : $(TARGET)

check :

download :

md5 :

###############################################################################
# Installation Details
###############################################################################
MNThdd   := /install/harddisk
IMG      := /install/images/image.img

# All sizes in blocks
ifeq "$(BUILD_PLATFORM)" "arm"
 BOOTLOADER =
 S_OFFSET = 8192

 # FAT32
 PART_TYPE = c
else
 BOOTLOADER = grub
 S_OFFSET = 8192

 # Logical
 PART_TYPE = L
endif

GRUB_CFGS = /boot/grub/grub.cfg

ifeq "$(EFI)" "1"
 S_EFI = 204800
 GRUB_CFGS += /boot/efi/EFI/ipfire/grub-efi/grub.cfg
else
 S_EFI = 0
endif

# /boot:  128MB - OFFSET
# /    : 1200 MB
S_BOOT := $(shell echo $$(( 262144 - $(S_OFFSET) )))
S_ROOT := 2477600

PADDING = 100 # MB

ifeq "$(EFI)" "1"
 SFDISK_EFI  = $(S_OFFSET),$(S_EFI),c,*\n
 SFDISK      = $(SFDISK_BOOT)$(SFDISK_EFI)$(SFDISK_ROOT)
else
 SFDISK_SWAP = ,0,0\n
 SFDISK      = $(SFDISK_BOOT)$(SFDISK_SWAP)$(SFDISK_ROOT)
endif

SFDISK_BOOT = $(shell echo $$(( $(S_OFFSET) + $(S_EFI) ))),$(S_BOOT),$(PART_TYPE)\n
SFDISK_ROOT = $(shell echo $$(( $(S_OFFSET) + $(S_EFI) + $(S_BOOT) ))),$(S_ROOT),L\n

$(TARGET) : $(patsubst %,$(DIR_DL)/%,$(objects))
	# Stop if $(MNThdd) is still mounted
	mountpoint $(MNThdd) && exit 1 || exit 0

	rm -rf $(IMG) $(MNThdd) && mkdir -p $(MNThdd)

	# Allocate image on disk
	dd if=/dev/zero of=$(IMG) bs=512 count=$$(( $(S_OFFSET) + $(S_BOOT) + $(S_EFI) + $(S_ROOT) ))
	losetup $(DEVICE) $(IMG)

	# Write Partition table
	echo -e "$(SFDISK)" | sfdisk -uS $(DEVICE)

	# Remove 2nd empty partition
ifneq "$(EFI)" "1"
	-echo -e "d\n2\nw\n\q\n" | fdisk $(DEVICE)
endif

	kpartx -v -a $(DEVICE)

	# Format them
ifeq "$(PART_TYPE)" "c"
	mkfs.vfat $(PART_BOOT)
else
	mkfs.ext2 -F $(PART_BOOT)
endif
ifeq "$(EFI)" "1"
	mkfs.vfat $(PART_EFI)
endif
	mkfs.ext4 -O ^has_journal,extent -F $(PART_ROOT)

	# Most systems that use Flashimages has no RTC at boot
	# so the interval check should disables
	tune2fs -i0 $(PART_ROOT)

	# Mount root partition
	mount $(PART_ROOT) $(MNThdd)

	# Mount boot partition
	mkdir -pv $(MNThdd)/boot
	mount $(PART_BOOT) $(MNThdd)/boot

ifeq "$(EFI)" "1"
	mkdir -pv $(MNThdd)/boot/efi
	mount $(PART_EFI) $(MNThdd)/boot/efi
endif

	# Install Pandaboard MLO and uboot first
ifeq "$(BUILD_ARCH)" "armv5tel"
	cp -v /boot/MLO $(MNThdd)/boot/
	cp -v /boot/u-boot.img $(MNThdd)/boot/
	sync
	umount $(MNThdd)/boot
	mount $(PART_BOOT) $(MNThdd)/boot
endif

	# Install IPFire
ifneq "$(BUILD_PLATFORM)" "arm"
	tar -x --xz -C $(MNThdd)/ -f /install/cdrom/distro.img
else
	tar -x -C $(MNThdd)/ -f /install/cdrom/distro.img
endif
	-touch $(MNThdd)/lib/modules/$(KVER)-ipfire/modules.dep
	mkdir $(MNThdd)/proc
	mount --bind /proc $(MNThdd)/proc
	mount --bind /dev  $(MNThdd)/dev
	mount --bind /sys  $(MNThdd)/sys

	chroot $(MNThdd) /usr/bin/perl -e "require '/var/ipfire/lang.pl'; &Lang::BuildCacheLang"

	# Create /etc/fstab
	printf "$(FSTAB_FMT)" "$$(blkid -o value -s UUID $(PART_BOOT))" "/boot" \
		"auto" "defaults" 1 2 >  $(MNThdd)/etc/fstab
ifeq "$(EFI)" "1"
	printf "$(FSTAB_FMT)" "$$(blkid -o value -s UUID $(PART_EFI))" "/boot/efi" \
		"auto" "defaults" 1 2 >> $(MNThdd)/etc/fstab
endif
	printf "$(FSTAB_FMT)" "$$(blkid -o value -s UUID $(PART_ROOT))" "/" \
		"auto" "defaults" 1 1 >> $(MNThdd)/etc/fstab

ifeq "$(BOOTLOADER)" "grub"
	# backup defaults file
	cp $(MNThdd)/etc/default/grub $(MNThdd)/etc/default/grub.backup

	# Enable also serial console on GRUB
	echo "GRUB_TERMINAL=\"serial console\"" >> $(MNThdd)/etc/default/grub
	echo "GRUB_SERIAL_COMMAND=\"serial --unit=0 --speed=115200\"" >> $(MNThdd)/etc/default/grub

	# Add additional entry for Serial console
	cp $(DIR_SRC)/config/flash-images/grub/11_linux_scon \
			$(MNThdd)/etc/grub.d/

	# Create configuration
	for i in $(GRUB_CFGS); do \
		mkdir -pv $$(dirname $(MNThdd)$(GRUB_CFG)); \
		chroot $(MNThdd) grub-mkconfig -o $$i || exit 1; \
	done

	# Boot the first kernel by default
	chroot $(MNThdd) grub-set-default 0

	# Insert the UUID because grub-mkconfig often fails to
	# detect that correctly
	sed -i $(addprefix $(MNThdd),$(GRUB_CFGS)) \
	sed -i $(MNThdd)/boot/grub/grub.cfg \
		-e "s/root=[A-Za-z0-9\/=-]*/root=UUID=$$(blkid -o value -s UUID $(PART_ROOT))/g"

	# Install GRUB
	grub-install --force --recheck --no-floppy --target=i386-pc \
		--root-directory=$(MNThdd) $(DEVICE)

	# restore orginal defaults
	mv -f $(MNThdd)/etc/default/grub.backup $(MNThdd)/etc/default/grub
	rm -f $(MNThdd)/etc/grub.d/11_linux_scon

ifeq "$(EFI)" "1"
	mkdir -pv $(MNThdd)/boot/efi/EFI/boot
	cp -vf $(MNThdd)/boot/efi/EFI/ipfire/grub-efi/grub.efi \
		$(MNThdd)/boot/efi/EFI/boot/bootia32.efi
endif
endif

	# Set ramdisk mode to automatic
	echo RAMDISK_MODE=2 > $(MNThdd)/etc/sysconfig/ramdisk

	# Automatically resize the root partition to its maximum size at first boot
	touch $(MNThdd)/.partresize

	# Unmount
	umount $(MNThdd)/proc
	umount $(MNThdd)/sys
	umount $(MNThdd)/dev
ifeq "$(EFI)" "1"
	umount $(MNThdd)/boot/efi
endif
	umount $(MNThdd)/boot
	umount $(MNThdd)

	# zerofree the ext2 images to get better compression
ifneq "$(PART_TYPE)" "c"
	zerofree $(PART_BOOT)
	-fsck.ext2 -f -y $(PART_BOOT)
	fsck.ext2 -f -y $(PART_BOOT)
endif
	zerofree $(PART_ROOT)
	-fsck.ext4 -f -y $(PART_ROOT)
	fsck.ext4 -f -y $(PART_ROOT)

	sleep 10     #Ubuntu compiling: allow time to automount/dismount
	kpartx -d -v $(DEVICE)
	losetup -d $(DEVICE)

	# Add padding at the end of the image (to fix alignment issues if the image is
	# not copied to a block device)
	dd if=/dev/zero bs=1M count=$(PADDING) >> $(IMG)

ifeq "$(BUILD_ARCH)" "armv5tel"
	# Install u-boot for Orangepi Zero/Nanopi DUO into image 8KB
	dd if=/usr/share/u-boot/orangepi_zero/u-boot-sunxi-with-spl.bin of=$(IMG) bs=1K seek=8 conv=notrunc
endif

	# Compress Image
	xz $(XZ_OPT) < $(IMG) > $(IMAGE_FILE)
	rm -rf $(IMG) $(MNThdd)