puppet-forge-mirror 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cc1e37ea1c172f05e43a6330616d5fb334135b05
-  data.tar.gz: 6e3e28361feb561f72b126af1cbb1666f810e97e
+  metadata.gz: 95ad37a72fd077978a6c8df5b4cfe729437c299e
+  data.tar.gz: e868c054f128679306ca5a50d58eb8b25d78e225
 SHA512:
-  metadata.gz: b8321c3692c06f251db3d4c0c4a4e8ac0c31ec78038d31bee9ea9b3926a8b9cb30476db6b4b073f8fec371ce4168fd9a0e518fabe95676766b62c051ca1d100d
-  data.tar.gz: 0ca262fbe7301c8f676cedaabdc5fe4e0927fbd2d0536889c0688d7487080efdf461e836a269ae57c12d3361dc82d26efd56d48148596e0ea8de91d8d74c5a06
+  metadata.gz: e8f6dcbdee362b072db3e8a627fc17d54e78d0cf06b2f720f800fb8338e1438f3bb9c213208ff823f58f94f527e62a2a28457541aa8c4d2f0e3633c4e71bf97e
+  data.tar.gz: 9585294c9c55e5fb17e50b897c316f96fba4a708931f3312e0c76544db2a2328220432a81589f8ce19eaeb33a9702ce8b90addef6bd068239a9466d4b5f731cc

data/.gemspec

@@ -1,7 +1,7 @@
 # -*- encoding: utf-8 -*-
 Gem::Specification.new do |s|
   s.name                  = 'puppet-forge-mirror'
-  s.version               = '0.0.2'
+  s.version               = '0.0.3'
   s.author                = 'Johan Haals'
   s.email                 = ['johan.haals@gmail.com']
   s.homepage              = 'https://github.com/jhaals/puppet-forge-mirror'

data/bin/puppet-forge-mirror

@@ -1,12 +1,25 @@
 #!/usr/bin/env ruby
 require 'puppet-forge-mirror'
+require 'optparse'
 
-if ARGV[0].nil?
-  puts "puppet-forge-mirror [module path]"
-  exit
-end
+options = {}
+OptionParser.new do |opts|
+  opts.banner = 'Usage: puppet-forge-mirror [options]'
+  opts.on('-m', '--modulepath PATH', 'Downloaded modules will be stored here') do |path|
+    options[:modulepath] = path
+  end
+  opts.on('--max-size MB', 'Maximum size of module in megabyte, bigger modules will be skipped') do |max_size|
+    options[:max_size] = max_size
+  end
+end.parse!
 
+if options[:modulepath].nil?
+  puts 'No modulepath specified'
+  exit 1
+end
+options[:max_size] = 2 if options[:max_size].nil?
 fm = PuppetForgeMirror.new(
   'https://forgeapi.puppetlabs.com',
-  ARGV[0])
+  options[:modulepath],
+  options[:max_size])
 fm.download_modules

data/lib/puppet-forge-mirror.rb

@@ -7,9 +7,10 @@ require 'uri'
 
 # Mirror /v3 forges
 class PuppetForgeMirror
-  def initialize(forge_url, modules_dir)
+  def initialize(forge_url, modules_dir, max_size)
     @modules_dir = modules_dir
     @forge_url = forge_url
+    @max_size = max_size
   end
 
   # Perform http GET request, return response.body if status code is 200
@@ -53,7 +54,7 @@ class PuppetForgeMirror
 
       r = JSON.parse(data)
       r['results'].each do |m|
-        if m['file_size'] > 1024 * 1024 * 1
+        if m['file_size'] > 1024 * 1024 * @max_size.to_i
           puts "#{m['metadata']['name']} is too big, skipping"
           next
         end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: puppet-forge-mirror
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Johan Haals
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-09 00:00:00.000000000 Z
+date: 2014-09-12 00:00:00.000000000 Z
 dependencies: []
 description: 
 email: