#!/usr/bin/env zsh
# Copyright (c) 2016-2021 Ivan J. <parazyd@dyne.org>
# This file is part of arm-sdk
#
# This source code 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 software 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 source code. If not, see <http://www.gnu.org/licenses/>.

## generic kernel build script for sunxi allwinner boards
##  http://linux-sunxi.org

## settings & config
vars+=(device_name arch size parted_type parted_boot parted_root bootfs inittab)
vars+=(gitkernel gitbranch)
arrs+=(custmodules)

device_name="sunxi"
arch="armhf"
size=1891
inittab=("T1:12345:respawn:/sbin/agetty -L ttyS0 115200 vt100")

parted_type="dos"
bootfs="ext2"
rootfs="ext4"
dos_boot="$bootfs 2048s 264191s"
dos_root="$rootfs 264192s 100%"

extra_packages+=()
custmodules=()

gitkernel="https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux-stable.git"
gitbranch="linux-5.11.y"

prebuild() {
	fn prebuild
    req=(device_name)
    ckreq || return 1

    notice "executing $device_name prebuild"

	copy-root-overlay

	mkdir -p $R/tmp/kernels/$device_name
}

postbuild() {
    fn postbuild
	req=(uboot_configs device_name compiler)
	ckreq || return 1

    notice "executing $device_name postbuild"

    notice "building u-boot"
	mkdir -p $R/dist/u-boot
	pushd $R/extra/u-boot
		for board in $uboot_configs; do
			notice "building u-boot for $board"

			make distclean
			make \
				$MAKEOPTS \
				ARCH=arm \
				CROSS_COMPILE=$compiler \
					"$board" || { zerr; return 1; }
			make \
				$MAKEOPTS \
				ARCH=arm \
				CROSS_COMPILE=$compiler || { zerr; return 1; }

			mv -v u-boot-sunxi-with-spl.bin $R/dist/u-boot/${board}.bin
		done
    popd

    notice "creating boot.cmd"
    cat <<EOF | sudo tee ${strapdir}/boot/boot.cmd
setenv bootargs console=ttyS0,115200 root=/dev/mmcblk0p2 rootwait panic=10 \${extra}
load mmc 0:1 0x43000000 dtbs/\${fdtfile} || load mmc 0:1 0x43000000 boot/dtbs/\${fdtfile}
load mmc 0:1 0x42000000 zImage || load mmc 0:1 0x42000000 boot/zImage
bootz 0x42000000 - 0x43000000
EOF

    notice "creating u-boot script image"
    sudo mkimage -A arm -T script -C none \
		-d $strapdir/boot/boot.cmd $strapdir/boot/boot.scr || { zerr; return 1; }

    postbuild-clean
}

build_kernel_armhf() {
    fn build_kernel_armhf
    req=(R arch device_name gitkernel gitbranch MAKEOPTS)
    req+=(strapdir)
    ckreq || return 1

    notice "building $arch kernel"

    prebuild || { zerr; return 1; }

    get-kernel-sources
    pushd $R/tmp/kernels/$device_name/${device_name}-linux
        #copy-kernel-config
		make $MAKEOPTS \
			ARCH=arm \
			CROSS_COMPILE=$compiler \
				sunxi_defconfig || { zerr; return 1; }

		# compile kernel and modules
        make \
			$MAKEOPTS \
            ARCH=arm \
			CROSS_COMPILE=$compiler \
				zImage dtbs modules || { zerr; return 1; }

		# install kernel modules
        sudo -E PATH="$PATH" \
            make \
				$MAKEOPTS \
				ARCH=arm \
				CROSS_COMPILE=$compiler \
				INSTALL_MOD_PATH=$strapdir \
					modules_install || { zerr; return 1; }

        sudo cp -v arch/arm/boot/zImage $strapdir/boot/ || { zerr; return 1; }
		sudo mkdir -p $strapdir/boot/dtbs
		for board in $board_dtbs; do
			sudo cp -v arch/arm/boot/dts/$board $strapdir/boot/dtbs/ || {
				zerr; return 1; }
		done
    popd

    postbuild || { zerr; return 1; }
}