%% @author Marc Worrell <marc@worrell.nl>
%% @copyright 2020 Marc Worrell
%% @doc Toggle database trace on/off

%% Copyright 2020 Marc Worrell
%%
%% 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(z_development_dbtrace).

-export([
    event/2,

    set_tracing/2,
    copy_from_session/1,
    is_tracing/1
    ]).

-include_lib("zotonic_core/include/zotonic.hrl").

event(#postback{ message=dbtrace_toggle }, Context) ->
    IsTracing = z_convert:to_bool( z_context:get_q(<<"triggervalue">>, Context) ),
    set_tracing(IsTracing, Context),
    case IsTracing of
        true ->
            z_render:growl(?__("Enabled database query tracing", Context), Context);
        false ->
            z_render:growl(?__("Disabled database query tracing", Context), Context)
    end.

set_tracing(IsTracing, Context) ->
    storage_store(IsTracing, Context),
    erlang:put(is_dbtrace, IsTracing).

copy_from_session(Context) ->
    Tracing = z_convert:to_bool( storage_lookup(Context) ),
    erlang:put(is_dbtrace, Tracing).

is_tracing(Context) ->
    case erlang:get(is_dbtrace) of
        true ->
            true;
        false ->
            false;
        _ ->
            Tracing = storage_lookup(Context),
            erlang:put(is_dbtrace, Tracing),
            Tracing
    end.

storage_lookup(Context) ->
    case z_notifier:first({server_storage, secure_lookup, is_dbtrace, undefined}, Context) of
        {ok, V} -> z_convert:to_bool(V);
        {error, _} -> false;
        undefined -> false
    end.

storage_store(V, Context) ->
    z_notifier:first({server_storage, secure_store, is_dbtrace, V}, Context).