1 files changed, 225 insertions, 0 deletions
diff --git a/mpv/scripts/file-browser/modules/ass.lua b/mpv/scripts/file-browser/modules/ass.lua
new file mode 100644
index 0000000..a64a47d
--- /dev/null
+++ b/mpv/scripts/file-browser/modules/ass.lua
@@ -0,0 +1,225 @@
+--------------------------------------------------------------------------------------------------------
+-----------------------------------------List Formatting------------------------------------------------
+--------------------------------------------------------------------------------------------------------
+--------------------------------------------------------------------------------------------------------
+
+local g = require 'modules.globals'
+local o = require 'modules.options'
+local fb_utils = require 'modules.utils'
+
+local state = g.state
+local style = g.style
+local ass = g.ass
+
+local function draw()
+    ass:update()
+end
+
+local function remove()
+    ass:remove()
+end
+
+---@type string[]
+local string_buffer = {}
+
+---appends the entered text to the overlay
+---@param ... string
+local function append(...)
+    for i = 1, select("#", ...) do
+        table.insert(string_buffer, select(i, ...) or '' )
+    end
+end
+
+--appends a newline character to the osd
+local function newline()
+    table.insert(string_buffer, '\\N')
+end
+
+local function flush_buffer()
+    ass.data = table.concat(string_buffer, '')
+    string_buffer = {}
+end
+
+---detects whether or not to highlight the given entry as being played
+---@param v Item
+---@return boolean
+local function highlight_entry(v)
+    if g.current_file.path == nil then return false end
+    local full_path = fb_utils.get_full_path(v)
+    local alt_path = v.name and g.state.directory..v.name or nil
+
+    if fb_utils.parseable_item(v) then
+        return (
+            string.find(g.current_file.directory, full_path, 1, true)
+            or (alt_path and string.find(g.current_file.directory, alt_path, 1, true))
+        ) ~= nil
+    else
+        return g.current_file.path == full_path
+            or (alt_path and g.current_file.path == alt_path)
+    end
+end
+
+---escape ass values and replace newlines
+---@param str string
+---@param style_reset string?
+---@return string
+local function ass_escape(str, style_reset)
+    return fb_utils.ass_escape(str, style_reset and style.warning..'␊'..style_reset or true)
+end
+
+local header_overrides = {['^'] = style.header}
+
+---@return number start
+---@return number finish
+---@return boolean is_overflowing
+local function calculate_view_window()
+    ---@type number
+    local start = 1
+    ---@type number
+    local finish = start+o.num_entries-1
+
+    --handling cursor positioning
+    local mid = math.ceil(o.num_entries/2)+1
+    if state.selected+mid > finish then
+        ---@type number
+        local offset = state.selected - finish + mid
+
+        --if we've overshot the end of the list then undo some of the offset
+        if finish + offset > #state.list then
+            offset = offset - ((finish+offset) - #state.list)
+        end
+
+        start = start + offset
+        finish = finish + offset
+    end
+
+    --making sure that we don't overstep the boundaries
+    if start < 1 then start = 1 end
+    local overflow = finish < #state.list
+    --this is necessary when the number of items in the dir is less than the max
+    if not overflow then finish = #state.list end
+
+    return start, finish, overflow
+end
+
+---@param i number index
+---@return string
+local function calculate_item_style(i)
+    local is_playing_file = highlight_entry(state.list[i])
+
+    --sets the selection colour scheme
+    local multiselected = state.selection[i]
+
+    --sets the colour for the item
+    local item_style = style.body
+
+    if multiselected then item_style = item_style..style.multiselect
+    elseif i == state.selected then item_style = item_style..style.selected end
+
+    if is_playing_file then item_style = item_style..(multiselected and style.playing_selected or style.playing) end
+
+    return item_style
+end
+
+local function draw_header()
+    append(style.header)
+    append(fb_utils.substitute_codes(o.format_string_header, header_overrides, nil, nil, function(str, code)
+        if code == '^' then return str end
+        return ass_escape(str, style.header)
+    end))
+    newline()
+end
+
+---@param wrapper_overrides ReplacerTable
+local function draw_top_wrapper(wrapper_overrides)
+    --adding a header to show there are items above in the list
+    append(style.footer_header)
+    append(fb_utils.substitute_codes(o.format_string_topwrapper, wrapper_overrides, nil, nil, function(str)
+        return ass_escape(str)
+    end))
+    newline()
+end
+
+---@param wrapper_overrides ReplacerTable
+local function draw_bottom_wrapper(wrapper_overrides)
+    append(style.footer_header)
+    append(fb_utils.substitute_codes(o.format_string_bottomwrapper, wrapper_overrides, nil, nil, function(str)
+        return ass_escape(str)
+    end))
+end
+
+---@param i number index
+---@param cursor string
+local function draw_cursor(i, cursor)
+    --handles custom styles for different entries
+    if i == state.selected or i == state.multiselect_start then
+        if not (i == state.selected) then append(style.selection_marker) end
+
+        if not state.multiselect_start then append(style.cursor)
+        else
+            if state.selection[state.multiselect_start] then append(style.cursor_select)
+            else append(style.cursor_deselect) end
+        end
+    else
+        append(g.style.indent)
+    end
+    append(cursor, '\\h', style.body)
+end
+
+--refreshes the ass text using the contents of the list
+local function update_ass()
+    if state.hidden then state.flag_update = true ; return end
+
+    append(style.global)
+    draw_header()
+
+    if #state.list < 1 then
+        append(state.empty_text)
+        flush_buffer()
+        draw()
+        return
+    end
+
+    local start, finish, overflow = calculate_view_window()
+
+    -- these are the number values to place into the wrappers
+    local wrapper_overrides = {['<'] = tostring(start-1), ['>'] = tostring(#state.list-finish)}
+    if o.format_string_topwrapper ~= '' and start > 1 then
+        draw_top_wrapper(wrapper_overrides)
+    end
+
+    for i=start, finish do
+        local v = state.list[i]
+        append(style.body)
+        if g.ALIGN_X ~= 'right' then draw_cursor(i, o.cursor_icon) end
+
+        local item_style = calculate_item_style(i)
+        append(item_style)
+
+        --sets the folder icon
+        if v.type == 'dir' then
+            append(style.folder, o.folder_icon, "\\h", style.body)
+            append(item_style)
+        end
+
+        --adds the actual name of the item
+        append(v.ass or ass_escape(v.label or v.name, item_style), '\\h')
+        if g.ALIGN_X == 'right' then draw_cursor(i, o.cursor_icon_flipped) end
+        newline()
+    end
+
+    if o.format_string_bottomwrapper ~= '' and overflow then
+        draw_bottom_wrapper(wrapper_overrides)
+    end
+
+    flush_buffer()
+    draw()
+end
+
+---@class ass
+return {
+    update_ass = update_ass,
+    highlight_entry = highlight_entry,
+    draw = draw,
+    remove = remove,
+}