package com.reactlibrary;

import android.content.Context;

import com.google.android.gms.cast.CastMediaControlIntent;
import com.google.android.gms.cast.MediaMetadata;
import com.google.android.gms.cast.framework.CastOptions;
import com.google.android.gms.cast.framework.OptionsProvider;
import com.google.android.gms.cast.framework.SessionProvider;
import com.google.android.gms.cast.framework.media.CastMediaOptions;
import com.google.android.gms.cast.framework.media.ImagePicker;
import com.google.android.gms.cast.framework.media.MediaIntentReceiver;
import com.google.android.gms.cast.framework.media.NotificationOptions;
import com.google.android.gms.common.images.WebImage;
import com.reactlibrary.components.RNExpandedCastControlsActivity;

import java.util.Arrays;
import java.util.List;

public class RnGoogleCastOptionsProvider implements OptionsProvider {
    @Override
    public CastOptions getCastOptions(Context context) {
        NotificationOptions notificationOptions =
                new NotificationOptions.Builder()
                        .setActions(
                                Arrays.asList(MediaIntentReceiver.ACTION_SKIP_NEXT,
                                        MediaIntentReceiver.ACTION_TOGGLE_PLAYBACK,
                                        MediaIntentReceiver.ACTION_STOP_CASTING),
                                new int[] {1, 2})
                        .setTargetActivityClassName(
                                RNExpandedCastControlsActivity.class.getName())
                        .build();

        CastMediaOptions mediaOptions =
                new CastMediaOptions.Builder()
                        .setImagePicker(new ImagePickerImpl())
                        .setNotificationOptions(notificationOptions)
                        .setExpandedControllerActivityClassName(
                                RNExpandedCastControlsActivity.class.getName())
                        .build();

        return new CastOptions.Builder()
                .setReceiverApplicationId(
                        CastMediaControlIntent.DEFAULT_MEDIA_RECEIVER_APPLICATION_ID)
                .setCastMediaOptions(mediaOptions)
                .build();
    }

    @Override
    public List<SessionProvider> getAdditionalSessionProviders(Context context) {
        return null;
    }

    private static class ImagePickerImpl extends ImagePicker {

        @Override
        public WebImage onPickImage(MediaMetadata mediaMetadata, int type) {
            if ((mediaMetadata == null) || !mediaMetadata.hasImages()) {
                return null;
            }
            List<WebImage> images = mediaMetadata.getImages();
            if (images.size() == 1) {
                return images.get(0);
            } else {
                if (type ==
                        ImagePicker.IMAGE_TYPE_MEDIA_ROUTE_CONTROLLER_DIALOG_BACKGROUND) {
                    return images.get(0);
                } else {
                    return images.get(1);
                }
            }
        }
    }
}