/*
 * 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.android.settings.applications.assist;

import android.content.Context;
import android.net.Uri;
import android.os.UserHandle;
import android.provider.Settings;
import android.support.v7.preference.Preference;
import android.support.v7.preference.PreferenceScreen;
import android.support.v7.preference.TwoStatePreference;

import com.android.internal.app.AssistUtils;
import com.android.settings.core.PreferenceControllerMixin;
import com.android.settingslib.core.AbstractPreferenceController;
import com.android.settingslib.core.lifecycle.Lifecycle;
import com.android.settingslib.core.lifecycle.LifecycleObserver;
import com.android.settingslib.core.lifecycle.events.OnPause;
import com.android.settingslib.core.lifecycle.events.OnResume;

import java.util.Arrays;
import java.util.List;

public class AssistScreenshotPreferenceController extends AbstractPreferenceController
        implements PreferenceControllerMixin, Preference.OnPreferenceChangeListener,
        LifecycleObserver, OnResume, OnPause {

    private static final String KEY_SCREENSHOT = "screenshot";

    private final AssistUtils mAssistUtils;
    private final SettingObserver mSettingObserver;

    private PreferenceScreen mScreen;
    private Preference mPreference;

    public AssistScreenshotPreferenceController(Context context, Lifecycle lifecycle) {
        super(context);
        mAssistUtils = new AssistUtils(context);
        mSettingObserver = new SettingObserver();
        if (lifecycle != null) {
            lifecycle.addObserver(this);
        }
    }

    @Override
    public boolean isAvailable() {
        return mAssistUtils.getAssistComponentForUser(UserHandle.myUserId()) != null;
    }

    @Override
    public void displayPreference(PreferenceScreen screen) {
        mScreen = screen;
        mPreference = screen.findPreference(getPreferenceKey());
        super.displayPreference(screen);
    }

    @Override
    public String getPreferenceKey() {
        return KEY_SCREENSHOT;
    }

    @Override
    public void onResume() {
        mSettingObserver.register(mContext.getContentResolver(), true);
        updatePreference();
    }

    @Override
    public void updateState(Preference preference) {
        updatePreference();
    }

    @Override
    public void onPause() {
        mSettingObserver.register(mContext.getContentResolver(), false);
    }

    @Override
    public boolean onPreferenceChange(Preference preference, Object newValue) {
        Settings.Secure.putInt(mContext.getContentResolver(),
                Settings.Secure.ASSIST_SCREENSHOT_ENABLED,
                (boolean) newValue ? 1 : 0);
        return true;
    }

    private void updatePreference() {
        if (mPreference == null || !(mPreference instanceof TwoStatePreference)) {
            return;
        }
        if (isAvailable()) {
            if (mScreen.findPreference(getPreferenceKey()) == null) {
                // add it if it's not on scree
                mScreen.addPreference(mPreference);
            }
        } else {
            mScreen.removePreference(mPreference);
        }
        final boolean checked = Settings.Secure.getInt(mContext.getContentResolver(),
                Settings.Secure.ASSIST_SCREENSHOT_ENABLED, 1) != 0;
        ((TwoStatePreference) mPreference).setChecked(checked);
        final boolean contextChecked = Settings.Secure.getInt(mContext.getContentResolver(),
                Settings.Secure.ASSIST_STRUCTURE_ENABLED, 1) != 0;
        mPreference.setEnabled(contextChecked);
    }

    class SettingObserver extends AssistSettingObserver {

        private final Uri URI =
                Settings.Secure.getUriFor(Settings.Secure.ASSIST_SCREENSHOT_ENABLED);
        private final Uri CONTEXT_URI =
                Settings.Secure.getUriFor(Settings.Secure.ASSIST_STRUCTURE_ENABLED);

        @Override
        protected List<Uri> getSettingUris() {
            return Arrays.asList(URI, CONTEXT_URI);
        }

        @Override
        public void onSettingChange() {
            updatePreference();
        }
    }
}
