From 91c229c9c4a9720aaf6aa85da40eb487339897bb Mon Sep 17 00:00:00 2001
From: Luis Fernando Molina <Luis-Fernando-Molina@users.noreply.github.com>
Date: Mon, 28 Oct 2019 16:05:57 +0100
Subject: [PATCH] References Being Saved

---
 .../Workspace/FunctionLibraries/ProtocolNode.js        | 10 +++++++++-
 DesignerSpace/Workspace/Workspace.js                   |  2 +-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/DesignerSpace/Workspace/FunctionLibraries/ProtocolNode.js b/DesignerSpace/Workspace/FunctionLibraries/ProtocolNode.js
index 80a9c7da9..d5a615e43 100644
--- a/DesignerSpace/Workspace/FunctionLibraries/ProtocolNode.js
+++ b/DesignerSpace/Workspace/FunctionLibraries/ProtocolNode.js
@@ -4,7 +4,7 @@ function newProtocolNode () {
   }
   return thisObject
 
-  function getProtocolNode (node, removePersonalData, parseCode, includeIds, includePayload) {
+  function getProtocolNode (node, removePersonalData, parseCode, includeIds, includePayload, includeReferences) {
     if (node === undefined) { return }
     switch (node.type) {
       case 'Code':
@@ -1013,6 +1013,14 @@ function newProtocolNode () {
         if (includePayload) {
           object.savedPayload = getSavedPayload(node)
         }
+        if (includeReferences) {
+          let referenceChildren = []
+          for (let i = 0; i < node.referenceChildren.length; i++) {
+            let child = node.referenceChildren[i]
+            referenceChildren.push(child.id)
+          }
+          object.referenceChildren = referenceChildren
+        }
         return object
       }
     }
diff --git a/DesignerSpace/Workspace/Workspace.js b/DesignerSpace/Workspace/Workspace.js
index 931604710..c087df957 100644
--- a/DesignerSpace/Workspace/Workspace.js
+++ b/DesignerSpace/Workspace/Workspace.js
@@ -142,7 +142,7 @@ function newWorkspace () {
     let stringifyReadyNodes = []
     for (let i = 0; i < thisObject.workspaceNode.rootNodes.length; i++) {
       let rootNode = thisObject.workspaceNode.rootNodes[i]
-      let node = functionLibraryProtocolNode.getProtocolNode(rootNode, removePersonalData, false, true, true)
+      let node = functionLibraryProtocolNode.getProtocolNode(rootNode, removePersonalData, false, true, true, true)
       if (node) {
         stringifyReadyNodes.push(node)
       }