feat(n-tree): adds option prop to the render-switcher-icon

CHANGELOG.en-US.md

@@ -33,6 +33,7 @@
 - `n-tree-select` adds `focusInput` `blurInput` methods.
 - `n-image-group` adds `on-preview-prev` `on-preview-next` prop
 - `n-tree` adds `show-line` prop, closes [#3796](https://github.com/tusen-ai/naive-ui/issues/3796), [#4554](https://github.com/tusen-ai/naive-ui/pull/4554)
+- `n-tree` adds node information for `render-switcher-icon` props, closes [#4815](https://github.com/tusen-ai/naive-ui/issues/4815).
 
 ## 2.34.4
 
@@ -62,6 +63,7 @@
 - `n-tabs` adds `placement` prop.
 - `n-spin` adds `delay` prop.
 
+
 ### i18n
 
 - Update `zhTW` locale.

CHANGELOG.zh-CN.md

@@ -33,6 +33,7 @@
 - `n-tree-select` 新增 `focusInput` `blurInput` 方法
 - `n-image-group` 新增 `on-preview-prev` `on-preview-next` 属性
 - `n-tree` 新增 `show-line` 属性，关闭 [#3796](https://github.com/tusen-ai/naive-ui/issues/3796)， [#4554](https://github.com/tusen-ai/naive-ui/issues/4554)
+- `n-tree` 为 `render-switcher-icon` 属性添加节点信息，关闭 [#4815](https://github.com/tusen-ai/naive-ui/issues/4815)
 
 ## 2.34.4
 

src/tree/demos/enUS/index.demo-entry.md

@@ -68,7 +68,7 @@ checkbox-placement.vue
 | render-label | `(info: { option: TreeOption, checked: boolean, selected: boolean }) => VNodeChild` | `undefined` | Render function of all the options' label. |  |
 | render-prefix | `(info: { option: TreeOption, checked: boolean, selected: boolean }) => VNodeChild` | `undefined` | Render function of all the options' prefix. |  |
 | render-suffix | `(info: { option: TreeOption, checked: boolean, selected: boolean }) => VNodeChild` | `undefined` | Render function of all the options' suffix. |  |
-| render-switcher-icon | `(props: { expanded: boolean, selected: boolean }) => VNodeChild` | `undefined` | Render function of option switcher icon. | 2.24.0, `props` 2.34.0 |
+| render-switcher-icon | `(props: { option: TreeOption, expanded: boolean, selected: boolean }) => VNodeChild` | `undefined` | Render function of option switcher icon. | 2.24.0, `props` 2.34.0 |
 | scrollbar-props | `object` | `undefined` | See [Scrollbar props](scrollbar#Scrollbar-Props) |  |
 | selectable | `boolean` | `true` | Whether the node can be selected. |  |
 | selected-keys | `Array<string \| number>` | `undefined` | If set, selected status will work in controlled manner. |  |

src/tree/demos/zhCN/index.demo-entry.md

@@ -74,7 +74,7 @@ expand-debug.vue
 | render-label | `(info: { option: TreeOption, checked: boolean, selected: boolean }) => VNodeChild` | `undefined` | 节点内容的渲染函数 |  |
 | render-prefix | `(info: { option: TreeOption, checked: boolean, selected: boolean }) => VNodeChild` | `undefined` | 节点前缀的渲染函数 |  |
 | render-suffix | `(info: { option: TreeOption, checked: boolean, selected: boolean }) => VNodeChild` | `undefined` | 节点后缀的渲染函数 |  |
-| render-switcher-icon | `(props: { expanded: boolean, selected: boolean }) => VNodeChild` | `undefined` | 节点展开开关的渲染函数 | 2.24.0, `props` 2.34.0 |
+| render-switcher-icon | `(props: { option: TreeOption, expanded: boolean, selected: boolean }) => VNodeChild` | `undefined` | 节点展开开关的渲染函数 | 2.24.0, `props` 2.34.0 |
 | scrollbar-props | `object` | `undefined` | 属性参考 [Scrollbar props](scrollbar#Scrollbar-Props) |   |
 | selectable | `boolean` | `true` | 节点是否可以被选中 |  |
 | selected-keys | `Array<string \| number>` | `undefined` | 如果设定则 `selected` 状态受控 |  |

src/tree/src/TreeNode.tsx

@@ -423,6 +423,7 @@ const TreeNode = defineComponent({
               selected={selected}
               loading={this.loading}
               hide={tmNode.isLeaf}
+              tmNode={this.tmNode}
               indent={indent}
               onClick={this.handleSwitcherClick}
             />

src/tree/src/TreeNodeSwitcher.tsx

@@ -1,7 +1,7 @@
 import { h, defineComponent, type PropType, inject } from 'vue'
 import { SwitcherIcon } from '../../_internal/icons'
 import { NIconSwitchTransition, NBaseLoading, NBaseIcon } from '../../_internal'
-import { treeInjectionKey } from './interface'
+import { type TmNode, treeInjectionKey } from './interface'
 
 export default defineComponent({
   name: 'NTreeSwitcher',
@@ -15,7 +15,11 @@ export default defineComponent({
     selected: Boolean,
     hide: Boolean,
     loading: Boolean,
-    onClick: Function as PropType<(e: MouseEvent) => void>
+    onClick: Function as PropType<(e: MouseEvent) => void>,
+    tmNode: {
+      type: Object as PropType<TmNode>,
+      required: true
+    }
   },
   setup (props) {
     // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
@@ -51,7 +55,8 @@ export default defineComponent({
                   return renderSwitcherIcon ? (
                     renderSwitcherIcon({
                       expanded: props.expanded,
-                      selected: props.selected
+                      selected: props.selected,
+                      option: props.tmNode.rawNode
                     })
                   ) : (
                     <NBaseIcon clsPrefix={clsPrefix} key="switcher">

src/tree/src/interface.ts

@@ -79,6 +79,7 @@ export interface InternalDropInfo {
 export type RenderSwitcherIcon = (props: {
   expanded: boolean
   selected: boolean
+  option: TreeOption
 }) => VNodeChild
 
 export type CheckOnClick = (option: TreeOption) => boolean