#!/usr/bin/env bash
#
# SPDX-License-Identifier: GPL-2.0
#
# Copyright (c) 2026 AMLOGIC
#
# Media modules build script for AMLbian
# This script builds media modules for different Amlogic chipsets

set -e

# Get script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
AMLBIAN_DIR="$(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
# ==============================================================================

KERNEL_VERSION="${KERNEL_VERSION:-6.12}"
BOARD="${BOARD:-s6-bq201}"
DEB_VERSION="5.15"

# LOAD_METHOD value: conf  script
LOAD_METHOD="${LOAD_METHOD:-conf}"

# 解码器支持的格式 - 根据需求配置
MEDIA_DECODER_SUPPORTED_FORMATS="${MEDIA_DECODER_SUPPORTED_FORMATS:-MPEG2 MPEG4 VC1 H264 H265 VP9 MJPEG REAL AVS AVS2 AV1}"
# 解码器 fb backend
MEDIA_DECODER_USE_FB="${MEDIA_DECODER_USE_FB:-0}"

# 编码器硬件配置 - 默认禁用
MEDIA_ENCODER_HW_CONFIG="${MEDIA_ENCODER_HW_CONFIG:-}"

# 默认配置
MEDIA_CONFIGS=" \
                 CONFIG_AMLOGIC_MEDIA_ENHANCEMENT_DOLBYVISION=y CONFIG_AMLOGIC_MEDIA_GE2D=y CONFIG_AMLOGIC_MEDIA_V4L_DEC=y"

export CONFIG_AMLOGIC_MEDIA_ENHANCEMENT_DOLBYVISION=y
export CONFIG_AMLOGIC_MEDIA_GE2D=y
export CONFIG_AMLOGIC_MEDIA_V4L_DEC=y

# ==============================================================================
# Section 4: Media Modules Build Functions
# ==============================================================================

# 配置解码器
function __configure_decoders {
    local formats=($MEDIA_DECODER_SUPPORTED_FORMATS)
    local use_fb=$MEDIA_DECODER_USE_FB
    
    for format in "${formats[@]}"; do
        case "$format" in
            VC1)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_VC1=m CONFIG_AMLOGIC_MEDIA_VDEC_VC1_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_VC1=m
                export CONFIG_AMLOGIC_MEDIA_VDEC_VC1_MULTI=m
                ;;
            H264)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_H264=m CONFIG_AMLOGIC_MEDIA_VDEC_H264_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_H264=m 
                export CONFIG_AMLOGIC_MEDIA_VDEC_H264_MULTI=m
                ;;
            H265)
                if [[ "$use_fb" == "1" ]]; then
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_H265_FB=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_H265_FB=m
                else
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_H265=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_H265=m
                fi
                ;;
            H266)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_H266=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_H266=m 
                ;;
            MPEG2)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_MPEG12=m CONFIG_AMLOGIC_MEDIA_VDEC_MPEG2_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_MPEG12=m
                export CONFIG_AMLOGIC_MEDIA_VDEC_MPEG2_MULTI=m
                ;;
            MPEG4)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_MPEG4=m CONFIG_AMLOGIC_MEDIA_VDEC_MPEG4_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_MPEG4=m
                export CONFIG_AMLOGIC_MEDIA_VDEC_MPEG4_MULTI=m
                ;;
            VP9)
                if [[ "$use_fb" == "1" ]]; then
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_VP9_FB=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_VP9_FB=m
                else
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_VP9=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_VP9=m
                fi
                ;;
            MJPEG)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_MJPEG=m CONFIG_AMLOGIC_MEDIA_VDEC_MJPEG_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_MJPEG=m
                export CONFIG_AMLOGIC_MEDIA_VDEC_MJPEG_MULTI=m
                ;;
            AVS)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_AVS_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_AVS_MULTI=m
                ;;
            AVS2)
                if [[ "$use_fb" == "1" ]]; then
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_AVS2_FB=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_AVS2_FB=m
                else
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_AVS2=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_AVS2=m
                fi
                ;;
            AVS3)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_AVS3=m"
                export CONFIG_AMLOGIC_MEDIA_VDEC_AVS3=m
                ;;
            AV1)
                if [[ "$use_fb" == "1" ]]; then
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_AV1_FB=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_AV1_FB=m
                else
                    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VDEC_AV1=m"
                    export CONFIG_AMLOGIC_MEDIA_VDEC_AV1=m
                fi
                ;;
            *)
                display_alert "MEDIA-MODULES" "Unsupported decoder format: $format" "warn"
                ;;
        esac
    done
}

# 配置编码器
function __configure_encoders {
    if [[ -z "$MEDIA_ENCODER_HW_CONFIG" ]]; then
        return 0
    fi
    
    MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VENC_COMMON=m"
    export CONFIG_AMLOGIC_MEDIA_VENC_COMMON=m
    
    local configs=($MEDIA_ENCODER_HW_CONFIG)
    for config in "${configs[@]}"; do
        case "$config" in
            HCODEC_AVC)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VENC_H264=m"
                export CONFIG_AMLOGIC_MEDIA_VENC_H264=m
                ;;
            HCODEC_JPEG)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VENC_JPEG=m"
                export CONFIG_AMLOGIC_MEDIA_VENC_JPEG=m
                ;;
            WAVE420L_HEVC)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VENC_H265=m"
                export CONFIG_AMLOGIC_MEDIA_VENC_H265=m
                ;;
            WAVE521)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VENC_MULTI=m"
                export CONFIG_AMLOGIC_MEDIA_VENC_MULTI=m
                ;;
            VC9000E)
                MEDIA_CONFIGS+=" CONFIG_AMLOGIC_MEDIA_VENC_MEMALLOC=m CONFIG_AMLOGIC_MEDIA_VENC_VCENC=m"
                export CONFIG_AMLOGIC_MEDIA_VENC_MEMALLOC=m
                export CONFIG_AMLOGIC_MEDIA_VENC_VCENC=m
                ;;
            *)
                display_alert "MEDIA-MODULES" "Unsupported encoder config: $config" "warn"
                ;;
        esac
    done
}

function __unset_media_modules_config {
    unset CONFIG_AMLOGIC_MEDIA_ENHANCEMENT_DOLBYVISION
    unset CONFIG_AMLOGIC_MEDIA_GE2D
    unset CONFIG_AMLOGIC_MEDIA_V4L_DEC

    unset CONFIG_AMLOGIC_MEDIA_VDEC_VC1
    unset CONFIG_AMLOGIC_MEDIA_VDEC_VC1_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VDEC_H264
    unset CONFIG_AMLOGIC_MEDIA_VDEC_H264_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VDEC_H265_FB
    unset CONFIG_AMLOGIC_MEDIA_VDEC_H265
    unset CONFIG_AMLOGIC_MEDIA_VDEC_H266
    unset CONFIG_AMLOGIC_MEDIA_VDEC_MPEG12
    unset CONFIG_AMLOGIC_MEDIA_VDEC_MPEG2_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VDEC_MPEG4
    unset CONFIG_AMLOGIC_MEDIA_VDEC_MPEG4_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VDEC_VP9_FB
    unset CONFIG_AMLOGIC_MEDIA_VDEC_VP9
    unset CONFIG_AMLOGIC_MEDIA_VDEC_MJPEG
    unset CONFIG_AMLOGIC_MEDIA_VDEC_MJPEG_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VDEC_AVS_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VDEC_AVS2_FB
    unset CONFIG_AMLOGIC_MEDIA_VDEC_AVS2
    unset CONFIG_AMLOGIC_MEDIA_VDEC_AVS3
    unset CONFIG_AMLOGIC_MEDIA_VDEC_AV1_FB
    unset CONFIG_AMLOGIC_MEDIA_VDEC_AV1

    unset CONFIG_AMLOGIC_MEDIA_VENC_COMMON

    unset CONFIG_AMLOGIC_MEDIA_VENC_H264
    unset CONFIG_AMLOGIC_MEDIA_VENC_JPEG
    unset CONFIG_AMLOGIC_MEDIA_VENC_H265
    unset CONFIG_AMLOGIC_MEDIA_VENC_MULTI
    unset CONFIG_AMLOGIC_MEDIA_VENC_MEMALLOC
    unset CONFIG_AMLOGIC_MEDIA_VENC_VCENC
}

function __media_modules_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 "MEDIA-MODULES" "aml-comp not found: ${aml_comp_path}" "err"
        exit 1
    fi
    
    local source_dir="${aml_comp_path}/hardware/aml-5.15/amlogic/media_modules"
    
    if [[ ! -d "${source_dir}" ]]; then
        display_alert "MEDIA-MODULES" "Source directory not found: ${source_dir}" "err"
        exit 1
    fi
    
    display_alert "MEDIA-MODULES" "Copying source from ${source_dir} to ${build_dir}" "info"
    
    # 清空构建目录
    rm -rf "${build_dir}"
    mkdir -p "${build_dir}"
    
    # 复制源代码到构建目录
    cp -r "${source_dir}"/* "${build_dir}/"
    
    display_alert "MEDIA-MODULES" "Source prepared successfully" "success"
}

function __media_modules_install_append {
    local install_dir="${1}"
    local kernel_version="${2:-KERNEL_VER}"
    
    # 复制固件文件
    display_alert "MEDIA-MODULES" "Copying firmware files chip ${CHIPSET_NAME:-}..." "info"
    
    local chipset_name="${CHIPSET_NAME:-}"
    local firmware_bin=""
    
    if [[ -n "${chipset_name}" && -f "firmware/${chipset_name}/video_ucode.bin" ]]; then
        firmware_bin="firmware/${chipset_name}/video_ucode.bin"
    elif [[ -f "firmware/video_ucode.bin" ]]; then
        firmware_bin="firmware/video_ucode.bin"
    fi
    
    if [[ -n "${firmware_bin}" ]]; then
        mkdir -p "${install_dir}/lib/firmware/video/"
        install -m 0666 "${firmware_bin}" "${install_dir}/lib/firmware/video/"
    fi

    display_alert "MEDIA-MODULES" "Copying ko files to ${kernel_version}..." "info"

    local tmp_dir="${install_dir}/lib/modules/tmp_dir"
    mkdir -p "${tmp_dir}"

    local cur_dir="${install_dir}/lib/modules/${kernel_version}"
    cd "${cur_dir}"
    find -name "*.ko" -exec cp {} ${tmp_dir}/ \;
    rm -rf "${cur_dir}"

    local dst_dir="${install_dir}/lib/modules/${kernel_version}/kernel/media"
    mkdir -p "${dst_dir}"
    cd "${dst_dir}"
    
    cp -r "${tmp_dir}"/* "${dst_dir}"

    rm -rf "${tmp_dir}" 

    if [ "$LOAD_METHOD" = "conf" ]; then
        display_alert "MEDIA-MODULES" "Generate modules-load.d configs to ${install_dir}/etc/modules-load.d ..." "info"

        # Generate modules-load.d configs
        local modules_load_dir="${install_dir}/etc/modules-load.d"
        mkdir -p "${modules_load_dir}"

        for ko_file in "${dst_dir}"/*.ko; do
            [[ -e "${ko_file}" ]] || continue

            module_name=$(basename "${ko_file}" .ko)
            echo "${module_name}" > "${modules_load_dir}/${module_name}.conf"
        done

    	display_alert "MEDIA-MODULES" "Configuring module parameters" "info" 

    elif [ "$LOAD_METHOD" = "script" ]; then
        display_alert "MEDIA-MODULES" "Copying ko load_media_modules.sh to ${install_dir}/etc ..." "info"

        local load_file="${SCRIPT_DIR}/load_media_modules.sh"
        if [[ -n "${load_file}" ]]; then
            mkdir -p "${install_dir}/etc"
            install -m 0666 "${load_file}" "${install_dir}/etc/"
        fi
    fi

	local modprobe_dir="$install_dir/etc/modprobe.d"
    mkdir -p "$modprobe_dir"

	local amvdec_ports_conf="$modprobe_dir/amvdec_ports.conf"
    if [[ -n "$kernel_version" && "$kernel_version" == *"6.12"* ]]; then
        # kernel 6.12版本使用不同的参数格式
        cat > "$amvdec_ports_conf" << 'EOF'
# Amlogic amvdec_ports module parameters
# For kernel 6.12+
options amvdec_ports amvdec_ports=multiplanar=1 amvdec_ports=vp9_need_prefix=1 amvdec_ports=av1_need_prefix=1
EOF
        display_alert "MEDIA-MODULES" "Created amvdec_ports configuration for kernel 6.12+: $amvdec_ports_conf" "info"
    else
        # 默认参数格式
        cat > "$amvdec_ports_conf" << 'EOF'
# Amlogic amvdec_ports module parameters
options amvdec_ports multiplanar=1 vp9_need_prefix=1 av1_need_prefix=1
EOF
        display_alert "MEDIA-MODULES" "Created amvdec_ports configuration: $amvdec_ports_conf" "info"
    fi
    
    # 为amvdec_mh264模块创建参数配置
    local amvdec_mh264_conf="$modprobe_dir/amvdec_mh264.conf"
    cat > "$amvdec_mh264_conf" << 'EOF'
# Amlogic amvdec_mh264 module parameters
options amvdec_mh264 error_proc_policy=8376310
EOF
    display_alert "MEDIA-MODULES" "Created amvdec_mh264 configuration: $amvdec_mh264_conf" "info"

    return 0
}

function __media_modules_build {
    local build_dir="${1}"
    local install_dir="${2}"
    
    display_alert "MEDIA-MODULES" "Starting media modules build..." "info"
    
    # 检查源目录
    if [[ ! -d "${build_dir}" ]]; then
        display_alert "MEDIA-MODULES" "Build directory not found: ${build_dir}" "err"
        return 1
    fi
    
    # 配置模块
    __configure_decoders
    __configure_encoders
    
    display_alert "MEDIA-MODULES" "MEDIA_CONFIGS: ${MEDIA_CONFIGS}" "debug"

    # 进入构建目录
    cd "${build_dir}"
    
    # 设置构建环境
    ARCH="${ARCH:-arm64}"
    CROSS_COMPILE="${CROSS_COMPILE:-aarch64-linux-gnu-}"
    
    # 查找内核构建目录
    local aml_comp_path="$(amlbian_get_aml_comp_dir)"
    if [[ -z "${aml_comp_path}" ]] || [[ ! -d "${aml_comp_path}" ]]; then
        display_alert "MEDIA-MODULES" "aml-comp not found: ${aml_comp_path}" "err"
        exit 1
    fi

    local kernel_build_dir="${aml_comp_path}/kernel/aml-${KERNEL_VERSION}/build"
    local STAGING_KERNEL_DIR="${aml_comp_path}/kernel/aml-${KERNEL_VERSION}"

    if [[ ! -d "${kernel_build_dir}" ]]; then
        display_alert "MEDIA-MODULES" "Kernel build directory not found. Expected at: ${kernel_build_dir}" "err"
        return 1
    fi

    local kernel_version=$(make -s -C "${kernel_build_dir}" kernelrelease 2>/dev/null || echo "")
    if [[ -z "${kernel_version}" ]]; then
        display_alert "MEDIA-MODULES" "Could not determine kernel version from ${kernel_build_dir}" "warn"
        kernel_version="${KERNEL_VERSION}"
    fi
    
    display_alert "MEDIA-MODULES" "Building media modules for kernel ${kernel_version}..." "info"

    # amlbian_checkout_toolchain

    # 为 a9 板子添加特定的编译标志
    if [[ "${BOARD}" == *"a9"* ]]; then
        display_alert "MEDIA-MODULES" "Adding -DDOS_REGISTERS_V3 flag for a9 board" "info"
        export KERNEL_CC="${KERNEL_CC} -DDOS_REGISTERS_V3"
        EXTRA_CFLAGS+=" -DDOS_REGISTERS_V3"
    fi

    display_alert "MEDIA-MODULES" "Building media modules..." "info"

    make -C "${kernel_build_dir}" \
        M="${build_dir}/drivers" \
        KERNELDIR="${STAGING_KERNEL_DIR}" \
        ARCH="${ARCH}" \
        CROSS_COMPILE="${CROSS_COMPILE}" \
        KERNEL_CC="${KERNEL_CC}" \
        EXTRA_CFLAGS="${EXTRA_CFLAGS}" \
        V=1 \
        modules \
        -j$(nproc) \
        2>&1 | tee make.log || {
            display_alert "MEDIA-MODULES" "Build failed" "err"
            return 1
        }
    
    # 安装模块
    display_alert "MEDIA-MODULES" "Installing media modules..." "info"
    
    make -C "${kernel_build_dir}" \
        M="${build_dir}/drivers" \
        KERNELDIR="${STAGING_KERNEL_DIR}" \
        ARCH="${ARCH}" \
        CROSS_COMPILE="${CROSS_COMPILE}" \
        INSTALL_MOD_PATH="${install_dir}" \
        INSTALL_MOD_STRIP=1 \
        modules_install \
        2>/dev/null | tee install.log || {
            display_alert "MEDIA-MODULES" "Install failed" "err"
            return 1
        }

    __media_modules_install_append "${install_dir}" "${kernel_version}"

    display_alert "MEDIA-MODULES" "Build completed successfully" "success"
    return 0
}

function __media_modules_create_deb {
    local build_dir="${1}"
    local install_dir="${2}"
    
    # 设置必要的变量
    local pkg_name="media-modules"
    local version="${DEB_VERSION}"
    local architecture="${ARCH:-arm64}"
    local maintainer="Amlbian Builder <build@amlbian.com>"
    local description="Amlogic media driver for kernel ${KERNEL_VERSION}"
    
    display_alert "MEDIA-MODULES" "Creating Debian package for ${pkg_name} ${version}..." "info"
    
    # 创建Debian包控制文件
    local deb_dir="${build_dir}/deb/${pkg_name}-${version}"
    local debian_dir="${deb_dir}/DEBIAN"
    
    mkdir -p "${debian_dir}"

    # 将安装目录的内容复制到debian-root
    if [[ -d "${install_dir}" ]] && [[ "$(ls -A "${install_dir}")" ]]; then
        cp -r "${install_dir}"/* "${deb_dir}/" 2>/dev/null || true
        display_alert "MEDIA-MODULES" "Copied installed files to ${deb_dir}" "info"
    else
        display_alert "MEDIA-MODULES" "No installed files found in ${install_dir}" "warn"
        return 0
    fi

    local aml_comp_path="$(amlbian_get_aml_comp_dir)"
    if [[ -z "${aml_comp_path}" ]] || [[ ! -d "${aml_comp_path}" ]]; then
        display_alert "MEDIA-MODULES" "aml-comp not found: ${aml_comp_path}" "err"
        exit 1
    fi

    local kernel_build_dir="${aml_comp_path}/kernel/aml-${KERNEL_VERSION}/build"

    if [[ ! -d "${kernel_build_dir}" ]]; then
        display_alert "MEDIA-MODULES" "Kernel build directory not found. Expected at: ${kernel_build_dir}" "err"
        return 1
    fi

    local kernel_version=$(make -s -C "${kernel_build_dir}" kernelrelease 2>/dev/null || echo "")
    
    # 创建control文件
    cat > "${debian_dir}/control" <<EOF
Package: ${pkg_name}
Version: ${version}
Architecture: ${architecture}
Maintainer: ${maintainer}
Description: ${description}
 This package provides kernel modules for Amlogic media hardware.
Depends:
Section: kernel
Priority: optional
EOF
    
    # 创建postinst脚本，用于更新模块依赖
    cat > "${debian_dir}/postinst" <<EOF
#!/bin/sh
set -e
  
if [ "\$1" = "configure" ]; then
    # 更新模块依赖
    if command -v depmod >/dev/null 2>&1; then
        depmod -a "${kernel_version:-}" || true
    fi
fi
EOF
     
    chmod 755 "${debian_dir}/postinst"
    
    display_alert "MEDIA-MODULES" "Building Debian package..." "info"
    
    # 构建Debian包
    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 "MEDIA-MODULES" "Debian package created: ${output_dir}/linux-${pkg_name}-${BOARD}-amlbian_${version}_${architecture}.deb" "success"
    else
        display_alert "MEDIA-MODULES" "Failed to build Debian package" "err"
        return 1
    fi
    
    return 0
}

function amlbian_build_media_modules {
    if ! is_unit_enabled "media-modules"; then
        display_alert "[SKIP] >> media-modules disable in BUILD_UNITS" "" "info"
        return
    fi
    display_alert "AMLOGIC: Building media modules ${KERNEL_VERSION} ${BOARD}" "info"
    
    # Define all directory paths (in .tmp directory)
    local build_root="${BUILD_DIR}/.tmp/media-modules-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"

    # 准备源代码
    __media_modules_prepare_source "${build_dir}"
    
    # 构建
    if ! __media_modules_build "${build_dir}" "${install_dir}"; then
        display_alert "AMLOGIC: Media modules build failed" "err"
        return 1
    fi

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

    __unset_media_modules_config

    return 0
}

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

display_alert "AMLIbian Media Modules Build Script" "Version: ${KERNEL_VERSION}, Board: ${BOARD}" "info"

amlbian_build_media_modules

display_alert "AMLOGIC: Media modules build completed" "${OUTPUT_DIR}" "info"

# ==============================================================================
# End of build-media-modules.sh
# ==============================================================================