BarInsta/app/src/main/java/awais/instagrabber/asyncs/direct_messages/DirectMessageInboxThreadFetcher.java

package awais.instagrabber.asyncs.direct_messages;

import android.os.AsyncTask;
import android.util.Log;

import androidx.annotation.Nullable;

import org.json.JSONObject;

import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

import awais.instagrabber.BuildConfig;
import awais.instagrabber.interfaces.FetchListener;
import awais.instagrabber.models.direct_messages.InboxThreadModel;
import awais.instagrabber.models.enums.UserInboxDirection;
import awais.instagrabber.utils.Constants;
import awais.instagrabber.utils.LocaleUtils;
import awais.instagrabber.utils.NetworkUtils;
import awais.instagrabber.utils.ResponseBodyUtils;
import awais.instagrabber.utils.TextUtils;

import static awais.instagrabber.utils.Utils.logCollector;
import static awaisomereport.LogCollector.LogFile;

public final class DirectMessageInboxThreadFetcher extends AsyncTask<Void, Void, InboxThreadModel> {
    private static final String TAG = "DMInboxThreadFetcher";

    private final String id;
    private final String endCursor;
    private final FetchListener<InboxThreadModel> fetchListener;
    private final UserInboxDirection direction;

    public DirectMessageInboxThreadFetcher(final String id,
                                           final UserInboxDirection direction,
                                           final String cursor,
                                           final FetchListener<InboxThreadModel> fetchListener) {
        this.id = id;
        this.direction = direction;
        this.endCursor = cursor;
        this.fetchListener = fetchListener;
    }

    @Nullable
    @Override
    protected InboxThreadModel doInBackground(final Void... voids) {
        InboxThreadModel result = null;
        final Map<String, String> queryParamsMap = new HashMap<>();
        queryParamsMap.put("visual_message_return_type", "unseen");
        if (direction != null) queryParamsMap.put("direction", direction.getValue());
        if (!TextUtils.isEmpty(endCursor)) {
            queryParamsMap.put("cursor", endCursor);
        }
        final String queryString = NetworkUtils.getQueryString(queryParamsMap);
        final String url = "https://i.instagram.com/api/v1/direct_v2/threads/" + id + "/?" + queryString;
        try {
            final HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection();
            conn.setRequestProperty("User-Agent", Constants.I_USER_AGENT);
            conn.setRequestProperty("Accept-Language", LocaleUtils.getCurrentLocale().getLanguage() + ",en-US;q=0.8");
            conn.setUseCaches(false);

            if (conn.getResponseCode() == HttpURLConnection.HTTP_OK) {
                final JSONObject data = new JSONObject(NetworkUtils.readFromConnection(conn)).getJSONObject("thread");
                result = ResponseBodyUtils.createInboxThreadModel(data, true);
            }

            conn.disconnect();
        } catch (final Exception e) {
            result = null;
            if (logCollector != null)
                logCollector.appendException(e, LogFile.ASYNC_DMS_THREAD, "doInBackground");
            if (BuildConfig.DEBUG) Log.e(TAG, "", e);
        }
        return result;
    }

    @Override
    protected void onPreExecute() {
        if (fetchListener != null) fetchListener.doBefore();
    }

    @Override
    protected void onPostExecute(final InboxThreadModel inboxThreadModel) {
        if (fetchListener != null) fetchListener.onResult(inboxThreadModel);
    }
}