refactor: Simplify code

app/src/main/java/dev/yokai/presentation/extension/repo/ExtensionRepoScreen.kt

@@ -24,12 +24,14 @@ import androidx.compose.ui.platform.LocalContext
 import androidx.compose.ui.res.stringResource
 import androidx.compose.ui.unit.sp
 import androidx.lifecycle.viewmodel.compose.viewModel
+import dev.yokai.domain.ComposableAlertDialog
 import dev.yokai.presentation.AppBarType
 import dev.yokai.presentation.YokaiScaffold
 import dev.yokai.presentation.component.EmptyScreen
 import dev.yokai.presentation.extension.repo.component.ExtensionRepoItem
 import eu.kanade.tachiyomi.R
 import eu.kanade.tachiyomi.util.compose.LocalAlertDialog
+import eu.kanade.tachiyomi.util.compose.currentOrThrow
 import eu.kanade.tachiyomi.util.system.toast
 import kotlinx.coroutines.flow.collectLatest
 
@@ -44,7 +46,7 @@ fun ExtensionRepoScreen(
     val repoState = viewModel.repoState.collectAsState()
     var inputText by remember { mutableStateOf("") }
     val listState = rememberLazyListState()
-    val alertDialog = LocalAlertDialog.current ?: throw RuntimeException("LocalAlertDialog not yet provided")
+    val alertDialog = LocalAlertDialog.currentOrThrow
 
     YokaiScaffold(
         onNavigationIconClicked = onBackPress,
@@ -59,7 +61,6 @@ fun ExtensionRepoScreen(
 
         val repos = (repoState.value as ExtensionRepoState.Success).repos
 
-        if (alertDialog.content != null)
         alertDialog.content?.let { it() }
 
         LazyColumn(
@@ -93,8 +94,30 @@ fun ExtensionRepoScreen(
                     ExtensionRepoItem(
                         repoUrl = repo,
                         onDeleteClick = { repoToDelete ->
-                            alertDialog.content =
+                            alertDialog.content = { ExtensionRepoDeletePrompt(repoToDelete, alertDialog, viewModel) }
-                                {
+                        },
+                    )
+                }
+            }
+        }
+    }
+
+    LaunchedEffect(repoUrl) {
+        repoUrl?.let { viewModel.addRepo(repoUrl) }
+    }
+    
+    LaunchedEffect(Unit) {
+        viewModel.event.collectLatest { event ->
+            if (event is ExtensionRepoEvent.LocalizedMessage)
+                context.toast(event.stringRes)
+            if (event is ExtensionRepoEvent.Success)
+                inputText = ""
+        }
+    }
+}
+
+@Composable
+fun ExtensionRepoDeletePrompt(repoToDelete: String, alertDialog: ComposableAlertDialog, viewModel: ExtensionRepoViewModel) {
     AlertDialog(
         containerColor = MaterialTheme.colorScheme.surface,
         title = {
@@ -138,24 +161,4 @@ fun ExtensionRepoScreen(
             }
         },
     )
-                                }
-                        },
-                    )
-                }
-            }
-        }
-    }
-
-    LaunchedEffect(repoUrl) {
-        repoUrl?.let { viewModel.addRepo(repoUrl) }
-    }
-    
-    LaunchedEffect(Unit) {
-        viewModel.event.collectLatest { event ->
-            if (event is ExtensionRepoEvent.LocalizedMessage)
-                context.toast(event.stringRes)
-            if (event is ExtensionRepoEvent.Success)
-                inputText = ""
-        }
-    }
 }

app/src/main/java/eu/kanade/tachiyomi/util/compose/Locals.kt

@@ -1,9 +1,14 @@
 package eu.kanade.tachiyomi.util.compose
 
+import androidx.compose.runtime.Composable
 import androidx.compose.runtime.ProvidableCompositionLocal
 import androidx.compose.runtime.compositionLocalOf
 import androidx.compose.runtime.staticCompositionLocalOf
 import dev.yokai.domain.ComposableAlertDialog
 
+val <T> ProvidableCompositionLocal<T?>.currentOrThrow
+    @Composable
+    get(): T = this.current ?: throw RuntimeException("CompositionLocal is null")
+
 val LocalBackPress: ProvidableCompositionLocal<(() -> Unit)?> = staticCompositionLocalOf { null }
 val LocalAlertDialog: ProvidableCompositionLocal<ComposableAlertDialog?> = compositionLocalOf { null }