%%
%% %CopyrightBegin%
%%
%% Copyright Ericsson AB 1996-2016. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%% http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%
%% %CopyrightEnd%
%%
-module(erl_pp).
%%% Pretty printer for Erlang code in the same format as returned from
%%% the parser. It does not always produce pretty code.
-export([form/1,form/2,
attribute/1,attribute/2,function/1,function/2,
guard/1,guard/2,exprs/1,exprs/2,exprs/3,expr/1,expr/2,expr/3,expr/4]).
-import(lists, [append/1,foldr/3,mapfoldl/3,reverse/1,reverse/2]).
-import(io_lib, [write/1,format/2]).
-import(erl_parse, [inop_prec/1,preop_prec/1,func_prec/0,max_prec/0,
type_inop_prec/1, type_preop_prec/1]).
-define(MAXLINE, 72).
-type(hook_function() :: none
| fun((Expr :: erl_parse:abstract_expr(),
CurrentIndentation :: integer(),
CurrentPrecedence :: non_neg_integer(),
Options :: options()) ->
io_lib:chars())).
-type(option() :: {hook, hook_function()}
| {encoding, latin1 | unicode | utf8}).
-type(options() :: hook_function() | [option()]).
-record(pp, {string_fun, char_fun}).
-record(options, {hook, encoding, opts}).
%-define(DEBUG, true).
-ifdef(DEBUG).
-define(TEST(T),
%% Assumes that erl_anno has been compiled with DEBUG=true.
%% erl_pp does not use the annoations, but test it anyway.
%% Note: hooks are not handled.
_ = try
erl_parse:map_anno(fun(A) when is_list(A) -> A end, T)
catch
_:_ ->
erlang:error(badarg, [T])
end).
-else.
-define(TEST(T), ok).
-endif.
%%%
%%% Exported functions
%%%
-spec(form(Form) -> io_lib:chars() when
Form :: erl_parse:abstract_form() | erl_parse:form_info()).
form(Thing) ->
form(Thing, none).
-spec(form(Form, Options) -> io_lib:chars() when
Form :: erl_parse:abstract_form() | erl_parse:form_info(),
Options :: options()).
form(Thing, Options) ->
?TEST(Thing),
State = state(Options),
frmt(lform(Thing, options(Options), State), State).
-spec(attribute(Attribute) -> io_lib:chars() when
Attribute :: erl_parse:abstract_form()).
attribute(Thing) ->
attribute(Thing, none).
-spec(attribute(Attribute, Options) -> io_lib:chars() when
Attribute :: erl_parse:abstract_form(),
Options :: options()).
attribute(Thing, Options) ->
?TEST(Thing),
State = state(Options),
frmt(lattribute(Thing, options(Options), State), State).
-spec(function(Function) -> io_lib:chars() when
Function :: erl_parse:abstract_form()).
function(F) ->
function(F, none).
-spec(function(Function, Options) -> io_lib:chars() when
Function :: erl_parse:abstract_form(),
Options :: options()).
function(F, Options) ->
?TEST(F),
frmt(lfunction(F, options(Options)), state(Options)).
-spec(guard(Guard) -> io_lib:chars() when
Guard :: [erl_parse:abstract_expr()]).
guard(Gs) ->
guard(Gs, none).
-spec(guard(Guard, Options) -> io_lib:chars() when
Guard :: [erl_parse:abstract_expr()],
Options :: options()).
guard(Gs, Options) ->
?TEST(Gs),
frmt(lguard(Gs, options(Options)), state(Options)).
-spec(exprs(Expressions) -> io_lib:chars() when
Expressions :: [erl_parse:abstract_expr()]).
exprs(Es) ->
exprs(Es, 0, none).
-spec(exprs(Expressions, Options) -> io_lib:chars() when
Expressions :: [erl_parse:abstract_expr()],
Options :: options()).
exprs(Es, Options) ->
exprs(Es, 0, Options).
-spec(exprs(Expressions, Indent, Options) -> io_lib:chars() when
Expressions :: [erl_parse:abstract_expr()],
Indent :: integer(),
Options :: options()).
exprs(Es, I, Options) ->
?TEST(Es),
frmt({seq,[],[],[$,],lexprs(Es, options(Options))}, I, state(Options)).
-spec(expr(Expression) -> io_lib:chars() when
Expression :: erl_parse:abstract_expr()).
expr(E) ->
?TEST(E),
frmt(lexpr(E, 0, options(none)), state(none)).
-spec(expr(Expression, Options) -> io_lib:chars() when
Expression :: erl_parse:abstract_expr(),
Options :: options()).
expr(E, Options) ->
?TEST(E),
frmt(lexpr(E, 0, options(Options)), state(Options)).
-spec(expr(Expression, Indent, Options) -> io_lib:chars() when
Expression :: erl_parse:abstract_expr(),
Indent :: integer(),
Options :: options()).
expr(E, I, Options) ->
?TEST(E),
frmt(lexpr(E, 0, options(Options)), I, state(Options)).
-spec(expr(Expression, Indent, Precedence, Options) -> io_lib:chars() when
Expression :: erl_parse:abstract_expr(),
Indent :: integer(),
Precedence :: non_neg_integer(),
Options :: options()).
expr(E, I, P, Options) ->
?TEST(E),
frmt(lexpr(E, P, options(Options)), I, state(Options)).
%%%
%%% Local functions
%%%
options(Options) when is_list(Options) ->
Hook = proplists:get_value(hook, Options, none),
Encoding = encoding(Options),
#options{hook = Hook, encoding = Encoding, opts = Options};
options(Hook) ->
#options{hook = Hook, encoding = encoding([]), opts = Hook}.
state(Options) when is_list(Options) ->
case encoding(Options) of
latin1 -> state();
unicode -> unicode_state()
end;
state(_Hook) ->
state().
state() ->
#pp{string_fun = fun io_lib:write_string_as_latin1/1,
char_fun = fun io_lib:write_char_as_latin1/1}.
unicode_state() ->
#pp{string_fun = fun io_lib:write_string/1,
char_fun = fun io_lib:write_char/1}.
encoding(Options) ->
case proplists:get_value(encoding, Options, epp:default_encoding()) of
latin1 -> latin1;
utf8 -> unicode;
unicode -> unicode
end.
lform({attribute,Line,Name,Arg}, Opts, State) ->
lattribute({attribute,Line,Name,Arg}, Opts, State);
lform({function,Line,Name,Arity,Clauses}, Opts, _State) ->
lfunction({function,Line,Name,Arity,Clauses}, Opts);
%% These are specials to make it easier for the compiler.
lform({error,E}, _Opts, _State) ->
leaf(format("~p\n", [{error,E}]));
lform({warning,W}, _Opts, _State) ->
leaf(format("~p\n", [{warning,W}]));
lform({eof,_Line}, _Opts, _State) ->
$\n.
lattribute({attribute,_Line,type,Type}, Opts, _State) ->
[typeattr(type, Type, Opts),leaf(".\n")];
lattribute({attribute,_Line,opaque,Type}, Opts, _State) ->
[typeattr(opaque, Type, Opts),leaf(".\n")];
lattribute({attribute,_Line,spec,Arg}, _Opts, _State) ->
[specattr(spec, Arg),leaf(".\n")];
lattribute({attribute,_Line,callback,Arg}, _Opts, _State) ->
[specattr(callback, Arg),leaf(".\n")];
lattribute({attribute,_Line,Name,Arg}, Opts, State) ->
[lattribute(Name, Arg, Opts, State),leaf(".\n")].
lattribute(module, {M,Vs}, _Opts, _State) ->
A = a0(),
attr("module",[{var,A,pname(M)},
foldr(fun(V, C) -> {cons,A,{var,A,V},C}
end, {nil,A}, Vs)]);
lattribute(module, M, _Opts, _State) ->
attr("module", [{var,a0(),pname(M)}]);
lattribute(export, Falist, _Opts, _State) ->
call({var,a0(),"-export"}, [falist(Falist)], 0, options(none));
lattribute(import, Name, _Opts, _State) when is_list(Name) ->
attr("import", [{var,a0(),pname(Name)}]);
lattribute(import, {From,Falist}, _Opts, _State) ->
attr("import",[{var,a0(),pname(From)},falist(Falist)]);
lattribute(export_type, Talist, _Opts, _State) ->
call({var,a0(),"-export_type"}, [falist(Talist)], 0, options(none));
lattribute(optional_callbacks, Falist, Opts, _State) ->
ArgL = try falist(Falist)
catch _:_ -> abstract(Falist, Opts)
end,
call({var,a0(),"-optional_callbacks"}, [ArgL], 0, options(none));
lattribute(file, {Name,Line}, _Opts, State) ->
attr("file", [{var,a0(),(State#pp.string_fun)(Name)},{integer,a0(),Line}]);
lattribute(record, {Name,Is}, Opts, _State) ->
Nl = leaf(format("-record(~w,", [Name])),
[{first,Nl,record_fields(Is, Opts)},$)];
lattribute(Name, Arg, Options, _State) ->
attr(write(Name), [abstract(Arg, Options)]).
abstract(Arg, #options{encoding = Encoding}) ->
erl_parse:abstract(Arg, [{encoding,Encoding}]).
typeattr(Tag, {TypeName,Type,Args}, _Opts) ->
{first,leaf("-"++atom_to_list(Tag)++" "),
typed(call({atom,a0(),TypeName}, Args, 0, options(none)), Type)}.
ltype(T) ->
ltype(T, 0).
ltype({ann_type,_Line,[V,T]}, Prec) ->
{_L,P,_R} = type_inop_prec('::'),
E = typed(lexpr(V, options(none)), T),
maybe_paren(P, Prec, E);
ltype({paren_type,_Line,[T]}, P) ->
%% Generated before Erlang/OTP 18.
ltype(T, P);
ltype({type,_Line,union,Ts}, Prec) ->
{_L,P,R} = type_inop_prec('|'),
E = {seq,[],[],[' |'],ltypes(Ts, R)},
maybe_paren(P, Prec, E);
ltype({type,_Line,list,[T]}, _) ->
{seq,$[,$],$,,[ltype(T)]};
ltype({type,_Line,nonempty_list,[T]}, _) ->
{seq,$[,$],[$,],[ltype(T),leaf("...")]};
ltype({type,Line,nil,[]}, _) ->
lexpr({nil,Line}, options(none));
ltype({type,Line,map,any}, _) ->
simple_type({atom,Line,map}, []);
ltype({type,_Line,map,Pairs}, Prec) ->
{P,_R} = type_preop_prec('#'),
E = map_type(Pairs),
maybe_paren(P, Prec, E);
ltype({type,Line,tuple,any}, _) ->
simple_type({atom,Line,tuple}, []);
ltype({type,_Line,tuple,Ts}, _) ->
tuple_type(Ts, fun ltype/2);
ltype({type,_Line,record,[{atom,_,N}|Fs]}, Prec) ->
{P,_R} = type_preop_prec('#'),
E = record_type(N, Fs),
maybe_paren(P, Prec, E);
ltype({type,_Line,range,[_I1,_I2]=Es}, Prec) ->
{_L,P,R} = type_inop_prec('..'),
F = fun(E, Opts) -> lexpr(E, R, Opts) end,
E = expr_list(Es, '..', F, options(none)),
maybe_paren(P, Prec, E);
ltype({type,_Line,binary,[I1,I2]}, _) ->
binary_type(I1, I2); % except binary()
ltype({type,_Line,'fun',[]}, _) ->
leaf("fun()");
ltype({type,_,'fun',[{type,_,any},_]}=FunType, _) ->
[fun_type(['fun',$(], FunType),$)];
ltype({type,_Line,'fun',[{type,_,product,_},_]}=FunType, _) ->
[fun_type(['fun',$(], FunType),$)];
ltype({type,Line,T,Ts}, _) ->
simple_type({atom,Line,T}, Ts);
ltype({user_type,Line,T,Ts}, _) ->
simple_type({atom,Line,T}, Ts);
ltype({remote_type,Line,[M,F,Ts]}, _) ->
simple_type({remote,Line,M,F}, Ts);
ltype({atom,_,T}, _) ->
leaf(write(T));
ltype(E, P) ->
lexpr(E, P, options(none)).
binary_type(I1, I2) ->
B = [[] || {integer,_,0} <- [I1]] =:= [],
U = [[] || {integer,_,0} <- [I2]] =:= [],
P = max_prec(),
E1 = [[leaf("_:"),lexpr(I1, P, options(none))] || B],
E2 = [[leaf("_:_*"),lexpr(I2, P, options(none))] || U],
{seq,'<<','>>',[$,],E1++E2}.
map_type(Fs) ->
{first,[$#],map_pair_types(Fs)}.
map_pair_types(Fs) ->
tuple_type(Fs, fun map_pair_type/2).
map_pair_type({type,_Line,map_field_assoc,[KType,VType]}, Prec) ->
{list,[{cstep,[ltype(KType, Prec),leaf(" =>")],ltype(VType, Prec)}]};
map_pair_type({type,_Line,map_field_exact,[KType,VType]}, Prec) ->
{list,[{cstep,[ltype(KType, Prec),leaf(" :=")],ltype(VType, Prec)}]}.
record_type(Name, Fields) ->
{first,[record_name(Name)],field_types(Fields)}.
field_types(Fs) ->
tuple_type(Fs, fun field_type/2).
field_type({type,_Line,field_type,[Name,Type]}, _Prec) ->
typed(lexpr(Name, options(none)), Type).
typed(B, Type) ->
{_L,_P,R} = type_inop_prec('::'),
{list,[{cstep,[B,' ::'],ltype(Type, R)}]}.
tuple_type(Ts, F) ->
{seq,${,$},[$,],ltypes(Ts, F, 0)}.
specattr(SpecKind, {FuncSpec,TypeSpecs}) ->
Func = case FuncSpec of
{F,_A} ->
format("~w", [F]);
{M,F,_A} ->
format("~w:~w", [M, F])
end,
{first,leaf(lists:concat(["-", SpecKind, " "])),
{list,[{first,leaf(Func),spec_clauses(TypeSpecs)}]}}.
spec_clauses(TypeSpecs) ->
{prefer_nl,[$;],[sig_type(T) || T <- TypeSpecs]}.
sig_type({type,_Line,bounded_fun,[T,Gs]}) ->
guard_type(fun_type([], T), Gs);
sig_type(FunType) ->
fun_type([], FunType).
guard_type(Before, Gs) ->
Opts = options(none),
Gl = {list,[{step,'when',expr_list(Gs, [$,], fun constraint/2, Opts)}]},
{list,[{step,Before,Gl}]}.
constraint({type,_Line,constraint,[{atom,_,is_subtype},[{var,_,_}=V,Type]]},
_Opts) ->
typed(lexpr(V, options(none)), Type);
constraint({type,_Line,constraint,[Tag,As]}, _Opts) ->
simple_type(Tag, As).
fun_type(Before, {type,_,'fun',[FType,Ret]}) ->
{first,Before,{step,[type_args(FType),' ->'],ltype(Ret)}}.
type_args({type,_Line,any}) ->
leaf("(...)");
type_args({type,_line,product,Ts}) ->
targs(Ts).
simple_type(Tag, Types) ->
{first,lexpr(Tag, options(none)),targs(Types)}.
targs(Ts) ->
{seq,$(,$),[$,],ltypes(Ts, 0)}.
ltypes(Ts, Prec) ->
ltypes(Ts, fun ltype/2, Prec).
ltypes(Ts, F, Prec) ->
[F(T, Prec) || T <- Ts].
attr(Name, Args) ->
call({var,a0(),format("-~s", [Name])}, Args, 0, options(none)).
pname(['' | As]) ->
[$. | pname(As)];
pname([A]) ->
write(A);
pname([A | As]) ->
[write(A),$.|pname(As)];
pname(A) when is_atom(A) ->
write(A).
falist([]) ->
{nil,a0()};
falist([{Name,Arity}|Falist]) ->
A = a0(),
{cons,A,{var,A,format("~w/~w", [Name,Arity])},falist(Falist)}.
lfunction({function,_Line,Name,_Arity,Cs}, Opts) ->
Cll = nl_clauses(fun (C, H) -> func_clause(Name, C, H) end, $;, Opts, Cs),
[Cll,leaf(".\n")].
func_clause(Name, {clause,Line,Head,Guard,Body}, Opts) ->
Hl = call({atom,Line,Name}, Head, 0, Opts),
Gl = guard_when(Hl, Guard, Opts),
Bl = body(Body, Opts),
{step,Gl,Bl}.
guard_when(Before, Guard, Opts) ->
guard_when(Before, Guard, Opts, ' ->').
guard_when(Before, Guard, Opts, After) ->
Gl = lguard(Guard, Opts),
[{list,[{step,Before,Gl}]},After].
lguard([E|Es], Opts) when is_list(E) ->
{list,[{step,'when',expr_list([E|Es], [$;], fun guard0/2, Opts)}]};
lguard([E|Es], Opts) -> % before R6
lguard([[E|Es]], Opts);
lguard([], _) ->
[].
guard0(Es, Opts) ->
expr_list(Es, [$,], fun lexpr/2, Opts).
%% body(Before, Es, Opts) -> [Char].
body([E], Opts) ->
lexpr(E, Opts);
body(Es, Opts) ->
{prefer_nl,[$,],lexprs(Es, Opts)}.
lexpr(E, Opts) ->
lexpr(E, 0, Opts).
lexpr({var,_,V}, _, _) when is_integer(V) -> %Special hack for Robert
leaf(format("_~w", [V]));
lexpr({var,_,V}, _, _) -> leaf(format("~ts", [V]));
lexpr({char,_,C}, _, _) -> {char,C};
lexpr({integer,_,N}, _, _) -> leaf(write(N));
lexpr({float,_,F}, _, _) -> leaf(write(F));
lexpr({atom,_,A}, _, _) -> leaf(write(A));
lexpr({string,_,S}, _, _) -> {string,S};
lexpr({nil,_}, _, _) -> '[]';
lexpr({cons,_,H,T}, _, Opts) ->
list(T, [H], Opts);
lexpr({lc,_,E,Qs}, _Prec, Opts) ->
Lcl = {list,[{step,[lexpr(E, Opts),leaf(" ||")],lc_quals(Qs, Opts)}]},
{list,[{seq,$[,[],[[]],[{force_nl,leaf(" "),[Lcl]}]},$]]};
%% {list,[{step,$[,Lcl},$]]};
lexpr({bc,_,E,Qs}, _Prec, Opts) ->
Lcl = {list,[{step,[lexpr(E, Opts),leaf(" ||")],lc_quals(Qs, Opts)}]},
{list,[{seq,'<<',[],[[]],[{force_nl,leaf(" "),[Lcl]}]},'>>']};
%% {list,[{step,'<<',Lcl},'>>']};
lexpr({tuple,_,Elts}, _, Opts) ->
tuple(Elts, Opts);
%%lexpr({struct,_,Tag,Elts}, _, Opts) ->
%% {first,format("~w", [Tag]),tuple(Elts, Opts)};
lexpr({record_index, _, Name, F}, Prec, Opts) ->
{P,R} = preop_prec('#'),
Nl = record_name(Name),
El = [Nl,$.,lexpr(F, R, Opts)],
maybe_paren(P, Prec, El);
lexpr({record, _, Name, Fs}, Prec, Opts) ->
{P,_R} = preop_prec('#'),
Nl = record_name(Name),
El = {first,Nl,record_fields(Fs, Opts)},
maybe_paren(P, Prec, El);
lexpr({record_field, _, Rec, Name, F}, Prec, Opts) ->
{L,P,R} = inop_prec('#'),
Rl = lexpr(Rec, L, Opts),
Nl = leaf(format("#~w.", [Name])),
El = [Rl,Nl,lexpr(F, R, Opts)],
maybe_paren(P, Prec, El);
lexpr({record, _, Rec, Name, Fs}, Prec, Opts) ->
{L,P,_R} = inop_prec('#'),
Rl = lexpr(Rec, L, Opts),
Nl = record_name(Name),
El = {first,[Rl,Nl],record_fields(Fs, Opts)},
maybe_paren(P, Prec, El);
lexpr({record_field, _, {atom,_,''}, F}, Prec, Opts) ->
{_L,P,R} = inop_prec('.'),
El = [$.,lexpr(F, R, Opts)],
maybe_paren(P, Prec, El);
lexpr({record_field, _, Rec, F}, Prec, Opts) ->
{L,P,R} = inop_prec('.'),
El = [lexpr(Rec, L, Opts),$.,lexpr(F, R, Opts)],
maybe_paren(P, Prec, El);
lexpr({map, _, Fs}, Prec, Opts) ->
{P,_R} = preop_prec('#'),
El = {first,leaf("#"),map_fields(Fs, Opts)},
maybe_paren(P, Prec, El);
lexpr({map, _, Map, Fs}, Prec, Opts) ->
{L,P,_R} = inop_prec('#'),
Rl = lexpr(Map, L, Opts),
El = {first,[Rl,leaf("#")],map_fields(Fs, Opts)},
maybe_paren(P, Prec, El);
lexpr({block,_,Es}, _, Opts) ->
{list,[{step,'begin',body(Es, Opts)},'end']};
lexpr({'if',_,Cs}, _, Opts) ->
{list,[{step,'if',if_clauses(Cs, Opts)},'end']};
lexpr({'case',_,Expr,Cs}, _, Opts) ->
{list,[{step,{list,[{step,'case',lexpr(Expr, Opts)},'of']},
cr_clauses(Cs, Opts)},
'end']};
lexpr({'cond',_,Cs}, _, Opts) ->
{list,[{step,leaf("cond"),cond_clauses(Cs, Opts)},'end']};
lexpr({'receive',_,Cs}, _, Opts) ->
{list,[{step,'receive',cr_clauses(Cs, Opts)},'end']};
lexpr({'receive',_,Cs,To,ToOpt}, _, Opts) ->
Al = {list,[{step,[lexpr(To, Opts),' ->'],body(ToOpt, Opts)}]},
{list,[{step,'receive',cr_clauses(Cs, Opts)},
{step,'after',Al},
'end']};
lexpr({'fun',_,{function,F,A}}, _Prec, _Opts) ->
leaf(format("fun ~w/~w", [F,A]));
lexpr({'fun',_,{function,F,A},Extra}, _Prec, _Opts) ->
{force_nl,fun_info(Extra),leaf(format("fun ~w/~w", [F,A]))};
lexpr({'fun',_,{function,M,F,A}}, _Prec, _Opts)
when is_atom(M), is_atom(F), is_integer(A) ->
%% For backward compatibility with pre-R15 abstract format.
leaf(format("fun ~w:~w/~w", [M,F,A]));
lexpr({'fun',_,{function,M,F,A}}, _Prec, Opts) ->
%% New format in R15.
NameItem = lexpr(M, Opts),
CallItem = lexpr(F, Opts),
ArityItem = lexpr(A, Opts),
["fun ",NameItem,$:,CallItem,$/,ArityItem];
lexpr({'fun',_,{clauses,Cs}}, _Prec, Opts) ->
{list,[{first,'fun',fun_clauses(Cs, Opts, unnamed)},'end']};
lexpr({named_fun,_,Name,Cs}, _Prec, Opts) ->
{list,[{first,['fun', " "],fun_clauses(Cs, Opts, {named, Name})},'end']};
lexpr({'fun',_,{clauses,Cs},Extra}, _Prec, Opts) ->
{force_nl,fun_info(Extra),
{list,[{first,'fun',fun_clauses(Cs, Opts, unnamed)},'end']}};
lexpr({named_fun,_,Name,Cs,Extra}, _Prec, Opts) ->
{force_nl,fun_info(Extra),
{list,[{first,['fun', " "],fun_clauses(Cs, Opts, {named, Name})},'end']}};
lexpr({'query',_,Lc}, _Prec, Opts) ->
{list,[{step,leaf("query"),lexpr(Lc, 0, Opts)},'end']};
lexpr({call,_,{remote,_,{atom,_,M},{atom,_,F}=N}=Name,Args}, Prec, Opts) ->
case erl_internal:bif(M, F, length(Args)) of
true ->
call(N, Args, Prec, Opts);
false ->
call(Name, Args, Prec, Opts)
end;
lexpr({call,_,Name,Args}, Prec, Opts) ->
call(Name, Args, Prec, Opts);
lexpr({'try',_,Es,Scs,Ccs,As}, _, Opts) ->
{list,[if
Scs =:= [] ->
{step,'try',body(Es, Opts)};
true ->
{step,{list,[{step,'try',body(Es, Opts)},'of']},
cr_clauses(Scs, Opts)}
end,
if
Ccs =:= [] ->
[];
true ->
{step,'catch',try_clauses(Ccs, Opts)}
end,
if
As =:= [] ->
[];
true ->
{step,'after',body(As, Opts)}
end,
'end']};
lexpr({'catch',_,Expr}, Prec, Opts) ->
{P,R} = preop_prec('catch'),
El = {list,[{step,'catch',lexpr(Expr, R, Opts)}]},
maybe_paren(P, Prec, El);
lexpr({match,_,Lhs,Rhs}, Prec, Opts) ->
{L,P,R} = inop_prec('='),
Pl = lexpr(Lhs, L, Opts),
Rl = lexpr(Rhs, R, Opts),
El = {list,[{cstep,[Pl,' ='],Rl}]},
maybe_paren(P, Prec, El);
lexpr({op,_,Op,Arg}, Prec, Opts) ->
{P,R} = preop_prec(Op),
Ol = leaf(format("~s ", [Op])),
El = [Ol,lexpr(Arg, R, Opts)],
maybe_paren(P, Prec, El);
lexpr({op,_,Op,Larg,Rarg}, Prec, Opts) when Op =:= 'orelse';
Op =:= 'andalso' ->
%% Breaks lines since R12B.
{L,P,R} = inop_prec(Op),
Ll = lexpr(Larg, L, Opts),
Ol = leaf(format("~s", [Op])),
Lr = lexpr(Rarg, R, Opts),
El = {prefer_nl,[[]],[Ll,Ol,Lr]},
maybe_paren(P, Prec, El);
lexpr({op,_,Op,Larg,Rarg}, Prec, Opts) ->
{L,P,R} = inop_prec(Op),
Ll = lexpr(Larg, L, Opts),
Ol = leaf(format("~s", [Op])),
Lr = lexpr(Rarg, R, Opts),
El = {list,[Ll,Ol,Lr]},
maybe_paren(P, Prec, El);
%% Special expressions which are not really legal everywhere.
lexpr({remote,_,M,F}, Prec, Opts) ->
{L,P,R} = inop_prec(':'),
NameItem = lexpr(M, L, Opts),
CallItem = lexpr(F, R, Opts),
maybe_paren(P, Prec, [NameItem,$:,CallItem]);
%% BIT SYNTAX:
lexpr({bin,_,Fs}, _, Opts) ->
bit_grp(Fs, Opts);
%% Special case for straight values.
lexpr({value,_,Val}, _,_) ->
leaf(write(Val));
%% Now do the hook.
lexpr(Other, _Precedence, #options{hook = none}) ->
leaf(format("INVALID-FORM:~w:",[Other]));
lexpr(HookExpr, Precedence, #options{hook = {Mod,Func,Eas}})
when Mod =/= 'fun' ->
{ehook,HookExpr,Precedence,{Mod,Func,Eas}};
lexpr(HookExpr, Precedence, #options{hook = Func, opts = Options}) ->
{hook,HookExpr,Precedence,Func,Options}.
call(Name, Args, Prec, Opts) ->
{F,P} = func_prec(),
Item = {first,lexpr(Name, F, Opts),args(Args, Opts)},
maybe_paren(P, Prec, Item).
fun_info(Extra) ->
leaf(format("% fun-info: ~w", [Extra])).
%% BITS:
bit_grp(Fs, Opts) ->
append([['<<'], [bit_elems(Fs, Opts)], ['>>']]).
bit_elems(Es, Opts) ->
expr_list(Es, $,, fun bit_elem/2, Opts).
bit_elem({bin_element,_,Expr,Sz,Types}, Opts) ->
P = max_prec(),
VChars = lexpr(Expr, P, Opts),
SChars = if
Sz =/= default ->
[VChars,$:,lexpr(Sz, P, Opts)];
true ->
VChars
end,
if
Types =/= default ->
[SChars,$/|bit_elem_types(Types)];
true ->
SChars
end.
bit_elem_types([T]) ->
[bit_elem_type(T)];
bit_elem_types([T | Rest]) ->
[bit_elem_type(T), $-|bit_elem_types(Rest)].
bit_elem_type({A,B}) ->
[lexpr(erl_parse:abstract(A), options(none)),
$:,
lexpr(erl_parse:abstract(B), options(none))];
bit_elem_type(T) ->
lexpr(erl_parse:abstract(T), options(none)).
%% end of BITS
record_name(Name) ->
leaf(format("#~w", [Name])).
record_fields(Fs, Opts) ->
tuple(Fs, fun record_field/2, Opts).
record_field({record_field,_,F,Val}, Opts) ->
{L,_P,R} = inop_prec('='),
Fl = lexpr(F, L, Opts),
Vl = lexpr(Val, R, Opts),
{list,[{cstep,[Fl,' ='],Vl}]};
record_field({typed_record_field,{record_field,_,F,Val},Type}, Opts) ->
{L,_P,R} = inop_prec('='),
Fl = lexpr(F, L, Opts),
Vl = typed(lexpr(Val, R, Opts), Type),
{list,[{cstep,[Fl,' ='],Vl}]};
record_field({typed_record_field,Field,Type}, Opts) ->
typed(record_field(Field, Opts), Type);
record_field({record_field,_,F}, Opts) ->
lexpr(F, 0, Opts).
map_fields(Fs, Opts) ->
tuple(Fs, fun map_field/2, Opts).
map_field({map_field_assoc,_,K,V}, Opts) ->
Pl = lexpr(K, 0, Opts),
{list,[{step,[Pl,leaf(" =>")],lexpr(V, 0, Opts)}]};
map_field({map_field_exact,_,K,V}, Opts) ->
Pl = lexpr(K, 0, Opts),
{list,[{step,[Pl,leaf(" :=")],lexpr(V, 0, Opts)}]}.
list({cons,_,H,T}, Es, Opts) ->
list(T, [H|Es], Opts);
list({nil,_}, Es, Opts) ->
proper_list(reverse(Es), Opts);
list(Other, Es, Opts) ->
improper_list(reverse(Es, [Other]), Opts).
%% if_clauses(Clauses, Opts) -> [Char].
%% Print 'if' clauses.
if_clauses(Cs, Opts) ->
clauses(fun if_clause/2, Opts, Cs).
if_clause({clause,_,[],G,B}, Opts) ->
Gl = [guard_no_when(G, Opts),' ->'],
{step,Gl,body(B, Opts)}.
guard_no_when([E|Es], Opts) when is_list(E) ->
expr_list([E|Es], $;, fun guard0/2, Opts);
guard_no_when([E|Es], Opts) -> % before R6
guard_no_when([[E|Es]], Opts);
guard_no_when([], _) -> % cannot happen
leaf("true").
%% cr_clauses(Clauses, Opts) -> [Char].
%% Print 'case'/'receive' clauses.
cr_clauses(Cs, Opts) ->
clauses(fun cr_clause/2, Opts, Cs).
cr_clause({clause,_,[T],G,B}, Opts) ->
El = lexpr(T, 0, Opts),
Gl = guard_when(El, G, Opts),
Bl = body(B, Opts),
{step,Gl,Bl}.
%% try_clauses(Clauses, Opts) -> [Char].
%% Print 'try' clauses.
try_clauses(Cs, Opts) ->
clauses(fun try_clause/2, Opts, Cs).
try_clause({clause,_,[{tuple,_,[{atom,_,throw},V,S]}],G,B}, Opts) ->
El = lexpr(V, 0, Opts),
Sl = stack_backtrace(S, [El], Opts),
Gl = guard_when(Sl, G, Opts),
Bl = body(B, Opts),
{step,Gl,Bl};
try_clause({clause,_,[{tuple,_,[C,V,S]}],G,B}, Opts) ->
Cs = lexpr(C, 0, Opts),
El = lexpr(V, 0, Opts),
CsEl = [Cs,$:,El],
Sl = stack_backtrace(S, CsEl, Opts),
Gl = guard_when(Sl, G, Opts),
Bl = body(B, Opts),
{step,Gl,Bl}.
stack_backtrace({var,_,'_'}, El, _Opts) ->
El;
stack_backtrace(S, El, Opts) ->
El++[$:,lexpr(S, 0, Opts)].
%% fun_clauses(Clauses, Opts) -> [Char].
%% Print 'fun' clauses.
fun_clauses(Cs, Opts, unnamed) ->
nl_clauses(fun fun_clause/2, [$;], Opts, Cs);
fun_clauses(Cs, Opts, {named, Name}) ->
nl_clauses(fun (C, H) ->
{step,Gl,Bl} = fun_clause(C, H),
{step,[atom_to_list(Name),Gl],Bl}
end, [$;], Opts, Cs).
fun_clause({clause,_,A,G,B}, Opts) ->
El = args(A, Opts),
Gl = guard_when(El, G, Opts),
Bl = body(B, Opts),
{step,Gl,Bl}.
%% cond_clauses(Clauses, Opts) -> [Char].
%% Print 'cond' clauses.
cond_clauses(Cs, Opts) ->
clauses(fun cond_clause/2, Opts, Cs).
cond_clause({clause,_,[],[[E]],B}, Opts) ->
{step,[lexpr(E, Opts),' ->'],body(B, Opts)}.
%% nl_clauses(Type, Opts, Clauses) -> [Char].
%% Generic clause printing function (always breaks lines).
nl_clauses(Type, Sep, Opts, Cs) ->
{prefer_nl,Sep,lexprs(Cs, Type, Opts)}.
%% clauses(Type, Opts, Clauses) -> [Char].
%% Generic clause printing function (breaks lines since R12B).
clauses(Type, Opts, Cs) ->
{prefer_nl,[$;],lexprs(Cs, Type, Opts)}.
%% lc_quals(Qualifiers, After, Opts)
%% List comprehension qualifiers (breaks lines since R12B).
lc_quals(Qs, Opts) ->
{prefer_nl,[$,],lexprs(Qs, fun lc_qual/2, Opts)}.
lc_qual({b_generate,_,Pat,E}, Opts) ->
Pl = lexpr(Pat, 0, Opts),
{list,[{step,[Pl,leaf(" <=")],lexpr(E, 0, Opts)}]};
lc_qual({generate,_,Pat,E}, Opts) ->
Pl = lexpr(Pat, 0, Opts),
{list,[{step,[Pl,leaf(" <-")],lexpr(E, 0, Opts)}]};
lc_qual(Q, Opts) ->
lexpr(Q, 0, Opts).
proper_list(Es, Opts) ->
{seq,$[,$],$,,lexprs(Es, Opts)}.
improper_list(Es, Opts) ->
{seq,$[,$],{$,,$|},lexprs(Es, Opts)}.
tuple(L, Opts) ->
tuple(L, fun lexpr/2, Opts).
tuple(Es, F, Opts) ->
{seq,${,$},$,,lexprs(Es, F, Opts)}.
args(As, Opts) ->
{seq,$(,$),[$,],lexprs(As, Opts)}.
expr_list(Es, Sep, F, Opts) ->
{seq,[],[],Sep,lexprs(Es, F, Opts)}.
lexprs(Es, Opts) ->
lexprs(Es, fun lexpr/2, Opts).
lexprs(Es, F, Opts) ->
[F(E, Opts) || E <- Es].
maybe_paren(P, Prec, Expr) when P < Prec ->
[$(,Expr,$)];
maybe_paren(_P, _Prec, Expr) ->
Expr.
leaf(S) ->
{leaf,chars_size(S),S}.
%%% Do the formatting. Currently nothing fancy. Could probably have
%%% done it in one single pass.
frmt(Item, PP) ->
frmt(Item, 0, PP).
frmt(Item, I, PP) ->
ST = spacetab(),
WT = wordtable(),
{Chars,_Length} = f(Item, I, ST, WT, PP),
[Chars].
%%% What the tags mean:
%%% - C: a character
%%% - [I|Is]: Is follow after I without newline or space
%%% - {list,IPs}: try to put all IPs on one line, if that fails newlines
%%% and indentation are inserted between IPs.
%%% - {first,I,IP2}: IP2 follows after I, and is output with an indentation
%%% updated with the width of I.
%%% - {seq,Before,After,Separator,IPs}: a sequence of Is separated by
%%% Separator. Before is output before IPs, and the indentation of IPs
%%% is updated with the width of Before. After follows after IPs.
%%% - {force_nl,ExtraInfo,I}: fun-info (a comment) forces linebreak before I.
%%% - {prefer_nl,Sep,IPs}: forces linebreak between Is unlesss negative
%%% indentation.
%%% - {char,C}: a character
%%% - {string,S}: a string.
%%% - {hook,...}, {ehook,...}: hook expressions.
%%%
%%% list, first, seq, force_nl, and prefer_nl all accept IPs, where each
%%% element is either an item or a tuple {step|cstep,I1,I2}. step means
%%% that I2 is output after linebreak and an incremented indentation.
%%% cstep works similarly, but no linebreak if the width of I1 is less
%%% than the indentation (this is for "A = <expression over several lines>).
f([]=Nil, _I0, _ST, _WT, _PP) ->
{Nil,0};
f(C, _I0, _ST, _WT, _PP) when is_integer(C) ->
{C,1};
f({leaf,Length,Chars}, _I0, _ST, _WT, _PP) ->
{Chars,Length};
f([Item|Items], I0, ST, WT, PP) ->
consecutive(Items, f(Item, I0, ST, WT, PP), I0, ST, WT, PP);
f({list,Items}, I0, ST, WT, PP) ->
f({seq,[],[],[[]],Items}, I0, ST, WT, PP);
f({first,E,Item}, I0, ST, WT, PP) ->
f({seq,E,[],[[]],[Item]}, I0, ST, WT, PP);
f({seq,Before,After,Sep,LItems}, I0, ST, WT, PP) ->
BCharsSize = f(Before, I0, ST, WT, PP),
I = indent(BCharsSize, I0),
CharsSizeL = fl(LItems, Sep, I, After, ST, WT, PP),
{CharsL,SizeL} = unz(CharsSizeL),
{BCharsL,BSizeL} = unz1([BCharsSize]),
Sizes = BSizeL ++ SizeL,
NSepChars = if
is_list(Sep), Sep =/= [] ->
erlang:max(0, length(CharsL)-1);
true ->
0
end,
case same_line(I0, Sizes, NSepChars) of
{yes,Size} ->
Chars = if
NSepChars > 0 -> insert_sep(CharsL, $\s);
true -> CharsL
end,
{BCharsL++Chars,Size};
no ->
{BCharsL++insert_newlines(CharsSizeL, I, ST),
nsz(lists:last(Sizes), I0)}
end;
f({force_nl,_ExtraInfoItem,Item}, I, ST, WT, PP) when I < 0 ->
%% Extra info is a comment; cannot have that on the same line
f(Item, I, ST, WT, PP);
f({force_nl,ExtraInfoItem,Item}, I, ST, WT, PP) ->
f({prefer_nl,[],[ExtraInfoItem,Item]}, I, ST, WT, PP);
f({prefer_nl,Sep,LItems}, I, ST, WT, PP) when I < 0 ->
f({seq,[],[],Sep,LItems}, I, ST, WT, PP);
f({prefer_nl,Sep,LItems}, I0, ST, WT, PP) ->
CharsSize2L = fl(LItems, Sep, I0, [], ST, WT, PP),
{_CharsL,Sizes} = unz(CharsSize2L),
if
Sizes =:= [] ->
{[], 0};
true ->
{insert_newlines(CharsSize2L, I0, ST),nsz(lists:last(Sizes), I0)}
end;
f({char,C}, I, ST, WT, PP) ->
f(write_a_char(C, PP), I, ST, WT, PP);
f({string,S}, I, ST, WT, PP) ->
f(write_a_string(S, I, PP), I, ST, WT, PP);
f({hook,HookExpr,Precedence,Func,Options}, I, _ST, _WT, _PP) ->
Chars = Func(HookExpr, I, Precedence, Options),
{Chars,indentation(Chars, I)};
f({ehook,HookExpr,Precedence,{Mod,Func,Eas}=ModFuncEas}, I, _ST, _WT, _PP) ->
Chars = apply(Mod, Func, [HookExpr,I,Precedence,ModFuncEas|Eas]),
{Chars,indentation(Chars, I)};
f(WordName, _I, _ST, WT, _PP) -> % when is_atom(WordName)
word(WordName, WT).
-define(IND, 4).
%% fl(ListItems, I0, ST, WT) -> [[CharsSize1,CharsSize2]]
%% ListItems = [{Item,Items}|Item]
fl([], _Sep, I0, After, ST, WT, PP) ->
[[f(After, I0, ST, WT, PP),{[],0}]];
fl(CItems, Sep0, I0, After, ST, WT, PP) ->
F = fun({step,Item1,Item2}, S) ->
[f(Item1, I0, ST, WT, PP),
f([Item2,S], incr(I0, ?IND), ST, WT, PP)];
({cstep,Item1,Item2}, S) ->
{_,Sz1} = CharSize1 = f(Item1, I0, ST, WT, PP),
if
is_integer(Sz1), Sz1 < ?IND ->
Item2p = [leaf("\s"),Item2,S],
[consecutive(Item2p, CharSize1, I0, ST, WT, PP),{[],0}];
true ->
[CharSize1,f([Item2,S], incr(I0, ?IND), ST, WT, PP)]
end;
(Item, S) ->
[f([Item,S], I0, ST, WT, PP),{[],0}]
end,
{Sep,LastSep} = case Sep0 of {_,_} -> Sep0; _ -> {Sep0,Sep0} end,
fl1(CItems, F, Sep, LastSep, After).
fl1([CItem], F, _Sep, _LastSep, After) ->
[F(CItem,After)];
fl1([CItem1,CItem2], F, _Sep, LastSep, After) ->
[F(CItem1, LastSep),F(CItem2, After)];
fl1([CItem|CItems], F, Sep, LastSep, After) ->
[F(CItem, Sep)|fl1(CItems, F, Sep, LastSep, After)].
consecutive(Items, CharSize1, I0, ST, WT, PP) ->
{CharsSizes,_Length} =
mapfoldl(fun(Item, Len) ->
CharsSize = f(Item, Len, ST, WT, PP),
{CharsSize,indent(CharsSize, Len)}
end, indent(CharSize1, I0), Items),
{CharsL,SizeL} = unz1([CharSize1|CharsSizes]),
{CharsL,line_size(SizeL)}.
unz(CharsSizesL) ->
unz1(append(CharsSizesL)).
unz1(CharSizes) ->
lists:unzip(nonzero(CharSizes)).
nonzero(CharSizes) ->
lists:filter(fun({_,Sz}) -> Sz =/= 0 end, CharSizes).
insert_newlines(CharsSizesL, I, ST) when I >= 0 ->
insert_nl(foldr(fun([{_C1,0},{_C2,0}], A) ->
A;
([{C1,_Sz1},{_C2,0}], A) ->
[C1|A];
([{C1,_Sz1},{C2,Sz2}], A) when Sz2 > 0 ->
[insert_nl([C1,C2], I+?IND, ST)|A]
end, [], CharsSizesL), I, ST).
insert_nl(CharsL, I, ST) ->
insert_sep(CharsL, nl_indent(I, ST)).
insert_sep([Chars1 | CharsL], Sep) ->
[Chars1 | [[Sep,Chars] || Chars <- CharsL]].
nl_indent(0, _T) ->
$\n;
nl_indent(I, T) when I > 0 ->
[$\n|spaces(I, T)].
same_line(I0, SizeL, NSepChars) ->
try
Size = lists:sum(SizeL) + NSepChars,
true = incr(I0, Size) =< ?MAXLINE,
{yes,Size}
catch _:_ ->
no
end.
line_size(SizeL) ->
line_size(SizeL, 0, false).
line_size([], Size, false) ->
Size;
line_size([], Size, true) ->
{line,Size};
line_size([{line,Len}|SizeL], _, _) ->
line_size(SizeL, Len, true);
line_size([Sz|SizeL], SizeSoFar, LF) ->
line_size(SizeL, SizeSoFar+Sz, LF).
nsz({line,_Len}=Sz, _I) ->
Sz;
nsz(Size, I) when I >= 0 ->
{line,Size+I}.
indent({_Chars,{line,Len}}, _I) ->
Len;
indent({_Chars,Size}, I) ->
incr(I, Size).
incr(I, _Incr) when I < 0 ->
I;
incr(I, Incr) ->
I+Incr.
indentation(E, I) when I < 0 ->
chars_size(E);
indentation(E, I0) ->
I = io_lib_format:indentation(E, I0),
case has_nl(E) of
true -> {line,I};
false -> I
end.
has_nl([$\n|_]) ->
true;
has_nl([C|Cs]) when is_integer(C) ->
has_nl(Cs);
has_nl([C|Cs]) ->
has_nl(C) orelse has_nl(Cs);
has_nl([]) ->
false.
write_a_char(C, PP) ->
flat_leaf(write_char(C, PP)).
-define(MIN_SUBSTRING, 5).
write_a_string(S, I, PP) when I < 0; S =:= [] ->
flat_leaf(write_string(S, PP));
write_a_string(S, I, PP) ->
Len = erlang:max(?MAXLINE-I, ?MIN_SUBSTRING),
{list,write_a_string(S, Len, Len, PP)}.
write_a_string([], _N, _Len, _PP) ->
[];
write_a_string(S, N, Len, PP) ->
SS = string:sub_string(S, 1, N),
Sl = write_string(SS, PP),
case (length(Sl) > Len) and (N > ?MIN_SUBSTRING) of
true ->
write_a_string(S, N-1, Len, PP);
false ->
[flat_leaf(Sl) |
write_a_string(lists:nthtail(length(SS), S), Len, Len, PP)]
end.
flat_leaf(S) ->
L = lists:flatten(S),
{leaf,length(L),L}.
write_string(S, PP) ->
lists:flatten((PP#pp.string_fun)(S)).
write_char(C, PP) ->
lists:flatten((PP#pp.char_fun)(C)).
%%
%% Utilities
%%
a0() ->
erl_anno:new(0).
chars_size([C | Es]) when is_integer(C) ->
1 + chars_size(Es);
chars_size([E | Es]) ->
chars_size(E) + chars_size(Es);
chars_size([]) ->
0;
chars_size(B) when is_binary(B) ->
byte_size(B).
-define(N_SPACES, 30).
spacetab() ->
{[_|L],_} = mapfoldl(fun(_, A) -> {A,[$\s|A]}
end, [], lists:seq(0, ?N_SPACES)),
list_to_tuple(L).
spaces(N, T) when N =< ?N_SPACES ->
element(N, T);
spaces(N, T) ->
[element(?N_SPACES, T)|spaces(N-?N_SPACES, T)].
wordtable() ->
L = [begin {leaf,Sz,S} = leaf(W), {S,Sz} end ||
W <- [" ->"," =","<<",">>","[]","after","begin","case","catch",
"end","fun","if","of","receive","try","when"," ::","..",
" |"]],
list_to_tuple(L).
word(' ->', WT) -> element(1, WT);
word(' =', WT) -> element(2, WT);
word('<<', WT) -> element(3, WT);
word('>>', WT) -> element(4, WT);
word('[]', WT) -> element(5, WT);
word('after', WT) -> element(6, WT);
word('begin', WT) -> element(7, WT);
word('case', WT) -> element(8, WT);
word('catch', WT) -> element(9, WT);
word('end', WT) -> element(10, WT);
word('fun', WT) -> element(11, WT);
word('if', WT) -> element(12, WT);
word('of', WT) -> element(13, WT);
word('receive', WT) -> element(14, WT);
word('try', WT) -> element(15, WT);
word('when', WT) -> element(16, WT);
word(' ::', WT) -> element(17, WT);
word('..', WT) -> element(18, WT);
word(' |', WT) -> element(19, WT).