ephp

-module(ephp_tracer).

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

-behaviour(gen_server).

-include("ephp.hrl").

-export([start_link/0, start_link/1, stop/0, calls/1, erase/1]).
% gen_server callbacks
-export([init/1, handle_call/3, handle_cast/2, handle_info/2, code_change/3,
         terminate/2]).

start_link() ->
    Output =
        fun(#stack_trace{file = File,
                         line = Line,
                         args = Args} =
                StackTrace) ->
           io:format("=> ~s:~b\n\tName => ~s\n\tArgs => ~p~n",
                     [File, Line, key_name(StackTrace), Args])
        end,
    start_link(Output).

start_link(Output) ->
    Args = [Output],
    gen_server:start_link({local, ?MODULE}, ?MODULE, Args, []).

stop() ->
    gen_server:stop(?MODULE).

calls(Name) ->
    gen_server:cast(?MODULE, {calls, Name}).

erase(Name) ->
    gen_server:cast(?MODULE, {erase, Name}).

init([OutputFun]) ->
    {ok, #{filters => [], output => OutputFun}}.

handle_cast(#stack_trace{} = StackTrace, #{filters := Filters} = State) ->
    Options =
        [{function, key_name(StackTrace)}, {file, filename:absname(StackTrace#stack_trace.file)}],
    Check = fun(Filter) -> lists:member(Filter, Options) end,
    case lists:any(Check, Filters) of
        true ->
            #{output := OutputFun} = State,
            OutputFun(StackTrace);
        false ->
            ok
    end,
    {noreply, State};
handle_cast({erase, Name}, #{filters := Filters} = State) ->
    {noreply, State#{filters => Filters -- [Name]}};
handle_cast({calls, Name}, #{filters := Filters} = State) ->
    case lists:member(Name, Filters) of
        true ->
            {noreply, State};
        false ->
            {noreply, State#{filters => [Name | Filters]}}
    end.

handle_call(_Msg, _From, State) ->
    {reply, ok, State}.

handle_info(_Info, State) ->
    {noreply, State}.

code_change(_OldVsn, State, _Extra) ->
    {ok, State}.

terminate(_Reason, _State) ->
    ok.

key_name(#stack_trace{type = undefined, function = Fun}) ->
    ephp_data:to_bin(Fun);
key_name(#stack_trace{type = <<"::">>,
                      class = Class,
                      function = Fun}) ->
    <<Class/binary, "::", (ephp_data:to_bin(Fun))/binary>>;
key_name(#stack_trace{type = <<"->">>,
                      class = Class,
                      function = Fun}) ->
    <<Class/binary, "->", (ephp_data:to_bin(Fun))/binary>>.