#!/usr/bin/env sh

#
#  helper script: generate release archive (ROM images)
#
#	Copyright (C) 2020,2021,2022,2023 Leah Rowe <info@minifree.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/>.
#

# This script assumes that the working directory is the root
# of git or release archive

[ "x${DEBUG+set}" = 'xset' ] && set -v
set -u -e

projectname="$(cat projectname)"

version="version-unknown"
if [ -f version ]; then
	version="$(cat version)"
fi
versiondate="version-date-unknown"
if [ -f versiondate ]; then
	versiondate="$(cat versiondate)"
fi

if [ ! -d "bin/" ]; then
	printf "build/release/roms: no ROMs built yet. Exiting.\n"
	exit 1
fi

[ ! -d "release/" ] && \
    mkdir -p release/
[ ! -d "release/${version}/" ] && \
    mkdir -p "release/${version}/"
[ -d "release/${version}/roms/" ] && \
    rm -Rf "release/${version}/roms/"
[ ! -d "release/${version}/roms/" ] && \
    mkdir -p "release/${version}/roms/"

printf "Building ROM image archives for version %s\n" "${version}"

for romdir in bin/*; do
	target="${romdir##*/}"

	echo ${target}
	if [ ! -d "${romdir}/" ]; then
	    continue
	fi

	CONFIG_HAVE_MRC="y"
	CONFIG_HAVE_ME_BIN="y"
	CONFIG_KBC1126_FIRMWARE="y"
	grep "CONFIG_HAVE_ME_BIN=y" "resources/coreboot/${target}/config/"* || CONFIG_HAVE_ME_BIN="n"
	grep "CONFIG_HAVE_MRC=y" "resources/coreboot/${target}/config/"* || CONFIG_HAVE_MRC="n"
	grep "CONFIG_KBC1126_FIRMWARE=y" "resources/coreboot/${target}/config"/* || CONFIG_KBC1126_FIRMWARE="n"

	# remove ME/MRC/EC firmware from ROM images
	if [ "${CONFIG_HAVE_ME_BIN}" = "y" ] || [ "${target}" = "e6400nvidia_4mb" ]; then
		if [ ! -d coreboot/default ]; then
			./download coreboot default || exit 1
		fi

		ifdtooldir="coreboot/default/util/ifdtool"
		ifdtool="${ifdtooldir}/ifdtool"
		if [ ! -f "${ifdtool}" ]; then
			./build module cbutils default || exit 1
		fi

		cbfstooldir="coreboot/default/util/cbfstool"
		cbfstool="${cbfstooldir}/cbfstool"
		if [ ! -f "${cbfstool}" ]; then
			./build module cbutils default || exit 1
		fi

		rm -Rf "${romdir}_tmp" # dirty hack, to reduce  disk io later
		# rather than using /tmp, which might not be tmpfs
		mkdir "${romdir}_tmp"
		
		if [ ! -f "${romdir}/blobhashes" ]; then
			printf "%s\n" "The roms in this archive should match the following hashes AFTER blobs are inserted" > "${romdir}/blobhashes"
		fi

		# Hash the rom before removing blobs
		( cd ${romdir}
		sha1sum *.rom >> blobhashes
		)

		for romfile in "${romdir}"/*.rom
		do
			if [ ! -f "${romfile}" ]; then
				continue
			fi

			if [ "${CONFIG_HAVE_ME_BIN}" = "y" ]; then
				${ifdtool} --nuke me "${romfile}" || exit 1
				mv "${romfile}" "${romdir}_tmp"/
				mv "${romfile}.new" "${romfile}"
			fi

			if [ "${CONFIG_HAVE_MRC}" = "y" ]
			then
				${cbfstool} "${romfile}" remove -n mrc.bin || exit 1
				${cbfstool} "${romfile}" print
			fi

			if [ "${CONFIG_KBC1126_FIRMWARE}" = "y" ]; then
				${cbfstool} "${romfile}" remove -n ecfw1.bin || exit 1
				${cbfstool} "${romfile}" remove -n ecfw2.bin || exit 1
			fi

			# TODO: replace this board-specific hack
			if [ "${target}" = "e6400nvidia_4mb" ]; then
				${cbfstool} "${romfile}" remove \
					-n "pci10de,06eb.rom" \
					|| exit 1
			fi
		done
	fi

	printf "Generating release/%s/roms/%s-%s_%s.tar.xz\n" "${version}" "${projectname}" "${version}" "${target##*/}"
	printf "%s\n" "${version}" > "${romdir}/version"
	printf "%s\n" "${versiondate}" > "${romdir}/versiondate"
	printf "%s\n" "${projectname}" > "${romdir}/projectname"
	tar -c "${romdir}/" | xz -9e >"release/${version}/roms/${projectname}-${version}_${target##*/}.tar.xz"

	if [ -d "${romdir}_tmp" ]
	then
		rm -Rf "${romdir}"
		mv "${romdir}_tmp" "${romdir}"
	fi
done

printf "\nROM image release archives available at release/%s/roms/\n\n" "${version}"