From 84ad494c52191539f892fd849701add7d6ca588a Mon Sep 17 00:00:00 2001
From: zzxming <zzxming77@163.com>
Date: Tue, 3 Dec 2024 14:29:00 +0800
Subject: [PATCH] fix: used color record out of max count

---
 src/modules/table-menu/constants.ts         |  3 ++-
 src/modules/table-menu/table-menu-common.ts | 28 +++++++++++----------
 2 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/src/modules/table-menu/constants.ts b/src/modules/table-menu/constants.ts
index 4fb5512..9bf6708 100644
--- a/src/modules/table-menu/constants.ts
+++ b/src/modules/table-menu/constants.ts
@@ -13,7 +13,6 @@ import SplitCell from '../../svg/split-cell.svg';
 
 export const menuColorSelectClassName = 'color-selector';
 export const contextmenuClassName = 'contextmenu';
-export const usedColors = new Set<string>();
 export const defaultTools: Tool[] = [
   {
     name: 'InsertTop',
@@ -128,6 +127,8 @@ export const defaultTools: Tool[] = [
   },
 ];
 
+export const maxSaveColorCount = 10;
+export const usedColors = new Set<string>();
 export const colorClassName = {
   used: 'table-color-used',
   item: 'table-color-item',
diff --git a/src/modules/table-menu/table-menu-common.ts b/src/modules/table-menu/table-menu-common.ts
index cb10002..96d734d 100644
--- a/src/modules/table-menu/table-menu-common.ts
+++ b/src/modules/table-menu/table-menu-common.ts
@@ -2,7 +2,7 @@ import type Quill from 'quill';
 import type { TableUp } from '../..';
 import type { TableMenuOptions, ToolOption, TooltipInstance } from '../../utils';
 import { createTooltip, debounce, defaultColorMap, isArray, isFunction, randomId } from '../../utils';
-import { colorClassName, defaultTools, menuColorSelectClassName, usedColors } from './constants';
+import { colorClassName, defaultTools, maxSaveColorCount, menuColorSelectClassName, usedColors } from './constants';
 
 export type TableMenuOptionsInput = Partial<Omit<TableMenuOptions, 'texts'>>;
 export class TableMenuCommon {
@@ -21,15 +21,15 @@ export class TableMenuCommon {
       if (!isArray(colorValue)) {
         colorValue = [];
       }
-      colorValue.map((c: string) => usedColors.add(c));
+      colorValue.slice(-1 * maxSaveColorCount).map((c: string) => usedColors.add(c));
     }
     catch {}
 
     this.updateUsedColor = debounce((color?: string) => {
       if (!color) return;
       usedColors.add(color);
-      if (usedColors.size > 10) {
-        const saveColors = Array.from(usedColors).slice(-10);
+      if (usedColors.size > maxSaveColorCount) {
+        const saveColors = Array.from(usedColors).slice(-1 * maxSaveColorCount);
         usedColors.clear();
         saveColors.map(v => usedColors.add(v));
       }
@@ -37,17 +37,19 @@ export class TableMenuCommon {
       localStorage.setItem(this.options.localstorageKey, JSON.stringify(Array.from(usedColors)));
       const usedColorWrappers = Array.from(document.querySelectorAll(`.${this.colorItemClass}.${colorClassName.used}`));
       for (const usedColorWrapper of usedColorWrappers) {
-        if (!usedColorWrapper) continue;
-
-        const colorItem = usedColorWrapper.querySelectorAll(`.${colorClassName.item}[style*="background-color: ${color}"]`);
-        for (const item of Array.from(colorItem)) {
-          item.remove();
-        }
-
         const newColorItem = document.createElement('div');
         newColorItem.classList.add(colorClassName.item);
         newColorItem.style.backgroundColor = String(color);
-        usedColorWrapper.appendChild(newColorItem);
+        // if already have same color item. doesn't need insert
+        const sameColorItem = Array.from(usedColorWrapper.querySelectorAll(`.${colorClassName.item}[style*="background-color: ${newColorItem.style.backgroundColor}"]`));
+        if (sameColorItem.length <= 0) {
+          usedColorWrapper.appendChild(newColorItem);
+        }
+
+        const colorItem = Array.from(usedColorWrapper.querySelectorAll(`.${colorClassName.item}`)).slice(0, -1 * maxSaveColorCount);
+        for (const item of colorItem) {
+          item.remove();
+        }
       }
     }, 1000);
   }
@@ -197,7 +199,7 @@ export class TableMenuCommon {
       const selectedTds = this.getSelectedTds();
       if (item && color && selectedTds.length > 0) {
         this.tableModule.setCellAttrs(selectedTds, key!, color, true);
-        if (item.closest(`.${colorClassName.item}`)) return;
+        if (!item.closest(`.${colorClassName.item}`)) return;
         this.updateUsedColor(color);
       }
     });