Merge branch 'master' into 499-admin-wildcard-search

app/controllers/admin/account_activities_controller.rb

@@ -35,7 +35,8 @@ module Admin
       respond_to do |format|
         format.html
         format.csv do
-          send_data @q.result.to_csv, filename: "account_activities_#{Time.zone.now.to_formatted_s(:number)}.csv"
+          raw_csv = CsvGenerator.generate_csv(@q.result)
+          send_data raw_csv, filename: "account_activities_#{Time.zone.now.to_formatted_s(:number)}.csv"
         end
       end
 

app/controllers/admin/base_controller.rb

@@ -26,7 +26,7 @@ module Admin
       respond_to do |format|
         format.html { render page }
         format.csv do
-          raw_csv = @q.result.to_csv
+          raw_csv = CsvGenerator.generate_csv(@q.result)
           send_data raw_csv,
                     filename: "#{filename}_#{Time.zone.now.to_formatted_s(:number)}.csv",
                     type: "#{Mime[:csv]}; charset=utf-8"

app/controllers/admin/contact_versions_controller.rb

@@ -1,13 +1,13 @@
 module Admin
   class ContactVersionsController < BaseController
-    include ObjectVersionsHelper
+    include ApplicationHelper
 
     load_and_authorize_resource class: Version::ContactVersion
 
     def index
       params[:q] ||= {}
 
-      search_params = PartialSearchFormatter.format(params[:q])
+      search_params = PartialSearchFormatter.format(fix_date_params)
       versions = Version::ContactVersion.includes(:item).order(created_at: :desc, id: :desc)
       @q = versions.ransack(polymorphic_association(search_params))
 
@@ -48,5 +48,14 @@ module Admin
 
       record_type
     end
+
+    def fix_date_params
+      params_copy = params[:q].deep_dup
+      if params_copy['created_at_lteq'].present?
+        params_copy['created_at_lteq'] = Date.parse(params_copy['created_at_lteq']) + 1.day
+      end
+
+      params_copy
+    end
   end
 end

app/controllers/admin/domain_versions_controller.rb

@@ -1,7 +1,5 @@
 module Admin
   class DomainVersionsController < BaseController
-    include ObjectVersionsHelper
-
     load_and_authorize_resource class: Version::DomainVersion
 
     def index
@@ -82,9 +80,8 @@ module Admin
 
     def fix_date_params
       params_copy = params[:q].deep_dup
-      if params_copy['created_at_lteq'].present?
-        params_copy['created_at_lteq'] = Date.parse(params_copy['created_at_lteq']) + 1.day
-      end
+      created_at = params_copy['created_at_lteq']
+      params_copy['created_at_lteq'] = Date.parse(created_at) + 1.day if created_at.present?
 
       params_copy
     end

app/controllers/api/v1/registrant/contacts_controller.rb

@@ -34,6 +34,17 @@ module Api
           end
         end
 
+        def do_need_update_contact
+          result = current_registrant_user.do_need_update_contact?
+          render json: { update_contacts: result[:result], counter: result[:counter] }
+        end
+
+        def update_company_contacts
+          companies = current_registrant_user.update_company_contacts
+
+          render json: { message: 'get it', companies: companies }
+        end
+
         def update
           logger.debug 'Received update request'
           logger.debug params

app/controllers/registrar/account_activities_controller.rb

@@ -20,7 +20,8 @@ class Registrar
       respond_to do |format|
         format.html { @account_activities = @q.result.page(params[:page]) }
         format.csv do
-          send_data @q.result.to_csv, filename: "account_activities_#{Time.zone.now.to_formatted_s(:number)}.csv"
+          raw_csv = CsvGenerator.generate_csv(@q.result)
+          send_data raw_csv, filename: "account_activities_#{Time.zone.now.to_formatted_s(:number)}.csv"
         end
       end
 

app/controllers/registrar/contacts_controller.rb

@@ -40,7 +40,7 @@ class Registrar
           @contacts = @contacts.per(contacts_per_page) if contacts_per_page.positive?
         end
         format.csv do
-          raw_csv = contacts.to_csv
+          raw_csv = CsvGenerator.generate_csv(contacts)
           send_data raw_csv, filename: 'contacts.csv', type: "#{Mime[:csv]}; charset=utf-8"
         end
         format.pdf do

app/helpers/application_helper.rb

@@ -10,27 +10,11 @@ module ApplicationHelper
   end
 
   def ident_for(contact)
-    if contact.is_a? Hash
-      ident_country_code = contact[:ident_country_code]
-      ident_type = contact[:ident_type]
-      ident = contact[:ident]
-    else
-      ident_country_code = contact.ident_country_code
-      ident_type = contact.ident_type
-      ident = contact.ident
-    end
+    ident = contact.ident
+    description = "[#{contact.ident_country_code} #{contact.ident_type}]"
+    description.prepend("#{ident} ") if ident.present?
 
-    case ident_type
-    when 'birthday'
-      "#{ident} [#{ident_country_code} #{ident_type}]"
-      else
-        if ident.present?
-          "#{ident} [#{ident_country_code} #{ident_type}]"
-        else
-          "[#{ident_country_code} #{ident_type}]"
-        end
-
-    end
+    description
   end
 
   def current_commit_link

app/helpers/object_versions_helper.rb

@@ -1,19 +0,0 @@
-module ObjectVersionsHelper
-  def attach_existing_fields(version, new_object)
-    version.object_changes.to_h.each do |key, value|
-      method_name = "#{key}=".to_sym
-      new_object.public_send(method_name, event_value(version, value)) if new_object.respond_to?(method_name)
-    end
-  end
-
-  def only_present_fields(version, model)
-    field_names = model.column_names
-    version.object.to_h.select { |key, _value| field_names.include?(key) }
-  end
-
-  private
-
-  def event_value(version, val)
-    version.event == 'destroy' ? val.first : val.last
-  end
-end

app/lib/to_csv.rb

@@ -1,10 +0,0 @@
-module ToCsv
-  def to_csv
-    CSV.generate do |csv|
-      csv << column_names
-      all.find_each do |item|
-        csv << item.attributes.values_at(*column_names)
-      end
-    end
-  end
-end

app/models/account.rb

@@ -1,5 +1,4 @@
 class Account < ApplicationRecord
-  extend ToCsv
   include Versions
 
   belongs_to :registrar, required: true
@@ -12,4 +11,12 @@ class Account < ApplicationRecord
   def activities
     account_activities
   end
+
+  def as_csv_row
+    [id, balance, currency, registrar]
+  end
+
+  def self.csv_header
+    ['Id', 'Balance', 'Currency', 'Registrar']
+  end
 end

app/models/account_activity.rb

@@ -11,6 +11,7 @@ class AccountActivity < ApplicationRecord
   UPDATE_CREDIT = 'update_credit'.freeze
 
   after_create :update_balance
+
   def update_balance
     account.balance += sum
     account.save
@@ -19,23 +20,17 @@ class AccountActivity < ApplicationRecord
     save
   end
 
+  def as_csv_row
+    [account.registrar.try(:code), description, I18n.t(activity_type), I18n.l(created_at), sum]
+  end
+
   class << self
     def types_for_select
       [CREATE, RENEW, ADD_CREDIT, UPDATE_CREDIT].map { |x| [I18n.t(x), x] }
     end
 
-    def to_csv
-      attributes = %w(description activity_type created_at sum)
-
-      CSV.generate(headers: true) do |csv|
-        csv << %w(registrar description activity_type receipt_date sum)
-
-        all.each do |x|
-          attrs  = [x.account.registrar.try(:code)]
-          attrs += attributes.map { |attr| x.send(attr) }
-          csv << attrs
-        end
-      end
+    def csv_header
+      ['Registrar', 'Description', 'Activity Type', 'Receipt Date', 'Sum']
     end
   end
 end

app/models/api_log/epp_log.rb

@@ -1,5 +1,3 @@
 module ApiLog
-  class EppLog < Db
-    extend ToCsv
-  end
+  class EppLog < Db; end
 end

app/models/api_log/repp_log.rb

@@ -1,5 +1,3 @@
 module ApiLog
-  class ReppLog < Db
-    extend ToCsv
-  end
+  class ReppLog < Db; end
 end

app/models/blocked_domain.rb

@@ -1,6 +1,5 @@
 class BlockedDomain < ApplicationRecord
   include Versions
-  extend ToCsv
   before_save   :generate_data
   after_destroy :remove_data
 

app/models/contact.rb

@@ -188,15 +188,6 @@ class Contact < ApplicationRecord
       ]
     end
 
-    def to_csv
-      CSV.generate do |csv|
-        csv << column_names
-        all.each do |contact|
-        csv << contact.attributes.values_at(*column_names)
-        end
-      end
-    end
-
     def pdf(html)
       kit = PDFKit.new(html)
       kit.to_pdf
@@ -569,4 +560,27 @@ class Contact < ApplicationRecord
   def deletable?
     !linked?
   end
+
+  def ident_human_description
+    description = "[#{ident_country_code} #{ident_type}]"
+    description.prepend("#{ident} ") if ident.present?
+
+    description
+  end
+
+  def as_csv_row
+    [
+      name,
+      code,
+      ident_human_description,
+      email,
+      created_at.to_formatted_s(:db),
+      registrar,
+      phone,
+    ]
+  end
+
+  def self.csv_header
+    ['Name', 'ID', 'Ident', 'E-mail', 'Created at', 'Registrar', 'Phone']
+  end
 end

app/models/dispute.rb

@@ -1,5 +1,4 @@
 class Dispute < ApplicationRecord
-  extend ToCsv
   include WhoisStatusPopulate
   validates :domain_name, :password, :starts_at, :expires_at, presence: true
   before_validation :fill_empty_passwords, :set_expiry_date

app/models/domain.rb

@@ -289,21 +289,6 @@ class Domain < ApplicationRecord
       )
     end
 
-    def to_csv
-      CSV.generate do |csv|
-        headers = column_names.dup
-        swap_elements(headers, [[0, 1], [1, 5]])
-        headers[0] = 'Domain'
-        headers[1] = headers[1].humanize
-        csv << headers
-        all.find_each do |item|
-          row = item.attributes.values_at(*column_names)
-          swap_elements(row, [[0, 1], [1, 5]])
-          csv << row
-        end
-      end
-    end
-
     private
 
     def registrant_user_domains_by_registrant(registrant_user)
@@ -753,6 +738,38 @@ class Domain < ApplicationRecord
     contacts.select(&:email_verification_failed?)&.map(&:email)&.uniq
   end
 
+  def as_csv_row
+    [
+      name,
+      registrant_name,
+      valid_to.to_formatted_s(:db),
+      registrar,
+      created_at.to_formatted_s(:db),
+      statuses,
+      contacts.pluck(:code),
+      force_delete_date,
+      force_delete_data,
+    ]
+  end
+
+  def registrant_name
+    return registrant.name if registrant
+
+    ver = Version::ContactVersion.where(item_id: registrant_id).last
+    contact = Contact.all_versions_for([registrant_id], created_at).first
+
+    contact = ObjectVersionsParser.new(ver).parse if contact.nil? && ver
+
+    contact.try(:name) || 'Deleted'
+  end
+
+  def self.csv_header
+    [
+      'Domain', 'Registrant', 'Valid to', 'Registrar', 'Created at',
+      'Statuses', 'Contacts code', 'Force delete date', 'Force delete data'
+    ]
+  end
+
   def self.pdf(html)
     kit = PDFKit.new(html)
     kit.to_pdf

app/models/invoice.rb

@@ -3,7 +3,6 @@ class Invoice < ApplicationRecord
   include Invoice::Cancellable
   include Invoice::Payable
   include Invoice::BookKeeping
-  extend ToCsv
 
   belongs_to :buyer, class_name: 'Registrar'
   has_one  :account_activity
@@ -117,6 +116,23 @@ class Invoice < ApplicationRecord
     e_invoice_sent_at.present?
   end
 
+  def as_csv_row
+    [
+      number,
+      buyer,
+      cancelled? ? I18n.t(:cancelled) : due_date,
+      receipt_date_status,
+      issue_date,
+      total,
+      currency,
+      seller_name,
+    ]
+  end
+
+  def self.csv_header
+    ['Number', 'Buyer', 'Due Date', 'Receipt Date', 'Issue Date', 'Total', 'Currency', 'Seller Name']
+  end
+
   def self.create_from_transaction!(transaction)
     registrar_user = Registrar.find_by(reference_no: transaction.parsed_ref_number)
     return unless registrar_user
@@ -128,6 +144,16 @@ class Invoice < ApplicationRecord
 
   private
 
+  def receipt_date_status
+    if paid?
+      receipt_date
+    elsif cancelled?
+      I18n.t(:cancelled)
+    else
+      I18n.t(:unpaid)
+    end
+  end
+
   def apply_default_buyer_vat_no
     self.buyer_vat_no = buyer.vat_no
   end

app/models/registrant_user.rb

@@ -20,16 +20,27 @@ class RegistrantUser < User
   def companies(company_register = CompanyRegister::Client.new)
     return [] if ident.include?('-')
 
-    companies = company_register.representation_rights(citizen_personal_code: ident,
-                                                       citizen_country_code: country.alpha3)
-
-    companies = update_contacts_before_receive(companies)
-    companies
+    company_register.representation_rights(citizen_personal_code: ident,
+                                           citizen_country_code: country.alpha3)
   rescue CompanyRegister::NotAvailableError
-    return []
+    []
   end
 
-  def update_contacts_before_receive(companies)
+  def do_need_update_contact?
+    return { result: false, counter: 0 } if companies.blank?
+
+    counter = 0
+    companies.each do |company|
+      counter += Contact.where(ident: company.registration_number, ident_country_code: 'EE')&.
+                  reject { |contact| contact.name == company.company_name }.size
+    end
+
+    return { result: true, counter: counter } if counter.positive?
+
+    { result: false, counter: 0 }
+  end
+
+  def update_company_contacts
     return [] if companies.blank?
 
     companies.each do |company|

app/models/reserved_domain.rb

@@ -1,5 +1,4 @@
 class ReservedDomain < ApplicationRecord
-  extend ToCsv
   include Versions # version/reserved_domain_version.rb
   include WhoisStatusPopulate
   before_save :fill_empty_passwords

app/models/version/contact_version.rb

@@ -1,7 +1,23 @@
 class Version::ContactVersion < PaperTrail::Version
-  extend ToCsv
   include VersionSession
 
   self.table_name    = :log_contacts
   self.sequence_name = :log_contacts_id_seq
+
+  def as_csv_row
+    contact = ObjectVersionsParser.new(self).parse
+
+    [
+      contact.name,
+      contact.code,
+      contact.ident_human_description,
+      contact.registrar,
+      event,
+      created_at.to_formatted_s(:db)
+    ]
+  end
+
+  def self.csv_header
+    ['Name', 'ID', 'Ident', 'Registrar', 'Action', 'Created at']
+  end
 end

app/models/version/domain_version.rb

@@ -1,5 +1,4 @@
 class Version::DomainVersion < PaperTrail::Version
-  extend ToCsv
   include VersionSession
 
   self.table_name    = :log_domains
@@ -7,6 +6,18 @@ class Version::DomainVersion < PaperTrail::Version
 
   scope :deleted, -> { where(event: 'destroy') }
 
+  def as_csv_row
+    domain = ObjectVersionsParser.new(self).parse
+
+    [
+      domain.name,
+      domain.registrant_name,
+      domain.registrar,
+      event,
+      created_at.to_formatted_s(:db)
+    ]
+  end
+
   def self.was_contact_linked?(contact_id)
     sql = <<-SQL
       SELECT
@@ -43,4 +54,8 @@ class Version::DomainVersion < PaperTrail::Version
 
     count_by_sql(sql).nonzero?
   end
+
+  def self.csv_header
+    ['Name', 'Registrant', 'Registrar', 'Action', 'Created at']
+  end
 end

app/services/csv_generator.rb

@@ -0,0 +1,29 @@
+class CsvGenerator
+  class << self
+    def generate_csv(objects)
+      class_name = objects.first.class
+      return default_generation(objects) unless custom_csv?(class_name)
+
+      CSV.generate do |csv|
+        csv << class_name.csv_header
+        objects.each { |object| csv << object.as_csv_row }
+      end
+    end
+
+    private
+
+    def default_generation(objects)
+      CSV.generate do |csv|
+        csv << objects.column_names
+        objects.all.find_each { |object| csv << object.attributes.values_at(*objects.column_names) }
+      end
+    end
+
+    def custom_csv?(class_name)
+      [
+        Version::DomainVersion, Version::ContactVersion, Domain,
+        Contact, Invoice, Account, AccountActivity
+      ].include?(class_name)
+    end
+  end
+end

app/services/object_versions_parser.rb

@@ -0,0 +1,28 @@
+class ObjectVersionsParser
+  def initialize(version)
+    @version = version
+  end
+
+  def parse
+    model = @version.item_type.constantize
+    attributes = only_present_fields(model)
+    history_object = model.new(attributes)
+    attach_existing_fields(history_object) unless @version.event == 'destroy'
+
+    history_object
+  end
+
+  private
+
+  def attach_existing_fields(history_object)
+    @version.object_changes.to_h.each do |key, value|
+      method_name = "#{key}=".to_sym
+      history_object.public_send(method_name, value.last) if history_object.respond_to?(method_name)
+    end
+  end
+
+  def only_present_fields(model)
+    field_names = model.column_names
+    @version.object.to_h.select { |key, _value| field_names.include?(key) }
+  end
+end

app/views/admin/contact_versions/index.haml

@@ -64,9 +64,7 @@
         %tbody
           - @versions.each do |version|
             - if version
-              - attributes = only_present_fields(version, Contact)
-              - contact = Contact.new(attributes)
-              - attach_existing_fields(version, contact)
+              - contact = ObjectVersionsParser.new(version).parse
 
               %tr
                 %td= link_to(contact.name, admin_contact_version_path(version.id))

app/views/admin/contact_versions/show.haml

@@ -1,6 +1,5 @@
-- attributes = only_present_fields(@version, Contact)
-- contact = Contact.new(attributes)
-- attach_existing_fields(@version, contact)
+- contact = ObjectVersionsParser.new(@version).parse
+
 = render 'shared/title', name: contact.name
 
 .row

app/views/admin/domain_versions/archive.haml

@@ -62,9 +62,7 @@
         %tbody
           - @versions.each do |version|
             - if version
-              - attributes = only_present_fields(version, Domain)
-              - domain = Domain.new(attributes)
-              - attach_existing_fields(version, domain) unless version.event == 'destroy'
+              - domain = ObjectVersionsParser.new(version).parse
 
               %tr
                 %td= link_to(domain.name, admin_domain_version_path(version.id))

app/views/admin/domain_versions/show.haml

@@ -1,6 +1,4 @@
-- present_fields = only_present_fields(@version, Domain)
-- domain = Domain.new(present_fields)
-- attach_existing_fields(@version, domain) unless @version.event == 'destroy'
+- domain = ObjectVersionsParser.new(@version).parse
 
 - if @version
   - children = HashWithIndifferentAccess.new(@version.children)