From 7bef50762c2032ce640aedcc10a16739e9471885 Mon Sep 17 00:00:00 2001
From: huxingyi1997 <xingyi.hu@shopee.com>
Date: Fri, 3 Mar 2023 19:02:19 +0800
Subject: [PATCH] feat: disable behaviors according to components in
 outline-tree

---
 .../src/views/tree-title.tsx                   | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/packages/plugin-outline-pane/src/views/tree-title.tsx b/packages/plugin-outline-pane/src/views/tree-title.tsx
index 7a3718366..05fa0f7be 100644
--- a/packages/plugin-outline-pane/src/views/tree-title.tsx
+++ b/packages/plugin-outline-pane/src/views/tree-title.tsx
@@ -6,7 +6,6 @@ import { IPublicModelPluginContext, IPublicApiEvent } from '@alilc/lowcode-types
 import TreeNode from '../controllers/tree-node';
 import { IconLock, IconUnlock, IconArrowRight, IconEyeClose, IconEye, IconCond, IconLoop, IconRadioActive, IconRadio, IconSetting } from '../icons';
 
-
 function emitOutlineEvent(event: IPublicApiEvent, type: string, treeNode: TreeNode, rest?: Record<string, unknown>) {
   const node = treeNode?.node;
   const npm = node?.componentMeta?.npm;
@@ -35,6 +34,8 @@ export default class TreeTitle extends Component<{
     title: '',
   };
 
+  private lastInput?: HTMLInputElement;
+
   private enableEdit = (e) => {
     e.preventDefault();
     this.setState({
@@ -66,8 +67,6 @@ export default class TreeTitle extends Component<{
     }
   };
 
-  private lastInput?: HTMLInputElement;
-
   private setCaret = (input: HTMLInputElement | null) => {
     if (!input || this.lastInput === input) {
       return;
@@ -96,6 +95,8 @@ export default class TreeTitle extends Component<{
     const { editing } = this.state;
     const isCNode = !treeNode.isRoot();
     const { node } = treeNode;
+    const { componentMeta } = node;
+    const availableActions = componentMeta ? componentMeta.availableActions.map((availableAction) => availableAction.name) : [];
     const isNodeParent = node.isParentalNode;
     const isContainer = node.isContainerNode;
     let style: any;
@@ -112,8 +113,11 @@ export default class TreeTitle extends Component<{
     const { intlNode, common, config } = pluginContext;
     const Tip = common.editorCabin.Tip;
     const Title = common.editorCabin.Title;
-    const shouldShowHideBtn = isCNode && isNodeParent && !isModal;
-    const shouldShowLockBtn = config.get('enableCanvasLock', false) && isContainer && isCNode && isNodeParent;
+    const couldHide = availableActions.includes('hide');
+    const couldLock = availableActions.includes('lock');
+    const couldUnlock = availableActions.includes('unlock');
+    const shouldShowHideBtn = isCNode && isNodeParent && !isModal && couldHide;
+    const shouldShowLockBtn = config.get('enableCanvasLock', false) && isContainer && isCNode && isNodeParent && ((couldLock && !node.isLocked) || (couldUnlock && node.isLocked));
     const shouldEditBtn = isCNode && isNodeParent;
     return (
       <div
@@ -221,7 +225,6 @@ class RenameBtn extends Component<{
   }
 }
 
-
 class LockBtn extends Component<{
   treeNode: TreeNode;
   pluginContext: IPublicModelPluginContext;
@@ -273,7 +276,6 @@ class HideBtn extends Component<{
   }
 }
 
-
 class ExpandBtn extends Component<{
   treeNode: TreeNode;
   pluginContext: IPublicModelPluginContext;
@@ -301,4 +303,4 @@ class ExpandBtn extends Component<{
       </div>
     );
   }
-}
\ No newline at end of file
+}