split menu load logic to lua

CMakeLists.txt

@@ -17,7 +17,6 @@ set(CMAKE_SHARED_LIBRARY_PREFIX "")
 add_library(menu SHARED
     src/mpv/misc/bstr.c
     src/mpv/misc/dispatch.c
-    src/mpv/misc/node.c
     src/mpv/ta/ta.c
     src/mpv/ta/ta_talloc.c
     src/mpv/ta/ta_utils.c
@@ -31,7 +30,6 @@ add_library(menu SHARED
 set_property(TARGET menu PROPERTY POSITION_INDEPENDENT_CODE ON)
 
 target_include_directories(menu PRIVATE src/mpv ${MPV_INCLUDE_DIRS})
-target_link_libraries(menu PRIVATE shlwapi)
 target_compile_definitions(menu PRIVATE MPV_CPLUGIN_DYNAMIC_SYM)
 
 install(TARGETS menu RUNTIME DESTINATION .)

src/lua/dyn_menu.lua

@@ -7,17 +7,18 @@ local msg = require('mp.msg')
 
 -- user options
 local o = {
+    uosc_syntax = false,     -- toggle uosc menu syntax support
     max_title_length = 80,   -- limit the title length, set to 0 to disable.
     max_playlist_items = 20, -- limit the playlist items in submenu, set to 0 to disable.
 }
 opts.read_options(o)
 
-local menu_prop = 'user-data/menu/items'                 -- menu data property
-local menu_items = mp.get_property_native(menu_prop, {}) -- raw menu data
-local menu_items_dirty = false                           -- menu data dirty flag
-local dyn_menus = {}                                     -- dynamic menu list
-local keyword_to_menu = {}                               -- keyword -> menu
-local has_uosc = false                                   -- uosc installed flag
+local menu_prop = 'user-data/menu/items' -- menu data property
+local menu_items = {}                    -- raw menu data
+local menu_items_dirty = false           -- menu data dirty flag
+local dyn_menus = {}                     -- dynamic menu list
+local keyword_to_menu = {}               -- keyword -> menu
+local has_uosc = false                   -- uosc installed flag
 
 -- lua expression compiler (copied from mpv auto_profiles.lua)
 ------------------------------------------------------------------------
@@ -501,15 +502,6 @@ local function load_dyn_menus()
     mp.commandv('script-message', 'menu-ready')
 end
 
--- menu data update callback
-local function menu_data_cb(name, items)
-    if not items or #items == 0 then return end
-    mp.unobserve_property(menu_data_cb)
-
-    menu_items = items
-    load_dyn_menus()
-end
-
 -- script message: get <keyword> <src>
 mp.register_script_message('get', function(keyword, src)
     if not src or src == '' then
@@ -570,13 +562,100 @@ mp.register_idle(function()
     end
 end)
 
--- parse menu data when menu items ready
---
--- NOTE: to simplify the code, we only procss the first valid update
---       event and ignore the rest, this make it conflict with other
---       scripts that also update the menu data property.
-if #menu_items > 0 then
+-- read input.conf content
+local function get_input_conf()
+    local prop = mp.get_property_native('input-conf')
+    if prop:sub(1, 9) == 'memory://' then return prop:sub(10) end
+
+    prop = prop == '' and '~~/input.conf' or prop
+    local conf_path = mp.command_native({ 'expand-path', prop })
+
+    local f, err = io.open(conf_path, 'rb')
+    if not f then
+        msg.error('failed to open file: ' .. conf_path)
+        return nil
+    end
+
+    local conf = f:read('*all')
+    f:close()
+    return conf
+end
+
+-- parse input.conf, return menu items
+local function parse_input_conf(conf)
+    local items = {}
+    local by_id = {}
+
+    local function extract_title(cmd)
+        if not cmd or cmd == '' then return '' end
+        local title = cmd:match('#menu:%s*(.*)%s*')
+        if not title and o.uosc_syntax then title = cmd:match('#!%s*(.*)%s*') end
+        return title or ''
+    end
+
+    local function split_title(title, ch)
+        local list = {}
+        if not title or title == '' then return list end
+        title = title:match('(.-)#%s*.-%s*$') or title -- strip trailing comment
+
+        local pattern = '(.-)%s*>%s*'
+        local last_ends = 1
+        local starts, ends, match = title:find(pattern)
+        while starts do
+            list[#list + 1] = match
+            last_ends = ends + 1
+            starts, ends, match = title:find(pattern, last_ends)
+        end
+        if last_ends < (#title + 1) then list[#list + 1] = title:sub(last_ends) end
+
+        return list
+    end
+
+    for line in conf:gmatch('[^\r\n]+') do
+        if line:sub(1, 1) ~= '#' or o.uosc_syntax then
+            local key, cmd = line:match('%s*([%S]+)%s+(.-)%s*$')
+            local title = extract_title(cmd)
+            local list = split_title(title)
+
+            local submenu_id = ''
+            local target_menu = items
+
+            for id, name in ipairs(list) do
+                if id < #list then
+                    submenu_id = submenu_id .. name
+                    if not by_id[submenu_id] then
+                        local submenu = {}
+                        by_id[submenu_id] = submenu
+                        target_menu[#target_menu + 1] = {
+                            title = name,
+                            type = 'submenu',
+                            submenu = submenu,
+                        }
+                    end
+                    target_menu = by_id[submenu_id]
+                else
+                    if name == '-' or (o.uosc_syntax and name:sub(1, 3) == '---') then
+                        target_menu[#target_menu + 1] = {
+                            type = 'separator',
+                        }
+                    else
+                        target_menu[#target_menu + 1] = {
+                            title = (key ~= '' and key ~= '_' and key ~= '#') and (name .. "\t" .. key) or name,
+                            cmd = cmd,
+                        }
+                    end
+                end
+            end
+        end
+    end
+
+    return items
+end
+
+-- NOTE: this script is conflict with other scripts that also update the menu data property
+local conf = get_input_conf()
+if conf then
+    menu_items = parse_input_conf(conf)
+    menu_items_dirty = true
     load_dyn_menus()
-else
-    mp.observe_property(menu_prop, 'native', menu_data_cb)
 end

src/menu.c

@@ -2,12 +2,8 @@
 // SPDX-License-Identifier: GPL-2.0-only
 
 #include "misc/bstr.h"
-#include "misc/node.h"
 #include "menu.h"
 
-#define MENU_PREFIX "#menu:"
-#define MENU_PREFIX_UOSC "#!"
-
 // escape & to && for menu title
 static wchar_t *escape_title(void *talloc_ctx, char *title) {
     void *tmp = talloc_new(NULL);
@@ -145,120 +141,14 @@ void build_menu(void *talloc_ctx, HMENU hmenu, mpv_node *node) {
     }
 }
 
-static bool is_separator(bstr text, bool uosc) {
-    return bstr_equals0(text, "-") || (uosc && bstr_startswith0(text, "---"));
-}
-
-// return submenu node if exists, otherwise create a new one
-static mpv_node *add_submenu(mpv_node *list, char *title) {
-    for (int i = 0; i < list->u.list->num; i++) {
-        mpv_node *item = &list->u.list->values[i];
-
-        mpv_node *type_node = node_map_get(item, "type");
-        if (!type_node || strcmp(type_node->u.string, "submenu") != 0) continue;
-
-        mpv_node *title_node = node_map_get(item, "title");
-        if (title_node && strcmp(title_node->u.string, title) == 0) {
-            mpv_node *submenu = node_map_get(item, "submenu");
-            if (!submenu)
-                submenu = node_map_add(item, "submenu", MPV_FORMAT_NODE_ARRAY);
-            return submenu;
-        }
-    }
-
-    mpv_node *node = node_array_add(list, MPV_FORMAT_NODE_MAP);
-    node_map_add_string(node, "title", title);
-    node_map_add_string(node, "type", "submenu");
-    return node_map_add(node, "submenu", MPV_FORMAT_NODE_ARRAY);
-}
-
-// parse menu line and add to menu node
-static void parse_menu(mpv_node *list, bstr key, bstr cmd, bstr text,
-                       bool uosc) {
-    bstr name, rest, comment;
-
-    name = bstr_split(text, ">", &rest);
-    name = bstr_split(name, "#", &comment);
-    name = bstr_strip(name);
-    if (!name.len) return;
-
-    void *tmp = talloc_new(NULL);
-
-    if (!rest.len) {
-        if (is_separator(name, uosc)) {
-            mpv_node *node = node_array_add(list, MPV_FORMAT_NODE_MAP);
-            node_map_add_string(node, "type", "separator");
-        } else {
-            bstr title = bstrdup(tmp, name);
-            if (key.len > 0 && !bstr_equals0(key, "_")) {
-                bstr_xappend(tmp, &title, bstr0("\t"));
-                bstr_xappend(tmp, &title, key);
-            }
-            mpv_node *node = node_array_add(list, MPV_FORMAT_NODE_MAP);
-            node_map_add_string(node, "title", bstrdup0(tmp, title));
-            node_map_add_string(node, "cmd", bstrdup0(tmp, cmd));
-        }
-    } else {
-        mpv_node *sub = add_submenu(list, bstrdup0(tmp, name));
-        if (!comment.len) parse_menu(sub, key, cmd, rest, uosc);
-    }
-
-    talloc_free(tmp);
-}
-
-static bool split_menu(bstr line, bstr *left, bstr *right, bool uosc) {
-    if (!line.len) return false;
-    if (!bstr_split_tok(line, MENU_PREFIX, left, right)) {
-        if (!uosc || !bstr_split_tok(line, MENU_PREFIX_UOSC, left, right))
-            return false;
-    }
-    *left = bstr_strip(*left);
-    *right = bstr_strip(*right);
-    return right->len > 0;
-}
-
-// build menu node from input.conf
-void load_menu(mpv_node *node, plugin_config *conf) {
-    void *tmp = talloc_new(NULL);
-    char *path = mp_get_prop_string(tmp, "input-conf");
-    if (path == NULL || strlen(path) == 0) path = "~~/input.conf";
-
-    bstr data = bstr0(mp_read_file(tmp, path));
-    node_init(node, MPV_FORMAT_NODE_ARRAY, NULL);
-
-    while (data.len > 0) {
-        bstr line = bstr_strip_linebreaks(bstr_getline(data, &data));
-        line = bstr_lstrip(line);
-        if (!line.len) continue;
-
-        bstr key, cmd, left, right;
-        if (bstr_eatstart0(&line, "#")) {
-            if (!conf->uosc) continue;
-            key = bstr0(NULL);
-            cmd = bstr_strip(line);
-        } else {
-            key = bstr_split(line, WHITESPACE, &cmd);
-            cmd = bstr_strip(cmd);
-        }
-        if (split_menu(cmd, &left, &right, conf->uosc))
-            parse_menu(node, key, cmd, right, conf->uosc);
-    }
-
-    talloc_free(tmp);
-}
-
 // update HMENU if menu node changed
 void update_menu(plugin_ctx *ctx, mpv_node *node) {
-    if (equal_mpv_node(ctx->node, node)) return;
-
-    copy_mpv_node(ctx->node, node);
+    if (!ctx->hmenu) return;
 
-    if (ctx->hmenu) {
-        while (GetMenuItemCount(ctx->hmenu) > 0)
-            RemoveMenu(ctx->hmenu, 0, MF_BYPOSITION);
-        talloc_free_children(ctx->hmenu_ctx);
-        build_menu(ctx->hmenu_ctx, ctx->hmenu, ctx->node);
-    }
+    while (GetMenuItemCount(ctx->hmenu) > 0)
+        RemoveMenu(ctx->hmenu, 0, MF_BYPOSITION);
+    talloc_free_children(ctx->hmenu_ctx);
+    build_menu(ctx->hmenu_ctx, ctx->hmenu, node);
 }
 
 // show menu at position if it is in window

src/menu.h

@@ -10,7 +10,6 @@
 
 #define MENU_DATA_PROP "user-data/menu/items"
 
-void load_menu(mpv_node *node, plugin_config *conf);
 void build_menu(void *talloc_ctx, HMENU hmenu, mpv_node *node);
 void update_menu(plugin_ctx *ctx, mpv_node *node);
 void show_menu(plugin_ctx *ctx, POINT *pt);

src/mpv/misc/dispatch.h

@@ -5,6 +5,7 @@
 
 typedef void (*mp_dispatch_fn)(void *data);
 struct mp_dispatch_queue;
+typedef struct mp_dispatch_queue mp_dispatch_queue;
 
 struct mp_dispatch_queue *mp_dispatch_create(void *talloc_parent);
 void mp_dispatch_set_wakeup_fn(struct mp_dispatch_queue *queue,