luau-unzip/.lune/docsgen/log.luau

local stdio = require("@lune/stdio")

local base = stdio.style("bold")

local STYLE_INFO = base .. `{stdio.color("green")}info{stdio.color("reset")}:`
local STYLE_WARN = base .. `{stdio.color("yellow")}warn{stdio.color("reset")}:`
local STYLE_ERROR = base .. `{stdio.color("red")}error{stdio.color("reset")}:`

export type LogType = "info" | "warn" | "error"
local styleMappings: { [LogType]: string } = {
  info = STYLE_INFO,
  warn = STYLE_WARN,
  error = STYLE_ERROR,
}

return {
  styles = styleMappings,
  log = function<T...>(type: LogType, ...: T...): ()
    local writer: (string) -> () = if type == "info" then stdio.write else stdio.ewrite
    local fmtMsg = stdio.format(styleMappings[type], ...) 

    return writer(fmtMsg .. "\n")
  end
}