csv2hash.rb in csv2hash-0.3.0

- old
+ new
@@ -1,94 +1,102 @@
-require 'csv2hash/version'
-require 'csv2hash/definition'
-require 'csv2hash/validator'
-require 'csv2hash/validator/mapping'
-require 'csv2hash/validator/collection'
-require 'csv2hash/structure_validator'
-require 'csv2hash/structure_validator/max_columns'
-require 'csv2hash/structure_validator/min_columns'
-require 'csv2hash/parser'
-require 'csv2hash/parser/mapping'
-require 'csv2hash/parser/collection'
-require 'csv2hash/csv_array'
-require 'csv2hash/data_wrapper'
-require 'csv2hash/notifier'
-require 'csv2hash/extra_validator'
+require_relative 'csv2hash/version'
+require_relative 'csv2hash/definition'
+require_relative 'csv2hash/validator'
+require_relative 'csv2hash/validator/mapping'
+require_relative 'csv2hash/validator/collection'
+require_relative 'csv2hash/structure_validator'
+require_relative 'csv2hash/structure_validator/max_columns'
+require_relative 'csv2hash/structure_validator/min_columns'
+require_relative 'csv2hash/parser'
+require_relative 'csv2hash/parser/mapping'
+require_relative 'csv2hash/parser/collection'
+require_relative 'csv2hash/csv_array'
+require_relative 'csv2hash/data_wrapper'
+require_relative 'csv2hash/notifier'
+require_relative 'csv2hash/extra_validator'
+require_relative 'csv2hash/adapters/base'
 
-require 'csv'
+begin
+  require 'pry'
+rescue LoadError
+end
 
-class Csv2hash
+module Csv2hash
+  class Main
+    include Csv2hash::StructureValidator
 
-  include Csv2hash::StructureValidator
+    attr_accessor :definition, :file_path_or_data, :data, :notifier, :exception_mode, :errors, :ignore_blank_line
 
-  attr_accessor :definition, :file_path, :data, :notifier, :exception_mode, :errors, :ignore_blank_line
+    def initialize definition, file_path_or_data, exception_mode=true, ignore_blank_line=false
+      self.definition, self.file_path_or_data = definition, file_path_or_data
+      @data_source = data_source
+      dynamic_lib_loading 'Parser'
+      self.exception_mode, self.errors = exception_mode, []
+      dynamic_lib_loading 'Validator'
+      self.notifier = Notifier.new
+      self.ignore_blank_line = ignore_blank_line
 
-  def initialize definition, file_path, exception_mode=true, data_source=nil, ignore_blank_line=false
-    @data_source = data_source
-    self.definition, self.file_path = definition, file_path
-    dynamic_lib_loading 'Parser'
-    self.exception_mode, self.errors = exception_mode, []
-    dynamic_lib_loading 'Validator'
-    self.notifier = Notifier.new
-    self.ignore_blank_line = ignore_blank_line
-    init_plugins
+      init_plugins
+    end
 
-  end
+    def init_plugins
+      begin
+        @plugins = []
+        ::Csv2hash::Plugins.constants.each do |name|
+          @plugins << ::Csv2hash::Plugins.const_get(name).new(self)
+        end
+      rescue; end
+    end
 
-  def init_plugins
-    begin
-      @plugins = []
-      ::Csv2hash::Plugins.constants.each do |name|
-        @plugins << ::Csv2hash::Plugins.const_get(name).new(self)
-      end
-    rescue; end
-  end
+    def parse
+      load_data_source
 
-  def parse
-    load_data_source
+      definition.validate!
+      definition.default!
+      validate_structure!
+      validate_data!
 
-    definition.validate!
-    definition.default!
-    validate_structure!
-    validate_data!
-
-    Csv2hash::DataWrapper.new.tap do |response|
-      if valid?
-        fill!
-        response.data = data[:data]
-      else
-        response.valid = false
-        response.errors = csv_with_errors
-        notifier.notify response
+      Csv2hash::DataWrapper.new.tap do |response|
+        if valid?
+          fill!
+          response.data = data[:data]
+        else
+          response.valid = false
+          response.errors = csv_with_errors
+          notifier.notify response
+        end
       end
     end
-  end
 
-  def csv_with_errors
-    @csv_with_errors ||= begin
-      CsvArray.new.tap do |rows|
-        errors.each do |error|
-          rows << error.merge({ value: (data_source[error[:y]][error[:x]] rescue nil) })
-        end
-      end #.to_csv
+    def csv_with_errors
+      @csv_with_errors ||= begin
+        CsvArray.new.tap do |rows|
+          errors.each do |error|
+            rows << error.merge({ value: (data_source[error[:y]][error[:x]] rescue nil) })
+          end
+        end #.to_csv
+      end
     end
-  end
 
-  # protected
+    # protected
 
-  def data_source
-    @data_source ||= CSV.read self.file_path
-  end
-  alias_method :load_data_source, :data_source
+    def data_source
+      @data_source ||= begin
+        adapter_name = self.file_path_or_data.is_a?(String) ? :csv : :memory
+        adapter = Adapter::Base.create(adapter_name, self.file_path_or_data)
+        adapter.source
+      end
+    end
+    alias_method :load_data_source, :data_source
 
-  private
+    private
 
-  def dynamic_lib_loading type
-    case definition.type
-    when Csv2hash::Definition::MAPPING
-      self.extend Module.module_eval("Csv2hash::#{type}::Mapping")
-    when Csv2hash::Definition::COLLECTION
-      self.extend Module.module_eval("Csv2hash::#{type}::Collection")
+    def dynamic_lib_loading type
+      case definition.type
+      when Csv2hash::Definition::MAPPING
+        self.extend Module.module_eval("Csv2hash::#{type}::Mapping")
+      when Csv2hash::Definition::COLLECTION
+        self.extend Module.module_eval("Csv2hash::#{type}::Collection")
+      end
     end
   end
-
-end
\ No newline at end of file
+end