/*
 * Copyright (C) 2022 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.server.appsearch.contactsindexer;

import android.annotation.NonNull;
import android.app.appsearch.AppSearchResult;
import android.app.appsearch.GenericDocument;
import android.app.appsearch.exceptions.AppSearchException;
import android.app.appsearch.testutil.TestContactsIndexerConfig;
import android.content.Context;
import android.util.ArrayMap;

import com.android.server.appsearch.contactsindexer.appsearchtypes.Person;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;

public final class FakeAppSearchHelper extends AppSearchHelper {
    final int mDocLimit;
    final int mDeleteLimit;
    // Contacts that have been deleted during the test
    List<String> mRemovedIds = new ArrayList<>();
    // Contacts that have been updated/inserted during the test
    List<Person> mIndexedContacts = new ArrayList<>();
    Map<String, Person> mExistingContacts = new ArrayMap<>();

    public FakeAppSearchHelper(@NonNull Context context) {
        this(context, Integer.MAX_VALUE, Integer.MAX_VALUE);
    }

    public FakeAppSearchHelper(@NonNull Context context, int docLimit, int deleteLimit) {
        super(context, Runnable::run, new TestContactsIndexerConfig());
        mDocLimit = docLimit;
        mDeleteLimit = deleteLimit;
    }

    public void setExistingContacts(@NonNull Collection<Person> contacts) {
        for (Person contact : contacts) {
            mExistingContacts.put(contact.getId(), contact);
        }
    }

    @Override
    public CompletableFuture<Void> indexContactsAsync(@NonNull Collection<Person> contacts,
            @NonNull ContactsUpdateStats updateStats,
            boolean shouldKeepUpdatingOnErrorFlagEnabled) {
        for (Person person : contacts) {
            if (mIndexedContacts.size() >= mDocLimit) {
                return CompletableFuture.failedFuture(
                        new AppSearchException(AppSearchResult.RESULT_OUT_OF_SPACE,
                                "Reached document limit."));
            } else {
                mExistingContacts.put(person.getId(), person);
                mIndexedContacts.add(person);
            }
        }
        return CompletableFuture.completedFuture(null);
    }

    @Override
    public CompletableFuture<Void> removeContactsByIdAsync(@NonNull Collection<String> ids,
            @NonNull ContactsUpdateStats updateStats,
            boolean shouldKeepUpdatingOnError) {
        mRemovedIds.addAll(ids);
        return CompletableFuture.completedFuture(null);
    }

    @Override
    public CompletableFuture<List<GenericDocument>> getContactsWithFingerprintsAsync(
            @NonNull List<String> ids, boolean shouldKeepUpdatingOnError,
            @NonNull ContactsUpdateStats updateStats) {
        return CompletableFuture.supplyAsync(() -> {
            List<GenericDocument> result = new ArrayList<>();
            for (String id : ids) {
                result.add(mExistingContacts.get(id));
            }
            return result;
        });
    }
}