summaryrefslogtreecommitdiff
path: root/mpv/scripts/sponsorblock_minimal.lua
diff options
context:
space:
mode:
authorMartial Simon <msimon_fr@hotmail.com>2023-05-17 00:09:51 +0200
committerMartial Simon <msimon_fr@hotmail.com>2023-05-17 00:09:51 +0200
commit5d3c0b24405381a41bd13a3135fa2969c2b47a48 (patch)
treec59a2521159b90daac9e139e5ebaeaf24dd3fa0f /mpv/scripts/sponsorblock_minimal.lua
parent436159f3542d4e11c81ff927ab6b22f5ad2619e0 (diff)
parentca1fba1ef518b2f55bc168e9782b500dfec67c16 (diff)
Merge branch 'main' of https://github.com/Laitram31/config
Diffstat (limited to 'mpv/scripts/sponsorblock_minimal.lua')
-rw-r--r--mpv/scripts/sponsorblock_minimal.lua145
1 files changed, 145 insertions, 0 deletions
diff --git a/mpv/scripts/sponsorblock_minimal.lua b/mpv/scripts/sponsorblock_minimal.lua
new file mode 100644
index 0000000..8ac0cca
--- /dev/null
+++ b/mpv/scripts/sponsorblock_minimal.lua
@@ -0,0 +1,145 @@
+-- sponsorblock_minimal.lua
+-- https://codeberg.org/jouni/mpv_sponsorblock_minimal
+-- LICENSE: GPLv3
+--
+-- b toggles between on/off
+--
+-- This script skips sponsored segments of YouTube videos
+-- using data from https://github.com/ajayyy/SponsorBlock
+
+local opt = require 'mp.options'
+local utils = require 'mp.utils'
+
+local options = {
+ server = "https://sponsor.ajay.app/api/skipSegments",
+
+ -- Categories to fetch and skip
+ categories = '"sponsor"',
+
+ -- Set this to "true" to use sha256HashPrefix instead of videoID
+ hash = ""
+}
+
+opt.read_options(options)
+
+function getranges(url)
+ local luacurl_available, cURL = pcall(require,'cURL')
+
+ local res = ""
+ if not(luacurl_available) then -- if Lua-cURL is not available on this system
+ local sponsors = mp.command_native{
+ name = "subprocess",
+ capture_stdout = true,
+ playback_only = false,
+ args = {"curl", "-L", "-s", "-g", url}
+ }
+ res = sponsors.stdout
+ else -- otherwise use Lua-cURL (binding to libcurl)
+ local buf={}
+ local c = cURL.easy_init()
+ c:setopt_followlocation(1)
+ c:setopt_url(url)
+ c:setopt_writefunction(function(chunk) table.insert(buf,chunk); return true; end)
+ c:perform()
+ res = table.concat(buf)
+ end
+
+ local json = utils.parse_json(res)
+ if options.hash == "true" and json ~= nil then
+ for _, i in pairs(json) do
+ if i.videoID == youtube_id then
+ return i.segments
+ end
+ end
+ else
+ return json
+ end
+
+ return nil
+end
+
+function skip_ads(name,pos)
+ if pos ~= nil then
+ for _, i in pairs(ranges) do
+ v = i.segment[2]
+ if i.segment[1] <= pos and v > pos then
+ --this message may sometimes be wrong
+ --it only seems to be a visual thing though
+ mp.osd_message(("[sponsorblock] skipping forward %ds"):format(math.floor(v-mp.get_property("time-pos"))))
+ --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",v+0.01)
+ return
+ end
+ end
+ end
+ return
+end
+
+function file_loaded()
+ local video_path = mp.get_property("path", "")
+ local video_referer = string.match(mp.get_property("http-header-fields", ""), "Referer:([^,]+)") or ""
+
+ local urls = {
+ "ytdl://youtu%.be/([%w-_]+).*",
+ "ytdl://w?w?w?%.?youtube%.com/v/([%w-_]+).*",
+ "https?://youtu%.be/([%w-_]+).*",
+ "https?://w?w?w?%.?youtube%.com/v/([%w-_]+).*",
+ "/watch.*[?&]v=([%w-_]+).*",
+ "/embed/([%w-_]+).*",
+ "^ytdl://([%w-_]+)$",
+ "-([%w-_]+)%."
+ }
+ youtube_id = nil
+ local purl = mp.get_property("metadata/by-key/PURL", "")
+ for i,url in ipairs(urls) do
+ youtube_id = youtube_id or string.match(video_path, url) or string.match(video_referer, url) or string.match(purl, url)
+ end
+
+ if not youtube_id or string.len(youtube_id) < 11 then return end
+ youtube_id = string.sub(youtube_id, 1, 11)
+
+ local url = ""
+ if options.hash == "true" then
+ local sha = mp.command_native{
+ name = "subprocess",
+ capture_stdout = true,
+ args = {"sha256sum"},
+ stdin_data = youtube_id
+ }
+ url = ("%s/%s?categories=[%s]"):format(options.server, string.sub(sha.stdout, 0, 4), options.categories)
+ else
+ url = ("%s?videoID=%s&categories=[%s]"):format(options.server, youtube_id, options.categories)
+ end
+
+ ranges = getranges(url)
+ if ranges ~= nil then
+ ON = true
+ mp.add_key_binding("b","sponsorblock",toggle)
+ mp.observe_property("time-pos", "native", skip_ads)
+ end
+ return
+end
+
+function end_file()
+ if not ON then return end
+ mp.unobserve_property(skip_ads)
+ ranges = nil
+ ON = false
+end
+
+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)
+mp.register_event("end-file", end_file) \ No newline at end of file