#!/bin/bash
set -e
# 获取系统架构 
pc_arch=`arch`
if [ $? -eq 0 ]; then
    echo "成功获取系统架构: $pc_arch"
else
    echo "获取系统架构失败"
    exit 1
fi

# 检查是否为特定机器
if cat /proc/cpuinfo|grep -i "kirin.*9.0" > /dev/null || cat /proc/cpuinfo|grep -i "pangu.*m900" > /dev/null ; then
    echo "990 或 pguw机器，不进行grub-install处理"
else
    if [ ! "x$pc_arch" = "xmips64" ] && [ ! "x$pc_arch" = "xloongarch64" ] && [ ! "x$pc_arch" = "xsw_64" ]; then
        grub-install -v
        if [ $? -eq 0 ]; then
            echo "grub-install 执行成功"
        else
            echo "grub-install 执行失败"
            exit 1
        fi
    fi
fi

# 更新grub配置
update-grub
if [ $? -eq 0 ]; then
    echo "update-grub 执行成功"
else
    echo "update-grub 执行失败"
    exit 1
fi

if [ "x$pc_arch" = "xaarch64" ]; then
	cp /boot/efi/EFI/kylin/grubaa64.efi /boot/efi/EFI/BOOT/grubaa64.efi
	cp /boot/efi/EFI/kylin/grubaa64.efi /boot/efi/EFI/neokylin/grubaa64.efi
fi

# 删除脚本文件
rm /home/fix-grub-real
if [ $? -eq 0 ]; then
    echo "成功删除 /home/fix-grub-real 文件"
else
    echo "删除 /home/fix-grub-real 文件失败"
    exit 1
fi
