feat: improve getRandomColor function

PancakePhilarmonych · Nov 21, 2023 · 942d53d · 942d53d
1 parent 5953636
commit 942d53d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/utils/index.ts b/src/utils/index.ts
@@ -32,9 +32,10 @@ const mainColors = [
   Colors.EMPTY,
 ];
 
-export const getRandomColor = () => {
-  const index = Math.floor(Math.random() * mainColors.length);
-  return mainColors[index];
+export const getRandomColor = (excludeEmpty = false) => {
+  const colors = excludeEmpty ? mainColors.filter(color => color !== Colors.EMPTY) : mainColors;
+  const index = Math.floor(Math.random() * colors.length);
+  return colors[index];
 }