PowerShell中隐藏TreeView父级/指定节点复选框的实现问题

• 基于WinForm搭建了带TreeView控件的窗体，需要实现一级节点不显示复选框、二级及以下节点保留复选框的效果，效果演示如下：
  
• 检索资料时仅找到对应C#实现，自行适配PowerShell过程中遇到问题，无法正常隐藏Level 0级节点的复选框。
• 现有测试代码运行后，Level 0节点完全显示为空白，仅子节点正常带复选框，结构如下：

• 空白节点
  • [复选框]子节点
  • [复选框]子节点
• 空白节点
  • [复选框]子节点

你之前写的代码有两个核心问题：

1. 绘制Level 0节点时只调用了DrawString画文本，没有绘制节点选中背景、展开/折叠图标、焦点框这些必备元素，而且用白色 SolidBrush 绘制文本，TreeView背景默认就是白色，看起来自然是一片空白。
2. 仅靠自定义绘制无法真正隐藏复选框：TreeView的复选框是控件底层通过Win32原生渲染的，不拦截底层渲染逻辑的话，哪怕你不画复选框，控件也会留出来复选框的占位空间。

参考方案的核心逻辑分两步，和具体语言无关：

1. 拦截TreeView的窗口消息，在节点预渲染阶段，给需要隐藏复选框的节点移除TVIS_STATEIMAGEMASK状态标记，从根源上告诉控件不要给这个节点渲染复选框，也不要留占位空间。
2. 开启TreeView的所有者绘制模式，不需要隐藏复选框的节点直接交给控件默认绘制；需要隐藏复选框的节点手动绘制背景、文本、展开折叠图标、焦点框，保证显示效果和原生节点一致。

# 加载WinForm和Drawing依赖
Add-Type -AssemblyName System.Windows.Forms
Add-Type -AssemblyName System.Drawing

# 定义Win32常量
$script:NM_CUSTOMDRAW = -12
$script:TVIS_STATEIMAGEMASK = 0xF000

# 定义互操作需要的结构体
Add-Type @"
using System;
using System.Runtime.InteropServices;
public struct NMTVCUSTOMDRAW {
    public NMCUSTOMDRAW nmcd;
    public int clrText;
    public int clrTextBk;
    public int iLevel;
}
public struct NMCUSTOMDRAW {
    public NMHDR hdr;
    public int dwDrawStage;
    public IntPtr hdc;
    public RECT rc;
    public IntPtr dwItemSpec;
    public int uItemState;
    public IntPtr lItemlParam;
    public POINT pt;
}
public struct NMHDR {
    public IntPtr hwndFrom;
    public IntPtr idFrom;
    public int code;
}
public struct RECT {
    public int left, top, right, bottom;
}
public struct POINT {
    public int x, y;
}
"@

# 创建测试窗体和TreeView
$form = New-Object System.Windows.Forms.Form
$form.Text = "TreeView隐藏一级节点复选框测试"
$form.Size = New-Object System.Drawing.Size(400, 500)

$treeView = New-Object System.Windows.Forms.TreeView
$treeView.Dock = [System.Windows.Forms.DockStyle]::Fill
$treeView.CheckBoxes = $true
$treeView.DrawMode = [System.Windows.Forms.TreeViewDrawMode]::OwnerDrawAll

# 添加测试节点
$root1 = $treeView.Nodes.Add("部门A")
$root1.Nodes.Add("员工张三")
$root1.Nodes.Add("员工李四")
$root2 = $treeView.Nodes.Add("部门B")
$root2.Nodes.Add("员工王五")
$root2.Nodes.Add("员工赵六")
$form.Controls.Add($treeView)

# 注册自定义绘制事件
$treeView.Add_DrawNode({
    param($sender, $e)
    # 非0级节点直接用默认绘制，保留复选框
    if ($e.Node.Level -ne 0) {
        $e.DrawDefault = $true
        return
    }

    $e.DrawDefault = $false
    # 绘制节点背景（选中/未选中状态适配系统主题）
    $backBrush = if ($e.State.HasFlag([System.Windows.Forms.TreeNodeStates]::Selected)) {
        [System.Drawing.SystemBrushes]::Highlight
    } else {
        [System.Drawing.SystemBrushes]::Window
    }
    $e.Graphics.FillRectangle($backBrush, $e.Bounds)

    # 绘制展开/折叠图标
    $iconRect = New-Object System.Drawing.Rectangle($e.Bounds.X - 20, $e.Bounds.Y + 3, 9, 9)
    if ($e.Node.Nodes.Count -gt 0) {
        $e.Graphics.DrawRectangle([System.Drawing.Pens]::DarkGray, $iconRect)
        $e.Graphics.DrawLine([System.Drawing.Pens]::Black, $iconRect.X + 2, $iconRect.Y + 4, $iconRect.Right - 2, $iconRect.Y + 4)
        if (-not $e.Node.IsExpanded) {
            $e.Graphics.DrawLine([System.Drawing.Pens]::Black, $iconRect.X + 4, $iconRect.Y + 2, $iconRect.X + 4, $iconRect.Bottom - 2)
        }
    }

    # 绘制节点文本
    $textBrush = if ($e.State.HasFlag([System.Windows.Forms.TreeNodeStates]::Selected)) {
        [System.Drawing.SystemBrushes]::HighlightText
    } else {
        [System.Drawing.SystemBrushes]::WindowText
    }
    $e.Graphics.DrawString($e.Node.Text, $e.Node.TreeView.Font, $textBrush, $e.Bounds.X, $e.Bounds.Y + 2)

    # 绘制焦点框
    if ($e.State.HasFlag([System.Windows.Forms.TreeNodeStates]::Focused)) {
        $focusBounds = [System.Drawing.Rectangle]::Inflate($e.Bounds, -2, -2)
        [System.Windows.Forms.ControlPaint]::DrawFocusRectangle($e.Graphics, $focusBounds)
    }
})

# 注册HandleCreated事件，子类化窗口过程拦截消息移除复选框
$treeView.Add_HandleCreated({
    $tvNativeWindow = [System.Windows.Forms.NativeWindow]::new()
    $tvNativeWindow.AssignHandle($this.Handle)
    $tvNativeWindow.Add_WndProc({
        param($m)
        # 拦截右键消息，避免点到原复选框位置触发异常
        if ($m.Msg -eq 0x0204 -or $m.Msg -eq 0x0205) {
            $m.Result = [IntPtr]::Zero
            return
        }
        # 处理WM_NOTIFY消息，在预渲染阶段移除0级节点的复选框状态
        if ($m.Msg -eq 0x004E) {
            $nmHeader = [System.Runtime.InteropServices.Marshal]::PtrToStructure($m.LParam, [type]'NMHDR')
            if ($nmHeader.code -eq $script:NM_CUSTOMDRAW) {
                $cdStruct = [System.Runtime.InteropServices.Marshal]::PtrToStructure($m.LParam, [type]'NMTVCUSTOMDRAW')
                # CDDS_ITEMPREPAINT阶段，即单个节点渲染前
                if ($cdStruct.nmcd.dwDrawStage -eq 0x00010001) {
                    $currentNode = $this.GetNodeAt($cdStruct.nmcd.pt.x, $cdStruct.nmcd.pt.y)
                    if ($currentNode -and $currentNode.Level -eq 0) {
                        # 移除复选框对应的状态标记
                        $cdStruct.nmcd.uItemState = $cdStruct.nmcd.uItemState -band (-bnot $script:TVIS_STATEIMAGEMASK)
                        [System.Runtime.InteropServices.Marshal]::StructureToPtr($cdStruct, $m.LParam, $true)
                    }
                }
            }
        }
        $tvNativeWindow.DefWndProc([ref]$m)
    })
})

# 启动窗体
$form.ShowDialog()

• 所有绘制逻辑都适配了系统主题色，不会出现硬编码白色/黑色导致的显示异常，选中、聚焦、展开折叠的效果和原生TreeView节点完全一致。
• 消息拦截逻辑直接从渲染层移除了一级节点的复选框标记，不会出现复选框占位空白的问题。
• 如果需要调整隐藏复选框的节点层级，只需要修改判断条件$e.Node.Level -eq 0即可，比如要隐藏Level 1的节点复选框就把判断条件改成对应层级。

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