From 67de4192f01c53f7261a1c84080bd2896dd31560 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Sun, 22 Jul 2018 14:09:53 -0500
Subject: [PATCH] Provide captions if no label is provided

---
 src/invidious.cr | 39 ++++++++++++++++++++++++++++-----------
 1 file changed, 28 insertions(+), 11 deletions(-)

diff --git a/src/invidious.cr b/src/invidious.cr
index f9f39d59..6a06635f 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -414,26 +414,45 @@ get "/captions/:id" do |env|
     halt env, status_code: 403
   end
 
-  env.response.content_type = "application/json"
-
   player_response = JSON.parse(video.info["player_response"])
   if !player_response["captions"]?
-    next "{}"
+    env.response.content_type = "application/json"
+    next {
+      "captions" => [] of String,
+    }.to_json
   end
 
-  tracks = player_response["captions"]["playerCaptionsTracklistRenderer"]["captionTracks"].as_a
+  tracks = player_response["captions"]["playerCaptionsTracklistRenderer"]["captionTracks"]?.try &.as_a
+  tracks ||= [] of JSON::Any
 
   label = env.params.query["label"]?
+  if !label
+    env.response.content_type = "application/json"
+
+    response = JSON.build do |json|
+      json.object do
+        json.field "captions" do
+          json.array do
+            tracks.each do |caption|
+              json.object do
+                json.field "label", caption["name"]["simpleText"]
+                json.field "languageCode", caption["languageCode"]
+              end
+            end
+          end
+        end
+      end
+    end
+
+    next response
+  end
+
   track = tracks.select { |tracks| tracks["name"]["simpleText"] == label }
 
   env.response.content_type = "text/vtt"
   if track.empty?
-    if tracks.empty?
       halt env, status_code: 403
     else
-      track = tracks[0]
-    end
-  else
     track = track[0]
   end
 
@@ -597,9 +616,7 @@ get "/comments/:id" do |env|
     end
 
     env.response.content_type = "application/json"
-    comments = JSON.parse(comments)
-
-    next comments.to_pretty_json
+    next comments
   elsif source == "reddit"
     client = make_client(REDDIT_URL)
     headers = HTTP::Headers{"User-Agent" => "web:invidio.us:v0.1.0 (by /u/omarroth)"}