fix: assign MetadataBadge's label (#311)

LuanRT · Feb 13, 2023 · e37cf62 · e37cf62
1 parent 567fdba
commit e37cf62
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/src/parser/classes/CompactVideo.ts b/src/parser/classes/CompactVideo.ts
@@ -58,18 +58,18 @@ class CompactVideo extends YTNode {
   }
 
   get is_fundraiser(): boolean {
-    return this.badges.some((badge) => badge.style === 'Fundraiser');
+    return this.badges.some((badge) => badge.label === 'Fundraiser');
   }
 
   get is_live(): boolean {
     return this.badges.some((badge) => {
-      if (badge.label === 'BADGE_STYLE_TYPE_LIVE_NOW' || badge.style === 'LIVE')
+      if (badge.style === 'BADGE_STYLE_TYPE_LIVE_NOW' || badge.label === 'LIVE')
         return true;
     });
   }
 
   get is_new(): boolean {
-    return this.badges.some((badge) => badge.style === 'New');
+    return this.badges.some((badge) => badge.label === 'New');
   }
 
   get is_premiere(): boolean {

diff --git a/src/parser/classes/MetadataBadge.ts b/src/parser/classes/MetadataBadge.ts
@@ -20,7 +20,7 @@ class MetadataBadge extends YTNode {
     }
 
     if (data?.label) {
-      this.style = data.label;
+      this.label = data.label;
     }
 
     if (data?.tooltip || data?.iconTooltip) {

diff --git a/src/parser/classes/Video.ts b/src/parser/classes/Video.ts
@@ -96,7 +96,7 @@ class Video extends YTNode {
 
   get is_live(): boolean {
     return this.badges.some((badge) => {
-      if (badge.label === 'BADGE_STYLE_TYPE_LIVE_NOW' || badge.style === 'LIVE')
+      if (badge.style === 'BADGE_STYLE_TYPE_LIVE_NOW' || badge.label === 'LIVE')
         return true;
     });
   }
@@ -106,15 +106,15 @@ class Video extends YTNode {
   }
 
   get is_premiere(): boolean {
-    return this.badges.some((badge) => badge.style === 'PREMIERE');
+    return this.badges.some((badge) => badge.label === 'PREMIERE');
   }
 
   get is_4k(): boolean {
-    return this.badges.some((badge) => badge.style === '4K');
+    return this.badges.some((badge) => badge.label === '4K');
   }
 
   get has_captions(): boolean {
-    return this.badges.some((badge) => badge.style === 'CC');
+    return this.badges.some((badge) => badge.label === 'CC');
   }
 
   get best_thumbnail(): Thumbnail | undefined {