class Lutaml::Qea::Validation::Formatters::JsonFormatter

def build_by_category
  result.messages.group_by(&:category).transform_values do |msgs|
    {
      count: msgs.size,
      messages: msgs.map(&:message),
    }
  end
end

def build_by_severity
  {
    errors: format_severity_group(result.errors),
    warnings: format_severity_group(result.warnings),
    info: format_severity_group(result.info),
  }
end

def build_data
  {
    summary: build_summary,
    messages: build_messages,
    by_category: build_by_category,
    by_severity: build_by_severity,
  }
end

def build_messages # rubocop:disable Metrics/MethodLength
  result.messages.map do |message|
    {
      severity: message.severity,
      category: message.category,
      entity_type: message.entity_type,
      entity_id: message.entity_id,
      entity_name: message.entity_name,
      message: message.message,
      context: message.context,
    }
  end
end

def build_summary
  {
    valid: result.valid?,
    total_messages: result.messages.size,
    error_count: result.errors.size,
    warning_count: result.warnings.size,
    info_count: result.info.size,
  }
end

def format
  data = build_data
  if options[:pretty]
    JSON.pretty_generate(data)
  else
    JSON.generate(data)
  end
end

def format_severity_group(messages) # rubocop:disable Metrics/MethodLength
  {
    count: messages.size,
    by_category: messages
      .group_by(&:category).transform_values do |msgs|
        msgs.map do |msg|
          {
            entity_type: msg.entity_type,
            entity_id: msg.entity_id,
            entity_name: msg.entity_name,
            message: msg.message,
          }
        end
      end,
  }
end

def initialize(result: nil, **options)
  @result = result
  @options = {
    pretty: false,
  }.merge(options)
end