path: root/input/autofill/AutofillFramework/afservice/src/main/java/com/example/android/autofill/service/settings/SettingsActivity.java

/*
 * Copyright (C) 2017 The Android Open Source Project
 *
 * 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.
 */
package com.example.android.autofill.service.settings;

import static com.example.android.autofill.service.util.Util.DalCheckRequirement.AllUrls;
import static com.example.android.autofill.service.util.Util.DalCheckRequirement.Disabled;
import static com.example.android.autofill.service.util.Util.DalCheckRequirement.LoginOnly;
import static com.example.android.autofill.service.util.Util.logd;
import static com.example.android.autofill.service.util.Util.logw;

import android.content.Context;
import android.content.DialogInterface;
import android.content.Intent;
import android.content.SharedPreferences;
import android.net.Uri;
import android.os.Bundle;
import android.provider.Settings;
import android.support.design.widget.Snackbar;
import android.support.v7.app.AlertDialog;
import android.support.v7.app.AppCompatActivity;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.view.autofill.AutofillManager;
import android.widget.CompoundButton;
import android.widget.EditText;
import android.widget.ImageView;
import android.widget.NumberPicker;
import android.widget.RadioGroup;
import android.widget.Switch;
import android.widget.TextView;

import com.example.android.autofill.service.R;
import com.example.android.autofill.service.data.AutofillDataBuilder;
import com.example.android.autofill.service.data.DataCallback;
import com.example.android.autofill.service.data.FakeAutofillDataBuilder;
import com.example.android.autofill.service.data.source.DefaultFieldTypesSource;
import com.example.android.autofill.service.data.source.PackageVerificationDataSource;
import com.example.android.autofill.service.data.source.local.DefaultFieldTypesLocalJsonSource;
import com.example.android.autofill.service.data.source.local.LocalAutofillDataSource;
import com.example.android.autofill.service.data.source.local.SharedPrefsPackageVerificationRepository;
import com.example.android.autofill.service.data.source.local.dao.AutofillDao;
import com.example.android.autofill.service.data.source.local.db.AutofillDatabase;
import com.example.android.autofill.service.model.DatasetWithFilledAutofillFields;
import com.example.android.autofill.service.model.FieldTypeWithHeuristics;
import com.example.android.autofill.service.util.AppExecutors;
import com.example.android.autofill.service.util.Util;
import com.google.gson.GsonBuilder;

import java.util.List;

public class SettingsActivity extends AppCompatActivity {
    private static final String TAG = "SettingsActivity";
    private static final int REQUEST_CODE_SET_DEFAULT = 1;
    private AutofillManager mAutofillManager;
    private LocalAutofillDataSource mLocalAutofillDataSource;
    private PackageVerificationDataSource mPackageVerificationDataSource;
    private MyPreferences mPreferences;
    private String mPackageName;

    @Override
    public void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.multidataset_service_settings_activity);
        SharedPreferences localAfDataSourceSharedPrefs =
                getSharedPreferences(LocalAutofillDataSource.SHARED_PREF_KEY, Context.MODE_PRIVATE);
        DefaultFieldTypesSource defaultFieldTypesSource =
                DefaultFieldTypesLocalJsonSource.getInstance(getResources(),
                        new GsonBuilder().create());
        AutofillDao autofillDao = AutofillDatabase.getInstance(
                this, defaultFieldTypesSource, new AppExecutors()).autofillDao();
        mPackageName = getPackageName();
        mLocalAutofillDataSource = LocalAutofillDataSource.getInstance(localAfDataSourceSharedPrefs,
                autofillDao, new AppExecutors());
        mAutofillManager = getSystemService(AutofillManager.class);
        mPackageVerificationDataSource =
                SharedPrefsPackageVerificationRepository.getInstance(this);
        mPreferences = MyPreferences.getInstance(this);
        setupSettingsSwitch(R.id.settings_auth_responses_container,
                R.id.settings_auth_responses_label,
                R.id.settings_auth_responses_switch,
                mPreferences.isResponseAuth(),
                (compoundButton, isResponseAuth) -> mPreferences.setResponseAuth(isResponseAuth));
        setupSettingsSwitch(R.id.settings_auth_datasets_container,
                R.id.settings_auth_datasets_label,
                R.id.settings_auth_datasets_switch,
                mPreferences.isDatasetAuth(),
                (compoundButton, isDatasetAuth) -> mPreferences.setDatasetAuth(isDatasetAuth));
        setupSettingsButton(R.id.settings_add_data_container,
                R.id.settings_add_data_label,
                R.id.settings_add_data_icon,
                (view) -> buildAddDataDialog().show());
        setupSettingsButton(R.id.settings_clear_data_container,
                R.id.settings_clear_data_label,
                R.id.settings_clear_data_icon,
                (view) -> buildClearDataDialog().show());
        setupSettingsButton(R.id.settings_auth_credentials_container,
                R.id.settings_auth_credentials_label,
                R.id.settings_auth_credentials_icon,
                (view) -> {
                    if (mPreferences.getMainPassword() != null) {
                        buildCurrentCredentialsDialog().show();
                    } else {
                        buildNewCredentialsDialog().show();
                    }
                });
        setupSettingsSwitch(R.id.settingsSetServiceContainer,
                R.id.settingsSetServiceLabel,
                R.id.settingsSetServiceSwitch,
                mAutofillManager.hasEnabledAutofillServices(),
                (compoundButton, serviceSet) -> setService(serviceSet));
        RadioGroup loggingLevelContainer = findViewById(R.id.loggingLevelContainer);
        Util.LogLevel loggingLevel = mPreferences.getLoggingLevel();
        Util.setLoggingLevel(loggingLevel);
        switch (loggingLevel) {
            case Off:
                loggingLevelContainer.check(R.id.loggingOff);
                break;
            case Debug:
                loggingLevelContainer.check(R.id.loggingDebug);
                break;
            case Verbose:
                loggingLevelContainer.check(R.id.loggingVerbose);
                break;
        }
        loggingLevelContainer.setOnCheckedChangeListener((group, checkedId) -> {
            switch (checkedId) {
                case R.id.loggingOff:
                    mPreferences.setLoggingLevel(Util.LogLevel.Off);
                    break;
                case R.id.loggingDebug:
                    mPreferences.setLoggingLevel(Util.LogLevel.Debug);
                    break;
                case R.id.loggingVerbose:
                    mPreferences.setLoggingLevel(Util.LogLevel.Verbose);
                    break;
            }
        });
        RadioGroup dalCheckRequirementContainer = findViewById(R.id.dalCheckRequirementContainer);
        Util.DalCheckRequirement dalCheckRequirement = mPreferences.getDalCheckRequirement();
        switch (dalCheckRequirement) {
            case Disabled:
                dalCheckRequirementContainer.check(R.id.dalDisabled);
                break;
            case LoginOnly:
                dalCheckRequirementContainer.check(R.id.dalLoginOnly);
                break;
            case AllUrls:
                dalCheckRequirementContainer.check(R.id.dalAllUrls);
                break;
        }
        dalCheckRequirementContainer.setOnCheckedChangeListener((group, checkedId) -> {
            switch (checkedId) {
                case R.id.dalDisabled:
                    mPreferences.setDalCheckRequired(Disabled);
                    break;
                case R.id.dalLoginOnly:
                    mPreferences.setDalCheckRequired(LoginOnly);
                    break;
                case R.id.dalAllUrls:
                    mPreferences.setDalCheckRequired(AllUrls);
                    break;
            }
        });
    }

    private AlertDialog buildClearDataDialog() {
        return new AlertDialog.Builder(SettingsActivity.this)
                .setMessage(R.string.settings_clear_data_confirmation)
                .setTitle(R.string.settings_clear_data_confirmation_title)
                .setNegativeButton(R.string.settings_cancel, null)
                .setPositiveButton(R.string.settings_ok, (dialog, which) -> {
                    mLocalAutofillDataSource.clear();
                    mPackageVerificationDataSource.clear();
                    mPreferences.clearCredentials();
                    dialog.dismiss();
                })
                .create();
    }

    private AlertDialog buildAddDataDialog() {
        NumberPicker numberOfDatasetsPicker = LayoutInflater
                .from(SettingsActivity.this)
                .inflate(R.layout.multidataset_service_settings_add_data_dialog, null)
                .findViewById(R.id.number_of_datasets_picker);
        numberOfDatasetsPicker.setMinValue(0);
        numberOfDatasetsPicker.setMaxValue(10);
        numberOfDatasetsPicker.setWrapSelectorWheel(false);
        return new AlertDialog.Builder(SettingsActivity.this)
                .setTitle(R.string.settings_add_data_title)
                .setNegativeButton(R.string.settings_cancel, null)
                .setMessage(R.string.settings_select_number_of_datasets)
                .setView(numberOfDatasetsPicker)
                .setPositiveButton(R.string.settings_ok, (dialog, which) -> {
                    int numOfDatasets = numberOfDatasetsPicker.getValue();
                    mPreferences.setNumberDatasets(numOfDatasets);
                    mLocalAutofillDataSource.getFieldTypes(new DataCallback<List<FieldTypeWithHeuristics>>() {
                        @Override
                        public void onLoaded(List<FieldTypeWithHeuristics> fieldTypes) {
                            boolean saved = buildAndSaveMockedAutofillFieldCollections(
                                    fieldTypes, numOfDatasets);
                            dialog.dismiss();
                            if (saved) {
                                Snackbar.make(findViewById(R.id.settings_layout),
                                        getResources().getQuantityString(
                                                R.plurals.settings_add_data_success,
                                                numOfDatasets, numOfDatasets),
                                        Snackbar.LENGTH_SHORT).show();
                            }
                        }

                        @Override
                        public void onDataNotAvailable(String msg, Object... params) {

                        }
                    });
                })
                .create();
    }

    public boolean buildAndSaveMockedAutofillFieldCollections(List<FieldTypeWithHeuristics> fieldTypes,
            int numOfDatasets) {
        if (numOfDatasets < 0 || numOfDatasets > 10) {
            logw("Number of Datasets (%d) out of range.", numOfDatasets);
        }
        for (int i = 0; i < numOfDatasets; i++) {
            int datasetNumber = mLocalAutofillDataSource.getDatasetNumber();
            AutofillDataBuilder autofillDataBuilder =
                    new FakeAutofillDataBuilder(fieldTypes, mPackageName, datasetNumber);
            List<DatasetWithFilledAutofillFields> datasetsWithFilledAutofillFields =
                    autofillDataBuilder.buildDatasetsByPartition(datasetNumber);
            // Save datasets to database.
            mLocalAutofillDataSource.saveAutofillDatasets(datasetsWithFilledAutofillFields);
        }
        return true;
    }

    private AlertDialog.Builder prepareCredentialsDialog() {
        return new AlertDialog.Builder(SettingsActivity.this)
                .setTitle(R.string.settings_auth_change_credentials_title)
                .setNegativeButton(R.string.settings_cancel, null);
    }

    private AlertDialog buildCurrentCredentialsDialog() {
        final EditText currentPasswordField = LayoutInflater
                .from(SettingsActivity.this)
                .inflate(R.layout.multidataset_service_settings_authentication_dialog, null)
                .findViewById(R.id.main_password_field);
        return prepareCredentialsDialog()
                .setMessage(R.string.settings_auth_enter_current_password)
                .setView(currentPasswordField)
                .setPositiveButton(R.string.settings_ok, new DialogInterface.OnClickListener() {
                    @Override
                    public void onClick(DialogInterface dialog, int which) {
                        String password = currentPasswordField.getText().toString();
                        if (mPreferences.getMainPassword()
                                .equals(password)) {
                            buildNewCredentialsDialog().show();
                            dialog.dismiss();
                        }
                    }
                })
                .create();
    }

    private AlertDialog buildNewCredentialsDialog() {
        final EditText newPasswordField = LayoutInflater
                .from(SettingsActivity.this)
                .inflate(R.layout.multidataset_service_settings_authentication_dialog, null)
                .findViewById(R.id.main_password_field);
        return prepareCredentialsDialog()
                .setMessage(R.string.settings_auth_enter_new_password)
                .setView(newPasswordField)
                .setPositiveButton(R.string.settings_ok, (dialog, which) -> {
                    String password = newPasswordField.getText().toString();
                    mPreferences.setMainPassword(password);
                    dialog.dismiss();
                })
                .create();
    }

    private void setupSettingsSwitch(int containerId, int labelId, int switchId, boolean checked,
            CompoundButton.OnCheckedChangeListener checkedChangeListener) {
        ViewGroup container = findViewById(containerId);
        String switchLabel = ((TextView) container.findViewById(labelId)).getText().toString();
        final Switch switchView = container.findViewById(switchId);
        switchView.setContentDescription(switchLabel);
        switchView.setChecked(checked);
        container.setOnClickListener((view) -> switchView.performClick());
        switchView.setOnCheckedChangeListener(checkedChangeListener);
    }

    private void setupSettingsButton(int containerId, int labelId, int imageViewId,
            final View.OnClickListener onClickListener) {
        ViewGroup container = findViewById(containerId);
        TextView buttonLabel = container.findViewById(labelId);
        String buttonLabelText = buttonLabel.getText().toString();
        ImageView imageView = container.findViewById(imageViewId);
        imageView.setContentDescription(buttonLabelText);
        container.setOnClickListener(onClickListener);
    }

    private void setService(boolean enableService) {
        if (enableService) {
            startEnableService();
        } else {
            disableService();
        }
    }

    private void disableService() {
        if (mAutofillManager != null && mAutofillManager.hasEnabledAutofillServices()) {
            mAutofillManager.disableAutofillServices();
            Snackbar.make(findViewById(R.id.settings_layout),
                    R.string.settings_autofill_disabled_message, Snackbar.LENGTH_SHORT).show();
        } else {
            logd("Sample service already disabled.");
        }
    }

    private void startEnableService() {
        if (mAutofillManager != null && !mAutofillManager.hasEnabledAutofillServices()) {
            Intent intent = new Intent(Settings.ACTION_REQUEST_SET_AUTOFILL_SERVICE);
            intent.setData(Uri.parse("package:com.example.android.autofill.service"));
            logd(TAG, "enableService(): intent=%s", intent);
            startActivityForResult(intent, REQUEST_CODE_SET_DEFAULT);
        } else {
            logd("Sample service already enabled.");
        }
    }

    @Override
    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
        logd(TAG, "onActivityResult(): req=%s", requestCode);
        switch (requestCode) {
            case REQUEST_CODE_SET_DEFAULT:
                onDefaultServiceSet(resultCode);
                break;
        }
    }

    private void onDefaultServiceSet(int resultCode) {
        logd(TAG, "resultCode=%d", resultCode);
        switch (resultCode) {
            case RESULT_OK:
                logd("Autofill service set.");
                Snackbar.make(findViewById(R.id.settings_layout),
                        R.string.settings_autofill_service_set, Snackbar.LENGTH_SHORT)
                        .show();
                break;
            case RESULT_CANCELED:
                logd("Autofill service not selected.");
                Snackbar.make(findViewById(R.id.settings_layout),
                        R.string.settings_autofill_service_not_set, Snackbar.LENGTH_SHORT)
                        .show();
                break;
        }
    }
}