Fixes #636

raamcosta · May 16, 2024 · 91f0732 · 91f0732
1 parent a18580e
commit 91f0732
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/...rc/main/java/com/ramcosta/composedestinations/codegen/writers/sub/SingleNavGraphWriter.kt b/...rc/main/java/com/ramcosta/composedestinations/codegen/writers/sub/SingleNavGraphWriter.kt
@@ -83,7 +83,19 @@ internal class SingleNavGraphWriter(
             .replace(NAV_GRAPH_DEEP_LINKS_PLACEHOLDER, navGraph.deepLinksCode())
             .replace(INNER_IMPORTED_ROUTES, navGraph.innerExternalRoutes())
             .replace(NAV_GRAPH_GEN_NAV_ARGS, navGraph.generatedNavArgsClass())
-            .replace(NAV_GRAPH_VISIBILITY_PLACEHOLDER, navGraph.visibility.name.lowercase())
+            .replace(
+                NAV_GRAPH_VISIBILITY_PLACEHOLDER,
+                navGraph.visibility.let {
+                    when (it) {
+                        Visibility.PUBLIC -> """
+                            @${importableHelper.addAndGetPlaceholder(Importable("Keep", "androidx.annotation.Keep"))}
+                            ${it.name.lowercase()}
+                        """.trimIndent()
+                        Visibility.INTERNAL,
+                        Visibility.PRIVATE -> it.name.lowercase()
+                    }
+                }
+            )
             .replace(
                 NAV_GRAPH_TYPE,
                 navGraph.graphSuperType()