chrome/android/features/autofill_assistant/java/src/org/chromium/chrome/browser/autofill_assistant/generic_ui/AssistantDrawable.java

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser.autofill_assistant.generic_ui;

import android.content.Context;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.graphics.drawable.BitmapDrawable;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.GradientDrawable;

import androidx.annotation.ColorInt;
import androidx.annotation.Nullable;
import androidx.appcompat.content.res.AppCompatResources;

import org.chromium.base.Callback;
import org.chromium.base.annotations.CalledByNative;
import org.chromium.base.annotations.JNINamespace;
import org.chromium.chrome.autofill_assistant.R;
import org.chromium.chrome.browser.autofill_assistant.drawable.AssistantDrawableIcon;
import org.chromium.chrome.browser.image_fetcher.ImageFetcher;
import org.chromium.chrome.browser.image_fetcher.ImageFetcherConfig;
import org.chromium.chrome.browser.image_fetcher.ImageFetcherFactory;

/** Represents a view background. */
@JNINamespace("autofill_assistant")
public abstract class AssistantDrawable {
    private static final int INVALID_ICON_ID = -1;

    /** Fetches the drawable. */
    public abstract void getDrawable(Context context, Callback<Drawable> callback);

    @CalledByNative
    public static AssistantDrawable createRectangleShape(
            @Nullable @ColorInt Integer backgroundColor, @Nullable @ColorInt Integer strokeColor,
            int strokeWidthInPixels, int cornerRadiusInPixels) {
        return new AssistantRectangleDrawable(
                backgroundColor, strokeColor, strokeWidthInPixels, cornerRadiusInPixels);
    }

    @CalledByNative
    public static AssistantDrawable createFromUrl(
            String url, int widthInPixels, int heightInPixels) {
        return new AssistantBitmapDrawable(url, widthInPixels, heightInPixels);
    }

    /** Returns whether {@code resourceId} is a valid resource identifier. */
    @CalledByNative
    public static boolean isValidDrawableResource(Context context, String resourceId) {
        int drawableId = context.getResources().getIdentifier(
                resourceId, "drawable", context.getPackageName());
        if (drawableId == 0) {
            return false;
        }
        return AppCompatResources.getDrawable(context, drawableId) != null;
    }

    @CalledByNative
    public static AssistantDrawable createFromResource(String resourceId) {
        return new AssistantResourceDrawable(resourceId);
    }

    @CalledByNative
    public static AssistantDrawable createFromIcon(@AssistantDrawableIcon int icon) {
        return new AssistantIconDrawable(icon);
    }

    @CalledByNative
    public static AssistantDrawable createFromBase64(byte[] base64) {
        return new AssistantBase64Drawable(base64);
    }

    private static class AssistantRectangleDrawable extends AssistantDrawable {
        private final @Nullable @ColorInt Integer mBackgroundColor;
        private final @Nullable @ColorInt Integer mStrokeColor;
        private final int mStrokeWidthInPixels;
        private final int mCornerRadiusInPixels;

        AssistantRectangleDrawable(@Nullable @ColorInt Integer backgroundColor,
                @Nullable @ColorInt Integer strokeColor, int strokeWidthInPixels,
                int cornerRadiusInPixels) {
            mBackgroundColor = backgroundColor;
            mStrokeColor = strokeColor;
            mStrokeWidthInPixels = strokeWidthInPixels;
            mCornerRadiusInPixels = cornerRadiusInPixels;
        }

        @Override
        public void getDrawable(Context context, Callback<Drawable> callback) {
            GradientDrawable shape = new GradientDrawable();
            shape.setShape(GradientDrawable.RECTANGLE);
            shape.setCornerRadius(mCornerRadiusInPixels);
            if (mBackgroundColor != null) {
                shape.setColor(mBackgroundColor);
            }
            if (mStrokeColor != null) {
                shape.setStroke(mStrokeWidthInPixels, mStrokeColor);
            }
            callback.onResult(shape);
        }
    }

    private static class AssistantBitmapDrawable extends AssistantDrawable {
        private final ImageFetcher mImageFetcher =
                ImageFetcherFactory.createImageFetcher(ImageFetcherConfig.DISK_CACHE_ONLY);
        private final String mUrl;
        private final int mWidthInPixels;
        private final int mHeightInPixels;

        AssistantBitmapDrawable(String url, int width, int height) {
            mUrl = url;
            mWidthInPixels = width;
            mHeightInPixels = height;
        }

        @Override
        public void getDrawable(Context context, Callback<Drawable> callback) {
            // TODO(b/143517837) Merge autofill assistant image fetcher UMA names.
            ImageFetcher.Params params = ImageFetcher.Params.create(
                    mUrl, ImageFetcher.ASSISTANT_DETAILS_UMA_CLIENT_NAME);
            mImageFetcher.fetchImage(
                    params, result -> {
                        if (result != null) {
                            callback.onResult(new BitmapDrawable(context.getResources(),
                                    Bitmap.createScaledBitmap(
                                            result, mWidthInPixels, mHeightInPixels, true)));
                        } else {
                            callback.onResult(null);
                        }
                    });
        }
    }

    private static class AssistantResourceDrawable extends AssistantDrawable {
        private final String mResourceId;

        AssistantResourceDrawable(String resourceId) {
            mResourceId = resourceId;
        }

        @Override
        public void getDrawable(Context context, Callback<Drawable> callback) {
            int drawableId = context.getResources().getIdentifier(
                    mResourceId, "drawable", context.getPackageName());
            if (drawableId == 0) {
                callback.onResult(null);
            }
            callback.onResult(AppCompatResources.getDrawable(context, drawableId));
        }
    }

    private static class AssistantIconDrawable extends AssistantDrawable {
        private final @AssistantDrawableIcon int mIcon;

        AssistantIconDrawable(@AssistantDrawableIcon int icon) {
            mIcon = icon;
        }

        @Override
        public void getDrawable(Context context, Callback<Drawable> callback) {
            int resourceId;
            switch (mIcon) {
                case AssistantDrawableIcon.PROGRESSBAR_DEFAULT_INITIAL_STEP:
                    resourceId = R.drawable.ic_autofill_assistant_default_progress_start_black_24dp;
                    break;
                case AssistantDrawableIcon.PROGRESSBAR_DEFAULT_DATA_COLLECTION:
                    resourceId = R.drawable.ic_shopping_basket_black_24dp;
                    break;
                case AssistantDrawableIcon.PROGRESSBAR_DEFAULT_PAYMENT:
                    resourceId = R.drawable.ic_payment_black_24dp;
                    break;
                case AssistantDrawableIcon.PROGRESSBAR_DEFAULT_FINAL_STEP:
                    resourceId = R.drawable.ic_check_circle_black_24dp;
                    break;
                default:
                    resourceId = INVALID_ICON_ID;
                    break;
            }

            callback.onResult(resourceId == INVALID_ICON_ID
                            ? null
                            : AppCompatResources.getDrawable(context, resourceId));
        }
    }

    private static class AssistantBase64Drawable extends AssistantDrawable {
        private final byte[] mBase64;

        AssistantBase64Drawable(byte[] base64) {
            mBase64 = base64;
        }

        @Override
        public void getDrawable(Context context, Callback<Drawable> callback) {
            Bitmap icon = BitmapFactory.decodeByteArray(mBase64, /* offset= */ 0, mBase64.length);
            callback.onResult(new BitmapDrawable(context.getResources(), icon));
        }
    }
}