From 9a02b5278a42b8785ed761d95f2d98a21df8ae71 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Tue, 7 Aug 2018 07:36:51 -0500
Subject: [PATCH] Fix duration for auto-generated captions

---
 src/invidious.cr | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/invidious.cr b/src/invidious.cr
index 9ad2e9b2..a9601256 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -1646,16 +1646,24 @@ get "/api/v1/captions/:id" do |env|
 
   END_VTT
 
-  caption_xml.xpath_nodes("//transcript/text").each do |node|
+  caption_nodes = caption_xml.xpath_nodes("//transcript/text")
+  caption_nodes.each_with_index do |node, i|
     start_time = node["start"].to_f.seconds
     duration = node["dur"]?.try &.to_f.seconds
     duration ||= start_time
-    end_time = start_time + duration
+
+    if caption_nodes.size > i + 1
+      end_time = caption_nodes[i + 1]["start"].to_f.seconds
+    else
+      end_time = start_time + duration
+    end
 
     start_time = "#{start_time.hours.to_s.rjust(2, '0')}:#{start_time.minutes.to_s.rjust(2, '0')}:#{start_time.seconds.to_s.rjust(2, '0')}.#{start_time.milliseconds.to_s.rjust(3, '0')}"
     end_time = "#{end_time.hours.to_s.rjust(2, '0')}:#{end_time.minutes.to_s.rjust(2, '0')}:#{end_time.seconds.to_s.rjust(2, '0')}.#{end_time.milliseconds.to_s.rjust(3, '0')}"
 
     text = HTML.unescape(node.content)
+    text = text.gsub(/<font color="#[a-fA-F0-9]{6}">/, "")
+    text = text.gsub(/<\/font>/, "")
     if md = text.match(/(?<name>.*) : (?<text>.*)/)
       text = "<v #{md["name"]}>#{md["text"]}</v>"
     end