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 kotlinx.coroutines.DelicateCoroutinesApi
import mx.trackermap.TrackerMap.android.details.commands.UnitCommandsFragment
import mx.trackermap.TrackerMap.android.details.information.UnitInformationFragment
import mx.trackermap.TrackerMap.android.details.reports.UnitReportsFragment

@DelicateCoroutinesApi
class UnitDetailsAdapter(
    activity: FragmentActivity,
    private val deviceId: Int,
    private val deviceName: String,
    private val deviceCategory: String
) : FragmentStateAdapter(activity) {

    companion object {
        const val DEVICE_ID_ARG = "device_id"
        const val DEVICE_NAME_ARG = "device_name"
        const val DEVICE_CATEGORY_ARG = "device_category"
    }

    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)
            putString(DEVICE_NAME_ARG, deviceName)
            putString(DEVICE_CATEGORY_ARG, deviceCategory)
        }
        return fragment
    }

}