From 0a6b5ba41c681294abd8ec259a0d1b45ce3a80c0 Mon Sep 17 00:00:00 2001
From: bin456789 <bin456789@gmail.com>
Date: Sun, 13 Oct 2024 22:47:52 +0800
Subject: [PATCH] =?UTF-8?q?windows:=20=E6=94=AF=E6=8C=81=E6=9B=BF=E6=8D=A2?=
 =?UTF-8?q?=20boot.wim?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 reinstall.sh | 12 ++++++++++--
 trans.sh     |  7 ++++++-
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/reinstall.sh b/reinstall.sh
index bd53e09..70e18cc 100644
--- a/reinstall.sh
+++ b/reinstall.sh
@@ -330,7 +330,9 @@ fix_file_type() {
     # x86 boot sector; partition 1: ...
 
     sed 's/^# //' | awk '{print $1}' | to_lower |
-        sed -e 's,dos/mbr,raw,' -e 's,x86,raw,'
+        sed -e 's,dos/mbr,raw,' \
+            -e 's,x86,raw,' \
+            -e 's,windows,wim,'
 }
 
 file_enhanced() {
@@ -1158,8 +1160,10 @@ Continue?
         # 注意 windows server 2008 r2 serverdatacenter 不用改
         image_name=${image_name/windows server 2008 server/windows longhorn server}
 
-        test_url $iso 'iso|raw'
+        test_url "$iso" 'iso|raw'
+        [ -n "$boot_wim" ] && test_url "$boot_wim" 'wim'
         eval "${step}_iso='$iso'"
+        eval "${step}_boot_wim='$boot_wim'"
         eval "${step}_image_name='$image_name'"
     }
 
@@ -3161,6 +3165,10 @@ while true; do
         iso=$2
         shift 2
         ;;
+    --boot-wim)
+        boot_wim=$2
+        shift 2
+        ;;
     --image-name)
         image_name=$(echo "$2" | to_lower)
         shift 2
diff --git a/trans.sh b/trans.sh
index 2fa95bb..bb38f21 100644
--- a/trans.sh
+++ b/trans.sh
@@ -3606,7 +3606,12 @@ install_windows() {
     mount -o ro /os/windows.iso /iso
 
     # 复制 boot.wim 到 /os，用于临时编辑
-    cp /iso/sources/boot.wim /os/boot.wim
+    if [ -n "$boot_wim" ]; then
+        # 自定义 boot.wim 链接
+        download "$boot_wim" /os/boot.wim
+    else
+        cp /iso/sources/boot.wim /os/boot.wim
+    fi
 
     # 从iso复制文件
     # 复制iso全部文件(除了boot.wim)到installer分区