#!/usr/bin/env sh
# SPDX-License-Identifier: GPL-3.0-or-later
# Copyright (c) 2022-2023 Alper Nebi Yasak <alpernebiyasak@gmail.com>
# Copyright (c) 2022 Ferass El Hafidi <vitali64pmemail@protonmail.com>
# Copyright (c) 2023-2024 Leah Rowe <leah@libreboot.org>

set -u -e

. "include/lib.sh"
. "include/git.sh"

eval "$(setvars "" xarch cfgsdir cdir config config_name xlang mode makeargs \
    listfile project target target_dir targets tree _f target1 bootstrapargs \
    autoconfargs cmakedir elfdir autogenargs xtree)"

main()
{
	while getopts f:b:m:u:c:x:s:l:n: option; do
		_f="$1"
		case "$1" in
		-b) : ;;
		-u) mode="oldconfig" ;;
		-m) mode="menuconfig" ;;
		-c) mode="distclean" ;;
		-x) mode="crossgcc-clean" ;;
		-f) mode="fetch" ;;
		-s) mode="savedefconfig" ;;
		-l) mode="olddefconfig" ;;
		-n) mode="nconfig" ;;
		*) $err "Invalid option" ;;
		esac
		shift; project="${OPTARG#src/}"; shift
	done
	[ -z "$_f" ] && $err "missing flag (-m/-u/-b/-c/-x/-f/-s/-l/-n)"
	[ -z "$project" ] && $err "project name not specified"
	elfdir="elf/$project"
	cfgsdir="config/$project"

	remkdir "${tmpgit%/*}"

	_cmd="build_projects"
	[ -f "config/$project/build.list" ] && _cmd="build_targets"
	$_cmd $@
}

build_projects()
{
	[ $# -gt 0 ] && x_ ./update trees $_f $@

	[ "$mode" = "fetch" ] && [ ! -f "CHANGELOG" ] && \
		eval "fetch_project_repo; return 0;"

	load_project_config "$cfgsdir"

	cdir="src/${project}"
	[ -d "$cdir" ] || x_ ./update trees -f "$project"

	[ "$mode" = "distclean" ] && mode="clean"
	run_make_command || return 0
}

build_targets()
{
	[ "$elfdir" = "elf/coreboot" ] && \
		elfdir="elf/coreboot_nopayload_DO_NOT_FLASH"

	[ -d "$cfgsdir" ] || $err "directory, $cfgsdir, does not exist"

	listfile="$cfgsdir/build.list"
	[ -f "$listfile" ] || $err "list file, $listfile, does not exist"

	# Build for all targets if no argument is given
	[ $# -gt 0 ] && target1="$1"
	[ "$target1" = "utils" ] && [ "$project" = "coreboot" ] && shift 1
	targets="$(items "$cfgsdir")" || $err "Can't get options for $cfgsdir"
	[ $# -gt 0 ] && targets=$@

	[ -z "$mode" ] && x_ mkdir -p "$elfdir"
	handle_targets
}

handle_targets()
{
	for x in $targets; do
		target="$x"
		printf "'make %s', '%s', '%s'\n" "$mode" "$project" "$target"
		x_ handle_defconfig
	done

	[ "$target1" = "utils" ] && [ "$project" = "coreboot" ] && return 0
	[ -z "$mode" ] && printf "Done! Check  %s/\n\n" "$elfdir"; return 0
}

handle_defconfig()
{
	handle_src_tree "$target" || return 0

	[ "$target1" = "utils" ] && [ "$project" = "coreboot" ] && \
		eval "handle_coreboot_utils \"$tree\"; return 0"

	for y in "$target_dir/config"/*; do
		[ -f "$y" ] || continue
		config="$y"
		config_name="${config#"$target_dir/config/"}"

		[ -n "$mode" ] || check_config || continue
		handle_makefile
		[ -n "$mode" ] || copy_elf
	done
}

handle_src_tree()
{
	target_dir="$cfgsdir/$target"
	[ "$mode" = "fetch" ] && [ ! -f "CHANGELOG" ] && \
		eval "fetch_project_trees; return 1;"

	load_project_config "$target_dir"
	x_ mkdir -p "$elfdir/$target"

	[ -z "$tree" ] && $err "handle_src_tree $project/$tree: tree unset"
	cdir="src/$project/$tree"

	if [ ! -d "$cdir" ]; then
		if [ "$mode" = "distclean" ] || \
		    [ "$mode" = "crossgcc-clean" ]; then
			printf "Directory %s missing; skip\n" "$cdir" 1>&2
			return 1
		fi
		x_ ./update trees -f "$project" "$target"
	fi

	[ "$target1" = "utils" ] && [ "$project" = "coreboot" ] && return 0
	[ -z "$mode" ] && check_cross_compiler; return 0
}

load_project_config()
{
	eval "$(setvars "" xarch xlang tree bootstrapargs autoconfargs xtree \
	    tree_depend makeargs)"

	[ -f "$1/target.cfg" ] || return 0
	. "$1/target.cfg" || $err "loadp $1: can't load target.cfg"; return 0
}

check_cross_compiler()
{
	for _xarch in $xarch; do
		cbdir="src/coreboot/$tree"
		[ "$project" != "coreboot" ] && cbdir="src/coreboot/default"
		[ -n "$xtree" ] && cbdir="src/coreboot/$xtree"

		x_ ./update trees -f coreboot ${cbdir#src/coreboot/}

		export PATH="$PWD/$cbdir/util/crossgcc/xgcc/bin:$PATH"
		export CROSS_COMPILE="${xarch% *}-"
		[ -n "$xlang" ] && export BUILD_LANGUAGES="$xlang"

		[ -d "$cbdir/util/crossgcc/xgcc/$_xarch/" ] && continue
		x_ make -C "$cbdir" crossgcc-${_xarch%-*} CPUS=$threads \
		    UPDATED_SUBMODULES=1
	done
}

check_config()
{
	[ -f "$config" ] || $err "check_config $project/$target: no config"

	dest_dir="$elfdir/$target/$config_name"
	# TODO: very hacky check. do it properly (based on build.list)
	for elftest in "$dest_dir"/*; do
		e "$elftest" f && return 1
	done
	x_ mkdir -p "$dest_dir"
}

handle_makefile()
{
	x_ make clean -C "$cdir" && x_ cp "$config" "$cdir/.config"
	[ -n "$mode" ] || make -C "$cdir" silentoldconfig || \
	    make -C "$cdir" oldconfig || :

	run_make_command || $err "handle_makefile $cdir: no makefile!"

	_copy=".config"
	[ "$mode" = "savedefconfig" ] && _copy="defconfig"
	[ "${mode%config}" = "$mode" ] || x_ cp "$cdir/$_copy" "$config"

	[ -e "$cdir/.git" ] && [ "$project" = "u-boot" ] && \
	    [ "$mode" = "distclean" ] && x_ git -C "$cdir" clean -fdx; return 0
}

run_make_command()
{
	check_cmake "$cdir"
	[ -z "$mode" ] && check_autoconf "$cdir"
	check_makefile "$cdir" || return 1

	[ "$project" = "coreboot" ] && [ -z "$mode" ] && x_ \
		printf "%s\n" "${version%%-*}" > "$cdir/.coreboot-version" \
		    && makeargs="$makeargs UPDATED_SUBMODULES=1 CPUS=$threads"

	make -C "$cdir" $mode -j$threads $makeargs || $err "!mk $cdir $mode"

	[ "$mode" != "clean" ] && return 0
	make -C "$cdir" distclean 2>/dev/null || :
}

check_cmake()
{
	[ -z "$cmakedir" ] || check_makefile "$1" || cmake -B "$1" \
	    "$1/$cmakedir" || check_makefile "$1" || $err "$1: !cmk $cmakedir"
	[ -z "$cmakedir" ] || check_makefile "$1" || \
		$err "check_cmake $1: can't generate Makefile"; return 0
}

check_autoconf()
{
	(
	cd "$1" || $err "!cd $1"
	[ -f "bootstrap" ] && x_ ./bootstrap $bootstrapargs
	[ -f "autogen.sh" ] && x_ ./autogen.sh $autogenargs
	[ -f "configure" ] && x_ ./configure $autoconfargs; return 0
	) || $err "can't bootstrap project: $1"
}

check_makefile()
{
	[ -f "$1/Makefile" ] || [ -f "$1/makefile" ] || \
	    [ -f "$1/GNUmakefile" ] || return 1; return 0
}

copy_elf()
{
	while read -r f; do
		[ -f "$cdir/$f" ] && x_ cp "$cdir/$f" "$dest_dir"
	done < "$listfile"
	x_ make clean -C "$cdir"
}

main $@