CartoDB/cartodb20

require 'active_record'
require_relative '../../../services/table-geocoder/lib/exceptions'

module Carto
  class Geocoding < ActiveRecord::Base

    PUBLIC_ATTRIBUTES = [:id, :table_id, :table_name, :state, :kind, :country_code, :region_code, :formatter,
                        :geocoder_type, :geometry_type, :error, :processed_rows, :cache_hits, :processable_rows,
                        :real_rows, :price, :used_credits, :remaining_quota, :country_column, :region_column,
                        :data_import_id, :error_code]

    GEOCODING_BLOCK_SIZE = 1000.0

    def self.processable_rows(table_service, force_all_rows=false)
      dataset = table_service.owner.in_database.select.from(table_service.sequel_qualified_table_name)
      if !force_all_rows && dataset.columns.include?(:cartodb_georef_status)
        dataset = dataset.exclude(cartodb_georef_status: true)
      end
      dataset.count
    end

    belongs_to :user

    def public_values
      Hash[PUBLIC_ATTRIBUTES.map{ |k| [k, (self.send(k) rescue self[k].to_s)] }]
    end

    def error
      additional_info = Carto::GeocoderErrors.additional_info(error_code)
      if additional_info
        { title: additional_info.title, description: additional_info.what_about }
      else
        { title: 'Geocoding error', description: '' }
      end
    end

    def price
      return 0 unless used_credits&.positive?

      (user.geocoding_block_price * used_credits) / GEOCODING_BLOCK_SIZE
    end

    def remaining_quota
      user.remaining_geocoding_quota
    end

    def log
      Carto::Log.find(log_id)
    end

  end
end