feat(filter): add 'primary' filter (#21)

feat: MPV done

Author: linrongbin16

lua/colorbox.lua

@@ -2,6 +2,17 @@ local logger = require("colorbox.logger")
 local LogLevels = require("colorbox.logger").LogLevels
 local json = require("colorbox.json")
 
+--- @param l any[]
+--- @param f fun(v:any):number
+--- @return number
+local function minimal_integer(l, f)
+    local result = 2 ^ 31 - 1
+    for _, i in ipairs(l) do
+        result = math.min(result, f(i))
+    end
+    return result
+end
+
 local function randint()
     local s1 = vim.loop.getpid()
     local s2, s3 = vim.loop.gettimeofday()
@@ -52,28 +63,19 @@ local function string_find(s, t, start)
     return nil
 end
 
---- @enum colorbox.PolicyConfigEnum
-local PolicyConfigEnum = {
-    SHUFFLE = "shuffle",
-    INORDER = "inorder",
-    SINGLE = "single",
-}
-
---- @enum colorbox.TimingConfigEnum
-local TimingConfigEnum = {
-    STARTUP = "startup",
-    INTERVAL = "interval",
-    FILETYPE = "filetype",
-}
+--- @enum
 
 --- @alias colorbox.Options table<any, any>
 --- @type colorbox.Options
 local Defaults = {
-    policy = PolicyConfigEnum.SHUFFLE,
+    --- @type "shuffle"|"inorder"|"single"
+    policy = "shuffle",
 
-    timing = TimingConfigEnum.STARTUP,
+    --- @type "startup"|"interval"|"filetype"
+    timing = "startup",
 
-    filter = function() end,
+    --- @type "primary"|fun(color:string):boolean|nil
+    filter = nil,
 
     setup = {
         ["projekt0n/github-nvim-theme"] = function()
@@ -175,6 +177,36 @@ local function _load_repo_meta_urls_map(cwd)
     return repos
 end
 
+--- @param color string
+--- @param spec colorbox.ColorSpec
+--- @return boolean
+local function _should_filter(color, spec)
+    if Configs.filter == nil then
+        return false
+    end
+    if Configs.filter == "primary" then
+        local unique = #spec.colors <= 1
+        local variants = spec.colors
+        local shortest = string.len(color)
+            == minimal_integer(variants, string.len)
+        local url_splits =
+            vim.split(spec.url, "/", { plain = true, trimempty = true })
+        local matched = url_splits[1]:lower() == color:lower()
+            or url_splits[2]:lower() == color:lower()
+        logger.debug(
+            "|colorbox._should_filter| color:%s, spec:%s, unique:%s, shortest: %s, matched:%s",
+            vim.inspect(color),
+            vim.inspect(spec),
+            vim.inspect(unique),
+            vim.inspect(shortest),
+            vim.inspect(matched)
+        )
+        return not unique and not shortest and not matched
+    elseif type(Configs.filter) == "function" then
+        return Configs.filter(color)
+    end
+end
+
 local function _init()
     local cwd = vim.fn["colorbox#base_dir"]()
     local packstart = string.format("%s/pack/colorbox/start", cwd)
@@ -217,6 +249,7 @@ local function _init()
                     vim.inspect(err)
                 )
             end
+            local color_candidates = {}
             while true do
                 local color_tmp = color_dir:readdir()
                 if type(color_tmp) == "table" and #color_tmp > 0 then
@@ -239,24 +272,32 @@ local function _init()
                             local color = color_file:sub(1, #color_file - 4)
                             table.insert(spec.colors, color)
                             ColorNamesMap[color] = spec
-                            table.insert(ColorNames, color)
+                            table.insert(color_candidates, color)
                         end
                     end
                 else
                     break
                 end
             end
+            for _, c in ipairs(color_candidates) do
+                if not _should_filter(c, spec) then
+                    table.insert(ColorNames, c)
+                end
+            end
         end
     end
+    logger.debug("|colorbox._init| ColorNames:%s", vim.inspect(ColorNames))
 end
 
 local function _policy_shuffle()
-    local n = 0
-    for _, spec in pairs(ColorSpecs) do
-        n = n + #spec.colors
-    end
-    local r = math.floor(math.fmod(randint(), n))
+    local r = math.floor(math.fmod(randint(), #ColorNames))
     local color = ColorNames[r + 1]
+    -- logger.debug(
+    --     "|colorbox._policy_shuffle| color:%s, ColorNames:%s (%d), r:%d",
+    --     vim.inspect(color),
+    --     vim.inspect(ColorNames),
+    --     vim.inspect()
+    -- )
     vim.cmd(string.format([[color %s]], color))
 end
 
@@ -293,13 +334,12 @@ local function setup(opts)
 
     vim.api.nvim_create_autocmd("ColorSchemePre", {
         callback = function(event)
-            logger.debug("|colorbox.init| event:%s", vim.inspect(event))
+            logger.debug("|colorbox.setup| event:%s", vim.inspect(event))
             if type(event) ~= "table" or ColorNamesMap[event.match] == nil then
                 return
             end
             local spec = ColorNamesMap[event.match]
             vim.cmd(string.format([[packadd %s]], spec.name))
-
             if
                 type(Configs.setup) == "table"
                 and type(Configs.setup[spec.url]) == "function"
@@ -342,16 +382,16 @@ local function update()
 
         local function _on_output(chanid, data, name)
             if type(data) == "table" then
+                logger.debug("%s: %s", vim.inspect(name), vim.inspect(data))
+                local lines = {}
                 for _, line in ipairs(data) do
-                    if string.len(line) > 0 then
-                        logger.debug(
-                            "(%s) %s: %s",
-                            vim.inspect(name),
-                            vim.inspect(url),
-                            vim.inspect(line)
-                        )
+                    if string.len(vim.trim(line)) > 0 then
+                        table.insert(lines, line)
                     end
                 end
+                if #lines > 0 then
+                    logger.info(table.concat(lines, ""))
+                end
             end
         end
 
@@ -367,7 +407,6 @@ local function update()
                 on_stdout = _on_output,
                 on_stderr = _on_output,
             })
-            logger.info("updating %s", vim.inspect(url))
             table.insert(jobs, jobid)
         else
             local cmd = string.format(
@@ -383,7 +422,7 @@ local function update()
                 on_stdout = _on_output,
                 on_stderr = _on_output,
             })
-            logger.info("installing %s", vim.inspect(url))
+            logger.debug("installing %s", vim.inspect(url))
             table.insert(jobs, jobid)
         end
     end