Merge pull request #28 from cadbuildr/fix-head-node-id

dag format

src/foundation/utils.py

@@ -1,9 +1,12 @@
 from foundation.types.component import Component
 from foundation.types.assembly import Assembly
+from foundation.types.serializable import serializable_nodes
 from foundation.geometry.plane import PlaneFromFrame, PlaneFactory
 import numpy as np
 import sys
 
+DAG_VERSION_FORMAT = "1.0"
+
 
 # TODO clean this code to remove copy paste
 
@@ -66,6 +69,24 @@ def check_dict(d) -> None:
                     print(k, v)
 
 
+def format_dag(dag: dict):
+    """Format the DAG to include extra information :
+    - the serializable nodes
+    - format version
+    - root node id
+
+    @param dag: the DAG to format ( see to_dag functions.)
+    """
+    # first of the key :
+
+    return {
+        "version": DAG_VERSION_FORMAT,
+        "rootNodeId": next(iter(dag.keys())),
+        "DAG": dag,
+        "serializableNodes": serializable_nodes,
+    }
+
+
 def show(component: Component) -> None:
     """Function that is actually mocked on
     server and browser, but not on the tests"""

src/foundation/utils_ext.py

@@ -1,5 +1,5 @@
 from foundation.types.component import Component
-
+from foundation.utils import format_dag
 
 try:
     import json
@@ -18,6 +18,7 @@ def create_connection(*args, **kwargs):
 
 def showExt(component: Component) -> None:
     dic = component.to_dag()
+    dic = format_dag(dic)
 
     try:
         ws = websocket.create_connection("ws://127.0.0.1:3000")