autocomplete.lua

local completion = require "cc.shell.completion"

---@class Autocomplete
local Autocomplete = {}

Autocomplete.structure = nil

---@param structure table
---@param name string | nil
---@return table
function Autocomplete:registerAutoComplete(structure, name)
    self.structure = structure
    shell.setCompletionFunction(
        name or shell.getRunningProgram(),
        function(shell, index, argument, previous)
            local commands = {}
            for k, _ in pairs(self.structure) do
                commands[k] = self.structure[k] or {}
            end

            local currArg = commands
            for i = 2, #previous do
                if currArg[previous[i]] then
                    currArg = currArg[previous[i]]
                else
                    return nil
                end
            end

            local results = {}
            for word, value in pairs(currArg) do
                if type(value) ~= "table" then
                    word = value
                end
                if word:sub(1, #argument) == argument then
                    results[#results + 1] = word:sub(#argument + 1)
                end
            end
            return results
        end
    )
end

return Autocomplete