import tables
import strutils
import math

proc colorEsc(color: string): string {.gcsafe.} =
  # Stored in proc to be gcsafe
  let colors = {
    "black": 30,
    "red": 31,
    "green": 32,
    "orange": 33,
    "blue": 34,
    "purple": 35,
    "cyan": 36,
    "reset": 0
  }.toTable()

  return "\e[" & intToStr(colors[color]) & "m"

proc colorize(text: string, color: string): string {.gcsafe.} =
  return colorEsc(color) & text & colorEsc("reset")

proc strCenter(text: string, filler: string = "-", length: int = 8): string {.gcsafe.} =
  # Simply truncate and return if it fills the entire length
  if text.len() > length:
    return text[0..length-1]

  # Find remaining characters and how much to pad each side
  var remainHalf: float = (length - text.len()) / 2
  var remainLeft: int = remainHalf.floor().int
  var remainRight: int = remainHalf.ceil().int

  return filler.repeat(remainLeft) & text & filler.repeat(remainRight)

proc logFormat(text: string, level: string = "undef"): string {.gcsafe.} =
  # Stored in proc to be gcsafe
  let colorMapping = {
    "Main": "green",
    "Error": "red",
    "Info": "purple",
    "Warning": "orange",
    "Web": "blue",
    "undef": "cyan"
  }.toTable()

  var prefix = "[" & colorize(text=strCenter(level), color=colorMapping[level]) & "]"
  return prefix & " " & text

proc logPrint(text: string = "Undefined error message", level: string = "undef") {.gcsafe.} =
  # In separate function in case I need to for example also write it to a file in the future.
  echo logFormat(text=text, level=level)

export logPrint