Contents

require "sshkit"

module EveApp
  module SDE
    class DataImporter < SSHKit::Backend::Local
      include Downloader
      include Normalizer

      def initialize
        super
        execute :mkdir, '-p', SDE.config.tmp_path
      end

      private

      def db_config
        ActiveRecord::Base.connection_config
      end

      def sql(sql)
        log "[SQL] #{sql}"
        db.execute(sql)
      end

      def db
        ActiveRecord::Base.connection
      end

      def table_list
        @_table_list ||= begin
          whitelist = Array[SDE.config.table_whitelist].flatten.compact.map(&:to_s)
          whitelist += %w[invMarketGroups invCategories invGroups] if whitelist.include?('invTypes')
          whitelist = whitelist.uniq

          tables = SDE.table_info
          tables.transform_values!(&:symbolize_keys)
          tables.select! { |(name,_)| whitelist.include?(name) } if whitelist.any?

          Hash[tables.map { |(name,info)| [name, { name: normalize_table_name(name) }.merge(info)] }]
        end
      end

      def table_names
        Hash[table_list.map { |(on,info)| [on, info[:name]] }]
      end

      def normalize_table_name(name)
        (SDE.config.table_prefix.to_s + name.gsub(/^#{SDE::PREFIXES.join('|')}/, '')).pluralize.underscore
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
eve_app-0.1.34	lib/eve_app/sde/data_importer.rb
eve_app-0.1.33	lib/eve_app/sde/data_importer.rb