/*
 * 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.search;

import static com.android.settings.search.InstalledAppResultLoader.getWordDifference;

import android.accessibilityservice.AccessibilityServiceInfo;
import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.content.pm.PackageManager;
import android.content.pm.ResolveInfo;
import android.content.pm.ServiceInfo;
import android.graphics.drawable.Drawable;
import android.os.UserHandle;
import android.support.annotation.VisibleForTesting;
import android.support.v4.content.ContextCompat;
import android.util.IconDrawableFactory;
import android.view.accessibility.AccessibilityManager;

import com.android.settings.R;
import com.android.settings.accessibility.AccessibilitySettings;
import com.android.settings.dashboard.SiteMapManager;
import com.android.settings.utils.AsyncLoader;

import java.util.HashSet;
import java.util.List;
import java.util.Objects;
import java.util.Set;

public class AccessibilityServiceResultLoader extends AsyncLoader<Set<? extends SearchResult>> {

    private static final int NAME_NO_MATCH = -1;

    private final Context mContext;

    private List<String> mBreadcrumb;
    private SiteMapManager mSiteMapManager;
    @VisibleForTesting
    final String mQuery;
    private final AccessibilityManager mAccessibilityManager;
    private final PackageManager mPackageManager;
    private final int mUserId;


    public AccessibilityServiceResultLoader(Context context, String query,
            SiteMapManager mapManager) {
        super(context);
        mContext = context;
        mUserId = UserHandle.myUserId();
        mSiteMapManager = mapManager;
        mPackageManager = context.getPackageManager();
        mAccessibilityManager =
                (AccessibilityManager) context.getSystemService(Context.ACCESSIBILITY_SERVICE);
        mQuery = query;
    }

    @Override
    public Set<? extends SearchResult> loadInBackground() {
        final Set<SearchResult> results = new HashSet<>();
        final Context context = getContext();
        final List<AccessibilityServiceInfo> services = mAccessibilityManager
                .getInstalledAccessibilityServiceList();
        final IconDrawableFactory iconFactory = IconDrawableFactory.newInstance(mContext);
        final String screenTitle = context.getString(R.string.accessibility_settings);
        for (AccessibilityServiceInfo service : services) {
            if (service == null) {
                continue;
            }
            final ResolveInfo resolveInfo = service.getResolveInfo();
            if (service.getResolveInfo() == null) {
                continue;
            }
            final ServiceInfo serviceInfo = resolveInfo.serviceInfo;
            final CharSequence title = resolveInfo.loadLabel(mPackageManager);
            final int wordDiff = getWordDifference(title.toString(), mQuery);
            if (wordDiff == NAME_NO_MATCH) {
                continue;
            }
            final Drawable icon;
            if (resolveInfo.getIconResource() == 0) {
                icon = ContextCompat.getDrawable(context, R.mipmap.ic_accessibility_generic);
            } else {
                icon = iconFactory.getBadgedIcon(
                        resolveInfo.serviceInfo,
                        resolveInfo.serviceInfo.applicationInfo,
                        mUserId);
            }
            final String componentName = new ComponentName(serviceInfo.packageName,
                    serviceInfo.name).flattenToString();
            final Intent intent = DatabaseIndexingUtils.buildSubsettingIntent(context,
                    AccessibilitySettings.class.getName(), componentName, screenTitle);

            results.add(new SearchResult.Builder()
                    .setTitle(title)
                    .addBreadcrumbs(getBreadCrumb())
                    .setPayload(new ResultPayload(intent))
                    .setRank(wordDiff)
                    .setIcon(icon)
                    .setStableId(Objects.hash(screenTitle, componentName))
                    .build());
        }
        return results;
    }

    private List<String> getBreadCrumb() {
        if (mBreadcrumb == null || mBreadcrumb.isEmpty()) {
            final Context context = getContext();
            mBreadcrumb = mSiteMapManager.buildBreadCrumb(
                    context, AccessibilitySettings.class.getName(),
                    context.getString(R.string.accessibility_settings));
        }
        return mBreadcrumb;
    }

    @Override
    protected void onDiscardResult(Set<? extends SearchResult> result) {

    }
}