#!/usr/bin/env bash
#
# SPDX-License-Identifier: GPL-2.0
#
# Copyright (c) 2026 AMLOGIC
#
# AML MediaHal SDK build script for AMLbian
# This script builds aml-mediahal-sdk for Amlogic SoCs
# Based on aml-mediahal-sdk_git.bb functionality

set -e

# Get script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
AMLBIAN_DIR="$(dirname "$(dirname "${SCRIPT_DIR}")")"
BUILD_DIR="${AMLBIAN_DIR}/../build"

# Source common functions
if [[ -f "${AMLBIAN_DIR}/lib/function-common.sh" ]]; then
    source "${AMLBIAN_DIR}/lib/function-common.sh"
fi

# ==============================================================================
# Section 2: Default Configuration
# ==============================================================================

BOARD="${BOARD:-s6-bq201}"
MEDIAHAL_SDK_VERSION="${MEDIAHAL_SDK_VERSION:-1.0.0}"
ARCH="${ARCH:-arm64}"
KERNEL_VERSION="${KERNEL_VERSION:-6.12}"
AML_CUST_SPEC="${AML_CUST_SPEC:-}"  # Could be "CUST_S_SMH" or empty
DISTRO_FEATURES="${DISTRO_FEATURES:-}"  # Could contain "wayland", "zapper"

# ==============================================================================
# Section 4: MediaHal SDK Build Functions
# ==============================================================================

function __unset_mediahal_sdk_config {
    # Clean up any environment variables set during build
    unset CFLAGS
    unset CXXFLAGS
    unset LDFLAGS
}

function __mediahal_sdk_prepare_source {
    local build_dir="${1}"

    local aml_comp_path="$(amlbian_get_aml_comp_dir)"
    if [[ -z "${aml_comp_path}" ]] || [[ ! -d "${aml_comp_path}" ]]; then
        display_alert "MEDIAHAL_SDK" "aml-comp not found: ${aml_comp_path}" "err"
        exit 1
    fi
    
    local source_dir="${aml_comp_path}/multimedia/mediahal-sdk"
    
    if [[ ! -d "${source_dir}" ]]; then
        display_alert "MEDIAHAL_SDK" "Source directory not found: ${source_dir}" "err"
        exit 1
    fi
    
    display_alert "MEDIAHAL_SDK" "Copying source from ${source_dir} to ${build_dir}" "info"
    
    # 清空构建目录
    rm -rf "${build_dir}"
    mkdir -p "${build_dir}"
    
    # 复制源代码到构建目录
    cp -r "${source_dir}"/* "${build_dir}/" 2>/dev/null || true
    
    # Check for necessary source files
    if [[ ! -f "${build_dir}/Makefile" ]] && [[ ! -d "${build_dir}/prebuilt" ]]; then
        display_alert "MEDIAHAL_SDK" "Makefile or prebuilt directory not found in source" "err"
        exit 1
    fi
    
    display_alert "MEDIAHAL_SDK" "Source prepared successfully" "success"
}

function __mediahal_sdk_build_examples {
    local build_dir="${1}"
    
    display_alert "MEDIAHAL_SDK" "Building examples..." "info"
    
    if [[ ! -d "${build_dir}" ]]; then
        display_alert "MEDIAHAL_SDK" "Build directory not found: ${build_dir}" "err"
        return 1
    fi
    
    # Set build environment
    ARCH="${ARCH:-arm64}"
    CROSS_COMPILE="${CROSS_COMPILE:-aarch64-linux-gnu-}"
    
    export ARCH="${ARCH}"
    export CROSS_COMPILE="${CROSS_COMPILE}"
    
    amlbian_checkout_toolchain
    
    # Set compiler
    export CC="${CROSS_COMPILE}gcc"
    export CXX="${CROSS_COMPILE}g++"
    export AR="${CROSS_COMPILE}ar"
    
    # Determine target architecture directory name
    local target_arch=""
    if [[ "${ARCH}" == "arm64" ]]; then
        # For 64-bit ARM, we need to check if we have aarch64.lp64 directory
        # If not, use arm.aapcs-linux.hard_y2038 or arm.aapcs-linux.hard
        if [[ -d "${build_dir}/prebuilt/aarch64.lp64." ]]; then
            target_arch="aarch64.lp64."
        elif [[ -d "${build_dir}/prebuilt/arm.aapcs-linux.hard_y2038" ]]; then
            target_arch="arm.aapcs-linux.hard_y2038"
        else
            target_arch="arm.aapcs-linux.hard"
        fi
    else
        # For 32-bit ARM
        target_arch="arm.aapcs-linux.hard"
    fi
    
    display_alert "MEDIAHAL_SDK" "Using target architecture: ${target_arch}" "info"
    
    if [[ -z "${AML_CUST_SPEC:-}" ]]; then
        display_alert "MEDIAHAL_SDK" "Skipping example builds for CUST_S_SMH" "info"
        return 0
    fi
    
    # Build AmTsPlayerExample
    if [[ -d "${build_dir}/example/AmTsPlayerExample" ]]; then
        display_alert "MEDIAHAL_SDK" "Building AmTsPlayerExample..." "info"
        cd "${build_dir}/example/AmTsPlayerExample"
        
        # Set CFLAGS and LDFLAGS for cross-compilation
        local cflags="-I${build_dir}/prebuilt/noarch/include"
        local ldflags="-L${build_dir}/prebuilt/${target_arch}"
        
        # Add wayland flags if needed
        if is_feature_contains "${DISTRO_FEATURES}" "wayland"; then
            cflags="${cflags} $(pkg-config --cflags wayland-client 2>/dev/null || echo '')"
            ldflags="${ldflags} $(pkg-config --libs wayland-client 2>/dev/null || echo '')"
        fi
        
        make CC="${CC}" CFLAGS="${cflags}" LDFLAGS="${ldflags}" -j$(nproc) 2>&1 | tee "${build_dir}/AmTsPlayerExample.log" || {
            display_alert "MEDIAHAL_SDK" "AmTsPlayerExample build failed" "err"
            return 1
        }
        display_alert "MEDIAHAL_SDK" "AmTsPlayerExample built successfully" "success"
    fi
    
    # Build AmTsPlayerMultiExample (if not zapper)
    if ! is_feature_contains "${DISTRO_FEATURES}" "zapper"; then
        if [[ -d "${build_dir}/example/AmTsPlayerMultiExample" ]]; then
            display_alert "MEDIAHAL_SDK" "Building AmTsPlayerMultiExample..." "info"
            cd "${build_dir}/example/AmTsPlayerMultiExample"
            
            local cflags="-I${build_dir}/prebuilt/noarch/include"
            local ldflags="-L${build_dir}/prebuilt/${target_arch}"
            
            # Add wayland flags if needed
            if is_feature_contains "${DISTRO_FEATURES}" "wayland"; then
                cflags="${cflags} $(pkg-config --cflags wayland-client 2>/dev/null || echo '')"
                ldflags="${ldflags} $(pkg-config --libs wayland-client 2>/dev/null || echo '')"
            fi
            
            make CC="${CC}" CFLAGS="${cflags}" LDFLAGS="${ldflags}" -j$(nproc) 2>&1 | tee "${build_dir}/AmTsPlayerMultiExample.log" || {
                display_alert "MEDIAHAL_SDK" "AmTsPlayerMultiExample build failed" "err"
                return 1
            }
            display_alert "MEDIAHAL_SDK" "AmTsPlayerMultiExample built successfully" "success"
        fi
    fi
    
    # Build EsVideoDecPlayer
    if [[ -d "${build_dir}/example/EsVideoDecPlayer" ]]; then
        display_alert "MEDIAHAL_SDK" "Building EsVideoDecPlayer..." "info"
        cd "${build_dir}/example/EsVideoDecPlayer"
        
        local cflags="-I${build_dir}/prebuilt/noarch/include"
        local ldflags="-L${build_dir}/prebuilt/${target_arch}"
        
        # Add wayland flags if needed
        if is_feature_contains "${DISTRO_FEATURES}" "wayland"; then
            cflags="${cflags} $(pkg-config --cflags wayland-client 2>/dev/null || echo '')"
            ldflags="${ldflags} $(pkg-config --libs wayland-client 2>/dev/null || echo '')"
        fi
        
        make CC="${CC}" CFLAGS="${cflags}" LDFLAGS="${ldflags}" -j$(nproc) 2>&1 | tee "${build_dir}/EsVideoDecPlayer.log" || {
            display_alert "MEDIAHAL_SDK" "EsVideoDecPlayer build failed" "err"
            return 1
        }
        display_alert "MEDIAHAL_SDK" "EsVideoDecPlayer built successfully" "success"
    fi
    
    return 0
}

function __mediahal_sdk_install {
    local build_dir="${1}"
    local install_dir="${2}"
    
    display_alert "MEDIAHAL_SDK" "Installing mediahal-sdk..." "info"
    
    if [[ ! -d "${build_dir}" ]]; then
        display_alert "MEDIAHAL_SDK" "Build directory not found: ${build_dir}" "err"
        return 1
    fi
    
    # Determine target architecture directory name
    local target_arch=""
    if [[ "${ARCH}" == "arm64" ]]; then
        if [[ -d "${build_dir}/prebuilt/aarch64.lp64." ]]; then
            target_arch="aarch64.lp64."
        elif [[ -d "${build_dir}/prebuilt/arm.aapcs-linux.hard_y2038" ]]; then
            target_arch="arm.aapcs-linux.hard_y2038"
        else
            target_arch="arm.aapcs-linux.hard"
        fi
    else
        target_arch="arm.aapcs-linux.hard"
    fi

    display_alert "MEDIAHAL_SDK" "target_arch ${target_arch}..." "info"
    
    # Create installation directories
    mkdir -p "${install_dir}/usr/lib"
    mkdir -p "${install_dir}/usr/include"
    mkdir -p "${install_dir}/usr/bin"
    
    # Install header files
    display_alert "MEDIAHAL_SDK" "Installing header files..." "info"
    local include_dir="${build_dir}/prebuilt/noarch/include"
    if [[ -d "${include_dir}" ]]; then
        for header_file in "${include_dir}"/*.h; do
            if [[ -f "${header_file}" ]]; then
                install -m 0644 -D "${header_file}" "${install_dir}/usr/include/"
                display_alert "MEDIAHAL_SDK" "Installed $(basename "${header_file}") to ${install_dir}/usr/include/" "debug"
            fi
        done
    else
        display_alert "MEDIAHAL_SDK" "Header directory not found: ${include_dir}" "warn"
    fi
    
    # Install library files
    display_alert "MEDIAHAL_SDK" "Installing library files..." "info"
    local lib_dir="${build_dir}/prebuilt/${target_arch}"
    if [[ -d "${lib_dir}" ]]; then
        # Install required libraries
        local required_libs=(
            "libmediahal_resman.so"
            "libmediahal_tsplayer.so"
            "libmediahal_videodec.so"
            "libmediahal_mediasync.so"
        )
        
        for lib in "${required_libs[@]}"; do
            if [[ -f "${lib_dir}/${lib}" ]]; then
                install -m 0644 "${lib_dir}/${lib}" "${install_dir}/usr/lib/"
                display_alert "MEDIAHAL_SDK" "Installed ${lib} to ${install_dir}/usr/lib/" "debug"
            else
                display_alert "MEDIAHAL_SDK" "Library not found: ${lib}" "warn"
            fi
        done
        
        # Install optional libraries (if not zapper)
        if ! is_feature_contains "${DISTRO_FEATURES}" "zapper"; then
            local optional_libs=(
                "libmediahal_videorender.so"
                "libmediahal_dmabufmanage.so"
            )
            
            for lib in "${optional_libs[@]}"; do
                if [[ -f "${lib_dir}/${lib}" ]]; then
                    install -m 0644 "${lib_dir}/${lib}" "${install_dir}/usr/lib/"
                    display_alert "MEDIAHAL_SDK" "Installed ${lib} to ${install_dir}/usr/lib/" "debug"
                else
                    display_alert "MEDIAHAL_SDK" "Optional library not found: ${lib}" "debug"
                fi
            done
        fi
    else
        display_alert "MEDIAHAL_SDK" "Library directory not found: ${lib_dir}" "err"
        return 1
    fi
    
    # Install binaries (examples)
    display_alert "MEDIAHAL_SDK" "Installing binaries..." "info"
    if [[ -z "${AML_CUST_SPEC:-}" ]]; then
        # Install AmTsPlayerExample
        if [[ -f "${build_dir}/example/AmTsPlayerExample/AmTsPlayerExample" ]]; then
            install -m 0755 "${build_dir}/example/AmTsPlayerExample/AmTsPlayerExample" "${install_dir}/usr/bin/"
            display_alert "MEDIAHAL_SDK" "Installed AmTsPlayerExample to ${install_dir}/usr/bin/" "debug"
        fi
        
        # Install EsVideoDecPlayer
        if [[ -f "${build_dir}/example/EsVideoDecPlayer/EsVideoDecPlayer" ]]; then
            install -m 0755 "${build_dir}/example/EsVideoDecPlayer/EsVideoDecPlayer" "${install_dir}/usr/bin/"
            display_alert "MEDIAHAL_SDK" "Installed EsVideoDecPlayer to ${install_dir}/usr/bin/" "debug"
        fi
        
        # Install AmTsPlayerMultiExample (if not zapper)
        if ! is_feature_contains "${DISTRO_FEATURES}" "zapper"; then
            if [[ -f "${build_dir}/example/AmTsPlayerMultiExample/AmTsPlayerMultiExample" ]]; then
                install -m 0755 "${build_dir}/example/AmTsPlayerMultiExample/AmTsPlayerMultiExample" "${install_dir}/usr/bin/"
                display_alert "MEDIAHAL_SDK" "Installed AmTsPlayerMultiExample to ${install_dir}/usr/bin/" "debug"
            fi
        fi
    fi
    
    # Install mediahal_setup.sh
    if [[ -f "${build_dir}/mediahal_setup.sh" ]]; then
        install -m 0755 "${build_dir}/mediahal_setup.sh" "${install_dir}/usr/bin/"
        display_alert "MEDIAHAL_SDK" "Installed mediahal_setup.sh to ${install_dir}/usr/bin/" "debug"
    fi
    
    display_alert "MEDIAHAL_SDK" "Installation completed successfully" "success"
    return 0
}

function __mediahal_sdk_create_deb {
    local build_dir="${1}"
    local install_dir="${2}"
    
    # Package details
    local pkg_name="aml-mediahal-sdk"
    local version="${MEDIAHAL_SDK_VERSION}"
    local architecture="${ARCH:-arm64}"
    local maintainer="Amlbian Builder <build@amlbian.com>"
    local description="Amlogic MediaHal SDK for multimedia applications"
    local description_long="This package provides the Amlogic MediaHal SDK libraries and tools for multimedia applications including video decoding, rendering, and playback."
    
    display_alert "MEDIAHAL_SDK" "Creating Debian package for ${pkg_name} ${version}..." "info"
    
    # Create Debian package directory structure
    local deb_dir="${build_dir}/deb/${pkg_name}-${version}"
    local debian_dir="${deb_dir}/DEBIAN"
    
    mkdir -p "${debian_dir}"
    
    # Copy installed files to package directory
    if [[ -d "${install_dir}" ]] && [[ "$(ls -A "${install_dir}")" ]]; then
        cp -r "${install_dir}"/* "${deb_dir}/" 2>/dev/null || true
        display_alert "MEDIAHAL_SDK" "Copied installed files to ${deb_dir}" "info"
    else
        display_alert "MEDIAHAL_SDK" "No installed files found in ${install_dir}" "warning"
        return 0
    fi
    
    # Create control file
    cat > "${debian_dir}/control" <<EOF
Package: ${pkg_name}
Version: ${version}
Architecture: ${architecture}
Maintainer: ${maintainer}
Description: ${description}
 ${description_long}
Depends: libc6
Section: libs
Priority: optional
EOF
    
    # Create postinst script for ldconfig
    cat > "${debian_dir}/postinst" << 'POSTINST_EOF'
#!/bin/sh
set -e

if [ "$1" = "configure" ]; then
    # Update ldconfig cache
    if command -v ldconfig >/dev/null 2>&1; then
        ldconfig
    fi
fi

exit 0
POSTINST_EOF
    
    chmod 755 "${debian_dir}/postinst"
    
    # Create prerm script
    cat > "${debian_dir}/prerm" << 'PRERM_EOF'
#!/bin/sh
set -e

if [ "$1" = "remove" ] || [ "$1" = "deconfigure" ]; then
    # Update ldconfig
    if command -v ldconfig >/dev/null 2>&1; then
        ldconfig
    fi
fi

exit 0
PRERM_EOF
    
    chmod 755 "${debian_dir}/prerm"
    
    display_alert "MEDIAHAL_SDK" "Building Debian package..." "info"
    
    # Build Debian package
    cd "${build_dir}/deb"
    dpkg-deb --build "${pkg_name}-${version}"
    
    local deb_file="${pkg_name}-${version}.deb"
    if [[ -f "${deb_file}" ]]; then
        local output_dir="${BUILD_DIR}/output"
        mkdir -p "${output_dir}"
        mv "${deb_file}" "${output_dir}/linux-${pkg_name}-${BOARD}-amlbian_${version}_${architecture}.deb"
        display_alert "MEDIAHAL_SDK" "Debian package created: ${output_dir}/linux-${pkg_name}-${BOARD}-amlbian_${version}_${architecture}.deb" "success"
    else
        display_alert "MEDIAHAL_SDK" "Failed to build Debian package" "err"
        return 1
    fi
    
    return 0
}

function amlbian_build_mediahal_sdk {
    if ! is_unit_enabled "mediahal-sdk" && ! is_unit_enabled "tvserver"; then
        display_alert "[SKIP] >> mediahal-sdk disable in BUILD_UNITS" "" "info"
        return
    fi

    if ! is_feature_contains "$DISTRO_FEATURES" "tvserver"; then
        display_alert "[SKIP] >> tvserver disable in DISTRO_FEATURES" "" "info"
        return
    fi

    display_alert "AMLOGIC: Building mediahal-sdk ${MEDIAHAL_SDK_VERSION} ${BOARD}" "info"
    
    # Define all directory paths (in .tmp directory)
    local build_root="${BUILD_DIR}/.tmp/mediahal-sdk-build-${ARCH:-arm64}"
    local build_dir="${build_root}/build"        # Build directory with source code
    local install_dir="${build_root}/install"    # Install directory

    rm -rf "${build_root}" 

    # 创建目录
    mkdir -p "${build_dir}" "${install_dir}"

    display_alert "AMLOGIC: build_dir  ${build_dir}" "info"
    display_alert "AMLOGIC: install_dir  ${install_dir}" "info"
    display_alert "AMLOGIC: output_dir  ${BUILD_DIR}/output" "info"

    # 准备源代码
    __mediahal_sdk_prepare_source "${build_dir}"
    
    # 构建示例程序
    if ! __mediahal_sdk_build_examples "${build_dir}"; then
        display_alert "AMLOGIC: mediahal-sdk examples build failed" "err"
        return 1
    fi
    
    # 安装到安装目录
    if ! __mediahal_sdk_install "${build_dir}" "${install_dir}"; then
        display_alert "AMLOGIC: mediahal-sdk installation failed" "err"
        return 1
    fi

    display_alert "AMLOGIC: Creating Debian package..." "info"
    __mediahal_sdk_create_deb "${build_dir}" "${install_dir}"

    __unset_mediahal_sdk_config

    return 0
}

# ==============================================================================
# Section 5: Main
# ==============================================================================

display_alert "AMLIbian MediaHal SDK Build Script" "Version: ${MEDIAHAL_SDK_VERSION}, Board: ${BOARD}" "info"

amlbian_build_mediahal_sdk

display_alert "AMLOGIC: mediahal-sdk build completed" "${BUILD_DIR}/output" "info"

# ==============================================================================
# End of build-aml-mediahal-sdk.sh
# ==============================================================================