fix(undo): Include customColors in visual stack

Author: AmirMohammad Cheraghali

src/App.tsx

@@ -197,8 +197,9 @@ function App() {
     showLigands,
     showIons,
     showSurface,
-    customBackgroundColor: customBackgroundColor || ''
-  }), [representation, coloring, colorPalette, showLigands, showIons, showSurface, customBackgroundColor]);
+    customBackgroundColor: customBackgroundColor || '',
+    customColors
+  }), [representation, coloring, colorPalette, showLigands, showIons, showSurface, customBackgroundColor, customColors]);
 
   const handleVisualStateChange = useCallback((newState: VisualState) => {
     setRepresentation(newState.representation);
@@ -208,6 +209,7 @@ function App() {
     setShowIons(newState.showIons);
     setShowSurface(newState.showSurface);
     setCustomBackgroundColor(newState.customBackgroundColor || null);
+    setCustomColors(newState.customColors);
   }, []);
 
   const { undo, redo, canUndo, canRedo } = useVisualStack({

src/hooks/useVisualStack.ts

@@ -1,5 +1,5 @@
 import { useState, useEffect, useCallback, useRef } from 'react';
-import type { RepresentationType, ColoringType, ColorPalette } from '../types';
+import type { RepresentationType, ColoringType, ColorPalette, CustomColorRule } from '../types';
 
 export interface VisualState {
     representation: RepresentationType;
@@ -9,6 +9,7 @@ export interface VisualState {
     showIons: boolean;
     showSurface: boolean;
     customBackgroundColor: string;
+    customColors: CustomColorRule[];
 }
 
 interface UseVisualStackProps {
@@ -53,7 +54,8 @@ export function useVisualStack({ state, onChange, resetTrigger }: UseVisualStack
             prev.showLigands !== state.showLigands ||
             prev.showIons !== state.showIons ||
             prev.showSurface !== state.showSurface ||
-            prev.customBackgroundColor !== state.customBackgroundColor;
+            prev.customBackgroundColor !== state.customBackgroundColor ||
+            prev.customColors !== state.customColors;
 
         if (hasChanged) {
             setPast(prevPast => {