fix(Block): fix block's selection

gravity-ui · Oct 8, 2024 · 5438d69 · 5438d69
1 parent c2e4cd2
commit 5438d69
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/react-component/Block.tsx b/src/react-component/Block.tsx
@@ -3,7 +3,7 @@ import { TBlock } from "../components/canvas/blocks/Block";
 import { Graph } from "../graph";
 import "./Block.css";
 import { setCssProps } from "../utils/functions/cssProp";
-import { useBlockViewState } from "./hooks/useBlockState";
+import { useBlockState, useBlockViewState } from "./hooks/useBlockState";
 
 export const GraphBlock = <T extends TBlock>({
   graph,
@@ -20,6 +20,7 @@ export const GraphBlock = <T extends TBlock>({
 }) => {
   const containerRef = useRef<HTMLDivElement>(null);
   const viewState = useBlockViewState(graph, block);
+  const state = useBlockState(graph, block);
 
   useLayoutEffect(() => {
     setCssProps(containerRef.current, {
@@ -44,7 +45,7 @@ export const GraphBlock = <T extends TBlock>({
 
   return (
     <div className={`graph-block-container ${containerClassName}`} ref={containerRef}>
-      <div className={`graph-block-wrapper ${className} ${block.selected ? "selected" : ""}`}>{children}</div>
+      <div className={`graph-block-wrapper ${className} ${state.selected ? "selected" : ""}`}>{children}</div>
     </div>
   );
 };