comand line parsting for rabbitmq-plugin

1 files changed, 33 insertions, 2 deletions

diff --git a/src/rabbit_plugin.erl b/src/rabbit_plugin.erl
index 6d94cee73a..b271652bee 100644
--- a/src/rabbit_plugin.erl
+++ b/src/rabbit_plugin.erl
@@ -19,6 +19,8 @@
 
 -export([start/0, stop/0]).
 
+-define(FORCE_OPT, "-f").
+
 %%----------------------------------------------------------------------------
 
 -ifdef(use_specs).
@@ -31,11 +33,40 @@
 %%----------------------------------------------------------------------------
 
 start() ->
-    io:format("Rabbitmq-plugin, GO!~n"),
+    {[Command0 | Args], Opts} =
+        case rabbit_misc:get_options([{flag, ?FORCE_OPT}],
+                                     init:get_plain_arguments()) of
+            {[], _Opts}    -> usage();
+            CmdArgsAndOpts -> CmdArgsAndOpts
+        end,
+    Command = list_to_atom(Command0),
 
-    rabbit_misc:quit(0).
+    case catch action(Command, Args, Opts) of
+        ok ->
+            rabbit_misc:quit(0);
+        {'EXIT', {function_clause, [{?MODULE, action, _} | _]}} ->
+            print_error("invalid command '~s'",
+                        [string:join([atom_to_list(Command) | Args], " ")]),
+            usage();
+        {error, Reason} ->
+            print_error("~p", [Reason]),
+            rabbit_misc:quit(2);
+        Other ->
+            print_error("~p", [Other]),
+            rabbit_misc:quit(2)
+    end.
 
 stop() ->
     ok.
 
+print_error(Format, Args) ->
+    rabbit_misc:format_stderr("Error: " ++ Format ++ "~n", Args).
+
+usage() ->
+    io:format("Insert USAGE here.~n"),
+    rabbit_misc:quit(1).
+
 %%----------------------------------------------------------------------------
+
+action(test, [], _Opts) ->
+    io:format("Test ok~n").