%% This Source Code Form is subject to the terms of the Mozilla Public
%% License, v. 2.0. If a copy of the MPL was not distributed with this
%% file, You can obtain one at https://mozilla.org/MPL/2.0/.
%%
%% Copyright (c) 2021-2022 VMware, Inc. or its affiliates. All rights reserved.
%% Copyright (c) 2021-2022 Viacheslav Katsuba.
%%
-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()) -> [binary() | list()].
overview(Element, Options) ->
Overview = edoc_layout:overview(Element, Options),
patch_html(Overview).
-spec type(term()) -> term().
type(Element) ->
edoc_layout:type(Element).
-spec run(#doclet_gen{}, tuple()) -> ok | no_return().
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),
case file:write_file(File, Content2) of
ok -> ok;
{error, Reason} -> exit({error, Reason})
end.
-spec patch_html(list()) -> list().
patch_html(Html) ->
Html2 = re:replace(
Html,
"",
"\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(), binary(), list()) -> list().
add_toc(App, Html, Dir) ->
case generate_toc(Dir, App) of
undefined ->
Html;
Toc ->
re:replace(
Html,
"(Modules
)",
Toc ++ "\\1",
[{return, list}])
end.
-spec generate_toc([binary() | list()], term()) -> [binary() | list()] | undefined.
generate_toc(Dir, App) ->
case get_overview(Dir) of
undefined ->
undefined;
Overview ->
Lines = re:split(Overview, "\\n", [{return, list}]),
Titles = [Line ||
Line <- Lines,
match =:= re:run(Line, "^=+.*=+$", [{capture, none}])],
generate_toc1(Titles, 0, [], App)
end.
-spec generate_toc1([binary() | list()], integer(), list(), term()) -> [binary() | 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() | undefined.
get_overview(Dir) ->
OverviewFile = filename:join(Dir, "overview.edoc"),
case file:read_file(OverviewFile) of
{ok, Overview} ->
Overview;
{error, _} ->
undefined
end.