From 9b8fa7730c9b57351bd1d2bd9ec9db67e9c5ee43 Mon Sep 17 00:00:00 2001
From: Arpandeep Khatua <arpandeepk@gmail.com>
Date: Fri, 22 Nov 2024 17:17:40 -0800
Subject: [PATCH] Handle saved and unsaved files (adding a * to unsaved files)

---
 .../src/components/CodeEditor/CodeEditor.tsx  | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx b/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx
index 18cb6336..aa5259c2 100644
--- a/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx
+++ b/examples/experimental/nodes/frontend/src/components/CodeEditor/CodeEditor.tsx
@@ -21,7 +21,7 @@
  *
  */
 
-import React from 'react';
+import React, { useState } from 'react';
 import CodeMirror from '@uiw/react-codemirror';
 import { javascript } from '@codemirror/lang-javascript';
 import { html } from '@codemirror/lang-html';
@@ -77,6 +77,8 @@ const CodeEditor: React.FC<CodeEditorProps> = ({
 }) => {
   // Function to extract the file name from the file path
   const getFileName = (path: string) => path.split('/').pop() || path;
+  const [unsavedChanges, setUnsavedChanges] = useState<{ [key: string]: boolean }>({}); // Track unsaved changes for each file
+  const activeFileContent = openFiles.find(f => f.path === activeFile)?.content;   // Get the content of the currently active file
 
   // Handle file close action
   const handleClose = (e: React.MouseEvent, path: string) => {
@@ -84,9 +86,6 @@ const CodeEditor: React.FC<CodeEditorProps> = ({
     onFileClose(path);
   };
 
-  // Get the content of the currently active file
-  const activeFileContent = openFiles.find(f => f.path === activeFile)?.content;
-
   // Add empty lines to fill the editor to a minimum height
   const fillEmptyLines = (content: string | undefined) => {
     if (!content) return '\n'.repeat(50); // Return 50 empty lines if no content
@@ -120,9 +119,15 @@ const CodeEditor: React.FC<CodeEditorProps> = ({
     if (activeFile) {
       console.log('Saving file:', activeFile, activeFileContent);
       socket.emit('save_file', { path: activeFile, content: activeFileContent });
+      setUnsavedChanges(prev => ({ ...prev, [activeFile]: false })); // Reset unsaved changes for the active file
     }
   };
 
+  const handleChange = (activeFile: string, value: string) => {
+    onChange(activeFile, value); // Call the onChange prop
+    setUnsavedChanges(prev => ({ ...prev, [activeFile]: true })); // Set unsaved changes for the active file
+  };
+
   return (
     <div className="editor-container">
       <div className="editor-toolbar">
@@ -134,7 +139,9 @@ const CodeEditor: React.FC<CodeEditorProps> = ({
               onClick={() => onFileSelect(file.path)}
             >
               {getFileIcon(file.path)}
-              <span className="tab-title">{getFileName(file.path)}</span>
+              <span className="tab-title">
+              {unsavedChanges[file.path] ? `* ${getFileName(file.path)}` : getFileName(file.path)}
+              </span>
               <span
                 className="tab-close"
                 onClick={(e) => handleClose(e, file.path)}
@@ -159,7 +166,7 @@ const CodeEditor: React.FC<CodeEditorProps> = ({
               ...getLanguageExtension(activeFile), // Set language-specific extensions
               EditorView.lineWrapping, // Enable line wrapping
             ]}
-            onChange={(value) => onChange(activeFile, value)} // Handle content changes
+            onChange={(value) => handleChange(activeFile, value)} // Handle content changes
             basicSetup={{
               lineNumbers: true,
               highlightActiveLineGutter: true,