/*
 * Copyright 2024 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.inputmethod;

import static com.android.settings.inputmethod.PhysicalKeyboardFragment.getHardKeyboards;

import android.app.settings.SettingsEnums;
import android.content.Context;
import android.hardware.input.InputManager;
import android.os.Bundle;
import android.view.InputDevice;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.preference.PreferenceScreen;
import androidx.recyclerview.widget.GridLayoutManager;
import androidx.recyclerview.widget.RecyclerView;

import com.android.internal.util.Preconditions;
import com.android.settings.R;
import com.android.settings.dashboard.DashboardFragment;
import com.android.settings.keyboard.Flags;
import com.android.settings.search.BaseSearchIndexProvider;
import com.android.settingslib.search.SearchIndexable;
import com.android.settingslib.utils.ThreadUtils;
import com.android.settingslib.widget.LayoutPreference;

import java.util.List;

@SearchIndexable
public class MouseKeysMainPageFragment extends DashboardFragment
        implements InputManager.InputDeviceListener {

    private static final String TAG = "MouseKeysMainPageFragment";
    private static final String KEY_MOUSE_KEY_LIST = "mouse_keys_list";

    private InputManager mInputManager;
    private LayoutPreference mMouseKeyImagesPreference;
    @Nullable
    private InputDevice mCurrentInputDevice;

    @Override
    public void onCreate(@NonNull Bundle bundle) {
        super.onCreate(bundle);
        mCurrentInputDevice = getInputDevice();
        final PreferenceScreen screen = getPreferenceScreen();
        mMouseKeyImagesPreference = screen.findPreference(KEY_MOUSE_KEY_LIST);
        mInputManager = Preconditions.checkNotNull(getActivity()
                .getSystemService(InputManager.class));
        String title = mCurrentInputDevice == null ? getActivity().getString(R.string.mouse_keys)
                : getActivity().getString(R.string.mouse_key_main_page_title,
                        mCurrentInputDevice.getName());
        getActivity().setTitle(title);
        configureImagesPreference();
    }

    @Override
    public void onResume() {
        super.onResume();
        finishEarlyIfNeeded();
        mInputManager.registerInputDeviceListener(this, null);
    }

    @Override
    public void onPause() {
        super.onPause();
        mInputManager.unregisterInputDeviceListener(this);
    }

    @Override
    public int getMetricsCategory() {
        return SettingsEnums.PHYSICAL_KEYBOARD_A11Y;
    }

    @Override
    protected int getPreferenceScreenResId() {
        return R.xml.mouse_keys_main_page;
    }

    @Override
    protected String getLogTag() {
        return TAG;
    }

    @Override
    public void onInputDeviceAdded(int deviceId) {
        finishEarlyIfNeeded();
    }

    @Override
    public void onInputDeviceRemoved(int deviceId) {
        finishEarlyIfNeeded();
    }

    @Override
    public void onInputDeviceChanged(int deviceId) {
        finishEarlyIfNeeded();
    }

    private void finishEarlyIfNeeded() {
        final Context context = getContext();
        ThreadUtils.postOnBackgroundThread(() -> {
            final List<PhysicalKeyboardFragment.HardKeyboardDeviceInfo> newHardKeyboards =
                    getHardKeyboards(context);
            if (newHardKeyboards.isEmpty()) {
                getActivity().finish();
            }
        });
    }

    private void configureImagesPreference() {
        final RecyclerView recyclerView = mMouseKeyImagesPreference.findViewById(
                R.id.mouse_keys_image_recycler_list);
        recyclerView.setLayoutManager(new GridLayoutManager(getActivity(), 2));
        recyclerView.setAdapter(new MouseKeysImageListAdapter(getActivity(), mCurrentInputDevice));
    }

    /**
     * Priority of picking input device:
     * 1. internal keyboard(built-in keyboard)
     * 2. first keyboard in the list
     */
    @Nullable
    private InputDevice getInputDevice() {
        InputDevice inputDevice = null;
        for (int deviceId : InputDevice.getDeviceIds()) {
            final InputDevice device = InputDevice.getDevice(deviceId);
            if (device == null || device.isVirtual() || !device.isFullKeyboard()) {
                continue;
            }
            if (inputDevice == null) {
                inputDevice = device;
            } else if (!device.isExternal()) {
                inputDevice = device;
                break;
            }
        }
        return inputDevice;
    }

    public static final BaseSearchIndexProvider SEARCH_INDEX_DATA_PROVIDER =
            new BaseSearchIndexProvider(R.xml.mouse_keys_main_page) {
                @Override
                protected boolean isPageSearchEnabled(Context context) {
                    return Flags.keyboardAndTouchpadA11yNewPageEnabled()
                            && !getHardKeyboards(context).isEmpty();
                }
            };
}