From 6ba134b1be3239920d264e0bac7e86c1c4dd6016 Mon Sep 17 00:00:00 2001
From: Harry Billinis <31311538+tiltorito@users.noreply.github.com>
Date: Fri, 25 Jan 2019 17:54:02 +0300
Subject: [PATCH] Better names suggestion for constant like reference
 expressions (KT-28485)

Code was simplified after @igorwojda suggestions

 #KT-28485 Fixed
---
 .../kotlin/utils/capitalizeDecapitalize.kt    | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/core/util.runtime/src/org/jetbrains/kotlin/utils/capitalizeDecapitalize.kt b/core/util.runtime/src/org/jetbrains/kotlin/utils/capitalizeDecapitalize.kt
index 9f3a78da602..42da285e0f0 100644
--- a/core/util.runtime/src/org/jetbrains/kotlin/utils/capitalizeDecapitalize.kt
+++ b/core/util.runtime/src/org/jetbrains/kotlin/utils/capitalizeDecapitalize.kt
@@ -20,6 +20,8 @@ package org.jetbrains.kotlin.util.capitalizeDecapitalize
  * "FooBar" -> "fooBar"
  * "FOOBar" -> "fooBar"
  * "FOO" -> "foo"
+ * "FOO_BAR" -> "fooBar"
+ * "__F_BAR" -> "fBar"
  */
 fun String.decapitalizeSmart(asciiOnly: Boolean = false): String {
     fun isUpperCaseCharAt(index: Int): Boolean {
@@ -27,6 +29,36 @@ fun String.decapitalizeSmart(asciiOnly: Boolean = false): String {
         return if (asciiOnly) c in 'A'..'Z' else c.isUpperCase()
     }
 
+    /**
+     * FOOBAR -> null
+     * FOO_BAR -> "fooBar"
+     * FOO_BAR_BAZ -> "fooBarBaz"
+     * "__F_BAR" -> "fBar"
+     * "F_BAR" -> "fBar"
+     */
+    fun decapitalizeConst(str: String): String? {
+        val words = str.split("_").filter { it.isNotEmpty() }
+
+        if (words.size <= 1) return null
+
+        val builder = StringBuilder()
+
+        words.forEachIndexed { index, word ->
+            if (index == 0) {
+                builder.append(word.toLowerCase())
+            } else {
+                builder.append(word.first().toUpperCase())
+                builder.append(word.drop(1).toLowerCase())
+            }
+        }
+
+        return builder.toString()
+    }
+
+    val constant = decapitalizeConst(this)
+
+    if (constant != null) return constant
+
     if (isEmpty() || !isUpperCaseCharAt(0)) return this
 
     if (length == 1 || !isUpperCaseCharAt(1)) {