1 files changed, 141 insertions, 0 deletions
diff --git a/mac/.config/mpv/scripts/sponsorblock_minimal.lua b/mac/.config/mpv/scripts/sponsorblock_minimal.lua
new file mode 100644
index 0000000..c375480
--- /dev/null
+++ b/mac/.config/mpv/scripts/sponsorblock_minimal.lua
@@ -0,0 +1,141 @@
+-- sponsorblock_minimal.lua
+--
+-- 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 ON = false
+local ranges = nil
+
+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 skip_ads(name, pos)
+	if pos 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
+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-_]+)%.",
+	}
+	local 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)
+		if youtube_id then
+			break
+		end
+	end
+
+	if not youtube_id or string.len(youtube_id) < 11 then
+		return
+	end
+	youtube_id = string.sub(youtube_id, 1, 11)
+
+	local args = { "curl", "-L", "-s", "-G", "--data-urlencode", ("categories=[%s]"):format(options.categories) }
+	local url = options.server
+	if options.hash == "true" then
+		local sha = mp.command_native({
+			name = "subprocess",
+			capture_stdout = true,
+			args = { "sha256sum" },
+			stdin_data = youtube_id,
+		})
+		url = ("%s/%s"):format(url, string.sub(sha.stdout, 0, 4))
+	else
+		table.insert(args, "--data-urlencode")
+		table.insert(args, "videoID=" .. youtube_id)
+	end
+	table.insert(args, url)
+
+	local sponsors = mp.command_native({
+		name = "subprocess",
+		capture_stdout = true,
+		playback_only = false,
+		args = args,
+	})
+	if sponsors.stdout then
+		local json = utils.parse_json(sponsors.stdout)
+		if type(json) == "table" then
+			if options.hash == "true" then
+				for _, i in pairs(json) do
+					if i.videoID == youtube_id then
+						ranges = i.segments
+						break
+					end
+				end
+			else
+				ranges = json
+			end
+
+			if ranges then
+				ON = true
+				mp.add_key_binding("", "sponsorblock", toggle)
+				mp.observe_property("time-pos", "native", skip_ads)
+			end
+		end
+	end
+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
+	else
+		mp.observe_property("time-pos", "native", skip_ads)
+		mp.osd_message("[sponsorblock] on")
+		ON = true
+	end
+end
+
+mp.register_event("file-loaded", file_loaded)
+mp.register_event("end-file", end_file)