From fe319d3d95e8c91e44587dba46b3b1423ca1eba4 Mon Sep 17 00:00:00 2001
From: Joel Keyser <keyser.joel@gmail.com>
Date: Wed, 14 Feb 2024 18:19:51 -0600
Subject: [PATCH] fix: cant upload a topic again due to id conflicts

---
 src/web/topic/store/utilActions.ts | 55 +++++++++++++++++++++++++++++-
 1 file changed, 54 insertions(+), 1 deletion(-)

diff --git a/src/web/topic/store/utilActions.ts b/src/web/topic/store/utilActions.ts
index 1a5322d0..b337d72a 100644
--- a/src/web/topic/store/utilActions.ts
+++ b/src/web/topic/store/utilActions.ts
@@ -1,8 +1,16 @@
+import { v4 as uuid } from "uuid";
 import { StorageValue } from "zustand/middleware";
 
 import { errorWithData } from "../../../common/errorHandling";
 import { emitter } from "../../common/event";
-import { TopicStoreState, initialState, playgroundUsername, useTopicStore } from "./store";
+import { Edge, Node } from "../utils/graph";
+import {
+  TopicStoreState,
+  UserScores,
+  initialState,
+  playgroundUsername,
+  useTopicStore,
+} from "./store";
 import { isPlaygroundTopic } from "./utils";
 
 export const getPersistState = () => {
@@ -19,6 +27,49 @@ export const getScoringUsernames = () => {
   return Object.keys(userScores);
 };
 
+/**
+ * Generate new ids for nodes and edges to avoid conflicts with the topic that this was downloaded
+ * from.
+ *
+ * Also, ensure that related scores, claims, and edges are updated accordingly.
+ */
+const ensureUnique = (nodes: Node[], edges: Edge[], userScores: UserScores) => {
+  [...nodes, ...edges].forEach((graphPart) => {
+    const newId = uuid();
+
+    // update edges
+    edges.forEach((edge) => {
+      // eslint-disable-next-line functional/immutable-data, no-param-reassign
+      if (edge.source === graphPart.id) edge.source = newId;
+      // eslint-disable-next-line functional/immutable-data, no-param-reassign
+      if (edge.target === graphPart.id) edge.target = newId;
+    });
+
+    // update claims
+    [...nodes, ...edges].forEach((otherGraphPart) => {
+      if (otherGraphPart.data.arguedDiagramPartId === graphPart.id) {
+        // eslint-disable-next-line functional/immutable-data, no-param-reassign
+        otherGraphPart.data.arguedDiagramPartId = newId;
+      }
+    });
+
+    // update scores
+    Object.entries(userScores).forEach(([_username, partScores]) => {
+      const partScore = partScores[graphPart.id];
+      if (partScore) {
+        // eslint-disable-next-line functional/immutable-data, no-param-reassign
+        partScores[newId] = partScore;
+        // eslint-disable-next-line functional/immutable-data, @typescript-eslint/no-dynamic-delete, no-param-reassign
+        delete partScores[graphPart.id];
+      }
+    });
+
+    // update graph part
+    // eslint-disable-next-line functional/immutable-data, no-param-reassign
+    graphPart.id = newId;
+  });
+};
+
 export const setTopicData = (state: TopicStoreState, sessionUsername?: string) => {
   // Don't override topic - this way, topic data from playground can be downloaded and uploaded as
   // a means of saving playground data to the db.
@@ -33,6 +84,8 @@ export const setTopicData = (state: TopicStoreState, sessionUsername?: string) =
   const myUsername = isPlaygroundTopic(topic) ? playgroundUsername : sessionUsername;
   const userScores = myScores && myUsername ? { [myUsername]: myScores } : {};
 
+  ensureUnique(state.nodes, state.edges, userScores);
+
   useTopicStore.setState({ ...state, topic, userScores }, false, "setState");
 
   emitter.emit("loadedTopicData");