From 3335c79591c7fa229406f8b5e97268227324e676 Mon Sep 17 00:00:00 2001
From: Harsh Shandilya <me@msfjarvis.dev>
Date: Sun, 18 Sep 2022 22:15:58 +0530
Subject: [PATCH] chore(common): add imports for `NetworkState` subclasses

---
 .../claw/common/comments/Comments.kt          | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/common/src/main/kotlin/dev/msfjarvis/claw/common/comments/Comments.kt b/common/src/main/kotlin/dev/msfjarvis/claw/common/comments/Comments.kt
index 2f2751e6..e655d1f4 100644
--- a/common/src/main/kotlin/dev/msfjarvis/claw/common/comments/Comments.kt
+++ b/common/src/main/kotlin/dev/msfjarvis/claw/common/comments/Comments.kt
@@ -18,6 +18,9 @@ import androidx.compose.ui.text.font.FontWeight
 import androidx.compose.ui.text.style.TextAlign
 import androidx.compose.ui.unit.dp
 import dev.msfjarvis.claw.common.NetworkState
+import dev.msfjarvis.claw.common.NetworkState.Error
+import dev.msfjarvis.claw.common.NetworkState.Loading
+import dev.msfjarvis.claw.common.NetworkState.Success
 import dev.msfjarvis.claw.common.posts.PostActions
 import dev.msfjarvis.claw.common.ui.NetworkError
 import dev.msfjarvis.claw.common.ui.ProgressBar
@@ -72,25 +75,25 @@ fun CommentsPage(
   modifier: Modifier = Modifier,
 ) {
   val postDetails by
-    produceState<NetworkState>(NetworkState.Loading) {
+    produceState<NetworkState>(Loading) {
       runCatching { getDetails(postId) }
         .fold(
-          onSuccess = { details -> value = NetworkState.Success(details) },
-          onFailure = { value = NetworkState.Error("Failed to load comments") }
+          onSuccess = { details -> value = Success(details) },
+          onFailure = { value = Error("Failed to load comments") }
         )
     }
 
   when (postDetails) {
-    is NetworkState.Success<*> -> {
+    is Success<*> -> {
       CommentsPageInternal(
-        (postDetails as NetworkState.Success<ExtendedPostDetails>).data,
+        (postDetails as Success<ExtendedPostDetails>).data,
         postActions,
         modifier.fillMaxSize(),
       )
     }
-    is NetworkState.Error -> {
-      NetworkError((postDetails as NetworkState.Error).message)
+    is Error -> {
+      NetworkError((postDetails as Error).message)
     }
-    NetworkState.Loading -> ProgressBar(modifier)
+    Loading -> ProgressBar(modifier)
   }
 }