/* Copyright 2017 Thomas Schneider
 *
 * This file is a part of Fedilab
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the
 * GNU General Public License as published by the Free Software Foundation; either version 3 of the
 * License, or (at your option) any later version.
 *
 * Fedilab is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
 * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
 * Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with Fedilab; if not,
 * see <http://www.gnu.org/licenses>. */
package app.fedilab.android.asynctasks;

import android.content.Context;
import android.database.sqlite.SQLiteDatabase;
import android.os.Handler;
import android.os.Looper;

import java.lang.ref.WeakReference;
import java.util.ArrayList;
import java.util.List;

import app.fedilab.android.activities.MainActivity;
import app.fedilab.android.client.API;
import app.fedilab.android.client.APIResponse;
import app.fedilab.android.client.Entities.Account;
import app.fedilab.android.client.Entities.Conversation;
import app.fedilab.android.client.Entities.ManageTimelines;
import app.fedilab.android.client.Entities.TagTimeline;
import app.fedilab.android.client.GNUAPI;
import app.fedilab.android.interfaces.OnRetrieveMissingFeedsInterface;
import app.fedilab.android.sqlite.SearchDAO;
import app.fedilab.android.sqlite.Sqlite;
import app.fedilab.android.sqlite.TimelinesDAO;


/**
 * Created by Thomas on 27/09/2017.
 * Retrieves missing toots since last pause
 */

public class RetrieveMissingFeedsAsyncTask {

    private final String since_id;
    private final OnRetrieveMissingFeedsInterface listener;
    private final List<app.fedilab.android.client.Entities.Status> statuses = new ArrayList<>();
    private final RetrieveFeedsAsyncTask.Type type;
    private final WeakReference<Context> contextReference;
    private String remoteInstance;
    private int timelineId;

    public RetrieveMissingFeedsAsyncTask(Context context, String since_id, RetrieveFeedsAsyncTask.Type type, OnRetrieveMissingFeedsInterface onRetrieveMissingFeedsInterface) {
        this.contextReference = new WeakReference<>(context);
        this.since_id = since_id;
        this.listener = onRetrieveMissingFeedsInterface;
        this.type = type;
        doInBackground();
    }

    public RetrieveMissingFeedsAsyncTask(Context context, String remoteInstance, String since_id, RetrieveFeedsAsyncTask.Type type, OnRetrieveMissingFeedsInterface onRetrieveMissingFeedsInterface) {
        this.contextReference = new WeakReference<>(context);
        this.since_id = since_id;
        this.listener = onRetrieveMissingFeedsInterface;
        this.type = type;
        this.remoteInstance = remoteInstance;
        doInBackground();
    }

    public RetrieveMissingFeedsAsyncTask(Context context, int timelineId, String since_id, RetrieveFeedsAsyncTask.Type type, OnRetrieveMissingFeedsInterface onRetrieveMissingFeedsInterface) {
        this.contextReference = new WeakReference<>(context);
        this.since_id = since_id;
        this.listener = onRetrieveMissingFeedsInterface;
        this.type = type;
        this.timelineId = timelineId;
        doInBackground();
    }

    protected void doInBackground() {
        new Thread(() -> {
            if (this.contextReference.get() == null) {
                Handler mainHandler = new Handler(Looper.getMainLooper());
                Runnable myRunnable = () -> listener.onRetrieveMissingFeeds(statuses);
                mainHandler.post(myRunnable);
                return;
            }
            if (MainActivity.social != UpdateAccountInfoAsyncTask.SOCIAL.GNU && MainActivity.social != UpdateAccountInfoAsyncTask.SOCIAL.FRIENDICA) {
                API api = new API(this.contextReference.get());
                List<app.fedilab.android.client.Entities.Status> tempStatus = null;
                APIResponse apiResponse = null;
                if (type == RetrieveFeedsAsyncTask.Type.HOME) {
                    apiResponse = api.getHomeTimelineSinceId(since_id);
                } else if (type == RetrieveFeedsAsyncTask.Type.DIRECT)
                    apiResponse = api.getDirectTimelineSinceId(since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.CONVERSATION)
                    apiResponse = api.getConversationTimelineSinceId(since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.LOCAL)
                    apiResponse = api.getPublicTimelineSinceId(true, since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.PUBLIC)
                    apiResponse = api.getPublicTimelineSinceId(false, since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.REMOTE_INSTANCE)
                    apiResponse = api.getInstanceTimelineSinceId(remoteInstance, since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.TAG) {
                    SQLiteDatabase db = Sqlite.getInstance(contextReference.get().getApplicationContext(), Sqlite.DB_NAME, null, Sqlite.DB_VERSION).open();
                    ManageTimelines manageTimelines = new TimelinesDAO(contextReference.get(), db).getById(timelineId);
                    if (manageTimelines != null && manageTimelines.getTagTimeline() != null) {
                        TagTimeline tagTimeline = manageTimelines.getTagTimeline();
                        boolean isArt = tagTimeline.isART();
                        if (isArt)
                            apiResponse = api.getCustomArtTimelineSinceId(false, manageTimelines.getTagTimeline().getName(), since_id, tagTimeline.getAny(), tagTimeline.getAll(), tagTimeline.getNone());
                        else
                            apiResponse = api.getPublicTimelineTagSinceId(manageTimelines.getTagTimeline().getName(), false, since_id, tagTimeline.getAny(), tagTimeline.getAll(), tagTimeline.getNone());
                    }
                } else if (type == RetrieveFeedsAsyncTask.Type.ART)
                    apiResponse = api.getArtTimelineSinceId(false, since_id, null, null, null);
                if (apiResponse != null) {
                    if (type != RetrieveFeedsAsyncTask.Type.CONVERSATION)
                        tempStatus = apiResponse.getStatuses();
                    else {
                        List<Conversation> conversations = apiResponse.getConversations();
                        tempStatus = new ArrayList<>();
                        if (conversations != null && conversations.size() > 0) {
                            for (Conversation conversation : conversations) {
                                app.fedilab.android.client.Entities.Status status = conversation.getLast_status();
                                if (status != null) {
                                    List<String> ppConversation = new ArrayList<>();
                                    for (Account account : conversation.getAccounts())
                                        ppConversation.add(account.getAvatar());
                                    status.setConversationProfilePicture(ppConversation);
                                    status.setConversationId(conversation.getId());
                                    tempStatus.add(status);
                                }
                            }
                        }
                    }

                    if (tempStatus != null)
                        statuses.addAll(0, tempStatus);
                }
                if (type == RetrieveFeedsAsyncTask.Type.HOME && statuses.size() > 0) {
                    MainActivity.lastHomeId = statuses.get(0).getId();
                }
            } else {
                GNUAPI gnuapi = new GNUAPI(this.contextReference.get());
                List<app.fedilab.android.client.Entities.Status> tempStatus = null;
                APIResponse apiResponse = null;
                if (type == RetrieveFeedsAsyncTask.Type.GNU_HOME) {
                    apiResponse = gnuapi.getHomeTimelineSinceId(since_id);
                } else if (type == RetrieveFeedsAsyncTask.Type.GNU_LOCAL)
                    apiResponse = gnuapi.getPublicTimelineSinceId(true, since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.GNU_WHOLE)
                    apiResponse = gnuapi.getPublicTimelineSinceId(false, since_id);
                else if (type == RetrieveFeedsAsyncTask.Type.GNU_TAG) {
                    SQLiteDatabase db = Sqlite.getInstance(contextReference.get().getApplicationContext(), Sqlite.DB_NAME, null, Sqlite.DB_VERSION).open();
                    List<TagTimeline> tagTimelines = new SearchDAO(contextReference.get(), db).getTimelineInfo(remoteInstance);
                    if (tagTimelines != null && tagTimelines.size() > 0) {
                        TagTimeline tagTimeline = tagTimelines.get(0);
                        boolean isArt = tagTimeline.isART();
                        if (isArt)
                            apiResponse = gnuapi.getCustomArtTimelineSinceId(false, remoteInstance, since_id, tagTimelines.get(0).getAny(), tagTimelines.get(0).getAll(), tagTimelines.get(0).getNone());
                        else
                            apiResponse = gnuapi.getPublicTimelineTagSinceId(remoteInstance, false, since_id, tagTimelines.get(0).getAny(), tagTimelines.get(0).getAll(), tagTimelines.get(0).getNone());
                    } else {
                        apiResponse = gnuapi.getPublicTimelineTag(remoteInstance, false, since_id, tagTimelines.get(0).getAny(), tagTimelines.get(0).getAll(), tagTimelines.get(0).getNone());
                    }
                }
                if (apiResponse != null) {
                    if (type != RetrieveFeedsAsyncTask.Type.CONVERSATION)
                        tempStatus = apiResponse.getStatuses();
                    else {
                        List<Conversation> conversations = apiResponse.getConversations();
                        tempStatus = new ArrayList<>();
                        if (conversations != null && conversations.size() > 0) {
                            for (Conversation conversation : conversations) {
                                app.fedilab.android.client.Entities.Status status = conversation.getLast_status();
                                if (status != null) {
                                    List<String> ppConversation = new ArrayList<>();
                                    for (Account account : conversation.getAccounts())
                                        ppConversation.add(account.getAvatar());
                                    status.setConversationProfilePicture(ppConversation);
                                    status.setConversationId(conversation.getId());
                                    tempStatus.add(status);
                                }
                            }
                        }
                    }

                    if (tempStatus != null)
                        statuses.addAll(0, tempStatus);
                }
            }
            Handler mainHandler = new Handler(Looper.getMainLooper());
            Runnable myRunnable = () -> listener.onRetrieveMissingFeeds(statuses);
            mainHandler.post(myRunnable);
        }).start();
    }

}