refactor: flash comment

xpadev-net · Nov 19, 2023 · 26da5f0 · 26da5f0
1 parent 8214920
commit 26da5f0
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/comments/FlashComment.ts b/src/comments/FlashComment.ts
@@ -368,12 +368,16 @@ class FlashComment extends BaseComment {
     const atButtonRadius = getConfig(config.atButtonRadius, true);
     const isHover = this.isHovered(cursor, posX, posY);
     context.save();
-    context.strokeStyle =
-      this.comment.button.limit < 1
-        ? "#777777"
-        : isHover
-        ? this.comment.color
-        : "white";
+    const getStrokeStyle = () => {
+      if (isHover) {
+        return this.comment.color;
+      }
+      if (this.comment.button && this.comment.button.limit < 1) {
+        return "#777777";
+      }
+      return "white";
+    };
+    context.strokeStyle = getStrokeStyle();
     drawLeftBorder(
       context,
       parts.left.left,