desc 'unidom:geo:china:import Rake Task imports China regions from the given CSV file.'

namespace :unidom do
  namespace :geo do
    namespace :china do

      namespace :region do

        # bundle exec rake unidom:geo:china:region:import
        #   file=/data.csv
        #   from_date=2020-01-01
        #   scheme_id=
        #   scheme_type=
        task import: :environment do

          include ::Unidom::Common::DataHelper

          file_name   = ENV['file']
          scheme_id   = ENV['scheme_id']||::Unidom::Common::NULL_UUID
          scheme_type = ENV['scheme_type']||''
          opened_at   = parse_time ENV['from_date']

          updated_count = 0
          created_count = 0

          region_entities = ::Unidom::Geo::China::Region.scheme_id_is(scheme_id).scheme_type_is(scheme_type).select('id, name, virtual, numeric_code, alphabetic_code, scheme_id, scheme_type, opened_at, closed_at, defunct, updated_at').to_a

          each_csv_row file_name do |region|

            numeric_code    = region['numeric_code']
            alphabetic_code = region['alphabetic_code']

            attributes = { name: region['name'], virtual: region['virtual'], scheme_id: scheme_id, scheme_type: scheme_type, opened_at: opened_at }
            attributes[:alphabetic_code] = alphabetic_code if alphabetic_code.present?

            if region_entities.present?
              found_region_entities = region_entities.select { |region_entity| region_entity.numeric_code==numeric_code }
              if found_region_entities.present?
                found_region_entities.each do |found_region_entity|
                  found_region_entity.assign_attributes attributes
                  if found_region_entity.changed?
                    found_region_entity.save!
                    updated_count += 1
                  end
                end
              else
                attributes[:numeric_code] = numeric_code
                ::Unidom::Geo::China::Region.create! attributes
                created_count += 1
              end
            else
              attributes[:numeric_code] = numeric_code
              ::Unidom::Geo::China::Region.create! attributes
              created_count += 1
            end

          end

          puts "#{created_count} China Regions were created. #{updated_count} China Regions were updated per CSV."
          puts "#{created_count+updated_count} China Regions were handled totally."

        end

      end

      namespace :town do

        # bundle exec rake unidom:geo:china:town:import
        #   file=/data.csv
        #   from_date=2020-01-01
        task import: :environment do

          include ::Unidom::Common::DataHelper

          file_name   = ENV['file']
          opened_at   = parse_time ENV['from_date']

          updated_count = 0
          created_count = 0

          each_csv_row file_name do |region|

            numeric_code = region['numeric_code']

            town   = Unidom::Geo::China::Town.numeric_coded_as(numeric_code).first
            county = Unidom::Geo::China::Region.numeric_coded_as(numeric_code[0..5]).valid_at.alive.first
            if town.present?
              town.assign_attributes region: county, name: region['name'], virtual: region['virtual']
              town.save!
              updated_count += 1
            else
              town = Unidom::Geo::China::Town.create! region: county, numeric_code: numeric_code, name: region['name'], virtual: region['virtual'], opened_at: opened_at
              created_count += 1
            end

          end

          puts "#{created_count} China Regions were created. #{updated_count} China Regions were updated per CSV."
          puts "#{created_count+updated_count} China Regions were handled totally."

        end

      end

    end
  end
end