#!/bin/bash

# 要添加的参数
PARAMS="cgroup_enable=cpuset cgroup_memory=1 cgroup_enable=memory swapaccount=1 systemd.unified_cgroup_hierarchy=0  extraboardargs="systemd.unified_cgroup_hierarchy=0" "
PARAMS_ARRAY=($PARAMS)

# 配置文件路径
UENV_PATH="/boot/uEnv.txt"
ARMBIANENV_PATH="/boot/armbianEnv.txt"
EXTLINUX_PATH="/boot/extlinux/extlinux.conf"

# 通用函数：检查并添加参数到配置文件
update_config_file() {
    local file_path=$1
    local added=false
    if grep -q "^cgroup_enable" "$file_path"; then
        for param in ${PARAMS_ARRAY[@]}; do
            if ! grep -q "$param" "$file_path"; then
                echo "正在将参数 '$param' 添加到 extraboardargs 中 $file_path"
                sed -i "/ / s/$/ $param/" "$file_path"
                added=true
            else
                echo "参数 '$param' 已存在于 extraboardargs 中 $file_path"
            fi
        done
    else
        echo "正在将参数添加到新的参数 中 $file_path"
        echo -n "" >> "$file_path"
        for param in ${PARAMS_ARRAY[@]}; do
            echo -n "$param " >> "$file_path"
        done
        echo "" >> "$file_path"
        added=true
    fi

    if [ "$added" = true ]; then
        echo "参数已成功添加到 $file_path"
    else
        echo "所有参数已经存在于 $file_path 中"
    fi
}

# 判断哪个文件存在并更新
if [ -f "$UENV_PATH" ]; then
    echo "更新 $UENV_PATH"
    update_config_file "$UENV_PATH"
elif [ -f "$ARMBIANENV_PATH" ]; then
    echo "更新 $ARMBIANENV_PATH"
    update_config_file "$ARMBIANENV_PATH"
elif [ -f "$EXTLINUX_PATH" ]; then
    echo "更新 $EXTLINUX_PATH"
    update_config_file "$EXTLINUX_PATH"
else
    echo "未找到合适的配置文件。请确保您在支持的设备上运行此脚本。"
    exit 1
fi

echo "所有更新已完成。请重启系统以使更改生效。"