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
86
87
88
|
-module(cc).
-export([epp/1, epp/2, dbg/1, dbg/2, cvr/1, cvr/2]).
-export([p/2, pp/2]).
%% epp(Module) - Creates Module.epp which contains all forms of Module
%% as obtained by using epp.
%%
%% dbg(Module) - Creates Module.dbg which contains all forms of Module
%% as obtained by using beam_lib:chunks/2.
%%
%% cvr(Module) - Creates Module.cvr which contains all forms of Module
%% as obtained by using cover:transform/3.
%%
epp(Module) ->
epp(Module, p).
epp(Module, P) ->
File = atom_to_list(Module)++".erl",
{ok,Cwd} = file:get_cwd(),
{ok, Fd1} = epp:open(File, [Cwd], []),
{ok, Fd2} = file:open(atom_to_list(Module)++".epp", write),
epp(Fd1, Fd2, P),
epp:close(Fd1),
file:close(Fd2),
ok.
epp(Fd1, Fd2, P) ->
case epp:parse_erl_form(Fd1) of
{ok, {attribute,Line,Attr,Data}} ->
epp(Fd1, Fd2, P);
{ok, Form} when P==p ->
io:format(Fd2, "~p.~n", [Form]),
epp(Fd1, Fd2, P);
{ok, Form} when P==pp ->
io:format(Fd2, "~p.~n", [erl_pp:form(Form)]),
epp(Fd1, Fd2, P);
{eof, Line} ->
ok
end.
cvr(Module) ->
cvr(Module, p).
cvr(Module, P) ->
case beam_lib:chunks(Module, [abstract_code]) of
{ok, {Module, [{abstract_code, no_abstract_code}]}} ->
{error, {no_debug_info,Module}};
{ok, {Module, [{abstract_code, {Vsn, Forms}}]}} ->
Vars = {vars,Module,Vsn, [],
undefined, undefined, undefined, undefined, undefined,
undefined,
false},
{ok, TForms, _Vars2} = cover:transform(Forms, [], Vars),
File = atom_to_list(Module)++".cvr",
apply(?MODULE, P, [File, TForms]);
Error ->
Error
end.
dbg(Module) ->
dbg(Module, p).
dbg(Module, P) ->
case beam_lib:chunks(Module, [abstract_code]) of
{ok, {Module, [{abstract_code, no_abstract_code}]}} ->
{error, {no_debug_info,Module}};
{ok, {Module, [{abstract_code, {Vsn, Forms}}]}} ->
File = atom_to_list(Module)++".dbg",
apply(?MODULE, P, [File, Forms]);
Error ->
Error
end.
p(File, Forms) ->
{ok, Fd} = file:open(File, write),
lists:foreach(fun(Form) ->
io:format(Fd, "~p.~n", [Form])
end,
Forms),
file:close(Fd).
pp(File, Forms) ->
{ok, Fd} = file:open(File, write),
lists:foreach(fun(Form) ->
io:format(Fd, "~s", [erl_pp:form(Form)])
end,
Forms),
file:close(Fd).
|
