Resolve enum constants when emitting types

kotlinpoet/src/main/java/com/squareup/kotlinpoet/CodeWriter.kt

@@ -403,7 +403,7 @@ internal class CodeWriter constructor(
     while (c != null) {
       val alias = memberImports[c.canonicalName]?.alias
       val simpleName = alias ?: c.simpleName
-      val resolved = resolve(simpleName)
+      val resolved = resolveClass(simpleName) ?: resolveEnumConstant(simpleName)
       nameResolved = resolved != null
 
       // We don't care about nullability and type annotations here, as it's irrelevant for imports.
@@ -509,7 +509,7 @@ internal class CodeWriter constructor(
    * imports.
    */
   // TODO(jwilson): also honor superclass members when resolving names.
-  private fun resolve(simpleName: String): ClassName? {
+  private fun resolveClass(simpleName: String): ClassName? {
     // Match a child of the current (potentially nested) class.
     for (i in typeSpecStack.indices.reversed()) {
       val typeSpec = typeSpecStack[i]
@@ -531,6 +531,20 @@ internal class CodeWriter constructor(
     return null
   }
 
+  /**
+   * Returns the qualified name of the `simpleName` enum constant if it's in the current scope
+   */
+  private fun resolveEnumConstant(simpleName: String): String? {
+    if (typeSpecStack.size > 0) {
+      val typeSpec = typeSpecStack[0]
+      if (typeSpec.isEnum && typeSpec.enumConstants.keys.contains(simpleName)) {
+        return "$packageName.${typeSpec.name}.$simpleName"
+      }
+    }
+
+    return null
+  }
+
   /** Returns the class named `simpleName` when nested in the class at `stackDepth`.  */
   private fun stackClassName(stackDepth: Int, simpleName: String): ClassName {
     var className = ClassName(packageName, typeSpecStack[0].name!!)

kotlinpoet/src/test/java/com/squareup/kotlinpoet/TypeSpecTest.kt

@@ -5056,6 +5056,36 @@ class TypeSpecTest {
     )
   }
 
+  // https://github.com/square/kotlinpoet/issues/1234
+  @Test fun `enum constants are resolved`() {
+    val file = FileSpec.builder("com.example", "test")
+      .addType(
+        TypeSpec.enumBuilder("Foo")
+          .addProperty(
+            PropertySpec.builder("rawValue", String::class)
+              .initializer("%S", "")
+              .build()
+          )
+          .addEnumConstant("String")
+          .build()
+      )
+      .build()
+
+    assertThat(file.toString()).isEqualTo(
+      """
+    package com.example
+
+    public enum class Foo {
+      String,
+      ;
+
+      public val rawValue: kotlin.String = ""
+    }
+
+      """.trimIndent()
+    )
+  }
+
   companion object {
     private const val donutsPackage = "com.squareup.donuts"
   }