kpm 0.7.2 → 0.10.0

data/Rakefile

@@ -1,4 +1,5 @@
 #!/usr/bin/env rake
+# frozen_string_literal: true
 
 Dir.glob('tasks/*.rake').each { |r| import r }
 
@@ -33,4 +34,4 @@ namespace :test do
 end
 
 # Run tests by default
-task :default => 'test:spec'
+task default: 'test:spec'

data/bin/kpm

@@ -1,10 +1,12 @@
 #!/usr/bin/env ruby
-$:.push File.expand_path("../../lib", __FILE__)
+# frozen_string_literal: true
+
+$LOAD_PATH.push File.expand_path('../lib', __dir__)
 require 'kpm'
 
 begin
   KPM::Cli.start
-rescue => e
+rescue StandardError => e
   KPM.ui.say "#{e.message}\n#{e.backtrace.join("\n")}", :red
   exit 1
 end

data/kpm.gemspec

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 #
 # Copyright 2014 The Billing Project, LLC
 #
@@ -14,7 +16,7 @@
 # under the License.
 #
 
-$LOAD_PATH.unshift File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift File.expand_path('lib', __dir__)
 require 'kpm/version'
 
 Gem::Specification.new do |s|
@@ -34,17 +36,18 @@ Gem::Specification.new do |s|
   s.files         = `git ls-files`.split("\n")
   s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
   s.bindir        = 'bin'
-  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
-  s.require_paths = ["lib"]
+  s.executables   = `git ls-files -- bin/*`.split("\n").map { |f| File.basename(f) }
+  s.require_paths = ['lib']
 
   s.rdoc_options << '--exclude' << '.'
 
   s.add_dependency 'highline', '~> 1.6.21'
-  s.add_dependency 'thor', '~> 0.19.1'
+  s.add_dependency 'killbill-client', '~> 3.2'
   s.add_dependency 'rubyzip', '~>1.2.0'
-  s.add_dependency 'killbill-client', '~> 2.5.0'
+  s.add_dependency 'thor', '~> 0.19.1'
 
-  s.add_development_dependency 'rake', '>= 10.0.0', '< 11.0.0'
-  s.add_development_dependency 'rspec', '~> 2.12.0'
+  s.add_development_dependency 'gem-release', '~> 2.2'
+  s.add_development_dependency 'rake', '~> 13.0'
+  s.add_development_dependency 'rspec', '~> 3.9'
+  s.add_development_dependency 'rubocop', '~> 0.88.0' if RUBY_VERSION >= '2.4'
 end
-

data/lib/kpm.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KPM
   autoload :Utils, 'kpm/utils'
   autoload :BaseArtifact, 'kpm/base_artifact'
@@ -19,6 +21,7 @@ module KPM
   autoload :Migrations, 'kpm/migrations'
   autoload :System, 'kpm/system'
   autoload :Account, 'kpm/account'
+  autoload :Blob, 'kpm/blob'
   autoload :Database, 'kpm/database'
   autoload :TenantConfig, 'kpm/tenant_config'
   autoload :DiagnosticFile, 'kpm/diagnostic_file'

data/lib/kpm/account.rb

@@ -1,17 +1,18 @@
+# frozen_string_literal: true
+
 require 'net/http'
 require 'tmpdir'
 require 'yaml'
 require 'date'
 require 'securerandom'
+require 'base64'
 require 'killbill_client'
 
 module KPM
-
   class Account
-
     # Killbill server
     KILLBILL_HOST = ENV['KILLBILL_HOST'] || '127.0.0.1'
-    KILLBILL_URL = 'http://'.concat(KILLBILL_HOST).concat(':8080')
+    KILLBILL_URL = "http://#{KILLBILL_HOST}:8080"
     KILLBILL_API_VERSION = '1.0'
 
     # USER/PWD
@@ -24,7 +25,7 @@ module KPM
 
     # Temporary directory
     TMP_DIR_PEFIX = 'killbill'
-    TMP_DIR = Dir.mktmpdir(TMP_DIR_PEFIX);
+    TMP_DIR = Dir.mktmpdir(TMP_DIR_PEFIX)
 
     # Created By
     WHO = 'kpm_export_import'
@@ -34,76 +35,72 @@ module KPM
     PLUGIN_NAME_COLUMN = 'plugin_name'
 
     # fields to remove from the export files
-    REMOVE_DATA_FROM = {:accounts => [:name, :address1, :address2, :city, :state_or_province, :phone, :email],
-                       :account_history => [:name, :address1, :address2, :city, :state_or_province, :phone, :email]}
+    REMOVE_DATA_FROM = { accounts: %i[name address1 address2 city state_or_province phone email],
+                         account_history: %i[name address1 address2 city state_or_province phone email] }.freeze
 
-    DATE_COLUMNS_TO_FIX = ['created_date','updated_date','processing_available_date','effective_date',
-                           'boot_date','start_timestamp','last_access_time','payment_date','original_created_date',
-                           'last_sys_update_date','charged_through_date','bundle_start_date','start_date', 'reference_time']
+    DATE_COLUMNS_TO_FIX = %w[created_date updated_date processing_available_date effective_date
+                             boot_date start_timestamp last_access_time payment_date original_created_date
+                             last_sys_update_date charged_through_date bundle_start_date start_date catalog_effective_date reference_time].freeze
 
     # round trip constants duplicate record
-    ROUND_TRIP_EXPORT_IMPORT_MAP = {:accounts => {:id => :accounts_id, :external_key => :accounts_id}, :all => {:account_id => :accounts_id},
-                                    :account_history => {:id => :account_history_id, :external_key => :accounts_id, :payment_method_id => :payment_methods_id},
-                                    :account_emails => {:id => :account_emails_id}, :account_email_history => {:id => :account_email_history_id},
-                                    :subscription_events => {:id => :subscription_events_id},:subscriptions => {:id => :subscriptions_id},
-                                    :bundles => {:id => :bundles_id},:blocking_states => {:id => :blocking_states_id, :blockable_id => nil},
-                                    :invoice_items => {:id => :invoice_items_id, :child_account_id => nil, :invoice_id => :invoices_id, :bundle_id => :bundles_id, :subscription_id => :subscriptions_id },
-                                    :invoices => {:id => :invoices_id},
-                                    :invoice_payments => {:id => :invoice_payments_id, :invoice_id => :invoices_id, :payment_id => :payments_id},
-                                    :invoice_parent_children => {:id => :invoice_parent_children_id, :parent_invoice_id  => nil, :child_invoice_id  => nil, :child_account_id => nil},
-                                    :payment_attempts => {:id => :payment_attempts_id, :payment_method_id  => :payment_methods_id, :transaction_id   => :payment_transactions_id},
-                                    :payment_attempt_history => {:id => :payment_attempt_history_id, :payment_method_id  => :payment_methods_id, :transaction_id   => :payment_transactions_id},
-                                    :payment_methods => {:id => :payment_methods_id, :external_key => :generate},:payment_method_history => {:id => :payment_method_history_id},
-                                    :payments => {:id => :payments_id, :payment_method_id  => :payment_methods_id},
-                                    :payment_history => {:id => :payment_history_id, :payment_method_id  => :payment_methods_id},
-                                    :payment_transactions => {:id => :payment_transactions_id, :payment_id   => :payments_id},
-                                    :payment_transaction_history => {:id => :payment_transaction_history_id, :payment_id   => :payments_id},
-                                    :_invoice_payment_control_plugin_auto_pay_off  => {:payment_method_id => :payment_methods_id, :payment_id   => :payments_id},
-                                    :rolled_up_usage => {:id => :rolled_up_usage_id, :subscription_id  => :subscriptions_id, :tracking_id => nil},
-                                    :custom_fields  => {:id => :custom_fields_id},:custom_field_history  => {:id => :custom_field_history_id},
-                                    :tag_definitions => {:id => :tag_definitions_id},:tag_definition_history => {:id => :tag_definition_history_id},
-                                    :tags => {:id => :tags_id, :object_id => nil},
-                                    :tag_history => {:id => :tag_history_id, :object_id => nil},
-                                    :audit_log => {:id => :audit_log_id}
-    }
-
-    #delimeters to sniff
-    DELIMITERS = [',','|']
-    DEFAULT_DELIMITER = "|"
+    ROUND_TRIP_EXPORT_IMPORT_MAP = { accounts: { id: :accounts_id, external_key: :accounts_id }, all: { account_id: :accounts_id },
+                                     account_history: { id: :account_history_id, external_key: :accounts_id, payment_method_id: :payment_methods_id },
+                                     account_emails: { id: :account_emails_id }, account_email_history: { id: :account_email_history_id },
+                                     subscription_events: { id: :subscription_events_id }, subscriptions: { id: :subscriptions_id },
+                                     bundles: { id: :bundles_id }, blocking_states: { id: :blocking_states_id, blockable_id: nil },
+                                     invoice_items: { id: :invoice_items_id, child_account_id: nil, invoice_id: :invoices_id, bundle_id: :bundles_id, subscription_id: :subscriptions_id },
+                                     invoices: { id: :invoices_id },
+                                     invoice_payments: { id: :invoice_payments_id, invoice_id: :invoices_id, payment_id: :payments_id },
+                                     invoice_parent_children: { id: :invoice_parent_children_id, parent_invoice_id: nil, child_invoice_id: nil, child_account_id: nil },
+                                     payment_attempts: { id: :payment_attempts_id, payment_method_id: :payment_methods_id, transaction_id: :payment_transactions_id },
+                                     payment_attempt_history: { id: :payment_attempt_history_id, payment_method_id: :payment_methods_id, transaction_id: :payment_transactions_id },
+                                     payment_methods: { id: :payment_methods_id, external_key: :generate }, payment_method_history: { id: :payment_method_history_id },
+                                     payments: { id: :payments_id, payment_method_id: :payment_methods_id },
+                                     payment_history: { id: :payment_history_id, payment_method_id: :payment_methods_id },
+                                     payment_transactions: { id: :payment_transactions_id, payment_id: :payments_id },
+                                     payment_transaction_history: { id: :payment_transaction_history_id, payment_id: :payments_id },
+                                     _invoice_payment_control_plugin_auto_pay_off: { payment_method_id: :payment_methods_id, payment_id: :payments_id },
+                                     rolled_up_usage: { id: :rolled_up_usage_id, subscription_id: :subscriptions_id, tracking_id: nil },
+                                     custom_fields: { id: :custom_fields_id }, custom_field_history: { id: :custom_field_history_id },
+                                     tag_definitions: { id: :tag_definitions_id }, tag_definition_history: { id: :tag_definition_history_id },
+                                     tags: { id: :tags_id, object_id: nil },
+                                     tag_history: { id: :tag_history_id, object_id: nil },
+                                     audit_log: { id: :audit_log_id } }.freeze
+
+    # delimeters to sniff
+    DELIMITERS = [',', '|'].freeze
+    DEFAULT_DELIMITER = '|'
+
+    B64_REGEX = %r{^([A-Za-z0-9+/]{4})*([A-Za-z0-9+/]{3}=|[A-Za-z0-9+/]{2}==)?$}.freeze
 
     def initialize(config_file = nil, killbill_api_credentials = nil, killbill_credentials = nil, killbill_url = nil,
                    database_name = nil, database_credentials = nil, database_host = nil, database_port = nil, data_delimiter = nil, logger = nil)
       @killbill_api_key = KILLBILL_API_KEY
-      @killbill_api_secrets = KILLBILL_API_SECRET
+      @killbill_api_secret = KILLBILL_API_SECRET
       @killbill_url = KILLBILL_URL
       @killbill_user = KILLBILL_USER
       @killbill_password = KILLBILL_PASSWORD
       @delimiter = data_delimiter || DEFAULT_DELIMITER
       @logger = logger
-      @tables_id = Hash.new
+      @tables_id = {}
 
+      set_killbill_options(killbill_api_credentials, killbill_credentials, killbill_url)
 
-      set_killbill_options(killbill_api_credentials,killbill_credentials,killbill_url)
-      set_database_options(database_host,database_port,database_name,database_credentials,logger)
+      database_credentials ||= [nil, nil]
+      @database = Database.new(database_name, database_host, database_port, database_credentials[0], database_credentials[1], logger)
 
       load_config_from_file(config_file)
-
     end
 
     def export_data(account_id = nil)
-
-      if account_id === :export.to_s
-        raise Interrupt, 'Need to specify an account id'
-      end
+      raise Interrupt, 'Need to specify an account id' if account_id == :export.to_s
 
       export_data = fetch_export_data(account_id)
       export_file = export(export_data)
 
-      unless File.exist?(export_file)
-        raise Interrupt, 'Account id not found'
-      else
-        @logger.info "\e[32mData exported under #{export_file}\e[0m"
-      end
+      raise Interrupt, 'Account id not found' unless File.exist?(export_file)
+
+      @logger.info "\e[32mData exported under #{export_file}\e[0m"
 
       export_file
     end
@@ -115,13 +112,9 @@ module KPM
       @tenant_record_id = tenant_record_id
       @round_trip_export_import = round_trip_export_import
 
-      if source_file === :import.to_s
-        raise Interrupt, 'Need to specify a file'
-      end
+      raise Interrupt, 'Need to specify a file' if source_file == :import.to_s
 
-      unless File.exist?(source_file)
-        raise Interrupt, "File #{source_file} does not exist"
-      end
+      raise Interrupt, "File #{source_file} does not exist" unless File.exist?(source_file)
 
       @delimiter = sniff_delimiter(source_file) || @delimiter
 
@@ -130,394 +123,331 @@ module KPM
 
     private
 
-      # export helpers: fetch_export_data; export; process_export_data; remove_export_data;
-      def fetch_export_data(account_id)
-        KillBillClient.url = @killbill_url
-        options = {
-          :username => @killbill_user,
-          :password => @killbill_password,
-          :api_key => @killbill_api_key,
-          :api_secret => @killbill_api_secrets
-        }
-
-        begin
-          account_data = KillBillClient::Model::Export.find_by_account_id(account_id, 'KPM', options)
-        rescue Exception => e
-          raise Interrupt, 'Account id not found'
-        end
-
-        account_data
+    # export helpers: fetch_export_data; export; process_export_data; remove_export_data;
+    def fetch_export_data(account_id)
+      KillBillClient.url = @killbill_url
+      options = {
+        username: @killbill_user,
+        password: @killbill_password,
+        api_key: @killbill_api_key,
+        api_secret: @killbill_api_secret
+      }
+
+      begin
+        account_data = KillBillClient::Model::Export.find_by_account_id(account_id, 'KPM', options)
+      rescue StandardError
+        raise Interrupt, 'Account id not found'
       end
 
-      def export(export_data)
-        export_file = TMP_DIR + File::SEPARATOR + 'kbdump'
-
-        open (export_file), 'w' do |io|
-
-          table_name = nil
-          cols_names = nil
-          export_data.split("\n").each do |line|
-            words = line.strip.split(" ")
-            clean_line = line
-            if not /--/.match(words[0]).nil?
-              table_name = words[1]
-              cols_names = words[2].strip.split(@delimiter)
-            elsif not table_name.nil?
-              clean_line = process_export_data(line,table_name,cols_names)
-            end
-            io.puts clean_line
+      account_data
+    end
 
+    def export(export_data)
+      export_file = TMP_DIR + File::SEPARATOR + 'kbdump'
+
+      File.open(export_file, 'w') do |io|
+        table_name = nil
+        cols_names = nil
+        export_data.split("\n").each do |line|
+          words = line.strip.split(' ')
+          clean_line = line
+          if !/--/.match(words[0]).nil?
+            table_name = words[1]
+            cols_names = words[2].strip.split(@delimiter)
+          elsif !table_name.nil?
+            clean_line = process_export_data(line, table_name, cols_names)
           end
-
+          io.puts clean_line
         end
-
-        export_file
       end
 
-      def process_export_data(line_to_process, table_name, cols_names)
-        clean_line = line_to_process
-
-        row = []
-        cols = clean_line.strip.split(@delimiter)
-        cols_names.each_with_index { |col_name, index|
-          sanitized_value = remove_export_data(table_name,col_name,cols[index])
-
-          row << sanitized_value
+      export_file
+    end
 
-        }
+    def process_export_data(line_to_process, table_name, cols_names)
+      clean_line = line_to_process
 
-        clean_line = row.join(@delimiter)
+      row = []
+      cols = clean_line.strip.split(@delimiter)
+      cols_names.each_with_index do |col_name, index|
+        sanitized_value = remove_export_data(table_name, col_name, cols[index])
 
-        clean_line
+        row << sanitized_value
       end
 
-      def remove_export_data(table_name,col_name,value)
-
-        if not REMOVE_DATA_FROM[table_name.to_sym].nil?
+      row.join(@delimiter)
+    end
 
-          if REMOVE_DATA_FROM[table_name.to_sym].include? col_name.to_sym
-            return nil
-          end
+    def remove_export_data(table_name, col_name, value)
+      unless REMOVE_DATA_FROM[table_name.to_sym].nil?
 
-        end
+        return nil if REMOVE_DATA_FROM[table_name.to_sym].include? col_name.to_sym
 
-        value
       end
 
-      # import helpers: sanitize_and_import; import; sanitize; replace_tenant_record_id; replace_account_record_id; replace_boolean;
-      # fix_dates; fill_empty_column;
-      def sanitize_and_import(source_file, skip_payment_methods)
-        tables = Hash.new
-        error_importing_data = false
-
-        open (source_file), 'r' do |data|
-
-          rows = nil;
-          table_name = nil;
-          cols_names = nil;
-
-          data.each_line do |line|
-            words = line.strip.split(" ")
-
-            if /--/.match(words[0])
-              unless table_name.nil?
-                if @generate_record_id
-                  cols_names.shift
-                end
+      value
+    end
 
-                tables[table_name] = { :col_names => cols_names, :rows => rows};
-              end
+    # import helpers: sanitize_and_import; import; sanitize; replace_tenant_record_id; replace_account_record_id; replace_boolean;
+    # fix_dates; fill_empty_column;
+    def sanitize_and_import(source_file, skip_payment_methods)
+      tables = {}
+      error_importing_data = false
 
-              table_name = words[1]
-              cols_names = words[2].strip.split(@delimiter)
+      File.open(source_file, 'r:UTF-8') do |data|
+        rows = nil
+        table_name = nil
+        cols_names = nil
 
-              rows = []
-            elsif not table_name.nil?
-              row = process_import_data(line, table_name,cols_names, skip_payment_methods, rows)
+        data.each_line do |line|
+          words = line.strip.split(' ')
 
-              next if row.nil?
+          if /--/.match(words[0])
+            unless table_name.nil?
+              cols_names.shift if @generate_record_id
 
-              rows.push(row)
-            else
-              error_importing_data = true
-              break
+              tables[table_name] = { col_names: cols_names, rows: rows }
             end
-          end
 
-          if not ( table_name.nil? || error_importing_data )
-            if @generate_record_id
-              cols_names.shift
-            end
+            table_name = words[1]
+            cols_names = words[2].strip.split(@delimiter)
 
-            tables[table_name] = { :col_names => cols_names, :rows => rows};
-          end
+            rows = []
+          elsif !table_name.nil?
+            row = process_import_data(line, table_name, cols_names, skip_payment_methods, rows)
 
-          if tables.empty?
+            next if row.nil?
+
+            rows.push(row)
+          else
             error_importing_data = true
+            break
           end
         end
 
-        unless error_importing_data
-          import(tables)
-        else
-          raise Interrupt, "Data on #{source_file} is invalid"
+        unless table_name.nil? || error_importing_data
+          cols_names.shift if @generate_record_id
+
+          tables[table_name] = { col_names: cols_names, rows: rows }
         end
 
+        error_importing_data = true if tables.empty?
       end
 
-      def process_import_data(line, table_name, cols_names, skip_payment_methods, rows)
-        # to make sure that the last column is not omitted if is empty
-        cols = line.strip.split(@delimiter,line.count(@delimiter)+1)
-
-        if cols_names.size != cols.size
-          @logger.warn "\e[32mWARNING!!! On #{table_name} table there is a mismatch on column count[#{cols.size}] versus header count[#{cols_names.size}]\e[0m"
-          return nil
-        end
+      raise Interrupt, "Data on #{source_file} is invalid" if error_importing_data
 
-        row = []
+      import(tables)
+    end
 
-        @logger.debug "Processing table_name=#{table_name}, line=#{line}"
-        cols_names.each_with_index do |col_name, index|
-          sanitized_value = sanitize(table_name,col_name,cols[index], skip_payment_methods)
+    def process_import_data(line, table_name, cols_names, skip_payment_methods, _rows)
+      # to make sure that the last column is not omitted if is empty
+      cols = line.strip.split(@delimiter, line.count(@delimiter) + 1)
 
-          unless sanitized_value.nil?
-            row << sanitized_value
-          end
-        end
-
-        return row
+      if cols_names.size != cols.size
+        @logger.warn "\e[32mWARNING!!! On #{table_name} table there is a mismatch on column count[#{cols.size}] versus header count[#{cols_names.size}]\e[0m"
+        return nil
       end
 
-      def import(tables)
-        record_id = nil;
-        statements = Database.generate_insert_statement(tables)
-        statements.each do |statement|
-          response = Database.execute_insert_statement(statement[:table_name],statement[:query], statement[:qty_to_insert], statement[:table_data],record_id)
+      row = []
 
-          if statement[:table_name] == 'accounts' && response.is_a?(String)
-            record_id = {:variable => '@account_record_id', :value => response}
-          end
-
-          if response === false
-            break
-          end
-        end
+      @logger.debug "Processing table_name=#{table_name}, line=#{line}"
+      cols_names.each_with_index do |col_name, index|
+        sanitized_value = sanitize(table_name, col_name, cols[index], skip_payment_methods)
 
+        row << sanitized_value unless sanitized_value.nil?
       end
 
-      def sanitize(table_name,column_name,value,skip_payment_methods)
-        sanitized_value = replace_boolean(value)
-        sanitized_value = fill_empty_column(sanitized_value)
+      row
+    end
 
-        if table_name == 'payment_methods' && skip_payment_methods && column_name == PLUGIN_NAME_COLUMN
-          sanitized_value = SAFE_PAYMENT_METHOD
-        end
+    def import(tables)
+      record_id = nil
+      statements = @database.generate_insert_statement(tables)
+      statements.each do |statement|
+        response = @database.execute_insert_statement(statement[:table_name], statement[:query], statement[:qty_to_insert], statement[:table_data], record_id)
 
-        if DATE_COLUMNS_TO_FIX.include? column_name
-          sanitized_value = fix_dates(sanitized_value)
-        end
+        record_id = { variable: '@account_record_id', value: response } if statement[:table_name] == 'accounts' && response.is_a?(String)
 
-        if not @tenant_record_id.nil?
-          sanitized_value = replace_tenant_record_id(table_name,column_name,sanitized_value)
-        end
+        break unless response
+      end
+    end
 
-        if @generate_record_id
-          sanitized_value = replace_account_record_id(table_name,column_name,sanitized_value)
-        end
+    def sanitize(table_name, column_name, value, skip_payment_methods)
+      sanitized_value = replace_boolean(value)
 
-        if @round_trip_export_import
-          sanitized_value = replace_uuid(table_name,column_name,sanitized_value)
-        end
+      sanitized_value = fill_empty_column(sanitized_value)
 
-        sanitized_value
-      end
+      sanitized_value = SAFE_PAYMENT_METHOD if table_name == 'payment_methods' && skip_payment_methods && column_name == PLUGIN_NAME_COLUMN
 
-      def replace_tenant_record_id(table_name,column_name,value)
-        return @tenant_record_id if column_name == 'tenant_record_id' || column_name == 'search_key2'
-        value
-      end
+      sanitized_value = fix_dates(sanitized_value) if DATE_COLUMNS_TO_FIX.include? column_name
 
-      def replace_account_record_id(table_name,column_name,value)
+      sanitized_value = replace_tenant_record_id(table_name, column_name, sanitized_value) unless @tenant_record_id.nil?
 
-        if column_name == 'account_record_id'
+      sanitized_value = replace_account_record_id(table_name, column_name, sanitized_value) if @generate_record_id
 
-          return :@account_record_id
-        end
+      sanitized_value = replace_uuid(table_name, column_name, sanitized_value) if @round_trip_export_import
 
-        if column_name == 'record_id'
-          return nil
-        end
+      sanitized_value = b64_decode_if_needed(sanitized_value) if column_name == 'billing_events'
 
-        if column_name == 'target_record_id'
+      sanitized_value
+    end
 
-          if table_name == 'account_history'
-            return :@account_record_id
-          end
-        end
+    def replace_tenant_record_id(_table_name, column_name, value)
+      return @tenant_record_id if %w[tenant_record_id search_key2].include?(column_name)
 
-        if column_name == 'search_key1' && table_name == 'bus_ext_events_history'
-            return :@account_record_id
-        end
+      value
+    end
 
-        if column_name == 'search_key1' && table_name == 'bus_events_history'
-          return :@account_record_id
-        end
+    def replace_account_record_id(table_name, column_name, value)
+      return :@account_record_id if column_name == 'account_record_id'
 
-        value
+      return nil if column_name == 'record_id'
 
-      end
+      if column_name == 'target_record_id'
 
-      def replace_boolean(value)
-        if value.to_s === 'true'
-          return 1
-        elsif value.to_s === 'false'
-          return 0
-        else
-          return value
-        end
+        return :@account_record_id if table_name == 'account_history'
       end
 
-      def fix_dates(value)
-        if !value.equal?(:DEFAULT)
+      return :@account_record_id if column_name == 'search_key1' && table_name == 'bus_ext_events_history'
 
-          dt = DateTime.parse(value)
-          return dt.strftime('%F %T').to_s
+      return :@account_record_id if column_name == 'search_key1' && table_name == 'bus_events_history'
 
-        end
+      value
+    end
 
+    def replace_boolean(value)
+      case value.to_s
+      when 'true'
+        1
+      when 'false'
+        0
+      else
         value
       end
+    end
 
-      def fill_empty_column(value)
-        if value.to_s.strip.empty?
-          return :DEFAULT
-        else
-          return value
-        end
-      end
-
-      def replace_uuid(table_name,column_name,value)
-
-        if column_name == 'id'
-          @tables_id["#{table_name}_id"] = SecureRandom.uuid
-        end
-
-        if ROUND_TRIP_EXPORT_IMPORT_MAP[table_name.to_sym] && ROUND_TRIP_EXPORT_IMPORT_MAP[table_name.to_sym][column_name.to_sym]
-          key = ROUND_TRIP_EXPORT_IMPORT_MAP[table_name.to_sym][column_name.to_sym]
+    def fix_dates(value)
+      unless value.equal?(:DEFAULT)
 
-          if key.equal?(:generate)
-            new_value = SecureRandom.uuid
-          else
-            new_value = @tables_id[key.to_s]
-          end
-
-          if new_value.nil?
-            new_value = SecureRandom.uuid
-            @tables_id[key.to_s] = new_value
-          end
-          return new_value
-        end
+        dt = DateTime.parse(value)
+        return dt.strftime('%F %T').to_s
 
-        if not ROUND_TRIP_EXPORT_IMPORT_MAP[:all][column_name.to_sym].nil?
-          key = ROUND_TRIP_EXPORT_IMPORT_MAP[:all][column_name.to_sym]
-          new_value = @tables_id[key.to_s]
+      end
 
-          return new_value
-        end
+      value
+    end
 
+    def fill_empty_column(value)
+      if value.to_s.strip.empty?
+        :DEFAULT
+      else
         value
       end
+    end
 
-      def sniff_delimiter(file)
-
-        return nil if File.size?(file).nil?
-
-        first_line = File.open(file) {|f| f.readline}
+    def replace_uuid(table_name, column_name, value)
+      @tables_id["#{table_name}_id"] = SecureRandom.uuid if column_name == 'id'
 
-        return nil if first_line.nil?
+      if ROUND_TRIP_EXPORT_IMPORT_MAP[table_name.to_sym] && ROUND_TRIP_EXPORT_IMPORT_MAP[table_name.to_sym][column_name.to_sym]
+        key = ROUND_TRIP_EXPORT_IMPORT_MAP[table_name.to_sym][column_name.to_sym]
 
-        sniff = {}
+        new_value = if key.equal?(:generate)
+                      SecureRandom.uuid
+                    else
+                      @tables_id[key.to_s]
+                    end
 
-        DELIMITERS.each do |delimiter|
-          sniff[delimiter] = first_line.count(delimiter)
+        if new_value.nil?
+          new_value = SecureRandom.uuid
+          @tables_id[key.to_s] = new_value
         end
-
-        sniff = sniff.sort {|a,b| b[1]<=>a[1]}
-        sniff.size > 0 ? sniff[0][0] : nil
+        return new_value
       end
 
-      # helper methods that set up killbill and database options: load_config_from_file; set_config; set_database_options;
-      # set_killbill_options;
-      def load_config_from_file(config_file)
+      unless ROUND_TRIP_EXPORT_IMPORT_MAP[:all][column_name.to_sym].nil?
+        key = ROUND_TRIP_EXPORT_IMPORT_MAP[:all][column_name.to_sym]
+        new_value = @tables_id[key.to_s]
 
-        set_config(config_file)
+        return new_value
+      end
 
-        if not @config.nil?
-          config_killbill = @config['killbill']
+      value
+    end
 
-          if not config_killbill.nil?
-            set_killbill_options([config_killbill['api_key'],config_killbill['api_secret']],
-                                 [config_killbill['user'],config_killbill['password']],
-                                 "http://#{config_killbill['host']}:#{config_killbill['port']}")
-          end
+    def b64_decode_if_needed(input)
+      # Exclude nil or non string
+      return input if input.nil? || !input.is_a?(String)
+      # Apply regex to check that string is built as a B64 string: the character set is [A-Z, a-z, 0-9, and + /]
+      # and if the rest length is less than 4, the string is padded with '=' characters.
+      return input if input.match(B64_REGEX).nil?
 
-          config_db = @config['database']
+      # Decode
+      result = Base64.decode64(input)
+      # Verify encoded of the decoded value == input prior return result
+      return input if  Base64.strict_encode64(result) != input
 
-          if not config_db.nil?
-            set_database_options(config_db['host'],config_db['name'],
-                                 [config_db['username'],config_db['password']],
-                                 @logger)
+      Blob.new(result, TMP_DIR)
+    end
 
-          end
-        end
-      end
+    def sniff_delimiter(file)
+      return nil if File.size?(file).nil?
 
-      def set_config(config_file = nil)
-        @config = nil
+      first_line = File.open(file, &:readline)
 
-        if not config_file.nil?
-          if not Dir[config_file][0].nil?
-            @config = YAML::load_file(config_file)
-          end
-        end
+      return nil if first_line.nil?
 
+      sniff = {}
+
+      DELIMITERS.each do |delimiter|
+        sniff[delimiter] = first_line.count(delimiter)
       end
 
-      def set_database_options(database_host = nil, database_port = nil, database_name = nil, database_credentials = nil, logger)
+      sniff = sniff.sort { |a, b| b[1] <=> a[1] }
+      !sniff.empty? ? sniff[0][0] : nil
+    end
 
-        Database.set_logger(logger)
+    def load_config_from_file(config_file)
+      self.config = config_file
 
-        Database.set_credentials(database_credentials[0],database_credentials[1]) unless database_credentials.nil?
-        Database.set_database_name(database_name) unless database_name.nil?
-        Database.set_host(database_host) unless database_host.nil?
-        Database.set_port(database_port) unless database_port.nil?
+      return if @config.nil?
 
-        Database.set_mysql_command_line
+      config_killbill = @config['killbill']
+
+      unless config_killbill.nil?
+        set_killbill_options([config_killbill['api_key'], config_killbill['api_secret']],
+                             [config_killbill['user'], config_killbill['password']],
+                             "http://#{config_killbill['host']}:#{config_killbill['port']}")
       end
 
-      def set_killbill_options(killbill_api_credentials, killbill_credentials, killbill_url)
+      config_db = @config['database']
 
-        if not killbill_api_credentials.nil?
+      @database = Database.new(config_db['name'], config_db['host'], config_db['port'], config_db['username'], config_db['password'], @logger) unless config_db.nil?
+    end
 
-          @killbill_api_key = killbill_api_credentials[0]
-          @killbill_api_secrets = killbill_api_credentials[1]
+    def config=(config_file = nil)
+      @config = nil
 
-        end
+      return if config_file.nil?
 
-        if not killbill_credentials.nil?
+      @config = YAML.load_file(config_file) unless Dir[config_file][0].nil?
+    end
 
-          @killbill_user = killbill_credentials[0]
-          @killbill_password = killbill_credentials[1]
+    def set_killbill_options(killbill_api_credentials, killbill_credentials, killbill_url)
+      unless killbill_api_credentials.nil?
 
-        end
+        @killbill_api_key = killbill_api_credentials[0]
+        @killbill_api_secret = killbill_api_credentials[1]
 
-        if not killbill_url.nil?
+      end
 
-          @killbill_url = killbill_url
+      unless killbill_credentials.nil?
+
+        @killbill_user = killbill_credentials[0]
+        @killbill_password = killbill_credentials[1]
 
-        end
       end
 
+      @killbill_url = killbill_url unless killbill_url.nil?
+    end
   end
-
-end
+end