From cc68272b487b8a4be6de08bc6542a9309c63e3bb Mon Sep 17 00:00:00 2001
From: Emir Karabeg <emirkarabeg@berkeley.edu>
Date: Tue, 21 Jan 2025 20:38:40 -0800
Subject: [PATCH] Fixed block deletion bug

---
 .../connection-blocks.tsx}                                    | 4 ++--
 app/w/components/workflow-block/workflow-block.tsx            | 4 ++--
 stores/workflow/workflow-store.ts                             | 3 ++-
 3 files changed, 6 insertions(+), 5 deletions(-)
 rename app/w/components/workflow-block/components/{connection-block/connection-block.tsx => connection-blocks/connection-blocks.tsx} (93%)

diff --git a/app/w/components/workflow-block/components/connection-block/connection-block.tsx b/app/w/components/workflow-block/components/connection-blocks/connection-blocks.tsx
similarity index 93%
rename from app/w/components/workflow-block/components/connection-block/connection-block.tsx
rename to app/w/components/workflow-block/components/connection-blocks/connection-blocks.tsx
index 17185c3a5d..b8e9243a13 100644
--- a/app/w/components/workflow-block/components/connection-block/connection-block.tsx
+++ b/app/w/components/workflow-block/components/connection-blocks/connection-blocks.tsx
@@ -1,11 +1,11 @@
 import { useBlockConnections } from '@/app/w/hooks/use-block-connections'
 import { Card } from '@/components/ui/card'
 
-interface ConnectionBlockProps {
+interface ConnectionBlocksProps {
   blockId: string
 }
 
-export function ConnectionBlock({ blockId }: ConnectionBlockProps) {
+export function ConnectionBlocks({ blockId }: ConnectionBlocksProps) {
   const { incomingConnections, hasIncomingConnections } =
     useBlockConnections(blockId)
 
diff --git a/app/w/components/workflow-block/workflow-block.tsx b/app/w/components/workflow-block/workflow-block.tsx
index 09eaa6c1b3..73e622e1e4 100644
--- a/app/w/components/workflow-block/workflow-block.tsx
+++ b/app/w/components/workflow-block/workflow-block.tsx
@@ -4,7 +4,7 @@ import { SubBlock } from './components/sub-block/sub-block'
 import { Handle, Position } from 'reactflow'
 import { cn } from '@/lib/utils'
 import { ActionBar } from './components/action-bar/action-bar'
-import { ConnectionBlock } from './components/connection-block/connection-block'
+import { ConnectionBlocks } from './components/connection-blocks/connection-blocks'
 
 interface WorkflowBlockProps {
   id: string
@@ -53,7 +53,7 @@ export function WorkflowBlock({
   return (
     <Card className="w-[320px] shadow-md select-none group relative cursor-default">
       {selected && <ActionBar blockId={id} />}
-      <ConnectionBlock blockId={id} />
+      <ConnectionBlocks blockId={id} />
 
       <Handle
         type="target"
diff --git a/stores/workflow/workflow-store.ts b/stores/workflow/workflow-store.ts
index 23f59d17e7..10ac3852d1 100644
--- a/stores/workflow/workflow-store.ts
+++ b/stores/workflow/workflow-store.ts
@@ -32,8 +32,9 @@ export const useWorkflowStore = create<WorkflowStoreWithHistory>()(
       updateSubBlock: (blockId: string, subBlockId: string, value: any) => {
         set((state) => {
           const block = state.blocks[blockId]
-          const blockConfig = getBlock(block.type)
+          if (!block) return state
           
+          const blockConfig = getBlock(block.type)
           if (!blockConfig) return state
 
           // Create new subBlocks state