//! 快速面板：检测前台 Explorer 窗口的当前目录。
//!
//! 必须在快捷键回调（`show_popup`）内调用：此时前台窗口仍是资源管理器，
//! 面板尚未取得焦点，`GetForegroundWindow` 拿到的才是 Explorer 主窗口；
//! 若等面板显示后再调用，前台就变成面板自身了。
//!
//! 思路（Listary / PowerToys Run 同款）：前台窗口 HWND 匹配 `IShellWindows`
//! 中某个 Shell 窗口 → 取其 `LocationURL`（file:///...）→ 转成本地路径。
//!
//! Win11 多选项卡：同一顶层窗口下每个选项卡都是独立的 `IShellWindows` 条目，
//! 共享顶层 HWND。活动选项卡的内容窗口（`ShellTabWindowClass`）在子窗口
//! z-order 顶层，用 `IID_IShellBrowser` 作为 `QueryService` 的 service ID 获取
//! 每个选项卡自己的 `IShellBrowser`（而非 `SID_STopLevelBrowser` 返回的顶层
//! browser），再通过 `GetWindow()` 拿到该选项卡的内容窗口句柄，与活动选项卡
//! 的内容窗口比对，从而定位当前正在浏览的选项卡（`IsWindowVisible` 对所有
//! 选项卡都成立，不可用）。

use std::path::Path;

#[cfg(windows)]
use windows::core::ComInterface;
#[cfg(windows)]
use windows::Win32::Foundation::HWND;
#[cfg(windows)]
use windows::Win32::System::Com::{
    CoCreateInstance, CoInitializeEx, CoUninitialize, CLSCTX_ALL, COINIT_APARTMENTTHREADED,
    IServiceProvider,
};
#[cfg(windows)]
use windows::Win32::System::Variant::{VARIANT, VT_I4};
#[cfg(windows)]
use windows::Win32::UI::Shell::{
    IWebBrowserApp, IShellBrowser, IShellWindows, ShellWindows,
};
#[cfg(windows)]
use windows::Win32::UI::WindowsAndMessaging::{
    GetClassNameW, GetForegroundWindow, GetWindow, GW_CHILD, GW_HWNDNEXT,
};

/// 检测前台 Explorer 窗口的当前目录。
/// 返回 `None`：前台不是 Explorer / COM 初始化失败 / URL 无法转路径。
#[cfg(windows)]
pub fn detect_explorer_folder() -> Option<String> {
    // 首次 COM 初始化失败（例如已在 MTA 线程）时，后续 COM 调用一般仍可用，忽略错误继续。
    let _ = unsafe { CoInitializeEx(None, COINIT_APARTMENTTHREADED) };
    let fg = unsafe { GetForegroundWindow() };
    let result = if fg.0 == 0 {
        None
    } else {
        unsafe { find_folder_for_hwnd(fg) }
    };
    unsafe { CoUninitialize() };
    result
}

#[cfg(windows)]
unsafe fn find_folder_for_hwnd(fg: HWND) -> Option<String> {
    let shell: IShellWindows = CoCreateInstance(&ShellWindows, None, CLSCTX_ALL).ok()?;
    let count = shell.Count().ok()?;
    // Win11 多选项卡：活动选项卡的内容窗口（ShellTabWindowClass）在子窗口 z-order
    // 顶层。在 IShellWindows 条目中，用 IShellBrowser::GetWindow() 取到的内容窗口
    // HWND 与它比对，即可定位当前正在浏览的选项卡（IsWindowVisible 对所有选项卡
    // 都成立，不可用）。
    let active_tab = find_active_shell_tab(fg);
    for i in 0..count {
        // 索引过期的窗口会返回失败，跳过继续即可，不能 `?` 提前结束整个循环。
        // 0.52 的 Win32 VARIANT 无 From<i32>，手动构造 VT_I4 变体。
        let mut index = VARIANT::default();
        {
            let value = &mut *index.Anonymous.Anonymous;
            value.vt = VT_I4;
            value.Anonymous.lVal = i;
        }
        let Ok(dispatch) = shell.Item(index) else { continue };
        let Ok(app) = dispatch.cast::<IWebBrowserApp>() else { continue };
        // 只考虑前台顶层窗口对应的条目；同一窗口的多个选项卡条目共享顶层句柄。
        let Ok(hwnd) = app.HWND() else { continue };
        if HWND(hwnd.0) != fg {
            continue;
        }
        // 有选项卡时，必须匹配活动选项卡的内容窗口；否则退化为任意条目（旧版无选项卡）。
        if let Some(active) = active_tab {
            let Ok(svc) = app.cast::<IServiceProvider>() else { continue };
            let Ok(browser) = svc.QueryService::<IShellBrowser>(&IShellBrowser::IID) else {
                continue;
            };
            let Ok(this_tab) = browser.GetWindow() else { continue };
            if this_tab != active {
                continue;
            }
        }
        if let Ok(url) = app.LocationURL() {
            return url_to_path(&url.to_string());
        }
    }
    None
}

/// 枚举前台窗口的子窗口（z-order 自上而下），返回第一个类名为 `ShellTabWindowClass`
/// 的窗口句柄，即 Win11 资源管理器活动选项卡的内容窗口；无选项卡时返回 `None`。
#[cfg(windows)]
unsafe fn find_active_shell_tab(fg: HWND) -> Option<HWND> {
    const CLASS: &str = "ShellTabWindowClass";
    let mut hwnd = GetWindow(fg, GW_CHILD);
    while hwnd.0 != 0 {
        let mut buf = [0u16; 64];
        let len = GetClassNameW(hwnd, &mut buf);
        if len > 0 {
            let name = String::from_utf16_lossy(&buf[..len as usize]);
            if name == CLASS {
                return Some(hwnd);
            }
        }
        hwnd = GetWindow(hwnd, GW_HWNDNEXT);
    }
    None
}

/// 把 `file:///C:/xxx`（可能带百分号编码）转成本地路径，仅接受目录。
#[cfg(windows)]
fn url_to_path(url: &str) -> Option<String> {
    if !url.starts_with("file:") {
        return None;
    }
    let parsed = url::Url::parse(url).ok()?;
    let path = parsed.to_file_path().ok()?;
    let path = Path::new(&path);
    if path.is_dir() {
        Some(path.to_string_lossy().to_string())
    } else {
        None
    }
}

/// 非 Windows 平台占位：保持模块可编译。
#[cfg(not(windows))]
pub fn detect_explorer_folder() -> Option<String> {
    None
}