fix(client): pass TGLink to .put()

TopGunBuild · Aug 19, 2023 · 272849c · 272849c
1 parent 7a5a9bb
commit 272849c
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 3 deletions.
diff --git a/src/client/graph/graph-utils.ts b/src/client/graph/graph-utils.ts
@@ -98,18 +98,45 @@ export function getPathData(
     };
 }
 
+const isReferenceLink = (data: TGValue) => Object.keys(data).length === 1 && typeof data['#'] === 'string';
+
 export function flattenGraphData(data: TGValue, fullPath: string[]): {
     graphData: TGGraphData,
     soul: string
 }
 {
     if (isObject(data))
     {
+        if (isReferenceLink(data))
+        {
+            const propertyName = fullPath.pop();
+            const soul         = fullPath.join('/').trim();
+
+            return {
+                graphData: {
+                    [soul]: {
+                        [propertyName]: data
+                    }
+                } as TGGraphData,
+                soul
+            }
+        }
+        else
+        {
+            const soul = fullPath.join('/');
+            return {
+                graphData: flattenGraphByPath(data, [soul]),
+                soul
+            };
+        }
+        /*console.log({
+            data, fullPath
+        });
         const soul = fullPath.join('/');
         return {
             graphData: flattenGraphByPath(data, [soul]),
             soul
-        };
+        };*/
     }
     else if (fullPath.length === 1 && data === null)
     {

diff --git a/src/client/link.ts b/src/client/link.ts
@@ -166,7 +166,7 @@ export class TGLink
      * You do not need to re-save the entire object every time, TopGun will automatically
      * merge your data into what already exists as a "partial" update.
      **/
-    put(value: TGValue, opt?: TGOptionsPut): Promise<TGMessage>
+    put(value: TGValue|TGLink, opt?: TGOptionsPut): Promise<TGMessage>
     {
         return new Promise<TGMessage>((resolve) =>
         {
@@ -176,12 +176,20 @@ export class TGLink
                     'You cannot save data to user space if the user is not authorized.',
                 );
             }
-            if (!this._parent && (!isObject(value) && value !== null))
+            else if (!this._parent && (!isObject(value) && value !== null))
             {
                 throw new Error(
                     'Data at root of graph must be a node (an object) or null.',
                 );
             }
+            else if (value instanceof TGLink)
+            {
+                if (value.getPath().length > 2)
+                {
+                    throw new Error('Data at root of graph must be a node (an object) or null.');
+                }
+                value = { '#': value.getPath().join('/') };
+            }
 
             this._client.graph.putPath(
                 this.getPath(),