Eancom::Definition.create(name: 'D01B', type: :pricat) do |document, config|
  document.header do |header|
    header.segment Eancom::Edifact::UNA.new

    header.segment Eancom::Edifact::UNB.new(
      syntax_identifier: 'UNOC',
      syntax_version_number: '3',
      interchange_sender_identification: config.header.interchange_sender_identification,
      sender_identifiction_code_qualifier: '14',
      interchange_recipient_identification: config.header.interchange_recipient_identification,
      recipient_identifiction_code_qualifier: '14',
      date: config.header.date,
      time: config.header.time,
      interchange_control_reference: config.header.interchange_control_reference,
      application_reference: config.header.application_reference,
      interchange_agreement_identifier: config.header.interchange_agreement_identifier,
      test_indicator: nil
    )

    header.segment Eancom::Edifact::UNH.new(
      message_reference_number: config.header.message_reference_number,
      message_type: config.header.message_type,
      message_version_number: config.header.message_version_number,
      message_release_number: config.header.message_release_number,
      controlling_agency: config.header.controlling_agency,
      association_assigned_code: config.header.association_assigned_code
    )
  end

  document.body do |body|
    messages = config.body.messages 
    
    messages.each do |message|
      body.segment Eancom::Edifact::BGM.new(
        document_name_code: message.document_name_code,
        document_identifier: message.document_identifier,
        message_function_code: message.message_function_code
      )

      date_time_messages = message.date_time_messages
      date_time_messages.each_with_index do |dtm, index|
        structure = Eancom.find_structure(tag: 'DTM')
        key = dtm.identifier.intern.to_sym
        body.segment Eancom::Edifact::DTM.new(
          type: structure.dictionary_lookup(:type, dtm.identifier),
          date_time: dtm[key].date_time,
          format: '102'
        )
      end

      nads = message.nads
      nads.each_with_index do |nad, index|
        next unless index == 0
        structure = Eancom.find_structure(tag: 'NAD')
        key = nad.identifier.intern.to_sym
        body.segment Eancom::Edifact::NAD.new(
          party_function_code_qualifier: structure.dictionary_lookup(:party_function_code_qualifier, nad.identifier),
          party_identifier: nad[key].party_identifier,
          code_list_responsible_agency_code_1: '9'
        )
      end
      
      nads.each_with_index do |nad, index|
        next unless index == 1
        structure = Eancom.find_structure(tag: 'NAD')
        key = nad.identifier.intern.to_sym
        body.segment Eancom::Edifact::NAD.new(
          party_function_code_qualifier: structure.dictionary_lookup(:party_function_code_qualifier, nad.identifier),
          party_identifier: nad[key].party_identifier,
          code_list_responsible_agency_code_1: '9'
        )
      end

      structure = Eancom.find_structure(tag: 'CUX')
      body.segment Eancom::Edifact::CUX.new(
        currency_usage_code_qualifier_1: structure.dictionary_lookup(:currency_usage_code_qualifier_1, message.currency.currency_usage_code_qualifier_1),
        currency_identification_code_1: message.currency.currency_identification_code_1,
        currency_type_code_qualifier_1: structure.dictionary_lookup(:currency_type_code_qualifier_1,
                                                                    message.currency.currency_type_code_qualifier_1)
      )

      structure = Eancom.find_structure(tag: 'PGI')
      body.segment Eancom::Edifact::PGI.new(
        product_group_type_code: structure.dictionary_lookup(:product_group_type_code, message.product_group_identifier)
      )

      total_quantity = 0

      items = message.items
      items.each_with_index do |item, index|
        total_quantity += item.quantity.to_i

        structure = Eancom.find_structure(tag: 'LIN')
        body.segment Eancom::Edifact::LIN.new(
          line_item_identifier_1: (index + 1).to_s,
          item_identifier: item.ean,
          action_request_notification_description_code: item.action_request_notification_description_code,
          item_type_identification_code: structure.dictionary_lookup(:item_type_identification_code, item.item_type_identification_code),
          sub_line_indicator_code: item.sub_line_indicator_code,
          line_item_identifier_2: item.line_item_identifier_2
        )

        if additional_product_id = item.additional_product_id
          additional_product_id.each do |id|
            next unless additional_information = id.additional_information

            structure = Eancom.find_structure(tag: 'PIA')
            body.segment Eancom::Edifact::PIA.new(
              product_identifier_code_qualifier: additional_information.product_identifier_code_qualifier,
              item_identifier_1: additional_information.item_identifier_1,
              item_type_identification_code_1: structure.dictionary_lookup(:item_type_identification_code_1,
                                                                           additional_information.item_type_identification_code_1),
              code_list_responsible_agency_code_1: additional_information.code_list_identification_code_1
            )
          end
        end

        item_descriptions = item.item_descriptions
        item_descriptions.each do |description|
          structure = Eancom.find_structure(tag: 'IMD')
          if description.item_description_code != ''
            item_description_code = structure.dictionary_lookup(:item_description_code,
                                                                description.item_description_code)
          else
            item_description_code = ''
          end
          body.segment Eancom::Edifact::IMD.new(
            description_format_code: structure.dictionary_lookup(:description_format_code,
                                                                 description.description_format_code),
            item_characteristic_code: structure.dictionary_lookup(:item_characteristic_code,
                                                                  description.item_characteristic_code),
            item_description_code: item_description_code,
            code_list_identification_code_1: description.code_list_identification_code_1,
            code_list_responsible_agency_code_1: structure.dictionary_lookup(:code_list_responsible_agency_code_1,
                                                                             description.code_list_responsible_agency_code_1),
            item_description_1: description.item_description_1,
            language_name_code: description.language_name_code
          )
        end

        date_time_messages = item.date_time_messages
        date_time_messages.each_with_index do |dtm, index|
          next unless index == 0
          structure = Eancom.find_structure(tag: 'DTM')
          key = dtm.identifier.intern.to_sym
          body.segment Eancom::Edifact::DTM.new(
            type: structure.dictionary_lookup(:type, dtm.identifier),
            date_time: dtm[key].date_time,
            format: '102'
          )
        end

        date_time_messages.each_with_index do |dtm, index|
          next unless index == 1
          structure = Eancom.find_structure(tag: 'DTM')
          key = dtm.identifier.intern.to_sym
          body.segment Eancom::Edifact::DTM.new(
            type: structure.dictionary_lookup(:type, dtm.identifier),
            date_time: dtm[key].date_time,
            format: '102'
          )
        end

        if taxes = item.taxes
          taxes.each do |tax|
            next unless tax = tax.tax

            structure = Eancom.find_structure(tag: 'TAX')
            body.segment Eancom::Edifact::TAX.new(
              duty_or_tax_or_fee_function_code_qualifier: '7',
              duty_or_tax_or_fee_name_code: structure.dictionary_lookup(:duty_or_tax_or_fee_name_code,
                                                                        tax.duty_or_tax_or_fee_name_code),
              duty_or_tax_or_fee_account_code: tax.duty_or_tax_or_fee_account_code,
              duty_tax_fee_rate_code: tax.duty_tax_fee_rate_code,
              code_list_identification_code_3: tax.code_list_identification_code_3,
              code_list_responsibility_agency_code_2: tax.code_list_responsibility_agency_code_2,
              code_list_responsibility_agency_code_3: tax.code_list_responsibility_agency_code_3,
              code_list_responsibility_agency_code_4: '21',
              duty_or_tax_or_fee_category_code: structure.dictionary_lookup(:duty_or_tax_or_fee_category_code,
                                                                            tax.duty_or_tax_or_fee_category_code)
            )
          end
        end

        prices = item.prices
        prices.each do |price|
          next unless price = price.price

          structure = Eancom.find_structure(tag: 'PRI')
          body.segment Eancom::Edifact::PRI.new(
            price_code_qualifier: structure.dictionary_lookup(:price_code_qualifier, price.price_code_qualifier),
            price_amount: price.price_amount,
            price_type_code: structure.dictionary_lookup(:price_type_code, price.price_type_code),
            price_specification_code: structure.dictionary_lookup(:price_specification_code,
                                                                  price.price_specification_code),
            unit_price_basis_value: price.unit_price_basis_value,
            measurment_unit_code: price.measurment_unit_code
          )
        end

        qty = item.quantities
        qty.each_with_index do |qty, index|
          structure = Eancom.find_structure(tag: 'QTY')
          key = qty.identifier.intern.to_sym
          body.segment Eancom::Edifact::QTY.new(
            quantity_type_code_qualifier: structure.dictionary_lookup(:quantity_type_code_qualifier, qty.identifier),
            quantity: qty[key].quantity
          )
        end
      end
    end
  end

  document.footer do |footer|

    footer.segment Eancom::Edifact::UNT.new(
      number_of_segments_in_message: (document.total_segments + 2).to_s,
      message_reference_number: config.header.message_reference_number
    )
    footer.segment Eancom::Edifact::UNZ.new(
      interchange_control_count: '1',
      interchange_control_reference: config.header.interchange_control_reference
    )
  end
end