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

package com.capstone.foodify.Fragment;

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.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 ProfileFragment();
            default: return new HomeFragment();
        }
    }

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