%% @author Marc Worrell <marc@worrell.nl>
%% @copyright 2010-2020 Marc Worrell
%% @doc Display a form to sign up.

%% Copyright 2010-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(controller_signup).
-author("Marc Worrell <marc@worrell.nl>").

-export([
    process/4
]).
-export([event/2]).

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


process(_Method, _AcceptedCT, _ProvidedCT, Context) ->
    Context2 = z_context:ensure_qs(Context),
    z_context:lager_md(Context2),
    Vars = case z_context:get_q(<<"xs">>, Context2) of
        undefined ->
            [];
        <<>> ->
            [];
        Check ->
            % Set in mod_signup when fetching signup_url
            case m_server_storage:secure_lookup(Check, Context) of
                {ok, {Check, Props, SignupProps}} ->
                    [
                        {xs_props, {Props, SignupProps}}
                        | Props
                    ];
                {error, _} ->
                    []
            end
    end,
    % z_session:set(signup_xs, undefined, Context),
    Rendered = z_template:render(<<"signup.tpl">>, Vars, Context2),
    z_context:output(Rendered, Context2).


%% @doc Handle the submit of the signup form.
event(#submit{message={signup, Args}, form= <<"signup_form">>}, Context) ->
    {XsProps0,XsSignupProps} = case proplists:get_value(xs_props, Args) of
        {A,B} -> {A,B};
        undefined -> {undefined, undefined}
    end,
    XsProps = case is_list(XsProps0) of
        false -> [];
        true -> XsProps0
    end,

    %% Call listeners to fetch the required signup form fields
    FormProps0 = [
        {email, true},
        {name_first, true},
        {name_surname_prefix, false},
        {name_surname, true}
    ],
    FormProps = z_notifier:foldr(signup_form_fields, FormProps0, Context),

    Props = lists:map(fun({Prop, Validate}) ->
                              {Prop, fetch_prop(Prop, Validate, XsProps, Context)}
                      end,
                      FormProps),

    Agree = z_convert:to_bool(z_context:get_q_validated(<<"signup_tos_agree">>, Context)),
    case Agree of
        true ->
            {email, Email} = proplists:lookup(email, Props),
            RequestConfirm = case proplists:lookup(request_confirm, Args) of
                {request_confirm, RC} when RC =/= undefined -> z_convert:to_bool(RC);
                _ -> z_convert:to_bool(m_config:get_value(mod_signup, request_confirm, true, Context))
            end,
            SignupProps = case is_set(XsSignupProps) of
                true ->
                    XsSignupProps;
                false ->
                    Username = case z_convert:to_bool(m_config:get_value(mod_signup, username_equals_email, false, Context)) of
                        false -> z_string:trim(z_context:get_q_validated(<<"username">>, Context));
                        true -> Email
                    end,
                    [
                        {identity,
                            {username_pw,
                                {   Username,
                                    z_context:get_q_validated(<<"password1">>, Context)},
                                    true,
                                    true}}
                  ]
            end,
            SignupProps1 = case Email of
                "" ->
                    SignupProps;
                <<>> ->
                    SignupProps;
                _ ->
                    case has_email_identity(Email, SignupProps) of
                        false ->
                            [
                                {identity, {email, Email, false, false}}
                                | SignupProps
                            ];
                        true ->
                            SignupProps
                    end
            end,
            signup(Props, SignupProps1, RequestConfirm, Context);
        false ->
            show_errors([error_tos_agree], Context)
    end.


fetch_prop(Prop, Validated, SignupProps, Context) ->
    case proplists:get_value(Prop, SignupProps) of
        undefined ->
            V = case Validated of
                true ->
                    z_context:get_q_validated(Prop, Context);
                false ->
                    case z_context:get_q(Prop, Context) of
                        undefined when Prop =:= name_surname_prefix ->
                            z_context:get_q(surprefix, Context);
                        QV ->
                            QV
                    end
            end,
            z_string:trim(z_convert:to_binary(V));
        V ->
            V
    end.

is_set(undefined) -> false;
is_set("") -> false;
is_set(<<>>) -> false;
is_set(_) -> true.

has_email_identity(_Email, []) -> false;
has_email_identity(Email, [ {identity, {email, Email, _, _}} | _ ]) -> true;
has_email_identity(Email, [ _ | Rest ]) -> has_email_identity(Email, Rest).


%% @doc Sign up a new user. Check if the identity is available.
signup(Props, SignupProps, RequestConfirm, Context) ->
    UserId = proplists:get_value(user_id, SignupProps),
    SignupProps1 = proplists:delete(user_id, SignupProps),
    case mod_signup:signup_existing(UserId, Props, SignupProps1, RequestConfirm, Context) of
        {ok, NewUserId} ->
            handle_confirm(NewUserId, SignupProps1, RequestConfirm, Context);
        {error, {identity_in_use, username}} ->
            show_errors([error_duplicate_username], Context);
        {error, {identity_in_use, _}} ->
            show_errors([error_duplicate_identity], Context);
        {error, #context{} = ContextError} ->
            show_errors([error_signup], ContextError);
        {error, _Reason} ->
            show_errors([error_signup], Context)
    end.


%% Handle sending a confirm, or redirect to the 'ready_page' location
handle_confirm(UserId, SignupProps, RequestConfirm, Context) ->
    case not RequestConfirm orelse m_identity:is_verified(UserId, Context) of
        true ->
            ensure_published(UserId, z_acl:sudo(Context)),
            {ok, ContextUser} = z_auth:logon(UserId, Context),
            Location = case get_redirect_page(SignupProps) of
                <<>> -> m_signup:confirm_redirect(ContextUser);
                Url -> Url
            end,
            % Post a onetime-token to the auth worker on the page
            % The auth worker will exchange it for a valid cookie and then perform
            % the redirect to the url.
            case z_authentication_tokens:encode_onetime_token(UserId, ContextUser) of
                {ok, Token} ->
                    AuthMsg = #{
                        token => Token,
                        url => Location
                    },
                    z_mqtt:publish(<<"~client/model/auth/post/onetime-token">>, AuthMsg, Context),
                    z_render:wire({mask, []}, Context);
                {error, _} = Error ->
                    lager:error("Error making onetime token: ~p", [ Error ]),
                    show_errors([internal], Context)
            end;
        false ->
            % User is not yet verified, send a verification message to the user's external identities
            case mod_signup:request_verification(UserId, Context) of
                {error, no_verifiable_identities} ->
                    % Problem, no email address or other identity that could be verified
                    show_errors([error_need_verification], Context);
                ok ->
                    % Show feedback that we sent a confirmation message
                    Context1 = show_errors([], Context),
                    z_render:update(
                        "signup_logon_box",
                        z_template:render("_signup_stage.tpl",
                                          [ {email, m_rsc:p_no_acl(UserId, email, Context1)}],
                                          Context1),
                        Context1)
            end
    end.

get_redirect_page(SignupProps) ->
    z_convert:to_binary(proplists:get_value(ready_page, SignupProps, <<>>)).


ensure_published(UserId, Context) ->
    case m_rsc:p(UserId, is_published, Context) of
        true -> {ok, UserId};
        false -> m_rsc:update(UserId, #{ <<"is_published">> => true }, Context)
    end.


show_errors(Errors, Context) ->
    Errors1 = [ z_convert:to_list(E) || E <- Errors ],
    z_render:wire({add_class, [{target, "signup_logon_box"}, {class, Errors1}]}, Context).