import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import com.jksalcedo.timer.CountdownTimer
import com.jksalcedo.timer.Stopwatch
import kotlinx.coroutines.flow.SharingStarted
import kotlinx.coroutines.flow.map
import kotlinx.coroutines.flow.stateIn
import kotlin.time.Duration
import kotlin.time.Duration.Companion.seconds

class TimerViewModel : ViewModel() {

    // --- Countdown Timer ---
    private val countdownTimer = CountdownTimer(viewModelScope)

    val countdownText = countdownTimer.remainingTime
        .map { duration -> formatDuration(duration) }
        .stateIn(
            viewModelScope,
            SharingStarted.WhileSubscribed(5000),
            formatDuration(Duration.ZERO)
        )

    val isCountdownRunning = countdownTimer.isRunning.asStateFlow()

    fun startCountdown(duration: Duration) { countdownTimer.start(duration) }
    fun pauseCountdown() { countdownTimer.pause() }
    fun resumeCountdown() { countdownTimer.resume() }
    fun stopCountdown() { countdownTimer.stop() }
    fun restartCountdown() { countdownTimer.restart() }

    // --- Stopwatch ---
    private val stopwatch = Stopwatch(viewModelScope)

    val stopwatchText = stopwatch.elapsedTime
        .map { duration -> formatDuration(duration, includeMilliseconds = true) }
        .stateIn(
            viewModelScope,
            SharingStarted.WhileSubscribed(5000),
            formatDuration(Duration.ZERO, includeMilliseconds = true)
        )

    val isStopwatchRunning = stopwatch.isRunning.asStateFlow()
    val lapTimes = stopwatch.lapTimes.asStateFlow()
    val splitTimes = stopwatch.splitTimes.asStateFlow()

    fun startStopwatch() { stopwatch.start() }
    fun pauseStopwatch() { stopwatch.pause() }
    fun stopStopwatch() { stopwatch.stop() }
    fun lapStopwatch() { stopwatch.lap() }
    fun splitStopwatch() { stopwatch.split() }
    fun resetStopwatch() { stopwatch.reset() }

    // --- Utility formatting function ---
    private fun formatDuration(duration: Duration, includeMilliseconds: Boolean = false): String {
        val totalSeconds = duration.inWholeSeconds
        val minutes = totalSeconds / 60
        val seconds = totalSeconds % 60
        val milliseconds = (duration - totalSeconds.seconds).inWholeMilliseconds

        return if (includeMilliseconds) {
            String.format("%02d:%02d.%03d", minutes, seconds, milliseconds)
        } else {
            String.format("%02d:%02d", minutes, seconds)
        }
    }
}

import android.os.Bundle
import android.widget.Button
import android.widget.TextView
import androidx.activity.viewModels
import androidx.appcompat.app.AppCompatActivity
import androidx.lifecycle.Lifecycle
import androidx.lifecycle.lifecycleScope
import androidx.lifecycle.repeatOnLifecycle
import kotlinx.coroutines.launch
import kotlin.time.Duration.Companion.seconds

class MainActivity : AppCompatActivity() {

    private val timerViewModel: TimerViewModel by viewModels()

    // UI elements (declare as lateinit var and initialize in onCreate)
    private lateinit var countdownTextView: TextView
    private lateinit var startCountdownButton: Button
    private lateinit var var pauseCountdownButton: Button
    private lateinit var var resumeCountdownButton: Button
    private lateinit var var stopCountdownButton: Button
    private lateinit var var restartCountdownButton: Button

    private lateinit var stopwatchTextView: TextView
    private lateinit var startStopwatchButton: Button
    private lateinit var pauseStopwatchButton: Button
    private lateinit var lapStopwatchButton: Button
    private lateinit var splitStopwatchButton: Button
    private lateinit var stopStopwatchButton: Button
    private lateinit var resetStopwatchButton: Button
    private lateinit var lapTimesTextView: TextView
    private lateinit var splitTimesTextView: TextView

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_main)

        // Initialize UI elements
        countdownTextView = findViewById(R.id.countdownTextView)
        startCountdownButton = findViewById(R.id.startCountdownButton)
        pauseCountdownButton = findViewById(R.id.pauseCountdownButton)
        resumeCountdownButton = findViewById(R.id.resumeCountdownButton)
        stopCountdownButton = findViewById(R.id.stopCountdownButton)
        restartCountdownButton = findViewById(R.id.restartCountdownButton)

        stopwatchTextView = findViewById(R.id.stopwatchTextView)
        startStopwatchButton = findViewById(R.id.startStopwatchButton)
        pauseStopwatchButton = findViewById(R.id.pauseStopwatchButton)
        lapStopwatchButton = findViewById(R.id.lapStopwatchButton)
        splitStopwatchButton = findViewById(R.id.splitStopwatchButton)
        stopStopwatchButton = findViewById(R.id.stopStopwatchButton)
        resetStopwatchButton = findViewById(R.id.resetStopwatchButton)
        lapTimesTextView = findViewById(R.id.lapTimesTextView)
        splitTimesTextView = findViewById(R.id.splitTimesTextView)

        // Set up click listeners for countdown timer buttons
        startCountdownButton.setOnClickListener { timerViewModel.startCountdown(60.seconds) }
        pauseCountdownButton.setOnClickListener { timerViewModel.pauseCountdown() }
        resumeCountdownButton.setOnClickListener { timerViewModel.resumeCountdown() }
        stopCountdownButton.setOnClickListener { timerViewModel.stopCountdown() }
        restartCountdownButton.setOnClickListener { timerViewModel.restartCountdown() }

        // Set up click listeners for stopwatch buttons
        startStopwatchButton.setOnClickListener { timerViewModel.startStopwatch() }
        pauseStopwatchButton.setOnClickListener { timerViewModel.pauseStopwatch() }
        lapStopwatchButton.setOnClickListener { timerViewModel.lapStopwatch() }
        splitStopwatchButton.setOnClickListener { timerViewModel.splitStopwatch() }
        stopStopwatchButton.setOnClickListener { timerViewModel.stopStopwatch() }
        resetStopwatchButton.setOnClickListener { timerViewModel.resetStopwatch() }

        // Observe the ViewModel's StateFlows and update the UI
        lifecycleScope.launch {
            repeatOnLifecycle(Lifecycle.State.STARTED) {
                launch { timerViewModel.countdownText.collect { text -> countdownTextView.text = text } }
                launch { timerViewModel.stopwatchText.collect { text -> stopwatchTextView.text = text } }
                launch {
                    timerViewModel.lapTimes.collect { lapTimes ->
                        val formattedLaps = lapTimes.joinToString(separator = "\n") { duration ->
                            val totalSeconds = duration.inWholeSeconds
                            val minutes = totalSeconds / 60
                            val seconds = totalSeconds % 60
                            val milliseconds = (duration - totalSeconds.seconds).inWholeMilliseconds
                            String.format("%02d:%02d.%03d", minutes, seconds, milliseconds)
                        }
                        lapTimesTextView.text = if (lapTimes.isEmpty()) "Lap Times: " else "Lap Times:\n$formattedLaps"
                    }
                }
                launch {
                    timerViewModel.splitTimes.collect { splitTimes ->
                        val formattedSplits = splitTimes.joinToString(separator = "\n") { duration ->
                            val totalSeconds = duration.inWholeSeconds
                            val minutes = totalSeconds / 60
                            val seconds = totalSeconds % 60
                            val milliseconds = (duration - totalSeconds.seconds).inWholeMilliseconds
                            String.format("%02d:%02d.%03d", minutes, seconds, milliseconds)
                        }
                        splitTimesTextView.text = if (splitTimes.isEmpty()) "Split Times: " else "Split Times:\n$formattedSplits"
                    }
                }
            }
        }
    }
}