diff options
Diffstat (limited to 'lua/mason-core/functional/data.lua')
| -rw-r--r-- | lua/mason-core/functional/data.lua | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/lua/mason-core/functional/data.lua b/lua/mason-core/functional/data.lua new file mode 100644 index 00000000..da6f1efd --- /dev/null +++ b/lua/mason-core/functional/data.lua @@ -0,0 +1,30 @@ +local _ = {} + +_.table_pack = function(...) + return { n = select("#", ...), ... } +end + +---@generic T : string +---@param values T[] +---@return table<T, T> +_.enum = function(values) + local result = {} + for i = 1, #values do + local v = values[i] + result[v] = v + end + return result +end + +---@generic T +---@param list T[] +---@return table<T, boolean> +_.set_of = function(list) + local set = {} + for i = 1, #list do + set[list[i]] = true + end + return set +end + +return _ |
