mirror of
https://github.com/ksyasuda/SubMiner.git
synced 2026-02-27 18:22:41 -08:00
437 lines
13 KiB
Lua
437 lines
13 KiB
Lua
local M = {}
|
|
|
|
local DEFAULT_HOVER_BASE_COLOR = "FFFFFF"
|
|
local DEFAULT_HOVER_COLOR = "C6A0F6"
|
|
|
|
function M.create(ctx)
|
|
local mp = ctx.mp
|
|
local msg = ctx.msg
|
|
local utils = ctx.utils
|
|
local state = ctx.state
|
|
|
|
local function to_hex_color(input)
|
|
if type(input) ~= "string" then
|
|
return nil
|
|
end
|
|
|
|
local hex = input:gsub("[%#%']", ""):gsub("^0x", "")
|
|
if #hex ~= 6 and #hex ~= 3 then
|
|
return nil
|
|
end
|
|
if #hex == 3 then
|
|
return hex:sub(1, 1) .. hex:sub(1, 1) .. hex:sub(2, 2) .. hex:sub(2, 2) .. hex:sub(3, 3) .. hex:sub(3, 3)
|
|
end
|
|
return hex
|
|
end
|
|
|
|
local function fix_ass_color(input, fallback)
|
|
local hex = to_hex_color(input)
|
|
if not hex then
|
|
return fallback or DEFAULT_HOVER_BASE_COLOR
|
|
end
|
|
local r, g, b = hex:sub(1, 2), hex:sub(3, 4), hex:sub(5, 6)
|
|
return b .. g .. r
|
|
end
|
|
|
|
local function escape_ass_text(text)
|
|
return (text or ""):gsub("\\", "\\\\"):gsub("{", "\\{"):gsub("}", "\\}"):gsub("\n", "\\N")
|
|
end
|
|
|
|
local function resolve_osd_dimensions()
|
|
local width = mp.get_property_number("osd-width", 0) or 0
|
|
local height = mp.get_property_number("osd-height", 0) or 0
|
|
|
|
if width <= 0 or height <= 0 then
|
|
local osd_dims = mp.get_property_native("osd-dimensions")
|
|
if type(osd_dims) == "table" and type(osd_dims.w) == "number" and osd_dims.w > 0 then
|
|
width = osd_dims.w
|
|
end
|
|
if type(osd_dims) == "table" and type(osd_dims.h) == "number" and osd_dims.h > 0 then
|
|
height = osd_dims.h
|
|
end
|
|
end
|
|
|
|
if width <= 0 then
|
|
width = 1280
|
|
end
|
|
if height <= 0 then
|
|
height = 720
|
|
end
|
|
|
|
return width, height
|
|
end
|
|
|
|
local function resolve_metrics()
|
|
local sub_font_size = mp.get_property_number("sub-font-size", 36) or 36
|
|
local sub_scale = mp.get_property_number("sub-scale", 1) or 1
|
|
local sub_scale_by_window = mp.get_property_bool("sub-scale-by-window", true) == true
|
|
local sub_pos = mp.get_property_number("sub-pos", 100) or 100
|
|
local sub_margin_y = mp.get_property_number("sub-margin-y", 0) or 0
|
|
local sub_font = mp.get_property("sub-font", "sans-serif") or "sans-serif"
|
|
local sub_spacing = mp.get_property_number("sub-spacing", 0) or 0
|
|
local sub_bold = mp.get_property_bool("sub-bold", false) == true
|
|
local sub_italic = mp.get_property_bool("sub-italic", false) == true
|
|
local sub_border_size = mp.get_property_number("sub-border-size", 2) or 2
|
|
local sub_shadow_offset = mp.get_property_number("sub-shadow-offset", 0) or 0
|
|
local osd_w, osd_h = resolve_osd_dimensions()
|
|
local window_scale = 1
|
|
if sub_scale_by_window and osd_h > 0 then
|
|
window_scale = osd_h / 720
|
|
end
|
|
local effective_margin_y = sub_margin_y * window_scale
|
|
|
|
return {
|
|
font_size = sub_font_size * (sub_scale > 0 and sub_scale or 1) * window_scale,
|
|
pos = sub_pos,
|
|
margin_y = effective_margin_y,
|
|
font = sub_font,
|
|
spacing = sub_spacing,
|
|
bold = sub_bold,
|
|
italic = sub_italic,
|
|
border = sub_border_size * window_scale,
|
|
shadow = sub_shadow_offset * window_scale,
|
|
base_color = fix_ass_color(mp.get_property("sub-color"), DEFAULT_HOVER_BASE_COLOR),
|
|
hover_color = fix_ass_color(DEFAULT_HOVER_COLOR, DEFAULT_HOVER_COLOR),
|
|
}
|
|
end
|
|
|
|
local function get_subtitle_ass_property()
|
|
local ass_text = mp.get_property("sub-text/ass")
|
|
if type(ass_text) == "string" and ass_text ~= "" then
|
|
return ass_text
|
|
end
|
|
ass_text = mp.get_property("sub-text-ass")
|
|
if type(ass_text) == "string" and ass_text ~= "" then
|
|
return ass_text
|
|
end
|
|
return nil
|
|
end
|
|
|
|
local function plain_text_and_ass_map(text)
|
|
local plain = {}
|
|
local map = {}
|
|
local plain_len = 0
|
|
local i = 1
|
|
local text_len = #text
|
|
|
|
while i <= text_len do
|
|
local ch = text:sub(i, i)
|
|
if ch == "{" then
|
|
local close = text:find("}", i + 1, true)
|
|
if not close then
|
|
break
|
|
end
|
|
i = close + 1
|
|
elseif ch == "\\" then
|
|
local esc = text:sub(i + 1, i + 1)
|
|
if esc == "N" or esc == "n" then
|
|
plain_len = plain_len + 1
|
|
plain[plain_len] = "\n"
|
|
map[plain_len] = i
|
|
i = i + 2
|
|
elseif esc == "h" then
|
|
plain_len = plain_len + 1
|
|
plain[plain_len] = " "
|
|
map[plain_len] = i
|
|
i = i + 2
|
|
elseif esc == "{" then
|
|
plain_len = plain_len + 1
|
|
plain[plain_len] = "{"
|
|
map[plain_len] = i
|
|
i = i + 2
|
|
elseif esc == "}" then
|
|
plain_len = plain_len + 1
|
|
plain[plain_len] = "}"
|
|
map[plain_len] = i
|
|
i = i + 2
|
|
elseif esc == "\\" then
|
|
plain_len = plain_len + 1
|
|
plain[plain_len] = "\\"
|
|
map[plain_len] = i
|
|
i = i + 2
|
|
else
|
|
local seq_end = i + 1
|
|
while seq_end <= text_len and text:sub(seq_end, seq_end):match("[%a]") do
|
|
seq_end = seq_end + 1
|
|
end
|
|
if text:sub(seq_end, seq_end) == "(" then
|
|
local close = text:find(")", seq_end, true)
|
|
if close then
|
|
i = close + 1
|
|
else
|
|
i = seq_end + 1
|
|
end
|
|
else
|
|
i = seq_end + 1
|
|
end
|
|
end
|
|
else
|
|
plain_len = plain_len + 1
|
|
plain[plain_len] = ch
|
|
map[plain_len] = i
|
|
i = i + 1
|
|
end
|
|
end
|
|
|
|
return table.concat(plain), map
|
|
end
|
|
|
|
local function find_hover_span(payload, plain)
|
|
local source_len = #plain
|
|
local cursor = 1
|
|
for _, token in ipairs(payload.tokens or {}) do
|
|
if type(token) ~= "table" or type(token.text) ~= "string" or token.text == "" then
|
|
goto continue
|
|
end
|
|
|
|
local token_text = token.text
|
|
local start_pos = nil
|
|
local end_pos = nil
|
|
|
|
if type(token.startPos) == "number" and type(token.endPos) == "number" then
|
|
if token.startPos >= 0 and token.endPos >= token.startPos then
|
|
local candidate_start = token.startPos + 1
|
|
local candidate_stop = token.endPos
|
|
if candidate_start >= 1 and candidate_stop <= source_len and candidate_stop >= candidate_start and plain:sub(candidate_start, candidate_stop) == token_text then
|
|
start_pos = candidate_start
|
|
end_pos = candidate_stop
|
|
end
|
|
end
|
|
end
|
|
|
|
if not start_pos or not end_pos then
|
|
local fallback_start, fallback_stop = plain:find(token_text, cursor, true)
|
|
if not fallback_start then
|
|
fallback_start, fallback_stop = plain:find(token_text, 1, true)
|
|
end
|
|
start_pos, end_pos = fallback_start, fallback_stop
|
|
end
|
|
|
|
if start_pos and end_pos then
|
|
if token.index == payload.hoveredTokenIndex then
|
|
return start_pos, end_pos
|
|
end
|
|
cursor = end_pos + 1
|
|
end
|
|
|
|
::continue::
|
|
end
|
|
|
|
return nil
|
|
end
|
|
|
|
local function inject_hover_color_to_ass(raw_ass, plain_map, hover_start, hover_end, hover_color, base_color)
|
|
if hover_start == nil or hover_end == nil then
|
|
return raw_ass
|
|
end
|
|
|
|
local raw_open_idx = plain_map[hover_start] or 1
|
|
local raw_close_idx = plain_map[hover_end + 1] or (#raw_ass + 1)
|
|
if raw_open_idx < 1 then
|
|
raw_open_idx = 1
|
|
end
|
|
if raw_close_idx < 1 then
|
|
raw_close_idx = 1
|
|
end
|
|
if raw_open_idx > #raw_ass + 1 then
|
|
raw_open_idx = #raw_ass + 1
|
|
end
|
|
if raw_close_idx > #raw_ass + 1 then
|
|
raw_close_idx = #raw_ass + 1
|
|
end
|
|
|
|
local open_tag = string.format("{\\1c&H%s&}", hover_color)
|
|
local close_tag = string.format("{\\1c&H%s&}", base_color)
|
|
local changes = {
|
|
{ idx = raw_open_idx, tag = open_tag },
|
|
{ idx = raw_close_idx, tag = close_tag },
|
|
}
|
|
table.sort(changes, function(a, b)
|
|
return a.idx < b.idx
|
|
end)
|
|
|
|
local output = {}
|
|
local cursor = 1
|
|
for _, change in ipairs(changes) do
|
|
if change.idx > #raw_ass + 1 then
|
|
change.idx = #raw_ass + 1
|
|
end
|
|
if change.idx < 1 then
|
|
change.idx = 1
|
|
end
|
|
if change.idx > cursor then
|
|
output[#output + 1] = raw_ass:sub(cursor, change.idx - 1)
|
|
end
|
|
output[#output + 1] = change.tag
|
|
cursor = change.idx
|
|
end
|
|
if cursor <= #raw_ass then
|
|
output[#output + 1] = raw_ass:sub(cursor)
|
|
end
|
|
|
|
return table.concat(output)
|
|
end
|
|
|
|
local function build_hover_subtitle_content(payload)
|
|
local source_ass = get_subtitle_ass_property()
|
|
if type(source_ass) == "string" and source_ass ~= "" then
|
|
state.hover_highlight.cached_ass = source_ass
|
|
else
|
|
source_ass = state.hover_highlight.cached_ass
|
|
end
|
|
if type(source_ass) ~= "string" or source_ass == "" then
|
|
return nil
|
|
end
|
|
|
|
local plain_source, plain_map = plain_text_and_ass_map(source_ass)
|
|
if type(plain_source) ~= "string" or plain_source == "" then
|
|
return nil
|
|
end
|
|
|
|
local hover_start, hover_end = find_hover_span(payload, plain_source)
|
|
if not hover_start or not hover_end then
|
|
return nil
|
|
end
|
|
|
|
local metrics = resolve_metrics()
|
|
local hover_color = fix_ass_color(payload.colors and payload.colors.hover or nil, metrics.hover_color)
|
|
local base_color = fix_ass_color(payload.colors and payload.colors.base or nil, metrics.base_color)
|
|
return inject_hover_color_to_ass(source_ass, plain_map, hover_start, hover_end, hover_color, base_color)
|
|
end
|
|
|
|
local function clear_hover_overlay()
|
|
if state.hover_highlight.clear_timer then
|
|
state.hover_highlight.clear_timer:kill()
|
|
state.hover_highlight.clear_timer = nil
|
|
end
|
|
if state.hover_highlight.overlay_active then
|
|
if type(state.hover_highlight.saved_sub_visibility) == "string" then
|
|
mp.set_property("sub-visibility", state.hover_highlight.saved_sub_visibility)
|
|
else
|
|
mp.set_property("sub-visibility", "yes")
|
|
end
|
|
if type(state.hover_highlight.saved_secondary_sub_visibility) == "string" then
|
|
mp.set_property("secondary-sub-visibility", state.hover_highlight.saved_secondary_sub_visibility)
|
|
end
|
|
state.hover_highlight.saved_sub_visibility = nil
|
|
state.hover_highlight.saved_secondary_sub_visibility = nil
|
|
state.hover_highlight.overlay_active = false
|
|
end
|
|
mp.set_osd_ass(0, 0, "")
|
|
state.hover_highlight.payload = nil
|
|
state.hover_highlight.revision = -1
|
|
state.hover_highlight.cached_ass = nil
|
|
state.hover_highlight.last_hover_update_ts = 0
|
|
end
|
|
|
|
local function schedule_hover_clear(delay_seconds)
|
|
if state.hover_highlight.clear_timer then
|
|
state.hover_highlight.clear_timer:kill()
|
|
state.hover_highlight.clear_timer = nil
|
|
end
|
|
state.hover_highlight.clear_timer = mp.add_timeout(delay_seconds or 0.08, function()
|
|
state.hover_highlight.clear_timer = nil
|
|
clear_hover_overlay()
|
|
end)
|
|
end
|
|
|
|
local function render_hover_overlay(payload)
|
|
if not payload or payload.hoveredTokenIndex == nil or payload.subtitle == nil then
|
|
clear_hover_overlay()
|
|
return
|
|
end
|
|
|
|
local ass = build_hover_subtitle_content(payload)
|
|
if not ass then
|
|
return
|
|
end
|
|
|
|
local osd_w, osd_h = resolve_osd_dimensions()
|
|
local metrics = resolve_metrics()
|
|
local osd_dims = mp.get_property_native("osd-dimensions")
|
|
local ml = (type(osd_dims) == "table" and type(osd_dims.ml) == "number") and osd_dims.ml or 0
|
|
local mr = (type(osd_dims) == "table" and type(osd_dims.mr) == "number") and osd_dims.mr or 0
|
|
local mt = (type(osd_dims) == "table" and type(osd_dims.mt) == "number") and osd_dims.mt or 0
|
|
local mb = (type(osd_dims) == "table" and type(osd_dims.mb) == "number") and osd_dims.mb or 0
|
|
local usable_w = math.max(1, osd_w - ml - mr)
|
|
local usable_h = math.max(1, osd_h - mt - mb)
|
|
local anchor_x = math.floor(ml + usable_w / 2)
|
|
local baseline_adjust = (metrics.border + metrics.shadow) * 5
|
|
local anchor_y = math.floor(mt + (usable_h * metrics.pos / 100) - metrics.margin_y + baseline_adjust)
|
|
local font_size = math.max(8, metrics.font_size)
|
|
local anchor_tag = string.format(
|
|
"{\\an2\\q2\\pos(%d,%d)\\fn%s\\fs%g\\b%d\\i%d\\fsp%g\\bord%g\\shad%g\\1c&H%s&}",
|
|
anchor_x,
|
|
anchor_y,
|
|
escape_ass_text(metrics.font),
|
|
font_size,
|
|
metrics.bold and 1 or 0,
|
|
metrics.italic and 1 or 0,
|
|
metrics.spacing,
|
|
metrics.border,
|
|
metrics.shadow,
|
|
metrics.base_color
|
|
)
|
|
if not state.hover_highlight.overlay_active then
|
|
state.hover_highlight.saved_sub_visibility = mp.get_property("sub-visibility")
|
|
state.hover_highlight.saved_secondary_sub_visibility = mp.get_property("secondary-sub-visibility")
|
|
mp.set_property("sub-visibility", "no")
|
|
mp.set_property("secondary-sub-visibility", "no")
|
|
state.hover_highlight.overlay_active = true
|
|
end
|
|
mp.set_osd_ass(osd_w, osd_h, anchor_tag .. ass)
|
|
end
|
|
|
|
local function handle_hover_message(payload_json)
|
|
local parsed, parse_error = utils.parse_json(payload_json)
|
|
if not parsed then
|
|
msg.warn("Invalid hover-highlight payload: " .. tostring(parse_error))
|
|
clear_hover_overlay()
|
|
return
|
|
end
|
|
|
|
if type(parsed.revision) ~= "number" then
|
|
clear_hover_overlay()
|
|
return
|
|
end
|
|
|
|
if parsed.revision < state.hover_highlight.revision then
|
|
return
|
|
end
|
|
|
|
if type(parsed.hoveredTokenIndex) == "number" and type(parsed.tokens) == "table" then
|
|
if state.hover_highlight.clear_timer then
|
|
state.hover_highlight.clear_timer:kill()
|
|
state.hover_highlight.clear_timer = nil
|
|
end
|
|
state.hover_highlight.revision = parsed.revision
|
|
state.hover_highlight.payload = parsed
|
|
state.hover_highlight.last_hover_update_ts = mp.get_time() or 0
|
|
render_hover_overlay(state.hover_highlight.payload)
|
|
return
|
|
end
|
|
|
|
local now = mp.get_time() or 0
|
|
local elapsed_since_hover = now - (state.hover_highlight.last_hover_update_ts or 0)
|
|
state.hover_highlight.revision = parsed.revision
|
|
state.hover_highlight.payload = nil
|
|
if state.hover_highlight.overlay_active then
|
|
if elapsed_since_hover > 0.35 then
|
|
return
|
|
end
|
|
schedule_hover_clear(0.08)
|
|
else
|
|
clear_hover_overlay()
|
|
end
|
|
end
|
|
|
|
return {
|
|
HOVER_MESSAGE_NAME = "subminer-hover-token",
|
|
HOVER_MESSAGE_NAME_LEGACY = "yomipv-hover-token",
|
|
handle_hover_message = handle_hover_message,
|
|
clear_hover_overlay = clear_hover_overlay,
|
|
}
|
|
end
|
|
|
|
return M
|