class LegalDocument < ApplicationRecord # rubocop:disable Metrics/ClassLength include EppErrors MIN_BODY_SIZE = (1.37 * 3.kilobytes).ceil MAX_BODY_SIZE = 8.megabytes if ENV['legal_document_types'].present? TYPES = ENV['legal_document_types'].split(',').map(&:strip) else TYPES = %w(pdf asice asics sce scs adoc edoc bdoc ddoc zip rar gz tar 7z odt doc docx).freeze end attr_accessor :body belongs_to :documentable, polymorphic: true validate :val_body_length, if: ->(file) { file.path.blank? && (Rails.env.production? || Rails.env.test?) } before_create :add_creator before_save :save_to_filesystem, if: :body def epp_code_map { '2308' => [ %i[body length_more_than], %i[body length_less_than], ], } end def val_body_length if body.nil? || body.size < MIN_BODY_SIZE errors.add(:body, :length_more_than) elsif body.size > MAX_BODY_SIZE errors.add(:body, :length_less_than) end end def save_to_filesystem binary = Base64.decode64(body) digest = Digest::SHA1.new.update(binary).to_s loop do rand = SecureRandom.random_number.to_s.last(4) next if rand.to_i.zero? || rand.length < 4 dir = "#{ENV['legal_documents_dir']}/#{Time.zone.now.strftime('%Y/%m/%d')}" FileUtils.mkdir_p(dir, mode: 0775) self.path = "#{dir}/#{Time.zone.now.to_formatted_s(:number)}_#{rand}.#{document_type}" break unless File.file?(path) end File.open(path, 'wb') { |f| f.write(binary) } unless Rails.env.test? self.path = path self.checksum = digest end def calc_checksum digest = Digest::SHA1.new digest.update File.binread(path) digest.to_s end def add_creator self.creator_str = ::PaperTrail.request.whodunnit true end def self.remove_duplicates start = Time.zone.now.to_f Rails.logger.info '-----> Removing legal documents duplicates' count = 0 modified = [] LegalDocument.where(documentable_type: 'Domain') .where.not(checksum: [nil, '']) .find_each do |orig_legal| next if modified.include?(orig_legal.checksum) next unless File.exist?(orig_legal.path) modified.push(orig_legal.checksum) LegalDocument.where(documentable_type: 'Domain', documentable_id: orig_legal.documentable_id) .where(checksum: orig_legal.checksum) .where.not(id: orig_legal.id) .where.not(path: orig_legal.path).each do |new_legal| next if modified.include?(orig_legal.id) File.delete(new_legal.path) if File.exist?(new_legal.path) new_legal.update(path: orig_legal.path) count += 1 Rails.logger.info "File #{new_legal.path} has been removed by Domain "\ "#{new_legal.documentable_id}. Document id: #{new_legal.id}" end contact_ids = Version::DomainVersion.where(item_id: orig_legal.documentable_id).distinct .pluck("object->>'registrant_id'", "object_changes->>'registrant_id'", "children->>'tech_contacts'", "children->>'admin_contacts'") .flatten.uniq contact_ids = contact_ids.map do |id| case id when Hash id['id'] when String JSON.parse(id) rescue id.to_i else id end end.flatten.compact.uniq LegalDocument.where(documentable_type: 'Contact', documentable_id: contact_ids) .where(checksum: orig_legal.checksum) .where.not(path: orig_legal.path).each do |new_legal| next if modified.include?(orig_legal.id) File.delete(new_legal.path) if File.exist?(new_legal.path) new_legal.update(path: orig_legal.path) count += 1 Rails.logger.info "File #{new_legal.path} has been removed by Contact "\ "#{new_legal.documentable_id}. Document id: #{new_legal.id}" end end Rails.logger.info "-----> Duplicates fixed for #{count} rows in #{(Time.zone.now.to_f - start).round(2)} seconds" end end