package mx.trackermap.TrackerMap.android.details

import android.os.Bundle
import androidx.fragment.app.Fragment
import androidx.fragment.app.FragmentActivity
import androidx.viewpager2.adapter.FragmentStateAdapter
import mx.trackermap.TrackerMap.android.details.commands.UnitCommandsFragment
import mx.trackermap.TrackerMap.android.details.information.UnitInformationFragment
import mx.trackermap.TrackerMap.android.details.reports.UnitReportsFragment

class UnitDetailsAdapter(
    activity: FragmentActivity,
    private val deviceId: Int
) : FragmentStateAdapter(activity) {

    companion object {
        val DEVICE_ID_ARG = "device_id"
    }

    override fun getItemCount(): Int = 3

    override fun createFragment(position: Int): Fragment {
        val fragment = when (position) {
            0 -> UnitInformationFragment()
            1 -> UnitReportsFragment()
            else -> UnitCommandsFragment()
        }
        fragment.arguments = Bundle().apply {
            putInt(DEVICE_ID_ARG, deviceId)
        }
        return fragment
    }

}