* picker: support for getting picker instance from picker menu.

easysoft · Aug 23, 2024 · 7eca974 · 7eca974
1 parent 4869b8a
commit 7eca974
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/lib/picker/src/component/picker-menu.tsx b/lib/picker/src/component/picker-menu.tsx
@@ -38,6 +38,10 @@ export class PickerMenu extends PickPop<PickerState, PickerMenuProps> {
         return this._menu.current;
     }
 
+    get picker() {
+        return this.props.picker;
+    }
+
     componentDidMount(): void {
         super.componentDidMount();
         if (this._firstSelected === undefined) {

diff --git a/lib/picker/src/component/picker.tsx b/lib/picker/src/component/picker.tsx
@@ -331,6 +331,7 @@ export class Picker<S extends PickerState = PickerState, O extends PickerOptions
     protected _getPopProps(props: RenderableProps<O>, state: Readonly<S>): PickerMenuProps<S> {
         return {
             ...super._getPopProps(props, state),
+            picker: this as unknown as Picker,
             menu: props.menu,
             tree: props.tree,
             checkbox: props.checkbox,

diff --git a/lib/picker/src/types/picker-menu-props.ts b/lib/picker/src/types/picker-menu-props.ts
@@ -3,6 +3,7 @@ import type {TreeOptions} from '@zui/tree';
 import type {PickPopProps} from '@zui/pick';
 import type {CustomContentType} from '@zui/core';
 import type {PickerState} from './picker-state';
+import type {Picker} from '../component/picker';
 
 export interface PickerMenuProps<S extends PickerState = PickerState> extends PickPopProps<S> {
     multiple?: boolean | number;
@@ -15,6 +16,7 @@ export interface PickerMenuProps<S extends PickerState = PickerState> extends Pi
     footer?: CustomContentType;
     noMatchHint?: string;
     maxItemsCount?: number;
+    picker?: Picker;
 
     onSelect: (values: string | string[]) => void;
     onDeselect: (values: string | string[]) => void;