#!/bin/bash
#
#    Copyright (c) 2022, 2025 Project CHIP Authors
#
#    Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.
#

set -e
set -x
helpFunction() {
    cat <<EOF
Usage: $0 -s|--src <src folder> -o|--out <out folder> [-d|--debug] [-n|--no-init] [-t|--trusty] [-m|--imx_ele]
    -s, --src       Source folder
    -o, --out       Output folder
    -d, --debug     Debug build (optional)
    -n, --no-init   No init mode (optional)
    -t, --trusty    Build with Trusty OS backed security enhancement (optional)
    -m, --imx_ele   Build with IMX ELE (EdgeLock Enclave) based security enhancement (optional)
EOF
    exit 1
}

trusty=false
imx_ele=false
imx_ele_path="third_party/imx-secure-enclave/repo/"
release_build=true
PARSED_OPTIONS="$(getopt -o s:o:tmdn --long src:,out:,trusty,imx_ele,debug,no-init -- "$@")"
if [ $? -ne 0 ]; then
    helpFunction
fi
eval set -- "$PARSED_OPTIONS"
while true; do
    case "$1" in
        -s | --src)
            src="$2"
            shift 2
            ;;
        -o | --out)
            out="$2"
            shift 2
            ;;
        -t | --trusty)
            trusty=true
            shift
            ;;
        -m | --imx_ele)
            imx_ele=true
            shift
            ;;
        -d | --debug)
            release_build=false
            shift
            ;;
        -n | --no-init)
            no_init=1
            shift
            ;;
        --)
            shift
            break
            ;;
        *)
            echo "Invalid option: $1" >&2
            exit 1
            ;;
    esac
done

if [ -z "$src" ] || [ -z "$out" ]; then
    echo "Some or all of the required -s|--src and -o|--out parameters are empty."
    helpFunction
fi

if [ "$no_init" != 1 ]; then
    source "$(dirname "$0")/../../scripts/activate.sh"
fi

if [ "$IMX_SDK_ROOT" = "" -o ! -d "$IMX_SDK_ROOT" ]; then
    echo "the Yocto SDK path is not specified with the shell env IMX_SDK_ROOT or an invalid path is specified"
    exit 1
fi

entries="$(echo "$(ls "$IMX_SDK_ROOT")" | tr -s '\n' ',')"
IFS=',' read -ra entry_array <<<"$entries"
for entry in "${entry_array[@]}"; do
    if [ "$(echo "$entry" | grep -E "^environment-setup-")" != "" ]; then
        env_setup_script=$entry
        break
    fi
done

if [ -z "$env_setup_script" ]; then
    echo "The SDK environment setup script is not found, make sure the env IMX_SDK_ROOT is correctly set."
    exit 1
fi

while read line; do
    # trim the potential whitespaces
    line=$(echo "$line" | xargs)

    if [ "$(echo "$line" | grep -E "^export SDKTARGETSYSROOT=")" != "" ]; then
        sdk_target_sysroot=${line#"export SDKTARGETSYSROOT="}
    fi

    if [ "$(echo "$line" | grep -E "^export CC=")" != "" ]; then
        cc=${line#"export CC="}
        cc=${cc#"\""}
        cc=${cc%"\""}
        cc=${cc/"\$SDKTARGETSYSROOT"/$sdk_target_sysroot}
    fi

    if [ "$(echo "$line" | grep -E "^export CXX=")" != "" ]; then
        cxx=${line#"export CXX="}
        cxx=${cxx#"\""}
        cxx=${cxx%"\""}
        cxx=${cxx/"\$SDKTARGETSYSROOT"/$sdk_target_sysroot}
    fi

    if [ "$(echo "$line" | grep -E "^export ARCH=")" != "" ]; then
        target_cpu=${line#"export ARCH="}

        if [ "$target_cpu" = "arm64" ]; then
            arm_arch="armv8-a"
        elif [ "$target_cpu" = "arm" ]; then
            arm_arch="armv7ve"
        else
            echo "ARCH should be arm64 or arm in the SDK environment setup script."
            exit 1
        fi
    fi

    if [ "$(echo "$line" | grep -E "^export CROSS_COMPILE=")" != "" ]; then
        cross_compile=${line#"export CROSS_COMPILE="}
        cross_compile=${cross_compile%"-"}
    fi
done <"$IMX_SDK_ROOT/$env_setup_script"

if [ -z "$sdk_target_sysroot" ]; then
    echo "SDKTARGETSYSROOT is not found in the SDK environment setup script."
    exit 1
fi

if [ -z "$cc" -o -z "$cxx" ]; then
    echo "CC and/or CXX are not found in the SDK environment setup script."
    exit 1
fi

if [ -z "$target_cpu" -o -z "$cross_compile" ]; then
    echo "ARCH and/or CROSS_COMPILE are not found in the SDK environment setup script."
    exit 1
fi

if [ "$imx_ele" = "true" ]; then
    if [ -f "$imx_ele_path/Makefile" ]; then
        make -C "$imx_ele_path" "install_version"
    else
        echo "Makefile does not exist in $imx_ele_path."
        exit 1
    fi
fi

PLATFORM_CFLAGS='-DCHIP_DEVICE_CONFIG_WIFI_STATION_IF_NAME=\"mlan0\"'
gn gen --check --fail-on-unused-args --root="$src" "$out" --args="target_os=\"linux\" target_cpu=\"$target_cpu\" arm_arch=\"$arm_arch\"
chip_with_trusty_os=$trusty
chip_with_imx_ele=$imx_ele
treat_warnings_as_errors=false
import(\"//build_overrides/build.gni\")
sysroot=\"$sdk_target_sysroot\"
target_cflags=[ \"$PLATFORM_CFLAGS\" ]
custom_toolchain=\"\${build_root}/toolchain/custom\"
target_cc=\"$IMX_SDK_ROOT/sysroots/x86_64-pokysdk-linux/usr/bin/$cross_compile/$cc\"
target_cxx=\"$IMX_SDK_ROOT/sysroots/x86_64-pokysdk-linux/usr/bin/$cross_compile/$cxx\"
target_ar=\"$IMX_SDK_ROOT/sysroots/x86_64-pokysdk-linux/usr/bin/$cross_compile/$cross_compile-ar\"
$(if [ "$release_build" = "true" ]; then echo "is_debug=false"; else echo "optimize_debug=true"; fi)"

ninja -C "$out"