#!/bin/bash
#
# Copyright (c) 2021-2022 Amlogic, Inc. All rights reserved.
#
# SPDX-License-Identifier: MIT
#

###############################################################
# Function: Auto-generate root CMakeLists.txt and Kconfig according to manifest.xml.
###############################################################

cmake_file="$PWD/CMakeLists.txt"
kconfig_file="$PWD/Kconfig"
build_dir="build_system"
exclude_dirs="boot products docs"
special_dirs="arch soc boards soc/$ARCH boards/$ARCH"

SOC_MANIFEST="$PWD/soc/$ARCH/$SOC/rtos_sdk_manifest.xml"
DEFAULT_RTOS_SDK_MANIFEST="$PWD/products/$PRODUCT/rtos_sdk_manifest.xml"
RTOS_SDK_MANIFEST_FILE="$kernel_BUILD_DIR/rtos_sdk_manifest.xml"
RTOS_SDK_MANIFEST_OLD_FILE="$kernel_BUILD_DIR/rtos_sdk_manifest_old.xml"
STAMP="$kernel_BUILD_DIR/.stamp"
RTOS_SDK_VERSION_FILE="$kernel_BUILD_DIR/sdk_ver.h"

# Use repo in .repo/repo dir,otherwise use in system default location
repo_root_dir=$PWD
while [ "$repo_root_dir" != "/" ]; do
	if [ -d "$repo_root_dir/.repo" ]; then
		break
	fi
	repo_root_dir=$(dirname "$repo_root_dir")
done
if [ "$repo_root_dir" == "/" ] && [ ! -d "/.repo" ]; then
	echo ".repo not found,use system default location."
else
	echo ".repo at: $repo_root_dir/.repo"
	export PATH=$repo_root_dir/.repo/repo:$PATH
fi

# Check whether the project is a valid repo
repo manifest 2>&1 | grep -q $build_dir
if [ "$?" -ne 0 ]; then
	echo "Non-repo source code"
	if [ -s $DEFAULT_RTOS_SDK_MANIFEST ]; then
		if [ -s $SOC_MANIFEST ]; then
			echo "Use specific manifest: $SOC_MANIFEST"
			cat $DEFAULT_RTOS_SDK_MANIFEST $SOC_MANIFEST > $RTOS_SDK_MANIFEST_FILE
		else
			echo "Use default manifest: $DEFAULT_RTOS_SDK_MANIFEST"
			cp -f $DEFAULT_RTOS_SDK_MANIFEST $RTOS_SDK_MANIFEST_FILE
		fi
	else
		echo "Default manifest.xml not found!"
		exit 0
	fi
else
	echo "Generate manifest.xml"
	repo manifest > $RTOS_SDK_MANIFEST_FILE
	if [ ! -f $RTOS_SDK_MANIFEST_FILE ]; then
		echo "Faild to save $RTOS_SDK_MANIFEST_FILE"
		exit 1
	fi
	# If the name and path is same in mainfext.xml,cmd repo manifest will ignore path section
	# and only generate name section cause they are same,so we need add path section with sed
	sed -i -r '/name=/ { /path=/! s/(name=("[^"]+"))/& path=\2/ }' $RTOS_SDK_MANIFEST_FILE
fi

# Get SDK_VERSION
if [ -s $DEFAULT_RTOS_SDK_MANIFEST ]; then
	source scripts/gen_version.sh $RTOS_SDK_VERSION_FILE \
	$RTOS_SDK_MANIFEST_FILE $DEFAULT_RTOS_SDK_MANIFEST
else
	source scripts/gen_version.sh $RTOS_SDK_VERSION_FILE $RTOS_SDK_MANIFEST_FILE
fi

if [ -s $RTOS_SDK_MANIFEST_OLD_FILE ] && [ -s $kconfig_file ] && [ $kconfig_file -ot $STAMP ]; then
	is_update=`comm -3 <(sort $RTOS_SDK_MANIFEST_FILE) <(sort $RTOS_SDK_MANIFEST_OLD_FILE)`
	if [ -z "$is_update" ]; then
		exit 0
	else
		echo "Update top Kconfig and CMakelists.txt."
	fi
fi

# Back up manifest.xml
cp -arf $RTOS_SDK_MANIFEST_FILE $RTOS_SDK_MANIFEST_OLD_FILE

# Write the fixed content to CMakeLists.txt
cat <<EOF > $cmake_file
enable_language(C CXX ASM)

EOF

# Clear Kconfig
cat <<EOF > $kconfig_file
EOF

# Figure out the $relative_dir and its column
[ -z "$REPO_DIR" ] && REPO_DIR=$PWD
pattern="path="
i=0
keyline=`grep "path=\".*$build_dir\"" $RTOS_SDK_MANIFEST_FILE`
for keyword in $keyline; do
	let i++
	if [[ $keyword == $pattern* ]]; then
		repo_path=`echo ${keyword#*${pattern}} | sed 's/\"//g' | sed 's/\/>//g'`
		relative_dir=`dirname $repo_path`
		# Filter current project
		if [[ $REPO_DIR == *$relative_dir ]]; then
			break
		fi
	fi
done

if [[ $relative_dir == . ]]; then
	pattern="path="
else
	pattern="path=\"${relative_dir}/"
fi

# Sort manifest.xml based on "path" to get an reasonable order for generating Kconfig
sort -k $i $RTOS_SDK_MANIFEST_FILE -o $RTOS_SDK_MANIFEST_FILE

drivers_found_flag=0
while IFS= read -r line
do
	keyline=`echo "$line" | grep "$pattern"`
	[ -z "$keyline" ] && continue

	for keyword in $keyline; do
		if [[ $keyword == path=* ]]; then
			repo_path=`echo ${keyword#*${pattern}} | sed 's/\"//g' | sed 's/\/>//g'`

			if [[ $repo_path == drivers* ]]; then
				[ "$drivers_found_flag" -eq 1 ] && continue
				if [ -d "drivers_$PRODUCT" ]; then
					category=drivers_$PRODUCT
				else
					category=drivers
				fi
				repo_path=$category
				drivers_found_flag=1
			elif [[ $repo_path == benchmark* ]]; then
				category=`basename $repo_path | sed 's/_/ /g'`
			else
				category=`dirname $repo_path`
			fi

			# exclude some dirs
			skip_flag=0
			for exclude_dir in $exclude_dirs; do
				if [[ $repo_path == $exclude_dir* ]]; then
					skip_flag=1
					break
				fi
			done
			[ "$skip_flag" -eq 1 ] && continue

			skip_flag=0
			for item in $special_dirs; do
				case "$repo_path" in
				"$item"*)
					if [[ "$item/$ARCH" == "$repo_path" ]] || [[ "$item/$ARCH/$PRODUCT" == "$repo_path" ]]; then
						break
					else
						skip_flag=1
						continue
					fi
				;;
				*)
				;;
				esac
			done
			[ "$skip_flag" -eq 1 ] && continue
			cmake_path=$repo_path
			kconfig_path=$repo_path

			# Generate root CMakeLists.txt
			if [ -f $repo_path/CMakeLists.txt ]; then
				echo "add_subdirectory($cmake_path)" >> $cmake_file
			fi

			# Generate root Kconfig
			if [ -f $repo_path/Kconfig ]; then
				if [ "$last_category" != "$category" ]; then
					if [ -n "$last_category" ]; then
						echo -e "endmenu\n" >> $kconfig_file
					fi

					if [ "$category" == "wcn" -o "$category" == "soc" ]; then
						echo "menu \"${category^^} Options\"" >> $kconfig_file
					else
						echo "menu \"${category^} Options\"" >> $kconfig_file
					fi
				fi

				echo "source \"$kconfig_path/Kconfig\"" >> $kconfig_file
				last_category=$category
			fi
			break;
		fi
	done
done < "$RTOS_SDK_MANIFEST_FILE"

echo "endmenu" >> $kconfig_file

sleep 1
touch $STAMP