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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
% 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.
-module(couch_epi_codegen).
-export([generate/2, scan/1, parse/1, function/1, format_term/1]).
generate(ModuleName, Forms) when is_atom(ModuleName) ->
generate(atom_to_list(ModuleName), Forms);
generate(ModuleName, Forms0) ->
Forms = scan("-module(" ++ ModuleName ++ ").") ++ Forms0,
ASTForms = parse(Forms),
{ok, Mod, Bin} = compile:forms(ASTForms, [verbose, report_errors]),
{module, Mod} = code:load_binary(Mod, atom_to_list(Mod) ++ ".erl", Bin),
ok.
scan(String) ->
Exprs = [E || E <- re:split(String, "\\.\n", [{return, list}, trim])],
FormsTokens = lists:foldl(fun(Expr, Acc) ->
case erl_scan:string(Expr) of
{ok, [], _} ->
Acc;
{ok, Tokens, _} ->
[{Expr, fixup_terminator(Tokens)} | Acc]
end
end, [], Exprs),
lists:reverse(FormsTokens).
parse(FormsTokens) ->
ASTForms = lists:foldl(fun(Tokens, Forms) ->
{ok, AST} = parse_form(Tokens),
[AST | Forms]
end, [], FormsTokens),
lists:reverse(ASTForms).
format_term(Data) ->
lists:flatten(io_lib:format("~w", [Data])).
parse_form(Tokens) ->
{Expr, Forms} = split_expression(Tokens),
case erl_parse:parse_form(Forms) of
{ok, AST} -> {ok, AST};
{error,{_,_, Reason}} ->
{error, Expr, Reason}
end.
split_expression({Expr, Forms}) -> {Expr, Forms};
split_expression(Tokens) ->
{Exprs, Forms} = lists:unzip(Tokens),
{string:join(Exprs, "\n"), lists:append(Forms)}.
function(Clauses) ->
[lists:flatten(Clauses)].
fixup_terminator(Tokens) ->
case lists:last(Tokens) of
{dot, _} -> Tokens;
{';', _} -> Tokens;
Token ->
Line = line(Token),
Tokens ++ [{dot, Line}]
end.
-ifdef(pre18).
line(Token) ->
{line, Line} = erl_scan:token_info(Token, line),
Line.
-else.
line(Token) ->
erl_scan:line(Token).
-endif.
|
