Module: Shef::Extensions

Defined in:
lib/chef/shef/ext.rb

Defined Under Namespace

Modules: FalseClass, ObjectCoreExtensions, String, Symbol, TrueClass Classes: Help

Constant Summary collapse

ObjectUIExtensions =

Methods that have associated help text need to be dynamically added to the main irb objects, so we define them in a proc and later instance_eval the proc in the object.

Proc.new do
  extend Shef::Extensions::ObjectCoreExtensions

  desc "prints this help message"
  explain(<<-E)
## SUMMARY ##
  When called with no argument, +help+ prints a table of all shef commands. When
  called with an argument COMMAND, +help+ prints a detailed explanation of the
  command if available, or the description if no explanation is available.
E
  def help(commmand=nil)
    if commmand
      explain_command(commmand)
    else
      puts help_banner
    end
    :ucanhaz_halp
  end
  alias :halp :help

  desc "prints information about chef"
  def version
    puts  "This is shef, the Chef shell.\n" +
          " Chef Version: #{::Chef::VERSION}\n" +
          " http://www.opscode.com/chef\n" +
          " http://wiki.opscode.com/display/chef/Home"
    :ucanhaz_automation
  end
  alias :shef :version

  desc "switch to recipe mode"
  def recipe
    find_or_create_session_for Shef.session.recipe
    :recipe
  end

  desc "switch to attributes mode"
  def attributes
    find_or_create_session_for Shef.session.node
    :attributes
  end

  desc "run chef using the current recipe"
  def run_chef
    Chef::Log.level = :debug
    session = Shef.session
    runrun = Chef::Runner.new(session.run_context).converge
    Chef::Log.level = :info
    runrun
  end

  desc "returns an object to control a paused chef run"
  subcommands :resume       => "resume the chef run",
              :step         => "run only the next resource",
              :skip_back    => "move back in the run list",
              :skip_forward => "move forward in the run list"
  def chef_run
    Shef.session.resource_collection.iterator
  end

  desc "resets the current recipe"
  def reset
    Shef.session.reset!
  end

  desc "assume the identity of another node."
  def become_node(node_name)
    Shef::DoppelGangerSession.instance.assume_identity(node_name)
    :doppelganger
  end
  alias :doppelganger :become_node

  desc "turns printout of return values on or off"
  def echo(on_or_off)
    conf.echo = on_or_off.on_off_to_bool
  end

  desc "says if echo is on or off"
  def echo?
    puts "echo is #{conf.echo.to_on_off_str}"
  end

  desc "turns on or off tracing of execution. *verbose*"
  def tracing(on_or_off)
    conf.use_tracer = on_or_off.on_off_to_bool
    tracing?
  end
  alias :trace :tracing

  desc "says if tracing is on or off"
  def tracing?
    puts "tracing is #{conf.use_tracer.to_on_off_str}"
  end
  alias :trace? :tracing?

  desc "simple ls style command"
  def ls(directory)
    Dir.entries(directory)
  end
end
MainContextExtensions =
Proc.new do
  desc "returns the current node (i.e., this host)"
  def node
    Shef.session.node
  end

  desc "pretty print the node's attributes"
  def ohai(key=nil)
    pp(key ? node.attribute[key] : node.attribute)
  end
end
RESTApiExtensions =
Proc.new do
  desc "edit an object in your EDITOR"
  explain(<<-E)
## SUMMARY ##
  +edit(object)+ allows you to edit any object that can be converted to JSON.
  When finished editing, this method will return the edited object:

  new_node = edit(existing_node)

## EDITOR SELECTION ##
  Shef looks for an editor using the following logic
  1. Looks for an EDITOR set by Shef.editor = "EDITOR"
  2. Looks for an EDITOR configured in your shef config file
  3. Uses the value of the EDITOR environment variable
E
  def edit(object)
    unless Shef.editor
      puts "Please set your editor with Shef.editor = \"vim|emacs|mate|ed\""
      return :failburger
    end

    filename = "shef-edit-#{object.class.name}-"
    if object.respond_to?(:name)
      filename += object.name
    elsif object.respond_to?(:id)
      filename += object.id
    end

    edited_data = Tempfile.open([filename, ".js"]) do |tempfile|
      tempfile.sync = true
      tempfile.puts Chef::JSONCompat.to_json(object)
      system("#{Shef.editor.to_s} #{tempfile.path}")
      tempfile.rewind
      tempfile.read
    end

    Chef::JSONCompat.from_json(edited_data)
  end

  desc "Find and edit API clients"
  explain(<<-E)
## SUMMARY ##
  +clients+ allows you to query you chef server for information about your api
  clients.

## LIST ALL CLIENTS ##
  To see all clients on the system, use

  clients.all #=> [<Chef::ApiClient...>, ...]

  If the output from all is too verbose, or you're only interested in a specific
  value from each of the objects, you can give a code block to +all+:

  clients.all { |client| client.name } #=> [CLIENT1_NAME, CLIENT2_NAME, ...]

## SHOW ONE CLIENT ##
  To see a specific client, use

  clients.show(CLIENT_NAME)

## SEARCH FOR CLIENTS ##
  You can also search for clients using +find+ or +search+. You can use the
  familiar string search syntax:

  clients.search("KEY:VALUE")

  Just as the +all+ subcommand, the +search+ subcommand can use a code block to
  filter or transform the information returned from the search:

  clients.search("KEY:VALUE") { |c| c.name }

  You can also use a Hash based syntax, multiple search conditions will be 
  joined with AND.

  clients.find :KEY => :VALUE, :KEY2 => :VALUE2, ...

## BULK-EDIT CLIENTS ##
                **BE CAREFUL, THIS IS DESTRUCTIVE**
  You can bulk edit API Clients using the +transform+ subcommand, which requires
  a code block. Each client will be saved after the code block is run. If the
  code block returns +nil+ or +false+, that client will be skipped:

  clients.transform("*:*") do |client|
    if client.name =~ /borat/i
      client.admin(false)
      true
    else
      nil
    end
  end

  This will strip the admin privileges from any client named after borat.
E
  subcommands :all        => "list all api clients",
              :show       => "load an api client by name",
              :search     => "search for API clients",
              :transform  => "edit all api clients via a code block and save them"
  def clients
    @clients ||= Shef::ModelWrapper.new(Chef::ApiClient, :client)
  end

  desc "Find and edit cookbooks"
  subcommands :all        => "list all cookbooks",
              :show       => "load a cookbook by name",
              :transform  => "edit all cookbooks via a code block and save them"
  def cookbooks
    @cookbooks ||= Shef::ModelWrapper.new(Chef::CookbookVersion)
  end

  desc "Find and edit nodes via the API"
  explain(<<-E)
## SUMMARY ##
  +nodes+ Allows you to query your chef server for information about your nodes.

## LIST ALL NODES ##
  You can list all nodes using +all+ or +list+

  nodes.all #=> [<Chef::Node...>, <Chef::Node...>, ...]

  To limit the information returned for each node, pass a code block to the +all+
  subcommand:

  nodes.all { |node| node.name } #=> [NODE1_NAME, NODE2_NAME, ...]

## SHOW ONE NODE ##
  You can show the data for a single node using the +show+ subcommand:

  nodes.show("NODE_NAME") => <Chef::Node @name="NODE_NAME" ...>

## SEARCH FOR NODES ##
  You can search for nodes using the +search+ or +find+ subcommands:

  nodes.find(:name => "app*") #=> [<Chef::Node @name="app1.example.com" ...>, ...]

  Similarly to +all+, you can pass a code block to limit or transform the
  information returned:

  nodes.find(:name => "app#") { |node| node.ec2 } 

## BULK EDIT NODES ##
          **BE CAREFUL, THIS OPERATION IS DESTRUCTIVE**

  Bulk edit nodes by passing a code block to the +transform+ or +bulk_edit+
  subcommand. The block will be applied to each matching node, and then the node
  will be saved. If the block returns +nil+ or +false+, that node will be 
  skipped.

  nodes.transform do |node|
    if node.fqdn =~ /.*\\.preprod\\.example\\.com/
      node.set[:environment] = "preprod"
    end
  end

  This will assign the attribute to every node with a FQDN matching the regex.
E
  subcommands :all        => "list all nodes",
              :show       => "load a node by name",
              :search     => "search for nodes",
              :transform  => "edit all nodes via a code block and save them"
  def nodes
    @nodes ||= Shef::ModelWrapper.new(Chef::Node)
  end

  desc "Find and edit roles via the API"
  explain(<<-E)
## SUMMARY ##
  +roles+ allows you to query and edit roles on your Chef server.

## SUBCOMMANDS ##
  * all       (list)
  * show      (load)
  * search    (find)
  * transform (bulk_edit)

## SEE ALSO ##
  See the help for +nodes+ for more information about the subcommands.
E
  subcommands :all        => "list all roles",
              :show       => "load a role by name",
              :search     => "search for roles",
              :transform  => "edit all roles via a code block and save them"
  def roles
    @roles ||= Shef::ModelWrapper.new(Chef::Role)
  end

  desc "Find and edit +databag_name+ via the api"
  explain(<<-E)
## SUMMARY ##
  +databags(DATABAG_NAME)+ allows you to query and edit data bag items on your
  Chef server. Unlike other commands for working with data on the server, 
  +databags+ requires the databag name as an argument, for example:
databags(:users).all

## SUBCOMMANDS ##
  * all       (list)
  * show      (load)
  * search    (find)
  * transform (bulk_edit)

## SEE ALSO ##
  See the help for +nodes+ for more information about the subcommands.

E
  subcommands :all        => "list all items in the data bag",
              :show       => "load a data bag item by id",
              :search     => "search for items in the data bag",
              :transform  => "edit all items via a code block and save them"
  def databags(databag_name)
    @named_databags_wrappers ||= {}
    @named_databags_wrappers[databag_name] ||= Shef::NamedDataBagWrapper.new(databag_name)
  end

  desc "Find and edit environments via the API"
  explain(<<-E)
## SUMMARY ##
  +environments+ allows you to query and edit environments on your Chef server.

## SUBCOMMANDS ##
  * all       (list)
  * show      (load)
  * search    (find)
  * transform (bulk_edit)

## SEE ALSO ##
  See the help for +nodes+ for more information about the subcommands.
E
  subcommands :all        => "list all environments",
              :show       => "load an environment by name",
              :search     => "search for environments",
              :transform  => "edit all environments via a code block and save them"
  def environments
    @environments ||= Shef::ModelWrapper.new(Chef::Environment)
  end

  desc "A REST Client configured to authenticate with the API"
  def api
    @rest = Shef::ShefREST.new(Chef::Config[:chef_server_url])
  end

end
RecipeUIExtensions =
Proc.new do
  alias :original_resources :resources

  desc "list all the resources on the current recipe"
  def resources(*args)
    if args.empty?
      pp run_context.resource_collection.instance_variable_get(:@resources_by_name).keys
    else
      pp resources = original_resources(*args)
      resources
    end
  end
end

Class Method Summary collapse

Class Method Details

.extend_context_node(node_obj) ⇒ Object



564
565
566
# File 'lib/chef/shef/ext.rb', line 564

def self.extend_context_node(node_obj)
  node_obj.instance_eval(&ObjectUIExtensions)
end

.extend_context_object(obj) ⇒ Object



556
557
558
559
560
561
562
# File 'lib/chef/shef/ext.rb', line 556

def self.extend_context_object(obj)
  obj.instance_eval(&ObjectUIExtensions)
  obj.instance_eval(&MainContextExtensions)
  obj.instance_eval(&RESTApiExtensions)
  obj.extend(FileUtils)
  obj.extend(Chef::Mixin::Language)
end

.extend_context_recipe(recipe_obj) ⇒ Object



568
569
570
571
# File 'lib/chef/shef/ext.rb', line 568

def self.extend_context_recipe(recipe_obj)
  recipe_obj.instance_eval(&ObjectUIExtensions)
  recipe_obj.instance_eval(&RecipeUIExtensions)
end