erlup

%% @copyright 2016 Hinagiku Soranoba All Rights Reserved.
%%
%% @doc A escript to support the upgrade and downgrade of OTP application.
%%
-module(erlup).

-include("erlup.hrl").

%%----------------------------------------------------------------------------------------------------------------------
%% Exported API
%%----------------------------------------------------------------------------------------------------------------------
-export([
         main/1, init/1, format_error/1,
         escript_opt_specs/1,
         do_task/3
        ]).

%%----------------------------------------------------------------------------------------------------------------------
%% Exported Functions
%%----------------------------------------------------------------------------------------------------------------------

%% rebar3 plugins.
%%
%% @private
-spec init(rebar_state:t()) -> {ok, rebar_state:t()}.
init(State0) ->
    Mods = [
            erlup_appup,
            erlup_relup
           ],
    lists:foldl(fun(Mod, {ok, State}) -> Mod:init(State) end, {ok, State0}, Mods).

%% escript.
%%
%% @private
main(Args) ->
    init_log(),
    case getopt:parse(escript_opt_specs(), Args) of
        {ok, {Options, _}} ->
            DoHelp  = proplists:get_value(help, Options, false),
            Task = proplists:get_value(task, Options, ""),
            case escript_opt_specs(Task) of
                [] ->
                    ?ERROR("Task ~s not found", [Task]), halt(1);
                Specs when DoHelp; Task =:= "" ->
                    case Task =:= "" of
                        true ->
                            getopt:usage(Specs, "erlup"),
                            providers:help(rebar_state:providers(element(2, init(rebar_state:new()))));
                        false ->
                            getopt:usage(Specs, "erlup" ++ " " ++ Task)
                    end,
                    halt(0);
                _ ->
                    try
                        do_task(Task, Options)
                    catch
                        throw:{error, {Module, Err}} when is_atom(Module) ->
                            ?ERROR("~s", [Module:format_error(Err)]),
                            halt(1)
                    end
            end;
        {error, Reason} ->
            ?ERROR("~s", [getopt:format_error(escript_opt_specs(), Reason)]),
            halt(1)
    end,
    halt(0).

%% @private
-spec format_error(iodata()) -> iolist().
format_error(Reason) ->
    io_lib:format("~s", [Reason]).

%%----------------------------------------------------------------------------------------------------------------------
%% Internal Functions
%%----------------------------------------------------------------------------------------------------------------------

-spec init_log() -> ok.
init_log() ->
    LogLevel = case os:getenv("QUIET") of
                   false ->
                       DefaultLevel = rebar_log:default_level(),
                       case os:getenv("DEBUG") of
                           false -> DefaultLevel;
                           _     -> DefaultLevel + 3
                       end;
                   _ ->
                       rebar_log:error_level()
               end,
    rebar_log:init(command_line, LogLevel).

-spec do_task(string(), [{atom(), term()}]) -> ok.
do_task(Task, Options) ->
    ConfPath  = proplists:get_value(conf, Options),
    ErlupConf = case file:consult(ConfPath) of
                    {ok, Terms} ->
                        case proplists:lookup(erlup, Terms) of
                            none            -> Terms;
                            {_, ErlupConf0} -> ErlupConf0
                        end;
                    {error, Reason} ->
                        ?WARN("~s", [file:format_error(Reason) ++ " " ++ ConfPath]),
                        []
                end,
    do_task(Task, Options, erlup_state:new(ErlupConf)).

-spec do_task(string(), [{atom(), term()}], erlup_state:t()) -> ok.
do_task(Task, Options, State) when Task =:= "appup"; Task =:= "relup" ->
    CurrentVsn0   = proplists:get_value(current, Options),
    PreviousVsns0 = proplists:get_all_values(previous, Options),
    Dirs0         = proplists:get_all_values(dir, Options),
    Dirs          = ?IIF(Dirs0 =:= [], ["."], Dirs0),

    CurrentVsn = ?IIF(CurrentVsn0 =:= undefined, default_current_vsn(hd(Dirs)), CurrentVsn0),
    ?IF(PreviousVsns0 =:= [CurrentVsn], ?throw("Current and previous are the same")),

    PreviousVsns = ?IIF(PreviousVsns0 =:= [], default_previous_vsns(Dirs, CurrentVsn), PreviousVsns0),
    case Task of
        "appup" -> erlup_appup:do(Dirs, PreviousVsns, CurrentVsn, State);
        "relup" -> erlup_relup:do(Dirs, PreviousVsns, CurrentVsn, State)
    end.

-spec default_current_vsn(file:filename()) -> string().
default_current_vsn(Dir) ->
    case erlup_utils:lookup_current_vsn(Dir) of
        {ok, CurrentVsn} ->
            ?INFO("Current vsn = ~p", [CurrentVsn]),
            CurrentVsn;
        {error, Reason}  ->
            ?throw(Reason)
    end.

-spec default_previous_vsns([file:filename()], string()) -> [string()].
default_previous_vsns(Dirs, CurrentVsn) ->
    Rels    = erlup_utils:find_rels(Dirs),
    RelName = case lists:keyfind(CurrentVsn, 2, Rels) of
                  false           -> ?throw("Can not find a rel file. (" ++ CurrentVsn ++ ")");
                  {RelName0, _, _} -> RelName0
              end,

    Vsns = [Vsn || {N, Vsn, _} <- Rels, RelName =:= N],
    case erlup_utils:split(erlup_utils:sort_vsns(Vsns), CurrentVsn) of
        {[], _}           -> ?throw("Can not find previous vsns");
        {PreviousVsns, _} -> PreviousVsns
    end.

-spec escript_opt_specs() -> [getopt:option_spec()].
escript_opt_specs() ->
    [
     {    task, undefined,  undefined,                 string, "Task to run"},
     {    help,        $h,  undefined,              undefined, "Display this help"},
     {    conf, undefined,     "conf", {string, "erlup.conf"}, "Path of configuration file"},
     { current,        $c,  undefined,                 string, "Vsn of current release"},
     {previous,        $p,  undefined,                 string, "Vsns of previous release. (e.g. -p 0.0.1 -p 0.0.2)"},
     {     dir,        $d,      "dir",                 string, "Release directories (e.g. -d _rel/${APP} -d /tmp/${APP})"}
    ].

-spec escript_opt_specs(string()) -> [getopt:option_spec()].
escript_opt_specs(Task) ->
    lists:filter(fun(X) -> lists:member(element(1, X), task_opts(Task)) end,
                 escript_opt_specs()).

-spec task_opts(Task :: string()) -> [OptKey :: atom()].
task_opts("appup") -> [current, previous, conf, dir, help];
task_opts("relup") -> [current, previous, conf, dir, help];
task_opts("")      -> [help, task, conf];
task_opts(_)       -> []. % not supported