BitLogger/src/utils/runtime_file_state.mbt

///|
pub struct RuntimeFileState {
  file : FileSinkState
  queued : Bool
  pending_count : Int
  dropped_count : Int
}

///|
pub fn RuntimeFileState::new(
  file : FileSinkState,
  queued? : Bool = false,
  pending_count? : Int = 0,
  dropped_count? : Int = 0,
) -> RuntimeFileState {
  { file, queued, pending_count, dropped_count }
}

///|
fn file_sink_policy_to_json_value(
  policy : FileSinkPolicy,
) -> @json_parser.JsonValue {
  let obj : Map[String, @json_parser.JsonValue] = {
    "append": @json_parser.JsonValue::Bool(policy.append),
    "auto_flush": @json_parser.JsonValue::Bool(policy.auto_flush),
  }
  match policy.rotation {
    None => obj["rotation"] = @json_parser.JsonValue::Null
    Some(rotation) => obj["rotation"] = file_rotation_config_to_json(rotation)
  }
  @json_parser.JsonValue::Object(obj)
}

///|
pub fn file_sink_policy_to_json(
  policy : FileSinkPolicy,
) -> @json_parser.JsonValue {
  file_sink_policy_to_json_value(policy)
}

///|
pub fn stringify_file_sink_policy(
  policy : FileSinkPolicy,
  pretty? : Bool = false,
) -> String {
  let value = file_sink_policy_to_json_value(policy)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}

///|
fn file_sink_state_to_json_value(
  state : FileSinkState,
) -> @json_parser.JsonValue {
  let obj : Map[String, @json_parser.JsonValue] = {
    "path": @json_parser.JsonValue::String(state.path),
    "available": @json_parser.JsonValue::Bool(state.available),
    "append": @json_parser.JsonValue::Bool(state.append),
    "auto_flush": @json_parser.JsonValue::Bool(state.auto_flush),
    "open_failures": @json_parser.JsonValue::Number(
      state.open_failures.to_double(),
    ),
    "write_failures": @json_parser.JsonValue::Number(
      state.write_failures.to_double(),
    ),
    "flush_failures": @json_parser.JsonValue::Number(
      state.flush_failures.to_double(),
    ),
    "rotation_failures": @json_parser.JsonValue::Number(
      state.rotation_failures.to_double(),
    ),
  }
  match state.rotation {
    None => obj["rotation"] = @json_parser.JsonValue::Null
    Some(rotation) => obj["rotation"] = file_rotation_config_to_json(rotation)
  }
  @json_parser.JsonValue::Object(obj)
}

///|
pub fn file_sink_state_to_json(state : FileSinkState) -> @json_parser.JsonValue {
  file_sink_state_to_json_value(state)
}

///|
pub fn stringify_file_sink_state(
  state : FileSinkState,
  pretty? : Bool = false,
) -> String {
  let value = file_sink_state_to_json_value(state)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}

///|
pub fn runtime_file_state_to_json(
  state : RuntimeFileState,
) -> @json_parser.JsonValue {
  @json_parser.JsonValue::Object({
    "file": file_sink_state_to_json_value(state.file),
    "queued": @json_parser.JsonValue::Bool(state.queued),
    "pending_count": @json_parser.JsonValue::Number(
      state.pending_count.to_double(),
    ),
    "dropped_count": @json_parser.JsonValue::Number(
      state.dropped_count.to_double(),
    ),
  })
}

///|
pub fn stringify_runtime_file_state(
  state : RuntimeFileState,
  pretty? : Bool = false,
) -> String {
  let value = runtime_file_state_to_json(state)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}