misc.lua

--[[

    https://github.com/stax76/mpv-scripts

    This script consist of various small unrelated features.

    Not used code sections can be removed.

    Bindings must be added manually to input.conf.



    Show media info on screen
    -------------------------
    Prints detailed media info on the screen.
    
    Depends on the CLI tool mediainfo:  
    https://mediaarea.net/en/MediaInfo/Download  
    On Ubuntu: sudo apt install mediainfo

    In input.conf add:
    i script-message-to misc print-media-info



    Load files/URLs from clipboard
    ------------------------------
    Loads one or multiple files/URLs from the clipboard.
    The clipboard format can be of type string or file object.
    Allows appending to the playlist.
    On Linux requires xclip being installed.

    In input.conf add:
    ctrl+v script-message-to misc load-from-clipboard
    ctrl+V script-message-to misc append-from-clipboard



    Cycle audio and subtitle tracks
    -------------------------------
    If there are 20+ subtitle tracks, it's annoying cycling through all
    of them. This feature allows you to cycle only through languages
    you actually know.

    In mpv.conf define your known languages:
    alang = de,deu,ger,en,eng #German/English
    slang = en,eng,de,deu,ger #English/German

    If you don't know the language IDs, use the terminal,
    mpv prints the language IDs there whenever a video file is loaded.

    In input.conf add:
    SHARP script-message-to misc cycle-known-tracks audio
    j     script-message-to misc cycle-known-tracks sub up
    J     script-message-to misc cycle-known-tracks sub down

    ~~/script-opts/misc.conf:
    #include_no_audio=no
    #include_no_sub=yes

    ## If more than 5 tracks exist, only known are cycled,
    ## define 0 to always cycle only known tracks.
    #max_audio_track_count=5
    #max_sub_track_count=5

    If you prefer a menu:
    https://github.com/stax76/mpv-scripts?tab=readme-ov-file#command_palette
    https://github.com/stax76/mpv-scripts#search-menu
    https://github.com/dyphire/mpv-scripts/blob/main/track-list.lua
    https://codeberg.org/NRK/mpv-toolbox/src/branch/master/mdmenu
    https://github.com/tomasklaen/uosc

    The code was originally written by stax76, it was later
    greatly improved by kaoneko making it much shorter.


    Jump to a random position in the playlist
    -----------------------------------------
    In input.conf add:
    ctrl+r script-message-to misc playlist-random

    If pos=last it jumps to first instead of random.



    Quick Bookmark
    --------------
    Creates or restores a bookmark. Supports one bookmark per video.

    Usage:
    Create a folder in the following location:
    ~~/script-settings/quick-bookmark/
    Or create it somewhere else, config at:
    ~~/script-opts/misc.conf:
    quick_bookmark_folder=<folder path>

    In input.conf add:
    ctrl+q script-message-to misc quick-bookmark

 

    Playlist Next/Prev
    ------------------
    Like the regular playlist-next/playlist-prev, but does not restart playback
    of the first or last file, in case the first or last track already plays,
    instead shows a OSD message.

    F11 script-message-to misc playlist-prev # Go to previous file in playlist
    F12 script-message-to misc playlist-next # Go to next file in playlist



    Playlist First/Last
    -------------------
    Navigates to the first or last track in the playlist,
    in case the first or last track already plays, it does not
    restart playback, instead shows a OSD message.

    Home script-message-to misc playlist-first # Go to first file in playlist
    End  script-message-to misc playlist-last  # Go to last file in playlist



    Restart mpv
    -----------
    Restarts mpv restoring the properties path, time-pos,
    pause and volume, the playlist is not restored.

    r script-message-to misc restart-mpv



    Execute Lua code
    ----------------
    Allows to execute Lua Code directly from input.conf.

    It's necessary to add a binding to input.conf:
    #Navigates to the last file in the playlist
    END script-message-to misc execute-lua-code "mp.set_property_number('playlist-pos', mp.get_property_number('playlist-count') - 1)"



    When seeking displays position and duration like so:
    ----------------------------------------------------
    70:00 / 80:00

    Which is different from most players which use:

    01:10:00 / 01:20:00

    input.conf:
    Right no-osd seek 5; script-message-to misc show-position

]]--

----- options

local o = {
    -- Cycle audio and subtitle tracks
    include_no_audio = false,
    include_no_sub = true,
    max_audio_track_count = 5,
    max_sub_track_count = 5,
    -- Quick Bookmark
    quick_bookmark_folder = "~~/script-settings/quick-bookmark/",
}

local opt = require "mp.options"
opt.read_options(o)

----- string

function is_empty(input)
    if input == nil or input == "" then
        return true
    end
end

function contains(input, find)
    if not is_empty(input) and not is_empty(find) then
        return input:find(find, 1, true)
    end
end

function trim(input)
    if not is_empty(input) then
        return input:match "^%s*(.-)%s*$"
    end
end

function split(input, sep)
    local tbl = {}

    if not is_empty(input) then
        for str in string.gmatch(input, "([^" .. sep .. "]+)") do
            table.insert(tbl, str)
        end
    end

    return tbl
end

----- list

function list_contains(list, value)
    for _, v in pairs(list) do
        if v == value then
            return true
        end
    end

    return false
end

----- math

function round(value)
    return value >= 0 and math.floor(value + 0.5) or math.ceil(value - 0.5)
end

----- file

function file_exists(path)
    if is_empty(path) then return false end
    local file = io.open(path, "r")

    if file ~= nil then
        io.close(file)
        return true
    end
end

function file_read(file_path)
    local file = assert(io.open(file_path, "r"))
    local content = file:read("*all")
    file:close()
    return content
end

function file_write(path, content)
    local file = assert(io.open(path, "w"))
    file:write(content)
    file:close()
end

----- shared

local is_windows = package.config:sub(1,1) == "\\"
local msg = require "mp.msg"
local utils = require "mp.utils"

function get_temp_dir()
    if is_windows then
        return os.getenv("TEMP") .. "\\"
    else
        return "/tmp/"
    end
end

----- Jump to a random position in the playlist

mp.register_script_message("playlist-random", function ()
    local count = mp.get_property_number("playlist-count")
    math.randomseed(os.time())
    local new_pos = math.random(0, count - 1)
    local current_pos = mp.get_property_number("playlist-pos")

    if current_pos == count - 1 then
        new_pos = 0
    end

    mp.set_property_number("playlist-pos", new_pos)
end)

----- Execute Lua code

mp.register_script_message("execute-lua-code", function (code)
    loadstring(code)()
end)

----- Alternative seek OSD message

function pad_zero(value)
    local value = round(value)

    if value > 9 then
        return "" .. value
    else
        return "0" .. value
    end
end

function format_pos(value)
    local seconds = round(value)

    if seconds < 0 then
        seconds = 0
    end

    local pos_min_floor = math.floor(seconds / 60)
    local sec_rest = seconds - pos_min_floor * 60

    return pad_zero(pos_min_floor) .. ":" .. pad_zero(sec_rest)
end

function show_pos()
    local position = mp.get_property_number("time-pos")
    local duration = mp.get_property_number("duration")

    if position > duration then
        position = duration
    end

    if position ~= 0 then
        mp.osd_message(format_pos(position) .. " / " .. format_pos(duration))
    end
end

mp.register_script_message("show-position", function (mode)
    mp.add_timeout(0.05, show_pos)
end)

----- Print media info on screen

local media_info_cache = {}

function show_text(text, duration, font_size)
    if text ~= "" and text ~= nil then
        mp.command('show-text "${osd-ass-cc/0}{\\\\fs' .. font_size ..
            '}${osd-ass-cc/1}' .. text .. '" ' .. duration)
    end
end

function get_media_info()
    local path = mp.get_property("path")

    if media_info_cache[path] then
        return media_info_cache[path]
    end

    local media_info_format = [[General;N: %FileNameExtension%\\nG: %Format%, %FileSize/String%, %Duration/String%, %OverallBitRate/String%, %Recorded_Date%\\n
Video;V: %Format%, %Format_Profile%, %Width%x%Height%, %BitRate/String%, %FrameRate% FPS\\n
Audio;A: %Language/String%, %Format%, %Format_Profile%, %BitRate/String%, %Channel(s)% ch, %SamplingRate/String%, %Title%\\n
Text;S: %Language/String%, %Format%, %Format_Profile%, %Title%\\n]]

    local format_file = get_temp_dir() .. "media-info-format-2.txt"

    if not file_exists(format_file) then
        file_write(format_file, media_info_format)
    end

    if contains(path, "://") or not file_exists(path) then
        return
    end

    local proc_result = mp.command_native({
        name = "subprocess",
        playback_only = false,
        capture_stdout = true,
        args = {"mediainfo", "--inform=file://" .. format_file, path},
    })

    if proc_result.status == 0 then
        local output = proc_result.stdout

        output = string.gsub(output, ", , ,", ",")
        output = string.gsub(output, ", ,", ",")
        output = string.gsub(output, ": , ", ": ")
        output = string.gsub(output, ", \\n\r*\n", "\\n")
        output = string.gsub(output, "\\n\r*\n", "\\n")
        output = string.gsub(output, ", \\n", "\\n")
        output = string.gsub(output, "%.000 FPS", " FPS")
        output = string.gsub(output, "MPEG Audio, Layer 3", "MP3")

        media_info_cache[path] = output

        return output
    else
        mp.osd_message("mediainfo failed to run")
    end
end

mp.register_script_message("print-media-info", function ()
    show_text(get_media_info(), 5000, 16)
end)

----- Playlist Next/Prev

mp.register_script_message("playlist-next", function ()
    local count = mp.get_property_number("playlist-count")
    if count == 0 then return end
    local pos = mp.get_property_number("playlist-pos")

    if pos == count - 1 then
        mp.osd_message("Already last track")
        return
    end

    mp.set_property_number("playlist-pos", pos + 1)
end)

mp.register_script_message("playlist-prev", function ()
    local count = mp.get_property_number("playlist-count")
    if count == 0 then return end
    local pos = mp.get_property_number("playlist-pos")

    if pos == 0 then
        mp.osd_message("Already first track")
        return
    end

    mp.set_property_number("playlist-pos", pos - 1)
end)

----- Playlist First/Last

mp.register_script_message("playlist-first", function ()
    local count = mp.get_property_number("playlist-count")
    if count == 0 then return end
    local pos = mp.get_property_number("playlist-pos")

    if pos == 0 then
        mp.osd_message("Already first track")
        return
    end

    mp.set_property_number("playlist-pos", 0)
end)

mp.register_script_message("playlist-last", function ()
    local count = mp.get_property_number("playlist-count")
    if count == 0 then return end
    local pos = mp.get_property_number("playlist-pos")

    if pos == count - 1 then
        mp.osd_message("Already last track")
        return
    end

    mp.set_property_number("playlist-pos", count - 1)
end)

----- Load files from clipboard

function loadfiles(mode)
    if is_windows then
        local ps_code = [[
            Add-Type -AssemblyName System.Windows.Forms
            $containsFiles = [Windows.Forms.Clipboard]::ContainsFileDropList()
            
            if ($containsFiles) {
                [Windows.Forms.Clipboard]::GetFileDropList() -join [Environment]::NewLine
            } else {
                (Get-Clipboard).Replace('"', "")
            }
        ]]

        proc_args = { "powershell", "-command", ps_code }
    else
        proc_args = { "xclip", "-o", "-selection", "clipboard" }
    end

    subprocess = {
        name = "subprocess",
        args = proc_args,
        playback_only = false,
        capture_stdout = true,
        capture_stderr = true
    }

    proc_result = mp.command_native(subprocess)

    if proc_result.status < 0 then
        msg.error("Error string: " .. proc_result.error_string)
        msg.error("Error stderr: " .. proc_result.stderr)
        return
    end

    proc_output = trim(proc_result.stdout)

    if is_empty(proc_output) then return end

    if contains(proc_output, "\n") then
        mp.commandv("loadlist", "memory://" .. proc_output, mode)
    else
        mp.commandv("loadfile", proc_output, mode)
    end
end

mp.register_script_message("load-from-clipboard", function ()
    loadfiles("replace")
end)

mp.register_script_message("append-from-clipboard", function ()
    loadfiles("append")
end)

----- Restart mpv

mp.register_script_message("restart-mpv", function ()
    local restart_args = {
        "mpv",
        "--pause=" .. mp.get_property("pause"),
        "--volume=" .. mp.get_property("volume"),
    }

    local playlist_pos = mp.get_property_number("playlist-pos")

    if playlist_pos > -1 then
        table.insert(restart_args, "--start=" .. mp.get_property("time-pos"))
        table.insert(restart_args, mp.get_property("path"))
    end

    mp.command_native({
        name = "subprocess",
        playback_only = false,
        detach = true,
        args = restart_args,
    })

    mp.command("quit")
end)

----- Cycle audio and subtitle tracks

mp.register_script_message("cycle-known-tracks", function (mode, dir)
    local m = mode:sub(1,1)
    local lang_list = {}
    for _,lang in pairs(mp.get_property_native(m.."lang")) do
        lang_list[lang:gsub(" ", "")] = true
    end
    local track_list = mp.get_property_native("track-list")
    local id_list = {o["include_no_"..mode] and "no" or nil}
    local count = 0
    local max_count = o["max_"..mode.."_track_count"]

    for _,track in pairs(track_list) do
        if track.type == mode then
            count = count + 1
            if lang_list[track.lang] or not track.lang or
                track.selected or not next(lang_list)
            then table.insert(id_list, track.id) end
        end
    end

    if #id_list < 2 then
        return
    elseif count <= max_count then
       mp.command("cycle "..mode.." "..(dir or ""))
    else
        mp.command("cycle-values "..(dir == "down" and "!reverse " or "")..
                    m.."id "..table.concat(id_list, " "))
    end
end)

----- Quick Bookmark

mp.register_script_message("quick-bookmark", function ()
    local path = mp.get_property("path")

    if is_empty(path) then
        return
    end

    local folder = mp.command_native({"expand-path", o.quick_bookmark_folder})

    if utils.file_info(folder) == nil then
        msg.error("Bookmark folder not found, create it at:\n" .. folder)
        return
    end

    if file_exists(path) then
        _, path = utils.split_path(path)
        path = utils.join_path(folder, path)
    else
        path = utils.join_path(folder, string.gsub(path, "[/\\:]", ""))
    end

    if file_exists(path) then
        mp.set_property_number("time-pos", tonumber(file_read(path)))
        os.remove(path)
    else
        file_write(path, mp.get_property("time-pos"))
        mp.osd_message("Bookmark saved")
    end
end)