Merge pull request #8 from trello/vr/internal-root-function

make sure root functions are internal so we don't have import errors
trello-archive · Mar 22, 2019 · 3fe008a · 3fe008a
2 parents 9874755 + b285644
commit 3fe008a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mr-clean-plugin/src/main/java/com/trello/mrclean/plugin/RootFunctionGenerator.kt b/mr-clean-plugin/src/main/java/com/trello/mrclean/plugin/RootFunctionGenerator.kt
@@ -2,10 +2,12 @@ package com.trello.mrclean.plugin
 
 import com.squareup.kotlinpoet.FileSpec
 import com.squareup.kotlinpoet.FunSpec
+import com.squareup.kotlinpoet.KModifier
 
 class RootFunctionGenerator {
   fun createRootFunction(packageName: String): FileSpec {
     val rootFunction = FunSpec.builder("sanitizedToString")
+        .addModifiers(KModifier.INTERNAL)
         .receiver(Any::class)
         .returns(String::class)
         .apply {

diff --git a/mr-clean-plugin/src/test/java/com/trello/mrclean/plugin/RootFunctionGeneratorTest.kt b/mr-clean-plugin/src/test/java/com/trello/mrclean/plugin/RootFunctionGeneratorTest.kt
@@ -19,7 +19,7 @@ class RootFunctionGeneratorTest {
       |import kotlin.Any
       |import kotlin.String
 
-      |fun Any.sanitizedToString(): String =
+      |internal fun Any.sanitizedToString(): String =
       |        error("No function generated! Make sure to annotate with @Sanitize")
       |
     """.trimMargin()