dotfiles/home/.config/mpv/scripts/yledl_hook.lua

77 lines
2.7 KiB
Lua
Raw Normal View History

-- Copied from https://github.com/pekkarr/mpv-yledl
-- Copyright 2021 Pekka Ristola
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program. If not, see <https://www.gnu.org/licenses/>.
local msg = require 'mp.msg'
local utils = require 'mp.utils'
local function add_series(data)
local playlist = { "#EXTM3U" }
for _, episode in pairs(data) do
table.insert(playlist, episode["webpage"])
end
mp.set_property("stream-open-filename", "memory://" .. table.concat(playlist, "\n"))
end
local function add_single_video(data)
local flavors = data["flavors"]
table.sort(flavors, function(a, b) return a["bitrate"] > b["bitrate"] end)
local best = flavors[1]
mp.set_property("stream-open-filename", best["url"])
mp.set_property("file-local-options/force-media-title", data["title"])
for _, sub in ipairs(data["subtitles"]) do
local lang = sub["language"]
msg.verbose("Adding subtitles for " .. lang)
mp.commandv("sub-add", sub["url"], "auto", sub["category"], lang)
end
end
mp.add_hook("on_load", 9, function()
msg.verbose('yle-dl hook')
local url = mp.get_property("stream-open-filename", "")
if (url:find("https?://%a+%.yle%.fi/") == 1) or (url:find("https?://yle%.fi/") == 1) then
local start_time = os.clock()
local command = { "yle-dl", "--showmetadata", url }
msg.debug("Running: " .. table.concat(command, ' '))
local ret = mp.command_native({name = "subprocess",
args = command,
capture_stdout = true,
capture_stderr = false})
if ret.killed_by_us then
return
end
if (ret.status < 0) or (ret.stdout == nil) or (ret.stdout == "") then
msg.error("yle-dl failed to parse url")
return
end
local json, err = utils.parse_json(ret.stdout)
if (json == nil) then
msg.error("failed to parse JSON: " .. err)
return
end
msg.verbose("yle-dl succeeded")
msg.debug("running yle-dl took " .. os.clock() - start_time .. " seconds")
if #json == 1 then
add_single_video(json[1])
else
add_series(json)
end
else
msg.verbose('not an areena url')
end
end)