feat: resourceTree获取resourceList支持自定义过滤

alibaba · Jan 10, 2024 · 68cdfa7 · 68cdfa7
1 parent 53125b8
commit 68cdfa7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/packages/plugin-view-manager-pane/src/components/resourceTree/index.tsx b/packages/plugin-view-manager-pane/src/components/resourceTree/index.tsx
@@ -14,13 +14,21 @@ import { AddFile } from '../addFile';
 const { Popup } = Overlay;
 const { Item } = Menu;
 
+function filterResourceList(resourceList: IPublicModelResource[] | undefined, handler?: Function) {
+  if (typeof handler === 'function') {
+    return handler(resourceList);
+  }
+
+  return resourceList;
+}
+
 export function ResourcePaneContent(props: IPluginOptions) {
   const { workspace } = props.pluginContext || {};
   const [resourceList, setResourceList] = useState<IPublicModelResource[] | undefined>(
-    workspace?.resourceList
+    filterResourceList(workspace?.resourceList, props?.options?.filterResourceList)
   );
   workspace?.onResourceListChange(() => {
-    setResourceList(workspace.resourceList);
+    setResourceList(filterResourceList(workspace?.resourceList, props?.options?.filterResourceList));
   });
   return (
     <ResourceListTree

diff --git a/packages/plugin-view-manager-pane/src/index.tsx b/packages/plugin-view-manager-pane/src/index.tsx
@@ -32,6 +32,8 @@ export interface IOptions {
 
   handleClose?: (force?: boolean) => void;
 
+  filterResourceList?: () => {};
+
   showIconText?: boolean;
 
   skeletonConfig?: IPublicTypeSkeletonConfig;