mbailey-chef 0.9.12.1

data/distro/redhat/etc/logrotate.d/chef-server-webui

@@ -0,0 +1,8 @@
+/var/log/chef/server-webui.log {
+  rotate 12
+  weekly
+  compress
+  postrotate
+    /etc/init.d/chef-server-webui condrestart >/dev/null || :
+  endscript
+}

data/distro/redhat/etc/logrotate.d/chef-solr

@@ -0,0 +1,8 @@
+/var/log/chef/solr.log {
+  rotate 12
+  weekly
+  compress
+  postrotate
+    /etc/init.d/chef-solr condrestart >/dev/null || :
+  endscript
+}

data/distro/redhat/etc/logrotate.d/chef-solr-indexer

@@ -0,0 +1,8 @@
+/var/log/chef/solr-indexer.log {
+  rotate 12
+  weekly
+  compress
+  postrotate
+    /etc/init.d/chef-solr-indexer condrestart >/dev/null || :
+  endscript
+}

data/distro/redhat/etc/sysconfig/chef-client

@@ -0,0 +1,15 @@
+# Configuration file for the chef-client service
+
+#CONFIG=/etc/chef/client.rb
+#PIDFILE=/var/run/chef/client.pid
+#LOCKFILE=/var/lock/subsys/chef-client
+#LOGFILE=/var/log/chef/client.log
+# Sleep interval between runs.
+# This value is in seconds.
+#INTERVAL=1800
+# Maximum amount of random delay before starting a run. Prevents every client
+# from contacting the server at the exact same time.
+# This value is in seconds.
+#SPLAY=20
+# Any additional chef-client options.
+#OPTIONS=

data/distro/redhat/etc/sysconfig/chef-server

@@ -0,0 +1,14 @@
+# Configuration file for the chef-server service
+
+#CONFIG=/etc/chef/server.rb
+#PIDFILE=/var/run/chef/server.pid
+#LOCKFILE=/var/lock/subsys/chef-server
+#LOGFILE=/var/log/chef/server.log
+#PORT=4000
+#ENVIRONMENT=production
+#ADAPTER=thin
+#CHILDPIDFILES=/var/run/chef/server.%s.pid
+#SERVER_USER=chef
+#SERVER_GROUP=chef
+# Any additional chef-server options.
+#OPTIONS=

data/distro/redhat/etc/sysconfig/chef-server-webui

@@ -0,0 +1,14 @@
+# Configuration file for the chef-server-webui service
+
+#CONFIG=/etc/chef/webui.rb
+#PIDFILE=/var/run/chef/server-webui.pid
+#LOCKFILE=/var/lock/subsys/chef-server-webui
+#LOGFILE=/var/log/chef/server-webui.log
+#PORT=4040
+#ENVIRONMENT=production
+#ADAPTER=thin
+#CHILDPIDFILES=/var/run/chef/server-webui.%s.pid
+#SERVER_USER=chef
+#SERVER_GROUP=chef
+# Any additional chef-server-webui options.
+#OPTIONS=

data/distro/redhat/etc/sysconfig/chef-solr

@@ -0,0 +1,8 @@
+# Configuration file for the chef-solr service
+
+#CONFIG=/etc/chef/solr.rb
+#PIDFILE=/var/run/chef/solr.pid
+#LOCKFILE=/var/lock/subsys/chef-solr
+#LOGFILE=/var/log/chef/solr.log
+# Options for Java. Need to start with -j followed by options.
+#OPTIONS="-j "

data/distro/redhat/etc/sysconfig/chef-solr-indexer

@@ -0,0 +1,7 @@
+# Configuration file for the chef-solr-indexer service
+
+#PIDFILE=/var/run/chef/solr-indexer.pid
+#CONFIG=/etc/chef/solr-indexer.rb
+#LOGFILE=/var/log/chef/solr-indexer.log
+# Any additional chef-solr-indexer options.
+#OPTIONS=

data/lib/chef/api_client.rb

@@ -0,0 +1,264 @@
+#
+# Author:: Adam Jacob (<adam@opscode.com>)
+# Author:: Nuo Yan (<nuo@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/config'
+require 'chef/mixin/params_validate'
+require 'chef/mixin/from_file'
+require 'chef/couchdb'
+require 'chef/certificate'
+require 'chef/index_queue'
+require 'extlib'
+require 'json'
+
+class Chef
+  class ApiClient 
+    
+    include Chef::Mixin::FromFile
+    include Chef::Mixin::ParamsValidate
+    include Chef::IndexQueue::Indexable
+    
+    
+    DESIGN_DOCUMENT = {
+      "version" => 1,
+      "language" => "javascript",
+      "views" => {
+        "all" => {
+          "map" => <<-EOJS
+          function(doc) { 
+            if (doc.chef_type == "client") {
+              emit(doc.name, doc);
+            }
+          }
+          EOJS
+        },
+        "all_id" => {
+          "map" => <<-EOJS
+          function(doc) { 
+            if (doc.chef_type == "client") {
+              emit(doc.name, doc.name);
+            }
+          }
+          EOJS
+        }
+      }
+    }
+
+    attr_accessor :couchdb_rev, :couchdb_id, :couchdb
+    
+    # Create a new Chef::ApiClient object.
+    def initialize(couchdb=nil)
+      @name = '' 
+      @public_key = nil
+      @private_key = nil
+      @couchdb_rev = nil
+      @couchdb_id = nil
+      @admin = false
+      @couchdb = (couchdb || Chef::CouchDB.new)
+    end
+
+    # Gets or sets the client name.
+    #
+    # @params [Optional String] The name must be alpha-numeric plus - and _.
+    # @return [String] The current value of the name.
+    def name(arg=nil) 
+      set_or_return(
+        :name,
+        arg,
+        :regex => /^[\-[:alnum:]_\.]+$/
+      )
+    end
+
+    # Gets or sets whether this client is an admin. 
+    #
+    # @params [Optional True/False] Should be true or false - default is false.
+    # @return [True/False] The current value
+    def admin(arg=nil)
+      set_or_return(
+        :admin,
+        arg,
+        :kind_of => [ TrueClass, FalseClass ]
+      )
+    end
+
+    # Gets or sets the public key.
+    # 
+    # @params [Optional String] The string representation of the public key. 
+    # @return [String] The current value.
+    def public_key(arg=nil) 
+      set_or_return(
+        :public_key,
+        arg,
+        :kind_of => String
+      )
+    end
+
+    # Gets or sets the private key.
+    # 
+    # @params [Optional String] The string representation of the private key.
+    # @return [String] The current value.
+    def private_key(arg=nil) 
+      set_or_return(
+        :private_key,
+        arg,
+        :kind_of => String
+      )
+    end
+
+    # Creates a new public/private key pair, and populates the public_key and
+    # private_key attributes.
+    # 
+    # @return [True]
+    def create_keys
+      results = Chef::Certificate.gen_keypair(self.name)
+      self.public_key(results[0].to_s)
+      self.private_key(results[1].to_s)
+      true
+    end
+
+    # The hash representation of the object.  Includes the name and public_key,
+    # but never the private key.
+    # 
+    # @return [Hash]
+    def to_hash
+      result = {
+        "name" => @name,
+        "public_key" => @public_key,
+        "admin" => @admin,
+        'json_class' => self.class.name,
+        "chef_type" => "client"
+      }
+      result["_rev"] = @couchdb_rev if @couchdb_rev
+      result
+    end
+
+    # The JSON representation of the object.
+    # 
+    # @return [String] the JSON string.
+    def to_json(*a)
+      to_hash.to_json(*a)
+    end
+    
+    def self.json_create(o)
+      client = Chef::ApiClient.new
+      client.name(o["name"] || o["clientname"])
+      client.public_key(o["public_key"])
+      client.admin(o["admin"])
+      client.couchdb_rev = o["_rev"]
+      client.couchdb_id = o["_id"]
+      client.index_id = client.couchdb_id
+      client
+    end
+    
+    # List all the Chef::ApiClient objects in the CouchDB.  If inflate is set
+    # to true, you will get the full list of all ApiClients, fully inflated.
+    def self.cdb_list(inflate=false, couchdb=nil)
+      rs = (couchdb || Chef::CouchDB.new).list("clients", inflate)
+      lookup = (inflate ? "value" : "key")
+      rs["rows"].collect { |r| r[lookup] }
+    end
+    
+    def self.list(inflate=false)
+      if inflate
+        response = Hash.new
+        Chef::Search::Query.new.search(:client) do |n|
+          n = self.json_create(n) if n.instance_of?(Hash)
+          response[n.name] = n
+        end
+        response
+      else
+        Chef::REST.new(Chef::Config[:chef_server_url]).get_rest("clients")
+      end
+    end
+    
+    # Load a client by name from CouchDB
+    # 
+    # @params [String] The name of the client to load
+    # @return [Chef::ApiClient] The resulting Chef::ApiClient object
+    def self.cdb_load(name, couchdb=nil)
+      (couchdb || Chef::CouchDB.new).load("client", name)
+    end
+    
+    # Load a client by name via the API
+    def self.load(name)
+      response = Chef::REST.new(Chef::Config[:chef_server_url]).get_rest("clients/#{name}")
+      if response.kind_of?(Chef::ApiClient)
+        response
+      else
+        client = Chef::ApiClient.new
+        client.name(response['clientname'])
+        client
+      end
+    end
+    
+    # Remove this client from the CouchDB
+    #
+    # @params [String] The name of the client to delete
+    # @return [Chef::ApiClient] The last version of the object
+    def cdb_destroy
+      @couchdb.delete("client", @name, @couchdb_rev)
+    end
+    
+    # Remove this client via the REST API
+    def destroy
+      Chef::REST.new(Chef::Config[:chef_server_url]).delete_rest("clients/#{@name}")
+    end
+    
+    # Save this client to the CouchDB
+    def cdb_save
+      @couchdb_rev = @couchdb.store("client", @name, self)["rev"]
+    end
+    
+    # Save this client via the REST API, returns a hash including the private key
+    def save(new_key=false, validation=false)
+      if validation
+        r = Chef::REST.new(Chef::Config[:chef_server_url], Chef::Config[:validation_client_name], Chef::Config[:validation_key])
+      else
+        r = Chef::REST.new(Chef::Config[:chef_server_url])
+      end
+      # First, try and create a new registration
+      begin
+        r.post_rest("clients", {:name => self.name, :admin => self.admin })
+      rescue Net::HTTPServerException => e
+        # If that fails, go ahead and try and update it
+        if e.response.code == "409"
+          r.put_rest("clients/#{name}", { :name => self.name, :admin => self.admin, :private_key => new_key }) 
+        else
+          raise e
+        end
+      end
+    end 
+    
+    # Create the client via the REST API
+    def create
+      Chef::REST.new(Chef::Config[:chef_server_url]).post_rest("clients", self)
+    end
+    
+    # Set up our CouchDB design document
+    def self.create_design_document(couchdb=nil)
+      (couchdb ||= Chef::CouchDB.new).create_design_document("clients", DESIGN_DOCUMENT)
+    end
+    
+    # As a string
+    def to_s
+      "client[#{@name}]"
+    end
+
+  end
+end
+

data/lib/chef/application/agent.rb

@@ -0,0 +1,18 @@
+#
+# Author:: AJ Christensen (<aj@opscode.comz>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'chef/application'

data/lib/chef/application/client.rb

@@ -0,0 +1,246 @@
+#
+# Author:: AJ Christensen (<aj@opscode.com)
+# Author:: Christopher Brown (<cb@opscode.com>)
+# Copyright:: Copyright (c) 2008 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'chef/application'
+require 'chef/client'
+require 'chef/config'
+require 'chef/daemon'
+require 'chef/log'
+require 'chef/rest'
+
+
+class Chef::Application::Client < Chef::Application
+  
+  option :config_file, 
+    :short => "-c CONFIG",
+    :long  => "--config CONFIG",
+    :default => "/etc/chef/client.rb",
+    :description => "The configuration file to use"
+
+  option :log_level, 
+    :short        => "-l LEVEL",
+    :long         => "--log_level LEVEL",
+    :description  => "Set the log level (debug, info, warn, error, fatal)",
+    :proc         => lambda { |l| l.to_sym }
+
+  option :log_location,
+    :short        => "-L LOGLOCATION",
+    :long         => "--logfile LOGLOCATION",
+    :description  => "Set the log file location, defaults to STDOUT - recommended for daemonizing",
+    :proc         => nil
+
+  option :verbose_logging,
+    :short        => "-V",
+    :long         => "--verbose",
+    :description  => "Ensures logging goes to STDOUT as well as to other configured log location(s).",
+    :proc         => lambda { |p| true }
+
+  option :help,
+    :short        => "-h",
+    :long         => "--help",
+    :description  => "Show this message",
+    :on           => :tail,
+    :boolean      => true,
+    :show_options => true,
+    :exit         => 0
+    
+  option :user,
+    :short => "-u USER",
+    :long => "--user USER",
+    :description => "User to set privilege to",
+    :proc => nil
+
+  option :group,
+    :short => "-g GROUP",
+    :long => "--group GROUP",
+    :description => "Group to set privilege to",
+    :proc => nil
+
+  option :daemonize,
+    :short => "-d",
+    :long => "--daemonize",
+    :description => "Daemonize the process",
+    :proc => lambda { |p| true }
+
+  option :pid_file,
+    :short        => "-P PID_FILE",
+    :long         => "--pid PIDFILE",
+    :description  => "Set the PID file location, defaults to /tmp/chef-client.pid",
+    :proc         => nil
+
+  option :interval,
+    :short => "-i SECONDS",
+    :long => "--interval SECONDS",
+    :description => "Run chef-client periodically, in seconds",
+    :proc => lambda { |s| s.to_i }
+
+  option :once,
+    :long => "--short",
+    :description => "Cancel any interval or splay options, run chef once and exit",
+    :boolean => true
+
+  option :json_attribs,
+    :short => "-j JSON_ATTRIBS",
+    :long => "--json-attributes JSON_ATTRIBS",
+    :description => "Load attributes from a JSON file or URL",
+    :proc => nil
+
+  option :node_name,
+    :short => "-N NODE_NAME",
+    :long => "--node-name NODE_NAME",
+    :description => "The node name for this client",
+    :proc => nil
+
+  option :splay,
+    :short => "-s SECONDS",
+    :long => "--splay SECONDS",
+    :description => "The splay time for running at intervals, in seconds",
+    :proc => lambda { |s| s.to_i }
+
+  option :chef_server_url,
+    :short => "-S CHEFSERVERURL",
+    :long => "--server CHEFSERVERURL",
+    :description => "The chef server URL",
+    :proc => nil
+
+  option :validation_key,
+    :short        => "-K KEY_FILE",
+    :long         => "--validation_key KEY_FILE",
+    :description  => "Set the validation key file location, used for registering new clients",
+    :proc         => nil
+
+  option :client_key,
+    :short        => "-k KEY_FILE",
+    :long         => "--client_key KEY_FILE",
+    :description  => "Set the client key file location",
+    :proc         => nil
+
+  option :version,
+    :short        => "-v",
+    :long         => "--version",
+    :description  => "Show chef version",
+    :boolean      => true,
+    :proc         => lambda {|v| puts "Chef: #{::Chef::VERSION}"},
+    :exit         => 0
+
+  def initialize
+    super
+
+    @chef_client = nil
+    @chef_client_json = nil
+  end
+  
+  # Reconfigure the chef client
+  # Re-open the JSON attributes and load them into the node
+  def reconfigure 
+    super 
+
+    Chef::Config[:chef_server_url] = config[:chef_server_url] if config.has_key? :chef_server_url
+   
+    if Chef::Config[:daemonize]
+      Chef::Config[:interval] ||= 1800
+    end
+
+    if Chef::Config[:once]
+      Chef::Config[:interval] = nil
+      Chef::Config[:splay] = nil
+    end
+
+    if Chef::Config[:json_attribs]
+      begin
+        json_io = case Chef::Config[:json_attribs]
+                  when /^(http|https):\/\//
+                    @rest = Chef::REST.new(Chef::Config[:json_attribs], nil, nil)
+                    @rest.get_rest(Chef::Config[:json_attribs], true).open
+                  else
+                    open(Chef::Config[:json_attribs])
+                  end
+      rescue SocketError => error
+        Chef::Application.fatal!("I cannot connect to #{Chef::Config[:json_attribs]}", 2)
+      rescue Errno::ENOENT => error
+        Chef::Application.fatal!("I cannot find #{Chef::Config[:json_attribs]}", 2)
+      rescue Errno::EACCES => error
+        Chef::Application.fatal!("Permissions are incorrect on #{Chef::Config[:json_attribs]}. Please chmod a+r #{Chef::Config[:json_attribs]}", 2)
+      rescue Exception => error
+        Chef::Application.fatal!("Got an unexpected error reading #{Chef::Config[:json_attribs]}: #{error.message}", 2)
+      end
+
+      begin
+        @chef_client_json = JSON.parse(json_io.read)
+        json_io.close unless json_io.closed?
+      rescue JSON::ParserError => error
+        Chef::Application.fatal!("Could not parse the provided JSON file (#{Chef::Config[:json_attribs]})!: " + error.message, 2)
+      end
+    end
+  end
+
+  def configure_logging
+    super
+    Chef::Log.verbose = Chef::Config[:verbose_logging]
+    Mixlib::Authentication::Log.logger = Ohai::Log.logger = Chef::Log.logger
+  end
+  
+  def setup_application
+    Chef::Daemon.change_privilege
+  end
+  
+  # Run the chef client, optionally daemonizing or looping at intervals.
+  def run_application
+    if Chef::Config[:version]
+      puts "Chef version: #{::Chef::VERSION}"
+    end
+
+    if Chef::Config[:daemonize]
+      Chef::Daemon.daemonize("chef-client")
+    end
+    
+    loop do
+      begin
+        if Chef::Config[:splay]
+          splay = rand Chef::Config[:splay]
+          Chef::Log.debug("Splay sleep #{splay} seconds")
+          sleep splay
+        end
+        @chef_client = Chef::Client.new(@chef_client_json)
+        @chef_client_json = nil
+
+        @chef_client.run
+        @chef_client = nil
+        if Chef::Config[:interval]
+          Chef::Log.debug("Sleeping for #{Chef::Config[:interval]} seconds")
+          sleep Chef::Config[:interval]
+        else
+          Chef::Application.exit! "Exiting", 0
+        end
+      rescue SystemExit => e
+        raise
+      rescue Exception => e
+        if Chef::Config[:interval]
+          Chef::Log.error("#{e.class}:#{e}\n#{e.backtrace.join("\n")}")
+          Chef::Log.error("Sleeping for #{Chef::Config[:interval]} seconds before trying again")
+          sleep Chef::Config[:interval]
+          retry
+        else
+          raise
+        end
+      ensure
+        GC.start
+      end
+    end
+  end
+end