nab 0.0.6 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e79f420590fcb47bb0b202a2b35b0ecc6f7cb27f
-  data.tar.gz: e6151e9936b15d208b729b123928f6e4f0ec696c
+  metadata.gz: 276ebcbc2b437f6832f7d40f63086cfb1beb1255
+  data.tar.gz: 2aa66966bec0c0f537e096a723187f282d963552
 SHA512:
-  metadata.gz: 5fcc55e128dea5ac025d3b9751c2514822c31c6a98732ce865513c061eaaff12ea739a4c5fc30d8450c2aed5ba3de4188530e597e6037b33e2f6997e2fca67cb
-  data.tar.gz: 7f50dde65fe78159e939f0fe3b28a02594386e32b6b46b80a9517a452a50e80a80bb5fcd861e2b9b3a75c16f9ec1372b93acfa4fd31977ed08c11c64e3a6d510
+  metadata.gz: 3da7c427f07ea594ad5fef284f2368842de4246b91c1551e9d3bff72837c1b53d77f91fe319ed055d5838f04610954a70c7aa5574ef9e65d6ea09e2ee9bbd285
+  data.tar.gz: 293bf411ad3fa723c8c05888b470f86e2af82bdc3daf1a6641f9fb625d7ef61e701e668917f1147e412be989071f628b70677edf13a69afc6c56c62013e1bddc

data/bin/nab

@@ -10,14 +10,31 @@ FATAL = Logger::FATAL
 INFO  = Logger::INFO
 
 # Default option values
-options             = OpenStruct.new
-options.log_level   = INFO
+options                 = OpenStruct.new
+options.log_level       = INFO
+options.destination_dir = '.'
+options.write_adapter   = Nab::FileWriter
 
 OptionParser.new do |o|
-  o.on('-l', '--log-level [LEVEL]', String, 'specify the log level (default is INFO)') do |l|
+  o.on('-l', '--log-level [LEVEL]', String,
+       'specify the log level (default is INFO)') do |l|
     options.log_level = Kernel.const_get l.upcase
   end
 
+  o.on('-d', '--destination-dir [DIR]', String,
+       'specify the directory to download to (default is .)') do |d|
+    options.destination_dir = d
+  end
+
+  o.on('-w', '--write-adapter [STRING]', String,
+       'specify the write adapter nab should use (default is File)') do |w|
+    options.write_adapter = Nab.const_get("#{w.capitalize}Writer")
+  end
+
+  o.on('-s', '--strip-dirs', 'Strip all leading directories') do |s|
+    options.strip_dirs = s
+  end
+
   o.on('-h') { Nab::Log.info o; exit }
 
   begin

data/changelog.md

@@ -1,6 +1,13 @@
 Nab Changelog
 ===
 
+1.0.0
+---
+  - Adds ability to change destination directory (where files go)
+  - Adds the ability to strip leading directories when saving files locally
+  - Major rewrite of how files are writter (file adapters)
+  - Adds storage adapters for STDERR and STDOUT
+
 0.0.5
 ---
   - Fixes bug because of missing require for the new aws-sdk-v1 gem.

data/lib/nab.rb

@@ -13,3 +13,6 @@ require_relative 'nab/client.rb'
 require_relative 'nab/logger.rb'
 require_relative 'nab/version.rb'
 require_relative 'nab/adapters/s3.rb'
+require_relative 'nab/writers/file_writer.rb'
+require_relative 'nab/writers/stdout_writer.rb'
+require_relative 'nab/writers/stderr_writer.rb'

data/lib/nab/adapter_manager.rb

@@ -2,18 +2,27 @@
 
 module Nab
   class AdapterManager
+    attr_accessor :write_adapter, :strip_dirs
 
     def initialize
       @adapters = Hash.new
+      @write_adapter = FileWriter
+      @strip_dirs = nil
     end
 
     def get(type)
       if @adapters.key? type
+        Nab::Log.debug "Retrieving cached adapter #{@adapters[type]}"
         @adapters[type]
       else
-        Log.debug "Loading adapter for #{type}"
         klass = Nab.const_get "#{type}Adapter"
+        Nab::Log.debug "Creating new #{type}Adapter"
         @adapters[type] = klass.new
+        Nab::Log.debug "Setting write_adapter to #{write_adapter} for #{type}"
+        @adapters[type].write_adapter = write_adapter
+        Nab::Log.debug "Setting strip_dirs to #{strip_dirs} for #{type}"
+        @adapters[type].strip_dirs = strip_dirs
+        @adapters[type]
       end
     end
   end

data/lib/nab/adapters/s3.rb

@@ -2,36 +2,27 @@
 
 module Nab
   class S3Adapter
-    attr_accessor :base_path
+    attr_accessor :write_adapter, :strip_dirs
 
-    def initialize(base_path = ".")
+    def initialize
       AWS.config
-      @base_path = "."
+      @write_adapter = FileWriter
+      @strip_dirs = false
     end
 
-    def get(uri)
+    def get(uri,destination)
       bucket_name = uri.host
       asset_name  = uri.path.gsub(/^\//, "") # remove leading slashes
-      name        = File.basename asset_name
-      remote_path = File.dirname asset_name
-      local_path  = File.join @base_path,remote_path
+      s3          = AWS::S3.new
+      bucket      = s3.buckets[bucket_name]
+      asset       = bucket.objects[asset_name]
 
-      Log.debug "Bucket: #{bucket_name}"
-      Log.debug "Asset name: #{asset_name}"
-      Log.debug "Local path: #{local_path}"
+      Nab::Log.info "Retrieving #{File.join(bucket_name,asset_name)}"
+      asset_name  = File.basename(asset_name) if strip_dirs
+      store       = write_adapter.new(destination, asset_name)
 
-      FileUtils.mkdir_p local_path
-
-      s3     = AWS::S3.new
-      bucket = s3.buckets[bucket_name]
-      asset  = bucket.objects[asset_name]
-
-      Log.info "Downloading #{File.join(bucket_name,asset_name)} to #{File.join(local_path,name)}"
-      File.open(File.join(local_path,name), 'wb') do |file|
-        asset.read do |chunk|
-          file.write chunk
-        end
-      end
+      asset.read { |chunk| store.write chunk }
+      store.close
     end
   end
 end

data/lib/nab/client.rb

@@ -3,15 +3,24 @@
 module Nab
   class Client
     def run(options)
+      Log.level options.log_level
 
-    Log.level options.log_level
+      Nab::Log.debug "Options received: #{options.inspect}"
 
-    my_manager = AdapterManager.new
-    ARGV.each do |a|
-      uri = URI(a)
-      adapter = my_manager.get(uri.scheme.capitalize)
-      adapter.get(uri)
-    end
+      if ARGV.size < 1
+        Nab::Log.fatal "No arguments specified"
+        exit 1
+      end
+
+      manager = AdapterManager.new
+      manager.write_adapter = options.write_adapter
+      manager.strip_dirs = options.strip_dirs
+
+      ARGV.each do |a|
+        uri = URI(a)
+        adapter = manager.get uri.scheme.capitalize
+        adapter.get uri,options.destination_dir
+      end
     end
   end
 end

data/lib/nab/version.rb

@@ -1,5 +1,5 @@
 # encoding: UTF-8
 
 module Nab
-  VERSION = '0.0.6'
+  VERSION = '1.0.0'
 end

data/lib/nab/writers/file_writer.rb

@@ -0,0 +1,21 @@
+# encoding: UTF-8
+
+module Nab
+  class FileWriter
+    def initialize(destination, asset_name)
+      full_path = File.join(destination,asset_name)
+      FileUtils.mkdir_p File.dirname(full_path)
+
+      Nab::Log.info "Writing to #{full_path}"
+      @file = File.new(full_path,"wb")
+    end
+
+    def write data
+      @file.write data
+    end
+
+    def close
+      @file.close
+    end
+  end
+end

data/lib/nab/writers/stderr_writer.rb

@@ -0,0 +1,18 @@
+# encoding: UTF-8
+
+module Nab
+  class StderrWriter
+    def initialize(destination, asset_name)
+      Nab::Log.info "Writing to STDERR"
+    end
+
+    def write data
+      STDERR.sync = true
+      STDERR.print data
+      STDERR.sync = false
+    end
+
+    def close
+    end
+  end
+end

data/lib/nab/writers/stdout_writer.rb

@@ -0,0 +1,18 @@
+# encoding: UTF-8
+
+module Nab
+  class StdoutWriter
+    def initialize(destination, asset_name)
+      Nab::Log.info "Writing to STDOUT"
+    end
+
+    def write data
+      STDOUT.sync = true
+      print data
+      STDOUT.sync = false
+    end
+
+    def close
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nab
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 1.0.0
 platform: ruby
 authors:
 - Brian Bianco
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-02-09 00:00:00.000000000 Z
+date: 2015-02-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-v1
@@ -56,6 +56,9 @@ files:
 - lib/nab/client.rb
 - lib/nab/logger.rb
 - lib/nab/version.rb
+- lib/nab/writers/file_writer.rb
+- lib/nab/writers/stderr_writer.rb
+- lib/nab/writers/stdout_writer.rb
 - nab.gemspec
 - readme.md
 homepage: http://www.github.com/brianbianco/nab