fix update warning dismiss

app/src/main/java/com/geode/launcher/MainActivity.kt

@@ -234,15 +234,13 @@ fun LoadFailedDialog(returnTitle: String, returnMessage: String) {
 }
 
 @Composable
-fun UpdateWarning(inSafeMode: Boolean = false) {
+fun UpdateWarning(inSafeMode: Boolean = false, onDismiss: () -> Unit) {
     val context = LocalContext.current
     val packageManager = context.packageManager
 
     val gdVersionCode = remember { LaunchUtils.getGeometryDashVersionCode(packageManager) }
     val gdVersionString = remember { LaunchUtils.getGeometryDashVersionString(packageManager) }
 
-    var showUpdateWarning by remember { mutableStateOf(true) }
-
     var lastDismissedVersion by PreferenceUtils.useLongPreference(
         preferenceKey = PreferenceUtils.Key.DISMISSED_GJ_UPDATE
     )
@@ -251,7 +249,6 @@ fun UpdateWarning(inSafeMode: Boolean = false) {
     val shouldDismiss = canDismissRelease && gdVersionCode == lastDismissedVersion
 
     if (gdVersionCode != Constants.SUPPORTED_VERSION_CODE && !shouldDismiss) {
-        if (showUpdateWarning) {
         AlertDialog(
             icon = {
                 Icon(
@@ -275,7 +272,7 @@ fun UpdateWarning(inSafeMode: Boolean = false) {
             },
             confirmButton = {
                 TextButton(onClick = {
-                        showUpdateWarning = false
+                    onDismiss()
                     lastDismissedVersion = gdVersionCode
 
                     onLaunch(context, inSafeMode)
@@ -284,13 +281,12 @@ fun UpdateWarning(inSafeMode: Boolean = false) {
                 }
             },
             dismissButton = {
-                    TextButton(onClick = { showUpdateWarning = false }) {
+                TextButton(onClick = { onDismiss() }) {
                     Text(stringResource(R.string.message_box_cancel))
                 }
             },
-                onDismissRequest = { showUpdateWarning = false }
+            onDismissRequest = { onDismiss() }
         )
-        }
     } else {
         LaunchedEffect(gdVersionCode) {
             onLaunch(context, inSafeMode)
@@ -621,7 +617,10 @@ fun MainScreen(
     }
 
     if (beginLaunch) {
-        UpdateWarning(launchInSafeMode)
+        UpdateWarning(launchInSafeMode) {
+            beginLaunch = false
+            launchInSafeMode = false
+        }
     }
 
     if (showSafeModeDialog) {
@@ -632,6 +631,8 @@ fun MainScreen(
             onLaunch = {
                 launchInSafeMode = true
                 beginLaunch = true
+
+                showSafeModeDialog = false
             }
         )
     }