/*
 * Copyright (C) 2021 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.providers.media.photopicker.data;

import android.content.ClipData;
import android.content.Intent;
import android.net.Uri;

import androidx.annotation.NonNull;

import com.android.providers.media.PickerUriResolver;
import com.android.providers.media.photopicker.data.model.Item;
import com.android.providers.media.photopicker.data.model.UserId;

import com.google.common.annotations.VisibleForTesting;

import java.util.ArrayList;
import java.util.List;

/**
 * This class is responsible for returning result to the caller of the PhotoPicker.
 */
public class PickerResult {

    /**
     * @return {@code Intent} which contains Uri that has been granted access on.
     */
    @NonNull
    public static Intent getPickerResponseIntent(String action, boolean canSelectMultiple,
            @NonNull List<Item> selectedItems) {
        // 1. Get Picker Uris corresponding to the selected items
        List<Uri> selectedUris = getPickerUrisForItems(action, selectedItems);

        // 2. Grant read access to picker Uris and return
        Intent intent = new Intent();
        final int size = selectedUris.size();
        if (size < 1) {
            // TODO (b/168783994): check if this is ever possible. If yes, handle properly,
            // if not, remove this if block.
            return intent;
        }
        if (!canSelectMultiple) {
            intent.setData(selectedUris.get(0));
        }
        // TODO (b/169737761): use correct mime types
        String[] mimeTypes = new String[]{"image/*", "video/*"};
        final ClipData clipData = new ClipData(null /* label */, mimeTypes,
                new ClipData.Item(selectedUris.get(0)));
        for (int i = 1; i < size; i++) {
            clipData.addItem(new ClipData.Item(selectedUris.get(i)));
        }
        intent.setClipData(clipData);
        intent.addFlags(Intent.FLAG_GRANT_READ_URI_PERMISSION);
        intent.addFlags(Intent.FLAG_GRANT_PERSISTABLE_URI_PERMISSION);

        return intent;
    }

    @VisibleForTesting
    static Uri getPickerUri(String action, Uri uri) {
        final String userInfo = uri.getUserInfo();
        final String userId = userInfo == null ? UserId.CURRENT_USER.toString() : userInfo;
        return PickerUriResolver.wrapProviderUri(uri, action, Integer.parseInt(userId));
    }

    /**
     * Returns list of PhotoPicker Uris corresponding to each {@link Item}
     *
     * @param action action name which opened PhotoPicker
     * @param items list of Item for which we return uri list.
     */
    @NonNull
    public static List<Uri> getPickerUrisForItems(String action, @NonNull List<Item> items) {
        List<Uri> uris = new ArrayList<>();
        for (Item item : items) {
            uris.add(getPickerUri(action, item.getContentUri()));
        }

        return uris;
    }

}