Check for invalid references on import

drawdb-io · Aug 2, 2024 · 52b4477 · 52b4477
1 parent ed12e3b
commit 52b4477
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/src/components/EditorHeader/Modal/ImportDiagram.jsx b/src/components/EditorHeader/Modal/ImportDiagram.jsx
@@ -86,6 +86,35 @@ export default function ImportDiagram({ setImportData, error, setError }) {
               return;
             }
 
+            let ok = true;
+            jsonObject.relationships.forEach((rel) => {
+              if (
+                !jsonObject.tables[rel.startTableId] ||
+                !jsonObject.tables[rel.endTableId]
+              ) {
+                setError({
+                  type: STATUS.ERROR,
+                  message: `Relationship ${rel.name} references a table that does not exist.`,
+                });
+                ok = false;
+                return;
+              }
+
+              if (
+                !jsonObject.tables[rel.startTableId].fields[rel.startFieldId] ||
+                !jsonObject.tables[rel.endTableId].fields[rel.endFieldId]
+              ) {
+                setError({
+                  type: STATUS.ERROR,
+                  message: `Relationship ${rel.name} references a field that does not exist.`,
+                });
+                ok = false;
+                return;
+              }
+            });
+
+            if (!ok) return;
+
             setImportData(jsonObject);
             if (diagramIsEmpty()) {
               setError({