From baa7e72ad6a4478f381f3255c68a07a4d94f1c8e Mon Sep 17 00:00:00 2001
From: yusing <yusing.wys@gmail.com>
Date: Mon, 15 Dec 2025 14:42:31 +0800
Subject: [PATCH] refactor(icon): improve handling in WithVariant

---
 internal/homepage/icon_url.go | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/internal/homepage/icon_url.go b/internal/homepage/icon_url.go
index 728b37f2..c820734c 100644
--- a/internal/homepage/icon_url.go
+++ b/internal/homepage/icon_url.go
@@ -96,13 +96,11 @@ func (u *IconURL) WithVariant(variant IconVariant) *IconURL {
 			Key:      u.Extra.Key,
 			Ref:      u.Extra.Ref,
 			FileType: u.Extra.FileType,
+			IsLight:  variant == IconVariantLight,
+			IsDark:   variant == IconVariantDark,
 		}
-		switch variant {
-		case IconVariantLight:
-			extra.IsLight = true
-		case IconVariantDark:
-			extra.IsDark = true
-		}
+		extra.Ref = strings.TrimSuffix(extra.Ref, "-light")
+		extra.Ref = strings.TrimSuffix(extra.Ref, "-dark")
 	}
 	return &IconURL{
 		IconSource: u.IconSource,