X-Git-Url: https://njoseph.me/gitweb/nimcoon.git/blobdiff_plain/62e4d409fe321dff3ef519f88df839c458fcfa49..f114ff9a9ec2477cf9b8fe62283b28794d49fcab:/src/peertube.nim

diff --git a/src/peertube.nim b/src/peertube.nim
index 946736f..16b6d23 100644
--- a/src/peertube.nim
+++ b/src/peertube.nim
@@ -7,7 +7,7 @@ import
 
 let PEERTUBE_REGEX = re"w\/[0-9a-zA-z]{22}"
 
-proc getPeerTubeMagnetLink*(url: string): string =
+proc getPeerTubeMagnetLink*(url: string, musicOnly: bool): string =
   ## Gets the magnet link of the best possible resolution from PeerTube
   let uuid = url.substr(find(url, PEERTUBE_REGEX) + "w/".len)
   let domainName = url.substr(8, find(url, '/', start=8) - 1)
@@ -15,4 +15,10 @@ proc getPeerTubeMagnetLink*(url: string): string =
   let client = newHttpClient()
   let response = get(client, apiURL)
   let jsonNode = parseJson($response.body)
-  jsonNode["files"][0]["magnetUri"].getStr()
+  var files = jsonNode["files"]
+  if len(jsonNode["files"]) == 0:
+    files = jsonNode["streamingPlaylists"][0]["files"]
+  if musicOnly:
+    files[len(files)-1]["magnetUri"].getStr()
+  else:
+    files[0]["magnetUri"].getStr()