From 13001396de1e16150d0ad29e5843cbace19875b0 Mon Sep 17 00:00:00 2001 From: Feng Date: Sun, 11 Oct 2015 03:03:07 +0800 Subject: [PATCH 1/2] cli --- src/emqttd_auth_username.erl | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/src/emqttd_auth_username.erl b/src/emqttd_auth_username.erl index 1c05d4e59..795546754 100644 --- a/src/emqttd_auth_username.erl +++ b/src/emqttd_auth_username.erl @@ -33,7 +33,7 @@ -include("emqttd_cli.hrl"). %% CLI callbacks --export([useradd/1, userdel/1]). +-export([cli/1]). -behaviour(emqttd_auth_mod). @@ -51,17 +51,15 @@ %%% CLI %%%============================================================================= -useradd([Username, Password]) -> +cli(["add", Username, Password]) -> ?PRINT("~p~n", [add_user(list_to_binary(Username), list_to_binary(Password))]); -useradd(_) -> - ?PRINT_CMD("useradd ", "add user"). - -userdel([Username]) -> +cli(["del", Username]) -> ?PRINT("~p~n", [remove_user(list_to_binary(Username))]); -userdel(_) -> - ?PRINT_CMD("userdel ", "delete user"). +cli(_) -> + ?USAGE([{"users add ", "add user"}, + {"users del ", "delete user"}]). %%%============================================================================= %%% API @@ -88,8 +86,7 @@ init(Opts) -> {disc_copies, [node()]}, {attributes, record_info(fields, ?AUTH_USERNAME_TAB)}]), mnesia:add_table_copy(?AUTH_USERNAME_TAB, node(), ram_copies), - emqttd_ctl:register_cmd(useradd, {?MODULE, useradd}, []), - emqttd_ctl:register_cmd(userdel, {?MODULE, userdel}, []), + emqttd_ctl:register_cmd(users, {?MODULE, cli}, []), {ok, Opts}. check(#mqtt_client{username = undefined}, _Password, _Opts) -> From 87b52454949cef300d05902ed9c91f3e4dd148f9 Mon Sep 17 00:00:00 2001 From: Feng Date: Sun, 11 Oct 2015 04:02:33 +0800 Subject: [PATCH 2/2] rm space --- src/emqttd_auth_username.erl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/emqttd_auth_username.erl b/src/emqttd_auth_username.erl index 795546754..39fa08a6a 100644 --- a/src/emqttd_auth_username.erl +++ b/src/emqttd_auth_username.erl @@ -51,7 +51,7 @@ %%% CLI %%%============================================================================= -cli(["add", Username, Password]) -> +cli(["add", Username, Password]) -> ?PRINT("~p~n", [add_user(list_to_binary(Username), list_to_binary(Password))]); cli(["del", Username]) ->