org.http4k.routing.inspect.style.kt Maven / Gradle / Ivy

Go to download
package org.http4k.routing.inspect

import org.http4k.routing.inspect.ForegroundColour.Default
import java.util.Locale.getDefault

fun String.styled(style: TextStyle, escapeMode: EscapeMode = EscapeMode.Ansi) = escapeMode(this, style)

data class TextStyle(
    val foregroundColour: ForegroundColour = Default,
    val backgroundColour: BackgroundColour = BackgroundColour.Default,
    val variation: Variation = Variation.Default,
)

enum class ForegroundColour(val ansi: String) {
    Black("\u001B[30m"), Red("\u001B[31m"), Green("\u001B[32m"), Yellow("\u001B[33m"), Blue("\u001B[34m"), Purple("\u001B[35m"), Cyan("\u001B[36m"), White("\u001B[37m"), Default("")
}

enum class BackgroundColour(val ansi: String) {
    Red("\u001B[41m"), Green("\u001B[42m"), Yellow("\u001B[43m"), Blue("\u001B[44m"), Purple("\u001B[45m"), Cyan("\u001B[46m"), White("\u001B[47m"), Black("\u001B[40m"), Default("")
}

enum class Variation(val ansi: String) {
    Bold("\u001B[1m"), Italic("\u001B[3m"), BoldItalic("\u001B[3m"), Underline("\u001B[4m"), Strikethrough("\u001B[9m"), Default("")
}

enum class EscapeMode(private val converter: (String, TextStyle) -> String) {
    Ansi({ text, style -> "${style.foregroundColour.ansi}${style.backgroundColour.ansi}${style.variation.ansi}$text$reset" }),
    Pseudo({ text, style ->
        style.variation.pseudo() + style.foregroundColour.pseudo() + text + style.foregroundColour.pseudo() + style.variation.pseudo()
    }),
    None({ text, _ -> text });

    operator fun invoke(text: String, style: TextStyle) = converter(text, style)
}

internal const val reset = "\u001B[0m"

private fun ForegroundColour.pseudo() = if (this == Default) "" else "[${name.lowercase(getDefault())}]"

private fun Variation.pseudo(): String = when (this) {
    Variation.Bold -> "*"
    Variation.Italic -> "_"
    Variation.BoldItalic -> "**"
    Variation.Underline -> "_"
    Variation.Strikethrough -> "~"
    Variation.Default -> ""
}