From 9a02b5278a42b8785ed761d95f2d98a21df8ae71 Mon Sep 17 00:00:00 2001 From: Omar Roth 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 9ad2e9b2a..a9601256c 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(//, "") + text = text.gsub(/<\/font>/, "") if md = text.match(/(?.*) : (?.*)/) text = "#{md["text"]}" end