ari_fetch 0.0.3 → 0.0.4

data/Gemfile

@@ -2,4 +2,7 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in ari_fetch.gemspec
 gemspec
-gem 'rspec'
+gem 'rspec'
+
+# gem "activesupport"
+# gem "rails"

data/ari_fetch.gemspec

@@ -17,7 +17,6 @@ Gem::Specification.new do |gem|
   gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
   gem.require_paths = ["lib"]
 
-  # gem.add_dependency "activesupport" , "~> 3.0.7"
-  # gem.add_dependency "rails"         , "~> 3.0.7"
+  gem.add_dependency "nokogiri"
 
 end

data/lib/ari_fetch.rb

@@ -1,11 +1,65 @@
 require "ari_fetch/version"
 
 require "active_support/dependencies"
+require "active_record"
+
+require "net/ftp"
 
 module AriFetch
   class Engine
-    def initialize(*args)
-      # do something
+
+    attr_accessor :read_files, :unread_files, :ftp_files, :ftp_url, :username,
+      :password, :ftp_instance, :options
+
+    def initialize(ftp_url, username, password, options = {})
+      @ftp_url, @username, @password, @options = ftp_url, username, password, options
+    end
+
+    def read_files
+      @read_files = AriFetch::AriReadFile.pluck(:name) rescue @read_files = AriFetch::Files.new([]).files
+    end
+
+    def ftp_files
+      @ftp_files = fetch! rescue @ftp_files = AriFetch::Files.new([]).files
+    end
+
+    def unread_files
+      @unread_files = ftp_files - read_files
+    end
+
+    def fetch_vehicles_from_files(how_many=1)
+      unread_files.first(how_many).map {|file| AriFetch::AriReadFile.create(name: file).fetch_data(ftp_instance, include_cancel?) }.flatten!
+    end
+
+    private
+
+    def fetch!
+      login
+      AriFetch::Files.new(ftp_instance.list()).filter!.sort!(reverse?)
+    end
+
+    def login
+      ftp_instance.connect(ftp_url, 21)
+      ftp_instance.login(username, password)
+      ftp_instance.passive = true
+      ftp_instance.chdir('/')
+    end
+
+    def ftp_instance
+      @ftp_instance ||= Net::FTP.new
     end
+
+    def include_cancel?
+      options[:cancel] || false
+    end
+
+    def reverse?
+      options[:reverse_order] || false
+    end
+
   end
-end
+end
+
+require "ari_fetch/ari_read_file.rb"
+require "ari_fetch/files.rb"
+require "ari_fetch/vehicle.rb"

data/lib/ari_fetch/ari_read_file.rb

@@ -0,0 +1,40 @@
+require "nokogiri"
+
+module AriFetch
+  class AriReadFile < ActiveRecord::Base
+    attr_accessible :name
+
+    attr_accessor :content, :vehicles
+
+    def fetch_data(ftp_instance, include_cancel)
+      ftp_instance.gettextfile(name)
+      @content = parse!(File.open(name, "r"))
+      FileUtils.rm name
+      filter_cancel_vehicles! unless include_cancel
+      result
+    end
+
+    private
+
+    def parse!(xml)
+      Hash.from_xml(::Nokogiri.XML(xml).to_s) rescue {}
+    end
+
+    def vehicles_hash
+      content["VEHICLES"]["VEHICLE"] rescue []
+    end
+
+    def vehicles
+      @vehicles ||= vehicles_hash.map {|vehicle| AriFetch::Vehicle.new(vehicle) }
+    end
+
+    def filter_cancel_vehicles!
+      @vehicles = vehicles.select { |vehicle| vehicle.available? }
+    end
+
+    def result
+      vehicles.map { |vehicle| vehicle.result }
+    end
+
+  end
+end

data/lib/ari_fetch/files.rb

@@ -0,0 +1,36 @@
+module AriFetch
+  class Files
+
+    attr_accessor :files
+
+    def initialize(files = [])
+      @files = files
+    end
+
+    def filter!
+      map_to_file_name.select{ |e| !useless_files.include?(e) && e.match(/\.xml\Z/) }
+      self
+    end
+
+    # reverse implies latest date will come first
+    def sort!(reverse=false)
+      return files.sort {|f1, f2| file_date(f1) <=> file_date(f2) } unless reverse
+      files.reverse
+    end
+
+    private
+
+    def map_to_file_name
+      @files = files.map { |e| e.split(/\s+/).last.to_s }
+    end
+
+    def useless_files
+      [".", ".."]
+    end
+
+    def file_date(file)
+      DateTime.strptime(file.to_s, "VehicleInformation%m_%d_%Y_%H_%M_%S_%p.xml") rescue DateTime.new
+    end
+
+  end
+end

data/lib/ari_fetch/vehicle.rb

@@ -0,0 +1,37 @@
+module AriFetch
+
+  class Vehicle
+
+    attr_accessor :vehicle
+
+    def initialize(vehicle)
+      @vehicle = vehicle
+    end
+
+    def result
+      OpenStruct.new(vehicle_data_with_downcased_keys)
+    end
+
+    def available?
+      ari_action == "AVAILABLE"
+    end
+
+    private
+
+    def vehicle_data_with_downcased_keys
+      downcased_hash = {}
+      vehicle.map{|k, v| downcased_hash.merge!({"#{k.downcase}" => v}) }
+      downcased_hash
+    end
+
+    def vin_number
+      vehicle["VIN"]
+    end
+
+    def ari_action
+      vehicle["ACTION"]
+    end
+
+  end
+
+end

data/lib/ari_fetch/version.rb

@@ -1,3 +1,3 @@
 module AriFetch
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/generators/ari_fetch/ari_fetch_generator.rb

@@ -17,6 +17,6 @@ class AriFetchGenerator < Rails::Generators::Base
   end
 
   def create_migration_file
-    migration_template 'migration.rb', 'db/migrate/create_ari_files_table.rb'
+    migration_template 'migration.rb', 'db/migrate/create_ari_read_files.rb'
   end
 end

data/lib/generators/ari_fetch/templates/migration.rb

@@ -1,6 +1,6 @@
-class CreateAriFiles < ActiveRecord::Migration
+class CreateAriReadFiles < ActiveRecord::Migration
   def self.up
-    create_table :ari_files do |t|
+    create_table :ari_read_files do |t|
       t.string :name
 
       t.timestamps
@@ -8,6 +8,6 @@ class CreateAriFiles < ActiveRecord::Migration
   end
 
   def self.down
-    drop_table :ari_files
+    drop_table :ari_read_files
   end
 end

data/spec/ari_spec.rb

@@ -2,6 +2,28 @@ require 'spec_helper'
 
 describe AriFetch::Engine do
 
-  pending "write it"
+  let!(:engine) { AriFetch::Engine.new("184.168.187.1", "arixml", "ARIuser1!") }
+
+  context "initialize ari fetch" do
+    subject { engine }
+
+    context "fetch files from ftp" do
+      let!(:ftp_files) { engine.ftp_files }
+      subject { ftp_files }
+      its(:count) { should be > 0 }
+    end
+
+    context "read and unread files" do
+      its(:read_files){ should == [] }
+      let!(:ftp_files) { engine.ftp_files }
+      its(:unread_files) { should == ftp_files }
+    end
+
+  end
+
+  context "reading files and fetching data" do
+    subject { engine }
+    its(:read_files_from_start) { should == "Array" }
+  end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ari_fetch
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,8 +9,24 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-06-03 00:00:00.000000000 Z
-dependencies: []
+date: 2013-06-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: ARI
 email:
 - vivek@devbrother.com
@@ -23,9 +39,11 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
-- app/models/ari_file.rb
 - ari_fetch.gemspec
 - lib/ari_fetch.rb
+- lib/ari_fetch/ari_read_file.rb
+- lib/ari_fetch/files.rb
+- lib/ari_fetch/vehicle.rb
 - lib/ari_fetch/version.rb
 - lib/generators/ari_fetch/ari_fetch_generator.rb
 - lib/generators/ari_fetch/templates/migration.rb

data/app/models/ari_file.rb

@@ -1,5 +0,0 @@
-module AriFetch
-  class AriFile < ActiveRecord::Base
-    attr_accessible :name
-  end
-end