WINAPI TreeView遍历问题：无法获取SHBrowseForFolder对话框子项

问题 #

在.NET Core WPF项目的MVVM架构中开发目录选择对话框，由于Forms命名空间集成性差，未使用FolderBrowserDialog，转而调用ShellNameSpace的API32原生SHBrowseForFolder对话框（FolderBrowserDialog是它的封装）。对话框能正常显示、设置根目录，但无法初始化选中指定目录。

尝试通过启动器提前获取对话框窗口句柄，进而拿到TreeView的句柄，遍历TreeView匹配目标目录时发现：使用WINAPI的TVM_GETNEXTITEM消息仅能获取根节点的子项，无法获取更深层级的子项（返回IntPtr.Zero），即使先选中父项触发动态加载也无效。

相关代码片段：

IntPtr hwndDlg = FindWindow(null, _topLevelSearchString);  // 获取对话框窗口句柄
if (hwndDlg != IntPtr.Zero)
{
    IntPtr hwndFolderCtrl = GetDlgItem(hwndDlg, _dlgControl); // 获取对话框内控件句柄
    if (hwndFolderCtrl != IntPtr.Zero)
    {
        IntPtr hwndTV = GetDlgItem(hwndFolderCtrl, _dlgTreeView);  // 获取TreeView句柄
        // ...
    }
}

遍历测试代码：

IntPtr RootItem = SendMessage(pTV, TVM_GETNEXTITEM, new(TVGN_ROOT), IntPtr.Zero); // 获取根节点
IntPtr Item1 = SendMessage(pTV, TVM_GETNEXTITEM, new(TVGN_CHILD), RootItem); // 获取根节点子项
// 选中父项并滚动到可见区域后仍无法获取子项
SendMessage(pTV, TVM_SELECTITEM, new IntPtr(TVGN_CARET), Item1);
SendMessage(pTV, TVGN_FIRSTVISIBLE, IntPtr.Zero, Item1);
IntPtr Item1_1 = SendMessage(pTV, TVM_GETNEXTITEM, new(TVGN_CHILD), Item1); // 返回IntPtr.Zero

测试结果：根节点子项可正常获取，但更深层级子项始终无法获取。

解决方案 #

1. 利用SHBrowseForFolder原生回调预设选中目录（推荐） #

不要绕弯路操作TreeView，SHBrowseForFolder本身支持通过回调函数初始化选中目录，这是最可靠的方式：

• 初始化BROWSEINFO结构体时，指定回调函数和目标目录路径。
• 在回调中收到BFFM_INITIALIZED消息时，发送BFFM_SETSELECTION消息传递目标路径，对话框会自动定位并选中对应节点。

示例代码：

[DllImport("shell32.dll", CharSet = CharSet.Auto)]
private static extern IntPtr SHBrowseForFolder(ref BROWSEINFO lpbi);

[DllImport("shell32.dll", CharSet = CharSet.Auto)]
private static extern bool SHGetPathFromIDList(IntPtr pidl, StringBuilder pszPath);

private delegate int BFFCALLBACK(IntPtr hwnd, int uMsg, IntPtr lParam, IntPtr lpData);

[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Auto)]
private struct BROWSEINFO
{
    public IntPtr hwndOwner;
    public IntPtr pidlRoot;
    public StringBuilder pszDisplayName;
    public string lpszTitle;
    public uint ulFlags;
    public BFFCALLBACK lpfn;
    public IntPtr lParam;
    public int iImage;
}

private const int BFFM_INITIALIZED = 1;
private const int BFFM_SETSELECTION = 0x0046;
private const uint BIF_RETURNONLYFSDIRS = 0x0001;

private static int BrowseCallbackProc(IntPtr hwnd, int uMsg, IntPtr lParam, IntPtr lpData)
{
    if (uMsg == BFFM_INITIALIZED)
    {
        string targetPath = Marshal.PtrToStringAuto(lpData);
        SendMessage(hwnd, BFFM_SETSELECTION, new IntPtr(1), Marshal.StringToHGlobalAuto(targetPath));
    }
    return 0;
}

// 调用示例
public string ShowFolderDialog(string initialPath)
{
    BROWSEINFO bi = new BROWSEINFO();
    bi.hwndOwner = IntPtr.Zero;
    bi.pidlRoot = IntPtr.Zero;
    bi.pszDisplayName = new StringBuilder(256);
    bi.lpszTitle = "请选择目录";
    bi.ulFlags = BIF_RETURNONLYFSDIRS;
    bi.lpfn = BrowseCallbackProc;
    bi.lParam = Marshal.StringToHGlobalAuto(initialPath);

    IntPtr pidl = SHBrowseForFolder(ref bi);
    if (pidl != IntPtr.Zero)
    {
        StringBuilder path = new StringBuilder(256);
        SHGetPathFromIDList(pidl, path);
        Marshal.FreeCoTaskMem(pidl);
        return path.ToString();
    }
    return string.Empty;
}

2. 若必须操作TreeView，解决延迟加载问题 #

原生Shell TreeView的子节点是延迟加载的，仅当节点被展开时才会创建子项。直接调用TVM_GETNEXTITEM会失败，需先触发展开并等待加载：

• 发送TVM_EXPAND消息（参数TVE_EXPAND）展开父节点。
• 加入短暂延迟或通过消息循环等待Shell完成子项加载，再尝试获取子项。

示例代码片段：

// 展开父节点
SendMessage(pTV, TVM_EXPAND, new IntPtr(TVE_EXPAND), Item1);
// 等待加载（根据实际情况调整延迟时间）
Thread.Sleep(100);
// 获取子项
IntPtr Item1_1 = SendMessage(pTV, TVM_GETNEXTITEM, new IntPtr(TVGN_CHILD), Item1);

3. 替代方案：使用Windows API Code Pack封装控件 #

若不想手动处理原生API，可使用Windows API Code Pack中的CommonOpenFileDialog，设置IsFolderPicker = true即可作为目录选择器。它是对原生对话框的封装，支持MVVM集成，可直接设置初始选中目录，无需手动操作TreeView。

内容的提问来源于stack exchange，提问作者perlfred