package ml.docilealligator.infinityforreddit.markdown;

import android.graphics.drawable.Animatable;
import android.graphics.drawable.Drawable;
import android.text.Spanned;
import android.text.style.ClickableSpan;
import android.view.View;
import android.widget.TextView;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;

import com.bumptech.glide.Glide;
import com.bumptech.glide.RequestBuilder;
import com.bumptech.glide.RequestManager;
import com.bumptech.glide.request.target.CustomTarget;
import com.bumptech.glide.request.target.Target;
import com.bumptech.glide.request.transition.Transition;

import org.commonmark.node.Link;
import org.commonmark.node.Node;

import java.util.HashMap;
import java.util.Map;

import io.noties.markwon.AbstractMarkwonPlugin;
import io.noties.markwon.MarkwonConfiguration;
import io.noties.markwon.MarkwonSpansFactory;
import io.noties.markwon.MarkwonVisitor;
import io.noties.markwon.RenderProps;
import io.noties.markwon.SpanFactory;
import io.noties.markwon.core.CoreProps;
import io.noties.markwon.image.AsyncDrawable;
import io.noties.markwon.image.AsyncDrawableLoader;
import io.noties.markwon.image.AsyncDrawableScheduler;
import io.noties.markwon.image.DrawableUtils;
import io.noties.markwon.image.ImageProps;
import ml.docilealligator.infinityforreddit.thing.MediaMetadata;
import ml.docilealligator.infinityforreddit.activities.BaseActivity;
import ml.docilealligator.infinityforreddit.utils.SharedPreferencesUtils;
import ml.docilealligator.infinityforreddit.utils.Utils;

public class EmotePlugin extends AbstractMarkwonPlugin {
    private final GlideAsyncDrawableLoader glideAsyncDrawableLoader;
    private boolean dataSavingMode;
    private final boolean disableImagePreview;
    private final boolean canShowEmote;
    private final OnEmoteClickListener onEmoteClickListener;

    public interface GlideStore {

        @NonNull
        RequestBuilder<Drawable> load(@NonNull AsyncDrawable drawable);

        void cancel(@NonNull Target<?> target);
    }

    public interface OnEmoteClickListener {
        void onEmoteClick(MediaMetadata mediaMetadata);
    }

    @NonNull
    public static EmotePlugin create(@NonNull final BaseActivity baseActivity, int embeddedMediaType,
                                     @NonNull final OnEmoteClickListener onEmoteClickListener) {
        // @since 4.5.0 cache RequestManager
        //  sometimes `cancel` would be called after activity is destroyed,
        //  so `Glide.with(baseActivity)` will throw an exception
        RequestManager requestManager = Glide.with(baseActivity);
        return new EmotePlugin(baseActivity, new GlideStore() {
            @NonNull
            @Override
            public RequestBuilder<Drawable> load(@NonNull AsyncDrawable drawable) {
                return requestManager.load(drawable.getDestination());
            }

            @Override
            public void cancel(@NonNull Target<?> target) {
                requestManager.clear(target);
            }
        }, embeddedMediaType, onEmoteClickListener);
    }

    @NonNull
    public static EmotePlugin create(@NonNull final BaseActivity baseActivity, int embeddedMediaType,
                                     boolean dataSavingMode, boolean disableImagePreview,
                                     @NonNull final OnEmoteClickListener onEmoteClickListener) {
        // @since 4.5.0 cache RequestManager
        //  sometimes `cancel` would be called after activity is destroyed,
        //  so `Glide.with(baseActivity)` will throw an exception
        RequestManager requestManager = Glide.with(baseActivity);
        return new EmotePlugin(baseActivity, new GlideStore() {
            @NonNull
            @Override
            public RequestBuilder<Drawable> load(@NonNull AsyncDrawable drawable) {
                return requestManager.load(drawable.getDestination());
            }

            @Override
            public void cancel(@NonNull Target<?> target) {
                requestManager.clear(target);
            }
        }, embeddedMediaType, dataSavingMode, disableImagePreview, onEmoteClickListener);
    }

    @SuppressWarnings("WeakerAccess")
    EmotePlugin(@NonNull final BaseActivity baseActivity,
                @NonNull GlideStore glideStore, int embeddedMediaType,
                @NonNull final OnEmoteClickListener onEmoteClickListener) {
        this.glideAsyncDrawableLoader = new GlideAsyncDrawableLoader(glideStore);
        String dataSavingModeString = baseActivity.getDefaultSharedPreferences().getString(SharedPreferencesUtils.DATA_SAVING_MODE, SharedPreferencesUtils.DATA_SAVING_MODE_OFF);
        if (dataSavingModeString.equals(SharedPreferencesUtils.DATA_SAVING_MODE_ALWAYS)) {
            dataSavingMode = true;
        } else if (dataSavingModeString.equals(SharedPreferencesUtils.DATA_SAVING_MODE_ONLY_ON_CELLULAR_DATA)) {
            dataSavingMode = Utils.getConnectedNetwork(baseActivity) == Utils.NETWORK_TYPE_CELLULAR;
        }
        disableImagePreview = baseActivity.getDefaultSharedPreferences().getBoolean(SharedPreferencesUtils.DISABLE_IMAGE_PREVIEW, false);
        canShowEmote = SharedPreferencesUtils.canShowEmote(embeddedMediaType);
        this.onEmoteClickListener = onEmoteClickListener;
    }

    @SuppressWarnings("WeakerAccess")
    EmotePlugin(@NonNull final BaseActivity baseActivity, @NonNull GlideStore glideStore,
                int embeddedMediaType, boolean dataSavingMode, boolean disableImagePreview,
                @NonNull final OnEmoteClickListener onEmoteClickListener) {
        this.glideAsyncDrawableLoader = new GlideAsyncDrawableLoader(glideStore);
        this.dataSavingMode = dataSavingMode;
        this.disableImagePreview = disableImagePreview;
        canShowEmote = SharedPreferencesUtils.canShowEmote(embeddedMediaType);
        this.onEmoteClickListener = onEmoteClickListener;
    }

    @Override
    public void configureSpansFactory(@NonNull MarkwonSpansFactory.Builder builder) {
        builder.setFactory(Emote.class, new EmoteSpanFactory());
    }

    @Override
    public void configureConfiguration(@NonNull MarkwonConfiguration.Builder builder) {
        builder.asyncDrawableLoader(glideAsyncDrawableLoader);
    }

    @Override
    public void configureVisitor(@NonNull MarkwonVisitor.Builder builder) {
        builder.on(Emote.class, (visitor, emote) -> {
            if ((dataSavingMode && disableImagePreview) || !canShowEmote) {
                Link link = new Link(emote.getMediaMetadata().original.url, emote.getTitle());

                final int length = visitor.length();
                visitor.visitChildren(emote);

                final String destination = link.getDestination();
                CoreProps.LINK_DESTINATION.set(visitor.renderProps(), destination);
                visitor.setSpansForNodeOptional(link, length);
                return;
            }

            // if there is no image spanFactory, ignore
            final SpanFactory spanFactory = visitor.configuration().spansFactory().get(Emote.class);
            if (spanFactory == null) {
                visitor.visitChildren(emote);
                return;
            }

            final int length = visitor.length();

            visitor.visitChildren(emote);

            // we must check if anything _was_ added, as we need at least one char to render
            if (length == visitor.length()) {
                visitor.builder().append('\uFFFC');
            }

            final MarkwonConfiguration configuration = visitor.configuration();

            final Node parent = emote.getParent();
            final boolean link = parent instanceof Link;

            final String destination = configuration
                    .imageDestinationProcessor()
                    .process(emote.getMediaMetadata().original.url);

            final RenderProps props = visitor.renderProps();

            // apply image properties
            // Please note that we explicitly set IMAGE_SIZE to null as we do not clear
            // properties after we applied span (we could though)
            ImageProps.DESTINATION.set(props, destination);
            ImageProps.REPLACEMENT_TEXT_IS_LINK.set(props, link);
            ImageProps.IMAGE_SIZE.set(props, null);

            visitor.setSpans(length, spanFactory.getSpans(configuration, props));
            visitor.setSpans(length, new ClickableSpan() {
                @Override
                public void onClick(@NonNull View widget) {
                    onEmoteClickListener.onEmoteClick(emote.getMediaMetadata());
                }
            });
        });
    }

    @Override
    public void beforeSetText(@NonNull TextView textView, @NonNull Spanned markdown) {
        AsyncDrawableScheduler.unschedule(textView);
    }

    @Override
    public void afterSetText(@NonNull TextView textView) {
        AsyncDrawableScheduler.schedule(textView);
    }

    public void setDataSavingMode(boolean dataSavingMode) {
        this.dataSavingMode = dataSavingMode;
    }

    private static class GlideAsyncDrawableLoader extends AsyncDrawableLoader {

        private final GlideStore glideStore;
        private final Map<AsyncDrawable, Target<?>> cache = new HashMap<>(2);

        GlideAsyncDrawableLoader(@NonNull GlideStore glideStore) {
            this.glideStore = glideStore;
        }

        @Override
        public void load(@NonNull AsyncDrawable drawable) {
            final Target<Drawable> target = new AsyncDrawableTarget(drawable);
            cache.put(drawable, target);
            glideStore.load(drawable)
                    .into(target);
        }

        @Override
        public void cancel(@NonNull AsyncDrawable drawable) {
            final Target<?> target = cache.remove(drawable);
            if (target != null) {
                glideStore.cancel(target);
            }
        }

        @Nullable
        @Override
        public Drawable placeholder(@NonNull AsyncDrawable drawable) {
            return null;
        }

        private class AsyncDrawableTarget extends CustomTarget<Drawable> {

            private final AsyncDrawable drawable;

            AsyncDrawableTarget(@NonNull AsyncDrawable drawable) {
                this.drawable = drawable;
            }

            @Override
            public void onResourceReady(@NonNull Drawable resource, @Nullable Transition<? super Drawable> transition) {
                if (cache.remove(drawable) != null) {
                    if (drawable.isAttached()) {
                        DrawableUtils.applyIntrinsicBoundsIfEmpty(resource);
                        drawable.setResult(resource);
                        if (resource instanceof Animatable) {
                            ((Animatable) resource).start();
                        }
                    }
                }
            }

            @Override
            public void onLoadStarted(@Nullable Drawable placeholder) {
                if (placeholder != null
                        && drawable.isAttached()) {
                    DrawableUtils.applyIntrinsicBoundsIfEmpty(placeholder);
                    drawable.setResult(placeholder);
                }
            }

            @Override
            public void onLoadFailed(@Nullable Drawable errorDrawable) {
                if (cache.remove(drawable) != null) {
                    if (errorDrawable != null
                            && drawable.isAttached()) {
                        DrawableUtils.applyIntrinsicBoundsIfEmpty(errorDrawable);
                        drawable.setResult(errorDrawable);
                    }
                }
            }

            @Override
            public void onLoadCleared(@Nullable Drawable placeholder) {
                // we won't be checking if target is still present as cancellation
                // must remove target anyway
                if (drawable.isAttached()) {
                    drawable.clearResult();
                }
            }
        }
    }
}