#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import sys
import subprocess
import shutil

def preprocess_markdown(content):
    """
    預(yù)處理Markdown文本：
    1. 普通文本行之間插入空行，防止Pandoc將它們合并為一段。
    2. 表格內(nèi)部（以|開頭）保持原樣，不插入空行。
    3. 表格前后確保有空行進(jìn)行隔離。
    """
    lines = content.split('\n')
    processed_lines = []
    in_table = False
    
    for i, line in enumerate(lines):
        stripped = line.strip()
        # 簡(jiǎn)單的Markdown表格行判斷：以豎線開頭
        is_table_row = stripped.startswith('|')
        
        if is_table_row:
            if not in_table:
                # 【狀態(tài)切換】剛進(jìn)入表格
                # 如果上一行不是空行，插入一個(gè)空行做隔離
                if processed_lines and processed_lines[-1].strip() != '':
                    processed_lines.append('') 
                in_table = True
            
            # 表格行直接追加，不加額外空行
            processed_lines.append(line)
            
        else:
            if in_table:
                # 【狀態(tài)切換】剛離開表格
                in_table = False
                # 離開表格后，立即追加一個(gè)空行做隔離
                processed_lines.append('')
            
            # 普通文本處理
            processed_lines.append(line)
            
            # 如果當(dāng)前行有文字（不是空行），且不是代碼塊標(biāo)記等特殊情況，
            # 則追加一個(gè)空行，強(qiáng)制Pandoc分段
            if stripped and not stripped.startswith('```'):
                processed_lines.append('')
                
    return '\n'.join(processed_lines)

def run_conversion(input_file):
    # 1. 檢查 Pandoc 是否安裝
    if not shutil.which("pandoc"):
        print("錯(cuò)誤：未檢測(cè)到 Pandoc。請(qǐng)先安裝 Pandoc 并添加到環(huán)境變量。")
        return

    # 2. 準(zhǔn)備文件名
    file_path = os.path.abspath(input_file)
    folder = os.path.dirname(file_path)
    filename = os.path.basename(file_path)
    filename_no_ext = os.path.splitext(filename)[0]
    
    # 臨時(shí)文件和輸出文件路徑
    temp_md_file = os.path.join(folder, f"{filename_no_ext}_temp_preprocessed.md")
    output_docx = os.path.join(folder, f"{filename_no_ext}.docx")

    try:
        # 3. 讀取并預(yù)處理 Markdown
        print(f"正在處理文件: {filename} ...")
        with open(file_path, 'r', encoding='utf-8') as f:
            raw_content = f.read()
        
        # 調(diào)用核心預(yù)處理邏輯
        clean_content = preprocess_markdown(raw_content)
        
        # 寫入臨時(shí)文件
        with open(temp_md_file, 'w', encoding='utf-8') as f:
            f.write(clean_content)
            
        # 4. 調(diào)用 Pandoc 進(jìn)行轉(zhuǎn)換
        # 命令解釋：pandoc 輸入文件 -o 輸出文件 --reference-doc=模板(可選)
        cmd = f'pandoc "{temp_md_file}" -o "{output_docx}"'
        
        print("正在調(diào)用 Pandoc 進(jìn)行轉(zhuǎn)換...")
        result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
        
        if result.returncode == 0:
            print(f"? 轉(zhuǎn)換成功！")
            print(f"?? 輸出文件: {output_docx}")
        else:
            print(f"? 轉(zhuǎn)換失?。篭n{result.stderr}")

    except Exception as e:
        print(f"發(fā)生異常: {e}")
    finally:
        # 5. 清理臨時(shí)文件
        if os.path.exists(temp_md_file):
            os.remove(temp_md_file)

if __name__ == '__main__':
    # 使用方法：直接運(yùn)行或拖入文件
    if len(sys.argv) > 1:
        target_file = sys.argv[1]
        run_conversion(target_file)
    else:
        print("使用說明：")
        print("請(qǐng)將 .md 文件直接拖拽到這個(gè)腳本文件上運(yùn)行，")
        print("或者在命令行輸入：python md2docx.py <你的文件名.md>")
        input("\n按回車鍵退出...")

python md2docx.py Mon.md
python md2docx.py Tue.md

# 假設(shè) custom.docx 和腳本在同一目錄
cmd = f'pandoc "{temp_md_file}" --reference-doc="custom.docx" -o "{output_docx}"'