✨ Add formatter style_tags config support

bitlogger/config.mbt

@@ -18,6 +18,7 @@ pub struct TextFormatterConfig {
   field_separator : String
   template : String
   color_mode : ColorMode
+  style_tags : Map[String, TextStyle]
 }
 
 pub fn TextFormatterConfig::new(
@@ -29,6 +30,7 @@ pub fn TextFormatterConfig::new(
   field_separator~ : String = " ",
   template~ : String = "",
   color_mode~ : ColorMode = ColorMode::Never,
+  style_tags~ : Map[String, TextStyle] = {},
 ) -> TextFormatterConfig {
   {
     show_timestamp,
@@ -39,9 +41,18 @@ pub fn TextFormatterConfig::new(
     field_separator,
     template,
     color_mode,
+    style_tags,
   }
 }
 
+fn style_tag_registry_from_config(style_tags : Map[String, TextStyle]) -> StyleTagRegistry {
+  let registry = style_tag_registry()
+  for name, style in style_tags {
+    ignore(registry.set_tag(name, style=style))
+  }
+  registry
+}
+
 pub fn TextFormatterConfig::to_formatter(self : TextFormatterConfig) -> TextFormatter {
   text_formatter(
     show_timestamp=self.show_timestamp,
@@ -52,6 +63,11 @@ pub fn TextFormatterConfig::to_formatter(self : TextFormatterConfig) -> TextForm
     field_separator=self.field_separator,
     template=self.template,
     color_mode=self.color_mode,
+    style_tags=if self.style_tags.length() == 0 {
+      None
+    } else {
+      Some(style_tag_registry_from_config(self.style_tags))
+    },
   )
 }
 
@@ -759,6 +775,19 @@ fn get_string(
   }
 }
 
+fn get_optional_string(
+  obj : Map[String, @json_parser.JsonValue],
+  key : String,
+) -> String? raise ConfigError {
+  match obj.get(key) {
+    None => None
+    Some(value) => match value.as_string() {
+      Some(text) => Some(text)
+      None => raise ConfigError::InvalidConfig("Expected string at key " + key)
+    }
+  }
+}
+
 fn get_bool(
   obj : Map[String, @json_parser.JsonValue],
   key : String,
@@ -848,9 +877,37 @@ fn parse_text_formatter_config(value : @json_parser.JsonValue) -> TextFormatterC
     field_separator=get_string(obj, "field_separator", default=" "),
     template=get_string(obj, "template", default=""),
     color_mode=parse_color_mode(get_string(obj, "color_mode", default="never")),
+    style_tags=match obj.get("style_tags") {
+      None => {}
+      Some(inner) => parse_style_tags_config(inner)
+    },
   )
 }
 
+fn parse_text_style_config(
+  value : @json_parser.JsonValue,
+  context : String,
+) -> TextStyle raise ConfigError {
+  let obj = expect_object(value, context)
+  text_style(
+    fg=get_optional_string(obj, "fg"),
+    bg=get_optional_string(obj, "bg"),
+    bold=get_bool(obj, "bold", default=false),
+    dim=get_bool(obj, "dim", default=false),
+    italic=get_bool(obj, "italic", default=false),
+    underline=get_bool(obj, "underline", default=false),
+  )
+}
+
+fn parse_style_tags_config(value : @json_parser.JsonValue) -> Map[String, TextStyle] raise ConfigError {
+  let obj = expect_object(value, "text_formatter.style_tags")
+  let style_tags : Map[String, TextStyle] = {}
+  for name, item in obj {
+    style_tags[name] = parse_text_style_config(item, "text_formatter.style_tags." + name)
+  }
+  style_tags
+}
+
 fn parse_queue_config(value : @json_parser.JsonValue) -> QueueConfig raise ConfigError {
   let obj = expect_object(value, "queue")
   QueueConfig::new(
@@ -934,7 +991,7 @@ pub fn stringify_queue_config(queue : QueueConfig, pretty~ : Bool = false) -> St
 }
 
 pub fn text_formatter_config_to_json(config : TextFormatterConfig) -> @json_parser.JsonValue {
-  @json_parser.JsonValue::Object({
+  let obj : Map[String, @json_parser.JsonValue] = {
     "show_timestamp": @json_parser.JsonValue::Bool(config.show_timestamp),
     "show_level": @json_parser.JsonValue::Bool(config.show_level),
     "show_target": @json_parser.JsonValue::Bool(config.show_target),
@@ -943,7 +1000,37 @@ pub fn text_formatter_config_to_json(config : TextFormatterConfig) -> @json_pars
     "field_separator": @json_parser.JsonValue::String(config.field_separator),
     "template": @json_parser.JsonValue::String(config.template),
     "color_mode": @json_parser.JsonValue::String(color_mode_label(config.color_mode)),
-  })
+  }
+  if config.style_tags.length() != 0 {
+    obj["style_tags"] = style_tags_config_to_json(config.style_tags)
+  }
+  @json_parser.JsonValue::Object(obj)
+}
+
+fn text_style_config_to_json(style : TextStyle) -> @json_parser.JsonValue {
+  let obj : Map[String, @json_parser.JsonValue] = {
+    "bold": @json_parser.JsonValue::Bool(style.bold),
+    "dim": @json_parser.JsonValue::Bool(style.dim),
+    "italic": @json_parser.JsonValue::Bool(style.italic),
+    "underline": @json_parser.JsonValue::Bool(style.underline),
+  }
+  match style.fg {
+    Some(value) => obj["fg"] = @json_parser.JsonValue::String(value)
+    None => ()
+  }
+  match style.bg {
+    Some(value) => obj["bg"] = @json_parser.JsonValue::String(value)
+    None => ()
+  }
+  @json_parser.JsonValue::Object(obj)
+}
+
+fn style_tags_config_to_json(style_tags : Map[String, TextStyle]) -> @json_parser.JsonValue {
+  let obj : Map[String, @json_parser.JsonValue] = {}
+  for name, style in style_tags {
+    obj[name] = text_style_config_to_json(style)
+  }
+  @json_parser.JsonValue::Object(obj)
 }
 
 pub fn stringify_text_formatter_config(