Add mpv lua script

notify_media.lua

@@ -0,0 +1,165 @@
+DEBUG = false
+
+local utils = require 'mp.utils'
+pid = utils.getpid()
+
+-- Print contents of `tbl`, with indentation.
+-- `indent` sets the initial level of indentation.
+function tprint (tbl, indent)
+    if not indent then indent = 0 end
+        for k, v in pairs(tbl) do
+            formatting = string.rep("  ", indent) .. k .. ": "
+        if type(v) == "table" then
+            print(formatting)
+            tprint(v, indent+1)
+        elseif type(v) == 'boolean' then
+            print(formatting .. tostring(v))      
+        else
+            print(formatting .. v)
+        end
+    end
+end
+
+function debug_log(message)
+    if DEBUG then
+        if not message then
+            print("DEBUG: nil")
+            return
+        end
+        if "table" == type(message) then
+            print("DEBUG: ")
+            tprint(message)
+        else
+            print("DEBUG: " .. message)
+        end
+    end
+end
+
+ipc_socket_file = "\\\\.\\pipe\\mpvmcsocket"
+
+function write_to_socket(message)
+    _, pipe = pcall(io.open, ipc_socket_file, "w")
+    if pipe then
+        pcall(pipe.write, pipe, message)
+        pcall(pipe.flush, pipe)
+        pcall(pipe.close, pipe)
+        debug_log(message)
+    end
+end
+
+function get_metadata(data, keys)
+    for _, v in pairs(keys) do
+        if data[v] and string.len(data[v]) > 0 then
+            return data[v]
+        end
+    end
+    return ""
+end
+
+function encode_element(str)
+    -- return str:gsub("%(", "\\\\["):gsub("%)", "\\\\]")
+    return tohex(str)
+end
+
+function tohex(str)
+    return (str:gsub('.', function (c)
+        return string.format('%02X', string.byte(c))
+    end))
+end
+
+function save_shot(path)
+    result = mp.commandv("screenshot-to-file", path)
+    if not result then
+        mp.add_timeout(0.1, function() save_shot(path) end)
+    end
+end
+
+function notify_current_file()
+    metadata = mp.get_property_native("metadata")
+    debug_log(metadata)
+    if not metadata then
+        return
+    end
+
+    artist = get_metadata(metadata, { "artist", "ARTIST", "Artist" })
+    title = get_metadata(metadata, { "title", "TITLE", "Title", "icy-title" })
+
+    if not artist or artist == "" or not title or title == "" then
+        chapter_metadata = mp.get_property_native("chapter-metadata")
+
+        if chapter_metadata then
+            chapter_artist = chapter_metadata["performer"]
+            if not artist or artist == "" then
+                artist = chapter_artist
+            end
+
+            chapter_title = chapter_metadata["title"]
+            if not title or title == "" then
+                title = chapter_title
+            end
+        end
+    end
+
+    if not title or title == "" then
+        title = mp.get_property_native("filename/no-ext")
+    end
+
+    path = mp.get_property_native("path")
+    if path:sub(2, 3) ~= ":\\" and path:sub(2, 3) ~= ":/" then
+        dir = mp.get_property_native("working-directory")
+        path = dir + "\\" + path
+    end
+
+    if not artist then
+        artist = ""
+    end
+
+    if title then
+        title = encode_element(title)
+    end
+    if artist then
+        artist = encode_element(artist)
+    end
+    path = encode_element(path)
+
+    local user_path = mp.command_native({"expand-path", "~~/"})
+    shot_path = user_path .. "\\" .. pid .. ".jpg"
+    save_shot(shot_path)
+    local shot_path_encoded = encode_element(shot_path)
+
+    message_content = "^[setFile](pid=" .. pid .. ")(title=" .. title .. ")(artist=" .. artist .. ")(path=" .. path .. ")(shot_path=" .. shot_path_encoded .. ")$"
+    write_to_socket(message_content)
+end
+
+function play_state_changed()
+    idle = mp.get_property_native("core-idle")
+    is_playing = not idle
+
+    message_content = "^[setState](pid=" .. pid .. ")(playing=" .. tostring(is_playing) .. ")$"
+    write_to_socket(message_content)
+
+    if not idle then
+        mp.add_timeout(10, play_state_changed)
+    end
+end
+
+function notify_metadata_updated()
+    notify_current_file()
+end
+
+mp.set_property("options/input-ipc-server", "\\\\.\\pipe\\mpvsocket_" .. pid)
+
+mp.register_event("file-loaded", notify_current_file)
+mp.observe_property("metadata", nil, notify_metadata_updated)
+mp.observe_property("chapter", nil, notify_metadata_updated)
+mp.register_event("end-file", play_state_changed)
+mp.observe_property("core-idle", nil, play_state_changed)
+
+function on_quit()
+    if shot_path then
+        os.remove(shot_path)
+    end
+    write_to_socket("^[setQuit](pid=" .. pid .. ")(quit=true)$")
+end
+
+mp.register_event("shutdown", on_quit)