From 9621175dc91d8f410dbc14d09bc0132e6a33ae6d Mon Sep 17 00:00:00 2001
From: Samantaz Fox <coding@samantaz.fr>
Date: Thu, 3 Feb 2022 00:57:44 +0100
Subject: [PATCH] extractors: Add helper for short view count text

---
 src/invidious/yt_backend/extractors.cr | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/invidious/yt_backend/extractors.cr b/src/invidious/yt_backend/extractors.cr
index 27ce550b..41d95962 100644
--- a/src/invidious/yt_backend/extractors.cr
+++ b/src/invidious/yt_backend/extractors.cr
@@ -519,6 +519,20 @@ module HelperExtractors
     end
   end
 
+  # Retrieves the amount of views/viewers a video has.
+  # Seems to be used on related videos only
+  #
+  # Returns "0" when unable to parse
+  def self.get_short_view_count(container : JSON::Any) : String
+    box = container["shortViewCountText"]?
+    return "0" if !box
+
+    # Simpletext: "4M views"
+    # runs: {"text": "1.1K"},{"text":" watching"}
+    return box["simpleText"]?.try &.as_s.sub(" views", "") ||
+      box.dig?("runs", 0, "text").try &.as_s || "0"
+  end
+
   # Retrieve lowest quality thumbnail from InnerTube data
   #
   # TODO allow configuration of image quality (-1 is highest)