emqx4.3/apps/emqx_lua_hook/src/emqx_lua_hook_cli.erl

%%--------------------------------------------------------------------
%% Copyright (c) 2020 EMQ Technologies Co., Ltd. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%%     http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%--------------------------------------------------------------------

-module(emqx_lua_hook_cli).

-export([ load/0
        , cmd/1
        , unload/0
        ]).

-include("emqx_lua_hook.hrl").
-include_lib("luerl/src/luerl.hrl").

-define(PRINT(Format, Args), io:format(Format, Args)).
-define(PRINT_CMD(Cmd, Descr), io:format("~-48s# ~s~n", [Cmd, Descr])).
-define(USAGE(CmdList), [?PRINT_CMD(Cmd, Descr) || {Cmd, Descr} <- CmdList]).

load() ->
    emqx_ctl:register_command(luahook, {?MODULE, cmd}, []).

unload() ->
    emqx_ctl:unregister_command(luahook).

cmd(["load", Script]) ->
    case emqx_lua_hook:load_script(fullname(Script)) of
        ok -> emqx_ctl:print("Load ~p successfully~n", [Script]);
        error -> emqx_ctl:print("Load ~p error~n", [Script])
    end;

cmd(["reload", Script]) ->
    FullName = fullname(Script),
    emqx_lua_hook:unload_script(FullName),
    case emqx_lua_hook:load_script(FullName) of
        ok -> emqx_ctl:print("Reload ~p successfully~n", [Script]);
        error -> emqx_ctl:print("Reload ~p error~n", [Script])
    end;

cmd(["unload", Script]) ->
    emqx_lua_hook:unload_script(fullname(Script)),
    emqx_ctl:print("Unload ~p successfully~n", [Script]);

cmd(["enable", Script]) ->
    FullName = fullname(Script),
    case file:rename(fullnamedisable(Script), FullName) of
        ok -> case emqx_lua_hook:load_script(FullName) of
                  ok ->
                      emqx_ctl:print("Enable ~p successfully~n", [Script]);
                  error ->
                      emqx_ctl:print("Fail to enable ~p~n", [Script])
              end;
        {error, Reason} ->
            emqx_ctl:print("Fail to enable ~p due to ~p~n", [Script, Reason])
    end;

cmd(["disable", Script]) ->
    FullName = fullname(Script),
    emqx_lua_hook:unload_script(FullName),
    case file:rename(FullName, fullnamedisable(Script)) of
        ok ->
            emqx_ctl:print("Disable ~p successfully~n", [Script]);
        {error, Reason} ->
            emqx_ctl:print("Fail to disable ~p due to ~p~n", [Script, Reason])
    end;

cmd(_) ->
    emqx_ctl:usage([{"luahook load <Script>",    "load lua script into hook"},
                    {"luahook unload <Script>",  "unload lua script from hook"},
                    {"luahook reload <Script>",  "reload lua script into hook"},
                    {"luahook enable <Script>",  "enable lua script and load it into hook"},
                    {"luahook disable <Script>", "unload lua script out of hook and disable it"}]).

fullname(Script) ->
    filename:join([emqx_lua_hook:lua_dir(), Script]).
fullnamedisable(Script) ->
    fullname(Script)++".x".