Delete previous highlight before highlighting again

NucleoidAI · May 10, 2024 · 7057887 · 7057887
1 parent 9b82389
commit 7057887
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/components/ReadOnlyEditor/ReadOnlyEditor.jsx b/src/components/ReadOnlyEditor/ReadOnlyEditor.jsx
@@ -1,10 +1,13 @@
 import "highlight.js/styles/github-dark.css";
+
 import EditIcon from "@mui/icons-material/Edit";
 import ExpandLessIcon from "@mui/icons-material/ExpandLess";
 import ExpandMoreIcon from "@mui/icons-material/ExpandMore";
 import hljs from "highlight.js";
+
 import { Box, Collapse, IconButton, Stack, Typography } from "@mui/material";
 import React, { useEffect, useRef, useState } from "react";
+
 import * as prettierStandalone from "prettier/standalone";
 import * as typescriptPlugin from "prettier/parser-typescript";
 
@@ -20,6 +23,9 @@ const ReadOnlyEditor = ({
 
   useEffect(() => {
     if (codeRef.current) {
+      if (codeRef.current.dataset.highlighted === "yes") {
+        delete codeRef.current.dataset.highlighted;
+      }
       hljs.highlightElement(codeRef.current);
     }
   }, [language, value]);

diff --git a/src/lib/Project.js b/src/lib/Project.js
@@ -276,7 +276,12 @@ function compile(blocks) {
 
   const api = createAPI(uniqueFunctions);
   api.unshift(rootObject);
-  return { api, functions: uniqueFunctions, declarations, imperatives };
+  return {
+    api,
+    functions: uniqueFunctions,
+    declarations: uniqueDeclarations,
+    imperatives,
+  };
 }
 
 export default { compile };