autotelik/datashift

# Copyright:: (c) Autotelik Media Ltd 2016
# Author ::   Tom Statter
# Date ::     Mar 2016
# License::   MIT.
#
require 'thor'
require_relative 'thor_behaviour'

# Note, not DataShift, case sensitive, create namespace for command line : datashift
module Datashift

  class Import < Thor

    include DataShift::ThorBehavior

=begin - On class_options

  We get      Thor::RequiredArgumentMissingError:
         No value provided for required options '--model', '--input'

  when we try
      Datashift::Import.new

    class_option :model, aliases: '-m', required: true, desc: 'The related active record model'

    class_option :input, aliases: '-i', required: true, desc: 'The input file'

    class_option :loader, aliases: '-l', required: false, desc: 'Loader class to use'

    class_option :verbose, aliases: '-v', type: :boolean, desc: 'Verbose logging'

    class_option :config, aliases: '-c', desc: 'YAML config file with defaults, over-rides etc'
=end

    desc 'load', 'Import data from file for specific active record model'

    method_option :model, aliases: '-m', required: true, desc: 'The related active record model'

    method_option :input, aliases: '-i', required: true, desc: 'The input file'

    method_option :loader, aliases: '-l', required: false, desc: 'Loader class to use'

    method_option :verbose, aliases: '-v', type: :boolean, desc: 'Verbose logging'

    method_option :config, aliases: '-c', desc: 'YAML config file with defaults, over-rides etc'

    method_option :force, aliases: '-f', type: :array, desc: "Call-able columns even though datashift does not recognise them as model methods. Space separated e.g : -f sku price"

    def load
      start_connections

      importer = if(options[:loader])
                   logger.info("Attempting to use supplied Loader : #{options[:loader]}")
                   DataShift::MapperUtils.class_from_string(options[:loader]).new
                 else
                   logger.info('No Loader specified - finding appropriate Loader for file type')
                   DataShift::Loader::Factory.get_loader(options[:input])
                 end

      import(importer)
    end

    desc 'excel', 'Import .xls file for specifiec active record model'

    method_option :model, aliases: '-m', required: true, desc: 'The related active record model'

    method_option :input, aliases: '-i', required: true, desc: 'The input file'

    method_option :loader, aliases: '-l', required: false, desc: 'Loader class to use'

    method_option :verbose, aliases: '-v', type: :boolean, desc: 'Verbose logging'

    method_option :config, aliases: '-c', desc: 'YAML config file with defaults, over-rides etc'

    method_option :force, aliases: '-f', type: :array, desc: "Call-able columns even though datashift does not recognise them as model methods. Space separated e.g : -f sku price"

    def excel

      start_connections

      importer = if(options[:loader])
                   logger.info("Attempting to use supplied Loader : #{options[:loader]}")
                   DataShift::MapperUtils.class_from_string(options[:loader]).new
                 else
                   logger.info('No Loader specified - using standard Excel Loader')
                   DataShift::ExcelLoader.new
                 end

      import(importer)
    end

    desc 'csv', 'Import CSV file for specified active record model'

    method_option :model, aliases: '-m', required: true, desc: 'The related active record model'

    method_option :input, aliases: '-i', required: true, desc: 'The input file'

    method_option :loader, aliases: '-l', required: false, desc: 'Loader class to use'

    method_option :verbose, aliases: '-v', type: :boolean, desc: 'Verbose logging'

    method_option :config, aliases: '-c', desc: 'YAML config file with defaults, over-rides etc'

    method_option :force, aliases: '-f', type: :array, desc: "Call-able columns even though datashift does not recognise them as model methods. Space separated e.g : -f sku price"

    def csv

      start_connections

      importer = if(options[:loader])
                   logger.info("Attempting to use supplied Loader : #{options[:loader]}")
                   DataShift::MapperUtils.class_from_string(options[:loader]).new
                 else
                   logger.info('No Loader specified - using standard Csv Loader')
                   DataShift::CsvLoader.new
                 end

      import(importer)
    end

    no_commands do

      def import(importer)
        logger.info "Datashift: Starting Import from #{options[:input]}"

        importer.configure_from( options[:config] ) if(options[:config])

        DataShift::Configuration.call.force_inclusion_of_columns = options[:force]

        importer.run(options[:input], options[:model])
      end

    end

  end
end