From 41f383d4e09a2fc13a3875a33021ba81c4240f4f Mon Sep 17 00:00:00 2001
From: Red-Giuliano <red.giuliano@zero-true.com>
Date: Tue, 17 Oct 2023 16:21:41 -0400
Subject: [PATCH] updated saving toml to deal better with multiline strings

---
 zt_backend/router.py | 55 ++++++++++++++++++++------------------------
 1 file changed, 25 insertions(+), 30 deletions(-)

diff --git a/zt_backend/router.py b/zt_backend/router.py
index 2cfe637e..38a90e8e 100644
--- a/zt_backend/router.py
+++ b/zt_backend/router.py
@@ -248,43 +248,38 @@ def globalStateUpdate(newCell: notebook.CodeCell=None, deletedCell: str=None, sa
 
 
     #print("Differences:", differences)
-def save_toml(zt_notebook: notebook.Notebook):
-    
-
-    tmp_uuid_file = 'notebook_'+ str(uuid.uuid4())+'.toml'
-    
-    
-    try:
-    # Create a TOML representation with only cell_id, cell_type, and code
-        toml_data = {
-            'notebookId': zt_notebook.notebookId,
-            'cells': {
-                cell_id: {
-                    'cellType': cell.cellType,
-                    'code': cell.code
-                } for cell_id, cell in zt_notebook.cells.items()
-            }
-        }
-        toml_str = toml.dumps(toml_data)
-        # Loop through each cell to reformat the 'code' field
-        for cell_id, cell in zt_notebook.cells.items():
-            code_single_line = cell.code.replace("\n", "\\n")
-            code_multi_line = '"""\n' + cell.code + '\n"""'
-            toml_str = re.sub(f'code = "{re.escape(code_single_line)}"', f'code = {code_multi_line}', toml_str)
 
+def save_toml(zt_notebook):
+    tmp_uuid_file = f'notebook_{uuid.uuid4()}.toml'
 
+    try:
         with open(tmp_uuid_file, "w") as project_file:
-            project_file.write(toml_str)
-            
+            # Write notebookId
+            project_file.write(f'notebookId = "{zt_notebook.notebookId}"\n\n')
+
+            for cell_id, cell in zt_notebook.cells.items():
+                # Write cell_id as a sub-section under cells
+                project_file.write(f'[cells.{cell_id}]\n')
+                
+                # Write cellType and code for this cell
+                project_file.write(f'cellType = "{cell.cellType}"\n')
+                
+                # Format code as a multi-line string
+                escaped_code = cell.code.encode().decode('unicode_escape')
+                project_file.write(f'code = """\n{escaped_code}"""\n\n')
+
         os.replace(tmp_uuid_file, 'notebook.toml')
 
     except Exception as e:
         print(e)
-            
-    try:
-        os.remove(tmp_uuid_file)
-    except Exception as e:
-        e
+        
+    finally:
+        try:
+            os.remove(tmp_uuid_file)
+        except Exception as e:
+            pass  # Handle error silently
+
+
 
 def remove_user_state(user_id):
     if user_id in user_states: