io.getquill.codegen.util.StringUtil.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quill-codegen_2.12 Show documentation

quill-codegen

The newest version!

package io.getquill.codegen.util

object StringUtil {

  def indent(code: String): String = {
    val lines = code.split("\n")
    lines.tail.foldLeft(lines.head) { (out, line) =>
      out + '\n' +
        (if (line.isEmpty) line else "  " + line)
    }
  }

  implicit final class StringExtensions(private val str: String) extends AnyVal {
    def snakeToUpperCamel: String = str.split("_").map(_.toLowerCase).map(_.capitalize).mkString
    def snakeToLowerCamel: String = str.split("_").map(_.toLowerCase).map(_.capitalize).mkString.uncapitalize
    def lowerCamelToSnake: String = str.split("(?=[A-Z])").mkString("_").toLowerCase
    def uncapitalize: String =
      new String(
        (str.toList match {
          case head :: tail => head.toLower :: tail
          case Nil          => Nil
        }).toArray
      )
    def unquote: String          = str.replaceFirst("^\"", "").replaceFirst("\"$", "")
    def trimFront: String        = str.dropWhile(_ == '\n')
    def notEmpty: Option[String] = if (str.trim == "") None else Some(str)
    def inSetNocase(seq: String*): Boolean =
      seq.map(_.toLowerCase).toSeq.contains(str.toLowerCase)
  }

  implicit final class OptionStringExtensions(private val str: Option[String]) extends AnyVal {
    def existsInSetNocase(seq: String*): Boolean =
      str.map(_.toLowerCase).exists(value => seq.map(_.toLowerCase).toSeq.contains(value))
  }
}