# -*- coding: utf-8 -*-
import subprocess
import re
import csv
import sys

def run_cmd(cmd):
    """執(zhí)行系統(tǒng)命令并返回輸出（永不返回 None）"""
    try:
        result = subprocess.run(
            cmd,
            shell=True,
            capture_output=True,
            text=True,
            encoding="gbk",
            errors="ignore"
        )
        return result.stdout or ""
    except Exception as e:
        return ""

def get_wifi_list():
    """獲取所有 WiFi 配置名稱（兼容中英文系統(tǒng)）"""
    output = run_cmd("netsh wlan show profiles")

    # 兼容中文：所有用戶配置文件
    # 兼容英文：All User Profile
    profiles = re.findall(
        r"(?:所有用戶配置文件|All User Profile)\s*:\s*(.*)",
        output
    )

    return [p.strip() for p in profiles]

def get_wifi_password(name):
    """獲取指定 WiFi 的密碼（兼容中英文系統(tǒng)）"""
    if not name:
        return ""

    output = run_cmd(f'netsh wlan show profile name="{name}" key=clear')

    # 兼容中文：關鍵內容
    # 兼容英文：Key Content
    match = re.search(
        r"(?:關鍵內容|Key Content)\s*:\s*(.*)",
        output
    )

    return match.group(1).strip() if match else ""

def export_wifi_passwords(csv_file="wifi_passwords.csv"):
    """導出所有 WiFi 名稱與密碼到 CSV，并打印到 stdout（供 C# 調用）"""
    wifi_list = get_wifi_list()
    data = []

    for wifi in wifi_list:
        pwd = get_wifi_password(wifi)
        data.append([wifi, pwd])
        print(f"{wifi} : {pwd}")

    # 寫入 CSV 文件
    try:
        with open(csv_file, "w", newline="", encoding="utf-8-sig") as f:
            writer = csv.writer(f)
            writer.writerow(["WiFi 名稱", "密碼"])
            # writer.writerows(data)
            for wifi, pwd in data:
                writer.writerow([wifi, "\t" + pwd]) # 把它當成純文本，不會吞掉前導 0
                # writer.writerow([wifi, f"'{pwd}"])

    except Exception as e:
        print(f"寫入 CSV 文件失敗: {e}")

    sys.stdout.flush()  # 關鍵：確保 C# 能讀取完整輸出

if __name__ == "__main__":
    export_wifi_passwords()

WXzhongshuge : 20130423
Ritchie : ritchie12345678
KFC FREE WIFI : 
ChinaNet-Starbucks : 
RD_Meeting_Room : 57744789
1205 : W136444564
高鐵WiFi : 
CMCC-JJJ2 : 12345678
H3C_E4C764 : 6575754635

已導出到文件：wifi_passwords.csv