94 lines
3.1 KiB
Lua
94 lines
3.1 KiB
Lua
-- sponsorblock_minimal.lua
|
|
--
|
|
-- This script skips sponsored segments of YouTube videos
|
|
-- using data from https://github.com/ajayyy/SponsorBlock
|
|
--
|
|
-- original from https://codeberg.org/jouni/mpv_sponsorblock_minimal
|
|
-- adapted for local playback skipping and some refactoring by me
|
|
local mp = require 'mp'
|
|
|
|
local options = {
|
|
API = "https://sponsor.ajay.app/api/skipSegments",
|
|
|
|
-- Categories to fetch and skip
|
|
categories = '"sponsor","intro","outro","interaction","selfpromo"'
|
|
}
|
|
|
|
local function getranges()
|
|
local args = {
|
|
"curl", "-s", "-d", "videoID=" .. Youtube_id, "-d",
|
|
"categories=[" .. options.categories .. "]", "-G", options.API
|
|
}
|
|
local sponsors = mp.command_native({
|
|
name = "subprocess",
|
|
capture_stdout = true,
|
|
playback_only = false,
|
|
args = args
|
|
})
|
|
|
|
if string.match(sponsors.stdout, "%[(.-)%]") then
|
|
Ranges = {}
|
|
for i in string.gmatch(string.sub(sponsors.stdout, 2, -2), "%[(.-)%]") do
|
|
local k, v = string.match(i, "(%d+.?%d*),(%d+.?%d*)")
|
|
Ranges[k] = v
|
|
end
|
|
end
|
|
end
|
|
|
|
local function skip_ads(_, pos)
|
|
if pos ~= nil then
|
|
for k, v in pairs(Ranges) do
|
|
if tonumber(k) <= pos and tonumber(v) > pos then
|
|
-- this message may sometimes be wrong
|
|
-- it only seems to be a visual thing though
|
|
mp.osd_message("[sponsorblock] skipping forward " ..
|
|
math.floor(
|
|
tonumber(v) - mp.get_property("time-pos")) ..
|
|
"s")
|
|
-- need to do the +0.01 otherwise mpv will start spamming skip sometimes
|
|
-- example: https://www.youtube.com/watch?v=4ypMJzeNooo
|
|
mp.set_property("time-pos", tonumber(v) + 0.01)
|
|
return
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function file_loaded()
|
|
local video_path = mp.get_property("path")
|
|
local youtube_id1 = string.match(video_path,
|
|
"https?://youtu%.be/([%w-_]+).*")
|
|
local youtube_id2 = string.match(video_path,
|
|
"https?://w?w?w?%.?youtube%.com/v/([%w-_]+).*")
|
|
local youtube_id3 = string.match(video_path, "/watch.*[?&]v=([%w-_]+).*")
|
|
local youtube_id4 = string.match(video_path, "/embed/([%w-_]+).*")
|
|
local localytfile = string.match(video_path,
|
|
"-([%a%d%-_]+)%.[mw][kpe][v4b][m]?$")
|
|
Youtube_id = youtube_id1 or youtube_id2 or youtube_id3 or youtube_id4 or
|
|
localytfile
|
|
if not Youtube_id or string.len(Youtube_id) < 11 then return end
|
|
Youtube_id = string.sub(Youtube_id, 1, 11)
|
|
|
|
getranges()
|
|
if Ranges then
|
|
ON = true
|
|
mp.add_key_binding("b", "sponsorblock", toggle)
|
|
mp.observe_property("time-pos", "native", skip_ads)
|
|
end
|
|
return
|
|
end
|
|
|
|
local function toggle()
|
|
if ON then
|
|
mp.unobserve_property(skip_ads)
|
|
mp.osd_message("[sponsorblock] off")
|
|
ON = false
|
|
return
|
|
end
|
|
mp.observe_property("time-pos", "native", skip_ads)
|
|
mp.osd_message("[sponsorblock] on")
|
|
ON = true
|
|
return
|
|
end
|
|
|
|
mp.register_event("file-loaded", file_loaded)
|