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
|
class Proc
def ===(*args)
call(*args)
end
def yield(*args)
call(*args)
end
def to_proc
self
end
def curry(arity=self.arity)
type = :proc
abs = lambda {|a| a < 0 ? -a - 1 : a}
arity = abs[arity]
if lambda?
type = :lambda
self_arity = self.arity
if (self_arity >= 0 && arity != self_arity) ||
(self_arity < 0 && abs[self_arity] > arity)
raise ArgumentError, "wrong number of arguments (given #{arity}, expected #{abs[self_arity]})"
end
end
pproc = self
make_curry = proc do |given_args=[]|
__send__(type) do |*args|
new_args = given_args + args
if new_args.size >= arity
pproc[*new_args]
else
make_curry[new_args]
end
end
end
make_curry.call
end
def <<(other)
->(*args, **opts, &block) { call(other.call(*args, **opts, &block)) }
end
def >>(other)
->(*args, **opts, &block) { other.call(call(*args, **opts, &block)) }
end
end
|