proton-mail-android/app/src/main/java/ch/protonmail/android/activities/BaseActivity.java

/*
 * Copyright (c) 2020 Proton Technologies AG
 *
 * This file is part of ProtonMail.
 *
 * ProtonMail is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ProtonMail is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with ProtonMail. If not, see https://www.gnu.org/licenses/.
 */
package ch.protonmail.android.activities;

import android.app.Activity;
import android.app.AlertDialog;
import android.content.BroadcastReceiver;
import android.content.Context;
import android.content.Intent;
import android.content.IntentFilter;
import android.graphics.Color;
import android.os.Bundle;
import android.os.Handler;
import android.os.SystemClock;
import android.text.TextUtils;
import android.text.method.LinkMovementMethod;
import android.view.LayoutInflater;
import android.view.View;
import android.view.WindowManager;
import android.widget.Button;
import android.widget.TextView;
import android.widget.Toast;

import androidx.annotation.Nullable;
import androidx.appcompat.app.AppCompatActivity;
import androidx.appcompat.app.AppCompatDelegate;
import androidx.appcompat.widget.Toolbar;
import androidx.work.WorkManager;

import com.birbit.android.jobqueue.JobManager;
import com.google.android.material.snackbar.Snackbar;
import com.squareup.otto.Subscribe;

import javax.inject.Inject;

import butterknife.BindView;
import butterknife.ButterKnife;
import ch.protonmail.android.R;
import ch.protonmail.android.activities.composeMessage.ComposeMessageActivity;
import ch.protonmail.android.activities.messageDetails.MessageDetailsActivity;
import ch.protonmail.android.adapters.swipe.SwipeProcessor;
import ch.protonmail.android.api.NetworkConfigurator;
import ch.protonmail.android.api.ProtonMailApiManager;
import ch.protonmail.android.api.models.MailSettings;
import ch.protonmail.android.api.models.User;
import ch.protonmail.android.api.segments.event.AlarmReceiver;
import ch.protonmail.android.bl.Html5Handler;
import ch.protonmail.android.bl.HtmlDivHandler;
import ch.protonmail.android.bl.HtmlProcessor;
import ch.protonmail.android.bl.XHtmlHandler;
import ch.protonmail.android.core.BigContentHolder;
import ch.protonmail.android.core.Constants;
import ch.protonmail.android.core.NetworkResults;
import ch.protonmail.android.core.ProtonMailApplication;
import ch.protonmail.android.core.QueueNetworkUtil;
import ch.protonmail.android.core.UserManager;
import ch.protonmail.android.events.ForceSwitchedAccountNotifier;
import ch.protonmail.android.events.LogoutEvent;
import ch.protonmail.android.events.MessageSentEvent;
import ch.protonmail.android.events.Status;
import ch.protonmail.android.jobs.FetchMailSettingsJob;
import ch.protonmail.android.jobs.organizations.GetOrganizationJob;
import ch.protonmail.android.jobs.payments.GetPaymentMethodsJob;
import ch.protonmail.android.settings.pin.ValidatePinActivity;
import ch.protonmail.android.utils.AppUtil;
import ch.protonmail.android.utils.CustomLocale;
import ch.protonmail.android.utils.INetworkConfiguratorCallback;
import ch.protonmail.android.utils.UiUtil;
import ch.protonmail.android.utils.extensions.TextExtensions;
import ch.protonmail.android.worker.FetchUserInfoWorker;
import dagger.hilt.android.AndroidEntryPoint;
import timber.log.Timber;

import static ch.protonmail.android.receivers.VerificationOnSendReceiver.EXTRA_MESSAGE_ADDRESS_ID;
import static ch.protonmail.android.receivers.VerificationOnSendReceiver.EXTRA_MESSAGE_ID;
import static ch.protonmail.android.receivers.VerificationOnSendReceiver.EXTRA_MESSAGE_INLINE;
import static ch.protonmail.android.settings.pin.ValidatePinActivityKt.EXTRA_FRAGMENT_TITLE;
import static ch.protonmail.android.settings.pin.ValidatePinActivityKt.EXTRA_LOGOUT;
import static ch.protonmail.android.settings.pin.ValidatePinActivityKt.EXTRA_PIN_VALID;
import static ch.protonmail.android.worker.FetchUserInfoWorkerKt.FETCH_USER_INFO_WORKER_NAME;
import static ch.protonmail.android.worker.FetchUserInfoWorkerKt.FETCH_USER_INFO_WORKER_RESULT;

@AndroidEntryPoint
public abstract class BaseActivity extends AppCompatActivity implements INetworkConfiguratorCallback {

    public static final String EXTRA_IN_APP = "extra_in_app";
    public static final int REQUEST_CODE_VALIDATE_PIN = 998;

    protected static boolean mPingHasConnection;

    @Inject
    protected ProtonMailApplication mApp;
    @Inject
    protected ProtonMailApiManager mApi;
    @Inject
    protected NetworkConfigurator networkConfigurator;
    @Inject
    protected UserManager mUserManager;
    @Inject
    protected JobManager mJobManager;
    @Inject
    protected QueueNetworkUtil mNetworkUtil;
    @Inject
    protected SwipeProcessor mSwipeProcessor;
    @Inject
    protected HtmlProcessor mHtmlProcessor;
    @Inject
    protected BigContentHolder mBigContentHolder;
    @Inject
    protected NetworkResults mNetworkResults;
    @Inject
    protected WorkManager workManager;
    @Inject
    protected FetchUserInfoWorker.Enqueuer fetchUserInfoWorkerEnqueuer;

    @Nullable
    @BindView(R.id.toolbar)
    protected Toolbar mToolbar;
    @Nullable
    @BindView(R.id.layout_no_connectivity_info)
    protected View mConnectivitySnackLayout;
    @Nullable
    @BindView(R.id.screenProtector)
    protected View mScreenProtectorLayout;

    private BroadcastReceiver mLangReceiver = null;
    private boolean inApp = false;
    private boolean checkForPin = true;
    private String mCurrentLocale;
    protected Snackbar mDraftedMessageSnack;
    protected Snackbar mRequestTimeoutSnack;

    private AlertDialog alertDelinquency;
    protected boolean mPinValid = false;
    private boolean shouldLock = false;

    protected abstract int getLayoutId();

    protected boolean shouldCheckForAutoLogout() {
        return true;
    }

    protected boolean isDohOngoing = false;
    protected boolean autoRetry = true;

    static {
        AppCompatDelegate.setCompatVectorFromResourcesEnabled(true);
    }

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        ProtonMailApplication.getApplication().setAppInBackground(false);
        inApp = getIntent().getBooleanExtra(EXTRA_IN_APP, false);
        if (savedInstanceState != null) {
            mCurrentLocale = savedInstanceState.getString("curr_loc");
            if (mCurrentLocale != null && !mCurrentLocale.equals(getResources().getConfiguration().locale.toString())) {
                inApp = false;
            }
        }
        mCurrentLocale = ProtonMailApplication.getApplication().getCurrentLocale();
        buildHtmlProcessor();

        setContentView(getLayoutId());
        handlePin();
        ButterKnife.bind(this);
        if (mToolbar != null) {
            try {
                setSupportActionBar(mToolbar);
            } catch (Exception e) {
                // Samsung Android 4.2 only issue
                // read more here: https://github.com/google/iosched/issues/79
                // and here: https://code.google.com/p/android/issues/detail?id=78377
            }
        }

        UiUtil.setStatusBarColor(this, getResources().getColor(R.color.dark_purple_statusbar));

        ForceSwitchedAccountNotifier.notifier.observe(this, event -> {
            if (event != null) {
                AppUtil.postEventOnUi(event);
                if (!(this instanceof NavigationActivity)) {
                    onBackPressed();
                }
            }
        });
    }

    @Override
    protected void onNewIntent(Intent intent) {
        super.onNewIntent(intent);
        inApp = intent.getBooleanExtra(EXTRA_IN_APP, false);
    }

    @Override
    public void onBackPressed() {
        if (!(this instanceof SplashActivity)) {
            saveLastInteraction();
        }
        finish();
    }

    @Override
    protected void onRestart() {
        super.onRestart();
        validationCanceled = true;
        if (checkForPin) {
            handlePin();
        }
    }

    protected boolean isPreventingScreenshots() {
        return false;
    }

    protected boolean secureContent() {
        return false;
    }

    protected void enableScreenshotProtector() { }

    protected void disableScreenshotProtector() { }

    @Override
    protected void onResume() {
        super.onResume();

        // Enable secure mode if screenshots are disabled, else disable it
        if (isPreventingScreenshots() || mUserManager.getUser().isPreventTakingScreenshots()) {
            getWindow().addFlags(WindowManager.LayoutParams.FLAG_SECURE);
        } else {
            getWindow().clearFlags(WindowManager.LayoutParams.FLAG_SECURE);
        }
        if (shouldLock && (secureContent() || mUserManager.getUser().isPreventTakingScreenshots())) {
            enableScreenshotProtector();
        }
        ProtonMailApplication.getApplication().setAppInBackground(false);
        NetworkConfigurator.Companion.setNetworkConfiguratorCallback(this);
    }

    @Override
    protected void onPause() {
        ProtonMailApplication.getApplication().setAppInBackground(true);
        super.onPause();
    }

    @Override
    protected void onSaveInstanceState(Bundle outState) {
        outState.putString("curr_loc", mCurrentLocale);
        super.onSaveInstanceState(outState);
    }

    protected void loadMailSettings() {
        mUserManager.setMailSettings(MailSettings.Companion.load(mUserManager.getUsername()));
        if (mUserManager.getMailSettings() == null) {
            mJobManager.addJobInBackground(new FetchMailSettingsJob());
        }
    }

    protected boolean isAutoShowRemoteImages() {
        return mUserManager.getMailSettings() != null && (mUserManager.getMailSettings().getShowImages() == 1 || mUserManager.getMailSettings().getShowImages() == 3);
    }

    protected boolean isAutoShowEmbeddedImages() {
        return mUserManager.getMailSettings() != null && (mUserManager.getMailSettings().getShowImages() == 2 || mUserManager.getMailSettings().getShowImages() == 3);
    }

    private void shouldLock() {
        User user = mUserManager.getUser();
        long diff = user.getLastInteractionDiff();

        if(user.isUsePin()) {
            if (diff >= 0) {
                shouldLock = user.shouldPINLockTheApp(diff);
            } else {
                shouldLock = true;
            }
        }
    }

    private void handlePin() {
        ProtonMailApplication.getApplication().setAppInBackground(false);
        mApp.setCurrentActivity(this);
        shouldLock();
        if (!shouldCheckForAutoLogout()) {
            return;
        }

        checkPinLock(shouldLock);
    }

    protected void checkPinLock(boolean shouldLock) {
        if (inApp || mPinValid) {
            this.shouldLock = false;
            mPinValid = false;
            return;
        }
        if (shouldLock) {
            Intent validatePinIntent = new Intent(BaseActivity.this, ValidatePinActivity.class);
            if (this instanceof MessageDetailsActivity) {
                validatePinIntent.putExtra(EXTRA_FRAGMENT_TITLE, R.string.enter_pin_message_details);
            }
            Intent pinIntent = AppUtil.decorInAppIntent(validatePinIntent);
            startActivityForResult(pinIntent, REQUEST_CODE_VALIDATE_PIN);
        } else {
            this.shouldLock = false;
        }
    }

    private boolean validationCanceled = true;

    @Override
    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
        if ((resultCode == RESULT_OK || resultCode == RESULT_CANCELED) && requestCode == REQUEST_CODE_VALIDATE_PIN) {
            if (data == null) {
                return;
            }
            if (resultCode == RESULT_CANCELED) {
                checkForPin = false;
            }
            boolean isValid = data.getBooleanExtra(EXTRA_PIN_VALID, false);
            if (!isValid) {
                boolean logout = data.getBooleanExtra(EXTRA_LOGOUT, false);
                if (logout) {
                    mUserManager.logoutLastActiveAccount();
                    AppUtil.postEventOnUi(new LogoutEvent(Status.SUCCESS));
                } else {
                    validationCanceled = true;
                    finish();
                }
            } else {
                if (this instanceof ValidatePinActivity) {
                    validationCanceled = false;
                }
            }
            shouldLock = false;
            disableScreenshotProtector();
        } else {
            super.onActivityResult(requestCode, resultCode, data);
        }
    }

    @Override
    protected void onStart() {
        super.onStart();
        ProtonMailApplication.getApplication().setAppInBackground(false);
        new Handler().postDelayed(this::deactivateScreenProtector, 500);
    }

    @Override
    protected void onStop() {
        super.onStop();
        // Enable secure mode for hide content from recent if pin is enabled, else disable it so
        // content will be visible in recent
        if (mUserManager.getUser().isUsePin())
            getWindow().addFlags(WindowManager.LayoutParams.FLAG_SECURE);
        else getWindow().clearFlags(WindowManager.LayoutParams.FLAG_SECURE);

        if (!validationCanceled && !(this instanceof ValidatePinActivity)) {
            saveLastInteraction();
        }
        if (!(this instanceof AddAttachmentsActivity)) {
            inApp = false;
            activateScreenProtector();
        }
    }

    private void deactivateScreenProtector() {
        if (mScreenProtectorLayout != null) {
            mScreenProtectorLayout.setVisibility(View.GONE);
        }
    }

    private void activateScreenProtector() {
        if (mUserManager.getUser().isUsePin()) {
            if (mScreenProtectorLayout != null) {
                mScreenProtectorLayout.setVisibility(View.VISIBLE);
            }
        }
    }

    protected void saveLastInteraction() {
        if (mUserManager.isLoggedIn()) {
            User user = mUserManager.getUser();
            user.setLastInteraction(SystemClock.elapsedRealtime());
        }
    }

    protected void checkDelinquency() {
        User user = mUserManager.getUser();
        if (user.getDelinquent() && (alertDelinquency == null || !alertDelinquency.isShowing())) {
            final AlertDialog.Builder builder = new AlertDialog.Builder(this);
            LayoutInflater inflater = this.getLayoutInflater();
            View dialogView = inflater.inflate(R.layout.layout_delinquency_dialog, null, false);
            TextView subtitle = dialogView.findViewById(R.id.subtitle);
            subtitle.setMovementMethod(LinkMovementMethod.getInstance());
            Button btnClose = dialogView.findViewById(R.id.close_app);
            Button btnCheckAgain = dialogView.findViewById(R.id.recheck);
            Button btnLogout = dialogView.findViewById(R.id.logout);

            btnLogout.setOnClickListener(v -> {
                mUserManager.logoutOffline();
                finish();
            });
            btnClose.setOnClickListener(v -> finish());
            btnCheckAgain.setOnClickListener(v -> {
                fetchUserInfoWorkerEnqueuer.invoke();
                workManager.getWorkInfosForUniqueWorkLiveData(FETCH_USER_INFO_WORKER_NAME)
                        .observe(this, workInfo -> {
                            boolean isDelinquent = workInfo.get(0).getOutputData().getBoolean(FETCH_USER_INFO_WORKER_RESULT, true);
                            if (!isDelinquent && alertDelinquency != null && alertDelinquency.isShowing()) {
                                alertDelinquency.dismiss();
                            }
                        });
            });
            builder.setView(dialogView);
            alertDelinquency = builder.create();
            alertDelinquency.setCanceledOnTouchOutside(false);
            alertDelinquency.setCancelable(false);
            if (!isFinishing()) {
                alertDelinquency.show();
            }
        }
    }

    public void showRequestTimeoutSnack() {
        mRequestTimeoutSnack = Snackbar.make(mConnectivitySnackLayout, getString(R.string.request_timeout), Snackbar.LENGTH_LONG);
        View view = mRequestTimeoutSnack.getView();
        TextView tv = view.findViewById(com.google.android.material.R.id.snackbar_text);
        tv.setTextColor(Color.WHITE);
        mRequestTimeoutSnack.show();
    }

    protected void fetchOrganizationData() {
        GetPaymentMethodsJob paymentMethodsJob = new GetPaymentMethodsJob();
        mJobManager.addJobInBackground(paymentMethodsJob);
        if (mUserManager.getUser().isPaidUser()) {
            GetOrganizationJob getOrganizationJob = new GetOrganizationJob();
            mJobManager.addJobInBackground(getOrganizationJob);
        } else {
            ProtonMailApplication.getApplication().setOrganization(null);
        }
    }

    protected final BroadcastReceiver humanVerificationBroadcastReceiver = new BroadcastReceiver() {

        @Override
        public void onReceive(final Context context, Intent intent) {
            Bundle extras = intent.getExtras();
            String errorText = getString(R.string.message_drafted);
            if (intent.hasExtra(Constants.ERROR)) {
                String extraText = intent.getStringExtra(Constants.ERROR);
                if (!TextUtils.isEmpty(extraText)) {
                    errorText = extraText;
                }
            }
            final String messageId = extras.getString(EXTRA_MESSAGE_ID);
            final boolean messageInline = extras.getBoolean(EXTRA_MESSAGE_INLINE);
            final String messageAddressId = extras.getString(EXTRA_MESSAGE_ADDRESS_ID);

            if (mConnectivitySnackLayout == null) {
                return;
            }
            mDraftedMessageSnack = Snackbar.make(mConnectivitySnackLayout, errorText, Snackbar.LENGTH_INDEFINITE);
            View view = mDraftedMessageSnack.getView();
            TextView tv = view.findViewById(com.google.android.material.R.id.snackbar_text);
            tv.setTextColor(Color.WHITE);
            mDraftedMessageSnack.setAction(getString(R.string.verify), v -> {
                mDraftedMessageSnack.dismiss();
                final Intent composeIntent = new Intent(BaseActivity.this, ComposeMessageActivity.class);
                composeIntent.putExtra(ComposeMessageActivity.EXTRA_MESSAGE_ID, messageId);
                composeIntent.putExtra(ComposeMessageActivity.EXTRA_MESSAGE_RESPONSE_INLINE, messageInline);
                composeIntent.putExtra(ComposeMessageActivity.EXTRA_MESSAGE_ADDRESS_ID, messageAddressId);
                composeIntent.putExtra(ComposeMessageActivity.EXTRA_VERIFY, true);
                startActivity(composeIntent);
            });
            mDraftedMessageSnack.setActionTextColor(getResources().getColor(R.color.icon_purple));
            mDraftedMessageSnack.show();

            setResultCode(Activity.RESULT_OK);
            abortBroadcast();
        }
    };

    @Subscribe
    public void onMessageSentEvent(MessageSentEvent event){
        switch (event.getStatus()) {
            case SUCCESS:
                TextExtensions.showToast(this, R.string.message_sent, Toast.LENGTH_SHORT);
                AlarmReceiver alarmReceiver = new AlarmReceiver();
                alarmReceiver.setAlarm(this, true);
                break;
            case FAILED:
                TextExtensions.showToast(this, R.string.message_failed, Toast.LENGTH_SHORT);
                break;
        }
    }

    protected BroadcastReceiver setupLangReceiver(){
        if (mLangReceiver == null) {
            mLangReceiver = new BroadcastReceiver() {
                @Override
                public void onReceive(Context context, Intent intent) {
                    // noop
                }
            };

            IntentFilter filter = new IntentFilter(Intent.ACTION_LOCALE_CHANGED);
            registerReceiver(mLangReceiver, filter);
        }

        return mLangReceiver;
    }

    @Override
    protected void attachBaseContext(Context base) {
        super.attachBaseContext(CustomLocale.Companion.apply(base));
    }

    private void buildHtmlProcessor() {
        mHtmlProcessor.addHandler(new HtmlDivHandler());
        mHtmlProcessor.addHandler(new Html5Handler());
        mHtmlProcessor.addHandler(new XHtmlHandler());
    }

    @Override
    public void onApiProvidersChanged() {
        networkConfigurator.refreshDomainsAsync();
    }

    @Override
    public void startDohSignal () {
        isDohOngoing = true;
        Timber.d("BaseActivity: startDohSignal");
    }

    @Override
    public void stopDohSignal () {
        isDohOngoing = false;
        Timber.d("BaseActivity: stopDohSignal");
    }

    @Override
    public void startAutoRetry() {
        autoRetry = true;
    }

    @Override
    public void stopAutoRetry() {
        autoRetry = false;
    }
}