aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/home/.config/mpv/scripts
diff options
context:
space:
mode:
authorLibravatar Arnold Sommerfeld <sommerfeld@strisemarx.com>2023-05-17 18:44:48 +0100
committerLibravatar Arnold Sommerfeld <sommerfeld@strisemarx.com>2023-10-18 11:16:43 +0100
commitb487984ecc61c6229cf92550030745c192fd3d0b (patch)
tree937e598c941fc1172467aeeed8ef51cbfceaca62 /home/.config/mpv/scripts
downloaddotfiles-b487984ecc61c6229cf92550030745c192fd3d0b.tar.gz
dotfiles-b487984ecc61c6229cf92550030745c192fd3d0b.tar.bz2
dotfiles-b487984ecc61c6229cf92550030745c192fd3d0b.zip
first commit
Diffstat (limited to 'home/.config/mpv/scripts')
-rw-r--r--home/.config/mpv/scripts/webtorrent-hook.lua138
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)