package ru.noties.markwon.image;

import android.graphics.drawable.Drawable;
import android.support.annotation.NonNull;
import android.support.annotation.Nullable;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class AsyncDrawableLoaderBuilder {

    ExecutorService executorService;
    final Map<String, SchemeHandler> schemeHandlers = new HashMap<>(3);
    final Map<String, MediaDecoder> mediaDecoders = new HashMap<>(3);
    MediaDecoder defaultMediaDecoder;
    DrawableProvider placeholderDrawableProvider;
    DrawableProvider errorDrawableProvider;

    @NonNull
    public AsyncDrawableLoaderBuilder executorService(@NonNull ExecutorService executorService) {
        this.executorService = executorService;
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder addSchemeHandler(@NonNull String scheme, @NonNull SchemeHandler schemeHandler) {
        schemeHandlers.put(scheme, schemeHandler);
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder addSchemeHandler(@NonNull Collection<String> schemes, @NonNull SchemeHandler schemeHandler) {
        for (String scheme : schemes) {
            schemeHandlers.put(scheme, schemeHandler);
        }
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder addMediaDecoder(@NonNull String contentType, @NonNull MediaDecoder mediaDecoder) {
        mediaDecoders.put(contentType, mediaDecoder);
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder addMediaDecoder(@NonNull Collection<String> contentTypes, @NonNull MediaDecoder mediaDecoder) {
        for (String contentType : contentTypes) {
            mediaDecoders.put(contentType, mediaDecoder);
        }
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder removeSchemeHandler(@NonNull String scheme) {
        schemeHandlers.remove(scheme);
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder removeMediaDecoder(@NonNull String contentType) {
        mediaDecoders.remove(contentType);
        return this;
    }

    @NonNull
    public AsyncDrawableLoaderBuilder defaultMediaDecoder(@Nullable MediaDecoder mediaDecoder) {
        this.defaultMediaDecoder = mediaDecoder;
        return this;
    }

    /**
     * @since 3.0.0
     */
    @NonNull
    public AsyncDrawableLoaderBuilder placeholderDrawableProvider(@NonNull DrawableProvider placeholderDrawableProvider) {
        this.placeholderDrawableProvider = placeholderDrawableProvider;
        return this;
    }

    /**
     * @since 3.0.0
     */
    @NonNull
    public AsyncDrawableLoaderBuilder errorDrawableProvider(@NonNull DrawableProvider errorDrawableProvider) {
        this.errorDrawableProvider = errorDrawableProvider;
        return this;
    }


    @NonNull
    public AsyncDrawableLoader build() {

        // if we have no schemeHandlers -> we cannot show anything
        // OR if we have no media decoders
        if (schemeHandlers.size() == 0
                || (mediaDecoders.size() == 0 && defaultMediaDecoder == null)) {
            return new AsyncDrawableLoaderNoOp();
        }

        if (executorService == null) {
            executorService = Executors.newCachedThreadPool();
        }

        return new AsyncDrawableLoaderImpl(this);
    }

}