From 8fb97a451153cc5cf1a20e6b76e25d6ee2a01c28 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Mon, 12 Mar 2018 20:36:49 -0500
Subject: [PATCH] Add suppor for dash manifest

---
 src/invidious.cr | 79 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git a/src/invidious.cr b/src/invidious.cr
index 4b820a790..ded6a8755 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -327,6 +327,85 @@ get "/redirect" do |env|
   end
 end
 
+# Return dash manifest for the given video ID
+get "/api/manifest/dash/id/:id" do |env|
+  env.response.headers.add("Access-Control-Allow-Origin", "*")
+  env.response.content_type = "application/dash+xml"
+
+  id = env.params.url["id"]
+
+  yt_client = get_client(youtube_pool)
+  begin
+    video = get_video(id, yt_client, PG_DB)
+  rescue ex
+    halt env, status_code: 403
+  ensure
+    youtube_pool << yt_client
+  end
+
+  adaptive_fmts = [] of HTTP::Params
+  if video.info.has_key?("adaptive_fmts")
+    video.info["adaptive_fmts"].split(",") do |string|
+      adaptive_fmts << HTTP::Params.parse(string)
+    end
+  else
+    halt env, status_code: 403
+  end
+
+  manifest = XML.build(indent: "  ") do |xml|
+    xml.element("MPD", xmlns: "urn:mpeg:dash:schema:mpd:2011", profiles: "urn:mpeg:dash:profile:full:2011",
+      mediaPresentationDuration: "PT#{video.info["length_seconds"]}S", minBufferTime: "PT2S", type: "static") do
+      host = URI.parse(adaptive_fmts[0]["url"]).host.to_s
+
+      xml.element("BaseURL") { xml.text "https://#{host}" }
+      xml.element("Period", id: "0") do
+        adaptive_fmts.each do |fmt|
+          mimetype, codecs = fmt["type"].split(";")
+          codecs = codecs[9..-2]
+          fmt_type = mimetype.split("/")[0]
+          fmt_id = fmt["itag"]
+          bandwidth = fmt["bitrate"]
+
+          url = fmt["url"]
+          if fmt["s"]?
+            url += "&signature="
+            url += decrypt_signature(fmt["s"])
+          end
+
+          url = URI.parse(url)
+          url = url.path.not_nil! + "?" + url.query.not_nil!
+
+          if fmt_type == "video"
+            width, height = fmt["size"].split("x")
+            framerate = fmt["fps"]
+
+            xml.element("AdaptationSet", contentType: "video", width: width,
+              height: height, frameRate: framerate, subsegmentAlignment: true, id: fmt_id) do
+              xml.element("Representation", id: fmt_id, bandwidth: bandwidth, codecs: codecs, mimeType: mimetype) do
+                xml.element("BaseURL") { xml.text url }
+                xml.element("SegmentBase", indexRange: fmt["init"]) do
+                  xml.element("Initialization", range: fmt["index"])
+                end
+              end
+            end
+          else
+            xml.element("AdaptationSet", contentType: "audio", subsegmentAlignment: true, id: fmt_id) do
+              xml.element("Representation", id: fmt_id, bandwidth: bandwidth, codecs: codecs, mimeType: mimetype) do
+                xml.element("BaseURL") { xml.text url }
+                xml.element("SegmentBase", indexRange: fmt["init"]) do
+                  xml.element("Initialization", range: fmt["index"])
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+
+  manifest
+end
+
 error 404 do |env|
   error_message = "404 Page not found"
   templated "error"