Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 16 additions & 13 deletions src/utils.jl
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
export @esc, isexpr, isline, iscall, rmlines, unblock, block, inexpr, namify, isdef,
longdef, shortdef, @expand, makeif, prettify, combinedef, splitdef, splitarg
longdef, shortdef, @expand, makeif, prettify, combinedef, splitdef, splitarg, combinearg

"""
assoc!(d, k, v)
Expand Down Expand Up @@ -406,7 +406,10 @@ end
`combinearg` is the inverse of [`splitarg`](@ref).
"""
function combinearg(arg_name, arg_type, is_splat, default)
a = arg_name===nothing ? :(::$arg_type) : :($arg_name::$arg_type)
@assert arg_name !== nothing || arg_type !== nothing
a = arg_name===nothing ? :(::$arg_type) :
arg_type==:Any ? arg_name :
:($arg_name::$arg_type)
a2 = is_splat ? Expr(:..., a) : a
return default === nothing ? a2 : Expr(:kw, a2, default)
end
Expand All @@ -430,19 +433,19 @@ julia> map(splitarg, (:(f(a=2, x::Int=nothing, y, args...))).args[2:end])
See also: [`combinearg`](@ref)
"""
function splitarg(arg_expr)
splitvar(arg) =
(@match arg begin
::T_ => (nothing, T)
name_::T_ => (name, T)
x_ => (x, :Any)
end)::NTuple{2,Any} # the pattern `x_` matches any expression
(is_splat = @capture(arg_expr, arg_expr2_...)) || (arg_expr2 = arg_expr)
if @capture(arg_expr2, arg_ = default_)
@assert default !== nothing "splitarg cannot handle `nothing` as a default. Use a quoted `nothing` if possible. (MacroTools#35)"
return (splitvar(arg)..., is_splat, default)
if @capture(arg_expr, arg_expr2_ = default_)
@assert default !== nothing "splitarg cannot handle `nothing` as a default. Use a quoted `nothing` if possible. (MacroTools#35)"
else
return (splitvar(arg_expr2)..., is_splat, nothing)
arg_expr2 = arg_expr
end
is_splat = @capture(arg_expr2, arg_expr3_...)
is_splat || (arg_expr3 = arg_expr2)
(arg_name, arg_type) = (@match arg_expr3 begin
::T_ => (nothing, T)
name_::T_ => (name, T)
x_ => (x, :Any)
end)::NTuple{2,Any} # the pattern `x_` matches any expression
return (arg_name, arg_type, is_splat, default)
end


Expand Down
17 changes: 16 additions & 1 deletion test/split.jl
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@ end

macro splitcombine(fundef) # should be a no-op
dict = splitdef(fundef)
dict[:args] = map(arg->combinearg(splitarg(arg)...), dict[:args])
dict[:kwargs] = map(arg->combinearg(splitarg(arg)...), dict[:kwargs])
esc(MacroTools.combinedef(dict))
end

Expand All @@ -24,10 +26,17 @@ let
@test longdef(:(f(x)::Int = 10)).head == :function
@test longdef(:(f(x::T) where U where T = 2)).head == :function
@test shortdef(:(function f(x)::Int 10 end)).head != :function
@test map(splitarg, (:(f(a=2, x::Int=nothing, y, args...))).args[2:end]) ==
@test map(splitarg, (:(f(a=2, x::Int=nothing, y::Any, args...))).args[2:end]) ==
[(:a, :Any, false, 2), (:x, :Int, false, :nothing),
(:y, :Any, false, nothing), (:args, :Any, true, nothing)]
@test splitarg(:(::Int)) == (nothing, :Int, false, nothing)
kwargs = splitdef(:(f(; a::Int = 1, b...) = 1))[:kwargs]
@test map(splitarg, kwargs) ==
[(:a, :Int, false, 1), (:b, :Any, true, nothing)]
args = splitdef(:(f(a::Int = 1) = 1))[:args]
@test map(splitarg, args) == [(:a, :Int, false, 1)]
args = splitdef(:(f(a::Int ... = 1) = 1))[:args]
@test map(splitarg, args) == [(:a, :Int, true, 1)]

@splitcombine foo(x) = x+2
@test foo(10) == 12
Expand All @@ -47,6 +56,12 @@ let
@splitcombine fmacro1() = @onearg 1
@test fmacro1() == 2

@splitcombine bar(; a::Int = 1, b...) = 2
@test bar(a=3, x = 1, y = 2) == 2
@splitcombine qux(a::Int... = 0) = sum(a)
@test qux(1, 2, 3) == 6
@test qux() == 0

struct Foo{A, B}
a::A
b::B
Expand Down