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.
This commit is contained in:
Nikhil-Doye
2025-10-23 14:47:59 -04:00
parent b4bb7608fb
commit b092917e68
+4 -3
View File
@@ -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",