1 files changed, 138 insertions, 0 deletions
diff --git a/home/.config/mpv/scripts/webtorrent-hook.lua b/home/.config/mpv/scripts/webtorrent-hook.lua
new file mode 100644
index 0000000..80d991f
--- /dev/null
+++ b/home/.config/mpv/scripts/webtorrent-hook.lua
@@ -0,0 +1,138 @@
+-- TODO prefetch if next in playlist?
+-- TODO handle torrent with multiple video files (if webtorrent can print json)
+-- - don't close kill webtorrent while still videos unplayed? or in playlist?
+-- - store titles/info when starting webtorrent and check stream-open-filename
+--   for any item in playlist to see if it matches stored entry
+
+local settings = {
+   close_webtorrent = true,
+   remove_files = true,
+   download_directory = "/tmp/webtorrent",
+   webtorrent_flags = "",
+   webtorrent_verbosity = "speed"
+}
+
+(require "mp.options").read_options(settings, "webtorrent-hook")
+
+local open_videos = {}
+
+-- http://lua-users.org/wiki/StringRecipes
+local function ends_with(str, ending)
+   return ending == "" or str:sub(-#ending) == ending
+end
+
+-- https://stackoverflow.com/questions/132397/get-back-the-output-of-os-execute-in-lua
+function os.capture(cmd, decolorize, raw)
+   if decolorize then
+      -- https://github.com/webtorrent/webtorrent-cli/issues/132
+      -- TODO webtorrent should have a way to just print json information with
+      -- no colors
+      -- https://stackoverflow.com/questions/19296667/remove-ansi-color-codes-from-a-text-file-using-bash/30938702#30938702
+      cmd = cmd .. " | sed -r 's/\\x1B\\[(([0-9]{1,2})?(;)?([0-9]{1,2})?)?[m,K,H,f,J]//g'"
+   end
+   local f = assert(io.popen(cmd, 'r'))
+   local s = assert(f:read('*a'))
+   f:close()
+   if raw then return s end
+   s = string.gsub(s, '^%s+', '')
+   s = string.gsub(s, '%s+$', '')
+   -- s = string.gsub(s, '[\n\r]+', ' ')
+   return s
+end
+
+function read_file(file)
+   local fh = assert(io.open(file, "rb"))
+   local contents = fh:read("*all")
+   fh:close()
+   return contents
+end
+
+function play_torrent()
+   local url = mp.get_property("stream-open-filename")
+   if (url:find("magnet:") == 1 or url:find("peerflix://") == 1
+       or url:find("webtorrent://") == 1 or ends_with(url, "torrent")) then
+      if url:find("webtorrent://") == 1 then
+         url = url:sub(14)
+      end
+      if url:find("peerflix://") == 1 then
+         url = url:sub(12)
+      end
+
+      os.execute("mkdir -p " .. settings.download_directory)
+      -- don't reuse files (so multiple mpvs works)
+      local output_file = settings.download_directory
+         .. "/webtorrent-output-" .. mp.get_time() .. ".log"
+      -- --keep-seeding is to prevent webtorrent from quitting once the download
+      -- is done
+      local webtorrent_command = "webtorrent "
+         .. settings.webtorrent_flags
+         .. " --out '" .. settings.download_directory .. "' --keep-seeding \""
+         .. url .. "\" > " .. output_file .. " 2>&1 & echo $!"
+      mp.msg.info("Starting webtorrent server")
+      mp.msg.info(webtorrent_command)
+      local pid = os.capture(webtorrent_command)
+      mp.msg.info("Waiting for webtorrent server")
+
+      local url_command = "tail -f " .. output_file
+         .. " | awk '/Server running at:/ {print $4; exit}'"
+      local url = os.capture(url_command, true)
+      mp.msg.info("Webtorrent server is up")
+
+      local title_command = "awk '/(Seeding|Downloading): / "
+         .. "{gsub(/(Seeding|Downloading): /, \"\"); print; exit}' "
+         .. output_file
+      local title = os.capture(title_command, true)
+      mp.msg.info("Setting media title to: " .. title)
+      mp.set_property("force-media-title", title)
+
+      local path
+      if title then
+         path = settings.download_directory .. "/\"" .. title .. "\""
+      end
+      open_videos[url] = {title=title,path=path,pid=pid}
+
+      mp.set_property("stream-open-filename", url)
+
+      if settings.webtorrent_verbosity == "speed" then
+         local printer_pid
+         local printer_pid_file = settings.download_directory
+            .. "/webtorrent-printer-" .. mp.get_time() .. ".pid"
+         os.execute("tail -f " .. output_file
+                       .. " | awk '/Speed:/' ORS='\r' & echo -n $! > "
+                       .. printer_pid_file)
+         printer_pid = read_file(printer_pid_file)
+         mp.register_event("file-loaded",
+                           function()
+                              os.execute("kill " .. printer_pid)
+                           end
+         )
+      end
+   end
+end
+
+function webtorrent_cleanup()
+   local url = mp.get_property("stream-open-filename")
+   if settings.close_webtorrent and open_videos[url] then
+      local title = open_videos[url].title
+      local path = open_videos[url].path
+      local pid = open_videos[url].pid
+
+      if pid then
+         mp.msg.info("Closing webtorrent for " .. title .. " with pid " .. pid)
+         os.execute("kill " .. pid)
+      end
+
+      if settings.remove_files then
+         if path then
+            mp.msg.info("Removing media file for " .. title .. " with path " .. path)
+            os.execute("rm -r " .. path)
+         end
+      end
+
+      open_videos[url] = {}
+   end
+end
+
+mp.add_hook("on_load", 50, play_torrent)
+
+mp.add_hook("on_unload", 10, webtorrent_cleanup)