From e32feb29e22a337472a7795604219a2dbb9d6042 Mon Sep 17 00:00:00 2001
From: Victor Woeltjen <victor.woeltjen@nasa.gov>
Date: Tue, 8 Dec 2015 15:41:33 -0800
Subject: [PATCH] [Duplicate] Rewrite identifiers in clones

Traverse object models of clones and rewrite domain object
identifiers that have changed during duplication.

Addresses https://github.com/nasa/openmctweb/issues/332
---
 .../entanglement/src/services/CopyTask.js     | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/platform/entanglement/src/services/CopyTask.js b/platform/entanglement/src/services/CopyTask.js
index d20c233b42..08bced4db8 100644
--- a/platform/entanglement/src/services/CopyTask.js
+++ b/platform/entanglement/src/services/CopyTask.js
@@ -45,6 +45,7 @@ define(
             this.policyService = policyService;
             this.persisted = 0;
             this.clones = [];
+            this.idMap = {};
         }
 
         function composeChild(child, parent, setLocation) {
@@ -104,6 +105,40 @@ define(
                 .then(function(){return self.firstClone;});
         }
 
+        /**
+         * Update identifiers in a cloned object model (or part of
+         * a cloned object model) to reflect new identifiers after
+         * copying.
+         * @private
+         */
+        CopyTask.prototype.rewriteIdentifiers = function (obj) {
+            var idMap = this.idMap;
+
+            if (Array.isArray(obj)) {
+                obj.forEach(function (value, index) {
+                    obj[index] = idMap[value] || value;
+                    this.rewriteIdentifiers(obj[index]);
+                }, this);
+            } else if (obj && typeof obj === 'object') {
+                Object.keys(obj).forEach(function (key) {
+                    var value = obj[key];
+                    obj[key] = idMap[value] || value;
+                    if (idMap[key]) {
+                        delete obj[key];
+                        obj[idMap[key]] = value;
+                    }
+                    this.rewriteIdentifiers(value);
+                }, this);
+            }
+        };
+
+        function rewriteIdentifiersInClones(self) {
+            self.clones.forEach(function (clone) {
+                self.rewriteIdentifiers(clone.getModel());
+            });
+            return self;
+        }
+
         /**
          * Given an array of objects composed by a parent, clone them, then
          * add them to the parent.
@@ -153,6 +188,9 @@ define(
                 // space is used.
                 clone = this.parent.useCapability("instantiation", cloneObjectModel(originalObject.getModel()));
 
+                // Record ID mappings so we can rewrite properties later
+                self.idMap[originalObject.getId()] = clone.getId();
+
                 //Iterate through child tree
                 return this.$q.when(originalObject.useCapability('composition')).then(function(composees){
                     self.deferred.notify({phase: "preparing"});
@@ -207,6 +245,7 @@ define(
             this.deferred = this.$q.defer();
 
             this.buildCopyPlan()
+                .then(rewriteIdentifiersInClones)
                 .then(persistObjects)
                 .then(addClonesToParent)
                 .then(this.deferred.resolve, this.deferred.reject);