-module(ephp_lib_control).

-author('manuel@altenwald.com').

-behaviour(ephp_lib).

-export([init_func/0, init_config/0, init_const/0, handle_error/3, include/3,
         include_once/3, require/3, require_once/3]).

-include("ephp.hrl").

-spec init_func() -> ephp_lib:php_function_results().
init_func() ->
    [{include, [{args, [string]}]},
     {include_once, [{args, [string]}]},
     {require, [{args, [string]}]},
     {require_once, [{args, [string]}]}].

-spec init_config() -> ephp_lib:php_config_results().
init_config() ->
    [{<<"include_path">>, <<".:">>}].

-spec init_const() -> ephp_lib:php_const_results().
init_const() ->
    [].

-spec handle_error(ephp_error:error_type(), ephp_error:error_level(), Args :: term()) ->
                      string() | ignore.
handle_error(erequired, _Level, {ReqFile, Func}) ->
    IncludePath = ephp_config:get(<<"include_path">>, <<".:">>),
    io_lib:format("~s(): Failed opening required '~s' (include_path='~s')",
                  [Func, ReqFile, IncludePath]);
handle_error(einclude, _Level, {ReqFile, Func}) ->
    IncludePath = ephp_config:get(<<"include_path">>, <<".:">>),
    io_lib:format("~s(): Failed opening '~s' for inclusion (include_path='~s')",
                  [Func, ReqFile, IncludePath]);
handle_error(_Type, _Level, _Args) ->
    ignore.

-spec include(ephp:context_id(), line(), InclFile :: var_value()) -> any().
include(Context, Line, {_Var, InclFile}) ->
    File = ephp_context:get_active_file(Context),
    case ephp_context:load(Context, InclFile) of
        {error, _} ->
            NoFileData = {InclFile, <<"include">>},
            ephp_error:handle_error(Context, {error, enofile, Line, File, ?E_WARNING, NoFileData}),
            ephp_error:handle_error(Context, {error, einclude, Line, File, ?E_WARNING, NoFileData}),
            undefined;
        Code ->
            include_file(Context, Code, InclFile)
    end.

-spec include_once(ephp:context_id(), line(), File :: var_value()) -> any().
include_once(Context, Line, {_Var, InclFile}) ->
    File = ephp_context:get_active_file(Context),
    case ephp_context:load_once(Context, InclFile) of
        {error, _} ->
            NoFileData = {InclFile, <<"include_once">>},
            ephp_error:handle_error(Context, {error, enofile, Line, File, ?E_WARNING, NoFileData}),
            ephp_error:handle_error(Context, {error, einclude, Line, File, ?E_WARNING, NoFileData}),
            undefined;
        {return, true} ->
            true;
        Code ->
            include_file(Context, Code, InclFile)
    end.

-spec require(ephp:context_id(), line(), File :: var_value()) -> any().
require(Context, Line, {_Var, ReqFile}) ->
    File = ephp_context:get_active_file(Context),
    case ephp_context:load(Context, ReqFile) of
        {error, _} ->
            NoFileData = {ReqFile, <<"require">>},
            ephp_error:handle_error(Context, {error, enofile, Line, File, ?E_WARNING, NoFileData}),
            ephp_error:error({error, erequired, Line, ?E_ERROR, NoFileData});
        Code ->
            include_file(Context, Code, ReqFile)
    end.

-spec require_once(ephp:context_id(), line(), File :: var_value()) -> any().
require_once(Context, Line, {_Var, ReqFile}) ->
    File = ephp_context:get_active_file(Context),
    case ephp_context:load_once(Context, ReqFile) of
        {error, _} ->
            NoFileData = {ReqFile, <<"require_once">>},
            ephp_error:handle_error(Context, {error, enofile, Line, File, ?E_WARNING, NoFileData}),
            ephp_error:error({error, erequired, Line, ?E_ERROR, NoFileData});
        {return, true} ->
            true;
        Code ->
            include_file(Context, Code, ReqFile)
    end.

%% ----------------------------------------------------------------------------
%% Internal functions
%% ----------------------------------------------------------------------------

include_file(Context, Code, File) ->
    OldValue = ephp_context:get_active_file(Context),
    ephp_context:set_active_file(Context, File),
    {ok, Res} = ephp_interpr:process(Context, Code),
    ephp_context:set_active_file(Context, OldValue),
    case Res of
        {return, Value} ->
            Value;
        _ ->
            undefined
    end.