src/com/android/providers/media/photopicker/data/glide/PickerModelLoader.java

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

import static com.android.providers.media.photopicker.ui.ImageLoader.THUMBNAIL_REQUEST;

import android.content.Context;
import android.content.UriMatcher;
import android.provider.CloudMediaProviderContract;

import com.bumptech.glide.load.Options;
import com.bumptech.glide.load.model.ModelLoader;
import com.bumptech.glide.signature.ObjectKey;

import java.io.InputStream;

/** Custom {@link ModelLoader} to load thumbnails from cloud media provider. */
public final class PickerModelLoader implements ModelLoader<GlideLoadable, InputStream> {
    private final Context mContext;

    PickerModelLoader(Context context) {
        mContext = context;
    }

    @Override
    public LoadData<InputStream> buildLoadData(
            GlideLoadable model, int width, int height, Options options) {
        final boolean isThumbRequest = Boolean.TRUE.equals(options.get(THUMBNAIL_REQUEST));
        return new LoadData<>(
                new ObjectKey(model.getLoadableSignature(/* prefix= */ null)),
                new PickerThumbnailFetcher(mContext, model, width, height, isThumbRequest));
    }

    @Override
    public boolean handles(GlideLoadable model) {
        final int pickerId = 1;
        final UriMatcher matcher = new UriMatcher(UriMatcher.NO_MATCH);
        matcher.addURI(
                model.getLoadableUri().getAuthority(),
                CloudMediaProviderContract.URI_PATH_MEDIA + "/*",
                pickerId);

        // Matches picker URIs of the form content://<authority>/media
        return matcher.match(model.getLoadableUri()) == pickerId;
    }
}