%% @author Arjan Scherpenisse <arjan@miraclethings.nl>
%% @copyright 2014-2017 Arjan Scherpenisse
%%
%% @doc Compilation of Zotonic files

%% 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(zotonic_filehandler_compile).
-author("Arjan Scherpenisse <arjan@miraclethings.nl>").

-export([
    start/0,

    all/0,
    all_sync/0,
    all_task/1,

    recompile/1,
    recompile_task/1,
    compile_options/1,

    run_cmd/1,
    run_cmd/3,
    run_cmd_task/3,
    run_cmd_task_buffalo/3,

    ld/0,
    ld/1,
    code_path_check/1
]).

-include_lib("kernel/include/logger.hrl").

%% @doc Compile all files. Called from zotonic-compile script
-dialyzer({no_return, start/0}).
start() ->
    Node = get_node_argument(),
    Result = case net_adm:ping(Node) of
                 pang -> all_sync();
                 pong -> rpc:call(Node, ?MODULE, all_sync, [], infinity)
             end,
    halt_with_result(Result).

halt_with_result(ok) -> halt(0);
halt_with_result(error) -> halt(1);
halt_with_result({badrpc, Reason}) ->
    io:fwrite(standard_error, "Could not compile: ~p.~n", [Reason]),
    halt(1).

get_node_argument() ->
    case init:get_argument(node) of
        error ->
            io:fwrite(standard_error, "No node option set.~n", []),
            halt(1);
        {ok, [[NodeArg]]} ->
            list_to_atom(NodeArg)
    end.

%% @doc Load all changed beam files, return list of reloaded modules.
-spec ld() -> list( code:load_ret() ).
ld() ->
    Ms = reloader:all_changed(),
    [ ld(M) || M <- Ms ].

%% @doc Load a specific beam file, reattach any observers etc.
-spec ld(module()) -> code:load_ret().
ld(Module) when is_atom(Module) ->
    code:purge(Module),
    case code:load_file(Module) of
        {error, Reason} = Error ->
            ?LOG_ERROR(#{
                text => <<"Error loading module">>,
                in => zotonic_core,
                module => Module,
                reason => Reason
            }),
            Error;
        {module, _} = Ok ->
            z_sites_manager:module_loaded(Module),
            Ok
    end.

%% @doc Check if the application is present in the current code path
-spec code_path_check( binary() ) -> boolean().
code_path_check(AppFilename) ->
    Path = filename:dirname( filename:dirname( AppFilename )),
    CurrentAppVsn = code_path_to_appvsn(),
    case maybe_add_path(Path, CurrentAppVsn) of
        {true, AppVsn} ->
            zotonic_filewatcher_sup:restart_watchers(),
            zotonic_filehandler:terminal_notifier("New application: "++z_convert:to_list(AppVsn)),
            true;
        false ->
            % App (with this vsn) exists in the code path
            false
    end.

%% @doc List all applications (with opt version number) as binaries.
-spec code_path_to_appvsn() -> list( binary() ).
code_path_to_appvsn() ->
    lists:map(
        fun(Path) ->
            z_convert:to_binary(filename:basename(filename:dirname(Path)))
        end,
        code:get_path()).

%% @doc Check if a new application (with optional vsn) directory needs to be added to the code path.
%%      Return 'true' if the code path has been changed.
-spec maybe_add_path( file:filename_all(), list( binary() )) -> false | {true, binary()}.
maybe_add_path(AppDir, Paths) ->
    AppVsn = z_convert:to_binary(filename:basename(AppDir)),
    case lists:member(AppVsn, Paths) of
        true ->
            false;
        false ->
            AppBeam = filename:join([ AppDir, "ebin" ]),
            code:add_patha(unicode:characters_to_list(AppBeam)),
            {true, AppVsn}
    end.


%% @doc Compile all files
-spec all() -> ok.
all() ->
    QueueOptions = #{
        timeout => 200,
        deadline => 10000,
        is_drop_running => true
    },
    {ok, _} = buffalo:queue(compile_all_task, {?MODULE, all_task, [ undefined ]}, QueueOptions),
    ok.

all_sync() ->
    QueueOptions = #{
        timeout => 10,
        deadline => 100,
        is_drop_running => true
    },
    {ok, _} = buffalo:queue(compile_all_task, {?MODULE, all_task, [ self() ]}, QueueOptions),
    receive
        {all_task_result, Result} ->
            Result
    end.

all_task( OptPid ) ->
    jobs:run(zotonic_filehandler_single_job, fun() -> do_all_task( OptPid ) end).

do_all_task( OptPid ) ->
    Cmd = case os:getenv("ZOTONIC") of
        false ->
            "./rebar3 compile";
        ZotonicDir ->
            lists:flatten([
                "cd ", z_filelib:os_filename(ZotonicDir),
                "; ./rebar3 compile"
            ])
    end,
    ?LOG_DEBUG(#{
        in => zotonic_core,
        text => <<"Compile all: start">>
    }),
    zotonic_filehandler:terminal_notifier("Compile all: start"),
    Result = run_cmd_task(Cmd, [], []),
    case Result of
        {error, _} ->
            zotonic_filehandler:terminal_notifier("Compile all: ERROR");
        _ ->
            zotonic_filehandler:terminal_notifier("Compile all: ready")
    end,
    ?LOG_DEBUG(#{
        text => <<"Compile all: ready">>,
        in => zotonic_core,
        result => Result
    }),
    Result1 = cleanup_stdout(Result),
    case is_pid(OptPid) of
        false -> ok;
        true -> OptPid ! {all_task_result, Result1}
    end,
    Result1.

cleanup_stdout({error, Props} = E) ->
    case proplists:get_value(stdout, Props) of
        Stdout when is_list(Stdout) ->
            Stdout1 = lists:filter(
                fun
                    (<<" - cotonic ", _/binary>>) ->
                        false;
                    (Line) ->
                        case binary:match(Line, <<" failed">>) of
                            nomatch ->
                                case binary:match(Line, <<"===> ">>) of
                                    nomatch ->
                                        binary:match(Line, <<" Warning: ">>) =:= nomatch;
                                    _ ->
                                        false
                                end;
                            _ ->
                                true
                        end
                end,
                Stdout),
            lists:foreach(
                fun(Line) ->
                    ?LOG_ERROR(#{
                        in => zotonic_core,
                        text => z_string:trim(Line)
                    })
                end,
                Stdout1),
            {error, [ {stdout, Stdout1} | proplists:delete(stdout, Props) ]};
        _ ->
            E
    end;
cleanup_stdout(Result) ->
    Result.

run_cmd(Cmd) ->
    run_cmd(Cmd, [], #{}).

run_cmd(Cmd, RunOpts, Opts) when is_binary(Cmd) ->
    run_cmd(unicode:characters_to_list(Cmd, utf8), RunOpts, Opts);
run_cmd(Cmd, RunOpts, Opts) ->
    QueueOptions = #{
        timeout => 150,
        deadline => 2000,
        is_drop_running => true
    },
    buffalo:queue({?MODULE, run_cmd_task_buffalo, [ Cmd, RunOpts, Opts ]}, QueueOptions).

run_cmd_task_buffalo(Cmd, RunOpts, Opts) ->
    run_cmd_task(Cmd, RunOpts, Opts),
    ok.

run_cmd_task(Cmd, RunOpts, Opts) ->
    try
        maybe_ignore_dir(Opts, true),
        case exec:run(lists:flatten(Cmd), [ sync, stdout, stderr ] ++ RunOpts) of
            {ok, Out} ->
                StdErr = proplists:get_value(stderr, Out, []),
                case StdErr of
                    [] ->
                        ok;
                    StdErr ->
                        ?LOG_ERROR(#{
                            text => <<"Unexpected output from run of command task">>,
                            in => zotonic_core,
                            command => Cmd,
                            result => error,
                            reason => iolist_to_binary(StdErr)
                        }),
                        ok
                end;
            {error, Args} = Error when is_list(Args) ->
                StdErr = proplists:get_value(stderr, Args, []),
                StdOut = proplists:get_value(stdout, Args, []),
                case {StdErr, StdOut} of
                    {[], []} ->
                        ?LOG_ERROR(#{
                            text => <<"Unexpected error from run of command task">>,
                            in => zotonic_core,
                            command => Cmd,
                            result => error,
                            reason => Args
                        });
                    {StdErr, _} when StdErr =/= [] ->
                        ?LOG_ERROR(#{
                            text => <<"Unexpected error from run of command task">>,
                            in => zotonic_core,
                            command => Cmd,
                            result => Error,
                            reason => iolist_to_binary(StdErr)
                        });
                    {_, StdOut} ->
                        ?LOG_ERROR(#{
                            text => <<"Unexpected error from run of command task">>,
                            in => zotonic_core,
                            command => Cmd,
                            result => error,
                            reason => iolist_to_binary(StdOut)
                        })
                end,
                Error
        end
    after
        maybe_ignore_dir(Opts, false)
    end.

maybe_ignore_dir(#{ ignore_dir := Dir }, false) ->
    % Let the build dir come to rest, drop all events
    timer:sleep(1000),
    zotonic_filehandler_handler:ignore_dir(Dir, false);
maybe_ignore_dir(#{ ignore_dir := Dir }, true) ->
    zotonic_filehandler_handler:ignore_dir(Dir, true);
maybe_ignore_dir(_, _) ->
    ok.

%% When "./rebar3 compile" runs, then all .../test/*.erl file are touched.
%% This forces a recompile of all these files. We might not want to
%% automatically recompile these files during a rebar3 run.
-spec recompile(file:filename_all()) -> ok.
recompile(File) when is_binary(File) ->
    recompile(unicode:characters_to_list(File, utf8));
recompile(File) ->
    case buffalo:status(compile_all_task) of
        {error, notfound} ->
            QueueOptions = #{
                timeout => 150,
                deadline => 2000,
                is_drop_running => true
            },
            {ok, _} = buffalo:queue({?MODULE, recompile_task, [ File ]}, QueueOptions),
            ok;
        {ok, _} ->
            % rebar3 is running, ignore recompile requests
            ok
    end.

recompile_task(File) ->
    case compile_options(File) of
        {ok, Options} ->
            ?LOG_DEBUG(#{
                text => <<"Recompile of erlang file using make">>,
                in => zotonic_core,
                file => File
            }),
            zotonic_filehandler:terminal_notifier("Compiling: " ++ filename:basename(File)),
            try
                case make:files([File], Options) of
                    up_to_date ->
                        ok;
                    Other ->
                        ?LOG_WARNING(#{
                            text => <<"Recompile of Erlang file unexpected result">>,
                            in => zotonic_core,
                            result => error,
                            reason => Other,
                            file => File
                        })
                end
            catch
                Type:Err:Stack ->
                    ?LOG_WARNING(#{
                        text => <<"Recompile of Erlang file exit">>,
                        in => zotonic_core,
                        file => File,
                        result => Type,
                        reason => Err,
                        stack => Stack
                    })
            end;
        false ->
            % Might be some new OTP app, so a manual build on the top level
            % should take care of this, we don't do anything now.
            ?LOG_WARNING(#{
                text => <<"Could not find compile options, no recompile for Erlang file">>,
                in => zotonic_core,
                result => error,
                reason => no_compile_options,
                file => File
            })
    end,
    ok.

-spec compile_options(file:filename_all()) -> {ok, list()} | false.
compile_options(File) when is_binary(File) ->
    compile_options(unicode:characters_to_list(File, utf8));
compile_options(Filename) ->
    case compile_options_1(Filename) of
        {ok, Options} ->
            Options1 = [ Opt || Opt <- Options, Opt =/= error_summary ],
            {ok, Options1};
        false ->
            false
    end.

compile_options_1(Filename) ->
    case previous_compile_options(Filename) of
        false ->
            % Guess the options based on other ebin files
            guess_compile_options(Filename);
        {ok, _} = Ok ->
            Ok
    end.

guess_compile_options(Filename) ->
    case guess_compile_options_src(Filename) of
        {ok, Options} -> {ok, Options};
        false -> guess_compile_options_ebin(Filename)
    end.

guess_compile_options_src(Filename) ->
    SrcFiles = filelib:wildcard(filename:join(filename:dirname(Filename), "*.erl")),
    first(fun previous_compile_options/1, SrcFiles).

first(_F, []) -> false;
first(F, [A|As]) ->
    case F(A) of
        false -> first(F, As);
        {ok, _} = Ok -> Ok
    end.

guess_compile_options_ebin(Filename) ->
    EbinDir = ebin_path(Filename),
    case filelib:wildcard(filename:join(EbinDir, "*.beam")) of
        [] -> false;
        BeamFiles -> previous_compile_options_any(BeamFiles)
    end.

ebin_path(SrcFile) ->
    Parts = filename:split(SrcFile),
    SrcPath = lists:takewhile(fun is_not_src/1, Parts),
    Dir = case SrcPath of
        Parts -> filename:dirname(SrcFile);
        [] -> [];
        Path1 -> filename:join(Path1)
    end,
    EbinDir = filename:join([Dir, "ebin"]),
    case filelib:is_dir(EbinDir) of
        true ->
            EbinDir;
        false ->
            % Could be that we are in _checkouts or in apps
            % Check the name of module/lib which is one level
            % above the src dir.
            case SrcPath of
                Parts -> EbinDir;
                [] -> EbinDir;
                _ ->
                    AppName = lists:last(SrcPath),
                    case code:priv_dir(z_convert:to_atom(AppName)) of
                        {error, bad_name} ->
                            filename:join([z_path:build_lib_dir(), AppName, "ebin"]);
                        PrivDir ->
                            filename:join(
                                filename:dirname(PrivDir),
                                "ebin")
                    end
            end
    end.

is_not_src("src") -> false;
is_not_src(<<"src">>) -> false;
is_not_src(_) -> true.

previous_compile_options(Filename) when is_list(Filename); is_binary(Filename) ->
    previous_compile_options(z_convert:to_atom(filename:rootname(filename:basename(Filename))));
previous_compile_options(Module) when is_atom(Module) ->
    try
        MInfo = Module:module_info(compile),
        case proplists:get_value(options, MInfo) of
            undefined -> false;
            Options -> {ok, with_outdir(Options, Module)}
        end
    catch
        error:undef -> false
    end.

previous_compile_options_any([]) -> false;
previous_compile_options_any([BeamFile|BeamFiles]) ->
    case previous_compile_options(BeamFile) of
        false -> previous_compile_options_any(BeamFiles);
        {ok, Options} -> {ok, Options}
    end.

%% @doc Add outdir if it's missing
%%      This seems to be the case after rebar3's first compile.
with_outdir(Options, Module) ->
    case proplists:lookup(outdir, Options) of
        none ->
            Outdir = filename:dirname(code:which(Module)),
            [{outdir, Outdir} | Options];
        _ ->
            Options
    end.

% zotonic_ebin_dir() ->
%     filename:join([build_lib_dir(), "zotonic_core", "ebin"]).

% zotonic_subdir(Path) when is_list(Path) ->
%     case os:getenv("ZOTONIC") of
%         false -> filename:join([file:get_cwd() | Path]);
%         ZotonicDir -> filename:join([ZotonicDir | Path])
%     end.