NameAllocator

Assigns Kotlin identifier names to avoid collisions, keywords, and invalid characters. To use, first create an instance and allocate all of the names that you need. Typically this is a mix of user-supplied names and constants:

val nameAllocator = NameAllocator()
for (property in properties) {
  nameAllocator.newName(property.name, property)
}
nameAllocator.newName("sb", "string builder")

Pass a unique tag object to each allocation. The tag scopes the name, and can be used to look up the allocated name later. Typically the tag is the object that is being named. In the above example we use property for the user-supplied property names, and "string builder" for our constant string builder.

Once we've allocated names we can use them when generating code:

val builder = FunSpec.builder("toString")
    .addModifiers(KModifier.OVERRIDE)
    .returns(String::class)

builder.addStatement("val %N = %T()",
    nameAllocator.get("string builder"), StringBuilder::class)

for (property in properties) {
  builder.addStatement("%N.append(%N)",
      nameAllocator.get("string builder"), nameAllocator.get(property))
}
builder.addStatement("return %N.toString()", nameAllocator.get("string builder"))
return builder.build()

The above code generates unique names if presented with conflicts. Given user-supplied properties with names ab and sb this generates the following:

override fun toString(): kotlin.String {
  val sb_ = java.lang.StringBuilder()
  sb_.append(ab)
  sb_.append(sb)
  return sb_.toString()
}

The underscore is appended to sb to avoid conflicting with the user-supplied sb property. Underscores are also prefixed for names that start with a digit, and used to replace name-unsafe characters like space or dash.

When dealing with multiple independent inner scopes, use a copy of the NameAllocator used for the outer scope to further refine name allocation for a specific inner scope.