import os
from docx import Document
from docx.oxml import OxmlElement
from docx.oxml.ns import qn
from docx.shared import Inches
from copy import deepcopy

# 定義 XML 命名空間常數
W_SECTPR = qn('w:sectPr')
# '{http://schemas.openxmlformats.org/wordprocessingml/2006/main}sectPr'

def get_xml_structure(doc, limit=500):
    """輔助函數：查看 document body 的最後幾個 XML tag"""
    xml = doc.element.body.xml
    # 簡單截取尾部方便觀察
    return "..." + xml[-limit:] if len(xml) > limit else xml

# 1. 產生一個有「特殊版面設定」的原始檔
source_path = "demo_source_layout.docx"
doc = Document()
doc.add_paragraph("這是原始文件，設定為特殊尺寸。")

# 修改最後一個 section 的版面設定
section = doc.sections[-1]
section.page_width = Inches(10)  # 寬度 10 吋
section.page_height = Inches(5)  # 高度 5 吋 (很扁)

doc.save(source_path)
print(f"已生成示範檔: {source_path}")

# 檢查它的 XML 尾巴，應該可以看到 <w:sectPr> 裡面有 <w:pgSz w:w="..." w:h="..."/>
doc_check = Document(source_path)
print("\n[原始檔 XML 尾部檢視]:")
print(get_xml_structure(doc_check, 300))

# 【關鍵步驟】把這個珍貴的原始設定存到全域變數，模擬程式運行時的狀態
global_CT_SectPr = doc_check.element.body.findall(W_SECTPR)[-1]
print(f"\n[已捕獲 Global SectPr]: {global_CT_SectPr}")

def save_with_protection(filename, use_global=True, use_backup=True):
    """
    這是一個模擬「存檔前最後檢查」的函數
    
    參數:
    use_global: 是否允許使用原始文件的設定 (Layer 1)
    use_backup: 是否允許使用 python-docx 預設的設定 (Layer 2)
    """
    
    # 1. 建立一個全新的小文件 (這是容器)
    new_doc = Document()
    new_doc.add_paragraph(f"這是測試檔案: {filename}")
    new_doc.add_paragraph("請檢查我的版面大小。")
    
    body = new_doc.element.body
    
    # --- [模擬操作] ---
    # 在這裡，我們模擬為了清理文件，不小心殺紅了眼，把原本 body 裡的 sectPr 也找出來了
    
    # [Layer 2 預備動作]：在殺掉它之前，先看看有沒有備份可以留
    backup_sectPr = None
    
    for el in list(body):
        if el.tag == W_SECTPR:
            backup_sectPr = el # 抓到了！這是 library 給我們的預設值
            body.remove(el)    # 把它從 body 移除
            
    print(f"\n--- 處理 {filename} ---")
    print(f"清理後 body 是否還有 sectPr? {'w:sectPr' in body.xml}")
    
    # --- [三層防護邏輯開始] ---
    
    final_sectPr = None
    
    # 層級 1: Global Source (最優先)
    # 我們看看全域變數有沒有壞掉，或者使用者是否允許使用
    if use_global and global_CT_SectPr is not None:
        print("✅ 使用 Layer 1: 原始文件設定 (Global)")
        final_sectPr = global_CT_SectPr
        
    # 層級 2: Library Backup (次優先)
    # 如果全域的沒了，那剛剛殺掉的那個預設值，還能用嗎？
    elif use_backup and backup_sectPr is not None:
        print("⚠️ 使用 Layer 2: 程式庫預設備份 (Backup)")
        final_sectPr = backup_sectPr
        
    # 層級 3: The Safety Net (最終手段)
    # 如果連備份都沒有 (例如 body 是早就被清空的)，我們必須無中生有
    else:
        print("🛡️ 使用 Layer 3: 最終空標籤 (Safety Net)")
        # 這就是我們的主角：憑空創造一個合法的印章
        final_sectPr = OxmlElement('w:sectPr')

    # [Action] 蓋章！
    if final_sectPr is not None:
        body.append(deepcopy(final_sectPr))
    
    new_doc.save(filename)
    return filename

# 情境 1: 快樂路徑 - 原始資料都在
save_with_protection("Result_L1_Global.docx", use_global=True, use_backup=True)

# 情境 2: 原始資料遺失 - 退回第二線
# (模擬 global 變數變成 None 或找不到)
save_with_protection("Result_L2_Backup.docx", use_global=False, use_backup=True)

# 情境 3: 絕境 - 連備份都沒了 (極端狀況 / 軟體工程防禦)
# 您可能會想：「Document() 產生時不是一定會有 sectPr 嗎？」
# 沒錯，但在軟體工程中，我們不能假設依賴的函式庫 (python-docx) 永遠不變，
# 或者這段程式碼被移動到「body 已經被外部程式清空」的流程中執行。
# 這層「不可由之」的防護，是為了確保程式在 0.01% 的異常狀態下也不會 Crash。
save_with_protection("Result_L3_Safety.docx", use_global=False, use_backup=False)

def inspect_sectpr(filename, label):
    """輔助函數：讀取並印出 docx 的最後一段 xml 與 sectPr 狀態"""
    if not os.path.exists(filename):
        print(f"[{label}] 檔案未生成: {filename}")
        return

    doc_inspect = Document(filename)
    xml_content = doc_inspect.element.body.xml
    sectpr_tags = doc_inspect.element.body.findall(W_SECTPR)
    
    print(f"\n=== 檢查 {label} ({filename}) ===")
    
    # 檢查 sectPr 標籤是否存在
    if sectpr_tags:
        last_sectpr = sectpr_tags[-1]
        print(f"✅ 成功找到 sectPr 標籤: {last_sectpr.tag}")
        
        # 進一步檢查是否有頁面大小設定 (w:pgSz)
        pgSz = last_sectpr.find(qn('w:pgSz'))
        if pgSz is not None:
             w = pgSz.get(qn('w:w'))
             h = pgSz.get(qn('w:h'))
             print(f"   📏 版面尺寸: 寬={w} (Twips), 高={h} (Twips)")
             # 註: Twips 是 'Twentieth of an Inch Point'，
             # 1 twip = 1/20 pt = 1/1440 英吋"
             # 註: 14400 Twips = 10 inches, 7200 Twips = 5 inches
        else:
             print("   ⚠️ 標籤內無尺寸設定 (這就是 Layer 3 的特徵: 空殼標籤 -> 交給 Word 預設值)")
    else:
        print("❌ 錯誤：找不到 sectPr 標籤！檔案可能損毀。")

    # 印出 XML 尾部供確認
    # print(f"   [XML Tail]: ...{xml_content[-150:]}")

# 一次檢查這三個檔案，見證三層防護的差異
inspect_sectpr("Result_L1_Global.docx", "情境 1 (Global)")
inspect_sectpr("Result_L2_Backup.docx", "情境 2 (Backup)")
inspect_sectpr("Result_L3_Safety.docx", "情境 3 (Safety Net)")