Foodify/app/src/main/java/com/capstone/foodify/ViewPagerAdapter.java

package com.capstone.foodify;

import androidx.annotation.NonNull;
import androidx.fragment.app.Fragment;
import androidx.fragment.app.FragmentActivity;
import androidx.viewpager2.adapter.FragmentStateAdapter;

import com.capstone.foodify.Fragment.BasketFragment;
import com.capstone.foodify.Fragment.FavoriteFragment;
import com.capstone.foodify.Fragment.HomeFragment;
import com.capstone.foodify.Fragment.ProfileFragment;
import com.capstone.foodify.Fragment.SearchFragment;

public class ViewPagerAdapter extends FragmentStateAdapter {

    public ViewPagerAdapter(@NonNull FragmentActivity fragmentActivity) {
        super(fragmentActivity);
    }

    @NonNull
    @Override
    public Fragment createFragment(int position) {
        switch(position) {
            case 1: return new SearchFragment();
            case 2: return new BasketFragment();
            case 3: return new FavoriteFragment();
            case 4: return new ProfileFragment();
            default: return new HomeFragment();
        }
    }

    @Override
    public int getItemCount() {
        return 5;
    }
}