lib/mindee/product/receipt/receipt_v5_page.rb


# frozen_string_literal: true

require_relative '../../parsing'
require_relative 'receipt_v5_document'

module Mindee
  module Product
    module Receipt
      # Receipt API version 5.3 page data.
      class ReceiptV5Page < Mindee::Parsing::Common::Page
        # @param prediction [Hash]
        def initialize(prediction)
          super
          @prediction = if prediction['prediction'].empty?
                          nil
                        else
                          ReceiptV5PagePrediction.new(
                            prediction['prediction'],
                            prediction['id']
                          )
                        end
        end
      end

      # Receipt V5 page prediction.
      class ReceiptV5PagePrediction < ReceiptV5Document
        # @return [String]
        def to_s
          out_str = String.new
          out_str << "\n#{super}"
          out_str
        end
      end
    end
  end
end