From d0805cd95dafd9ef4b050d998813fe3136d28cf4 Mon Sep 17 00:00:00 2001
From: manu <mmmateos@protonmail.com>
Date: Mon, 1 Jul 2024 15:09:52 +0200
Subject: [PATCH] fix: [ANDROAPP-6271] crash when date format make year out of
 range

---
 .../designsystem/component/InputDateTime.kt   | 91 +++++++++++--------
 1 file changed, 51 insertions(+), 40 deletions(-)

diff --git a/designsystem/src/commonMain/kotlin/org/hisp/dhis/mobile/ui/designsystem/component/InputDateTime.kt b/designsystem/src/commonMain/kotlin/org/hisp/dhis/mobile/ui/designsystem/component/InputDateTime.kt
index e0fcc5b2f..7ca8f7182 100644
--- a/designsystem/src/commonMain/kotlin/org/hisp/dhis/mobile/ui/designsystem/component/InputDateTime.kt
+++ b/designsystem/src/commonMain/kotlin/org/hisp/dhis/mobile/ui/designsystem/component/InputDateTime.kt
@@ -18,6 +18,7 @@ import androidx.compose.material3.DatePicker
 import androidx.compose.material3.DatePickerColors
 import androidx.compose.material3.DatePickerDefaults
 import androidx.compose.material3.DatePickerDialog
+import androidx.compose.material3.DatePickerState
 import androidx.compose.material3.ExperimentalMaterial3Api
 import androidx.compose.material3.Icon
 import androidx.compose.material3.MaterialTheme
@@ -238,31 +239,7 @@ fun InputDateTime(
         },
         inputStyle = uiModel.inputStyle,
     )
-    var datePickerState = rememberDatePickerState(
-        selectableDates = getSelectableDates(uiModel),
-    )
-    if (!uiModel.inputTextFieldValue?.text.isNullOrEmpty() && uiModel.actionType != DateTimeActionType.TIME) {
-        datePickerState = if (uiModel.actionType == DateTimeActionType.DATE_TIME && uiModel.inputTextFieldValue?.text?.length == 12 && yearIsInRange(uiModel.inputTextFieldValue.text.substring(0, 8), uiModel.yearRange)) {
-            rememberDatePickerState(
-                initialSelectedDateMillis = parseStringDateToMillis(
-                    uiModel.inputTextFieldValue.text.substring(0, uiModel.inputTextFieldValue.text.length - 4),
-                    pattern = uiModel.format,
-                ),
-                yearRange = uiModel.yearRange,
-                selectableDates = getSelectableDates(uiModel),
-            )
-        } else {
-            if (uiModel.inputTextFieldValue?.text?.length == 8 && yearIsInRange(uiModel.inputTextFieldValue.text, uiModel.yearRange)) {
-                rememberDatePickerState(
-                    initialSelectedDateMillis = parseStringDateToMillis(uiModel.inputTextFieldValue.text, uiModel.format),
-                    yearRange = uiModel.yearRange,
-                    selectableDates = getSelectableDates(uiModel),
-                )
-            } else {
-                datePickerState
-            }
-        }
-    }
+    val datePickerState = provideDatePickerState(uiModel)
 
     if (showDatePicker) {
         MaterialTheme(
@@ -428,7 +405,7 @@ private fun getSupportingTextList(uiModel: InputDateTimeModel, dateOutOfRangeIte
                         ),
                         uiModel.selectableDates, uiModel.format,
                     )
-                    dateIsInYearRange = yearIsInRange(uiModel.inputTextFieldValue.text.substring(0, uiModel.inputTextFieldValue.text.length - 4), uiModel.yearRange)
+                    dateIsInYearRange = yearIsInRange(uiModel.inputTextFieldValue.text, getDefaultFormat(uiModel.actionType), uiModel.yearRange)
                     isValidHourFormat = isValidHourFormat(uiModel.inputTextFieldValue.text.substring(8, 12))
                     if (!dateIsInRange || !dateIsInYearRange) supportingTextList.add(dateOutOfRangeItem)
                     if (!isValidHourFormat) supportingTextList.add(incorrectHourFormatItem)
@@ -437,7 +414,7 @@ private fun getSupportingTextList(uiModel: InputDateTimeModel, dateOutOfRangeIte
             DateTimeActionType.DATE -> {
                 if (uiModel.inputTextFieldValue?.text!!.length == 8) {
                     dateIsInRange = dateIsInRange(parseStringDateToMillis(uiModel.inputTextFieldValue.text), uiModel.selectableDates, uiModel.format)
-                    dateIsInYearRange = yearIsInRange(uiModel.inputTextFieldValue.text, uiModel.yearRange)
+                    dateIsInYearRange = yearIsInRange(uiModel.inputTextFieldValue.text, getDefaultFormat(uiModel.actionType), uiModel.yearRange)
                     if (!dateIsInRange || !dateIsInYearRange) supportingTextList.add(dateOutOfRangeItem)
                 }
             }
@@ -446,6 +423,32 @@ private fun getSupportingTextList(uiModel: InputDateTimeModel, dateOutOfRangeIte
     return supportingTextList.toList()
 }
 
+@Composable
+@OptIn(ExperimentalMaterial3Api::class)
+private fun provideDatePickerState(uiModel: InputDateTimeModel): DatePickerState {
+    return uiModel.inputTextFieldValue?.text?.takeIf {
+        it.isNotEmpty() &&
+            yearIsInRange(it, getDefaultFormat(uiModel.actionType), uiModel.yearRange)
+    }?.let {
+        rememberDatePickerState(
+            initialSelectedDateMillis = parseStringDateToMillis(
+                dateString = it,
+                pattern = getDefaultFormat(uiModel.actionType),
+            ),
+            yearRange = uiModel.yearRange,
+            selectableDates = getSelectableDates(uiModel),
+        )
+    } ?: rememberDatePickerState(selectableDates = getSelectableDates(uiModel))
+}
+
+private fun getDefaultFormat(actionType: DateTimeActionType): String {
+    return when (actionType) {
+        DateTimeActionType.DATE -> "ddMMyyyy"
+        DateTimeActionType.TIME -> "HHmm"
+        DateTimeActionType.DATE_TIME -> "ddMMyyyyHHmm"
+    }
+}
+
 enum class DateTimeActionType {
     DATE, TIME, DATE_TIME
 }
@@ -547,16 +550,12 @@ private fun getTime(timePickerState: TimePickerState, format: String? = "HHmm"):
     return formater.format(cal.time)
 }
 
-private fun parseStringDateToMillis(dateString: String, pattern: String = "ddMMyyyy", locale: Locale = Locale.getDefault()): Long {
-    return if (dateString.isNotEmpty()) {
-        val cal = Calendar.getInstance()
-        val sdf = SimpleDateFormat(pattern, locale)
-        sdf.timeZone = TimeZone.getTimeZone("UTC")
-        cal.time = sdf.parse(dateString) ?: Date()
+private fun parseStringDateToMillis(dateString: String, pattern: String = "ddMMyyyy"): Long {
+    val cal = Calendar.getInstance()
+    return dateString.parseDate(pattern)?.let {
+        cal.time = it
         cal.timeInMillis
-    } else {
-        0L
-    }
+    } ?: 0L
 }
 
 data class SelectableDates(
@@ -607,10 +606,22 @@ internal fun dateIsInRange(date: Long, allowedDates: SelectableDates, format: St
         )
 }
 
-fun yearIsInRange(date: String, yearRange: IntRange): Boolean {
-    return (
-        yearRange.contains(date.substring(date.length - 4, date.length).toInt())
-        )
+fun yearIsInRange(date: String, pattern: String, yearRange: IntRange): Boolean {
+    val cal = Calendar.getInstance()
+    return date.parseDate(pattern)?.let {
+        cal.time = it
+        yearRange.contains(cal.get(Calendar.YEAR))
+    } ?: false
+}
+
+fun String.parseDate(pattern: String): Date? {
+    return if (isNotEmpty() && length == pattern.length) {
+        val sdf = SimpleDateFormat(pattern, Locale.getDefault())
+        sdf.timeZone = TimeZone.getTimeZone("UTC")
+        sdf.parse(this)
+    } else {
+        null
+    }
 }
 
 @OptIn(ExperimentalMaterial3Api::class)