aboutsummaryrefslogtreecommitdiffstats
path: root/lua/mason-core/functional/logic.lua
diff options
context:
space:
mode:
Diffstat (limited to 'lua/mason-core/functional/logic.lua')
-rw-r--r--lua/mason-core/functional/logic.lua63
1 files changed, 63 insertions, 0 deletions
diff --git a/lua/mason-core/functional/logic.lua b/lua/mason-core/functional/logic.lua
new file mode 100644
index 00000000..0e0044d5
--- /dev/null
+++ b/lua/mason-core/functional/logic.lua
@@ -0,0 +1,63 @@
+local fun = require "mason-core.functional.function"
+
+local _ = {}
+
+---@generic T
+---@param predicates (fun(item: T): boolean)[]
+---@return fun(item: T): boolean
+_.all_pass = fun.curryN(function(predicates, item)
+ for i = 1, #predicates do
+ if not predicates[i](item) then
+ return false
+ end
+ end
+ return true
+end, 2)
+
+---@generic T
+---@param predicates (fun(item: T): boolean)[]
+---@return fun(item: T): boolean
+_.any_pass = fun.curryN(function(predicates, item)
+ for i = 1, #predicates do
+ if predicates[i](item) then
+ return true
+ end
+ end
+ return false
+end, 2)
+
+---@generic T
+---@param predicate fun(item: T): boolean
+---@param on_true fun(item: T): any
+---@param on_false fun(item: T): any
+---@param value T
+_.if_else = fun.curryN(function(predicate, on_true, on_false, value)
+ if predicate(value) then
+ return on_true(value)
+ else
+ return on_false(value)
+ end
+end, 4)
+
+---@param value boolean
+_.is_not = function(value)
+ return not value
+end
+
+---@generic T
+---@param predicate fun(value: T): boolean
+---@param value T
+_.complement = fun.curryN(function(predicate, value)
+ return not predicate(value)
+end, 2)
+
+_.cond = fun.curryN(function(predicate_transformer_pairs, value)
+ for _, pair in ipairs(predicate_transformer_pairs) do
+ local predicate, transformer = pair[1], pair[2]
+ if predicate(value) then
+ return transformer(value)
+ end
+ end
+end, 2)
+
+return _