lib/mindee/input/sources/base64_input_source.rb


# frozen_string_literal: true

require 'stringio'

module Mindee
  module Input
    module Source
      # Load a document from a base64 string.
      class Base64InputSource < LocalInputSource
        # @param base64_string [String]
        # @param filename [String]
        # @param fix_pdf [Boolean]
        def initialize(base64_string, filename, fix_pdf: false)
          io_stream = StringIO.new(base64_string.unpack1('m*'))
          io_stream.set_encoding Encoding::BINARY
          super(io_stream, filename, fix_pdf: fix_pdf)
        end

        # Overload of the same function to prevent a base64 from being re-encoded.
        # @param close [Boolean]
        # @return [Array<String, [String, aBinaryString ], [Hash, nil] >]
        def read_document(close: true)
          @io_stream.seek(0)
          data = @io_stream.read
          @io_stream.close if close
          ['document', [data].pack('m'), { filename: Source.convert_to_unicode_escape(@filename) }]
        end
      end
    end
  end
end