aboutsummaryrefslogtreecommitdiffstats
path: root/lua/mason-core/functional/logic.lua
blob: 0e0044d5d4562128875e974e0c6ed11ba0d1e606 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
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 _