From b092917e687f6ad466dc8a52dfc15a6e5ec012ca Mon Sep 17 00:00:00 2001
From: Nikhil-Doye <nikhil.doye@gmail.com>
Date: Thu, 23 Oct 2025 14:47:59 -0400
Subject: [PATCH] Enhance UnifiedDatabaseNode with type safety for operations

- Updated the UnifiedDatabaseNode component to use the DatabaseOperation type for better type safety in operation handling.
- Refined operationIcons and operationLabels to utilize TypeScript's Record utility, improving code clarity and maintainability.
---
 src/components/nodes/UnifiedDatabaseNode.tsx | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/components/nodes/UnifiedDatabaseNode.tsx b/src/components/nodes/UnifiedDatabaseNode.tsx
index cdf5c5d..544cfad 100644
--- a/src/components/nodes/UnifiedDatabaseNode.tsx
+++ b/src/components/nodes/UnifiedDatabaseNode.tsx
@@ -1,6 +1,7 @@
 import React from "react";
 import { BaseNode } from "./BaseNode";
 import { NodeData } from "../../types";
+import { DatabaseOperation } from "../../types/database";
 import {
   Database,
   CheckCircle,
@@ -25,8 +26,8 @@ export const UnifiedDatabaseNode: React.FC<UnifiedDatabaseNodeProps> = ({
 }) => {
   // Get operation display info
   const getOperationInfo = () => {
-    const operation = data.config?.operation;
-    const operationIcons = {
+    const operation = data.config?.operation as DatabaseOperation;
+    const operationIcons: Record<DatabaseOperation, React.ComponentType<any>> = {
       query: Search,
       insert: Plus,
       update: Edit,
@@ -35,7 +36,7 @@ export const UnifiedDatabaseNode: React.FC<UnifiedDatabaseNodeProps> = ({
       transaction: GitBranch,
     };
 
-    const operationLabels = {
+    const operationLabels: Record<DatabaseOperation, string> = {
       query: "Query",
       insert: "Insert",
       update: "Update",