More optimal way for get cache...

This commit is contained in:
localhost_frssoft 2023-03-04 03:25:09 +03:00
parent 5f5b35b887
commit 145dcf904b
3 changed files with 913 additions and 4 deletions

View File

@ -12,7 +12,8 @@ import re
fzf = FzfPrompt() fzf = FzfPrompt()
if get_config('enable_persistent_cache'): if get_config('enable_persistent_cache'):
player = mpv.MPV(cache=True, demuxer_max_bytes=25*1024*1024, scripts='src/mpv_scripts/mpv_cache.lua') player = mpv.MPV(cache=True, scripts='src/mpv_scripts/mpv_cache.lua:src/mpv_scripts/streamsave.lua')
player.command('script-message', 'streamsave-path', 'cache')
else: else:
player = mpv.MPV(cache=True, demuxer_max_bytes=25*1024*1024) player = mpv.MPV(cache=True, demuxer_max_bytes=25*1024*1024)
player.ytdl = False # Prevent attempts load track with yt-dlp player.ytdl = False # Prevent attempts load track with yt-dlp

View File

@ -28,12 +28,15 @@ function make_cache_track(url)
find_uuid = "%x+-%x+-%x+-%x+-%x+" find_uuid = "%x+-%x+-%x+-%x+-%x+"
uuid = string.sub(url, string.find(url, find_uuid)) uuid = string.sub(url, string.find(url, find_uuid))
host = get_url_host(url) host = get_url_host(url)
cache_path_file = 'cache/' .. host .. '/' .. uuid .. '' cache_path_file = 'cache/' .. host .. '/' .. uuid .. '.webm'
if false == file_exists(cache_path_file) then if false == file_exists(cache_path_file) then
createDir('cache/' .. host .. '/') createDir('cache/' .. host .. '/')
msg.verbose('Caching ' .. cache_path_file .. '') msg.verbose('Caching ' .. cache_path_file .. '')
os.execute('curl -s --retry 3 "' .. url .. '" -o "' .. cache_path_file .. '"') mp.command('script-message streamsave-title ' .. uuid .. '')
mp.set_property("stream-open-filename", cache_path_file) mp.set_property('script-opts/media-uuid', uuid)
mp.command('script-message streamsave-extension .webm')
mp.command('script-message streamsave-path cache/' .. host .. '')
mp.command('script-message streamsave-rec')
else else
msg.verbose('Already cached ' .. cache_path_file .. '') msg.verbose('Already cached ' .. cache_path_file .. '')
os.execute('touch ' .. cache_path_file .. '') os.execute('touch ' .. cache_path_file .. '')

View File

@ -0,0 +1,905 @@
--[[
streamsave.lua
Version 0.20.6
2022-10-13
https://github.com/Sagnac/streamsave
NOTE: Modified
mpv script aimed at saving live streams and clipping online videos without encoding.
Essentially a wrapper around mpv's cache dumping commands, the script adds the following functionality:
* Automatic determination of the output file name and format
* Option to specify the preferred output directory
* Switch between 3 different dump modes (clip mode, full/continuous dump, write from beginning to current position)
* Prevention of file overwrites
* Acceptance of inverted loop ranges, allowing the end point to be set first
* Dynamic chapter indicators on the OSC displaying the clipping interval
* Automated stream saving
* Workaround for some DAI HLS streams served from .m3u8 where the host changes
By default the A-B loop points (set using the `l` key in mpv) determine the portion of the cache written to disk.
It is advisable that you set --demuxer-max-bytes and --demuxer-max-back-bytes to larger values
(e.g. at least 1GiB) in order to have a larger cache.
If you want to use with local files set cache=yes in mpv.conf
Options are specified in ~~/script-opts/streamsave.conf
Runtime changes to all user options are supported via the `script-opts` property by using mpv's `set` or
`change-list` input commands and the `streamsave-` prefix.
General Options:
save_directory sets the output file directory. Don't use quote marks or a trailing slash when specifying paths here.
Example: save_directory=C:\User Directory
mpv double tilde paths ~~/ and home path shortcuts ~/ are also accepted.
By default files are dumped in the current directory.
dump_mode=continuous will use dump-cache, setting the initial timestamp to 0 and leaving the end timestamp unset.
Use this mode if you want to dump the entire cache.
This process will continue as packets are read and until the streams change, the player is closed,
or the user presses the stop keybind.
Under this mode pressing the cache-write keybind again will stop writing the first file and
initiate another file starting at 0 and continuing as the cache increases.
If you want continuous dumping with a different starting point use the default A-B mode instead
and only set the first loop point then press the cache-write keybind.
dump_mode=current will dump the cache from timestamp 0 to the current playback position in the file.
The output_label option allows you to choose how the output filename is tagged.
The default uses iterated step increments for every file output; i.e. file-1.mkv, file-2.mkv, etc.
There are 3 other choices:
output_label=timestamp will append Unix timestamps to the file name.
output_label=range will tag the file with the A-B loop range instead using the format HH.MM.SS
e.g. file-[00.15.00 - 00.20.00].mkv
output_label=overwrite will not tag the file and will overwrite any existing files with the same name.
The force_extension option allows you to force a preferred format and sidestep the automatic detection.
If using this option it is recommended that a highly flexible container is used (e.g. Matroska).
The format is specified as the extension including the dot (e.g. force_extension=.mkv).
If this option is set, `script-message streamsave-extension revert` will run the automatic determination at runtime;
running this command again will reset the extension to what's specified in force_extension.
The force_title option will set the title used for the filename. By default the script uses the media-title.
This is specified without double quote marks in streamsave.conf, e.g. force_title=Example Title
The output_label is still used here and file overwrites are prevented if desired.
Changing the filename title to the media-title is still possible at runtime by using the revert argument,
as in the force_extension example.
The range_marks option allows the script to set temporary chapters at A-B loop points.
If chapters already exist they are stored and cleared whenever any A-B points are set.
Once the A-B points are cleared the original chapters are restored.
Any chapters added after A-B mode is entered are added to the initial chapter list.
This option is disabled by default; set range_marks=yes in streamsave.conf in order to enable it.
Automation Options:
The autostart and autoend options are used for automated stream capturing.
Set autostart=yes if you want the script to trigger cache writing immediately on stream load.
Set autoend to a time format of the form HH:MM:SS (e.g. autoend=01:20:08) if you want the file writing
to stop at that time.
The hostchange option enables an experimental workaround for DAI HLS .m3u8 streams in which the host changes.
If enabled this will result in multiple files being output as the stream reloads.
The autostart option must also be enabled in order to autosave these types of streams.
The `quit=HH:MM:SS` option will set a one shot timer from script load to the specified time,
at which point the player will exit. This serves as a replacement for autoend when using hostchange.
Running `script-message streamsave-quit HH:MM:SS` at runtime will reset and restart the timer.
Set piecewise=yes if you want to save a stream in parts automatically, useful for
e.g. saving long streams on slow systems. Set autoend to the duration preferred for each output file.
This feature requires autostart=yes.
mpv's script-message command can be used at runtime to set the dump mode, override the output title
or file extension, change the save directory, or switch the output label.
If you override the title, the file extension, or the directory, the revert argument can be used
to set it back to the default value.
Examples:
script-message streamsave-mode continuous
script-message streamsave-title "Example Title"
script-message streamsave-extension .mkv
script-message streamsave-extension revert
script-message streamsave-path ~/streams
script-message streamsave-label range
]]
local options = require 'mp.options'
local utils = require 'mp.utils'
local msg = require 'mp.msg'
local unpack = unpack or table.unpack
-- default user options
-- change these in streamsave.conf
local opts = {
save_directory = [[.]], -- output file directory
dump_mode = "continuous", -- <ab|current|continuous>
output_label = "overwrite", -- <increment|range|timestamp|overwrite>
force_extension = ".webm", -- <no|.ext> extension will be .ext if set
force_title = "no", -- <no|title> custom title used for the filename
range_marks = false, -- <yes|no> set chapters at A-B loop points?
autostart = true, -- <yes|no> automatically dump cache at start?
autoend = "no", -- <no|HH:MM:SS> cache time to stop at
hostchange = false, -- <yes|no> use if the host changes mid stream
quit = "no", -- <no|HH:MM:SS> quits player at specified time
piecewise = false, -- <yes|no> writes stream in parts with autoend
}
-- for internal use
local file = {
name, -- file name (full path to file)
path, -- directory the file is written to
title, -- media title
inc, -- filename increments
ext, -- file extension
pending, -- number of files pending write completion (max 2)
queue, -- cache_write queue in case of multiple write requests
oldtitle, -- initialized if title is overridden, allows revert
oldext, -- initialized if format is overridden, allows revert
oldpath, -- initialized if directory is overriden, allows revert
}
local loop = {
a, -- A loop point as number type
b, -- B loop point as number type
a_revert, -- A loop point prior to keyframe alignment
b_revert, -- B loop point prior to keyframe alignment
range, -- A-B loop range
aligned, -- are the loop points aligned to keyframes?
}
local cache = {
dumped, -- autowrite cache state (serves as an autowrite request)
observed, -- whether the cache time is being observed
endsec, -- user specified autoend cache time in seconds
prior, -- previous cache time
seekend, -- seekable cache end timestamp
part, -- approx. end time of last piece / start time of next piece
switch, -- request to observe track switches and seeking
use, -- use cache_time instead of seekend for initial piece
restart, -- hostchange interval where subsequent reloads are immediate
}
local convert_time
local observe_cache
local continuous
local write_file
local reset
local title_change
local container
local chapter_list = {} -- initial chapter list
local ab_chapters = {} -- A-B loop point chapters
local chapter_points
local get_seekable_cache
local reload
local automatic
local quitseconds
local quit_timer
local autoquit
function convert_time(value)
local i, j, H, M, S = value:find("(%d+):(%d+):(%d+)")
if not i then
return
else
return H*3600 + M*60 + S
end
end
local function validate_opts()
if opts.output_label ~= "increment" and
opts.output_label ~= "range" and
opts.output_label ~= "timestamp" and
opts.output_label ~= "overwrite"
then
msg.warn("Invalid output_label '" .. opts.output_label .. "'")
opts.output_label = "increment"
end
if opts.dump_mode ~= "ab" and
opts.dump_mode ~= "current" and
opts.dump_mode ~= "continuous"
then
msg.warn("Invalid dump_mode '" .. opts.dump_mode .. "'")
opts.dump_mode = "ab"
end
if opts.autoend ~= "no" then
if not cache.part then
cache.endsec = convert_time(opts.autoend)
end
if not convert_time(opts.autoend) then
msg.warn("Invalid autoend value '" .. opts.autoend ..
"'. Use HH:MM:SS format.")
opts.autoend = "no"
end
end
if opts.quit ~= "no" then
quitseconds = convert_time(opts.quit)
if not quitseconds then
msg.warn("Invalid quit value '" .. opts.quit ..
"'. Use HH:MM:SS format.")
opts.quit = "no"
end
end
end
local function update_opts(changed)
validate_opts()
-- expand mpv meta paths (e.g. ~~/directory)
file.path = mp.command_native({"expand-path", opts.save_directory})
if opts.force_title ~= "no" then
file.title = opts.force_title
elseif changed["force_title"] then
title_change(_, mp.get_property("media-title"), true)
end
if opts.force_extension ~= "no" then
file.ext = opts.force_extension
elseif changed["force_extension"] then
container(_, _, true)
end
if changed["range_marks"] then
if opts.range_marks then
chapter_points()
else
ab_chapters = {}
mp.set_property_native("chapter-list", chapter_list)
end
end
if changed["autoend"] then
cache.endsec = convert_time(opts.autoend)
observe_cache()
end
if changed["autostart"] or changed["hostchange"] then
observe_cache()
end
if changed["quit"] then
autoquit()
end
if changed["piecewise"] and not opts.piecewise then
cache.part = 0
elseif changed["piecewise"] then
cache.endsec = convert_time(opts.autoend)
end
end
options.read_options(opts, "streamsave", update_opts)
update_opts{}
-- dump mode switching
local function mode_switch(value)
value = value or opts.dump_mode
if value == "cycle" then
if opts.dump_mode == "ab" then
value = "current"
elseif opts.dump_mode == "current" then
value = "continuous"
else
value = "ab"
end
end
if value == "continuous" then
opts.dump_mode = "continuous"
print("Continuous mode")
mp.osd_message("Cache write mode: Continuous")
elseif value == "ab" then
opts.dump_mode = "ab"
print("A-B loop mode")
mp.osd_message("Cache write mode: A-B loop")
elseif value == "current" then
opts.dump_mode = "current"
print("Current position mode")
mp.osd_message("Cache write mode: Current position")
else
msg.warn("Invalid dump mode '" .. value .. "'")
end
end
-- Set the principal part of the file name using the media title
function title_change(_, media_title, req)
if opts.force_title ~= "no" and not req then
file.title = opts.force_title
return end
if media_title then
-- Replacement of reserved file name characters on Windows
file.title = media_title:gsub("[\\/:*?\"<>|]", ".")
file.oldtitle = nil
end
end
-- Determine container for standard formats
function container(_, _, req)
local audio = mp.get_property("audio-codec-name")
local video = mp.get_property("video-format")
local file_format = mp.get_property("file-format")
if not file_format then
reset()
return end
if opts.force_extension ~= "no" and not req then
file.ext = opts.force_extension
observe_cache()
return end
if string.find(file_format, "mp4")
or ((video == "h264" or video == "av1" or not video) and
(audio == "aac" or not audio))
then
file.ext = ".mp4"
elseif (video == "vp8" or video == "vp9" or not video)
and (audio == "opus" or audio == "vorbis" or not audio)
then
file.ext = ".webm"
else
file.ext = ".mkv"
end
observe_cache()
file.oldext = nil
end
local function format_override(ext)
ext = ext or file.ext
file.oldext = file.oldext or file.ext
if ext == "revert" and file.ext == opts.force_extension then
container(_, _, true)
elseif ext == "revert" and opts.force_extension ~= "no" then
file.ext = opts.force_extension
elseif ext == "revert" then
file.ext = file.oldext
else
file.ext = ext
end
print("file extension changed to " .. file.ext)
mp.osd_message("streamsave: file extension changed to " .. file.ext)
end
local function title_override(title)
title = title or file.title
file.oldtitle = file.oldtitle or file.title
if title == "revert" and file.title == opts.force_title then
title_change(_, mp.get_property("media-title"), true)
elseif title == "revert" and opts.force_title ~= "no" then
file.title = opts.force_title
elseif title == "revert" then
file.title = file.oldtitle
else
file.title = title
end
print("title changed to " .. file.title)
mp.osd_message("streamsave: title changed to " .. file.title)
end
local function path_override(value)
value = value or opts.save_directory
file.oldpath = file.oldpath or opts.save_directory
if value == "revert" then
opts.save_directory = file.oldpath
else
opts.save_directory = value
end
file.path = mp.command_native({"expand-path", opts.save_directory})
print("Output directory changed to " .. opts.save_directory)
mp.osd_message("streamsave: directory changed to " .. opts.save_directory)
end
local function label_override(value)
if value == "cycle" then
if opts.output_label == "increment" then
value = "range"
elseif opts.output_label == "range" then
value = "timestamp"
elseif opts.output_label == "timestamp" then
value = "overwrite"
else
value = "increment"
end
end
opts.output_label = value or opts.output_label
validate_opts()
print("File label changed to " .. opts.output_label)
mp.osd_message("streamsave: label changed to " .. opts.output_label)
end
local function marks_override(value)
if not value or value == "no" then
opts.range_marks = false
ab_chapters = {}
mp.set_property_native("chapter-list", chapter_list)
print("Range marks disabled")
mp.osd_message("streamsave: range marks disabled")
elseif value == "yes" then
opts.range_marks = true
chapter_points()
print("Range marks enabled")
mp.osd_message("streamsave: range marks enabled")
else
msg.warn("Invalid input '" .. value .. "'. Use yes or no.")
mp.osd_message("streamsave: invalid input; use yes or no")
end
end
local function autostart_override(value)
if value and value ~= "no" and value ~= "yes" then
msg.warn("Invalid input '" .. value .. "'. Use yes or no.")
mp.osd_message("streamsave: invalid input; use yes or no")
return
end
if not value or value == "no" then
opts.autostart = false
print("Autostart disabled")
mp.osd_message("streamsave: autostart disabled")
elseif value == "yes" then
opts.autostart = true
print("Autostart enabled")
mp.osd_message("streamsave: autostart enabled")
end
observe_cache()
end
local function autoend_override(value)
opts.autoend = value or opts.autoend
validate_opts()
cache.endsec = convert_time(opts.autoend)
observe_cache()
print("Autoend set to " .. opts.autoend)
mp.osd_message("streamsave: autoend set to " .. opts.autoend)
end
local function hostchange_override(value)
value = value == "cycle" and (not opts.hostchange and "yes" or "no") or value
if value and value ~= "no" and value ~= "yes" then
msg.warn("Invalid input '" .. value .. "'. Use yes or no.")
mp.osd_message("streamsave: invalid input; use yes or no")
return
end
if not value or value == "no" then
opts.hostchange = false
mp.unobserve_property(reload)
local timer = cache.restart and cache.restart:kill()
print("Hostchange disabled")
mp.osd_message("streamsave: hostchange disabled")
elseif value == "yes" then
opts.hostchange = true
print("Hostchange enabled")
mp.osd_message("streamsave: hostchange enabled")
end
observe_cache()
end
local function quit_override(value)
opts.quit = value or opts.quit
validate_opts()
autoquit()
print("Quit set to " .. opts.quit)
mp.osd_message("streamsave: quit set to " .. opts.quit)
end
local function piecewise_override(value)
if not value or value == "no" then
opts.piecewise = false
cache.part = 0
print("Piecewise dumping disabled")
mp.osd_message("streamsave: piecewise dumping disabled")
elseif value == "yes" then
opts.piecewise = true
cache.endsec = convert_time(opts.autoend)
print("Piecewise dumping enabled")
mp.osd_message("streamsave: piecewise dumping enabled")
else
msg.warn("Invalid input '" .. value .. "'. Use yes or no.")
mp.osd_message("streamsave: invalid input; use yes or no")
end
end
local function range_flip()
loop.a = mp.get_property_number("ab-loop-a")
loop.b = mp.get_property_number("ab-loop-b")
if (loop.a and loop.b) and (loop.a > loop.b) then
loop.a, loop.b = loop.b, loop.a
mp.set_property_number("ab-loop-a", loop.a)
mp.set_property_number("ab-loop-b", loop.b)
end
end
local function loop_range()
local a_loop_osd = mp.get_property_osd("ab-loop-a")
local b_loop_osd = mp.get_property_osd("ab-loop-b")
loop.range = a_loop_osd .. " - " .. b_loop_osd
return loop.range
end
local function set_name(label)
return file.path .. "/" .. file.title .. label .. file.ext
end
local function increment_filename()
if set_name(-(file.inc or 1)) ~= file.name then
file.inc = 1
file.name = set_name(-file.inc)
end
-- check if file exists
while utils.file_info(file.name) do
file.inc = file.inc + 1
file.name = set_name(-file.inc)
end
end
local function range_stamp(mode)
local file_range
if mode == "ab" then
file_range = "-[" .. loop_range():gsub(":", ".") .. "]"
elseif mode == "current" then
local file_pos = mp.get_property_osd("playback-time", "0")
file_range = "-[" .. 0 .. " - " .. file_pos:gsub(":", ".") .. "]"
else
-- range tag is incompatible with full dump, fallback to increments
increment_filename()
return
end
file.name = set_name(file_range)
-- check if file exists, append increments if so
local i = 1
while utils.file_info(file.name) do
i = i + 1
file.name = set_name(file_range .. -i)
end
end
local function write_set(mode, file_name, file_pos, quiet)
local command = {
_flags = {
(not quiet or nil) and "osd-msg",
},
filename = file_name,
}
if mode == "ab" then
command["name"] = "ab-loop-dump-cache"
else
command["name"] = "dump-cache"
command["start"] = 0
command["end"] = file_pos or "no"
end
return command
end
local function cache_write(mode, quiet)
if not (file.title and file.ext) then
return end
if file.pending == 2 then
file.queue = file.queue or {}
-- honor extra write requests when pending queue is full
-- but limit number of outstanding write requests to be fulfilled
if #file.queue < 10 then
table.insert(file.queue, {mode, quiet})
end
return end
range_flip()
-- evaluate tagging conditions and set file name
if opts.output_label == "increment" then
increment_filename()
elseif opts.output_label == "range" then
range_stamp(mode)
elseif opts.output_label == "timestamp" then
file.name = set_name(-os.time())
elseif opts.output_label == "overwrite" then
file.name = set_name("")
end
-- dump cache according to mode
local file_pos
local file_name = file.name -- scope reduction so callback verifies correct file
file.pending = (file.pending or 0) + 1
continuous = mode == "continuous" or loop.a and not loop.b
if mode == "current" then
file_pos = mp.get_property_number("playback-time", 0)
elseif continuous and file.pending == 1 then
print("Dumping cache continuously to:" .. file_name)
end
write_file = mp.command_native_async (
write_set(mode, file_name, file_pos, quiet),
function(success, _, command_error)
command_error = command_error and msg.error(command_error)
-- check if file is written
if utils.file_info(file_name) then
if success then
print("Finished writing cache to:" .. file_name)
else
msg.warn("Possibly broken file created at:" .. file_name)
end
else
msg.error("File not written.")
end
if continuous and file.pending == 2 then
print("Dumping cache continuously to:" .. file.name)
end
file.pending = file.pending - 1
-- fulfil any write requests now that the pending queue has been serviced
if file.queue and #file.queue > 0 then
cache_write(file.queue[1][1], file.queue[1][2])
table.remove(file.queue, 1)
end
end
)
return true
end
--[[ This command attempts to align the A-B loop points to keyframes.
Use align-cache if you want to know which range will likely be dumped.
Keep in mind this changes the A-B loop points you've set.
This is sometimes inaccurate. Calling align_cache() again will reset the points
to their initial values. ]]
local function align_cache()
if not loop.aligned then
range_flip()
loop.a_revert = loop.a
loop.b_revert = loop.b
mp.command("ab-loop-align-cache")
loop.aligned = true
print("Adjusted range: " .. loop_range())
else
mp.set_property_native("ab-loop-a", loop.a_revert)
mp.set_property_native("ab-loop-b", loop.b_revert)
loop.aligned = false
print("Loop points reverted to: " .. loop_range())
mp.osd_message("A-B loop: " .. loop.range)
end
end
-- creates chapters at A-B loop points
function chapter_points()
if not opts.range_marks then
return end
local current_chapters = mp.get_property_native("chapter-list", {})
-- make sure master list is up to date
if current_chapters[1] and
not string.match(current_chapters[1]["title"], "^[AB] loop point$")
then
chapter_list = current_chapters
-- if a script has added chapters after A-B points are set then
-- add those to the original chapter list
elseif #current_chapters > #ab_chapters then
for i = #ab_chapters + 1, #current_chapters do
table.insert(chapter_list, current_chapters[i])
end
end
ab_chapters = {}
-- restore original chapter list if A-B points are cleared
-- otherwise set chapters to A-B points
range_flip()
if not loop.a and not loop.b then
mp.set_property_native("chapter-list", chapter_list)
else
if loop.a then
ab_chapters[1] = {
title = "A loop point",
time = loop.a
}
end
if loop.b and not loop.a then
ab_chapters[1] = {
title = "B loop point",
time = loop.b
}
elseif loop.b then
ab_chapters[2] = {
title = "B loop point",
time = loop.b
}
end
mp.set_property_native("chapter-list", ab_chapters)
end
end
-- stops writing the file
local function stop()
mp.abort_async_command(write_file or {})
end
function reset()
if cache.observed or cache.dumped then
stop()
mp.unobserve_property(automatic)
mp.unobserve_property(reload)
mp.unobserve_property(get_seekable_cache)
cache.endsec = convert_time(opts.autoend)
cache.observed = false
end
cache.prior = 0
cache.part = 0
cache.dumped = false
cache.switch = true
end
reset()
function get_seekable_cache(prop, range_check, underrun)
-- use the seekable part of the cache for more accurate timestamps
local cache_state = mp.get_property_native("demuxer-cache-state", {})
if underrun then
return cache_state["underrun"]
end
local seekable_ranges = cache_state["seekable-ranges"] or {}
if prop then
if range_check ~= false and
(#seekable_ranges == 0
or not mp.get_property_number("demuxer-cache-time"))
then
reset()
cache.use = opts.piecewise
observe_cache()
end
return
end
local seekable_ends = {0}
for i, range in ipairs(seekable_ranges) do
seekable_ends[i] = range["end"] or 0
end
cache.seekend = math.max(0, unpack(seekable_ends))
return cache.seekend
end
function reload(_, play_time)
local cache_duration = mp.get_property_number("demuxer-cache-duration")
if play_time and play_time >= cache.seekend - 0.25
or cache_duration and math.abs(cache.prior - cache_duration) > 4800
or get_seekable_cache(nil, false, true)
then
reset()
cache.restart = cache.restart or mp.add_timeout(300, function() end)
cache.restart:resume()
msg.warn("Reloading stream due to host change.")
mp.command("playlist-play-index current")
end
end
function automatic(_, cache_time)
if opts.hostchange and cache.prior ~= 0
and (not cache_time or math.abs(cache_time - cache.prior) > 300
or mp.get_property_number("demuxer-cache-duration", 0) > 11000)
and not mp.get_property_bool("seeking")
then
if not cache.restart or not cache.restart:is_enabled() then
reset()
cache.observed = true
cache.prior = mp.get_property_number("demuxer-cache-duration", 0)
get_seekable_cache()
mp.observe_property("playback-time", "number", reload)
else
-- reload stream
cache.restart:kill()
reset()
msg.warn("Reloading stream due to host change.")
mp.command("playlist-play-index current")
end
return
elseif not cache_time then
reset()
cache.use = opts.piecewise
observe_cache()
return
end
-- cache write according to automatic options
if opts.autostart and not cache.dumped
and (not cache.endsec or cache_time < cache.endsec
or opts.piecewise)
then
if opts.piecewise and cache.part ~= 0 then
cache.dumped = cache_write("ab")
else
cache.dumped = cache_write("continuous", opts.hostchange)
-- update the piece time if there's a track/seeking reset
cache.part = cache.use and cache.dumped and cache_time or 0
cache.use = cache.use and cache.part == 0
end
end
-- the seekable ranges update slowly, which is why they're used to check
-- against switches for increased certainty, but this means the switch properties
-- should be watched only when the ranges exist
if cache.switch and get_seekable_cache() ~= 0 then
cache.switch = false
mp.observe_property("current-tracks/audio/id", "number", get_seekable_cache)
mp.observe_property("current-tracks/video/id", "number", get_seekable_cache)
mp.observe_property("seeking", "bool", get_seekable_cache)
end
-- unobserve cache time if not needed
if cache.dumped and not cache.switch
and not cache.endsec and not opts.hostchange
then
mp.unobserve_property(automatic)
cache.observed = false
cache.prior = 0
return
end
-- stop cache dump
if cache.endsec and cache.dumped and
cache_time - cache.part >= cache.endsec
then
if opts.piecewise then
cache.part = get_seekable_cache()
mp.set_property_number("ab-loop-a", cache.part)
mp.set_property("ab-loop-b", "no")
-- try and make the next piece start on the final keyframe of this piece
loop.aligned = false
align_cache()
cache.dumped = false
else
cache.endsec = nil
end
stop()
end
cache.prior = cache_time
end
function autoquit()
if opts.quit == "no" then
if quit_timer then
quit_timer:kill()
end
elseif not quit_timer then
quit_timer = mp.add_timeout(quitseconds,
function()
stop()
mp.command("quit")
print("Quit after " .. opts.quit)
end)
else
quit_timer["timeout"] = quitseconds
quit_timer:kill()
quit_timer:resume()
end
end
autoquit()
-- cache time observation switch for runtime changes
function observe_cache()
local network = mp.get_property_bool("demuxer-via-network")
local obs_xyz = opts.autostart or cache.endsec or opts.hostchange
if not cache.observed and obs_xyz and network then
mp.observe_property("demuxer-cache-time", "number", automatic)
cache.observed = true
elseif (cache.observed or cache.dumped) and (not obs_xyz or not network) then
reset()
end
end
mp.observe_property("media-uuid", "string", title_change)
--[[ video and audio formats observed in order to handle track changes
useful if e.g. --script-opts=ytdl_hook-all_formats=yes
or script-opts=ytdl_hook-use_manifests=yes ]]
mp.observe_property("audio-codec-name", "string", container)
mp.observe_property("video-format", "string", container)
mp.observe_property("file-format", "string", container)
--[[ Loading chapters can be slow especially if they're passed from
an external file, so make sure existing chapters are not overwritten
by observing A-B loop changes only after the file is loaded. ]]
local function on_file_load()
mp.observe_property("ab-loop-a", "native", chapter_points)
mp.observe_property("ab-loop-b", "native", chapter_points)
end
mp.register_event("file-loaded", on_file_load)
mp.register_script_message("streamsave-rec", function() cache_write(opts.dump_mode)
end)
mp.register_script_message("streamsave-mode", mode_switch)
mp.register_script_message("streamsave-title", title_override)
mp.register_script_message("streamsave-extension", format_override)
mp.register_script_message("streamsave-path", path_override)
mp.register_script_message("streamsave-label", label_override)
mp.register_script_message("streamsave-marks", marks_override)
mp.register_script_message("streamsave-autostart", autostart_override)
mp.register_script_message("streamsave-autoend", autoend_override)
mp.register_script_message("streamsave-hostchange", hostchange_override)
mp.register_script_message("streamsave-quit", quit_override)
mp.register_script_message("streamsave-piecewise", piecewise_override)
mp.add_key_binding("Alt+z", "mode-switch", function() mode_switch("cycle") end)
mp.add_key_binding("Ctrl+x", "stop-cache-write", stop)
mp.add_key_binding("Alt+x", "align-cache", align_cache)
mp.add_key_binding("Ctrl+z", "cache-write",
function() cache_write(opts.dump_mode)
end)