package app.fedilab.android.ui.pageadapter;
/* Copyright 2022 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>. */

import android.os.Bundle;
import android.view.ViewGroup;

import androidx.annotation.NonNull;
import androidx.fragment.app.Fragment;
import androidx.fragment.app.FragmentManager;
import androidx.fragment.app.FragmentStatePagerAdapter;

import app.fedilab.android.client.entities.Pinned;
import app.fedilab.android.client.entities.Timeline;
import app.fedilab.android.client.entities.app.PinnedTimeline;
import app.fedilab.android.helper.Helper;
import app.fedilab.android.ui.fragment.timeline.FragmentMastodonConversation;
import app.fedilab.android.ui.fragment.timeline.FragmentMastodonTimeline;
import app.fedilab.android.ui.fragment.timeline.FragmentNotificationContainer;

public class FedilabPageAdapter extends FragmentStatePagerAdapter {

    public static final int BOTTOM_TIMELINE_COUNT = 5; //home, local, public, notification, DM
    private final Pinned pinned;
    private Fragment mCurrentFragment;

    public FedilabPageAdapter(FragmentManager fm, Pinned pinned) {
        super(fm, BEHAVIOR_RESUME_ONLY_CURRENT_FRAGMENT);
        this.pinned = pinned;
    }

    public Fragment getCurrentFragment() {
        return mCurrentFragment;
    }

    @Override
    public void setPrimaryItem(@NonNull ViewGroup container, int position, @NonNull Object object) {
        if (getCurrentFragment() != object) {
            mCurrentFragment = ((Fragment) object);
        }
        super.setPrimaryItem(container, position, object);
    }

    @Override
    public int getItemPosition(@NonNull Object object) {
        // POSITION_NONE makes it possible to reload the PagerAdapter
        return POSITION_NONE;
    }

    @NonNull
    @Override
    public Fragment getItem(int position) {

        //Position 3 is for notifications
        if (position == 3) {
            return new FragmentNotificationContainer();
        } else if (position == 4) { //Position 4 is for DM
            return new FragmentMastodonConversation();
        } else {
            FragmentMastodonTimeline fragment = new FragmentMastodonTimeline();
            Bundle bundle = new Bundle();
            if (position == 0) { //Home timeline
                bundle.putSerializable(Helper.ARG_TIMELINE_TYPE, Timeline.TimeLineEnum.HOME);
            } else if (position == 1) { //Local timeline
                bundle.putSerializable(Helper.ARG_TIMELINE_TYPE, Timeline.TimeLineEnum.LOCAL);
            } else if (position == 2) { //Public timeline
                bundle.putSerializable(Helper.ARG_TIMELINE_TYPE, Timeline.TimeLineEnum.PUBLIC);
            } else { //Pinned timeline
                int pinnedPosition = position - BOTTOM_TIMELINE_COUNT; //Real position has an offset.
                PinnedTimeline pinnedTimeline = pinned.pinnedTimelines.get(pinnedPosition);
                bundle.putSerializable(Helper.ARG_TIMELINE_TYPE, pinnedTimeline.type);

                if (pinnedTimeline.type == Timeline.TimeLineEnum.LIST) {
                    bundle.putString(Helper.ARG_LIST_ID, pinnedTimeline.mastodonList.id);
                } else if (pinnedTimeline.type == Timeline.TimeLineEnum.TAG) {
                    String tag = pinnedTimeline.tagTimeline.name.replaceAll("#", "");
                    bundle.putString(Helper.ARG_SEARCH_KEYWORD, tag);
                } else if (pinnedTimeline.type == Timeline.TimeLineEnum.REMOTE) {
                    String instance = pinnedTimeline.remoteInstance.host;
                    bundle.putString(Helper.ARG_REMOTE_INSTANCE, instance);
                }
            }
            bundle.putString(Helper.ARG_VIEW_MODEL_KEY, "FEDILAB_" + position);
            fragment.setArguments(bundle);
            return fragment;
        }
    }

    @Override
    public int getCount() {
        if (pinned != null && pinned.pinnedTimelines != null) {
            return pinned.pinnedTimelines.size() + BOTTOM_TIMELINE_COUNT;
        } else {
            return BOTTOM_TIMELINE_COUNT;
        }
    }
}