generate-puppetfile 0.9.0

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    ZmU2ODhjY2FlMTczMTAxMzIzY2ZmOWVlZmRkYzBjMTBjNGFiN2JmMA==
+  data.tar.gz: !binary |-
+    ZjhmMzUwOTM4Yzc1YTFmZjUxMjA2Njg2NGVkZWIwZDRiMjFjNGNlNA==
+SHA512:
+  metadata.gz: !binary |-
+    ZDE0Njg0ZmNhY2Q1MzU4MTI1MjUzZTgwM2NiYmQ1MDJiMjQ3YzUwZmJkNWIy
+    NGMxODA0M2Y3ZDJiNjllNDYxY2VkMzE2ZjQxZDFjZDVlYThjMDc0NmRlZTFi
+    YTY4ZWM3Y2IyNjkwYmVlZGIyYjM2NmU4YjcwMWJkZjU1NGIwMmQ=
+  data.tar.gz: !binary |-
+    YzM5MTUyZDNiOWQ4MDE3YzcyMjNhNWE0MTA3MmVkYmI4YmQ2NzExYjM2NzIx
+    NzdlMTU3M2MwOThlZmNiYzFkZmM0ZDAyY2IwNDE2ZDZkOWQzZDliNzM5NDE3
+    N2ZjOGE3MmE5NGJiYWFjZjBiZjk1NDgwNjdmZmRlOTUwZTUxMzk=

data/README.md

@@ -0,0 +1,22 @@
+Tired of searching for dependencies on the forge and hoping you got everything? Use generate-puppetfile to generate a list of modules and their dependencies in a Puppetfile format for use with r10k or librarian-puppet.
+
+To use this, simply clone the repository and run the generate-puppetfile script with a list of valid module names:
+
+```
+$ ./generate-puppetfile echocat/nfs rnelson0/certs
+Download echocat/nfs and its dependencies...
+Download rnelson0/certs and its dependencies...
+Module download complete. Generating Puppetfile.
+
+Here is your suggested Puppetfile:
+----------------------------------
+
+forge 'http://forge.puppetlabs.com'
+
+mod 'echocat/nfs', '1.7.1'
+mod 'herculesteam/augeasproviders_core', '2.1.2'
+mod 'herculesteam/augeasproviders_shellvar', '2.2.0'
+mod 'puppetlabs/concat', '1.2.4'
+mod 'puppetlabs/stdlib', '4.9.0'
+mod 'rnelson0/certs', '0.6.2'
+```

data/bin/generate-puppetfile

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+require 'generate_puppetfile'
+
+begin
+  GeneratePuppetfile::Bin.new(ARGV).run
+end

data/lib/generate_puppetfile.rb

@@ -0,0 +1,5 @@
+module GeneratePuppetfile
+  require 'generate_puppetfile/bin'
+  require 'generate_puppetfile/optparser'
+  require 'generate_puppetfile/version'
+end

data/lib/generate_puppetfile/bin.rb

@@ -0,0 +1,153 @@
+# encoding: utf-8
+require 'generate_puppetfile'
+require 'generate_puppetfile/optparser'
+require 'fileutils'
+
+module GeneratePuppetfile
+  class Bin
+    Module_regex = Regexp.new("mod ['\"]([a-z0-9_]+\/[a-z0-9_]+)['\"](, ['\"](\\d\.\\d\.\\d)['\"])?", Regexp::IGNORECASE)
+    @options = {}
+
+    def initialize(args)
+      @args = args
+    end
+
+    def run
+      @options = GeneratePuppetfile::OptParser.parse(@args)
+
+      helpmsg = "generate-puppetfile: try 'generate-puppetfile --help' for more information."
+
+      if @args[0].nil? && (! @options[:puppetfile])
+        puts "generate-puppetfile: No modules or existing Puppetfile specified."
+        puts helpmsg
+        return 1
+      end
+
+      forge_module_list = Array.new
+
+      if @args
+	puts "\nProcessing modules from the command line...\n\n" if @options[:debug]
+        cli_modules = Array.new
+        @args.each do |modulename|
+	  validate(modulename) && (cli_modules.push(modulename))
+        end
+      end
+
+      puppetfile_contents = Hash.new
+      extras = Array.new
+      if @options[:puppetfile]
+	puts "\nProcessing the puppetfile '#{@options[:puppetfile]}'...\n\n" if @options[:debug]
+	puppetfile_contents = read_puppetfile(@options[:puppetfile])
+	extras = puppetfile_contents[:extras]
+      end
+
+      forge_module_list.push(*cli_modules) if @args
+      forge_module_list.push(*puppetfile_contents[:modules]) if puppetfile_contents[:modules]
+
+      list_modules(forge_module_list) if puppetfile_contents && @options[:debug]
+      list_extras(puppetfile_contents[:extras]) if puppetfile_contents[:extras] && @options[:debug]
+
+      generate_puppetfile_contents(forge_module_list, puppetfile_contents[:extras])
+    end
+
+    def validate (modulename)
+      success = (modulename =~ /[a-z0-9_]\/[a-z0-9_]/i)
+      puts "    '#{modulename}' is not a valid module name. Skipping." unless success
+      success
+    end
+
+    def list_modules (module_list)
+      puts "\nListing discovered modules from CLI and/or Puppetfile:\n\n"
+      module_list.each do |name|
+	puts "    #{name}"
+      end
+      puts ""
+    end
+
+    def list_extras (extras)
+      puts "\nExtras found in the existing Puppetfile:\n\n"
+      extras.each do |line|
+	puts "    #{line}"
+      end
+      puts ""
+    end
+
+    def read_puppetfile (puppetfile)
+      puppetfile_contents = {
+	:modules => Array.new,
+	:extras  => Array.new,
+      }
+
+      File.foreach(puppetfile) do |line|
+	if Module_regex.match(line)
+	  name = $1
+	  print "    #{name} looks like a forge module.\n" if @options[:debug]
+	  puppetfile_contents[:modules].push(name)
+	else
+	  next if line =~ /^forge/
+	  next if line =~ /^\s+$/
+	  next if line =~ /# Discovered elements from existing Puppetfile/
+
+	  puppetfile_contents[:extras].push(line)
+	end
+      end
+
+      puppetfile_contents
+    end
+
+    def generate_puppetfile_contents (module_list, extras = [])
+      # cli_modules is a hash of module name => version
+      # puppetfile_contents is a hash with two keys:
+      #   module_list is an array of forge module names to be downloaded
+      #   extras is an array of strings
+      workspace = `mktemp -d`
+
+      modulepath = "--modulepath #{workspace.chomp} "
+      silence    = '>/dev/null 2>&1 '
+      strip_colors = 'sed -r "s/\x1B\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]//g"'
+
+      puts "\nInstalling modules. This may take a few minutes.\n\n"
+      module_list.each do |name|
+	command  = "puppet module install #{name} "
+	command += modulepath + silence
+	
+	system(command)
+      end
+
+      puppetfile_header = <<-EOF
+forge 'http://forge.puppetlabs.com'
+
+# Modules discovered by generate-puppetfile
+      EOF
+
+      puppetfile_body = `puppet module list #{modulepath} 2>/dev/null | #{strip_colors}`
+
+      puppetfile_body.gsub!(/^\/.*$/, '')
+      puppetfile_body.gsub!(/-/,    '/')
+      puppetfile_body.gsub!(/├── /, "mod '")
+      puppetfile_body.gsub!(/└── /, "mod '")
+      puppetfile_body.gsub!(/ \(v/, "', '")
+      puppetfile_body.gsub!(/\)$/,  "'")
+      puppetfile_body.gsub!(/^$\n/, '')
+
+      puppetfile_footer = "# Discovered elements from existing Puppetfile\n"
+      extras.each do |line|
+	puppetfile_footer += "#{line}"
+      end
+
+      puts <<-EOF
+
+
+Your Puppetfile has been generated. Copy and paste between the markers:
+
+=======================================================================
+#{puppetfile_header}
+#{puppetfile_body}
+#{puppetfile_footer}
+=======================================================================
+      EOF
+
+      FileUtils.rm_rf(workspace)
+    end
+  end
+end

data/lib/generate_puppetfile/optparser.rb

@@ -0,0 +1,33 @@
+require 'generate_puppetfile'
+require 'optparse'
+
+module GeneratePuppetfile
+  class OptParser
+    def self.parse(args)
+      options = {}
+      opts = OptionParser.new do |opts|
+        opts.banner = "generate-puppetfile [OPTIONS] [<MODULE> ... <MODULE>]"
+
+        opts.on('-p', '--puppetfile FILE', 'Name of existing Puppetfile to verify and update') do |file|
+          unless File.readable?(file)
+            puts "\nPuppetfile '#{file}' cannot be read. Are you sure you passed in the correct filename?\n\n"
+            exit 1
+          end
+
+	  options[:puppetfile] = file
+        end
+
+	opts.on('-d', '--debug', 'Enable debug logging') do
+	  options[:debug] = true
+	end
+
+	opts.on_tail('-v', '--version', 'Show version') do
+	  puts "generate-puppetfile v#{GeneratePuppetfile::VERSION}"
+	end
+      end
+     
+      opts.parse!(args)
+      options
+    end
+  end
+end

data/lib/generate_puppetfile/version.rb

@@ -0,0 +1,3 @@
+module GeneratePuppetfile
+  VERSION = '0.9.0'
+end

metadata

@@ -0,0 +1,51 @@
+--- !ruby/object:Gem::Specification
+name: generate-puppetfile
+version: !ruby/object:Gem::Version
+  version: 0.9.0
+platform: ruby
+authors:
+- Rob Nelson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-10-26 00:00:00.000000000 Z
+dependencies: []
+description: Generate a Puppetfile for use with r10k based on an existing file or
+  a list of modules.
+email: rnelson0@gmail.com
+executables:
+- generate-puppetfile
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- bin/generate-puppetfile
+- lib/generate_puppetfile.rb
+- lib/generate_puppetfile/bin.rb
+- lib/generate_puppetfile/optparser.rb
+- lib/generate_puppetfile/version.rb
+homepage: https://github.com/rnelson0/puppet-generate-puppetfile
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.3
+signing_key: 
+specification_version: 4
+summary: Generate a Puppetfile
+test_files: []