module RubyLLM::Providers::Bedrock::Media

# File lib/ruby_llm/providers/bedrock/media.rb, line 88
def document_format(attachment)
  attachment.extension || attachment.format
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 19
def empty_content?(content)
  content.nil? || (content.respond_to?(:empty?) && content.empty?)
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 37
def render_attachment(attachment, used_document_names:)
  case attachment.type
  when :image
    render_image_attachment(attachment)
  when :pdf, :document
    render_document_attachment(attachment, used_document_names:)
  when :text
    render_text_attachment(attachment)
  else
    raise UnsupportedAttachmentError, attachment.mime_type
  end
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 10
def render_content(content, used_document_names: nil)
  return [] if empty_content?(content)
  return render_raw_content(content) if content.is_a?(RubyLLM::Content::Raw)
  return [{ text: content.to_json }] if content.is_a?(Hash) || content.is_a?(Array)
  return [{ text: content }] unless content.is_a?(RubyLLM::Content)

  render_content_object(content, used_document_names || {})
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 23
def render_content_object(content, used_document_names)
  blocks = []
  blocks << { text: content.text } if content.text
  content.attachments.each do |attachment|
    blocks << render_attachment(attachment, used_document_names:)
  end
  blocks
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 67
def render_document_attachment(attachment, used_document_names:)
  format = document_format(attachment)

  raise UnsupportedAttachmentError, attachment.mime_type unless supported_document_format?(attachment)

  document_name = unique_document_name(sanitize_document_name(attachment.filename), used_document_names)
  {
    document: {
      format: format,
      name: document_name,
      source: {
        bytes: attachment.encoded
      }
    }
  }
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 52
def render_image_attachment(attachment)
  {
    image: {
      format: attachment.format,
      source: {
        bytes: attachment.encoded
      }
    }
  }
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 32
def render_raw_content(content)
  value = content.value
  value.is_a?(Array) ? value : [value]
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 63
def render_text_attachment(attachment)
  { text: attachment.for_llm }
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 92
def sanitize_document_name(filename)
  base = File.basename(filename.to_s, '.*')
  safe = base.gsub(/[^a-zA-Z0-9_-]/, '_')
  safe.empty? ? 'document' : safe
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 84
def supported_document_format?(attachment)
  SUPPORTED_DOCUMENT_FORMATS.include?(document_format(attachment))
end

# File lib/ruby_llm/providers/bedrock/media.rb, line 98
def unique_document_name(base_name, used_names)
  count = used_names[base_name].to_i
  used_names[base_name] = count + 1
  return base_name if count.zero?

  "#{base_name}_#{count + 1}"
end