-module(rebar3_edoc_extensions_wrapper).
-include_lib("edoc/include/edoc_doclet.hrl").
-include("rebar3_edoc_extensions.hrl").
-export([module/2, overview/2, type/1]).
-export([run/2]).
-spec module(term(), term()) -> term().
module(Element, Options) ->
edoc_layout:module(Element, Options).
-spec overview(term(), term()) -> term().
overview(Element, Options) ->
Overview = edoc_layout:overview(Element, Options),
patch_html(Overview).
-spec type(term()) -> term().
type(Element) ->
edoc_layout:type(Element).
-spec run(list(), term()) -> list().
run(#doclet_gen{app = App} = Cmd, Ctxt) ->
ok = edoc_doclet:run(Cmd, Ctxt),
%% Ctxt is a #context{} record in Erlang 23 and #doclet_context{} in Erlang
%% 24. The directory is the second field in that record in both cases.
Dir = element(2, Ctxt),
File = filename:join(Dir, "modules-frame.html"),
{ok, Content0} = file:read_file(File),
Content1 = add_toc(App, Content0, Dir),
Content2 = patch_html(Content1),
ok = copy_static_files(Dir),
case file:write_file(File, Content2) of
ok -> ok;
{error, Reason} -> exit({error, Reason})
end.
-spec patch_html(list()) -> list().
patch_html(Html) ->
Html1 = re:replace(
Html,
"",
?SYNTAX_HIGHLIGHTING_CSS "\n"
?ADDITIONAL_STYLE "\n"
"",
[{return, list}]),
Html2 = re:replace(
Html1,
"",
"\n"
?SYNTAX_HIGHLIGHTING_JS,
[{return, list}]),
Html3 = re:replace(
Html2,
"(.*)
",
"\\1
",
[{return, list}, ungreedy, dotall, global]),
Html4 = re:replace(
Html3,
" +(" ?LANG_REGEX ")(?:\n)(.*)
",
"\\2
",
[{return, list}, ungreedy, dotall, global]),
Html4.
-spec add_toc(term(), list(), list()) -> list().
add_toc(App, Html, Dir) ->
Toc = generate_toc(Dir, App),
re:replace(
Html,
"(Modules
)",
Toc ++ "\\1",
[{return, list}]).
-spec generate_toc(list(), term()) -> list().
generate_toc(Dir, App) ->
Overview = get_overview(Dir),
Lines = re:split(Overview, "\\n", [{return, list}]),
Titles = [Line ||
Line <- Lines,
match =:= re:run(Line, "^=+.*=+$", [{capture, none}])],
generate_toc1(Titles, 0, [], App).
-spec generate_toc1(list(), integer(), list(), term()) -> list().
generate_toc1([Title | Rest], CurrentLevel, Result, App) ->
ReOpts = [{capture, all_but_first, list}],
{match, [Equals, Title1]} = re:run(Title, "^(=+) *(.*)", ReOpts),
Title2 = re:replace(Title1, " *=+$", "", [{return, list}]),
Anchor = "#" ++ re:replace(Title2, " ", "_", [{return, list}, global]),
Link = "" ++ Title2 ++ "",
Level = length(Equals) - 1,
if
Level =:= CurrentLevel ->
Result1 = Result ++ ["\n", "", Link],
generate_toc1(Rest, CurrentLevel, Result1, App);
CurrentLevel =:= 0 andalso
Level =:= CurrentLevel + 1 ->
OverviewLink = "Overview",
Result1 = Result ++ ["\n",
"- ", OverviewLink, "
\n",
"- ", Link],
generate_toc1(Rest, Level, Result1, App);
Level =:= CurrentLevel + 1 ->
Result1 = Result ++ ["\n", "
\n", "- ", Link],
generate_toc1(Rest, Level, Result1, App);
Level < CurrentLevel ->
Result1 = Result ++ ["
\n"
"
\n"
|| _ <- lists:seq(Level, CurrentLevel - 1)],
Result2 = Result1 ++ [" \n", "- ", Link],
generate_toc1(Rest, Level, Result2, App)
end;
generate_toc1([], CurrentLevel, Result, App) ->
Name = atom_to_list(App),
["
", Name, "
\n",
Result,
" \n",
["
\n" || _ <- lists:seq(0, CurrentLevel - 1)]].
-spec get_overview(list()) -> binary().
get_overview(Dir) ->
OverviewFile = filename:join(Dir, "overview.edoc"),
case file:read_file(OverviewFile) of
{ok, Overview} ->
Overview;
_ ->
NewDir = code:priv_dir(rebar3_edoc_extensions),
OverviewFile2 = filename:join(NewDir, "overview.edoc"),
{ok, Overview} = file:read_file(OverviewFile2),
Overview
end.
-spec copy_static_files(list()) -> ok.
copy_static_files(Dir) ->
PrivDir = code:priv_dir(rebar3_edoc_extensions),
PrismJSPathPriv = filename:join(PrivDir, "prism.js"),
PrismCSSPathPriv = filename:join(PrivDir, "prism.css"),
GithubMarkdownCSSPathPriv = filename:join(PrivDir, "github-markdown.css"),
PrismJSPath = filename:join(Dir, "prism.js"),
PrismCSSPath = filename:join(Dir, "prism.css"),
GithubMarkdownCSSPath = filename:join(Dir, "github-markdown.css"),
{ok, _} = file:copy(PrismJSPathPriv, PrismJSPath),
{ok, _} = file:copy(PrismCSSPathPriv, PrismCSSPath),
{ok, _} = file:copy(GithubMarkdownCSSPathPriv, GithubMarkdownCSSPath),
ok.