feat: live grep

lua/fzfx.lua

@@ -42,18 +42,42 @@ local Defaults = {
         },
         action = {
             unrestricted_switch = "ctrl-u",
-            restricted = "ctrl-r",
-            unrestricted = "ctrl-u",
         },
     },
     live_grep = {
+        command = {
+            normal = {
+                name = "FzfxLiveGrep",
+                desc = "Live grep",
+            },
+            unrestricted = {
+                name = "FzfxLiveGrepU",
+                desc = "Live grep unrestrictly",
+            },
+            visual = {
+                name = "FzfxLiveGrepV",
+                desc = "Live grep by visual select",
+            },
+            unrestricted_visual = {
+                name = "FzfxLiveGrepUV",
+                desc = "Live grep unrestrictly by visual select",
+            },
+            cword = {
+                name = "FzfxLiveGrepW",
+                desc = "Live grep by cursor word",
+            },
+            unrestricted_cword = {
+                name = "FzfxLiveGrepUW",
+                desc = "Live grep unrestrictly by cursor word",
+            },
+        },
         provider = {
             restricted = default_rg_command,
             unrestricted = default_rg_command .. " -uu",
         },
         action = {
-            fzf_mode = "ctrl-f",
-            rg_mode = "ctrl-r",
+            fuzzy_switch = "ctrl-f",
+            unrestricted_switch = "ctrl-u",
         },
     },
     debug = {

lua/fzfx/files.lua

@@ -238,7 +238,6 @@ local function setup(files_configs)
 end
 
 local M = {
-    files = files,
     setup = setup,
 }
 

lua/fzfx/live_grep.lua

@@ -0,0 +1,264 @@
+local log = require("fzfx.log")
+local utils = require("fzfx.utils")
+local path = require("fzfx.path")
+local legacy = require("fzfx.legacy")
+
+--- @type table<string, string|nil>
+local Runtime = {
+    --- @type string|nil
+    current_fuzzy_header = nil,
+    --- @type string|nil
+    next_fuzzy_header = nil,
+    --- @type string|nil
+    swap_fuzzy_header = nil,
+    --- @type string|nil
+    current_unrestricted_header = nil,
+    --- @type string|nil
+    next_unrestricted_header = nil,
+    --- @type string|nil
+    swap_unrestricted_header = nil,
+    --- @type string|nil
+    current_provider = nil,
+    --- @type string|nil
+    next_provider = nil,
+    --- @type string|nil
+    swap_provider = nil,
+}
+
+--- @param query string
+--- @param fullscreen boolean|integer
+--- @param opts Config
+local function live_grep(query, fullscreen, opts)
+    local switch_action = string.lower(opts.action.unrestricted_switch)
+
+    -- header
+    vim.fn.writefile(
+        {
+            string.format(
+                ":: Press %s to restricted mode",
+                legacy.magenta(string.upper(switch_action))
+            ),
+        },
+        opts.unrestricted and Runtime.current_fuzzy_header
+            or Runtime.next_fuzzy_header
+    )
+    vim.fn.writefile(
+        {
+            string.format(
+                ":: Press %s to unrestricted mode",
+                legacy.magenta(string.upper(switch_action))
+            ),
+        },
+        opts.unrestricted and Runtime.next_fuzzy_header
+            or Runtime.current_fuzzy_header
+    )
+    vim.fn.writefile({
+        string.format(
+            ":: Press %s to fzf mode",
+            legacy.magenta(string.upper(fuzzy_switch_action))
+        ),
+    })
+
+    -- provider
+    vim.fn.writefile(
+        { opts.provider.unrestricted },
+        opts.unrestricted and Runtime.current_provider or Runtime.next_provider
+    )
+    vim.fn.writefile(
+        { opts.provider.restricted },
+        opts.unrestricted and Runtime.next_provider or Runtime.current_provider
+    )
+
+    -- query command, both initial query + reload query
+    local query_command = string.format(
+        "nvim --headless -l %sfiles_provider.lua || true",
+        path.plugin_bin()
+    )
+    log.debug(
+        "|fzfx.live_grep - files| query_command:%s",
+        vim.inspect(query_command)
+    )
+
+    local spec = {
+        source = query_command,
+        options = {
+            "--ansi",
+            "--query",
+            query,
+            "--bind",
+            -- unrestricted switch action: swap header, swap provider, then change header + reload
+            string.format(
+                "%s:unbind(%s)+execute-silent(mv %s %s && mv %s %s && mv %s %s)+execute-silent(mv %s %s && mv %s %s && mv %s %s)+rebind(%s)+transform-header(cat %s)+reload(%s)",
+                switch_action,
+                switch_action,
+                Runtime.current_fuzzy_header,
+                Runtime.swap_fuzzy_header,
+                Runtime.next_fuzzy_header,
+                Runtime.current_fuzzy_header,
+                Runtime.swap_fuzzy_header,
+                Runtime.next_fuzzy_header,
+                Runtime.current_provider,
+                Runtime.swap_provider,
+                Runtime.next_provider,
+                Runtime.current_provider,
+                Runtime.swap_provider,
+                Runtime.next_provider,
+                switch_action,
+                Runtime.current_fuzzy_header,
+                query_command
+            ),
+            "--header",
+            opts.unrestricted and string.format(
+                ":: Press %s to restricted mode",
+                legacy.magenta(string.upper(switch_action))
+            ) or string.format(
+                ":: Press %s to unrestricted mode",
+                legacy.magenta(string.upper(switch_action))
+            ),
+        },
+    }
+    spec = vim.fn["fzf#vim#with_preview"](spec)
+    log.debug("|fzfx.live_grep - files| spec:%s", vim.inspect(spec))
+    return vim.fn["fzf#vim#files"]("", spec, fullscreen)
+end
+
+--- @param live_grep_configs Config
+local function setup(live_grep_configs)
+    log.debug(
+        "|fzfx.live_grep - setup| plugin_bin:%s, files_configs:%s",
+        vim.inspect(path.plugin_bin()),
+        vim.inspect(live_grep_configs)
+    )
+    local restricted_opts = vim.tbl_deep_extend(
+        "force",
+        vim.deepcopy(live_grep_configs),
+        { unrestricted = false }
+    )
+    local unrestricted_opts = vim.tbl_deep_extend(
+        "force",
+        vim.deepcopy(live_grep_configs),
+        { unrestricted = true }
+    )
+
+    -- user commands opts
+    local normal_command_opts = {
+        bang = true,
+        nargs = "?",
+        complete = "dir",
+    }
+    local visual_command_opts = {
+        bang = true,
+        range = true,
+    }
+    local cword_command_opts = {
+        bang = true,
+    }
+
+    -- FzfxFiles
+    utils.define_command(live_grep_configs.command.normal, function(opts)
+        log.debug(
+            "|fzfx.live_grep - setup| normal command opts:%s",
+            vim.inspect(opts)
+        )
+        return files(opts.args, opts.bang, restricted_opts)
+    end, normal_command_opts)
+    -- FzfxFilesU
+    utils.define_command(live_grep_configs.command.unrestricted, function(opts)
+        log.debug(
+            "|fzfx.live_grep - setup| unrestricted command opts:%s",
+            vim.inspect(opts)
+        )
+        return files(opts.args, opts.bang, unrestricted_opts)
+    end, normal_command_opts)
+    -- FzfxFilesV
+    utils.define_command(live_grep_configs.command.visual, function(opts)
+        local visual_select = utils.visual_select()
+        log.debug(
+            "|fzfx.live_grep - setup| visual command select:%s, opts:%s",
+            vim.inspect(visual_select),
+            vim.inspect(opts)
+        )
+        return files(visual_select, opts.bang, restricted_opts)
+    end, visual_command_opts)
+    -- FzfxFilesUV
+    utils.define_command(
+        live_grep_configs.command.unrestricted_visual,
+        function(opts)
+            local visual_select = utils.visual_select()
+            log.debug(
+                "|fzfx.live_grep - setup| visual command select:%s, opts:%s",
+                vim.inspect(visual_select),
+                vim.inspect(opts)
+            )
+            return files(visual_select, opts.bang, unrestricted_opts)
+        end,
+        visual_command_opts
+    )
+    -- FzfxFilesW
+    utils.define_command(live_grep_configs.command.cword, function(opts)
+        local word = vim.fn.expand("<cword>")
+        log.debug(
+            "|fzfx.live_grep - setup| cword command word:%s, opts:%s",
+            vim.inspect(word),
+            vim.inspect(opts)
+        )
+        return files(word, opts.bang, restricted_opts)
+    end, cword_command_opts)
+    -- FzfxFilesUW
+    utils.define_command(
+        live_grep_configs.command.unrestricted_cword,
+        function(opts)
+            local word = vim.fn.expand("<cword>")
+            log.debug(
+                "|fzfx.live_grep - setup| cword command word:%s, opts:%s",
+                vim.inspect(word),
+                vim.inspect(opts)
+            )
+            return files(word, opts.bang, unrestricted_opts)
+        end,
+        cword_command_opts
+    )
+
+    -- runtime
+    if live_grep_configs.debug then
+        Runtime.current_fuzzy_header = string.format(
+            "%s/fzfx.nvim/files_current_header",
+            vim.fn.stdpath("data")
+        )
+        Runtime.next_fuzzy_header = string.format(
+            "%s/fzfx.nvim/files_next_header",
+            vim.fn.stdpath("data")
+        )
+        Runtime.swap_fuzzy_header = string.format(
+            "%s/fzfx.nvim/files_swap_header",
+            vim.fn.stdpath("data")
+        )
+        Runtime.current_provider = string.format(
+            "%s/fzfx.nvim/files_current_provider",
+            vim.fn.stdpath("data")
+        )
+        Runtime.next_provider = string.format(
+            "%s/fzfx.nvim/files_next_provider",
+            vim.fn.stdpath("data")
+        )
+        Runtime.swap_provider = string.format(
+            "%s/fzfx.nvim/files_swap_provider",
+            vim.fn.stdpath("data")
+        )
+    else
+        Runtime.current_fuzzy_header = path.tempname()
+        Runtime.next_fuzzy_header = path.tempname()
+        Runtime.swap_fuzzy_header = path.tempname()
+        Runtime.current_provider = path.tempname()
+        Runtime.next_provider = path.tempname()
+        Runtime.swap_provider = path.tempname()
+    end
+    vim.env._FZFX_FILES_PROVIDER = Runtime.current_provider
+    log.debug("|fzfx.live_grep - setup| Runtime:%s", vim.inspect(Runtime))
+end
+
+local M = {
+    setup = setup,
+}
+
+return M