puppet-parse 0.0.1 → 0.0.2

data/lib/vendor/puppet/application/queue.rb

@@ -0,0 +1,161 @@
+require 'puppet/application'
+require 'puppet/util'
+
+class Puppet::Application::Queue < Puppet::Application
+  should_parse_config
+
+  attr_accessor :daemon
+
+  def preinit
+    require 'puppet/daemon'
+    @daemon = Puppet::Daemon.new
+    @daemon.argv = ARGV.dup
+
+    # Do an initial trap, so that cancels don't get a stack trace.
+
+    # This exits with exit code 1
+    Signal.trap(:INT) do
+      $stderr.puts "Caught SIGINT; shutting down"
+      exit(1)
+    end
+
+    # This is a normal shutdown, so code 0
+    Signal.trap(:TERM) do
+      $stderr.puts "Caught SIGTERM; shutting down"
+      exit(0)
+    end
+
+    {
+      :verbose => false,
+      :debug => false
+    }.each do |opt,val|
+      options[opt] = val
+    end
+  end
+
+  option("--debug","-d")
+  option("--verbose","-v")
+
+  def help
+    <<-HELP
+
+puppet-queue(8) -- Queuing daemon for asynchronous storeconfigs
+========
+
+SYNOPSIS
+--------
+Retrieves serialized storeconfigs records from a queue and processes
+them in order.
+
+
+USAGE
+-----
+puppet queue [-d|--debug] [-v|--verbose]
+
+
+DESCRIPTION
+-----------
+This application runs as a daemon and processes storeconfigs data,
+retrieving the data from a stomp server message queue and writing it to
+a database.
+
+For more information, including instructions for properly setting up
+your puppet master and message queue, see the documentation on setting
+up asynchronous storeconfigs at:
+http://projects.puppetlabs.com/projects/1/wiki/Using_Stored_Configuration
+
+
+OPTIONS
+-------
+Note that any configuration parameter that's valid in the configuration
+file is also a valid long argument. For example, 'server' is a valid
+configuration parameter, so you can specify '--server <servername>' as
+an argument.
+
+See the configuration file documentation at
+http://docs.puppetlabs.com/references/stable/configuration.html for the
+full list of acceptable parameters. A commented list of all
+configuration options can also be generated by running puppet queue with
+'--genconfig'.
+
+* --debug:
+  Enable full debugging.
+
+* --help:
+  Print this help message
+
+* --verbose:
+  Turn on verbose reporting.
+
+* --version:
+  Print the puppet version number and exit.
+
+
+EXAMPLE
+-------
+    $ puppet queue
+
+
+AUTHOR
+------
+Luke Kanies
+
+
+COPYRIGHT
+---------
+Copyright (c) 2011 Puppet Labs, LLC Licensed under the Apache 2.0 License
+
+    HELP
+  end
+
+  option("--logdest DEST", "-l DEST") do |arg|
+    begin
+      Puppet::Util::Log.newdestination(arg)
+      options[:setdest] = true
+    rescue => detail
+      puts detail.backtrace if Puppet[:debug]
+      $stderr.puts detail.to_s
+    end
+  end
+
+  def main
+    require 'puppet/indirector/catalog/queue' # provides Puppet::Indirector::Queue.subscribe
+    Puppet.notice "Starting puppetqd #{Puppet.version}"
+    Puppet::Resource::Catalog::Queue.subscribe do |catalog|
+      # Once you have a Puppet::Resource::Catalog instance, passing it to save should suffice
+      # to put it through to the database via its active_record indirector (which is determined
+      # by the terminus_class = :active_record setting above)
+      Puppet::Util.benchmark(:notice, "Processing queued catalog for #{catalog.name}") do
+        begin
+          Puppet::Resource::Catalog.indirection.save(catalog)
+        rescue => detail
+          puts detail.backtrace if Puppet[:trace]
+          Puppet.err "Could not save queued catalog for #{catalog.name}: #{detail}"
+        end
+      end
+    end
+
+    Thread.list.each { |thread| thread.join }
+  end
+
+  def setup
+    unless Puppet.features.stomp?
+      raise ArgumentError, "Could not load the 'stomp' library, which must be present for queueing to work.  You must install the required library."
+    end
+
+    setup_logs
+
+    exit(Puppet.settings.print_configs ? 0 : 1) if Puppet.settings.print_configs?
+
+    require 'puppet/resource/catalog'
+    Puppet::Resource::Catalog.indirection.terminus_class = :store_configs
+
+    daemon.daemonize if Puppet[:daemonize]
+
+    # We want to make sure that we don't have a cache
+    # class set up, because if storeconfigs is enabled,
+    # we'll get a loop of continually caching the catalog
+    # for storage again.
+    Puppet::Resource::Catalog.indirection.cache_class = nil
+  end
+end

data/lib/vendor/puppet/application/report.rb

@@ -0,0 +1,4 @@
+require 'puppet/application/indirection_base'
+
+class Puppet::Application::Report < Puppet::Application::IndirectionBase
+end

data/lib/vendor/puppet/application/resource.rb

@@ -0,0 +1,237 @@
+require 'puppet/application'
+
+class Puppet::Application::Resource < Puppet::Application
+
+  should_not_parse_config
+
+  attr_accessor :host, :extra_params
+
+  def preinit
+    @extra_params = []
+    Facter.loadfacts
+  end
+
+  option("--debug","-d")
+  option("--verbose","-v")
+  option("--edit","-e")
+
+  option("--host HOST","-H") do |arg|
+    @host = arg
+  end
+
+  option("--types", "-t") do |arg|
+    types = []
+    Puppet::Type.loadall
+    Puppet::Type.eachtype do |t|
+      next if t.name == :component
+      types << t.name.to_s
+    end
+    puts types.sort
+    exit
+  end
+
+  option("--param PARAM", "-p") do |arg|
+    @extra_params << arg.to_sym
+  end
+
+  def help
+    <<-'HELP'
+
+puppet-resource(8) -- The resource abstraction layer shell
+========
+
+SYNOPSIS
+--------
+Uses the Puppet RAL to directly interact with the system.
+
+
+USAGE
+-----
+puppet resource [-h|--help] [-d|--debug] [-v|--verbose] [-e|--edit]
+  [-H|--host <host>] [-p|--param <parameter>] [-t|--types] <type>
+  [<name>] [<attribute>=<value> ...]
+
+
+DESCRIPTION
+-----------
+This command provides simple facilities for converting current system
+state into Puppet code, along with some ability to modify the current
+state using Puppet's RAL.
+
+By default, you must at least provide a type to list, in which case
+puppet resource will tell you everything it knows about all resources of
+that type. You can optionally specify an instance name, and puppet
+resource will only describe that single instance.
+
+If given a type, a name, and a series of <attribute>=<value> pairs,
+puppet resource will modify the state of the specified resource.
+Alternately, if given a type, a name, and the '--edit' flag, puppet
+resource will write its output to a file, open that file in an editor,
+and then apply the saved file as a Puppet transaction.
+
+
+OPTIONS
+-------
+Note that any configuration parameter that's valid in the configuration
+file is also a valid long argument. For example, 'ssldir' is a valid
+configuration parameter, so you can specify '--ssldir <directory>' as an
+argument.
+
+See the configuration file documentation at
+http://docs.puppetlabs.com/references/stable/configuration.html for the
+full list of acceptable parameters. A commented list of all
+configuration options can also be generated by running puppet with
+'--genconfig'.
+
+* --debug:
+  Enable full debugging.
+
+* --edit:
+  Write the results of the query to a file, open the file in an editor,
+  and read the file back in as an executable Puppet manifest.
+
+* --host:
+  When specified, connect to the resource server on the named host
+  and retrieve the list of resouces of the type specified.
+
+* --help:
+  Print this help message.
+
+* --param:
+  Add more parameters to be outputted from queries.
+
+* --types:
+  List all available types.
+
+* --verbose:
+  Print extra information.
+
+
+EXAMPLE
+-------
+This example uses `puppet resource` to return a Puppet configuration for
+the user `luke`:
+
+    $ puppet resource user luke
+    user { 'luke':
+     home => '/home/luke',
+     uid => '100',
+     ensure => 'present',
+     comment => 'Luke Kanies,,,',
+     gid => '1000',
+     shell => '/bin/bash',
+     groups => ['sysadmin','audio','video','puppet']
+    }
+
+
+AUTHOR
+------
+Luke Kanies
+
+
+COPYRIGHT
+---------
+Copyright (c) 2011 Puppet Labs, LLC Licensed under the Apache 2.0 License
+
+    HELP
+  end
+
+  def main
+    type, name, params = parse_args(command_line.args)
+
+    raise "You cannot edit a remote host" if options[:edit] and @host
+
+    resources = find_or_save_resources(type, name, params)
+    text = resources.
+      map { |resource| resource.prune_parameters(:parameters_to_include => @extra_params).to_manifest }.
+      join("\n")
+
+    options[:edit] ?
+      handle_editing(text) :
+      (puts text)
+  end
+
+  def setup
+    Puppet::Util::Log.newdestination(:console)
+
+    Puppet.parse_config
+
+    if options[:debug]
+      Puppet::Util::Log.level = :debug
+    elsif options[:verbose]
+      Puppet::Util::Log.level = :info
+    end
+  end
+
+  private
+
+  def remote_key(type, name)
+    Puppet::Resource.indirection.terminus_class = :rest
+    port = Puppet[:puppetport]
+    ["https://#{@host}:#{port}", "production", "resources", type, name].join('/')
+  end
+
+  def local_key(type, name)
+    [type, name].join('/')
+  end
+
+  def handle_editing(text)
+    require 'tempfile'
+    # Prefer the current directory, which is more likely to be secure
+    # and, in the case of interactive use, accessible to the user.
+    tmpfile = Tempfile.new('x2puppet', Dir.pwd)
+    begin
+      # sync write, so nothing buffers before we invoke the editor.
+      tmpfile.sync = true
+      tmpfile.puts text
+
+      # edit the content
+      system(ENV["EDITOR"] || 'vi', tmpfile.path)
+
+      # ...and, now, pass that file to puppet to apply.  Because
+      # many editors rename or replace the original file we need to
+      # feed the pathname, not the file content itself, to puppet.
+      system('puppet apply -v ' + tmpfile.path)
+    ensure
+      # The temporary file will be safely removed.
+      tmpfile.close(true)
+    end
+  end
+
+  def parse_args(args)
+    type = args.shift or raise "You must specify the type to display"
+    Puppet::Type.type(type) or raise "Could not find type #{type}"
+    name = args.shift
+    params = {}
+    args.each do |setting|
+      if setting =~ /^(\w+)=(.+)$/
+        params[$1] = $2
+      else
+        raise "Invalid parameter setting #{setting}"
+      end
+    end
+
+    [type, name, params]
+  end
+
+  def find_or_save_resources(type, name, params)
+    key = @host ? remote_key(type, name) : local_key(type, name)
+
+    if name
+      if params.empty?
+        [ Puppet::Resource.indirection.find( key ) ]
+      else
+        resource = Puppet::Resource.new( type, name, :parameters => params )
+
+        # save returns [resource that was saved, transaction log from applying the resource]
+        save_result = Puppet::Resource.indirection.save(resource, key)
+        [ save_result.first ]
+      end
+    else
+      if type == "file"
+        raise "Listing all file instances is not supported.  Please specify a file or directory, e.g. puppet resource file /etc"
+      end
+      Puppet::Resource.indirection.search( key, {} )
+    end
+  end
+end

data/lib/vendor/puppet/application/resource_type.rb

@@ -0,0 +1,4 @@
+require 'puppet/application/indirection_base'
+
+class Puppet::Application::Resource_type < Puppet::Application::IndirectionBase
+end

data/lib/vendor/puppet/application/secret_agent.rb

@@ -0,0 +1,6 @@
+require 'puppet/application/face_base'
+require 'puppet/face'
+
+class Puppet::Application::Secret_agent < Puppet::Application::FaceBase
+  run_mode :agent
+end

data/lib/vendor/puppet/application/status.rb

@@ -0,0 +1,4 @@
+require 'puppet/application/indirection_base'
+
+class Puppet::Application::Status < Puppet::Application::IndirectionBase
+end