feat: resourceTree获取resourceList支持自定义过滤 (#98)

alibaba · Jan 15, 2024 · 492ce98 · 492ce98
1 parent 0c8961f
commit 492ce98
Show file tree

Hide file tree

Showing 2 changed files with 17 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;
@@ -135,6 +137,11 @@ ViewManagerPane.meta = {
         type: 'function',
         description: '',
       },
+      {
+        key: 'filterResourceList',
+        type: 'function',
+        description: '',
+      },
       {
         key: 'showIconText',
         type: 'boolean',