From a1e8d3c3dc45b68953b0f9100f78650345fae350 Mon Sep 17 00:00:00 2001
From: Badlop <badlop@process-one.net>
Date: Fri, 17 Feb 2017 12:37:42 +0100
Subject: [PATCH] Commands documentation lack arguments when policy=user
 (#1514)

---
 src/ejabberd_commands_doc.erl | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/ejabberd_commands_doc.erl b/src/ejabberd_commands_doc.erl
index f34449011..fc921c066 100644
--- a/src/ejabberd_commands_doc.erl
+++ b/src/ejabberd_commands_doc.erl
@@ -403,13 +403,20 @@ generate_html_output(File, RegExp, Languages) ->
     Cmds3 = lists:sort(fun(#ejabberd_commands{name=N1}, #ejabberd_commands{name=N2}) ->
                                N1 =< N2
                        end, Cmds2),
+    Cmds4 = [maybe_add_policy_arguments(Cmd) || Cmd <- Cmds3],
     Langs = binary:split(Languages, <<",">>, [global]),
-    Out = lists:map(fun(C) -> gen_doc(C, true, Langs) end, Cmds3),
+    Out = lists:map(fun(C) -> gen_doc(C, true, Langs) end, Cmds4),
     {ok, Fh} = file:open(File, [write]),
     io:format(Fh, "~s", [[html_pre(), Out, html_post()]]),
     file:close(Fh),
     ok.
 
+maybe_add_policy_arguments(#ejabberd_commands{args=Args1, policy=user}=Cmd) ->
+    Args2 = [{user, binary}, {server, binary} | Args1],
+    Cmd#ejabberd_commands{args = Args2};
+maybe_add_policy_arguments(Cmd) ->
+    Cmd.
+
 generate_md_output(File, RegExp, Languages) ->
     Cmds = find_commands_definitions(),
     {ok, RE} = re:compile(RegExp),